diff --git a/internal/app/wwctl/ssh/root.go b/internal/app/wwctl/ssh/root.go index 81233e23..be6be9c9 100644 --- a/internal/app/wwctl/ssh/root.go +++ b/internal/app/wwctl/ssh/root.go @@ -1,6 +1,7 @@ package ssh import ( + "github.com/hpcng/warewulf/internal/pkg/node" "github.com/spf13/cobra" ) @@ -12,6 +13,19 @@ var ( Long: "Easily ssh into nodes in parallel to run non-interactive commands\n", RunE: CobraRunE, Args: cobra.MinimumNArgs(2), + ValidArgsFunction: func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) { + if len(args) != 0 { + return nil, cobra.ShellCompDirectiveNoFileComp + } + + nodeDB, _ := node.New() + nodes, _ := nodeDB.FindAllNodes() + var node_names []string + for _, node := range nodes { + node_names = append(node_names, node.Id.Get()) + } + return node_names, cobra.ShellCompDirectiveNoFileComp + }, } DryRun bool FanOut int