Add interactive stale branch pruning with PIN confirmation

Enhances the git prune stale command to interactively display stale branches, allow users to select which branches to delete, and require PIN confirmation before deletion. Adds helpers for generating a random PIN and multi-select prompts, with corresponding tests for PIN generation.

Author: YoungMayor

internal/commands/git.go

@@ -62,9 +62,9 @@ var GitPruneStaleCmd = &cobra.Command{
 		}
 		currentBranch := strings.TrimSpace(string(currentOutput))
 
-		// Parse local branches
+		// Parse local branches and identify stale branches
 		localBranches := strings.Split(string(localOutput), "\n")
-		deletedCount := 0
+		var staleBranches []string
 
 		for _, branch := range localBranches {
 			branch = strings.TrimSpace(branch)
@@ -87,24 +87,99 @@ var GitPruneStaleCmd = &cobra.Command{
 				continue
 			}
 
-			// If branch doesn't exist on remote, delete it
+			// If branch doesn't exist on remote, add to stale list
 			if strings.TrimSpace(string(remoteOutput)) == "" {
-				fmt.Printf("Deleting stale branch: %s\n", branch)
-				deleteCmd := exec.Command("git", "branch", "-D", branch)
-				if err := deleteCmd.Run(); err != nil {
-					fmt.Printf("⚠️  Failed to delete %s: %v\n", branch, err)
-				} else {
-					deletedCount++
-				}
+				staleBranches = append(staleBranches, branch)
 			}
 		}
 
-		if deletedCount == 0 {
+		// If no stale branches, exit early
+		if len(staleBranches) == 0 {
 			fmt.Println("✅ No stale branches found!")
-		} else {
-			fmt.Printf("✅ Deleted %d stale branch(es)!\n", deletedCount)
+			return nil
+		}
+
+		// Display stale branches count
+		fmt.Printf("\n⚠️  Found %d stale branch(es):\n", len(staleBranches))
+		for _, branch := range staleBranches {
+			fmt.Printf("   - %s\n", branch)
+		}
+		fmt.Println()
+
+		// Stern warning
+		fmt.Println("⚠️  WARNING: This is a DESTRUCTIVE action that CANNOT be reversed!")
+		fmt.Println("⚠️  Deleted branches cannot be recovered unless backed up elsewhere.")
+		fmt.Println()
+
+		// Prompt user for action
+		action, err := utils.PromptSelect(
+			"How would you like to proceed?",
+			[]string{"Prune All", "Select Branches to Prune", "Abort"},
+		)
+		if err != nil {
+			return err
 		}
 
+		var branchesToDelete []string
+
+		switch action {
+		case "Abort":
+			fmt.Println("❌ Operation aborted. No branches were deleted.")
+			return nil
+
+		case "Prune All":
+			branchesToDelete = staleBranches
+
+		case "Select Branches to Prune":
+			selected, err := utils.PromptMultiSelect(
+				"Select branches to delete:",
+				staleBranches,
+			)
+			if err != nil {
+				return err
+			}
+
+			if len(selected) == 0 {
+				fmt.Println("❌ No branches selected. Operation aborted.")
+				return nil
+			}
+
+			branchesToDelete = selected
+		}
+
+		// Generate PIN for confirmation
+		pin, err := utils.GeneratePIN()
+		if err != nil {
+			return fmt.Errorf("failed to generate PIN: %w", err)
+		}
+
+		// Display PIN and ask for confirmation
+		fmt.Printf("\n🔐 To confirm deletion of %d branch(es), enter this PIN: %s\n", len(branchesToDelete), pin)
+		userPIN, err := utils.SurveyInput("Enter PIN to confirm", "")
+		if err != nil {
+			return err
+		}
+
+		if userPIN != pin {
+			fmt.Println("❌ Incorrect PIN. Operation aborted. No branches were deleted.")
+			return nil
+		}
+
+		// Delete selected branches
+		fmt.Println("\n🗑️  Deleting branches...")
+		deletedCount := 0
+		for _, branch := range branchesToDelete {
+			fmt.Printf("   Deleting: %s\n", branch)
+			deleteCmd := exec.Command("git", "branch", "-D", branch)
+			if err := deleteCmd.Run(); err != nil {
+				fmt.Printf("   ⚠️  Failed to delete %s: %v\n", branch, err)
+			} else {
+				deletedCount++
+			}
+		}
+
+		fmt.Printf("\n✅ Successfully deleted %d branch(es)!\n", deletedCount)
+
 		return nil
 	},
 }

internal/utils/helpers.go

@@ -205,3 +205,24 @@ func SurveyConfirm(message string, defaultValue bool) (bool, error) {
 	err := survey.AskOne(prompt, &result)
 	return result, err
 }
+
+// GeneratePIN generates a random 6-digit PIN
+func GeneratePIN() (string, error) {
+	num, err := rand.Int(rand.Reader, big.NewInt(1000000))
+	if err != nil {
+		return "", err
+	}
+	// Ensure 6 digits with leading zeros if needed
+	return fmt.Sprintf("%06d", num.Int64()), nil
+}
+
+// PromptMultiSelect prompts the user to select multiple items from a list
+func PromptMultiSelect(message string, options []string) ([]string, error) {
+	var result []string
+	prompt := &survey.MultiSelect{
+		Message: message,
+		Options: options,
+	}
+	err := survey.AskOne(prompt, &result)
+	return result, err
+}

internal/utils/helpers_test.go

@@ -239,3 +239,52 @@ func TestGeneratePasswordByType(t *testing.T) {
 		})
 	}
 }
+
+func TestGeneratePIN(t *testing.T) {
+	// Test that PIN is generated without error
+	pin, err := GeneratePIN()
+	if err != nil {
+		t.Errorf("GeneratePIN() error = %v", err)
+		return
+	}
+
+	// Test that PIN is exactly 6 characters
+	if len(pin) != 6 {
+		t.Errorf("GeneratePIN() length = %v, want 6", len(pin))
+	}
+
+	// Test that PIN contains only digits
+	for _, c := range pin {
+		if c < '0' || c > '9' {
+			t.Errorf("GeneratePIN() contains non-digit character: %c", c)
+		}
+	}
+
+	// Test that multiple calls generate different PINs (with high probability)
+	pin2, err := GeneratePIN()
+	if err != nil {
+		t.Errorf("GeneratePIN() second call error = %v", err)
+		return
+	}
+
+	// Generate several more PINs to check randomness
+	// Note: There's a tiny chance all could be the same, but highly unlikely
+	allSame := pin == pin2
+	if allSame {
+		// Try a few more times
+		for i := 0; i < 5; i++ {
+			pin3, err := GeneratePIN()
+			if err != nil {
+				t.Errorf("GeneratePIN() call %d error = %v", i+3, err)
+				return
+			}
+			if pin3 != pin {
+				allSame = false
+				break
+			}
+		}
+		if allSame {
+			t.Errorf("GeneratePIN() appears to not be random - all 7 calls returned same PIN: %s", pin)
+		}
+	}
+}