Support tfenv export (#188)

segmentio · Feb 23, 2019 · 063d1b0 · 063d1b0
1 parent 059f1d1
commit 063d1b0
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/cmd/export.go b/cmd/export.go
@@ -32,7 +32,7 @@ var (
 )
 
 func init() {
-	exportCmd.Flags().StringVarP(&exportFormat, "format", "f", "json", "Output format (json, java-properties, csv, tsv, dotenv)")
+	exportCmd.Flags().StringVarP(&exportFormat, "format", "f", "json", "Output format (json, java-properties, csv, tsv, dotenv, tfvars)")
 	exportCmd.Flags().StringVarP(&exportOutput, "output-file", "o", "", "Output file (default is standard output)")
 	RootCmd.AddCommand(exportCmd)
 }
@@ -97,6 +97,8 @@ func runExport(cmd *cobra.Command, args []string) error {
 		err = exportAsTsv(params, w)
 	case "dotenv":
 		err = exportAsEnvFile(params, w)
+	case "tfvars":
+		err = exportAsTfvars(params, w)
 	default:
 		err = errors.Errorf("Unsupported export format: %s", exportFormat)
 	}
@@ -120,6 +122,15 @@ func exportAsEnvFile(params map[string]string, w io.Writer) error {
 	return nil
 }
 
+func exportAsTfvars(params map[string]string, w io.Writer) error {
+	// Terraform Variables is like dotenv, but removes the TF_VAR and keeps lowercase
+	for _, k := range sortedKeys(params) {
+		key := strings.TrimPrefix(k, "tf_var_")
+		w.Write([]byte(fmt.Sprintf(`%s = "%s"`+"\n", key, doubleQuoteEscape(params[k]))))
+	}
+	return nil
+}
+
 func exportAsJson(params map[string]string, w io.Writer) error {
 	// JSON like:
 	// {"param1":"value1","param2":"value2"}