package model

import (
	"gitee.com/mosache/YFrame/tools/Y/model/conf"
	"gitee.com/mosache/YFrame/tools/Y/model/db"
	"gitee.com/mosache/YFrame/tools/Y/model/gen"
	"gitee.com/mosache/YFrame/tools/Y/utils"
	"gitee.com/mosache/YFrame/tools/Y/utils/cobraErr"
	"github.com/spf13/cobra"
)

var (
	databaseName   string   // 数据库名
	tables         []string // 表名
	configFilePath string   // 配置文件地址
	targetPath     string   // 生成文件路径
	packageName    string   // 包名
	tablePrefix    string   // 表名前缀
	isDoc          bool     // 是否是doc
)

var Cmd = &cobra.Command{Use: "model",
	Long: `generate model file`,
	Run: func(cmd *cobra.Command, args []string) {
		if err := modelGen(); err != nil {
			cobraErr.Panic(cmd, err.Error())
		}
	},
}

func init() {
	Cmd.Flags().StringVarP(&databaseName, "database", "d", "", "database name for generate")
	Cmd.Flags().StringArrayVarP(&tables, "table", "t", []string{}, "tables to scan for generate")
	Cmd.Flags().StringVarP(&configFilePath, "conf", "c", "../../appConfig/gomg.yaml", "conf file path")
	Cmd.Flags().StringVarP(&targetPath, "target", "", "", "generate file target path")
	Cmd.Flags().StringVarP(&packageName, "package", "p", utils.GetPwd(), "package name")
	Cmd.Flags().StringVarP(&tablePrefix, "tablePrefix", "", "", "tableNamePrefix")
	Cmd.Flags().BoolVarP(&isDoc, "isDoc", "", false, "is doc ")

}

func modelGen() error {
	if err := conf.InitConf(configFilePath); err != nil {
		return err
	}

	if err := db.InitDB(conf.Config); err != nil {
		return err
	}

	opts := make([]gen.Option, 0)

	if len(databaseName) > 0 {
		opts = append(opts, gen.WithDataBaseName(databaseName))
	}

	if len(tables) > 0 {
		opts = append(opts, gen.WithTables(tables))
	}

	if err := gen.NewGenerator(opts...).Generate(targetPath, packageName, tablePrefix, isDoc); err != nil {
		panic(err)
	}

	return nil
}