Quickfix for input sets (support for boolean)

depado · May 27, 2019 · 77344bb · 77344bb
1 parent da3a2c3
commit 77344bb
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/conf/input.go b/conf/input.go
@@ -51,7 +51,17 @@ func GetSetContext(set []string) (InputCtx, error) {
 		if len(tmp) != 2 {
 			return out, fmt.Errorf("invalid set option: %s", s)
 		}
-		out = append(out, yaml.MapItem{Key: tmp[0], Value: tmp[1]})
+		v := yaml.MapItem{Key: tmp[0]}
+		// Convert to bool if needed
+		switch strings.ToLower(tmp[1]) {
+		case "1", "true":
+			v.Value = true
+		case "0", "false":
+			v.Value = false
+		default:
+			v.Value = tmp[1]
+		}
+		out = append(out, v)
 	}
 	return out, nil
 }
diff --git a/renderer/analyze.go b/renderer/analyze.go
@@ -56,7 +56,7 @@ func Analyze(dir, output, input string, set []string) {
 		}
 		utils.OkPrintln("Input file", utils.Green.Sprint(input), "found")
 	}
-	if set != nil {
+	if len(set) > 0 {
 		setCtx, err := conf.GetSetContext(set)
 		if err != nil {
 			utils.FatalPrintln("Could not parse set flags:", err)