move logic from cmd/helm to pkg/cmd

Signed-off-by: Austin Abro <AustinAbro321@gmail.com>

cmd/helm/helm.go

@@ -23,7 +23,6 @@ import (
 	// Import to initialize client auth plugins.
 	_ "k8s.io/client-go/plugin/pkg/client/auth"
 
-	"helm.sh/helm/v4/pkg/action"
 	helmcmd "helm.sh/helm/v4/pkg/cmd"
 	"helm.sh/helm/v4/pkg/kube"
 )
@@ -39,8 +38,7 @@ func main() {
 	// manager as picked up by the automated name detection.
 	kube.ManagedFieldsManager = "helm"
 
-	actionConfig := new(action.Configuration)
+	cmd, err := helmcmd.NewRootCmd(os.Stdout, os.Args[1:])
-	cmd, err := helmcmd.NewRootCmd(actionConfig, os.Stdout, os.Args[1:])
 	if err != nil {
 		helmcmd.Warning("%+v", err)
 		os.Exit(1)

pkg/cmd/helpers_test.go

@@ -95,7 +95,7 @@ func executeActionCommandStdinC(store *storage.Storage, in *os.File, cmd string)
 		Log:          func(_ string, _ ...interface{}) {},
 	}
 
-	root, err := NewRootCmd(actionConfig, buf, args)
+	root, err := newRootCmdWithConfig(actionConfig, buf, args)
 	if err != nil {
 		return nil, "", err
 	}

pkg/cmd/root.go

@@ -106,7 +106,26 @@ func Warning(format string, v ...interface{}) {
 	fmt.Fprintf(os.Stderr, "WARNING: "+format+"\n", v...)
 }
 
-func NewRootCmd(actionConfig *action.Configuration, out io.Writer, args []string) (*cobra.Command, error) {
+func NewRootCmd(out io.Writer, args []string) (*cobra.Command, error) {
+	actionConfig := new(action.Configuration)
+	cmd, err := newRootCmdWithConfig(actionConfig, out, args)
+	if err != nil {
+		return nil, err
+	}
+	cobra.OnInitialize(func() {
+		helmDriver := os.Getenv("HELM_DRIVER")
+		if err := actionConfig.Init(settings.RESTClientGetter(), settings.Namespace(), helmDriver, Debug); err != nil {
+			log.Fatal(err)
+		}
+		if helmDriver == "memory" {
+			loadReleasesInMemory(actionConfig)
+		}
+		actionConfig.SetHookOutputFunc(hookOutputWriter)
+	})
+	return cmd, nil
+}
+
+func newRootCmdWithConfig(actionConfig *action.Configuration, out io.Writer, args []string) (*cobra.Command, error) {
 	cmd := &cobra.Command{
 		Use:          "helm",
 		Short:        "The Helm package manager for Kubernetes.",
@@ -124,16 +143,6 @@ func NewRootCmd(actionConfig *action.Configuration, out io.Writer, args []string
 		},
 	}
 
-	cobra.OnInitialize(func() {
-		helmDriver := os.Getenv("HELM_DRIVER")
-		if err := actionConfig.Init(settings.RESTClientGetter(), settings.Namespace(), helmDriver, Debug); err != nil {
-			log.Fatal(err)
-		}
-		if helmDriver == "memory" {
-			loadReleasesInMemory(actionConfig)
-		}
-		actionConfig.SetHookOutputFunc(hookOutputWriter)
-	})
 	flags := cmd.PersistentFlags()
 
 	settings.AddFlags(flags)