99 lines
2.0 KiB
Go
99 lines
2.0 KiB
Go
// Copyright 2016 Tim Shannon. All rights reserved.
|
|
// Use of this source code is governed by the MIT license
|
|
// that can be found in the LICENSE file.
|
|
|
|
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"strings"
|
|
)
|
|
|
|
func runCmd(cmd, dir string, env []string) ([]byte, error) {
|
|
s := strings.Fields(strings.Replace(cmd, "@dir", dir, -1))
|
|
|
|
for i := range env {
|
|
env[i] = strings.Replace(env[i], "@dir", dir, -1)
|
|
}
|
|
|
|
var args []string
|
|
|
|
if len(s) > 1 {
|
|
args = s[1:]
|
|
}
|
|
|
|
name := s[0]
|
|
ec := &exec.Cmd{
|
|
Path: name,
|
|
Args: append([]string{name}, args...),
|
|
Dir: dir,
|
|
Env: env,
|
|
}
|
|
|
|
if filepath.Base(name) == name {
|
|
lp, err := lookPath(name, env)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
ec.Path = lp
|
|
}
|
|
|
|
vlog("Executing command: %s in dir %s\n", cmd, dir)
|
|
|
|
result, err := ec.CombinedOutput()
|
|
if err != nil {
|
|
return nil, fmt.Errorf("%s\n%s", err, result)
|
|
}
|
|
return result, nil
|
|
}
|
|
|
|
// similar to os/exec.LookPath, except it checks if the passed in
|
|
// custom environment includes a path definitions and uses that path instead
|
|
// note this probably only works on unix, that's all I care about for now
|
|
func lookPath(file string, env []string) (string, error) {
|
|
if strings.Contains(file, "/") {
|
|
err := findExecutable(file)
|
|
if err == nil {
|
|
return file, nil
|
|
}
|
|
return "", &exec.Error{Name: file, Err: err}
|
|
}
|
|
|
|
for i := range env {
|
|
if strings.HasPrefix(env[i], "PATH=") {
|
|
pathenv := env[i][5:]
|
|
if pathenv == "" {
|
|
return "", &exec.Error{Name: file, Err: exec.ErrNotFound}
|
|
}
|
|
for _, dir := range strings.Split(pathenv, ":") {
|
|
if dir == "" {
|
|
// Unix shell semantics: path element "" means "."
|
|
dir = "."
|
|
}
|
|
path := dir + "/" + file
|
|
if err := findExecutable(path); err == nil {
|
|
return path, nil
|
|
}
|
|
}
|
|
return "", &exec.Error{Name: file, Err: exec.ErrNotFound}
|
|
}
|
|
}
|
|
|
|
return exec.LookPath(file)
|
|
}
|
|
|
|
func findExecutable(file string) error {
|
|
d, err := os.Stat(file)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if m := d.Mode(); !m.IsDir() && m&0111 != 0 {
|
|
return nil
|
|
}
|
|
|
|
return os.ErrPermission
|
|
}
|