Add flagcompletion to --container flag

Signed-off-by: Hadi Chokr <[email protected]>

Author: silverhadch

src/cmd/export.go

@@ -38,12 +38,19 @@ var exportCmd = &cobra.Command{
 	Use:   "export",
 	Short: "Export binaries or applications from a toolbox container",
 	RunE:  runExport,
+	ValidArgsFunction: completionContainerNamesFiltered,
 }
 
 func init() {
 	exportCmd.Flags().StringVar(&exportBin, "bin", "", "Path or name of binary to export")
 	exportCmd.Flags().StringVar(&exportApp, "app", "", "Path or name of application to export")
 	exportCmd.Flags().StringVar(&exportContainer, "container", "", "Name of the toolbox container")
+
+	// Register container flag completion
+	if err := exportCmd.RegisterFlagCompletionFunc("container", completionContainerNames); err != nil {
+		panic(fmt.Sprintf("failed to register flag completion function: %v", err))
+	}
+
 	exportCmd.SetHelpFunc(exportHelp)
 	rootCmd.AddCommand(exportCmd)
 }

src/cmd/unexport.go

@@ -38,13 +38,20 @@ var unexportCmd = &cobra.Command{
 	Use:   "unexport",
 	Short: "Remove exported binaries and applications for a specific toolbox container",
 	RunE:  runUnexport,
+	ValidArgsFunction: completionContainerNamesFiltered,
 }
 
 func init() {
 	unexportCmd.Flags().StringVar(&unexportContainer, "container", "", "Name of the toolbox container")
 	unexportCmd.Flags().StringVar(&unexportBin, "bin", "", "Name of the exported binary to remove")
 	unexportCmd.Flags().StringVar(&unexportApp, "app", "", "Name of the exported application to remove")
 	unexportCmd.Flags().BoolVar(&unexportAll, "all", false, "Remove all exported binaries and applications for the container")
+
+	// Register container flag completion
+	if err := unexportCmd.RegisterFlagCompletionFunc("container", completionContainerNames); err != nil {
+		panic(fmt.Sprintf("failed to register flag completion function: %v", err))
+	}
+
 	unexportCmd.SetHelpFunc(unexportHelp)
 	rootCmd.AddCommand(unexportCmd)
 }