summaryrefslogtreecommitdiff
path: root/pkg/cli/command.go
blob: a9938d1175510053dec2c583041d47d9783d00b9 (plain)
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
/*******************************************************************************
*
* Copyright 2017 Stefan Majewsky <majewsky@gmx.net>
*
* This program is free software: you can redistribute it and/or modify it under
* the terms of the GNU General Public License as published by the Free Software
* Foundation, either version 3 of the License, or (at your option) any later
* version.
*
* This program is distributed in the hope that it will be useful, but WITHOUT ANY
* WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
* A PARTICULAR PURPOSE. See the GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License along with
* this program. If not, see <http://www.gnu.org/licenses/>.
*
*******************************************************************************/

package cli

import (
	"fmt"
	"io"
	"os/exec"
	"strings"
)

//Command describes a command that can be run using the methods in the
//Implementation interface.
type Command struct {
	Program []string
	WorkDir string
}

type commandError struct {
	Cmd Command
	Err error
}

func (e commandError) Error() string {
	cmdline := strings.Join(e.Cmd.Program, " ")
	if e.Cmd.WorkDir == "" {
		return fmt.Sprintf("exec `%s`: %s",
			cmdline, e.Err.Error(),
		)
	}
	return fmt.Sprintf("exec `%s` in %s: %s",
		cmdline, e.Cmd.WorkDir, e.Err.Error(),
	)
}

//CommandRunner is a function that can execute commands given to it.
//This interface is only useful for unit tests; the default CommandRunner
//suffices for all regular operation.
type CommandRunner func(c Command, stdin io.Reader, stdout, stderr io.Writer) error

//DefaultCommandRunner is a CommandRunner that actually executes the command.
func DefaultCommandRunner(c Command, stdin io.Reader, stdout, stderr io.Writer) error {
	cmd := exec.Command(c.Program[0], c.Program[1:]...)
	cmd.Stdin = stdin
	cmd.Stdout = stdout
	cmd.Stderr = stderr
	cmd.Dir = c.WorkDir

	err := cmd.Run()
	if err != nil {
		err = commandError{c, err}
	}
	return err
}