postman: Make HostToEnv variable name configurable

bukalapak · Nov 8, 2017 · 7b261ae · 7b261ae
1 parent 2275473
commit 7b261ae
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 6 deletions.
diff --git a/config/config.go b/config/config.go
@@ -15,7 +15,9 @@ type Config struct {
 		Name    string `yaml:"Name,omitempty"`
 	} `yaml:"SchemeToEnv,omitempty"`
 	HostToEnv struct {
-		Segments int `yaml:"segments,omitempty"`
+		Enabled  bool   `yaml:"Enabled,omitempty"`
+		Segments int    `yaml:"Segments,omitempty"`
+		Name     string `yaml:"Name,omitempty"`
 	} `yaml:"HostToEnv,omitempty"`
 	AuthTokenToEnv struct {
 		Enabled bool   `yaml:"Enabled,omitempty"`

diff --git a/main.go b/main.go
@@ -84,8 +84,8 @@ func main() {
 	if cfg.SchemeToEnv.Enabled {
 		postman.SchemeToEnv(&collection, cfg.SchemeToEnv.Name)
 	}
-	if cfg.HostToEnv.Segments > 0 {
-		postman.HostToEnv(&collection, cfg.HostToEnv.Segments)
+	if cfg.HostToEnv.Enabled && cfg.HostToEnv.Segments > 0 {
+		postman.HostToEnv(&collection, cfg.HostToEnv.Segments, cfg.HostToEnv.Name)
 	}
 	if cfg.AuthTokenToEnv.Enabled {
 		postman.AuthTokenToEnv(&collection, cfg.AuthTokenToEnv.Name)

diff --git a/postman/transformer.go b/postman/transformer.go
@@ -15,7 +15,7 @@ func SchemeToEnv(c *Collection, placeholder string) {
 	}
 }
 
-func HostToEnv(c *Collection, n int) {
+func HostToEnv(c *Collection, n int, placeholder string) {
 	items := getItemsFromCollection(c)
 
 	for i := range items {
@@ -30,7 +30,7 @@ func HostToEnv(c *Collection, n int) {
 					hostSegments[0:len(hostSegments)-2]...,
 				)
 			}
-			newHostSegments = append(newHostSegments, "{{host}}")
+			newHostSegments = append(newHostSegments, fmt.Sprintf("{{%s}}", placeholder))
 			items[i].Request.Url.Host = strings.Join(newHostSegments, ".")
 		}
 	}

diff --git a/vanadia.yml b/vanadia.yml
@@ -3,7 +3,9 @@ SchemeToEnv:
   Name: scheme
 
 HostToEnv:
-  segments: 2
+  Enabled: true
+  Segments: 2
+  Name: host
 
 AuthTokenToEnv:
   Enabled: true