-
Notifications
You must be signed in to change notification settings - Fork 47
/
cmd.go
102 lines (89 loc) · 2.79 KB
/
cmd.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
/*
Copyright (c) 2021 - Present. Blend Labs, Inc. All rights reserved
Use of this source code is governed by a MIT license that can be found in the LICENSE file.
*/
package sh
import (
"context"
"os"
"os/exec"
"strings"
"github.com/blend/go-sdk/ex"
)
// MustCmds returns a list of commands for a given set of statements.
// It is useful for running a batch of commands.
// It panics on error.
func MustCmds(statements ...string) []*exec.Cmd {
var output []*exec.Cmd
for _, statement := range statements {
output = append(output, MustCmdParsed(statement))
}
return output
}
// Cmds returns a list of commands for a given set of statements.
// It is useful for running a batch of commands.
// It will return the first error it encounters.
func Cmds(statements ...string) ([]*exec.Cmd, error) {
var output []*exec.Cmd
for _, statement := range statements {
cmd, err := CmdParsed(statement)
if err != nil {
return nil, err
}
output = append(output, cmd)
}
return output, nil
}
// MustCmdParsed returns a command for a full comamnd statement.
func MustCmdParsed(statement string) *exec.Cmd {
cmd, err := CmdParsed(statement)
if err != nil {
panic(err)
}
return cmd
}
// CmdParsed returns a command for a full comamnd statement.
func CmdParsed(statement string) (*exec.Cmd, error) {
parts := strings.Split(statement, " ")
if len(parts) > 1 {
return Cmd(parts[0], parts[1:]...)
}
return Cmd(parts[0])
}
// CmdParsedContext returns a command for a full comamnd statement within a context..
func CmdParsedContext(ctx context.Context, statement string) (*exec.Cmd, error) {
parts := strings.Split(statement, " ")
if len(parts) > 1 {
return CmdContext(ctx, parts[0], parts[1:]...)
}
return CmdContext(ctx, parts[0])
}
// MustCmd returns a new command with the fully qualified path of the executable.
// It panics on error.
func MustCmd(command string, args ...string) *exec.Cmd {
cmd, err := Cmd(command, args...)
if err != nil {
panic(ex.New(err, ex.OptMessagef("looking for: %s", command)))
}
return cmd
}
// Cmd returns a new command with the fully qualified path of the executable.
func Cmd(command string, args ...string) (*exec.Cmd, error) {
absoluteCommand, err := exec.LookPath(command)
if err != nil {
return nil, ex.New(err, ex.OptMessagef("looking for: %s", command))
}
cmd := exec.Command(absoluteCommand, args...)
cmd.Env = os.Environ()
return cmd, nil
}
// CmdContext returns a new command with the fully qualified path of the executable within a context.
func CmdContext(ctx context.Context, command string, args ...string) (*exec.Cmd, error) {
absoluteCommand, err := exec.LookPath(command)
if err != nil {
return nil, ex.New(err, ex.OptMessagef("looking for: %s", command))
}
cmd := exec.CommandContext(ctx, absoluteCommand, args...)
cmd.Env = os.Environ()
return cmd, nil
}