fix -f flag for etcd alarm list / disarm commands

cozystack · May 20, 2024 · 7b3fe03 · 7b3fe03
1 parent d752a63
commit 7b3fe03
Show file tree

Hide file tree

Showing 2 changed files with 46 additions and 26 deletions.
diff --git a/pkg/commands/imported_etcd.go b/pkg/commands/imported_etcd.go
diff --git a/tools/import_commands.go b/tools/import_commands.go
@@ -214,6 +214,15 @@ func processFile(filename, cmdName string) {
 		}
 	}
 
+	if cmdName == "etcd" {
+		for _, subCmdName := range []string{"etcdAlarmListCmd", "etcdAlarmDisarmCmd"} {
+			initCode = fmt.Sprintf("%s\n%s", initCode, fmt.Sprintf(`
+	%s.Flags().StringSliceVarP(&etcdCmdFlags.configFiles, "file", "f", nil, "specify config files or patches in a YAML file (can specify multiple)")
+	%s.PreRunE = etcdCmd.PreRunE
+		`, subCmdName, subCmdName))
+		}
+	}
+
 	insertInitCode(node, cmdName, initCode)
 
 	var buf bytes.Buffer