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.
43 lines
842 B
Go
43 lines
842 B
Go
package execx
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"fmt"
|
|
"os/exec"
|
|
"runtime"
|
|
"strings"
|
|
|
|
"github.com/tal-tech/go-zero/tools/goctl/util"
|
|
"github.com/tal-tech/go-zero/tools/goctl/vars"
|
|
)
|
|
|
|
func Run(arg string, dir string) (string, error) {
|
|
goos := runtime.GOOS
|
|
var cmd *exec.Cmd
|
|
switch goos {
|
|
case vars.OsMac, vars.OsLinux:
|
|
cmd = exec.Command("sh", "-c", arg)
|
|
case vars.OsWindows:
|
|
cmd = exec.Command("cmd.exe", "/c", arg)
|
|
default:
|
|
return "", fmt.Errorf("unexpected os: %v", goos)
|
|
}
|
|
if len(dir) > 0 {
|
|
cmd.Dir = dir
|
|
}
|
|
stdout := new(bytes.Buffer)
|
|
stderr := new(bytes.Buffer)
|
|
cmd.Stdout = stdout
|
|
cmd.Stderr = stderr
|
|
err := cmd.Run()
|
|
if err != nil {
|
|
if stderr.Len() > 0 {
|
|
return "", errors.New(strings.TrimSuffix(stderr.String(), util.NL))
|
|
}
|
|
return "", err
|
|
}
|
|
|
|
return strings.TrimSuffix(stdout.String(), util.NL), nil
|
|
}
|