130 lines
3.4 KiB
Go
130 lines
3.4 KiB
Go
package helmexec
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"strings"
|
|
"syscall"
|
|
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
// Runner interface for shell commands
|
|
type Runner interface {
|
|
Execute(cmd string, args []string, env map[string]string) ([]byte, error)
|
|
ExecuteStdIn(cmd string, args []string, env map[string]string, stdin io.Reader) ([]byte, error)
|
|
}
|
|
|
|
// ShellRunner implemention for shell commands
|
|
type ShellRunner struct {
|
|
Dir string
|
|
|
|
Logger *zap.SugaredLogger
|
|
}
|
|
|
|
// Execute a shell command
|
|
func (shell ShellRunner) Execute(cmd string, args []string, env map[string]string) ([]byte, error) {
|
|
preparedCmd := exec.Command(cmd, args...)
|
|
preparedCmd.Dir = shell.Dir
|
|
preparedCmd.Env = mergeEnv(os.Environ(), env)
|
|
return Output(preparedCmd, &logWriterGenerator{
|
|
log: shell.Logger,
|
|
})
|
|
}
|
|
|
|
// Execute a shell command
|
|
func (shell ShellRunner) ExecuteStdIn(cmd string, args []string, env map[string]string, stdin io.Reader) ([]byte, error) {
|
|
preparedCmd := exec.Command(cmd, args...)
|
|
preparedCmd.Dir = shell.Dir
|
|
preparedCmd.Env = mergeEnv(os.Environ(), env)
|
|
preparedCmd.Stdin = stdin
|
|
return Output(preparedCmd, &logWriterGenerator{
|
|
log: shell.Logger,
|
|
})
|
|
}
|
|
|
|
func Output(c *exec.Cmd, logWriterGenerators ...*logWriterGenerator) ([]byte, error) {
|
|
if c.Stdout != nil {
|
|
return nil, errors.New("exec: Stdout already set")
|
|
}
|
|
if c.Stderr != nil {
|
|
return nil, errors.New("exec: Stderr already set")
|
|
}
|
|
|
|
var stdout bytes.Buffer
|
|
var stderr bytes.Buffer
|
|
var combined bytes.Buffer
|
|
|
|
var logWriters []io.Writer
|
|
|
|
id := newExecutionID()
|
|
for _, g := range logWriterGenerators {
|
|
logPrefix := fmt.Sprintf("%s:%s> ", filepath.Base(c.Path), id)
|
|
logWriters = append(logWriters, g.Writer(logPrefix))
|
|
}
|
|
|
|
c.Stdout = io.MultiWriter(append([]io.Writer{&stdout, &combined}, logWriters...)...)
|
|
c.Stderr = io.MultiWriter(append([]io.Writer{&stderr, &combined}, logWriters...)...)
|
|
|
|
err := c.Run()
|
|
|
|
if err != nil {
|
|
// TrimSpace is necessary, because otherwise helmfile prints the redundant new-lines after each error like:
|
|
//
|
|
// err: release "envoy2" in "helmfile.yaml" failed: exit status 1: Error: could not find a ready tiller pod
|
|
// <redundant new line!>
|
|
// err: release "envoy" in "helmfile.yaml" failed: exit status 1: Error: could not find a ready tiller pod
|
|
switch ee := err.(type) {
|
|
case *exec.ExitError:
|
|
// Propagate any non-zero exit status from the external command, rather than throwing it away,
|
|
// so that helmfile could return its own exit code accordingly
|
|
waitStatus := ee.Sys().(syscall.WaitStatus)
|
|
exitStatus := waitStatus.ExitStatus()
|
|
err = newExitError(c.Path, c.Args, exitStatus, ee, stderr.String(), combined.String())
|
|
default:
|
|
panic(fmt.Sprintf("unexpected error: %v", err))
|
|
}
|
|
}
|
|
|
|
return stdout.Bytes(), err
|
|
}
|
|
|
|
func mergeEnv(orig []string, new map[string]string) []string {
|
|
wanted := env2map(orig)
|
|
for k, v := range new {
|
|
wanted[k] = v
|
|
}
|
|
return map2env(wanted)
|
|
}
|
|
|
|
func map2env(wanted map[string]string) []string {
|
|
result := []string{}
|
|
for k, v := range wanted {
|
|
result = append(result, k+"="+v)
|
|
}
|
|
return result
|
|
}
|
|
|
|
func env2map(env []string) map[string]string {
|
|
wanted := map[string]string{}
|
|
for _, cur := range env {
|
|
pair := strings.SplitN(cur, "=", 2)
|
|
|
|
var v string
|
|
|
|
// An environment can completely miss `=` and the right side.
|
|
// If we didn't deal with that, this may fail due to an index-out-of-range error
|
|
if len(pair) > 1 {
|
|
v = pair[1]
|
|
}
|
|
|
|
wanted[pair[0]] = v
|
|
}
|
|
return wanted
|
|
}
|