github.com/ppphp/yayagf@v0.0.1/cmd/generate/curd/crud.go

package curd

import (
	stdlog "log"
	"os"
	"path/filepath"

	"github.com/sirupsen/logrus"
	"github.com/ppphp/yayagf/internal/log"

	"github.com/ppphp/yayagf/internal/ent"

	"github.com/ppphp/yayagf/internal/file"
	"github.com/ppphp/yayagf/pkg/cli"
)

func CommandFactory() (*cli.Command, error) {
	c := &cli.Command{
		Run: func(args []string, flags map[string]string) (int, error) {
			template := flags["t"]
			if template == "" {
				template = flags["template"]
			}
			templates := []string{}
			if template != "" {
				templates = append(templates, template)
			}
			_, debug := flags["d"]
			if !debug {
				_, debug = flags["debug"]
			}
			if debug {
				stdlog.SetFlags(stdlog.Flags() | stdlog.Llongfile)
				log.Logger.SetReportCaller(true)
				log.Logger.SetLevel(logrus.DebugLevel)
			}

			if debug {
				log.Printf("%v", args)
			}
			root, err := file.GetAppRoot()
			if err != nil {
				log.Printf("get project name failed: %v", err.Error())
				return 1, err
			}
			mod, err := file.GetMod(root)
			if err != nil {
				return 1, err
			}
			if st, err := os.Stat(filepath.Join(root, "app", "schema", "template")); err == nil {
				if st.IsDir() {
					templates = append(templates, filepath.Join(root, "app", "schema", "template"))
				}
			}
			if debug {
				log.Printf("%v", templates)
			}
			if err := ent.GenerateCRUDFiles(mod, filepath.Join(root, "app", "schema"), filepath.Join(root, "app", "crud"), templates); err != nil {
				return 1, err
			}

			return 0, nil
		},
	}
	return c, nil
}