feat: support custom table primary key type and name

pkg/sql2code/parser/sqlite.go

@@ -1,24 +1,12 @@
 package parser
 
 import (
+	"fmt"
 	"strings"
 
 	"github.com/zhufuyi/sponge/pkg/ggorm"
 )
 
-var sqliteToMysqlTypeMap = map[string]string{
-	" INTEGER ":     " INT ",
-	" REAL ":        " FLOAT ",
-	" BOOLEAN ":     " TINYINT ",
-	" NUMERIC ":     " VARCHAR(255) ",
-	"AUTOINCREMENT": "auto_increment",
-	" integer ":     " INT ",
-	" real ":        " FLOAT ",
-	" boolean ":     " TINYINT ",
-	" numeric ":     " VARCHAR(255) ",
-	"autoincrement": "auto_increment",
-}
-
 // GetSqliteTableInfo get table info from sqlite
 func GetSqliteTableInfo(dbFile string, tableName string) (string, error) {
 	db, err := ggorm.InitSqlite(dbFile)
@@ -27,16 +15,102 @@ func GetSqliteTableInfo(dbFile string, tableName string) (string, error) {
 	}
 	defer closeDB(db)
 
-	var sql string
-	err = db.Raw("select sql from sqlite_master where type = ? and name = ?", "table", tableName).Scan(&sql).Error
+	var sqliteFields SqliteFields
+	sql := fmt.Sprintf("PRAGMA table_info('%s')", tableName)
+	err = db.Raw(sql).Scan(&sqliteFields).Error
 	if err != nil {
 		return "", err
 	}
 
-	for k, v := range sqliteToMysqlTypeMap {
-		sql = strings.ReplaceAll(sql, k, v)
-	}
-	sql = strings.ReplaceAll(sql, "\"", "")
-
-	return sql, nil
+	return convertToSQLBySqliteFields(tableName, sqliteFields), nil
+}
+
+// SqliteField sqlite field struct
+type SqliteField struct {
+	Cid          int    `gorm:"column:cid" json:"cid"`
+	Name         string `gorm:"column:name" json:"name"`
+	Type         string `gorm:"column:type" json:"type"`
+	Notnull      int    `gorm:"column:notnull" json:"notnull"`
+	DefaultValue string `gorm:"column:dflt_value" json:"dflt_value"`
+	Pk           int    `gorm:"column:pk" json:"pk"`
+}
+
+var sqliteToMysqlType = map[string]string{
+	"integer":       "INT",
+	"text":          "TEXT",
+	"real":          "FLOAT",
+	"datetime":      "DATETIME",
+	"blob":          "BLOB",
+	"boolean":       "TINYINT",
+	"numeric":       " VARCHAR(255)",
+	"autoincrement": "auto_increment",
+}
+
+func (field *SqliteField) getMysqlType() string {
+	sqliteType := strings.ToLower(field.Type)
+	if mysqlType, ok := sqliteToMysqlType[sqliteType]; ok {
+		if field.Name == "id" && sqliteType == "text" {
+			return "VARCHAR(50)"
+		}
+		return mysqlType
+	}
+	return "VARCHAR(100)"
+}
+
+// SqliteFields sqlite fields
+type SqliteFields []*SqliteField
+
+func (fields SqliteFields) getPrimaryField() *SqliteField {
+	var f *SqliteField
+	for _, field := range fields {
+		if field.Pk == 1 || field.Name == "id" {
+			f = field
+			return f
+		}
+	}
+	/*
+		// if no primary key, find the first xxx_id field
+		if f == nil {
+			for _, field := range fields {
+				if strings.HasSuffix(field.Name, "_id") {
+					f = field
+					f.Pk = 1
+					return f
+				}
+			}
+		}
+
+		// if no xxx_id field, find the first field
+		if f == nil {
+			for _, field := range fields {
+				f = field
+				f.Pk = 1
+				return f
+			}
+		}
+	*/
+	return f
+}
+
+func convertToSQLBySqliteFields(tableName string, fields SqliteFields) string {
+	if len(fields) == 0 {
+		return ""
+	}
+
+	fieldStr := ""
+	for _, field := range fields {
+		notnullStr := "not null"
+		if field.Notnull == 0 {
+			notnullStr = "null"
+		}
+		fieldStr += fmt.Sprintf("    `%s` %s %s comment '%s',\n", field.Name, field.getMysqlType(), notnullStr, "")
+	}
+
+	primaryField := fields.getPrimaryField()
+	if primaryField != nil {
+		fieldStr += fmt.Sprintf("    PRIMARY KEY (`%s`)\n", primaryField.Name)
+	} else {
+		fieldStr = strings.TrimSuffix(fieldStr, ",\n")
+	}
+	return fmt.Sprintf("CREATE TABLE `%s` (\n%s\n);", tableName, fieldStr)
 }