Implement --binary to set path to alternative kustomize

puzzle · Apr 8, 2024 · f1b3511 · f1b3511
1 parent 8a59629
commit f1b3511
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 4 deletions.
diff --git a/cmd/kustomize.go b/cmd/kustomize.go
@@ -12,6 +12,7 @@ import (
 
 var outputDotDir *string
 var version *bool
+var binary *string
 
 var kustomizeCmd = &cobra.Command{
 	Use:   "kustomize [rootDir]",
@@ -24,8 +25,13 @@ var kustomizeCmd = &cobra.Command{
 	},
 	Long: `Generate a DOT file to visualize the dependencies between your kustomize components`,
 	RunE: func(cmd *cobra.Command, args []string) error {
+		kustomizeCmd := "kustomize"
+		if *binary != "" {
+			kustomizeCmd = *binary
+		}
+
 		if *version {
-			kustomizeCmd := exec.CommandContext(cmd.Context(), "kustomize", "version")
+			kustomizeCmd := exec.CommandContext(cmd.Context(), kustomizeCmd, "version")
 			kustomizeCmd.Stdout = cmd.OutOrStdout()
 			kustomizeCmd.Stderr = cmd.OutOrStderr()
 			if err := kustomizeCmd.Run(); err != nil {
@@ -43,6 +49,7 @@ func init() {
 	kustomizeCmd.AddCommand(kustomize.KustomizeBuildCmd)
 	rootCmd.AddCommand(kustomizeCmd)
 
+	binary = kustomizeCmd.Flags().String("binary", "", "Alternative kustomize binary")
 	version = kustomizeCmd.Flags().BoolP("version", "v", false, "Display version of kustomize")
 	outputDotDir = kustomizeCmd.Flags().StringP("output-dir", "o", ".", "Output directory")
 }
diff --git a/cmd/kustomize/build.go b/cmd/kustomize/build.go
@@ -20,7 +20,11 @@ var KustomizeBuildCmd = &cobra.Command{
 	Args:  cobra.ExactArgs(1),
 	Long:  `Build all kustomize file within parent directory`,
 	RunE: func(cmd *cobra.Command, args []string) error {
-		return kustomize.BuildAll(args[0], *outputBuildDir)
+		kustomizeCommand, err := cmd.Flags().GetString("binary")
+		if err != nil {
+			return err
+		}
+		return kustomize.BuildAll(kustomizeCommand, args[0], *outputBuildDir)
 	},
 }
 

diff --git a/kustomize/kustomize.go b/kustomize/kustomize.go
@@ -12,7 +12,7 @@ import (
 	"github.com/puzzle/goff/util"
 )
 
-func BuildAll(sourceDir, targetDir string) error {
+func BuildAll(kustomizeCommand, sourceDir, targetDir string) error {
 
 	dirs, err := kustomizationfile.New().GetDirectories(sourceDir)
 	if err != nil {
@@ -27,7 +27,7 @@ func BuildAll(sourceDir, targetDir string) error {
 		var stdout strings.Builder
 
 		// TODO: Make customizable
-		cmd := exec.Command("kustomize", "build", absoluteKustomizationPath)
+		cmd := exec.Command(kustomizeCommand, "build", absoluteKustomizationPath)
 		cmd.Stdout = &stdout
 		if err := cmd.Run(); err != nil {
 			return fmt.Errorf("error running kustomize: %w", err)