dynamic nouns for 'wwctl power'

internal/app/wwctl/power/off/root.go

@@ -1,6 +1,7 @@
 package poweroff
 
 import (
+	"github.com/hpcng/warewulf/internal/pkg/node"
 	"github.com/spf13/cobra"
 )
 
@@ -10,6 +11,19 @@ var (
 		Short: "Power off the given node(s)",
 		Long:  "This command will shutdown the power to a given set of nodes.",
 		RunE:  CobraRunE,
+		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
+		},
 	}
 )
 

internal/app/wwctl/power/on/root.go

@@ -1,6 +1,7 @@
 package poweron
 
 import (
+	"github.com/hpcng/warewulf/internal/pkg/node"
 	"github.com/spf13/cobra"
 )
 
@@ -10,6 +11,19 @@ var (
 		Short: "Power on the given node(s)",
 		Long:  "This command will power on a given set of nodes.",
 		RunE:  CobraRunE,
+		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
+		},
 	}
 )
 

internal/app/wwctl/power/reset/root.go

@@ -1,6 +1,7 @@
 package powerreset
 
 import (
+	"github.com/hpcng/warewulf/internal/pkg/node"
 	"github.com/spf13/cobra"
 )
 
@@ -10,6 +11,19 @@ var (
 		Short: "Issue a reset to the given node(s)",
 		Long:  "This command will issue a reset to the given set of nodes.",
 		RunE:  CobraRunE,
+		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
+		},
 	}
 )
 

internal/app/wwctl/power/soft/root.go

@@ -1,6 +1,7 @@
 package powersoft
 
 import (
+	"github.com/hpcng/warewulf/internal/pkg/node"
 	"github.com/spf13/cobra"
 )
 
@@ -10,6 +11,19 @@ var (
 		Short: "Gracefully shuts down the given node(s)",
 		Long:  "This command will gracefully shutdown the given set of nodes.",
 		RunE:  CobraRunE,
+		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
+		},
 	}
 )
 

internal/app/wwctl/power/status/root.go

@@ -1,6 +1,7 @@
 package powerstatus
 
 import (
+	"github.com/hpcng/warewulf/internal/pkg/node"
 	"github.com/spf13/cobra"
 )
 
@@ -10,6 +11,19 @@ var (
 		Short: "Show power status for the given node(s)",
 		Long:  "This command will show the power status of a given set of nodes.",
 		RunE:  CobraRunE,
+		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
+		},
 	}
 )