add validation for switch command (#102)

cmd/switch.go

@@ -1,40 +1,46 @@
 /*
 Copyright © 2023 NAME HERE <EMAIL ADDRESS>
-
 */
 package cmd
 
 import (
-	"fmt"
+	"errors"
 
 	"github.com/spf13/cobra"
 )
 
+var (
+	createOption string
+)
+
 // switchCmd represents the switch command
 var switchCmd = &cobra.Command{
 	Use:   "switch",
-	Short: "A brief description of your command",
-	Long: `A longer description that spans multiple lines and likely contains examples
-and usage of using your command. For example:
-
-Cobra is a CLI library for Go that empowers applications.
-This application is a tool to generate the needed files
-to quickly create a Cobra application.`,
-	Run: func(cmd *cobra.Command, args []string) {
-		fmt.Println("switch called")
+	Short: "switch branches",
+	Long:  "switch branches",
+	PreRunE: func(cmd *cobra.Command, args []string) error {
+		if client.RootGoitPath == "" {
+			return ErrGoitNotInitialized
+		}
+		return nil
+	},
+	RunE: func(cmd *cobra.Command, args []string) error {
+		// validation
+		if len(args) >= 2 {
+			return errors.New("fatal: only one reference expected")
+		}
+		if createOption == "" && len(args) == 0 {
+			return errors.New("fatal: missing branch")
+		} else if createOption != "" && len(args) >= 1 {
+			return errors.New("invalid create option format")
+		}
+
+		return nil
 	},
 }
 
 func init() {
 	rootCmd.AddCommand(switchCmd)
 
-	// Here you will define your flags and configuration settings.
-
-	// Cobra supports Persistent Flags which will work for this command
-	// and all subcommands, e.g.:
-	// switchCmd.PersistentFlags().String("foo", "", "A help for foo")
-
-	// Cobra supports local flags which will only run when this command
-	// is called directly, e.g.:
-	// switchCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
+	switchCmd.Flags().StringVarP(&createOption, "create", "c", "", "create new branch")
 }