Add preprocess function for apply/preview/destroy commands

Author: Liujingfang1

cmd/apply/cmdapply.go

@@ -18,6 +18,7 @@ import (
 	"sigs.k8s.io/cli-utils/cmd/printers"
 	"sigs.k8s.io/cli-utils/pkg/apply"
 	"sigs.k8s.io/cli-utils/pkg/common"
+	"sigs.k8s.io/cli-utils/pkg/inventory"
 	"sigs.k8s.io/cli-utils/pkg/manifestreader"
 	"sigs.k8s.io/cli-utils/pkg/provider"
 )
@@ -70,11 +71,12 @@ func ApplyCommand(f cmdutil.Factory, ioStreams genericclioptions.IOStreams) *cob
 }
 
 type ApplyRunner struct {
-	Command   *cobra.Command
-	ioStreams genericclioptions.IOStreams
-	Applier   *apply.Applier
-	provider  provider.Provider
-	loader    manifestreader.ManifestLoader
+	Command    *cobra.Command
+	PreProcess func(info inventory.InventoryInfo, strategy common.DryRunStrategy) (inventory.InventoryPolicy, error)
+	ioStreams  genericclioptions.IOStreams
+	Applier    *apply.Applier
+	provider   provider.Provider
+	loader     manifestreader.ManifestLoader
 
 	serverSideOptions      common.ServerSideOptions
 	output                 string
@@ -125,6 +127,13 @@ func (r *ApplyRunner) RunE(cmd *cobra.Command, args []string) error {
 		return err
 	}
 
+	if r.PreProcess != nil {
+		inventoryPolicy, err = r.PreProcess(inv, common.DryRunNone)
+		if err != nil {
+			return err
+		}
+	}
+
 	// Run the applier. It will return a channel where we can receive updates
 	// to keep track of progress and any issues.
 	if err := r.Applier.Initialize(); err != nil {

cmd/destroy/cmddestroy.go

@@ -14,6 +14,8 @@ import (
 	"sigs.k8s.io/cli-utils/cmd/flagutils"
 	"sigs.k8s.io/cli-utils/cmd/printers"
 	"sigs.k8s.io/cli-utils/pkg/apply"
+	"sigs.k8s.io/cli-utils/pkg/common"
+	"sigs.k8s.io/cli-utils/pkg/inventory"
 	"sigs.k8s.io/cli-utils/pkg/manifestreader"
 	"sigs.k8s.io/cli-utils/pkg/provider"
 )
@@ -52,11 +54,12 @@ func DestroyCommand(f cmdutil.Factory, ioStreams genericclioptions.IOStreams) *c
 
 // DestroyRunner encapsulates data necessary to run the destroy command.
 type DestroyRunner struct {
-	Command   *cobra.Command
-	ioStreams genericclioptions.IOStreams
-	Destroyer *apply.Destroyer
-	provider  provider.Provider
-	loader    manifestreader.ManifestLoader
+	Command    *cobra.Command
+	PreProcess func(info inventory.InventoryInfo, strategy common.DryRunStrategy) (inventory.InventoryPolicy, error)
+	ioStreams  genericclioptions.IOStreams
+	Destroyer  *apply.Destroyer
+	provider   provider.Provider
+	loader     manifestreader.ManifestLoader
 
 	output          string
 	inventoryPolicy string
@@ -82,6 +85,13 @@ func (r *DestroyRunner) RunE(cmd *cobra.Command, args []string) error {
 		return err
 	}
 
+	if r.PreProcess != nil {
+		inventoryPolicy, err = r.PreProcess(inv, r.Destroyer.DryRunStrategy)
+		if err != nil {
+			return err
+		}
+	}
+
 	// Run the destroyer. It will return a channel where we can receive updates
 	// to keep track of progress and any issues.
 	err = r.Destroyer.Initialize()

cmd/preview/cmdpreview.go

@@ -12,11 +12,13 @@ import (
 	"k8s.io/cli-runtime/pkg/genericclioptions"
 	cmdutil "k8s.io/kubectl/pkg/cmd/util"
 	"k8s.io/kubectl/pkg/util/i18n"
+
 	"sigs.k8s.io/cli-utils/cmd/flagutils"
 	"sigs.k8s.io/cli-utils/cmd/printers"
 	"sigs.k8s.io/cli-utils/pkg/apply"
 	"sigs.k8s.io/cli-utils/pkg/apply/event"
 	"sigs.k8s.io/cli-utils/pkg/common"
+	"sigs.k8s.io/cli-utils/pkg/inventory"
 	"sigs.k8s.io/cli-utils/pkg/manifestreader"
 	"sigs.k8s.io/cli-utils/pkg/provider"
 )
@@ -70,12 +72,13 @@ func PreviewCommand(f cmdutil.Factory, ioStreams genericclioptions.IOStreams) *c
 
 // PreviewRunner encapsulates data necessary to run the preview command.
 type PreviewRunner struct {
-	Command   *cobra.Command
-	ioStreams genericclioptions.IOStreams
-	Applier   *apply.Applier
-	Destroyer *apply.Destroyer
-	provider  provider.Provider
-	loader    manifestreader.ManifestLoader
+	Command    *cobra.Command
+	PreProcess func(info inventory.InventoryInfo, strategy common.DryRunStrategy) (inventory.InventoryPolicy, error)
+	ioStreams  genericclioptions.IOStreams
+	Applier    *apply.Applier
+	Destroyer  *apply.Destroyer
+	provider   provider.Provider
+	loader     manifestreader.ManifestLoader
 
 	serverSideOptions common.ServerSideOptions
 	output            string
@@ -114,6 +117,13 @@ func (r *PreviewRunner) RunE(cmd *cobra.Command, args []string) error {
 		return err
 	}
 
+	if r.PreProcess != nil {
+		inventoryPolicy, err = r.PreProcess(inv, drs)
+		if err != nil {
+			return err
+		}
+	}
+
 	// if destroy flag is set in preview, transmit it to destroyer DryRunStrategy flag
 	// and pivot execution to destroy with dry-run
 	if !r.Destroyer.DryRunStrategy.ClientOrServerDryRun() {