restructure

model.go

@@ -0,0 +1,93 @@
+package atom
+
+import (
+	"errors"
+	"log"
+
+	"github.com/glebarez/sqlite"
+	"github.com/rogeecn/atom/container"
+	"github.com/spf13/cobra"
+	"go.uber.org/dig"
+	"gorm.io/driver/mysql"
+	"gorm.io/driver/postgres"
+	"gorm.io/gen"
+	"gorm.io/gorm"
+)
+
+func WithModel(rootCmd *cobra.Command) *cobra.Command {
+	rootCmd.AddCommand(modelCmd)
+	return rootCmd
+}
+
+// MigrationInfo http service container
+type GenQueryGenerator struct {
+	dig.In
+
+	DB *gorm.DB
+}
+
+// // Dynamic SQL
+// type Querier interface {
+// 	// SELECT * FROM @@table WHERE name = @name{{if role !=""}} AND role = @role{{end}}
+// 	FilterWithNameAndRole(name, role string) ([]gen.T, error)
+// }
+
+// modelCmd represents the gen command
+var modelCmd = &cobra.Command{
+	Use:   "model",
+	Short: "gorm model&query generator",
+	Long:  `gorm model&query generator. more info, see https://gorm.io/gen`,
+	RunE: func(cmd *cobra.Command, args []string) error {
+		return container.Container.Invoke(func(gq GenQueryGenerator) error {
+			var tables []string
+
+			switch gq.DB.Dialector.Name() {
+			case mysql.Dialector{}.Name():
+				err := gq.DB.Raw("show tables").Scan(&tables).Error
+				if err != nil {
+					log.Fatal(err)
+				}
+			case postgres.Dialector{}.Name():
+				err := gq.DB.Raw("SELECT table_name FROM information_schema.tables WHERE table_schema = 'public'").Scan(&tables).Error
+				if err != nil {
+					log.Fatal(err)
+				}
+			case sqlite.DriverName:
+				err := gq.DB.Raw("SELECT name FROM sqlite_master WHERE type='table'").Scan(&tables).Error
+				if err != nil {
+					log.Fatal(err)
+				}
+			}
+
+			if len(tables) == 0 {
+				return errors.New("no tables in database, run migrate first")
+			}
+
+			g := gen.NewGenerator(gen.Config{
+				OutPath:          "database/query",
+				OutFile:          "query.gen.go",
+				ModelPkgPath:     "database/models",
+				FieldSignable:    true,
+				FieldWithTypeTag: true,
+				Mode:             gen.WithDefaultQuery | gen.WithQueryInterface,
+			})
+
+			g.UseDB(gq.DB) // reuse your gorm db
+
+			models := []interface{}{}
+			for _, table := range tables {
+				models = append(models, g.GenerateModel(table))
+			}
+
+			// Generate basic type-safe DAO API for struct `model.User` following conventions
+			g.ApplyBasic(models...)
+
+			// Generate Type Safe API with Dynamic SQL defined on Querier interface for `model.User` and `model.Company`
+			// g.ApplyInterface(func(Querier) {}, model.User{}, model.Company{})
+
+			// Generate the code
+			g.Execute()
+			return nil
+		})
+	},
+}