Refactor cmd_add.go to utilize ui package for consistent output styling

This commit updates the cmd_add.go file to replace direct color formatting functions with calls to the ui package, ensuring consistent styling across the application. Additionally, it removes obsolete color functions from cmd_helpers.go, streamlining the codebase and enhancing maintainability.

Author: utkarsh5026

sourcecontrol/cmd/sourcecontrol/cmd_add.go

@@ -4,6 +4,7 @@ import (
 	"fmt"
 
 	"github.com/spf13/cobra"
+	"github.com/utkarsh5026/SourceControl/cmd/ui"
 	"github.com/utkarsh5026/SourceControl/pkg/index"
 	"github.com/utkarsh5026/SourceControl/pkg/store"
 )
@@ -16,38 +17,33 @@ func newAddCmd() *cobra.Command {
 This stages changes for the next commit.`,
 		Args: cobra.MinimumNArgs(1),
 		RunE: func(cmd *cobra.Command, args []string) error {
-			// Find repository
 			repo, err := findRepository()
 			if err != nil {
 				return err
 			}
 
-			// Create index manager
 			repoRoot := repo.WorkingDirectory()
 			indexMgr := index.NewManager(repoRoot)
 			if err := indexMgr.Initialize(); err != nil {
 				return fmt.Errorf("failed to initialize index: %w", err)
 			}
 
-			// Create object store
 			objectStore := store.NewFileObjectStore()
 			objectStore.Initialize(repo.WorkingDirectory())
 
-			// Add files
 			result, err := indexMgr.Add(args, objectStore)
 			if err != nil {
 				return fmt.Errorf("failed to add files: %w", err)
 			}
 
-			// Display results
 			for _, path := range result.Added {
-				fmt.Printf("%s %s\n", colorGreen("added:"), path)
+				fmt.Printf("%s %s\n", ui.Green("added:"), path)
 			}
 			for _, path := range result.Modified {
-				fmt.Printf("%s %s\n", colorYellow("modified:"), path)
+				fmt.Printf("%s %s\n", ui.Yellow("modified:"), path)
 			}
 			for _, failure := range result.Failed {
-				fmt.Printf("%s %s: %s\n", colorRed("failed:"), failure.Path, failure.Reason)
+				fmt.Printf("%s %s: %s\n", ui.Red("failed:"), failure.Path, failure.Reason)
 			}
 
 			return nil

sourcecontrol/cmd/sourcecontrol/cmd_helpers.go

@@ -5,7 +5,6 @@ import (
 	"os"
 	"path/filepath"
 
-	"github.com/utkarsh5026/SourceControl/cmd/ui"
 	"github.com/utkarsh5026/SourceControl/pkg/refs/branch"
 	"github.com/utkarsh5026/SourceControl/pkg/repository/scpath"
 	"github.com/utkarsh5026/SourceControl/pkg/repository/sourcerepo"
@@ -64,16 +63,3 @@ func getCurrentBranchName(repo *sourcerepo.SourceRepository) (string, error) {
 
 	return branchName, nil
 }
-
-// Wrapper functions for backward compatibility - delegate to ui package
-func colorGreen(s string) string {
-	return ui.Green(s)
-}
-
-func colorRed(s string) string {
-	return ui.Red(s)
-}
-
-func colorYellow(s string) string {
-	return ui.Yellow(s)
-}

sourcecontrol/cmd/sourcecontrol/commit.go

@@ -85,7 +85,7 @@ Displays the commit history starting from the current HEAD.`,
 			}
 
 			if len(history) == 0 {
-				fmt.Println(colorYellow("📝 No commits yet"))
+				fmt.Println(ui.Yellow("📝 No commits yet"))
 				return nil
 			}