fix: improve version and help flag handling with proper Cobra integration

Author: rajamohan-rj

cmd/kdiff/main.go

@@ -48,15 +48,7 @@ Prerequisites:
   • kubectl command-line tool
   • kubectl neat plugin (optional, for cleaner YAML output)
   • colordiff (optional, for colored output)`,
-	Args: func(cmd *cobra.Command, args []string) error {
-		// Allow version flag to bypass argument validation
-		showVersion, _ := cmd.Flags().GetBool("version")
-		if showVersion {
-			fmt.Printf("kdiff %s (commit %s, built %s)\n", version, commit, date)
-			os.Exit(0)
-		}
-		return cobra.ExactArgs(4)(cmd, args)
-	},
+	Args: cobra.ExactArgs(4),
 	Run: func(cmd *cobra.Command, args []string) {
 		opts.Namespace1 = args[0]
 		opts.Resource1 = args[1]
@@ -70,12 +62,27 @@ Prerequisites:
 	},
 }
 
+var versionCmd = &cobra.Command{
+	Use:   "version",
+	Short: "Print the version number",
+	Run: func(cmd *cobra.Command, args []string) {
+		fmt.Printf("kdiff %s (commit %s, built %s)\n", version, commit, date)
+	},
+}
+
 func init() {
 	rootCmd.Flags().BoolVar(&opts.NoColor, "no-color", false, "disable colored output (useful for CI/CD pipelines)")
 	rootCmd.Flags().BoolVar(&opts.NoNeat, "no-neat", false, "skip kubectl neat processing (show raw YAML)")
 	rootCmd.Flags().StringVar(&opts.OutputFormat, "output", "unified", "output format: unified, context, side-by-side")
 	rootCmd.Flags().BoolVar(&opts.Verbose, "verbose", false, "enable verbose logging for debugging")
-	rootCmd.Flags().Bool("version", false, "show version information")
+	
+	// Add version flag that shows version and exits
+	rootCmd.Flags().BoolP("version", "v", false, "show version information")
+	rootCmd.SetVersionTemplate("kdiff {{.Version}} (commit " + commit + ", built " + date + ")\n")
+	rootCmd.Version = version
+	
+	// Add version subcommand
+	rootCmd.AddCommand(versionCmd)
 }
 
 // Execute runs the root command - called from main.go