mirror of
https://github.com/rancher/os.git
synced 2025-04-28 19:34:45 +00:00
43 lines
727 B
Go
43 lines
727 B
Go
package control
|
|
|
|
import (
|
|
"log"
|
|
"os"
|
|
"os/exec"
|
|
"syscall"
|
|
|
|
"github.com/rancher/os/config"
|
|
"github.com/rancher/os/pkg/util"
|
|
|
|
"github.com/codegangsta/cli"
|
|
)
|
|
|
|
func envAction(c *cli.Context) error {
|
|
cfg := config.LoadConfig()
|
|
|
|
args := c.Args()
|
|
if len(args) == 0 {
|
|
return nil
|
|
}
|
|
osEnv := os.Environ()
|
|
|
|
envMap := make(map[string]string, len(cfg.Rancher.Environment)+len(osEnv))
|
|
for k, v := range cfg.Rancher.Environment {
|
|
envMap[k] = v
|
|
}
|
|
for k, v := range util.KVPairs2Map(osEnv) {
|
|
envMap[k] = v
|
|
}
|
|
|
|
if cmd, err := exec.LookPath(args[0]); err != nil {
|
|
log.Fatal(err)
|
|
} else {
|
|
args[0] = cmd
|
|
}
|
|
if err := syscall.Exec(args[0], args, util.Map2KVPairs(envMap)); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
return nil
|
|
}
|