bee/internal/app/module/beegopro/util.go

package beegopro

import (
	"errors"
	"fmt"
	"github.com/beego/bee/internal/pkg/utils"
	beeLogger "github.com/beego/bee/logger"
	"go/format"
	"io/ioutil"
	"os"
	"path"
	"path/filepath"
	"strings"
	"time"
)

// write to file
func (c *RenderFile) write(filename string, buf string) (err error) {
	if utils.IsExist(filename) && !isNeedOverwrite(filename) {
		return
	}

	filePath := filepath.Dir(filename)
	err = createPath(filePath)
	if err != nil {
		err = errors.New("write create path " + err.Error())
		return
	}

	filePathBak := filePath + "/bak"
	err = createPath(filePathBak)
	if err != nil {
		err = errors.New("write create path bak " + err.Error())
		return
	}

	name := path.Base(filename)

	if utils.IsExist(filename) {
		bakName := fmt.Sprintf("%s/%s.%s.bak", filePathBak, filepath.Base(name), time.Now().Format("2006.01.02.15.04.05"))
		beeLogger.Log.Infof("bak file '%s'", bakName)
		if err := os.Rename(filename, bakName); err != nil {
			err = errors.New("file is bak error, path is " + bakName)
			return err
		}
	}

	file, err := os.Create(filename)
	defer func() {
		err = file.Close()
		if err != nil {
			beeLogger.Log.Fatalf("file close error, err %s", err)
		}
	}()
	if err != nil {
		err = errors.New("write create file " + err.Error())
		return
	}

	output := []byte(buf)

	if c.Option.EnableFormat && filepath.Ext(filename) == ".go" {
		// format code
		var bts []byte
		bts, err = format.Source([]byte(buf))
		if err != nil {
			err = errors.New("format buf error " + err.Error())
			return
		}
		output = bts
	}

	err = ioutil.WriteFile(filename, output, 0644)
	if err != nil {
		err = errors.New("write write file " + err.Error())
		return
	}
	return
}

func isNeedOverwrite(fileName string) (flag bool) {
	seg := "//"
	ext := filepath.Ext(fileName)
	if ext == ".sql" {
		seg = "--"
	}

	f, err := os.Open(fileName)
	if err != nil {
		return
	}
	defer f.Close()
	overwrite := ""
	var contentByte []byte
	contentByte, err = ioutil.ReadAll(f)
	if err != nil {
		return
	}
	for _, s := range strings.Split(string(contentByte), "\n") {
		s = strings.TrimSpace(strings.TrimPrefix(s, seg))
		if strings.HasPrefix(s, "@BeeOverwrite") {
			overwrite = strings.TrimSpace(s[len("@BeeOverwrite"):])
		}
	}
	if strings.ToLower(overwrite) == "yes" {
		flag = true
		return
	}
	return
}

// createPath 调用os.MkdirAll递归创建文件夹
func createPath(filePath string) error {
	if !utils.IsExist(filePath) {
		err := os.MkdirAll(filePath, os.ModePerm)
		return err
	}
	return nil
}

func getPackagePath() (packagePath string) {
	f, err := os.Open("go.mod")
	if err != nil {
		return
	}
	defer f.Close()
	var contentByte []byte
	contentByte, err = ioutil.ReadAll(f)
	if err != nil {
		return
	}
	for _, s := range strings.Split(string(contentByte), "\n") {
		packagePath = strings.TrimSpace(strings.TrimPrefix(s, "module"))
		return
	}
	return
}

func getModelType(orm string) (inputType, goType, mysqlType, tag string) {
	kv := strings.SplitN(orm, ",", 2)
	inputType = kv[0]
	switch inputType {
	case "string":
		goType = "string"
		tag = "size(255)"
		// todo use orm data
		mysqlType = "varchar(255) NOT NULL"
	case "text":
		goType = "string"
		tag = "type(longtext)"
		mysqlType = "longtext  NOT NULL"
	case "auto":
		goType = "int"
		tag = "auto"
		mysqlType = "int(11) NOT NULL AUTO_INCREMENT"
	case "pk":
		goType = "int"
		tag = "pk"
		mysqlType = "int(11) NOT NULL"
	case "datetime":
		goType = "time.Time"
		tag = "type(datetime)"
		mysqlType = "datetime NOT NULL"
	case "int", "int8", "int16", "int32", "int64":
		fallthrough
	case "uint", "uint8", "uint16", "uint32", "uint64":
		goType = inputType
		tag = ""
		mysqlType = "int(11) DEFAULT NULL"
	case "bool":
		goType = inputType
		tag = ""
		mysqlType = "int(11) DEFAULT NULL"
	case "float32", "float64":
		goType = inputType
		tag = ""
		mysqlType = "float NOT NULL"
	case "float":
		goType = "float64"
		tag = ""
		mysqlType = "float NOT NULL"
	default:
		beeLogger.Log.Fatalf("not support type: %s", inputType)
	}
	// user set orm tag
	if len(kv) == 2 {
		tag = kv[1]
	}
	return
}