初步实现安装界面

cmd/sql-generate/main.go

@@ -0,0 +1,103 @@
+package main
+
+import (
+	"bytes"
+	"flag"
+	"fmt"
+	"github.com/iwind/TeaGo/Tea"
+	_ "github.com/iwind/TeaGo/bootstrap"
+	"github.com/iwind/TeaGo/logs"
+	"go/format"
+	"io/ioutil"
+	"os"
+	"path/filepath"
+	"strings"
+)
+
+func main() {
+	var dir string
+	flag.StringVar(&dir, "dir", "", "SQL dir")
+	flag.Parse()
+
+	if len(dir) == 0 {
+		fmt.Println("[ERROR]'dir' should not be empty")
+		return
+	}
+
+	sourceDir := filepath.Dir(Tea.Root)
+
+	// full
+	fullSQLFile := dir + "/full.sql"
+	_, err := os.Stat(fullSQLFile)
+	if err != nil {
+		if os.IsNotExist(err) {
+			fmt.Println("[ERROR]'full.sql' not found")
+			return
+		}
+		fmt.Println("[ERROR]checking 'full.sql' failed: " + err.Error())
+		return
+	}
+
+	matches, err := filepath.Glob(dir + "/*.sql")
+	if err != nil {
+		fmt.Println("[ERROR]" + err.Error())
+		return
+	}
+
+	versionsCode := "// generated\npackage sqls\n\nvar SQLVersions = []map[string]string{"
+
+	for _, match := range matches {
+		baseName := filepath.Base(match)
+		logs.Println("reading " + baseName + " ...")
+
+		data, err := ioutil.ReadFile(match)
+		if err != nil {
+			fmt.Println("[ERROR]" + err.Error())
+			return
+		}
+
+		version := baseName[:strings.LastIndex(baseName, ".")]
+		versionsCode += "\n" + `{ "version": "` + version + `", "sql": SQL_` + version + ` },`
+
+		code := "// generated\npackage sqls \n\n"
+		lines := bytes.Split(data, []byte{'\n'})
+		for index, line := range lines {
+			if index == 0 {
+				code += "var SQL_" + version + " = "
+			}
+			code += `"` + string(line) + `\n"`
+			if index != len(lines)-1 {
+				code += "+"
+			}
+			code += "\n"
+		}
+		code += "\n"
+
+		codeBytes, err := format.Source([]byte(code))
+		if err != nil {
+			fmt.Println("[ERROR]" + err.Error())
+			return
+		}
+		fmt.Println("writing sql_" + version + ".go ...")
+		err = ioutil.WriteFile(sourceDir+"/internal/setup/sqls/sql_"+version+".go", codeBytes, 0666)
+		if err != nil {
+			fmt.Println("[ERROR]" + err.Error())
+			return
+		}
+		fmt.Println("ok")
+	}
+
+	versionsCode += "\n}"
+	versionsCodeBytes, err := format.Source([]byte(versionsCode))
+	if err != nil {
+		fmt.Println("[ERROR]" + err.Error())
+		return
+	}
+	fmt.Println("writing sqls.go ...")
+	err = ioutil.WriteFile(sourceDir+"/internal/setup/sqls/sqls.go", versionsCodeBytes, 0666)
+	if err != nil {
+		fmt.Println("[ERROR]" + err.Error())
+		return
+	}
+	fmt.Println("ok")
+}