main.go

package main

import (
	"bytes"
	"flag"
	"fmt"
	"log"
	"os/exec"

	"github.com/ibbd-dev/go-db-models/models"
)

var debug = false

func main() {
	log.SetFlags(0)

	// 命令参数设置
	database := flag.String("d", "", "")
	charset := flag.String("c", "utf8", "")
	username := flag.String("u", "", "")
	password := flag.String("p", "", "")
	packagename := flag.String("n", "", "")
	host := flag.String("h", "127.0.0.1", "")
	port := flag.Int("r", 3306, "")
	version := flag.Bool("v", false, "")
	help := flag.Bool("e", false, "")
	flag.StringVar(database, "database", "", "")
	flag.StringVar(charset, "charset", "utf8", "")
	flag.StringVar(username, "username", "", "")
	flag.StringVar(password, "password", "", "")
	flag.StringVar(host, "host", "127.0.0.1", "")
	flag.StringVar(packagename, "package", "", "")
	flag.IntVar(port, "port", 3306, "")
	flag.BoolVar(version, "version", false, "")
	flag.BoolVar(help, "help", false, "")
	flag.Usage = func() { log.Println(usageText) } // call on flag error
	flag.Parse()

	if debug {
		fmt.Println("**********************************")
		fmt.Printf("database: %s\n", *database)
		fmt.Printf("charset: %s\n", *charset)
		fmt.Printf("username: %s\n", *username)
		fmt.Printf("password: %s\n", *password)
		fmt.Printf("host: %s\n", *host)
		fmt.Printf("port: %d\n", *port)
		fmt.Println("**********************************")
	}

	if *help {
		// not an error, send to stdout
		// that way people can: scaneo -h | less
		fmt.Println(usageText)
		return
	}

	if len(*packagename) == 0 {
		fmt.Println("packagename is empty! use -n or --package")
		fmt.Println(usageText)
		return
	}

	if *version {
		fmt.Println(versionText)
		return
	}
	fmt.Println("Begin ...")

	// 生成数据库的表结构
	dbConf := &models.DbConf{
		Host:     *host,
		Port:     *port,
		DbName:   *database,
		UserName: *username,
		Password: *password,
	}
	tables, err := dbConf.ShowTables()
	if err != nil {
		panic(err)
	}

	// 处理model配置文件
	jsonFile := flag.Arg(0)
	var modelsConf = &models.JsonConf{}
	if jsonFile != "" {
		modelsConf, err = models.JsonUnmarshal(jsonFile)
		if err != nil {
			panic(err)
		}
	}

	if debug {
		fmt.Println("Input Filename: ", jsonFile)
		fmt.Print("Models config: ")
		fmt.Println(modelsConf)
	}

	// 解释数据库的表结构
	ptables, err := models.ParseTablesStruct(tables, *packagename, modelsConf)
	if err != nil {
		panic(err)
	}
	if debug {
		fmt.Print("Tables after parse: ")
		fmt.Println(ptables)
	}

	// 格式化生成的代码
	runFmt()
	fmt.Println("\nAll is ok!")
}

func runFmt() {
	in := bytes.NewBuffer(nil)
	cmd := exec.Command("sh")
	cmd.Stdin = in
	go func() {
		in.WriteString("go fmt\n")
		in.WriteString("exit\n")
	}()

	if err := cmd.Run(); err != nil {
		panic(err)
	}
}