This commit is contained in:
cdle
2021-09-09 18:08:52 +08:00
parent b4dac8396b
commit ae9b3b2916
2 changed files with 37 additions and 33 deletions
+3 -1
View File
@@ -95,12 +95,14 @@ func init() {
f["JD_CASH_SHARECODES"] = e[k]
}
}
envs := []qinglong.Env{}
for i := range f {
qinglong.SetEnv(qinglong.Env{
envs = append(envs, qinglong.Env{
Name: i,
Value: f[i],
})
}
qinglong.SetEnv(envs...)
return
}
}
+3 -1
View File
@@ -70,12 +70,13 @@ func GetEnvs(searchValue string) ([]Env, error) {
return envs, nil
}
func SetEnv(env Env) error {
func SetEnv(envs ...Env) error {
config, err := GetConfig()
if err != nil {
return err
}
lines := strings.Split(config, "\n")
for _, env := range envs {
set := false
for j, line := range lines {
for i, pattern := range []string{`^\s*export\s+([^'"=\s]+)=[ '"]?(.*?)['"]?$`, `^\s*#[#\s]*export\s+([^'"=\s]+)=[ '"]?(.*?)['"]?$`, `^\s*([^'"=\s]+)=[ '"]?(.*?)['"]?$`, `^\s*#[#\s]*([^'"=\s]+)=[ '"]?(.*?)['"]?$`} {
@@ -111,5 +112,6 @@ func SetEnv(env Env) error {
if !set {
lines = append(lines, fmt.Sprintf("export %s=\"%s\"", env.Name, env.Value))
}
}
return SvaeConfig(strings.Join(lines, "\n"))
}