mirror of
https://github.com/gomods/athens
synced 2026-02-03 08:40:31 +00:00
67 lines
1.6 KiB
Go
67 lines
1.6 KiB
Go
package module
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"runtime"
|
|
)
|
|
|
|
// prepareEnv will return all the appropriate
|
|
// environment variables for a Go Command to run
|
|
// successfully (such as GOPATH, GOCACHE, PATH etc).
|
|
func prepareEnv(gopath string, envVars []string) []string {
|
|
gopathEnv := fmt.Sprintf("GOPATH=%s", gopath)
|
|
cacheEnv := fmt.Sprintf("GOCACHE=%s", filepath.Join(gopath, "cache"))
|
|
disableCgo := "CGO_ENABLED=0"
|
|
enableGoModules := "GO111MODULE=on"
|
|
cmdEnv := []string{
|
|
gopathEnv,
|
|
cacheEnv,
|
|
disableCgo,
|
|
enableGoModules,
|
|
}
|
|
keys := []string{
|
|
"PATH",
|
|
"HOME",
|
|
"GIT_SSH",
|
|
"GIT_SSH_COMMAND",
|
|
"HTTP_PROXY",
|
|
"HTTPS_PROXY",
|
|
"NO_PROXY",
|
|
// GOROOT needs to be propagated if it exists.
|
|
"GOROOT",
|
|
// Need to also check the lower case version of just these three env variables.
|
|
"http_proxy",
|
|
"https_proxy",
|
|
"no_proxy",
|
|
}
|
|
if runtime.GOOS == "windows" {
|
|
windowsSpecificKeys := []string{
|
|
"USERPROFILE",
|
|
"SystemRoot",
|
|
"ALLUSERSPROFILE",
|
|
"HOMEDRIVE",
|
|
"HOMEPATH",
|
|
}
|
|
keys = append(keys, windowsSpecificKeys...)
|
|
}
|
|
for _, key := range keys {
|
|
// Prepend only if environment variable is present.
|
|
if v, ok := os.LookupEnv(key); ok {
|
|
cmdEnv = append(cmdEnv, fmt.Sprintf("%s=%s", key, v))
|
|
}
|
|
}
|
|
cmdEnv = append(cmdEnv, envVars...)
|
|
|
|
if sshAuthSockVal, hasSSHAuthSock := os.LookupEnv("SSH_AUTH_SOCK"); hasSSHAuthSock {
|
|
// Verify that the ssh agent unix socket exists and is a unix socket.
|
|
st, err := os.Stat(sshAuthSockVal)
|
|
if err == nil && st.Mode()&os.ModeSocket != 0 {
|
|
sshAuthSock := fmt.Sprintf("SSH_AUTH_SOCK=%s", sshAuthSockVal)
|
|
cmdEnv = append(cmdEnv, sshAuthSock)
|
|
}
|
|
}
|
|
return cmdEnv
|
|
}
|