mirror of https://github.com/rocboss/paopao-ce
alimy
4 months ago
parent
fb552babcd
commit
60206a8a22
@ -0,0 +1,116 @@
|
|||||||
|
//go:build mage
|
||||||
|
// +build mage
|
||||||
|
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"go/build"
|
||||||
|
"os"
|
||||||
|
"os/exec"
|
||||||
|
"path/filepath"
|
||||||
|
|
||||||
|
"github.com/alimy/mir/v5/core"
|
||||||
|
"github.com/alimy/mir/v5/engine"
|
||||||
|
"github.com/livekit/mageutil"
|
||||||
|
)
|
||||||
|
|
||||||
|
var Default = Generate
|
||||||
|
|
||||||
|
// regenerate mir and protobuf
|
||||||
|
func Generate() (err error) {
|
||||||
|
return anyFun(cleanAuto, Mir, Rpc, Enum, Fmt)
|
||||||
|
}
|
||||||
|
|
||||||
|
// regenerate mir RESTful API interface
|
||||||
|
func Mir() error {
|
||||||
|
fmt.Println("Generating mir")
|
||||||
|
if err := engine.Generate(
|
||||||
|
core.UseGin(),
|
||||||
|
core.Schema("mirc"),
|
||||||
|
core.SinkPath("auto"),
|
||||||
|
core.WatchCtxDone(true),
|
||||||
|
core.RunMode(core.InSerialMode),
|
||||||
|
); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return gofmt("./auto/api/...")
|
||||||
|
}
|
||||||
|
|
||||||
|
// regenerate RPC API interface
|
||||||
|
func Rpc() error {
|
||||||
|
fmt.Println("Generating gRPC source code")
|
||||||
|
cmd := exec.Command("buf", "generate", "proto")
|
||||||
|
mageutil.ConnectStd(cmd)
|
||||||
|
if err := cmd.Run(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return gofmt("./auto/rpc/...")
|
||||||
|
}
|
||||||
|
|
||||||
|
func Enum() error {
|
||||||
|
fmt.Println("Generating Enum source code")
|
||||||
|
cmd := exec.Command("go", "generate", "./internal/model/enum/...")
|
||||||
|
mageutil.ConnectStd(cmd)
|
||||||
|
if err := cmd.Run(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return gofmt("./internal/model/enum/...")
|
||||||
|
}
|
||||||
|
|
||||||
|
// run tests
|
||||||
|
func Test() error {
|
||||||
|
fmt.Println("Testing start")
|
||||||
|
cmd := exec.Command("go", "test", "-race", "./...")
|
||||||
|
connectStd(cmd)
|
||||||
|
return cmd.Run()
|
||||||
|
}
|
||||||
|
|
||||||
|
// format source code
|
||||||
|
func Fmt() error {
|
||||||
|
return gofmt("./...")
|
||||||
|
}
|
||||||
|
|
||||||
|
func gofmt(pattern string) error {
|
||||||
|
fmt.Printf("Formating source code [%s]\n", pattern)
|
||||||
|
cmd := exec.Command("go", "fmt", pattern)
|
||||||
|
mageutil.ConnectStd(cmd)
|
||||||
|
return cmd.Run()
|
||||||
|
}
|
||||||
|
|
||||||
|
func getToolPath(name string) (string, error) {
|
||||||
|
if p, err := exec.LookPath(name); err == nil {
|
||||||
|
return p, nil
|
||||||
|
}
|
||||||
|
// check under gopath
|
||||||
|
gopath := os.Getenv("GOPATH")
|
||||||
|
if gopath == "" {
|
||||||
|
gopath = build.Default.GOPATH
|
||||||
|
}
|
||||||
|
p := filepath.Join(gopath, "bin", name)
|
||||||
|
if _, err := os.Stat(p); err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
return p, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func cleanAuto() error {
|
||||||
|
fmt.Println("Delete auto directory")
|
||||||
|
cmd := exec.Command("rm", "-rf", "auto")
|
||||||
|
connectStd(cmd)
|
||||||
|
return cmd.Run()
|
||||||
|
}
|
||||||
|
|
||||||
|
func connectStd(cmd *exec.Cmd) {
|
||||||
|
cmd.Stdout = os.Stdout
|
||||||
|
cmd.Stderr = os.Stderr
|
||||||
|
}
|
||||||
|
|
||||||
|
func anyFun(fn ...func() error) (err error) {
|
||||||
|
for _, f := range fn {
|
||||||
|
if err = f(); err != nil {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
Loading…
Reference in new issue