如何使用 Go 获取 shell 脚本

Question

我想使用 Go 来获取 shell 脚本。理想情况下，以下代码

cmd := exec.Command("/bin/bash", "source", file.Name())

但是，我知道"源"是一个 bash 内置函数，而不是可执行文件。

但是，我找到了一些在Python中模仿这种行为的方法：

http://pythonwise.blogspot.fr/2010/04/sourcing-shell-script.html

不幸的是，我不知道如何在 Go 中翻译它。有人有想法吗？

谢谢！

Answer 1

您可以使用以下命令

在运行程序时设置环境变量exec：

cmd := exec.Command("whatever")
cmd.Env = []string{"A=B"}
cmd.Run()

如果你真的需要源代码，那么你可以通过bash运行你的命令：

cmd := exec.Command("bash", "-c", "source " + file.Name() + " ; echo 'hi'")
cmd.Run()

查看此库以获取功能更全的工作流程：https://github.com/progrium/go-basher。

更新：下面是修改当前环境的示例：

package main
import (
    "bufio"
    "bytes"
    "io/ioutil"
    "log"
    "os"
    "os/exec"
    "strings"
)
func main() {
    err := ioutil.WriteFile("example_source", []byte("export FOO=bar; echo $FOO"), 0777)
    if err != nil {
        log.Fatal(err)
    }
    cmd := exec.Command("bash", "-c", "source example_source ; echo '<<<ENVIRONMENT>>>' ; env")
    bs, err := cmd.CombinedOutput()
    if err != nil {
        log.Fatalln(err)
    }
    s := bufio.NewScanner(bytes.NewReader(bs))
    start := false
    for s.Scan() {
        if s.Text() == "<<<ENVIRONMENT>>>" {
            start = true
        } else if start {
            kv := strings.SplitN(s.Text(), "=", 2)
            if len(kv) == 2 {
                os.Setenv(kv[0], kv[1])
            }
        }
    }
}
log.Println(os.Getenv("FOO"))

Answer 2

我最近在我的 shell/bash Golang 库中添加了这样一个实用程序函数：

https://godoc.org/mvdan.cc/sh/shell#SourceFile

例如，您可以执行以下操作：

vars, err := shell.SourceFile("foo.sh")
if err != nil { ... }
fmt.Println(vars["URL"].Value)
// http://the.url/value

它相当安全，因为它从未真正调用bash或任何其他程序。它解析 shell 脚本，然后对其进行解释。但是在解释时，它有一个白名单，其中包含脚本可以打开的文件以及脚本可以执行的程序。

解释器也有上下文。上下文，因此，如果您希望受到永久循环或其他错误代码的保护，则可以设置超时。