add override env configuration in home config file (#80)

ggn/home.go

@@ -15,6 +15,17 @@ type Config struct {
 	WorkPath   string `yaml:"workPath,omitempty"`
 	User       string `yaml:"user,omitempty"`
 	EnvVarUser string `yaml:"envVarUser,omitempty"`
+
+	OverrideConfig map[string]struct {
+		Fleet struct {
+			Endpoint                 string `yaml:"endpoint,omitempty"`
+			Username                 string `yaml:"username,omitempty"`
+			Password                 string `yaml:"password,omitempty"`
+			Strict_host_key_checking *bool  `yaml:"strict_host_key_checking,omitempty"`
+			Sudo                     *bool  `yaml:"sudo,omitempty"`
+			Driver                   string `yaml:"driver,omitempty"`
+		} `yaml:"fleet,omitempty"`
+	} `yaml:"overrideConfig,omitempty"`
 }
 
 type HomeStruct struct {

work/env.go

@@ -149,6 +149,38 @@ func (e *Env) loadConfig() {
 		dest[v] = src[i]
 	}
 	e.config.Fleet.Endpoint = strings.Join(dest, ",")
+
+	if _, ok := ggn.Home.Config.OverrideConfig[e.config.EnvName]; ok {
+		e.overrideFleetEnv()
+	}
+}
+
+func (e *Env) overrideFleetEnv() {
+	overridedEnv := ggn.Home.Config.OverrideConfig[e.config.EnvName]
+	if overridedEnv.Fleet.Endpoint != "" {
+		src := strings.Split(overridedEnv.Fleet.Endpoint, ",")
+		dest := make([]string, len(src))
+		perm := rand.Perm(len(src))
+		for i, v := range perm {
+			dest[v] = src[i]
+		}
+		e.config.Fleet.Endpoint = strings.Join(dest, ",")
+	}
+	if overridedEnv.Fleet.Username != "" {
+		e.config.Fleet.Username = overridedEnv.Fleet.Username
+	}
+	if overridedEnv.Fleet.Driver != "" {
+		e.config.Fleet.Driver = overridedEnv.Fleet.Driver
+	}
+	if overridedEnv.Fleet.Password != "" {
+		e.config.Fleet.Password = overridedEnv.Fleet.Password
+	}
+	if overridedEnv.Fleet.Strict_host_key_checking != nil {
+		e.config.Fleet.Strict_host_key_checking = *overridedEnv.Fleet.Strict_host_key_checking
+	}
+	if overridedEnv.Fleet.Sudo != nil {
+		e.config.Fleet.Sudo = *overridedEnv.Fleet.Sudo
+	}
 }
 
 func (e *Env) loadPartials() {