First pass at a rework of the overlay subsystem in Warewulf

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

@@ -5,11 +5,10 @@ import (
 	"path"
 	"strconv"
 
-	"github.com/hpcng/warewulf/internal/pkg/config"
+	"github.com/hpcng/warewulf/internal/pkg/overlay"
 	"github.com/hpcng/warewulf/internal/pkg/util"
 	"github.com/hpcng/warewulf/internal/pkg/wwlog"
 
-	"github.com/pkg/errors"
 	"github.com/spf13/cobra"
 )
 
@@ -19,13 +18,8 @@ func CobraRunE(cmd *cobra.Command, args []string) error {
 	var gid int
 	var err error
 
-	overlayKind := args[0]
-	overlayName := args[1]
-	fileName := args[2]
-
-	if overlayKind != "system" && overlayKind != "runtime" {
-		return errors.New("overlay kind must be of type 'system' or 'runtime'")
-	}
+	overlayName := args[0]
+	fileName := args[1]
 
 	uid, err = strconv.Atoi(args[3])
 	if err != nil {
@@ -43,11 +37,7 @@ func CobraRunE(cmd *cobra.Command, args []string) error {
 		gid = 0
 	}
 
-	if overlayKind == "system" {
-		overlaySourceDir = config.SystemOverlaySource(overlayName)
-	} else if overlayKind == "runtime" {
-		overlaySourceDir = config.RuntimeOverlaySource(overlayName)
-	}
+	overlaySourceDir = overlay.OverlaySourceDir(overlayName)
 
 	if !util.IsDir(overlaySourceDir) {
 		wwlog.Printf(wwlog.ERROR, "Overlay does not exist: %s\n", overlayName)