reset link goctl (#1232)

This commit is contained in:
anqiansong
2021-11-13 18:39:07 +08:00
committed by GitHub
parent 57d2f22c24
commit 0cb86c6990
7 changed files with 40 additions and 93 deletions

View File

@@ -3,7 +3,6 @@ package main
import (
"fmt"
"os"
"path/filepath"
"runtime"
"github.com/logrusorgru/aurora"
@@ -29,16 +28,10 @@ import (
rpc "github.com/tal-tech/go-zero/tools/goctl/rpc/cli"
"github.com/tal-tech/go-zero/tools/goctl/tpl"
"github.com/tal-tech/go-zero/tools/goctl/upgrade"
"github.com/tal-tech/go-zero/tools/goctl/util/console"
"github.com/tal-tech/go-zero/tools/goctl/util/env"
"github.com/urfave/cli"
pluginCtl "github.com/zeromicro/protobuf/protoc-gen-go"
)
const (
protocGenGoctl = "protoc-gen-goctl"
codeFailure = 1
)
const codeFailure = 1
var commands = []cli.Command{
{
@@ -650,46 +643,10 @@ var commands = []cli.Command{
},
}
func init() {
err := linkProtocGenGoctl()
if err != nil {
console.Error("%+v", err)
}
}
func linkProtocGenGoctl() error {
path, err := env.LookPath("goctl")
if err != nil {
return err
}
dir := filepath.Dir(path)
ext := filepath.Ext(path)
target := filepath.Join(dir, protocGenGoctl)
if len(ext) > 0 {
target = target + ext
}
_, err = os.Lstat(target)
if err != nil && !os.IsNotExist(err) {
return err
}
if os.IsNotExist(err) {
return os.Symlink(path, target)
}
return nil
}
func main() {
logx.Disable()
load.Disable()
args := os.Args
pluginName := filepath.Base(args[0])
if pluginName == protocGenGoctl {
pluginCtl.Generate()
return
}
app := cli.NewApp()
app.Usage = "a cli tool to generate code"
app.Version = fmt.Sprintf("%s %s/%s", version.BuildVersion, runtime.GOOS, runtime.GOARCH)