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:
parent
6f4d7a177a
commit
d62eb3de5d
@ -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
39
cmd/control/env.go
Normal 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)
|
||||
}
|
||||
}
|
18
util/util.go
18
util/util.go
@ -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
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user