beego pro init

internal/app/module/beegopro/schema.go

@@ -0,0 +1,151 @@
+package beegopro
+
+import (
+	"fmt"
+	"github.com/beego/bee/internal/pkg/command"
+	"github.com/beego/bee/internal/pkg/system"
+	beeLogger "github.com/beego/bee/logger"
+	"github.com/beego/bee/utils"
+	"github.com/flosch/pongo2"
+	"github.com/smartwalle/pongo2render"
+	"path/filepath"
+	"strings"
+	"sync"
+)
+
+// store all data
+type Container struct {
+	BeegoProFile     string                 // beego pro toml
+	TimestampFile    string                 // store ts file
+	GoModFile        string                 // go mod file
+	Option           Option                 // user option
+	Tmpl             Tmpl                   // remote tmpl
+	CurPath          string                 // user current path
+	EnableModules    map[string]interface{} // beego pro provider a collection of module
+	FunctionOnce     map[string]sync.Once   // exec function once
+	Timestamp        Timestamp
+	GenerateTime     string
+	GenerateTimeUnix int64
+}
+
+// user option
+type Option struct {
+	Debug          bool                    `json:"debug"`
+	ContextDebug   bool                    `json:"contextDebug"`
+	Dsn            string                  `json:"dsn"`
+	Driver         string                  `json:"driver"`
+	ProType        string                  `json:"proType"`
+	ApiPrefix      string                  `json:"apiPrefix"`
+	EnableModule   []string                `json:"enableModule"`
+	Models         map[string]ModelContent `json:"models"`
+	GitRemotePath  string                  `json:"gitRemotePath"`
+	Branch         string                  `json:"branch"`
+	GitLocalPath   string                  `json:"gitLocalPath"`
+	EnableFormat   bool                    `json:"enableFormat"`
+	SourceGen      string                  `json:"sourceGen"`
+	EnableGitPull  bool                    `json:"enbaleGitPull"`
+	Path           map[string]string       `json:"path"`
+	EnableGomod    bool                    `json:"enableGomod"`
+	RefreshGitTime int64                   `json:"refreshGitTime"`
+	Extend         map[string]string       `json:"extend"` // extend user data
+}
+
+type BeegoPro struct {
+	FilePath string
+	CurPath  string
+	Option   Option
+}
+
+// tmpl option
+type Tmpl struct {
+	RenderPath string `toml:"renderPath"`
+	Descriptor []Descriptor
+}
+
+type Descriptor struct {
+	Module  string `toml:"module"`
+	SrcName string `toml:"srcName"`
+	DstPath string `toml:"dstPath"`
+	Once    bool   `toml:"once"`
+	Script  string `toml:"script"`
+}
+
+func (descriptor Descriptor) Parse(modelName string, paths map[string]string) (newDescriptor Descriptor, ctx pongo2.Context) {
+	var (
+		err             error
+		relativeDstPath string
+		absFile         string
+		relPath         string
+	)
+
+	newDescriptor = descriptor
+	render := pongo2render.NewRender("")
+	ctx = make(pongo2.Context, 0)
+	for key, value := range paths {
+		absFile, err = filepath.Abs(value)
+		if err != nil {
+			beeLogger.Log.Fatalf("absolute path error %s from key %s and value %s", err, key, value)
+		}
+		relPath, err = filepath.Rel(system.CurrentDir, absFile)
+		if err != nil {
+			beeLogger.Log.Fatalf("Could not get the relative path: %s", err)
+		}
+		// user input path
+		ctx["path"+utils.CamelCase(key)] = value
+		// relativePath
+		ctx["pathRel"+utils.CamelCase(key)] = relPath
+	}
+	ctx["modelName"] = lowerFirst(utils.CamelString(modelName))
+	relativeDstPath, err = render.TemplateFromString(descriptor.DstPath).Execute(ctx)
+	if err != nil {
+		beeLogger.Log.Fatalf("beego tmpl exec error, err: %s", err)
+		return
+	}
+
+	newDescriptor.DstPath, err = filepath.Abs(relativeDstPath)
+	if err != nil {
+		beeLogger.Log.Fatalf("absolute path error %s from flush file %s", err, relativeDstPath)
+	}
+
+	newDescriptor.Script, err = render.TemplateFromString(descriptor.Script).Execute(ctx)
+	if err != nil {
+		beeLogger.Log.Fatalf("parse script %s, error %s", descriptor.Script, err)
+	}
+	return
+}
+
+func (descriptor Descriptor) IsExistScript() bool {
+	if descriptor.Script != "" {
+		return true
+	}
+	return false
+}
+
+func (d Descriptor) ExecScript(path string) (err error) {
+	arr := strings.Split(d.Script, " ")
+	if len(arr) == 0 {
+		return
+	}
+	fmt.Println("path------>", path)
+
+	stdout, stderr, err := command.ExecCmdDir(path, arr[0], arr[1:]...)
+	if err != nil {
+		return concatenateError(err, stderr)
+	}
+
+	beeLogger.Log.Info(stdout)
+
+	return nil
+}
+
+type Timestamp struct {
+	GitCacheLastRefresh int64 `toml:"gitCacheLastRefresh"`
+	Generate            int64 `toml:"generate"`
+}
+
+func concatenateError(err error, stderr string) error {
+	if len(stderr) == 0 {
+		return err
+	}
+	return fmt.Errorf("%v: %s", err, stderr)
+}