feat: add shell completion for ucind CLI (#344)

For feature parity with uc. Only _really_ usefull for ucind cluster rm,
but still useful, and not many lines of code.

Signed-off-by: Miek Gieben <miek@miek.nl>
This commit is contained in:
Miek Gieben
2026-05-04 13:42:23 +10:00
committed by GitHub
parent d23f1bdc00
commit dd68079005
2 changed files with 36 additions and 0 deletions
+7
View File
@@ -4,6 +4,7 @@ import (
"fmt" "fmt"
"github.com/psviderski/uncloud/internal/ucind" "github.com/psviderski/uncloud/internal/ucind"
"github.com/psviderski/uncloud/internal/ucind/completion"
"github.com/spf13/cobra" "github.com/spf13/cobra"
) )
@@ -26,6 +27,12 @@ func NewRemoveCommand() *cobra.Command {
fmt.Printf("Cluster '%s' removed.\n", name) fmt.Printf("Cluster '%s' removed.\n", name)
return nil return nil
}, },
ValidArgsFunction: func(cmd *cobra.Command, args []string, toComplete string) ([]cobra.Completion, cobra.ShellCompDirective) {
if len(args) > 0 {
return nil, cobra.ShellCompDirectiveNoFileComp
}
return completion.Clusters(cmd, args, toComplete)
},
} }
return cmd return cmd
} }
+29
View File
@@ -0,0 +1,29 @@
// Package completion implements completion functions for the ucind cli.
package completion
import (
"slices"
"strings"
"github.com/psviderski/uncloud/internal/ucind"
"github.com/spf13/cobra"
)
func Clusters(cmd *cobra.Command, args []string, toComplete string) ([]cobra.Completion, cobra.ShellCompDirective) {
p := cmd.Context().Value("provisioner").(*ucind.Provisioner)
clusters, err := p.ListClusters(cmd.Context())
if err != nil {
return nil, cobra.ShellCompDirectiveError
}
names := []cobra.Completion{}
for _, cluster := range clusters {
if slices.Contains(args, cluster.Name) {
continue
}
if strings.HasPrefix(cluster.Name, toComplete) {
names = append(names, cluster.Name)
}
}
return names, cobra.ShellCompDirectiveNoFileComp
}