Merge pull request #1658 from GitoxideLabs/merge

merge-base flexibility

gitoxide-core/src/repository/merge/commit.rs

@@ -0,0 +1,88 @@
+use crate::OutputFormat;
+use anyhow::{anyhow, bail, Context};
+use gix::bstr::BString;
+use gix::bstr::ByteSlice;
+use gix::merge::tree::UnresolvedConflict;
+use gix::prelude::Write;
+
+use super::tree::Options;
+
+#[allow(clippy::too_many_arguments)]
+pub fn commit(
+    mut repo: gix::Repository,
+    out: &mut dyn std::io::Write,
+    err: &mut dyn std::io::Write,
+    ours: BString,
+    theirs: BString,
+    Options {
+        format,
+        file_favor,
+        in_memory,
+    }: Options,
+) -> anyhow::Result<()> {
+    if format != OutputFormat::Human {
+        bail!("JSON output isn't implemented yet");
+    }
+    repo.object_cache_size_if_unset(repo.compute_object_cache_size_for_tree_diffs(&**repo.index_or_empty()?));
+    if in_memory {
+        repo.objects.enable_object_memory();
+    }
+    let (ours_ref, ours_id) = refname_and_commit(&repo, ours)?;
+    let (theirs_ref, theirs_id) = refname_and_commit(&repo, theirs)?;
+
+    let options = repo.tree_merge_options()?.with_file_favor(file_favor);
+    let ours_id_str = ours_id.to_string();
+    let theirs_id_str = theirs_id.to_string();
+    let labels = gix::merge::blob::builtin_driver::text::Labels {
+        ancestor: None,
+        current: ours_ref
+            .as_ref()
+            .map_or(ours_id_str.as_str().into(), |n| n.as_bstr())
+            .into(),
+        other: theirs_ref
+            .as_ref()
+            .map_or(theirs_id_str.as_str().into(), |n| n.as_bstr())
+            .into(),
+    };
+    let res = repo
+        .merge_commits(ours_id, theirs_id, labels, options.into())?
+        .tree_merge;
+    let has_conflicts = res.conflicts.is_empty();
+    let has_unresolved_conflicts = res.has_unresolved_conflicts(UnresolvedConflict::Renames);
+    {
+        let _span = gix::trace::detail!("Writing merged tree");
+        let mut written = 0;
+        let tree_id = res
+            .tree
+            .detach()
+            .write(|tree| {
+                written += 1;
+                repo.write(tree)
+            })
+            .map_err(|err| anyhow!("{err}"))?;
+        writeln!(out, "{tree_id} (wrote {written} trees)")?;
+    }
+
+    if !has_conflicts {
+        writeln!(err, "{} possibly resolved conflicts", res.conflicts.len())?;
+    }
+    if has_unresolved_conflicts {
+        bail!("Tree conflicted")
+    }
+    Ok(())
+}
+
+fn refname_and_commit(
+    repo: &gix::Repository,
+    revspec: BString,
+) -> anyhow::Result<(Option<BString>, gix::hash::ObjectId)> {
+    let spec = repo.rev_parse(revspec.as_bstr())?;
+    let commit_id = spec
+        .single()
+        .context("Expected revspec to expand to a single rev only")?
+        .object()?
+        .peel_to_commit()?
+        .id;
+    let refname = spec.first_reference().map(|r| r.name.shorten().as_bstr().to_owned());
+    Ok((refname, commit_id))
+}

gitoxide-core/src/repository/merge/mod.rs

@@ -3,3 +3,6 @@ pub use file::file;
 
 pub mod tree;
 pub use tree::function::tree;
+
+mod commit;
+pub use commit::commit;