refactor: code refactor and beta tag added to mongo command

* docs update

README.md

@@ -16,7 +16,7 @@ cbmigrate [command]
 
 ### Available Commands
 - `help` - Displays help information about any command.
-- `mongo` - Specific commands for migrating data from MongoDB.
+- `mongo` - Migrate data from MongoDB to Couchbase
 
 ### Flags
 - `-h, --help` - help for `cbmigrate`.

cmd/command.go

@@ -32,7 +32,7 @@ func Execute() {
 			Usage: "Displays the version of this tool.",
 		},
 	}
-	cmd := common.NewCommand("cbmigrate", nil, nil, "", "", flags)
+	cmd := common.NewCommand(common.CBMigrate, nil, nil, "", "", flags)
 	cmd.AddCommand(completionCommand())
 	cmd.AddCommand(mongo.GetMongoMigrateCommand())
 	cmd.RunE = func(cmd *cobra.Command, args []string) error {

cmd/common/command.go

@@ -1,20 +1,32 @@
 package common
 
 import (
+	"slices"
 	"strings"
 
 	"github.com/couchbaselabs/cbmigrate/cmd/flag"
 	"github.com/spf13/cobra"
 )
 
+type Command string
+
+const (
+	Mongo     = "mongo"
+	CBMigrate = "cbmigrate"
+)
+
+var BetaCommands = []Command{
+	Mongo,
+}
+
 // Example command example
 type Example struct {
 	Value string
 	Usage string
 }
 
 // NewCommand command constructor
-func NewCommand(name string, alias []string, examples []Example, short string, long string, flags []flag.Flag) *cobra.Command {
+func NewCommand(name Command, alias []string, examples []Example, short string, long string, flags []flag.Flag) *cobra.Command {
 
 	var example strings.Builder
 	exsLen := len(examples)
@@ -40,8 +52,17 @@ func NewCommand(name string, alias []string, examples []Example, short string, l
 	flagUsages = append(flagUsages, "[--help HELP]")
 	flagUsage := strings.Join(flagUsages, " ")
 
+	if slices.Index(BetaCommands, name) > -1 {
+		if short != "" {
+			short = "[Beta] " + short
+		}
+		if long != "" {
+			long = "[Beta] " + long
+		}
+	}
+
 	cmd := &cobra.Command{
-		Use:                   name + " " + flagUsage,
+		Use:                   string(name) + " " + flagUsage,
 		Aliases:               alias,
 		Example:               example.String(),
 		Short:                 short,

cmd/mongo/command/command.go

@@ -208,5 +208,6 @@ func NewCommand() *cobra.Command {
 			Usage: "Imports the data from mongo to couchbase.",
 		},
 	}
-	return common.NewCommand("mongo", []string{"m"}, examples, "", "", flags)
+	usage := "Migrate data from MongoDB to Couchbase"
+	return common.NewCommand(common.Mongo, []string{"m"}, examples, usage, usage, flags)
 }