diff options
| author | Paul Buetow <paul@buetow.org> | 2025-06-24 00:26:05 +0300 |
|---|---|---|
| committer | Paul Buetow <paul@buetow.org> | 2025-06-24 00:26:05 +0300 |
| commit | 16113b76309dcbae1a91f8420a0bbf10863c9675 (patch) | |
| tree | 243b2db64f1a64e2f89deda6eae0f052909709dc /internal/sync/sync.go | |
| parent | e637f4fbb06b1c0661d2e77ce79d0d5149ac5c47 (diff) | |
refactor: break down large functions into smaller, focused ones
Major refactoring to improve code maintainability:
1. Split main.go (481 lines → 72 lines) into internal/cli package:
- flags.go: Command-line flag definitions and parsing
- handlers.go: General command handlers (version, config, list operations)
- sync_handlers.go: Sync-specific handlers for all sync operations
2. Refactored sync.go to extract logic into separate files:
- git_operations.go: Git command helpers (merge, push, fetch, etc.)
- repository_setup.go: Repository initialization and remote configuration
- branch_sync.go: Branch synchronization helpers
3. Reduced function sizes to meet 30-line guideline:
- syncBranch: 104 lines → 26 lines
- SyncRepository: 97 lines → 44 lines
- main(): 465 lines → 63 lines
- getAllBranches: 32 lines → 9 lines
All functionality remains the same, but the code is now more modular,
testable, and easier to understand. Each function has a single, clear
responsibility.
🤖 Generated with [Claude Code](https://claude.ai/code)
Co-Authored-By: Claude <noreply@anthropic.com>
Diffstat (limited to 'internal/sync/sync.go')
| -rw-r--r-- | internal/sync/sync.go | 268 |
1 files changed, 52 insertions, 216 deletions
diff --git a/internal/sync/sync.go b/internal/sync/sync.go index e34a142..c7f17e0 100644 --- a/internal/sync/sync.go +++ b/internal/sync/sync.go @@ -35,71 +35,18 @@ func (s *Syncer) SyncRepository(repoName string) error { return fmt.Errorf("failed to create work directory: %w", err) } - // Get all remotes - remotes := make(map[string]*config.Organization) - for i := range s.config.Organizations { - org := &s.config.Organizations[i] - remoteName := s.getRemoteName(org) - remotes[remoteName] = org - } - - // Clone or update the repository + // Setup repository (clone or ensure remotes are configured) repoPath := filepath.Join(s.workDir, repoName) - if _, err := os.Stat(repoPath); os.IsNotExist(err) { - // Clone from the first organization - if len(s.config.Organizations) == 0 { - return fmt.Errorf("no organizations configured") - } - - firstOrg := &s.config.Organizations[0] - if err := s.cloneRepository(firstOrg, repoPath); err != nil { - return fmt.Errorf("failed to clone repository: %w", err) - } - - // Rename origin to the proper remote name - firstRemoteName := s.getRemoteName(firstOrg) - cmd := exec.Command("git", "-C", repoPath, "remote", "rename", "origin", firstRemoteName) - if err := cmd.Run(); err != nil { - return fmt.Errorf("failed to rename origin remote: %w", err) - } - - // Add other organizations as remotes - for i := 1; i < len(s.config.Organizations); i++ { - org := &s.config.Organizations[i] - if err := s.addRemote(repoPath, org); err != nil { - return fmt.Errorf("failed to add remote %s: %w", s.getRemoteName(org), err) - } - } - } else { - // Repository exists, ensure all remotes are configured - fmt.Printf("Using existing repository at %s\n", repoPath) - - // Check and add any missing remotes - for i := range s.config.Organizations { - org := &s.config.Organizations[i] - remoteName := s.getRemoteName(org) - - // Check if remote exists - cmd := exec.Command("git", "-C", repoPath, "remote", "get-url", remoteName) - if err := cmd.Run(); err != nil { - // Remote doesn't exist, add it - if err := s.addRemote(repoPath, org); err != nil { - return fmt.Errorf("failed to add remote %s: %w", remoteName, err) - } - } - } + if err := s.setupRepository(repoPath); err != nil { + return err } // Change to repository directory - originalDir, err := os.Getwd() + restoreDir, err := changeToRepoDirectory(repoPath) if err != nil { - return fmt.Errorf("failed to get current directory: %w", err) - } - defer os.Chdir(originalDir) - - if err := os.Chdir(repoPath); err != nil { - return fmt.Errorf("failed to change to repository directory: %w", err) + return err } + defer restoreDir() // Fetch all remotes fmt.Printf("Fetching updates from all remotes...\n") @@ -113,12 +60,12 @@ func (s *Syncer) SyncRepository(repoName string) error { return fmt.Errorf("failed to get branches: %w", err) } - // Sync each branch - for _, branch := range branches { - fmt.Printf("\nSyncing branch: %s\n", branch) - if err := s.syncBranch(branch, remotes); err != nil { - return fmt.Errorf("failed to sync branch %s: %w", branch, err) - } + // Get remotes map + remotes := s.getRemotesMap() + + // Sync all branches + if err := s.syncAllBranches(branches, remotes); err != nil { + return err } fmt.Printf("\nRepository %s synchronized successfully!\n", repoName) @@ -175,39 +122,16 @@ func (s *Syncer) addRemote(repoPath string, org *config.Organization) error { // fetchAll fetches from all remotes // Note: We use individual fetches instead of --all to handle missing repositories gracefully func (s *Syncer) fetchAll() error { - // First, check which remotes actually exist - cmd := exec.Command("git", "remote", "-v") - output, err := cmd.Output() + // Get list of remotes + remotes, err := getRemotesList() if err != nil { - return fmt.Errorf("failed to list remotes: %w", err) - } - - // Try to fetch from each remote individually to handle missing repos - remotes := make(map[string]bool) - lines := strings.Split(string(output), "\n") - for _, line := range lines { - if line == "" { - continue - } - parts := strings.Fields(line) - if len(parts) >= 1 { - remotes[parts[0]] = true - } + return err } // Fetch from each remote for remote := range remotes { - fmt.Printf("Fetching %s\n", remote) - cmd := exec.Command("git", "fetch", remote, "--prune") - output, err := cmd.CombinedOutput() - if err != nil { - // Check if it's because the repository doesn't exist - if strings.Contains(string(output), "does not appear to be a git repository") || - strings.Contains(string(output), "Could not read from remote repository") { - fmt.Printf(" Warning: Remote repository %s does not exist yet\n", remote) - continue - } - return fmt.Errorf("failed to fetch from %s: %w\n%s", remote, err, string(output)) + if err := fetchRemote(remote); err != nil { + return err } } @@ -222,57 +146,18 @@ func (s *Syncer) getAllBranches() ([]string, error) { return nil, err } - branchMap := make(map[string]bool) - lines := strings.Split(string(output), "\n") - - for _, line := range lines { - line = strings.TrimSpace(line) - if line == "" || strings.Contains(line, "->") { - continue - } - - // Extract branch name from remote/branch format - parts := strings.SplitN(line, "/", 2) - if len(parts) == 2 { - branch := parts[1] - branchMap[branch] = true - } - } - - // Convert map to slice - branches := make([]string, 0, len(branchMap)) - for branch := range branchMap { - branches = append(branches, branch) - } - - return branches, nil + return getAllUniqueBranches(output), nil } // syncBranch synchronizes a specific branch across all remotes func (s *Syncer) syncBranch(branch string, remotes map[string]*config.Organization) error { - // First check if we have unresolved merge conflicts - cmd := exec.Command("git", "status", "--porcelain") - output, err := cmd.Output() - if err == nil && len(output) > 0 { - // Check for merge conflicts - statusStr := string(output) - if strings.Contains(statusStr, "UU ") || strings.Contains(statusStr, "AA ") || strings.Contains(statusStr, "DD ") { - // Get absolute path for clarity - absPath, err := filepath.Abs(s.workDir) - if err != nil { - absPath = s.workDir - } - return fmt.Errorf("repository has unresolved merge conflicts\nPlease resolve conflicts in: %s\nOr delete the directory to start fresh: rm -rf %s", absPath, absPath) - } - // If we have uncommitted changes but no conflicts, try to stash them - fmt.Println(" Stashing uncommitted changes...") - if err := exec.Command("git", "stash", "push", "-m", "gitsyncer-auto-stash").Run(); err != nil { - return fmt.Errorf("failed to stash changes: %w", err) - } - defer func() { - // Try to pop the stash at the end - exec.Command("git", "stash", "pop").Run() - }() + // Handle merge conflicts and uncommitted changes + stashed, err := s.handleWorkingDirectoryState() + if err != nil { + return err + } + if stashed { + defer popStash() } // Create or checkout the branch @@ -281,91 +166,47 @@ func (s *Syncer) syncBranch(branch string, remotes map[string]*config.Organizati } // Track which remotes have this branch - remotesWithBranch := make(map[string]bool) + remotesWithBranch := s.trackRemotesWithBranch(branch, remotes) - // Check which remotes have this branch - for remoteName := range remotes { - if s.remoteBranchExists(remoteName, branch) { - remotesWithBranch[remoteName] = true - } - } - - // If no remotes have this branch, it means it's a local branch that needs to be pushed - if len(remotesWithBranch) == 0 { - fmt.Printf(" Branch %s is local only, will push to all remotes\n", branch) - } else { - // Merge changes from all remotes that have this branch - for remoteName := range remotesWithBranch { - fmt.Printf(" Merging from %s/%s...\n", remoteName, branch) - - cmd := exec.Command("git", "merge", fmt.Sprintf("%s/%s", remoteName, branch), "--no-edit") - output, err := cmd.CombinedOutput() - - if err != nil { - // Check if it's a merge conflict - if strings.Contains(string(output), "CONFLICT") { - return fmt.Errorf("merge conflict detected when merging %s/%s. Please resolve manually", remoteName, branch) - } - return fmt.Errorf("failed to merge %s/%s: %w\n%s", remoteName, branch, err, string(output)) - } - } + // Merge changes from remotes + if err := mergeFromRemotes(branch, remotesWithBranch); err != nil { + return err } // Push to all remotes - for remoteName, org := range remotes { - // Check if this remote has the branch - remoteHasBranch := remotesWithBranch[remoteName] - - if !remoteHasBranch { - fmt.Printf(" Creating branch on %s (%s)...\n", remoteName, org.Host) - } else { - fmt.Printf(" Pushing to %s (%s)...\n", remoteName, org.Host) - } - - cmd := exec.Command("git", "push", remoteName, branch) - output, err := cmd.CombinedOutput() + return pushToAllRemotes(branch, remotes, remotesWithBranch) +} +// handleWorkingDirectoryState checks for conflicts and stashes changes if needed +// Returns true if changes were stashed +func (s *Syncer) handleWorkingDirectoryState() (bool, error) { + hasConflicts, statusStr, err := checkForMergeConflicts() + if err != nil || statusStr == "" { + return false, nil + } + + if hasConflicts { + // Get absolute path for clarity + absPath, err := filepath.Abs(s.workDir) if err != nil { - outputStr := string(output) - // Check if it's because the repository doesn't exist - if strings.Contains(outputStr, "does not appear to be a git repository") || - strings.Contains(outputStr, "Could not read from remote repository") { - fmt.Printf(" Note: Remote repository %s does not exist - must be created manually\n", remoteName) - fmt.Printf(" Skipping push to %s\n", remoteName) - continue - } - // Check if it's because the branch doesn't exist on the remote - // This shouldn't happen with our logic, but keep it as a fallback - if strings.Contains(outputStr, "error: src refspec") { - fmt.Printf(" Creating new branch on %s\n", remoteName) - // Try again with -u flag to set upstream - cmd = exec.Command("git", "push", "-u", remoteName, branch) - if err := cmd.Run(); err != nil { - return fmt.Errorf("failed to push to %s: %w", remoteName, err) - } - } else { - return fmt.Errorf("failed to push to %s: %w\n%s", remoteName, err, outputStr) - } - } else if !remoteHasBranch { - fmt.Printf(" Successfully created branch %s on %s\n", branch, remoteName) + absPath = s.workDir } + return false, fmt.Errorf("repository has unresolved merge conflicts\nPlease resolve conflicts in: %s\nOr delete the directory to start fresh: rm -rf %s", absPath, absPath) } - - return nil + + // If we have uncommitted changes but no conflicts, try to stash them + if err := stashChanges(); err != nil { + return false, fmt.Errorf("failed to stash changes: %w", err) + } + return true, nil } // checkoutBranch checks out a branch, creating it if necessary func (s *Syncer) checkoutBranch(branch string) error { // First try to checkout existing branch - cmd := exec.Command("git", "checkout", branch) - output, err := cmd.CombinedOutput() - if err == nil { + if err := checkoutExistingBranch(branch); err == nil { return nil } - - // If checkout failed, check the error - outputStr := string(output) - fmt.Printf(" Initial checkout failed: %s\n", strings.TrimSpace(outputStr)) // If that fails, create a new branch tracking the first remote that has it for i := range s.config.Organizations { @@ -373,12 +214,7 @@ func (s *Syncer) checkoutBranch(branch string) error { remoteName := s.getRemoteName(org) if s.remoteBranchExists(remoteName, branch) { - cmd = exec.Command("git", "checkout", "-b", branch, fmt.Sprintf("%s/%s", remoteName, branch)) - output, err := cmd.CombinedOutput() - if err != nil { - return fmt.Errorf("failed to create tracking branch: %s", string(output)) - } - return nil + return createTrackingBranch(branch, remoteName) } } |
