go-script/cmd/go-orm/main.go

package main

import (
	"encoding/json"
	"flag"
	"fmt"
	"os"
	"os/exec"
	"path/filepath"
	"strings"

	"gitea.ckfah.com/go-script/logger"
	"gitea.ckfah.com/go-script/orm"
)

const (
	generatorVersion = "1.0.0"
	successExit      = -1
)

var (
	config *DBConfig = new(DBConfig)
)

var (
	configFile       string
	help             bool
	showVersion      bool
	destDir          string
	modelPackageName string
	daoPackageName   string
	dtoPackageName   string
	tableNames       strFlags
	dbType           string
	tags             strFlags
	dbName           string
	dbHost           string
	dbPort           int
	dbUserName       string
	dbPassword       string
	dbCharset        string
)

type strFlags []string

func (i *strFlags) String() string {
	return "table names"
}

func (i *strFlags) Set(value string) error {
	*i = append(*i, value)
	return nil
}

func init() {
	flag.BoolVar(&help, "h", false, "this help")
	flag.BoolVar(&showVersion, "v", false, "generator version")
	flag.StringVar(&dbType, "db_type", "mysql", "database type, eg: -db_type=mysql")
	flag.StringVar(&dbName, "d", "", "database name, eg: -d=xorm")
	flag.StringVar(&dbHost, "host", "localhost", "database host, eg: -port=localhost")
	flag.IntVar(&dbPort, "port", 3306, "database port, eg: -port=3306")
	flag.StringVar(&dbUserName, "u", "root", "database username, eg: -u=root")
	flag.StringVar(&dbPassword, "p", "123456", "database password, eg: -p=123456")
	flag.Var(&tableNames, "t", "database table names default all tables , eg: -t=class -t=user")
	flag.StringVar(&dbCharset, "charset", "utf8", "database table names, eg: -charset=utf8")
	flag.Var(&tags, "tag", "modle tag names support add many tags,default xorm, eg: -tag=xorm -tag=json")
	flag.StringVar(&configFile, "config", "", "orm config file, the shell instrument is priority than config file")
	flag.StringVar(&destDir, "dir", "./tmp", "generated directory default ./tmp, eg: -dir=./tmp")
	flag.StringVar(&modelPackageName, "model_package", "model", "package name default model, eg:-model_package=model")
	flag.StringVar(&daoPackageName, "dao_package", "dao", "package name default dao, eg:-dao_package=dao")
	flag.StringVar(&dtoPackageName, "dto_package", "dto", "package name default dao, eg:-dto_package=dao")
}

func main() {
	initConfig("go-orm-config.json", true, false)
	// 解析输入
	initFlag()
	if configFile != "" {
		initConfig(configFile, false, true)
	}
	// 初始化属性
	config := initProperties()
	err := config.Generator()
	if err != nil {
		logger.FatalF("generate fail,err=%s", err)
	}
	fmt.Println("generate template finished")
}

func GetRootPath() string {
	curFilename := os.Args[0]
	binaryPath, err := exec.LookPath(curFilename)
	if err != nil {
		panic(err)
	}
	binaryPath, err = filepath.Abs(binaryPath)
	if err != nil {
		panic(err)
	}
	return filepath.Dir(binaryPath)
}

// 是否忽略error
func initConfig(fileName string, fail, asb bool) {
	if !asb {
		fileName = filepath.Clean(fmt.Sprintf("%s/%s", GetRootPath(), fileName))
	}
	file, err := os.Open(fileName)
	fmt.Println(fmt.Sprintf("load config %s", fileName))
	if err != nil {
		if fail {
			return
		}
		logger.FatalF("not fond config-file", err)
	}
	err = json.NewDecoder(file).Decode(config)
	if err != nil {
		bytes, _ := json.Marshal(config)
		logger.InfoF(`
your config should be as follow:
%s
`, bytes)
		logger.FatalF("decode config-file err,err=%s", err)
	}
	reloadConfig()
}

// 初始化属性
func initProperties() *orm.Config {
	config := new(orm.Config)
	config.DbType = orm.GetDbType(dbType)
	config.DbName = dbName
	config.DbUserName = dbUserName
	config.DbPassword = dbPassword
	config.DbHost = dbHost
	config.DbPort = dbPort
	config.DbCharset = dbCharset
	config.GeneratorModel = true
	config.GeneratorDao = true
	config.GeneratorDto = true
	config.DaoPackageName = daoPackageName
	config.ModelPackageName = modelPackageName
	config.DtoPackageName = dtoPackageName
	config.Tags = addTag(tags)
	config.SaveFile = destDir
	config.TableNames = tableNames
	return config
}

// 指示
func initFlag() {
	flag.Parse()
	if help {
		fmt.Println(`generator version: anthony/1.0.0
Usage: generator -host=localhost -port=3306 -d=urban_v -u=root -p=123456 -t=class -t=student -tag=xorm -dir=./tmp
Option:`)
		flag.PrintDefaults()
		os.Exit(successExit)
	}

	if showVersion {
		fmt.Printf("generator version: anthony/%s", generatorVersion)
		os.Exit(successExit)
	}
}

func addTag(tags strFlags) []string {
	str := "xorm"
	result := make([]string, 0)
	result = append(result, str)
	for _, e := range tags {
		if strings.Compare(str, e) == 0 {
			continue
		}
		result = append(result, e)
	}
	return result
}

func reloadConfig() {
	dbType = config.DbType
	tags = config.Tags
	dbName = config.DbName
	dbHost = config.DbHost
	dbPort = config.DbPort
	dbUserName = config.DbUserName
	dbPassword = config.DbPassword
	dbCharset = config.DbCharset
}

type DBConfig struct {
	DbType     string   `json:"db_type"`
	Tags       strFlags `json:"tags"`
	DbName     string   `json:"db_name"`
	DbHost     string   `json:"db_host"`
	DbPort     int      `json:"db_port"`
	DbUserName string   `json:"db_user_name"`
	DbPassword string   `json:"db_password"`
	DbCharset  string   `json:"db_charset"`
}