Added overlay build option to wwctl

internal/app/wwctl/overlay/build/main.go

@@ -12,28 +12,36 @@ import (
 
 func CobraRunE(cmd *cobra.Command, args []string) error {
 
-	nodeDB, err := node.New()
-	if err != nil {
-		wwlog.Printf(wwlog.ERROR, "Could not open node configuration: %s\n", err)
-		os.Exit(1)
+	if BuildHost || (!BuildHost && !BuildNodes) {
+		err := overlay.BuildHostOverlay()
+		if err != nil {
+			wwlog.Printf(wwlog.WARN, "host overlay could not be built: %s\n", err)
+		}
 	}
+	if BuildNodes || (!BuildHost && !BuildNodes) {
+		nodeDB, err := node.New()
+		if err != nil {
+			wwlog.Printf(wwlog.ERROR, "Could not open node configuration: %s\n", err)
+			os.Exit(1)
+		}
 
-	nodes, err := nodeDB.FindAllNodes()
-	if err != nil {
-		wwlog.Printf(wwlog.ERROR, "Could not get node list: %s\n", err)
-		os.Exit(1)
+		nodes, err := nodeDB.FindAllNodes()
+		if err != nil {
+			wwlog.Printf(wwlog.ERROR, "Could not get node list: %s\n", err)
+			os.Exit(1)
+		}
+
+		if len(args) > 0 {
+			args = hostlist.Expand(args)
+			err = overlay.BuildAllOverlays(node.FilterByName(nodes, args))
+		} else {
+			err = overlay.BuildAllOverlays(nodes)
+		}
+
+		if err != nil {
+			wwlog.Printf(wwlog.WARN, "Some system overlays failed to be generated: %s\n", err)
+
+		}
 	}
-
-	if len(args) > 0 {
-		args = hostlist.Expand(args)
-		err = overlay.BuildAllOverlays(node.FilterByName(nodes, args))
-	} else {
-		err = overlay.BuildAllOverlays(nodes)
-	}
-
-	if err != nil {
-		wwlog.Printf(wwlog.WARN, "Some system overlays failed to be generated: %s\n", err)
-	}
-
 	return nil
 }

internal/app/wwctl/overlay/build/root.go

@@ -10,10 +10,13 @@ var (
 		Long:                  "This command builds overlays for given nodes.",
 		RunE:                  CobraRunE,
 	}
+	BuildHost  bool
+	BuildNodes bool
 )
 
 func init() {
-	//baseCmd.PersistentFlags().BoolVarP(&BuildAll, "all", "a", false, "Build overlays for all nodes")
+	baseCmd.PersistentFlags().BoolVarP(&BuildHost, "host", "H", false, "Build overlays only for the host")
+	baseCmd.PersistentFlags().BoolVarP(&BuildNodes, "nodes", "N", false, "Build overlays only for the nodes")
 }
 
 // GetRootCommand returns the root cobra.Command for the application.