1
0
mirror of https://github.com/rancher/os.git synced 2025-08-01 23:17:50 +00:00

rancherctl env subcommand

Usage:
rancherctl env <command>

Executes <command> with environment from rancher.environment. Real env vars override those from rancher.environment.
This commit is contained in:
Ivan Mikushin 2015-04-29 15:32:12 +05:00
parent 6f4d7a177a
commit d62eb3de5d
3 changed files with 65 additions and 0 deletions

View File

@ -24,6 +24,14 @@ func Main() {
HideHelp: true,
Subcommands: configSubcommands(),
},
{
Name: "env",
ShortName: "e",
Usage: "env command",
HideHelp: true,
SkipFlagParsing: true,
Action: envAction,
},
{
Name: "service",
ShortName: "s",

39
cmd/control/env.go Normal file
View File

@ -0,0 +1,39 @@
package control
import (
"log"
"os"
"os/exec"
"syscall"
"github.com/codegangsta/cli"
"github.com/rancherio/os/config"
"github.com/rancherio/os/util"
)
func envAction(c *cli.Context) {
cfg, err := config.LoadConfig()
if err != nil {
log.Fatal(err)
}
args := c.Args()
osEnv := os.Environ()
envMap := make(map[string]string, len(cfg.Environment) + len(osEnv))
for k, v := range cfg.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)
}
}

View File

@ -302,3 +302,21 @@ func GetValue(kvPairs []string, key string) string {
return ""
}
func Map2KVPairs(m map[string]string) []string {
r := make([]string, 0, len(m))
for k, v := range m {
r = append(r, k + "=" + v)
}
return r
}
func KVPairs2Map(kvs []string) map[string]string {
r := make(map[string]string, len(kvs))
for _, kv := range kvs {
sepIdx := strings.Index(kv, "=")
k, v := kv[:sepIdx], kv[(sepIdx + 1):]
r[k] = v
}
return r
}