add completion for limactl disk

Signed-off-by: Nikita Vasilchenko <[email protected]>

Author: vasileknik76

cmd/limactl/completion.go

@@ -23,3 +23,11 @@ func bashCompleteTemplateNames(_ *cobra.Command) ([]string, cobra.ShellCompDirec
 	}
 	return comp, cobra.ShellCompDirectiveDefault
 }
+
+func bashCompleteDiskNames(_ *cobra.Command) ([]string, cobra.ShellCompDirective) {
+	disks, err := store.Disks()
+	if err != nil {
+		return nil, cobra.ShellCompDirectiveDefault
+	}
+	return disks, cobra.ShellCompDirectiveNoFileComp
+}

cmd/limactl/disk.go

@@ -203,10 +203,11 @@ $ limactl disk delete DISK
 To delete multiple disks:
 $ limactl disk delete DISK1 DISK2 ...
 `,
-		Aliases: []string{"remove", "rm"},
-		Short:   "Delete one or more Lima disks",
-		Args:    WrapArgsError(cobra.MinimumNArgs(1)),
-		RunE:    diskDeleteAction,
+		Aliases:           []string{"remove", "rm"},
+		Short:             "Delete one or more Lima disks",
+		Args:              WrapArgsError(cobra.MinimumNArgs(1)),
+		RunE:              diskDeleteAction,
+		ValidArgsFunction: diskBashComplete,
 	}
 	diskDeleteCommand.Flags().BoolP("force", "f", false, "force delete")
 	return diskDeleteCommand
@@ -294,9 +295,10 @@ $ limactl disk unlock DISK
 To unlock multiple disks:
 $ limactl disk unlock DISK1 DISK2 ...
 `,
-		Short: "Unlock one or more Lima disks",
-		Args:  WrapArgsError(cobra.MinimumNArgs(1)),
-		RunE:  diskUnlockAction,
+		Short:             "Unlock one or more Lima disks",
+		Args:              WrapArgsError(cobra.MinimumNArgs(1)),
+		RunE:              diskUnlockAction,
+		ValidArgsFunction: diskBashComplete,
 	}
 	return diskUnlockCommand
 }
@@ -342,9 +344,10 @@ func newDiskResizeCommand() *cobra.Command {
 		Example: `
 Resize a disk:
 $ limactl disk resize DISK --size SIZE`,
-		Short: "Resize existing Lima disk",
-		Args:  WrapArgsError(cobra.ExactArgs(1)),
-		RunE:  diskResizeAction,
+		Short:             "Resize existing Lima disk",
+		Args:              WrapArgsError(cobra.ExactArgs(1)),
+		RunE:              diskResizeAction,
+		ValidArgsFunction: diskBashComplete,
 	}
 	diskResizeCommand.Flags().String("size", "", "Disk size")
 	_ = diskResizeCommand.MarkFlagRequired("size")
@@ -390,3 +393,7 @@ func diskResizeAction(cmd *cobra.Command, args []string) error {
 	logrus.Infof("Resized disk %q (%q)", diskName, disk.Dir)
 	return nil
 }
+
+func diskBashComplete(cmd *cobra.Command, _ []string, _ string) ([]string, cobra.ShellCompDirective) {
+	return bashCompleteDiskNames(cmd)
+}