зеркало из https://github.com/mislav/hub.git
77 строки
1.3 KiB
Go
77 строки
1.3 KiB
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"os/exec"
|
|
"strings"
|
|
"syscall"
|
|
)
|
|
|
|
type Cmd struct {
|
|
Name string
|
|
Args []string
|
|
}
|
|
|
|
func (cmd Cmd) String() string {
|
|
return fmt.Sprintf("%s %s", cmd.Name, strings.Join(cmd.Args, " "))
|
|
}
|
|
|
|
func (cmd *Cmd) WithArg(arg string) *Cmd {
|
|
if arg != "" {
|
|
cmd.Args = append(cmd.Args, arg)
|
|
}
|
|
|
|
return cmd
|
|
}
|
|
|
|
func (cmd *Cmd) WithArgs(args ...string) *Cmd {
|
|
for _, arg := range args {
|
|
cmd.WithArg(arg)
|
|
}
|
|
|
|
return cmd
|
|
}
|
|
|
|
func (cmd *Cmd) ExecOutput() (string, error) {
|
|
output, err := exec.Command(cmd.Name, cmd.Args...).CombinedOutput()
|
|
|
|
return string(output), err
|
|
}
|
|
|
|
func (cmd *Cmd) Exec() error {
|
|
binary, lookErr := exec.LookPath(cmd.Name)
|
|
if lookErr != nil {
|
|
return fmt.Errorf("command not found: %s", cmd.Name)
|
|
}
|
|
|
|
c := exec.Command(binary, cmd.Args...)
|
|
c.Stdin = os.Stdin
|
|
c.Stdout = os.Stdout
|
|
c.Stderr = os.Stderr
|
|
|
|
return c.Run()
|
|
}
|
|
|
|
func (cmd *Cmd) SysExec() error {
|
|
binary, lookErr := exec.LookPath(cmd.Name)
|
|
if lookErr != nil {
|
|
return fmt.Errorf("command not found: %s", cmd.Name)
|
|
}
|
|
|
|
args := []string{cmd.Name}
|
|
args = append(args, cmd.Args...)
|
|
|
|
env := os.Environ()
|
|
|
|
return syscall.Exec(binary, args, env)
|
|
}
|
|
|
|
func New(name string) *Cmd {
|
|
return &Cmd{Name: name, Args: make([]string, 0)}
|
|
}
|
|
|
|
func NewWithArray(cmd []string) *Cmd {
|
|
return &Cmd{Name: cmd[0], Args: cmd[1:]}
|
|
}
|