SetupCommonFlags

Author: lantoli

internal/cli/adv2new/adv2new.go

@@ -3,7 +3,6 @@ package adv2new
 import (
 	"github.com/mongodb-labs/atlas-cli-plugin-terraform/internal/cli"
 	"github.com/mongodb-labs/atlas-cli-plugin-terraform/internal/convert"
-	"github.com/mongodb-labs/atlas-cli-plugin-terraform/internal/flag"
 	"github.com/spf13/afero"
 	"github.com/spf13/cobra"
 )
@@ -26,13 +25,6 @@ func Builder() *cobra.Command {
 			return o.Run()
 		},
 	}
-	cmd.Flags().StringVarP(&o.File, flag.File, flag.FileShort, "", "input file")
-	_ = cmd.MarkFlagRequired(flag.File)
-	cmd.Flags().StringVarP(&o.Output, flag.Output, flag.OutputShort, "", "output file")
-	_ = cmd.MarkFlagRequired(flag.Output)
-	cmd.Flags().BoolVarP(&o.ReplaceOutput, flag.ReplaceOutput, flag.ReplaceOutputShort, false,
-		"replace output file if exists")
-	cmd.Flags().BoolVarP(&o.Watch, flag.Watch, flag.WatchShort, false,
-		"keeps the plugin running and watches the input file for changes")
+	cli.SetupCommonFlags(cmd, o)
 	return cmd
 }

internal/cli/clu2adv/clu2adv.go

@@ -33,14 +33,7 @@ func Builder() *cobra.Command {
 			return o.Run()
 		},
 	}
-	cmd.Flags().StringVarP(&o.File, flag.File, flag.FileShort, "", "input file")
-	_ = cmd.MarkFlagRequired(flag.File)
-	cmd.Flags().StringVarP(&o.Output, flag.Output, flag.OutputShort, "", "output file")
-	_ = cmd.MarkFlagRequired(flag.Output)
-	cmd.Flags().BoolVarP(&o.ReplaceOutput, flag.ReplaceOutput, flag.ReplaceOutputShort, false,
-		"replace output file if exists")
-	cmd.Flags().BoolVarP(&o.Watch, flag.Watch, flag.WatchShort, false,
-		"keeps the plugin running and watches the input file for changes")
+	cli.SetupCommonFlags(cmd, o.BaseOpts)
 	cmd.Flags().BoolVarP(&o.includeMoved, flag.IncludeMoved, flag.IncludeMovedShort, false,
 		"include moved blocks in the output file")
 	return cmd

internal/cli/common.go

@@ -6,7 +6,9 @@ import (
 
 	"github.com/fsnotify/fsnotify"
 	"github.com/mongodb-labs/atlas-cli-plugin-terraform/internal/file"
+	"github.com/mongodb-labs/atlas-cli-plugin-terraform/internal/flag"
 	"github.com/spf13/afero"
+	"github.com/spf13/cobra"
 )
 
 type ConvertFn func(config []byte) ([]byte, error)
@@ -106,3 +108,15 @@ func (o *BaseOpts) waitForFileEvent(watcher *fsnotify.Watcher) error {
 	}
 	return nil
 }
+
+// SetupCommonFlags sets up the common flags used by all commands.
+func SetupCommonFlags(cmd *cobra.Command, opts *BaseOpts) {
+	cmd.Flags().StringVarP(&opts.File, flag.File, flag.FileShort, "", "input file")
+	_ = cmd.MarkFlagRequired(flag.File)
+	cmd.Flags().StringVarP(&opts.Output, flag.Output, flag.OutputShort, "", "output file")
+	_ = cmd.MarkFlagRequired(flag.Output)
+	cmd.Flags().BoolVarP(&opts.ReplaceOutput, flag.ReplaceOutput, flag.ReplaceOutputShort, false,
+		"replace output file if exists")
+	cmd.Flags().BoolVarP(&opts.Watch, flag.Watch, flag.WatchShort, false,
+		"keeps the plugin running and watches the input file for changes")
+}