rewrite: add conflict labels for squashed commits

An example of squashing `ysrnknol` into `rtsqusxu`:

```
<<<<<<< conflict 1 of 1
+++++++ squash destination (rtsqusxu 2768b0b9)
updated in destination
%%%%%%% squashed commit (ysrnknol 7a20f389) compared with parents of ysrnknol 7a20f389
-base
+squashed
>>>>>>> conflict 1 of 1 ends
```

Author: scott2000

cli/tests/test_squash_command.rs

@@ -869,26 +869,26 @@ fn test_squash_from_multiple() {
 
     // Squash a few commits sideways
     let output = work_dir.run_jj(["squash", "--from=b", "--from=c", "--into=d"]);
-    insta::assert_snapshot!(output, @r###"
+    insta::assert_snapshot!(output, @r"
     ------- stderr -------
     Rebased 2 descendant commits
-    Working copy  (@) now at: kpqxywon 703c6f0c f | (no description set)
-    Parent commit (@-)      : yostqsxw 3d6a1899 e | (no description set)
+    Working copy  (@) now at: kpqxywon 49727075 f | (no description set)
+    Parent commit (@-)      : yostqsxw 2e4142e6 e | (no description set)
     New conflicts appeared in 1 commits:
-      yqosqzyt a3221d7a d | (conflict) (no description set)
+      yqosqzyt 5ae324ae d | (conflict) (no description set)
     Hint: To resolve the conflicts, start by creating a commit on top of
     the conflicted commit:
       jj new yqosqzyt
     Then use `jj resolve`, or edit the conflict markers in the file directly.
     Once the conflicts are resolved, you can inspect the result with `jj diff`.
     Then run `jj squash` to move the resolution into the conflicted commit.
     [EOF]
-    "###);
+    ");
     insta::assert_snapshot!(get_log_output(&work_dir), @r"
-    @  703c6f0cae6f f
-    ○    3d6a18995cae e
+    @  49727075f875 f
+    ○    2e4142e67bae e
     ├─╮
-    × │  a3221d7ae02a d
+    × │  5ae324aed855 d
     ├─╯
     ○  e88768e65e67 a b c
     ◆  000000000000 (empty)
@@ -898,13 +898,13 @@ fn test_squash_from_multiple() {
     let output = work_dir.run_jj(["file", "show", "-r=d", "file"]);
     insta::assert_snapshot!(output, @r"
     <<<<<<< conflict 1 of 1
-    %%%%%%% side #1 compared with base #1
+    %%%%%%% squash destination (yqosqzyt 8acbb715) compared with parents of kkmpptxz fed4d1a2
     -a
     +d
-    %%%%%%% side #2 compared with base #2
+    %%%%%%% squashed commit (kkmpptxz fed4d1a2) compared with parents of mzvwutvl d7e94ec7
     -a
     +b
-    +++++++ side #3
+    +++++++ squashed commit (mzvwutvl d7e94ec7)
     c
     >>>>>>> conflict 1 of 1 ends
     [EOF]
@@ -1013,29 +1013,29 @@ fn test_squash_from_multiple_partial() {
 
     // Partially squash a few commits sideways
     let output = work_dir.run_jj(["squash", "--from=b|c", "--into=d", "file1"]);
-    insta::assert_snapshot!(output, @r###"
+    insta::assert_snapshot!(output, @r"
     ------- stderr -------
     Rebased 2 descendant commits
-    Working copy  (@) now at: kpqxywon f3ae0274 f | (no description set)
-    Parent commit (@-)      : yostqsxw 45ad30bd e | (no description set)
+    Working copy  (@) now at: kpqxywon de844869 f | (no description set)
+    Parent commit (@-)      : yostqsxw c6dd948e e | (no description set)
     New conflicts appeared in 1 commits:
-      yqosqzyt 15efa8c0 d | (conflict) (no description set)
+      yqosqzyt 95cc1bb7 d | (conflict) (no description set)
     Hint: To resolve the conflicts, start by creating a commit on top of
     the conflicted commit:
       jj new yqosqzyt
     Then use `jj resolve`, or edit the conflict markers in the file directly.
     Once the conflicts are resolved, you can inspect the result with `jj diff`.
     Then run `jj squash` to move the resolution into the conflicted commit.
     [EOF]
-    "###);
+    ");
     insta::assert_snapshot!(get_log_output(&work_dir), @r"
-    @  f3ae0274fb6c f
-    ○      45ad30bdccc6 e
+    @  de8448696599 f
+    ○      c6dd948e8e55 e
     ├─┬─╮
     │ │ ○  e9db15b956c4 b
     │ ○ │  83cbe51db94d c
     │ ├─╯
-    × │  15efa8c069e0 d
+    × │  95cc1bb77ab6 d
     ├─╯
     ○  64ea60be8d77 a
     ◆  000000000000 (empty)
@@ -1056,13 +1056,13 @@ fn test_squash_from_multiple_partial() {
     let output = work_dir.run_jj(["file", "show", "-r=d", "file1"]);
     insta::assert_snapshot!(output, @r"
     <<<<<<< conflict 1 of 1
-    %%%%%%% side #1 compared with base #1
+    %%%%%%% squash destination (yqosqzyt f6812ff8) compared with parents of kkmpptxz f2c9709f
     -a
     +d
-    %%%%%%% side #2 compared with base #2
+    %%%%%%% squashed commit (selected changes from kkmpptxz f2c9709f) compared with parents of mzvwutvl aa908686
     -a
     +b
-    +++++++ side #3
+    +++++++ squashed commit (selected changes from mzvwutvl aa908686)
     c
     >>>>>>> conflict 1 of 1 ends
     [EOF]

lib/src/merge.rs

@@ -22,6 +22,7 @@ use std::fmt::Formatter;
 use std::fmt::Write as _;
 use std::future::Future;
 use std::hash::Hash;
+use std::iter;
 use std::iter::zip;
 use std::slice;
 use std::sync::Arc;
@@ -196,6 +197,15 @@ impl<T> Merge<T> {
         Self { values }
     }
 
+    /// Creates a `Merge` from a first side and a series of diffs to apply to
+    /// that side.
+    pub fn from_diffs(first_side: T, diffs: impl IntoIterator<Item = Diff<T>>) -> Self {
+        let values = iter::once(first_side)
+            .chain(diffs.into_iter().flat_map(|diff| [diff.before, diff.after]))
+            .collect();
+        Self { values }
+    }
+
     /// Creates a `Merge` with a single resolved value.
     pub const fn resolved(value: T) -> Self {
         Self {

lib/src/rewrite.rs

@@ -39,6 +39,7 @@ use crate::index::Index;
 use crate::index::IndexResult;
 use crate::matchers::Matcher;
 use crate::matchers::Visit;
+use crate::merge::Diff;
 use crate::merge::Merge;
 use crate::merged_tree::MergedTree;
 use crate::merged_tree::MergedTreeBuilder;
@@ -1152,6 +1153,20 @@ impl CommitWithSelection {
     pub fn is_empty_selection(&self) -> bool {
         !self.selected_tree.id().has_changes(&self.parent_tree.id())
     }
+
+    /// Returns conflict labels for the diff represented by this selection.
+    pub fn conflict_labels(&self) -> BackendResult<Diff<String>> {
+        let commit_label = self.commit.conflict_label();
+        let parent_label = format!("parents of {commit_label}");
+        if self.is_full_selection() {
+            Ok(Diff::new(parent_label, commit_label))
+        } else {
+            Ok(Diff::new(
+                parent_label,
+                format!("selected changes from {commit_label}"),
+            ))
+        }
+    }
 }
 
 /// Resulting commit builder and stats to be returned by [`squash_commits()`].
@@ -1174,6 +1189,7 @@ pub fn squash_commits<'repo>(
 ) -> BackendResult<Option<SquashedCommit<'repo>>> {
     struct SourceCommit<'a> {
         commit: &'a CommitWithSelection,
+        labels: Diff<String>,
         abandon: bool,
     }
     let mut source_commits = vec![];
@@ -1190,6 +1206,7 @@ pub fn squash_commits<'repo>(
         // squash -r`)? The source tree will be unchanged in that case.
         source_commits.push(SourceCommit {
             commit: source,
+            labels: source.conflict_labels()?,
             abandon,
         });
     }
@@ -1206,12 +1223,18 @@ pub fn squash_commits<'repo>(
         } else {
             let source_tree = source.commit.commit.tree()?;
             // Apply the reverse of the selected changes onto the source
-            let new_source_tree = source_tree
-                .merge_unlabeled(
+            let new_source_tree = MergedTree::merge(Merge::from_vec(vec![
+                (source_tree, source.commit.commit.conflict_label()),
+                (
                     source.commit.selected_tree.clone(),
+                    source.labels.after.clone(),
+                ),
+                (
                     source.commit.parent_tree.clone(),
-                )
-                .block_on()?;
+                    source.labels.before.clone(),
+                ),
+            ]))
+            .block_on()?;
             repo.rewrite_commit(&source.commit.commit)
                 .set_tree_id(new_source_tree.id().clone())
                 .write()?;
@@ -1238,22 +1261,29 @@ pub fn squash_commits<'repo>(
             };
         })?;
     }
-    // Apply the selected changes onto the destination
-    let mut destination_tree = rewritten_destination.tree()?;
-    for source in &source_commits {
-        destination_tree = destination_tree
-            .merge_unlabeled(
-                source.commit.parent_tree.clone(),
-                source.commit.selected_tree.clone(),
-            )
-            .block_on()?;
-    }
     let mut predecessors = vec![destination.id().clone()];
     predecessors.extend(
         source_commits
             .iter()
             .map(|source| source.commit.commit.id().clone()),
     );
+    // Apply the selected changes onto the destination
+    let destination_tree = MergedTree::merge(Merge::from_diffs(
+        (
+            rewritten_destination.tree()?,
+            format!("squash destination ({})", destination.conflict_label()),
+        ),
+        source_commits.into_iter().map(|source| {
+            Diff::new(
+                (source.commit.parent_tree.clone(), source.labels.before),
+                (
+                    source.commit.selected_tree.clone(),
+                    format!("squashed commit ({})", source.labels.after),
+                ),
+            )
+        }),
+    ))
+    .block_on()?;
 
     let commit_builder = repo
         .rewrite_commit(&rewritten_destination)