mirror of https://github.com/databricks/cli.git
40 lines
610 B
Go
40 lines
610 B
Go
|
package process
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"os"
|
||
|
|
"os/exec"
|
||
|
|
)
|
||
|
|
|
||
|
|
type execOption func(*exec.Cmd) error
|
||
|
|
|
||
|
|
func WithEnv(key, value string) execOption {
|
||
|
|
return func(c *exec.Cmd) error {
|
||
|
|
if c.Env == nil {
|
||
|
|
c.Env = os.Environ()
|
||
|
|
}
|
||
|
|
v := fmt.Sprintf("%s=%s", key, value)
|
||
|
|
c.Env = append(c.Env, v)
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func WithEnvs(envs map[string]string) execOption {
|
||
|
|
return func(c *exec.Cmd) error {
|
||
|
|
for k, v := range envs {
|
||
|
|
err := WithEnv(k, v)(c)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func WithDir(dir string) execOption {
|
||
|
|
return func(c *exec.Cmd) error {
|
||
|
|
c.Dir = dir
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
}
|