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.
go-zero/tools/goctl/util/env/env.go

81 lines
1.8 KiB
Go

package env
import (
"os"
"os/exec"
"path/filepath"
"runtime"
"strings"
"github.com/tal-tech/go-zero/tools/goctl/vars"
)
const bin = "bin"
const binGo = "go"
const binProtoc = "protoc"
const binProtocGenGo = "protoc-gen-go"
// LookUpGo searches an executable go in the directories
// named by the GOROOT/bin or PATH environment variable.
func LookUpGo() (string, error) {
goRoot := runtime.GOROOT()
suffix := getExeSuffix()
xGo := binGo + suffix
path := filepath.Join(goRoot, bin, xGo)
if _, err := os.Stat(path); err == nil {
return path, nil
}
return LookPath(xGo)
}
// LookUpProtoc searches an executable protoc in the directories
// named by the PATH environment variable.
func LookUpProtoc() (string, error) {
suffix := getExeSuffix()
xProtoc := binProtoc + suffix
return LookPath(xProtoc)
}
// LookUpProtocGenGo searches an executable protoc-gen-go in the directories
// named by the PATH environment variable.
func LookUpProtocGenGo() (string, error) {
suffix := getExeSuffix()
xProtocGenGo := binProtocGenGo + suffix
return LookPath(xProtocGenGo)
}
// LookPath searches for an executable named file in the
// directories named by the PATH environment variable,
// for the os windows, the named file will be spliced with the
// .exe suffix.
func LookPath(xBin string) (string, error) {
suffix := getExeSuffix()
if len(suffix) > 0 && !strings.HasSuffix(xBin, suffix) {
xBin = xBin + suffix
}
bin, err := exec.LookPath(xBin)
if err != nil {
return "", err
}
return bin, nil
}
// CanExec reports whether the current system can start new processes
// using os.StartProcess or (more commonly) exec.Command.
func CanExec() bool {
switch runtime.GOOS {
case vars.OsJs, vars.OsIOS:
return false
default:
return true
}
}
func getExeSuffix() string {
if runtime.GOOS == vars.OsWindows {
return ".exe"
}
return ""
}