-
Notifications
You must be signed in to change notification settings - Fork 0
/
run.go
63 lines (49 loc) · 1.4 KB
/
run.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
package binit
import (
"log"
"os"
"os/exec"
"strconv"
"syscall"
"github.com/kballard/go-shellquote"
)
func Run(config Config, args []string) {
waiter := NewWaiter()
runCommandWithEnv(config.BEFORE, waiter, nil)
if len(args) == 0 {
waiter.Fatalf("error, need a program to execute")
}
cmd := exec.Command(args[0], args[1:]...)
stat, err := os.Stdin.Stat()
if err != nil {
waiter.Fatalf("error while stating stdin: %v", err)
}
if (stat.Mode() & os.ModeCharDevice) == 0 {
cmd.Stdin = os.Stdin
}
cmd.Stdout = os.Stdout
cmd.Stderr = os.Stderr
cmd.SysProcAttr = &syscall.SysProcAttr{Setpgid: true}
if config.WORKDIR != "" {
if err := os.Chdir(config.WORKDIR); err != nil {
waiter.Fatalf("error changing to workdir %s for %s: %v", config.WORKDIR, cmd.Path, err)
}
}
signaller := NewSignaller(config)
signaller.Start(cmd)
defer signaller.Shutdown()
log.Printf("Running command { %v } now.", shellquote.Join(cmd.Args...))
if err := cmd.Start(); err != nil {
waiter.Fatalf("error while running %s: %v", cmd.Path, err)
}
err = cmd.Wait()
exitCode := strconv.Itoa(cmd.ProcessState.ExitCode())
env := map[string]string{"BINIT_EXIT_CODE": exitCode}
runCommandWithEnv(config.AFTER, waiter, &env)
if err != nil {
if exitErr, ok := err.(*exec.ExitError); !ok {
waiter.Fatalf("error while waiting for %s with exit code %s: %v", cmd.Path, exitCode, exitErr)
}
}
waiter.Quit(0)
}