1use crate::errors::{CascadeError, Result};
2use crate::git::{ConflictAnalyzer, GitRepository};
3use crate::stack::{Stack, StackManager};
4use chrono::Utc;
5use serde::{Deserialize, Serialize};
6use std::collections::HashMap;
7use tracing::{debug, info, warn};
8use uuid::Uuid;
9
10#[derive(Debug, Clone)]
12enum ConflictResolution {
13 Resolved,
15 TooComplex,
17}
18
19#[derive(Debug, Clone)]
21#[allow(dead_code)]
22struct ConflictRegion {
23 start: usize,
25 end: usize,
27 start_line: usize,
29 end_line: usize,
31 our_content: String,
33 their_content: String,
35}
36
37#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
39pub enum RebaseStrategy {
40 ForcePush,
43 Interactive,
45}
46
47#[derive(Debug, Clone)]
49pub struct RebaseOptions {
50 pub strategy: RebaseStrategy,
52 pub interactive: bool,
54 pub target_base: Option<String>,
56 pub preserve_merges: bool,
58 pub auto_resolve: bool,
60 pub max_retries: usize,
62 pub skip_pull: Option<bool>,
64 pub original_working_branch: Option<String>,
67}
68
69#[derive(Debug)]
71pub struct RebaseResult {
72 pub success: bool,
74 pub branch_mapping: HashMap<String, String>,
76 pub conflicts: Vec<String>,
78 pub new_commits: Vec<String>,
80 pub error: Option<String>,
82 pub summary: String,
84}
85
86#[allow(dead_code)]
92struct TempBranchCleanupGuard {
93 branches: Vec<String>,
94 cleaned: bool,
95}
96
97#[allow(dead_code)]
98impl TempBranchCleanupGuard {
99 fn new() -> Self {
100 Self {
101 branches: Vec::new(),
102 cleaned: false,
103 }
104 }
105
106 fn add_branch(&mut self, branch: String) {
107 self.branches.push(branch);
108 }
109
110 fn cleanup(&mut self, git_repo: &GitRepository) {
112 if self.cleaned || self.branches.is_empty() {
113 return;
114 }
115
116 info!("๐งน Cleaning up {} temporary branches", self.branches.len());
117 for branch in &self.branches {
118 if let Err(e) = git_repo.delete_branch_unsafe(branch) {
119 warn!("Failed to delete temp branch {}: {}", branch, e);
120 }
122 }
123 self.cleaned = true;
124 }
125}
126
127impl Drop for TempBranchCleanupGuard {
128 fn drop(&mut self) {
129 if !self.cleaned && !self.branches.is_empty() {
130 warn!(
133 "โ ๏ธ {} temporary branches were not cleaned up: {}",
134 self.branches.len(),
135 self.branches.join(", ")
136 );
137 warn!("Run 'ca cleanup' to remove orphaned temporary branches");
138 }
139 }
140}
141
142pub struct RebaseManager {
144 stack_manager: StackManager,
145 git_repo: GitRepository,
146 options: RebaseOptions,
147 conflict_analyzer: ConflictAnalyzer,
148}
149
150impl Default for RebaseOptions {
151 fn default() -> Self {
152 Self {
153 strategy: RebaseStrategy::ForcePush,
154 interactive: false,
155 target_base: None,
156 preserve_merges: true,
157 auto_resolve: true,
158 max_retries: 3,
159 skip_pull: None,
160 original_working_branch: None,
161 }
162 }
163}
164
165impl RebaseManager {
166 pub fn new(
168 stack_manager: StackManager,
169 git_repo: GitRepository,
170 options: RebaseOptions,
171 ) -> Self {
172 Self {
173 stack_manager,
174 git_repo,
175 options,
176 conflict_analyzer: ConflictAnalyzer::new(),
177 }
178 }
179
180 pub fn into_stack_manager(self) -> StackManager {
182 self.stack_manager
183 }
184
185 pub fn rebase_stack(&mut self, stack_id: &Uuid) -> Result<RebaseResult> {
187 debug!("Starting rebase for stack {}", stack_id);
188
189 let stack = self
190 .stack_manager
191 .get_stack(stack_id)
192 .ok_or_else(|| CascadeError::config(format!("Stack {stack_id} not found")))?
193 .clone();
194
195 match self.options.strategy {
196 RebaseStrategy::ForcePush => self.rebase_with_force_push(&stack),
197 RebaseStrategy::Interactive => self.rebase_interactive(&stack),
198 }
199 }
200
201 fn rebase_with_force_push(&mut self, stack: &Stack) -> Result<RebaseResult> {
205 use crate::cli::output::Output;
206
207 if self.has_in_progress_cherry_pick()? {
209 return self.handle_in_progress_cherry_pick(stack);
210 }
211
212 Output::section(format!("Rebasing stack: {}", stack.name));
213
214 let mut result = RebaseResult {
215 success: true,
216 branch_mapping: HashMap::new(),
217 conflicts: Vec::new(),
218 new_commits: Vec::new(),
219 error: None,
220 summary: String::new(),
221 };
222
223 let target_base = self
224 .options
225 .target_base
226 .as_ref()
227 .unwrap_or(&stack.base_branch)
228 .clone(); let original_branch = self
234 .options
235 .original_working_branch
236 .clone()
237 .or_else(|| self.git_repo.get_current_branch().ok());
238
239 if let Some(ref orig) = original_branch {
242 if orig == &target_base {
243 debug!(
244 "Original working branch is base branch '{}' - will skip working branch update",
245 orig
246 );
247 }
248 }
249
250 if !self.options.skip_pull.unwrap_or(false) {
253 if let Err(e) = self.pull_latest_changes(&target_base) {
254 Output::warning(format!("Could not pull latest changes: {}", e));
255 }
256 }
257
258 if let Err(e) = self.git_repo.reset_to_head() {
260 Output::warning(format!("Could not reset working directory: {}", e));
261 }
262
263 let mut current_base = target_base.clone();
264 let entry_count = stack.entries.len();
265 let mut temp_branches: Vec<String> = Vec::new(); let mut branches_to_push: Vec<(String, String)> = Vec::new(); if entry_count == 0 {
270 println!(); Output::info("Stack has no entries yet");
272 Output::tip("Use 'ca push' to add commits to this stack");
273
274 result.summary = "Stack is empty".to_string();
275
276 println!(); Output::success(&result.summary);
279
280 self.stack_manager.save_to_disk()?;
282 return Ok(result);
283 }
284
285 println!(); let plural = if entry_count == 1 { "entry" } else { "entries" };
287 println!("Rebasing {} {}...", entry_count, plural);
288
289 for (index, entry) in stack.entries.iter().enumerate() {
291 let original_branch = &entry.branch;
292
293 if self
296 .git_repo
297 .is_commit_based_on(&entry.commit_hash, ¤t_base)
298 .unwrap_or(false)
299 {
300 tracing::debug!(
301 "Entry '{}' is already correctly based on '{}', skipping rebase",
302 original_branch,
303 current_base
304 );
305
306 let tree_char = if index + 1 == entry_count {
308 "โโ"
309 } else {
310 "โโ"
311 };
312
313 if let Some(pr_num) = &entry.pull_request_id {
314 println!(" {} {} (PR #{})", tree_char, original_branch, pr_num);
315 branches_to_push.push((original_branch.clone(), pr_num.clone()));
316 } else {
317 println!(" {} {} (not submitted)", tree_char, original_branch);
318 }
319
320 result
321 .branch_mapping
322 .insert(original_branch.clone(), original_branch.clone());
323
324 current_base = original_branch.clone();
326 continue;
327 }
328
329 let temp_branch = format!("{}-temp-{}", original_branch, Utc::now().timestamp());
332 temp_branches.push(temp_branch.clone()); self.git_repo
334 .create_branch(&temp_branch, Some(¤t_base))?;
335 self.git_repo.checkout_branch_silent(&temp_branch)?;
336
337 match self.cherry_pick_commit(&entry.commit_hash) {
339 Ok(new_commit_hash) => {
340 result.new_commits.push(new_commit_hash.clone());
341
342 let rebased_commit_id = self.git_repo.get_head_commit()?.id().to_string();
344
345 self.git_repo
348 .update_branch_to_commit(original_branch, &rebased_commit_id)?;
349
350 let tree_char = if index + 1 == entry_count {
352 "โโ"
353 } else {
354 "โโ"
355 };
356
357 if let Some(pr_num) = &entry.pull_request_id {
358 println!(" {} {} (PR #{})", tree_char, original_branch, pr_num);
359 branches_to_push.push((original_branch.clone(), pr_num.clone()));
360 } else {
361 println!(" {} {} (not submitted)", tree_char, original_branch);
362 }
363
364 result
365 .branch_mapping
366 .insert(original_branch.clone(), original_branch.clone());
367
368 self.update_stack_entry(
370 stack.id,
371 &entry.id,
372 original_branch,
373 &rebased_commit_id,
374 )?;
375
376 current_base = original_branch.clone();
378 }
379 Err(e) => {
380 result.conflicts.push(entry.commit_hash.clone());
381
382 if !self.options.auto_resolve {
383 println!(); Output::error(e.to_string());
385 result.success = false;
386 result.error = Some(format!(
387 "Conflict in {}: {}\n\n\
388 MANUAL CONFLICT RESOLUTION REQUIRED\n\
389 =====================================\n\n\
390 Step 1: Analyze conflicts\n\
391 โ Run: ca conflicts\n\
392 โ This shows which conflicts are in which files\n\n\
393 Step 2: Resolve conflicts in your editor\n\
394 โ Open conflicted files and edit them\n\
395 โ Remove conflict markers (<<<<<<, ======, >>>>>>)\n\
396 โ Keep the code you want\n\
397 โ Save the files\n\n\
398 Step 3: Mark conflicts as resolved\n\
399 โ Run: git add <resolved-files>\n\
400 โ Or: git add -A (to stage all resolved files)\n\n\
401 Step 4: Complete the sync\n\
402 โ Run: ca sync\n\
403 โ Cascade will detect resolved conflicts and continue\n\n\
404 Alternative: Abort and start over\n\
405 โ Run: git cherry-pick --abort\n\
406 โ Then: ca sync (starts fresh)\n\n\
407 TIP: Enable auto-resolution for simple conflicts:\n\
408 โ Run: ca sync --auto-resolve\n\
409 โ Only complex conflicts will require manual resolution",
410 entry.commit_hash, e
411 ));
412 break;
413 }
414
415 match self.auto_resolve_conflicts(&entry.commit_hash) {
417 Ok(fully_resolved) => {
418 if !fully_resolved {
419 result.success = false;
420 result.error = Some(format!(
421 "Conflicts in commit {}\n\n\
422 To resolve:\n\
423 1. Fix conflicts in your editor\n\
424 2. Run: ca sync --continue\n\n\
425 Or abort:\n\
426 โ Run: git cherry-pick --abort",
427 &entry.commit_hash[..8]
428 ));
429 break;
430 }
431
432 let commit_message =
434 format!("Auto-resolved conflicts in {}", &entry.commit_hash[..8]);
435
436 debug!("Checking staged files before commit");
438 let staged_files = self.git_repo.get_staged_files()?;
439
440 if staged_files.is_empty() {
441 result.success = false;
445 result.error = Some(format!(
446 "CRITICAL BUG DETECTED: Cherry-pick failed but no files were staged!\n\n\
447 This indicates a Git state issue after cherry-pick failure.\n\n\
448 RECOVERY STEPS:\n\
449 ================\n\n\
450 Step 1: Check Git status\n\
451 โ Run: git status\n\
452 โ Check if there are any changes in working directory\n\n\
453 Step 2: Check for conflicts manually\n\
454 โ Run: git diff\n\
455 โ Look for conflict markers (<<<<<<, ======, >>>>>>)\n\n\
456 Step 3: Abort the cherry-pick\n\
457 โ Run: git cherry-pick --abort\n\n\
458 Step 4: Report this bug\n\
459 โ This is a known issue we're investigating\n\
460 โ Cherry-pick failed for commit {}\n\
461 โ But Git reported no conflicts and no staged files\n\n\
462 Step 5: Try manual resolution\n\
463 โ Run: ca sync --no-auto-resolve\n\
464 โ Manually resolve conflicts as they appear",
465 &entry.commit_hash[..8]
466 ));
467 warn!("CRITICAL - No files staged after auto-resolve!");
468 break;
469 }
470
471 debug!("{} files staged", staged_files.len());
472
473 match self.git_repo.commit(&commit_message) {
474 Ok(new_commit_id) => {
475 debug!(
476 "Created commit {} with message '{}'",
477 &new_commit_id[..8],
478 commit_message
479 );
480
481 Output::success("Auto-resolved conflicts");
482 result.new_commits.push(new_commit_id.clone());
483 let rebased_commit_id = new_commit_id;
484
485 self.git_repo.update_branch_to_commit(
487 original_branch,
488 &rebased_commit_id,
489 )?;
490
491 let tree_char = if index + 1 == entry_count {
493 "โโ"
494 } else {
495 "โโ"
496 };
497
498 if let Some(pr_num) = &entry.pull_request_id {
499 println!(
500 " {} {} (PR #{})",
501 tree_char, original_branch, pr_num
502 );
503 branches_to_push
504 .push((original_branch.clone(), pr_num.clone()));
505 } else {
506 println!(
507 " {} {} (not submitted)",
508 tree_char, original_branch
509 );
510 }
511
512 result
513 .branch_mapping
514 .insert(original_branch.clone(), original_branch.clone());
515
516 self.update_stack_entry(
518 stack.id,
519 &entry.id,
520 original_branch,
521 &rebased_commit_id,
522 )?;
523
524 current_base = original_branch.clone();
526 }
527 Err(commit_err) => {
528 result.success = false;
529 result.error = Some(format!(
530 "Could not commit auto-resolved conflicts: {}\n\n\
531 This usually means:\n\
532 - Git index is locked (another process accessing repo)\n\
533 - File permissions issue\n\
534 - Disk space issue\n\n\
535 Recovery:\n\
536 1. Check if another Git operation is running\n\
537 2. Run 'rm -f .git/index.lock' if stale lock exists\n\
538 3. Run 'git status' to check repo state\n\
539 4. Retry 'ca sync' after fixing the issue",
540 commit_err
541 ));
542 break;
543 }
544 }
545 }
546 Err(resolve_err) => {
547 result.success = false;
548 result.error = Some(format!(
549 "Could not resolve conflicts: {}\n\n\
550 Recovery:\n\
551 1. Check repo state: 'git status'\n\
552 2. If files are staged, commit or reset them: 'git reset --hard HEAD'\n\
553 3. Remove any lock files: 'rm -f .git/index.lock'\n\
554 4. Retry 'ca sync'",
555 resolve_err
556 ));
557 break;
558 }
559 }
560 }
561 }
562 }
563
564 if !temp_branches.is_empty() {
567 if let Err(e) = self.git_repo.checkout_branch_unsafe(&target_base) {
570 debug!("Could not checkout base for cleanup: {}", e);
571 } else {
574 for temp_branch in &temp_branches {
576 if let Err(e) = self.git_repo.delete_branch_unsafe(temp_branch) {
577 debug!("Could not delete temp branch {}: {}", temp_branch, e);
578 }
579 }
580 }
581 }
582
583 let pushed_count = branches_to_push.len();
586 let skipped_count = entry_count - pushed_count;
587
588 if !branches_to_push.is_empty() {
589 for (branch_name, _pr_num) in &branches_to_push {
592 match self.git_repo.force_push_single_branch_auto(branch_name) {
593 Ok(_) => {
594 debug!("Pushed {} successfully", branch_name);
595 }
596 Err(e) => {
597 Output::warning(format!("Could not push '{}': {}", branch_name, e));
598 }
600 }
601 }
602 }
603
604 if let Some(ref orig_branch) = original_branch {
607 if orig_branch != &target_base {
609 if let Some(last_entry) = stack.entries.last() {
611 let top_branch = &last_entry.branch;
612
613 if let (Ok(working_head), Ok(top_commit)) = (
616 self.git_repo.get_branch_head(orig_branch),
617 self.git_repo.get_branch_head(top_branch),
618 ) {
619 if working_head != top_commit {
621 if let Ok(commits) = self.git_repo.get_commits_between(&top_commit, &working_head) {
623 if !commits.is_empty() {
624 Output::error(format!(
626 "Cannot sync: Working branch '{}' has {} commit(s) not in the stack",
627 orig_branch,
628 commits.len()
629 ));
630 println!();
631 Output::sub_item("These commits would be lost if we proceed:");
632 for (i, commit) in commits.iter().take(5).enumerate() {
633 let message = commit.summary().unwrap_or("(no message)");
634 Output::sub_item(format!(" {}. {} - {}", i + 1, &commit.id().to_string()[..8], message));
635 }
636 if commits.len() > 5 {
637 Output::sub_item(format!(" ... and {} more", commits.len() - 5));
638 }
639 println!();
640 Output::tip("Add these commits to the stack first:");
641 Output::bullet(format!("Run: ca stack push"));
642 Output::bullet(format!("Then run: ca sync"));
643 println!();
644
645 return Err(CascadeError::validation(
646 format!(
647 "Working branch '{}' has {} untracked commit(s). Add them to the stack with 'ca stack push' before syncing.",
648 orig_branch, commits.len()
649 )
650 ));
651 }
652 }
653 }
654
655 debug!(
657 "Updating working branch '{}' to match top of stack ({})",
658 orig_branch,
659 &top_commit[..8]
660 );
661
662 if let Err(e) = self
663 .git_repo
664 .update_branch_to_commit(orig_branch, &top_commit)
665 {
666 Output::warning(format!(
667 "Could not update working branch '{}' to top of stack: {}",
668 orig_branch, e
669 ));
670 }
671 }
672 }
673
674 if let Err(e) = self.git_repo.checkout_branch_unsafe(orig_branch) {
677 debug!(
678 "Could not return to original branch '{}': {}",
679 orig_branch, e
680 );
681 }
683 } else {
684 debug!(
687 "Skipping working branch update - user was on base branch '{}'",
688 orig_branch
689 );
690 if let Err(e) = self.git_repo.checkout_branch_unsafe(orig_branch) {
691 debug!("Could not return to base branch '{}': {}", orig_branch, e);
692 }
693 }
694 }
695
696 result.summary = if pushed_count > 0 {
698 let pr_plural = if pushed_count == 1 { "" } else { "s" };
699 let entry_plural = if entry_count == 1 { "entry" } else { "entries" };
700
701 if skipped_count > 0 {
702 format!(
703 "{} {} rebased ({} PR{} updated, {} not yet submitted)",
704 entry_count, entry_plural, pushed_count, pr_plural, skipped_count
705 )
706 } else {
707 format!(
708 "{} {} rebased ({} PR{} updated)",
709 entry_count, entry_plural, pushed_count, pr_plural
710 )
711 }
712 } else {
713 let plural = if entry_count == 1 { "entry" } else { "entries" };
714 format!("{} {} rebased (no PRs to update yet)", entry_count, plural)
715 };
716
717 println!(); if result.success {
720 Output::success(&result.summary);
721 } else {
722 let error_msg = result
724 .error
725 .as_deref()
726 .unwrap_or("Rebase failed for unknown reason");
727 Output::error(error_msg);
728 }
729
730 self.stack_manager.save_to_disk()?;
732
733 if !result.success {
736 let detailed_error = result.error.as_deref().unwrap_or("Rebase failed");
738 return Err(CascadeError::Branch(detailed_error.to_string()));
739 }
740
741 Ok(result)
742 }
743
744 fn rebase_interactive(&mut self, stack: &Stack) -> Result<RebaseResult> {
746 info!("Starting interactive rebase for stack '{}'", stack.name);
747
748 let mut result = RebaseResult {
749 success: true,
750 branch_mapping: HashMap::new(),
751 conflicts: Vec::new(),
752 new_commits: Vec::new(),
753 error: None,
754 summary: String::new(),
755 };
756
757 println!("Interactive Rebase for Stack: {}", stack.name);
758 println!(" Base branch: {}", stack.base_branch);
759 println!(" Entries: {}", stack.entries.len());
760
761 if self.options.interactive {
762 println!("\nChoose action for each commit:");
763 println!(" (p)ick - apply the commit");
764 println!(" (s)kip - skip this commit");
765 println!(" (e)dit - edit the commit message");
766 println!(" (q)uit - abort the rebase");
767 }
768
769 for entry in &stack.entries {
772 println!(
773 " {} {} - {}",
774 entry.short_hash(),
775 entry.branch,
776 entry.short_message(50)
777 );
778
779 match self.cherry_pick_commit(&entry.commit_hash) {
781 Ok(new_commit) => result.new_commits.push(new_commit),
782 Err(_) => result.conflicts.push(entry.commit_hash.clone()),
783 }
784 }
785
786 result.summary = format!(
787 "Interactive rebase processed {} commits",
788 stack.entries.len()
789 );
790 Ok(result)
791 }
792
793 fn cherry_pick_commit(&self, commit_hash: &str) -> Result<String> {
795 let new_commit_hash = self.git_repo.cherry_pick(commit_hash)?;
797
798 if let Ok(staged_files) = self.git_repo.get_staged_files() {
800 if !staged_files.is_empty() {
801 let cleanup_message = format!("Cleanup after cherry-pick {}", &commit_hash[..8]);
803 let _ = self.git_repo.commit_staged_changes(&cleanup_message);
804 }
805 }
806
807 Ok(new_commit_hash)
808 }
809
810 fn auto_resolve_conflicts(&self, commit_hash: &str) -> Result<bool> {
812 debug!("Starting auto-resolve for commit {}", commit_hash);
813
814 let has_conflicts = self.git_repo.has_conflicts()?;
816 debug!("has_conflicts() = {}", has_conflicts);
817
818 let cherry_pick_head = self.git_repo.path().join(".git").join("CHERRY_PICK_HEAD");
820 let cherry_pick_in_progress = cherry_pick_head.exists();
821
822 if !has_conflicts {
823 debug!("No conflicts detected by Git index");
824
825 if cherry_pick_in_progress {
827 warn!("CHERRY_PICK_HEAD exists but no conflicts in index - aborting cherry-pick");
828
829 let _ = std::process::Command::new("git")
831 .args(["cherry-pick", "--abort"])
832 .current_dir(self.git_repo.path())
833 .output();
834
835 return Err(CascadeError::Branch(format!(
836 "Cherry-pick failed for {} but Git index shows no conflicts. \
837 This usually means the cherry-pick was aborted or failed in an unexpected way. \
838 Please try manual resolution.",
839 &commit_hash[..8]
840 )));
841 }
842
843 return Ok(true);
844 }
845
846 let conflicted_files = self.git_repo.get_conflicted_files()?;
847
848 if conflicted_files.is_empty() {
849 debug!("Conflicted files list is empty");
850 return Ok(true);
851 }
852
853 debug!(
854 "Found conflicts in {} files: {:?}",
855 conflicted_files.len(),
856 conflicted_files
857 );
858
859 let analysis = self
861 .conflict_analyzer
862 .analyze_conflicts(&conflicted_files, self.git_repo.path())?;
863
864 debug!(
865 "Conflict analysis: {} total conflicts, {} auto-resolvable",
866 analysis.total_conflicts, analysis.auto_resolvable_count
867 );
868
869 for recommendation in &analysis.recommendations {
871 debug!("{}", recommendation);
872 }
873
874 let mut resolved_count = 0;
875 let mut resolved_files = Vec::new(); let mut failed_files = Vec::new();
877
878 for file_analysis in &analysis.files {
879 debug!(
880 "Processing file: {} (auto_resolvable: {}, conflicts: {})",
881 file_analysis.file_path,
882 file_analysis.auto_resolvable,
883 file_analysis.conflicts.len()
884 );
885
886 if file_analysis.auto_resolvable {
887 match self.resolve_file_conflicts_enhanced(
888 &file_analysis.file_path,
889 &file_analysis.conflicts,
890 ) {
891 Ok(ConflictResolution::Resolved) => {
892 resolved_count += 1;
893 resolved_files.push(file_analysis.file_path.clone());
894 debug!("Successfully resolved {}", file_analysis.file_path);
895 }
896 Ok(ConflictResolution::TooComplex) => {
897 debug!(
898 "{} too complex for auto-resolution",
899 file_analysis.file_path
900 );
901 failed_files.push(file_analysis.file_path.clone());
902 }
903 Err(e) => {
904 debug!("Failed to resolve {}: {}", file_analysis.file_path, e);
905 failed_files.push(file_analysis.file_path.clone());
906 }
907 }
908 } else {
909 failed_files.push(file_analysis.file_path.clone());
910 debug!(
911 "{} requires manual resolution ({} conflicts)",
912 file_analysis.file_path,
913 file_analysis.conflicts.len()
914 );
915 }
916 }
917
918 if resolved_count > 0 {
919 debug!(
920 "Resolved {}/{} files",
921 resolved_count,
922 conflicted_files.len()
923 );
924 debug!("Resolved files: {:?}", resolved_files);
925
926 let file_paths: Vec<&str> = resolved_files.iter().map(|s| s.as_str()).collect();
929 debug!("Staging {} files", file_paths.len());
930 self.git_repo.stage_files(&file_paths)?;
931 debug!("Files staged successfully");
932 } else {
933 debug!("No files were resolved (resolved_count = 0)");
934 }
935
936 let all_resolved = failed_files.is_empty();
938
939 debug!(
940 "all_resolved = {}, failed_files = {:?}",
941 all_resolved, failed_files
942 );
943
944 if !all_resolved {
945 debug!("{} files still need manual resolution", failed_files.len());
946 }
947
948 debug!("Returning all_resolved = {}", all_resolved);
949 Ok(all_resolved)
950 }
951
952 fn resolve_file_conflicts_enhanced(
954 &self,
955 file_path: &str,
956 conflicts: &[crate::git::ConflictRegion],
957 ) -> Result<ConflictResolution> {
958 let repo_path = self.git_repo.path();
959 let full_path = repo_path.join(file_path);
960
961 let mut content = std::fs::read_to_string(&full_path)
963 .map_err(|e| CascadeError::config(format!("Failed to read file {file_path}: {e}")))?;
964
965 if conflicts.is_empty() {
966 return Ok(ConflictResolution::Resolved);
967 }
968
969 info!(
970 "Resolving {} conflicts in {} using enhanced analysis",
971 conflicts.len(),
972 file_path
973 );
974
975 let mut any_resolved = false;
976
977 for conflict in conflicts.iter().rev() {
979 match self.resolve_single_conflict_enhanced(conflict) {
980 Ok(Some(resolution)) => {
981 let before = &content[..conflict.start_pos];
983 let after = &content[conflict.end_pos..];
984 content = format!("{before}{resolution}{after}");
985 any_resolved = true;
986 debug!(
987 "โ
Resolved {} conflict at lines {}-{} in {}",
988 format!("{:?}", conflict.conflict_type).to_lowercase(),
989 conflict.start_line,
990 conflict.end_line,
991 file_path
992 );
993 }
994 Ok(None) => {
995 debug!(
996 "โ ๏ธ {} conflict at lines {}-{} in {} requires manual resolution",
997 format!("{:?}", conflict.conflict_type).to_lowercase(),
998 conflict.start_line,
999 conflict.end_line,
1000 file_path
1001 );
1002 return Ok(ConflictResolution::TooComplex);
1003 }
1004 Err(e) => {
1005 debug!("โ Failed to resolve conflict in {}: {}", file_path, e);
1006 return Ok(ConflictResolution::TooComplex);
1007 }
1008 }
1009 }
1010
1011 if any_resolved {
1012 let remaining_conflicts = self.parse_conflict_markers(&content)?;
1014
1015 if remaining_conflicts.is_empty() {
1016 debug!(
1017 "All conflicts resolved in {}, content length: {} bytes",
1018 file_path,
1019 content.len()
1020 );
1021
1022 if content.trim().is_empty() {
1024 warn!(
1025 "SAFETY CHECK: Resolved content for {} is empty! Aborting auto-resolution.",
1026 file_path
1027 );
1028 return Ok(ConflictResolution::TooComplex);
1029 }
1030
1031 let backup_path = full_path.with_extension("cascade-backup");
1033 if let Ok(original_content) = std::fs::read_to_string(&full_path) {
1034 debug!(
1035 "Backup for {} (original: {} bytes, resolved: {} bytes)",
1036 file_path,
1037 original_content.len(),
1038 content.len()
1039 );
1040 let _ = std::fs::write(&backup_path, original_content);
1041 }
1042
1043 crate::utils::atomic_file::write_string(&full_path, &content)?;
1045
1046 debug!("Wrote {} bytes to {}", content.len(), file_path);
1047 return Ok(ConflictResolution::Resolved);
1048 } else {
1049 info!(
1050 "Partially resolved conflicts in {} ({} remaining)",
1051 file_path,
1052 remaining_conflicts.len()
1053 );
1054 }
1055 }
1056
1057 Ok(ConflictResolution::TooComplex)
1058 }
1059
1060 #[allow(dead_code)]
1062 fn count_whitespace_consistency(content: &str) -> usize {
1063 let mut inconsistencies = 0;
1064 let lines: Vec<&str> = content.lines().collect();
1065
1066 for line in &lines {
1067 if line.contains('\t') && line.contains(' ') {
1069 inconsistencies += 1;
1070 }
1071 }
1072
1073 lines.len().saturating_sub(inconsistencies)
1075 }
1076
1077 fn resolve_single_conflict_enhanced(
1079 &self,
1080 conflict: &crate::git::ConflictRegion,
1081 ) -> Result<Option<String>> {
1082 debug!(
1083 "Resolving {} conflict in {} (lines {}-{})",
1084 format!("{:?}", conflict.conflict_type).to_lowercase(),
1085 conflict.file_path,
1086 conflict.start_line,
1087 conflict.end_line
1088 );
1089
1090 use crate::git::ConflictType;
1091
1092 match conflict.conflict_type {
1093 ConflictType::Whitespace => {
1094 let our_normalized = conflict
1097 .our_content
1098 .split_whitespace()
1099 .collect::<Vec<_>>()
1100 .join(" ");
1101 let their_normalized = conflict
1102 .their_content
1103 .split_whitespace()
1104 .collect::<Vec<_>>()
1105 .join(" ");
1106
1107 if our_normalized == their_normalized {
1108 Ok(Some(conflict.their_content.clone()))
1113 } else {
1114 debug!(
1116 "Whitespace conflict has content differences - requires manual resolution"
1117 );
1118 Ok(None)
1119 }
1120 }
1121 ConflictType::LineEnding => {
1122 let normalized = conflict
1124 .our_content
1125 .replace("\r\n", "\n")
1126 .replace('\r', "\n");
1127 Ok(Some(normalized))
1128 }
1129 ConflictType::PureAddition => {
1130 if conflict.our_content.is_empty() && !conflict.their_content.is_empty() {
1134 Ok(Some(conflict.their_content.clone()))
1136 } else if conflict.their_content.is_empty() && !conflict.our_content.is_empty() {
1137 Ok(Some(String::new()))
1139 } else if conflict.our_content.is_empty() && conflict.their_content.is_empty() {
1140 Ok(Some(String::new()))
1142 } else {
1143 debug!(
1149 "PureAddition conflict has content on both sides - requires manual resolution"
1150 );
1151 Ok(None)
1152 }
1153 }
1154 ConflictType::ImportMerge => {
1155 let our_lines: Vec<&str> = conflict.our_content.lines().collect();
1160 let their_lines: Vec<&str> = conflict.their_content.lines().collect();
1161
1162 let all_simple = our_lines.iter().chain(their_lines.iter()).all(|line| {
1164 let trimmed = line.trim();
1165 trimmed.starts_with("import ")
1166 || trimmed.starts_with("from ")
1167 || trimmed.starts_with("use ")
1168 || trimmed.starts_with("#include")
1169 || trimmed.is_empty()
1170 });
1171
1172 if !all_simple {
1173 debug!("ImportMerge contains non-import lines - requires manual resolution");
1174 return Ok(None);
1175 }
1176
1177 let mut all_imports: Vec<&str> = our_lines
1179 .into_iter()
1180 .chain(their_lines)
1181 .filter(|line| !line.trim().is_empty())
1182 .collect();
1183 all_imports.sort();
1184 all_imports.dedup();
1185 Ok(Some(all_imports.join("\n")))
1186 }
1187 ConflictType::Structural | ConflictType::ContentOverlap | ConflictType::Complex => {
1188 Ok(None)
1190 }
1191 }
1192 }
1193
1194 fn parse_conflict_markers(&self, content: &str) -> Result<Vec<ConflictRegion>> {
1196 let lines: Vec<&str> = content.lines().collect();
1197 let mut conflicts = Vec::new();
1198 let mut i = 0;
1199
1200 while i < lines.len() {
1201 if lines[i].starts_with("<<<<<<<") {
1202 let start_line = i + 1;
1204 let mut separator_line = None;
1205 let mut end_line = None;
1206
1207 for (j, line) in lines.iter().enumerate().skip(i + 1) {
1209 if line.starts_with("=======") {
1210 separator_line = Some(j + 1);
1211 } else if line.starts_with(">>>>>>>") {
1212 end_line = Some(j + 1);
1213 break;
1214 }
1215 }
1216
1217 if let (Some(sep), Some(end)) = (separator_line, end_line) {
1218 let start_pos = lines[..i].iter().map(|l| l.len() + 1).sum::<usize>();
1220 let end_pos = lines[..end].iter().map(|l| l.len() + 1).sum::<usize>();
1221
1222 let our_content = lines[(i + 1)..(sep - 1)].join("\n");
1223 let their_content = lines[sep..(end - 1)].join("\n");
1224
1225 conflicts.push(ConflictRegion {
1226 start: start_pos,
1227 end: end_pos,
1228 start_line,
1229 end_line: end,
1230 our_content,
1231 their_content,
1232 });
1233
1234 i = end;
1235 } else {
1236 i += 1;
1237 }
1238 } else {
1239 i += 1;
1240 }
1241 }
1242
1243 Ok(conflicts)
1244 }
1245
1246 fn update_stack_entry(
1249 &mut self,
1250 stack_id: Uuid,
1251 entry_id: &Uuid,
1252 _new_branch: &str,
1253 new_commit_hash: &str,
1254 ) -> Result<()> {
1255 debug!(
1256 "Updating entry {} in stack {} with new commit {}",
1257 entry_id, stack_id, new_commit_hash
1258 );
1259
1260 let stack = self
1262 .stack_manager
1263 .get_stack_mut(&stack_id)
1264 .ok_or_else(|| CascadeError::config(format!("Stack {stack_id} not found")))?;
1265
1266 if let Some(entry) = stack.entries.iter_mut().find(|e| e.id == *entry_id) {
1268 debug!(
1269 "Found entry {} - updating commit from '{}' to '{}' (keeping original branch '{}')",
1270 entry_id, entry.commit_hash, new_commit_hash, entry.branch
1271 );
1272
1273 entry.commit_hash = new_commit_hash.to_string();
1276
1277 debug!(
1280 "Successfully updated entry {} in stack {}",
1281 entry_id, stack_id
1282 );
1283 Ok(())
1284 } else {
1285 Err(CascadeError::config(format!(
1286 "Entry {entry_id} not found in stack {stack_id}"
1287 )))
1288 }
1289 }
1290
1291 fn pull_latest_changes(&self, branch: &str) -> Result<()> {
1293 info!("Pulling latest changes for branch {}", branch);
1294
1295 match self.git_repo.fetch() {
1297 Ok(_) => {
1298 debug!("Fetch successful");
1299 match self.git_repo.pull(branch) {
1301 Ok(_) => {
1302 info!("Pull completed successfully for {}", branch);
1303 Ok(())
1304 }
1305 Err(e) => {
1306 warn!("Pull failed for {}: {}", branch, e);
1307 Ok(())
1309 }
1310 }
1311 }
1312 Err(e) => {
1313 warn!("Fetch failed: {}", e);
1314 Ok(())
1316 }
1317 }
1318 }
1319
1320 pub fn is_rebase_in_progress(&self) -> bool {
1322 let git_dir = self.git_repo.path().join(".git");
1324 git_dir.join("REBASE_HEAD").exists()
1325 || git_dir.join("rebase-merge").exists()
1326 || git_dir.join("rebase-apply").exists()
1327 }
1328
1329 pub fn abort_rebase(&self) -> Result<()> {
1331 info!("Aborting rebase operation");
1332
1333 let git_dir = self.git_repo.path().join(".git");
1334
1335 if git_dir.join("REBASE_HEAD").exists() {
1337 std::fs::remove_file(git_dir.join("REBASE_HEAD")).map_err(|e| {
1338 CascadeError::Git(git2::Error::from_str(&format!(
1339 "Failed to clean rebase state: {e}"
1340 )))
1341 })?;
1342 }
1343
1344 if git_dir.join("rebase-merge").exists() {
1345 std::fs::remove_dir_all(git_dir.join("rebase-merge")).map_err(|e| {
1346 CascadeError::Git(git2::Error::from_str(&format!(
1347 "Failed to clean rebase-merge: {e}"
1348 )))
1349 })?;
1350 }
1351
1352 if git_dir.join("rebase-apply").exists() {
1353 std::fs::remove_dir_all(git_dir.join("rebase-apply")).map_err(|e| {
1354 CascadeError::Git(git2::Error::from_str(&format!(
1355 "Failed to clean rebase-apply: {e}"
1356 )))
1357 })?;
1358 }
1359
1360 info!("Rebase aborted successfully");
1361 Ok(())
1362 }
1363
1364 pub fn continue_rebase(&self) -> Result<()> {
1366 info!("Continuing rebase operation");
1367
1368 if self.git_repo.has_conflicts()? {
1370 return Err(CascadeError::branch(
1371 "Cannot continue rebase: there are unresolved conflicts. Resolve conflicts and stage files first.".to_string()
1372 ));
1373 }
1374
1375 self.git_repo.stage_conflict_resolved_files()?;
1377
1378 info!("Rebase continued successfully");
1379 Ok(())
1380 }
1381
1382 fn has_in_progress_cherry_pick(&self) -> Result<bool> {
1384 let git_dir = self.git_repo.path().join(".git");
1385 Ok(git_dir.join("CHERRY_PICK_HEAD").exists())
1386 }
1387
1388 fn handle_in_progress_cherry_pick(&mut self, stack: &Stack) -> Result<RebaseResult> {
1390 use crate::cli::output::Output;
1391
1392 let git_dir = self.git_repo.path().join(".git");
1393
1394 Output::section("Resuming in-progress sync");
1395 println!();
1396 Output::info("Detected unfinished cherry-pick from previous sync");
1397 println!();
1398
1399 if self.git_repo.has_conflicts()? {
1401 let conflicted_files = self.git_repo.get_conflicted_files()?;
1402
1403 let result = RebaseResult {
1404 success: false,
1405 branch_mapping: HashMap::new(),
1406 conflicts: conflicted_files.clone(),
1407 new_commits: Vec::new(),
1408 error: Some(format!(
1409 "Cannot continue: {} file(s) still have unresolved conflicts\n\n\
1410 MANUAL CONFLICT RESOLUTION REQUIRED\n\
1411 =====================================\n\n\
1412 Conflicted files:\n{}\n\n\
1413 Step 1: Analyze conflicts\n\
1414 โ Run: ca conflicts\n\
1415 โ Shows detailed conflict analysis\n\n\
1416 Step 2: Resolve conflicts in your editor\n\
1417 โ Open conflicted files and edit them\n\
1418 โ Remove conflict markers (<<<<<<, ======, >>>>>>)\n\
1419 โ Keep the code you want\n\
1420 โ Save the files\n\n\
1421 Step 3: Mark conflicts as resolved\n\
1422 โ Run: git add <resolved-files>\n\
1423 โ Or: git add -A (to stage all resolved files)\n\n\
1424 Step 4: Complete the sync\n\
1425 โ Run: ca sync\n\
1426 โ Cascade will continue from where it left off\n\n\
1427 Alternative: Abort and start over\n\
1428 โ Run: git cherry-pick --abort\n\
1429 โ Then: ca sync (starts fresh)",
1430 conflicted_files.len(),
1431 conflicted_files
1432 .iter()
1433 .map(|f| format!(" - {}", f))
1434 .collect::<Vec<_>>()
1435 .join("\n")
1436 )),
1437 summary: "Sync paused - conflicts need resolution".to_string(),
1438 };
1439
1440 return Ok(result);
1441 }
1442
1443 Output::info("Conflicts resolved, continuing cherry-pick...");
1445
1446 self.git_repo.stage_conflict_resolved_files()?;
1448
1449 let cherry_pick_msg_file = git_dir.join("CHERRY_PICK_MSG");
1451 let commit_message = if cherry_pick_msg_file.exists() {
1452 std::fs::read_to_string(&cherry_pick_msg_file)
1453 .unwrap_or_else(|_| "Resolved conflicts".to_string())
1454 } else {
1455 "Resolved conflicts".to_string()
1456 };
1457
1458 match self.git_repo.commit(&commit_message) {
1459 Ok(_new_commit_id) => {
1460 Output::success("Cherry-pick completed");
1461
1462 if git_dir.join("CHERRY_PICK_HEAD").exists() {
1464 let _ = std::fs::remove_file(git_dir.join("CHERRY_PICK_HEAD"));
1465 }
1466 if cherry_pick_msg_file.exists() {
1467 let _ = std::fs::remove_file(&cherry_pick_msg_file);
1468 }
1469
1470 println!();
1471 Output::info("Continuing with rest of stack...");
1472 println!();
1473
1474 self.rebase_with_force_push(stack)
1477 }
1478 Err(e) => {
1479 let result = RebaseResult {
1480 success: false,
1481 branch_mapping: HashMap::new(),
1482 conflicts: Vec::new(),
1483 new_commits: Vec::new(),
1484 error: Some(format!(
1485 "Failed to complete cherry-pick: {}\n\n\
1486 This usually means:\n\
1487 - Git index is locked (another process accessing repo)\n\
1488 - File permissions issue\n\
1489 - Disk space issue\n\n\
1490 Recovery:\n\
1491 1. Check if another Git operation is running\n\
1492 2. Run 'rm -f .git/index.lock' if stale lock exists\n\
1493 3. Run 'git status' to check repo state\n\
1494 4. Retry 'ca sync' after fixing the issue\n\n\
1495 Or abort and start fresh:\n\
1496 โ Run: git cherry-pick --abort\n\
1497 โ Then: ca sync",
1498 e
1499 )),
1500 summary: "Failed to complete cherry-pick".to_string(),
1501 };
1502
1503 Ok(result)
1504 }
1505 }
1506 }
1507}
1508
1509impl RebaseResult {
1510 pub fn get_summary(&self) -> String {
1512 if self.success {
1513 format!("โ
{}", self.summary)
1514 } else {
1515 format!(
1516 "โ Rebase failed: {}",
1517 self.error.as_deref().unwrap_or("Unknown error")
1518 )
1519 }
1520 }
1521
1522 pub fn has_conflicts(&self) -> bool {
1524 !self.conflicts.is_empty()
1525 }
1526
1527 pub fn success_count(&self) -> usize {
1529 self.new_commits.len()
1530 }
1531}
1532
1533#[cfg(test)]
1534mod tests {
1535 use super::*;
1536 use std::path::PathBuf;
1537 use std::process::Command;
1538 use tempfile::TempDir;
1539
1540 #[allow(dead_code)]
1541 fn create_test_repo() -> (TempDir, PathBuf) {
1542 let temp_dir = TempDir::new().unwrap();
1543 let repo_path = temp_dir.path().to_path_buf();
1544
1545 Command::new("git")
1547 .args(["init"])
1548 .current_dir(&repo_path)
1549 .output()
1550 .unwrap();
1551 Command::new("git")
1552 .args(["config", "user.name", "Test"])
1553 .current_dir(&repo_path)
1554 .output()
1555 .unwrap();
1556 Command::new("git")
1557 .args(["config", "user.email", "test@test.com"])
1558 .current_dir(&repo_path)
1559 .output()
1560 .unwrap();
1561
1562 std::fs::write(repo_path.join("README.md"), "# Test").unwrap();
1564 Command::new("git")
1565 .args(["add", "."])
1566 .current_dir(&repo_path)
1567 .output()
1568 .unwrap();
1569 Command::new("git")
1570 .args(["commit", "-m", "Initial"])
1571 .current_dir(&repo_path)
1572 .output()
1573 .unwrap();
1574
1575 (temp_dir, repo_path)
1576 }
1577
1578 #[test]
1579 fn test_conflict_region_creation() {
1580 let region = ConflictRegion {
1581 start: 0,
1582 end: 50,
1583 start_line: 1,
1584 end_line: 3,
1585 our_content: "function test() {\n return true;\n}".to_string(),
1586 their_content: "function test() {\n return true;\n}".to_string(),
1587 };
1588
1589 assert_eq!(region.start_line, 1);
1590 assert_eq!(region.end_line, 3);
1591 assert!(region.our_content.contains("return true"));
1592 assert!(region.their_content.contains("return true"));
1593 }
1594
1595 #[test]
1596 fn test_rebase_strategies() {
1597 assert_eq!(RebaseStrategy::ForcePush, RebaseStrategy::ForcePush);
1598 assert_eq!(RebaseStrategy::Interactive, RebaseStrategy::Interactive);
1599 }
1600
1601 #[test]
1602 fn test_rebase_options() {
1603 let options = RebaseOptions::default();
1604 assert_eq!(options.strategy, RebaseStrategy::ForcePush);
1605 assert!(!options.interactive);
1606 assert!(options.auto_resolve);
1607 assert_eq!(options.max_retries, 3);
1608 }
1609
1610 #[test]
1611 fn test_cleanup_guard_tracks_branches() {
1612 let mut guard = TempBranchCleanupGuard::new();
1613 assert!(guard.branches.is_empty());
1614
1615 guard.add_branch("test-branch-1".to_string());
1616 guard.add_branch("test-branch-2".to_string());
1617
1618 assert_eq!(guard.branches.len(), 2);
1619 assert_eq!(guard.branches[0], "test-branch-1");
1620 assert_eq!(guard.branches[1], "test-branch-2");
1621 }
1622
1623 #[test]
1624 fn test_cleanup_guard_prevents_double_cleanup() {
1625 use std::process::Command;
1626 use tempfile::TempDir;
1627
1628 let temp_dir = TempDir::new().unwrap();
1630 let repo_path = temp_dir.path();
1631
1632 Command::new("git")
1633 .args(["init"])
1634 .current_dir(repo_path)
1635 .output()
1636 .unwrap();
1637
1638 Command::new("git")
1639 .args(["config", "user.name", "Test"])
1640 .current_dir(repo_path)
1641 .output()
1642 .unwrap();
1643
1644 Command::new("git")
1645 .args(["config", "user.email", "test@test.com"])
1646 .current_dir(repo_path)
1647 .output()
1648 .unwrap();
1649
1650 std::fs::write(repo_path.join("test.txt"), "test").unwrap();
1652 Command::new("git")
1653 .args(["add", "."])
1654 .current_dir(repo_path)
1655 .output()
1656 .unwrap();
1657 Command::new("git")
1658 .args(["commit", "-m", "initial"])
1659 .current_dir(repo_path)
1660 .output()
1661 .unwrap();
1662
1663 let git_repo = GitRepository::open(repo_path).unwrap();
1664
1665 git_repo.create_branch("test-temp", None).unwrap();
1667
1668 let mut guard = TempBranchCleanupGuard::new();
1669 guard.add_branch("test-temp".to_string());
1670
1671 guard.cleanup(&git_repo);
1673 assert!(guard.cleaned);
1674
1675 guard.cleanup(&git_repo);
1677 assert!(guard.cleaned);
1678 }
1679
1680 #[test]
1681 fn test_rebase_result() {
1682 let result = RebaseResult {
1683 success: true,
1684 branch_mapping: std::collections::HashMap::new(),
1685 conflicts: vec!["abc123".to_string()],
1686 new_commits: vec!["def456".to_string()],
1687 error: None,
1688 summary: "Test summary".to_string(),
1689 };
1690
1691 assert!(result.success);
1692 assert!(result.has_conflicts());
1693 assert_eq!(result.success_count(), 1);
1694 }
1695}