You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
135 lines
2.5 KiB
Go
135 lines
2.5 KiB
Go
package gogen
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
"os/exec"
|
|
"path"
|
|
"path/filepath"
|
|
"strconv"
|
|
"strings"
|
|
"sync"
|
|
"time"
|
|
|
|
"zero/core/lang"
|
|
apiformat "zero/tools/goctl/api/format"
|
|
"zero/tools/goctl/api/parser"
|
|
apiutil "zero/tools/goctl/api/util"
|
|
"zero/tools/goctl/util"
|
|
|
|
"github.com/logrusorgru/aurora"
|
|
"github.com/urfave/cli"
|
|
)
|
|
|
|
const tmpFile = "%s-%d"
|
|
|
|
var tmpDir = path.Join(os.TempDir(), "goctl")
|
|
|
|
func GoCommand(c *cli.Context) error {
|
|
apiFile := c.String("api")
|
|
dir := c.String("dir")
|
|
if len(apiFile) == 0 {
|
|
return errors.New("missing -api")
|
|
}
|
|
if len(dir) == 0 {
|
|
return errors.New("missing -dir")
|
|
}
|
|
|
|
p, err := parser.NewParser(apiFile)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
api, err := p.Parse()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
lang.Must(util.MkdirIfNotExist(dir))
|
|
lang.Must(genEtc(dir, api))
|
|
lang.Must(genConfig(dir, api))
|
|
lang.Must(genMain(dir, api))
|
|
lang.Must(genServiceContext(dir, api))
|
|
lang.Must(genTypes(dir, api))
|
|
lang.Must(genHandlers(dir, api))
|
|
lang.Must(genRoutes(dir, api))
|
|
lang.Must(genLogic(dir, api))
|
|
// it does not work
|
|
format(dir)
|
|
|
|
if err := backupAndSweep(apiFile); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err = apiformat.ApiFormat(apiFile, false); err != nil {
|
|
return err
|
|
}
|
|
|
|
fmt.Println(aurora.Green("Done."))
|
|
return nil
|
|
}
|
|
|
|
func backupAndSweep(apiFile string) error {
|
|
var err error
|
|
var wg sync.WaitGroup
|
|
|
|
wg.Add(2)
|
|
_ = os.MkdirAll(tmpDir, os.ModePerm)
|
|
|
|
go func() {
|
|
_, fileName := filepath.Split(apiFile)
|
|
_, e := apiutil.Copy(apiFile, fmt.Sprintf(path.Join(tmpDir, tmpFile), fileName, time.Now().Unix()))
|
|
if e != nil {
|
|
err = e
|
|
}
|
|
wg.Done()
|
|
}()
|
|
go func() {
|
|
if e := sweep(); e != nil {
|
|
err = e
|
|
}
|
|
wg.Done()
|
|
}()
|
|
wg.Wait()
|
|
|
|
return err
|
|
}
|
|
|
|
func format(dir string) {
|
|
cmd := exec.Command("go", "fmt", "./"+dir+"...")
|
|
_, err := cmd.CombinedOutput()
|
|
if err != nil {
|
|
print(err.Error())
|
|
}
|
|
}
|
|
|
|
func sweep() error {
|
|
keepTime := time.Now().AddDate(0, 0, -7)
|
|
return filepath.Walk(tmpDir, func(fpath string, info os.FileInfo, err error) error {
|
|
if info.IsDir() {
|
|
return nil
|
|
}
|
|
|
|
pos := strings.LastIndexByte(info.Name(), '-')
|
|
if pos > 0 {
|
|
timestamp := info.Name()[pos+1:]
|
|
seconds, err := strconv.ParseInt(timestamp, 10, 64)
|
|
if err != nil {
|
|
// print error and ignore
|
|
fmt.Println(aurora.Red(fmt.Sprintf("sweep ignored file: %s", fpath)))
|
|
return nil
|
|
}
|
|
|
|
tm := time.Unix(seconds, 0)
|
|
if tm.Before(keepTime) {
|
|
if err := os.Remove(fpath); err != nil {
|
|
fmt.Println(aurora.Red(fmt.Sprintf("failed to remove file: %s", fpath)))
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
})
|
|
}
|