actions

var GenConfig = func(context *cli.Context) error {

	parseTemp, err := temp.ParseTemp(temp.Config, nil)
	if err != nil {
		return err
	}
	err = utils.GenFile("cli.yaml", parseTemp, false)
	if err != nil {
		return err
	}
	return nil
}

var GenController = func(context *cli.Context) error {
	db, _ := utils.GetDb()

	tables, err := db.Migrator().GetTables()
	if err != nil {
		return err
	}

	for _, table := range tables {

		args := struct {
			ControllerName string
			AppName        string
			CName          string
		}{
			utils.FirstUpper(table),
			viper.GetString("app_name"),
			table,
		}
		parseTemp, err := temp.ParseTemp(temp.Controller, args)
		if err != nil {
			fmt.Println(err)
			return err
		}
		err = utils.GenFile("controller/"+table+".go", parseTemp, false)
		if err != nil {
			fmt.Println(err)
			return err
		}

	}
	return nil
}

var GenDatabase = func(context *cli.Context) error {

	pConfig, err := utils.GetProjectConfig()
	if err != nil {
		fmt.Println(err.Error())
		return err
	}
	if !pConfig.UseDb {
		err = errors.New("请开启配置文件中的use_db")
		return err
	}

	start := time.Now()
	dbconfig := viper.GetStringMapString("database")

	dsn := dbconfig["username"] + ":" + dbconfig["password"] + "@tcp(" + dbconfig["host"] + ":" + dbconfig["port"] + ")/" + dbconfig["dbname"] + "?charset=" + dbconfig["charset"] + "&parseTime=True&loc=Local"
	args := struct {
		Dsn string
	}{
		dsn,
	}
	parseTemp, err := temp.ParseTemp(temp.Db, args)
	if err != nil {
		return err
	}
	err = utils.GenFile("dal/db.go", parseTemp, false)
	if err != nil {
		return err
	}

	parseDiy, err := temp.ParseTemp(temp.Diy, nil)
	if err != nil {
		return err
	}
	err = utils.GenFile("dal/diy.go", parseDiy, false)
	if err != nil {
		return err
	}

	db, _ := utils.GetDb()
	g := gen.NewGenerator(gen.Config{
		OutPath:      "dal",
		Mode:         gen.WithDefaultQuery,
		ModelPkgPath: "dal/models",
		OutFile:      "gen.go",
	})
	g.UseDB(db)
	g.ApplyBasic(g.GenerateAllTable()...)
	g.Execute()
	fmt.Println("耗时：", time.Since(start))
	return nil
}

var GenProject = func(context *cli.Context) error {
	start := time.Now()
	fmt.Println("开始生成目录结构...")

	GenDirs()

	genFiles := []fileInfo{
		{"main.go", temp.Index, false},
		{"go.mod", temp.Mod, false},
		{"router/base.go", temp.Router, true},
	}

	for _, file := range genFiles {
		args, err := utils.GetProjectConfig()
		parseTemp, err := temp.ParseTemp(file.Temp, args)
		if err != nil {
			return err
		}
		err = utils.GenFile(file.Name, parseTemp, file.IsOverlay)
		if err != nil {
			return err
		}
	}

	fmt.Println("生成目录结构结束!请执行go mod tidy")
	fmt.Println("耗时：", time.Since(start))
	return nil
}

var GenServices = func(context *cli.Context) error {

	db, _ := utils.GetDb()
	tables, err := db.Migrator().GetTables()
	if err != nil {
		fmt.Println(err)
		return err
	}

	for _, table := range tables {

		err := os.MkdirAll("services/"+table, os.ModePerm)
		if err != nil {
			fmt.Println(err)
			return err
		}

		GenQueryFile(table)
		var cols []colInfo
		db.Raw("desc " + table).Scan(&cols)
	}
	return nil
}