Implement reword_commit as an example of rebase engine usage

Author: Caleb-T-Owens

apps/desktop/src/components/profileSettings/ExperimentalSettings.svelte

@@ -1,6 +1,6 @@
 <script lang="ts">
 	import { SETTINGS_SERVICE } from '$lib/config/appSettingsV2';
-	import { ircEnabled, ircServer, fModeEnabled } from '$lib/config/uiFeatureFlags';
+	import { ircEnabled, ircServer, fModeEnabled, useNewRebaseEngine } from '$lib/config/uiFeatureFlags';
 	import { USER } from '$lib/user/user';
 	import { inject } from '@gitbutler/core/context';
 	import { CardGroup, Textbox, Toggle } from '@gitbutler/ui';
@@ -83,6 +83,21 @@
 			/>
 		{/snippet}
 	</CardGroup.Item>
+	<CardGroup.Item labelFor="new-rebase-engine">
+		{#snippet title()}
+			New rebase engine
+		{/snippet}
+		{#snippet caption()}
+			Use the new graph-based rebase engine for stack operations.
+		{/snippet}
+		{#snippet actions()}
+			<Toggle
+				id="new-rebase-engine"
+				checked={$useNewRebaseEngine}
+				onclick={() => useNewRebaseEngine.set(!$useNewRebaseEngine)}
+			/>
+		{/snippet}
+	</CardGroup.Item>
 
 	{#if $user?.role === 'admin'}
 		<CardGroup.Item labelFor="single-branch">

apps/desktop/src/lib/config/uiFeatureFlags.ts

@@ -15,3 +15,4 @@ export type StagingBehavior = 'all' | 'selection' | 'none';
 export const stagingBehaviorFeature = persisted<StagingBehavior>('all', 'feature-staging-behavior');
 export const fModeEnabled = persisted(true, 'f-mode');
 export const newlineOnEnter = persisted(false, 'feature-newline-on-enter');
+export const useNewRebaseEngine = persisted(false, 'feature-use-new-rebase-engine');

apps/desktop/src/lib/stacks/stackService.svelte.ts

@@ -1,3 +1,4 @@
+import { useNewRebaseEngine } from '$lib/config/uiFeatureFlags';
 import { ConflictEntries, type ConflictEntriesObj } from '$lib/files/conflicts';
 import { sortLikeFileTree } from '$lib/files/filetreeV3';
 import { showToast } from '$lib/notifications/toasts';
@@ -28,6 +29,7 @@ import {
 	type ThunkDispatch,
 	type UnknownAction
 } from '@reduxjs/toolkit';
+import { get } from 'svelte/store';
 import type { StackOrder } from '$lib/branches/branch';
 import type { Commit, CommitDetails, UpstreamCommit } from '$lib/branches/v3';
 import type { MoveCommitIllegalAction } from '$lib/commits/commit';
@@ -630,7 +632,11 @@ export class StackService {
 	}
 
 	get updateCommitMessage() {
-		return this.api.endpoints.updateCommitMessage.useMutation();
+		if (get(useNewRebaseEngine)) {
+			return this.api.endpoints.updateCommitMessage.useMutation();
+		} else {
+			return this.api.endpoints.legacyUpdateCommitMessage.useMutation();
+		}
 	}
 
 	get newBranch() {
@@ -1181,7 +1187,7 @@ function injectEndpoints(api: ClientState['backendApi'], uiState: UiState) {
 					};
 				}
 			}),
-			updateCommitMessage: build.mutation<
+			legacyUpdateCommitMessage: build.mutation<
 				string,
 				{ projectId: string; stackId: string; commitId: string; message: string }
 			>({
@@ -1192,6 +1198,17 @@ function injectEndpoints(api: ClientState['backendApi'], uiState: UiState) {
 				query: (args) => args,
 				invalidatesTags: () => [invalidatesList(ReduxTag.HeadSha)]
 			}),
+			updateCommitMessage: build.mutation<
+				string,
+				{ projectId: string; commitId: string; message: string }
+			>({
+				extraOptions: {
+					command: 'reword_commit',
+					actionName: 'Update Commit Message'
+				},
+				query: (args) => args,
+				invalidatesTags: () => [invalidatesList(ReduxTag.HeadSha)]
+			}),
 			newBranch: build.mutation<
 				void,
 				{ projectId: string; stackId: string; request: { targetPatch?: string; name: string } }

crates/but-api/src/commit.rs

@@ -0,0 +1,44 @@
+use bstr::{BString, ByteSlice};
+use but_api_macros::but_api;
+use but_oplog::legacy::{OperationKind, SnapshotDetails};
+use tracing::instrument;
+
+/// Rewords a commit, but without updating the oplog.
+///
+/// Returns the ID of the newly renamed commit
+#[but_api]
+#[instrument(err(Debug))]
+pub fn reword_commit_only(
+    ctx: &but_ctx::Context,
+    commit_id: gix::ObjectId,
+    message: BString,
+) -> anyhow::Result<gix::ObjectId> {
+    let mut guard = ctx.exclusive_worktree_access();
+    let (repo, _, graph) = ctx.graph_and_meta_mut_and_repo_from_head(guard.write_permission())?;
+
+    but_workspace::commit::reword(&graph, &repo, commit_id, message.as_bstr())
+}
+
+/// Apply `existing_branch` to the workspace in the repository that `ctx` refers to, or create the workspace with default name.
+///
+/// Returns the ID of the newly renamed commit
+#[but_api]
+#[instrument(err(Debug))]
+pub fn reword_commit(
+    ctx: &but_ctx::Context,
+    commit_id: gix::ObjectId,
+    message: BString,
+) -> anyhow::Result<gix::ObjectId> {
+    // NOTE: since this is optional by nature, the same would be true if snapshotting/undo would be disabled via `ctx` app settings, for instance.
+    let maybe_oplog_entry = but_oplog::UnmaterializedOplogSnapshot::from_details(
+        ctx,
+        SnapshotDetails::new(OperationKind::UpdateCommitMessage),
+    )
+    .ok();
+
+    let res = reword_commit_only(ctx, commit_id, message);
+    if let Some(snapshot) = maybe_oplog_entry.filter(|_| res.is_ok()) {
+        snapshot.commit(ctx).ok();
+    };
+    res
+}

crates/but-api/src/lib.rs

@@ -16,6 +16,9 @@ pub mod github;
 /// Functions that take a branch as input.
 pub mod branch;
 
+/// Functions that operate commits
+pub mod commit;
+
 /// Functions that show what changed in various Git entities, like trees, commits and the worktree.
 pub mod diff;
 

crates/but-rebase/src/graph_rebase/materialize.rs

@@ -1,4 +1,6 @@
 //! Functions for materializing a rebase
+use std::collections::HashMap;
+
 use crate::graph_rebase::{Checkouts, rebase::SuccessfulRebase};
 use anyhow::Result;
 use but_core::{
@@ -9,9 +11,16 @@ use but_core::{
     },
 };
 
+/// The outcome of a materialize
+#[derive(Debug, Clone)]
+pub struct MaterializeOutcome {
+    /// A mapping of any commits that were rewritten as part of the rebase
+    pub commit_mapping: HashMap<gix::ObjectId, gix::ObjectId>,
+}
+
 impl SuccessfulRebase {
     /// Materializes a history rewrite
-    pub fn materialize(mut self) -> Result<()> {
+    pub fn materialize(mut self) -> Result<MaterializeOutcome> {
         let repo = self.repo.clone();
         if let Some(memory) = self.repo.objects.take_object_memory() {
             memory.persist(self.repo)?;
@@ -40,6 +49,8 @@ impl SuccessfulRebase {
 
         repo.edit_references(self.ref_edits.clone())?;
 
-        Ok(())
+        Ok(MaterializeOutcome {
+            commit_mapping: self.commit_mapping,
+        })
     }
 }

crates/but-workspace/src/commit.rs renamed to crates/but-workspace/src/commit/mod.rs

@@ -4,6 +4,9 @@ use but_core::ref_metadata::MaybeDebug;
 
 use crate::WorkspaceCommit;
 
+pub mod reword;
+pub use reword::function::reword;
+
 /// A minimal stack for use by [WorkspaceCommit::new_from_stacks()].
 #[derive(Clone)]
 pub struct Stack {

crates/but-workspace/src/commit/reword.rs

@@ -0,0 +1,35 @@
+//! An action to perform a reword of a commit
+
+pub(crate) mod function {
+    use anyhow::Result;
+    use bstr::BStr;
+    use but_rebase::{
+        commit::DateMode,
+        graph_rebase::{GraphExt, Step},
+    };
+
+    /// This action will rewrite a commit and any relevant history so it uses
+    /// the new name.
+    ///
+    /// Returns the ID of the newly renamed commit
+    pub fn reword(
+        graph: &but_graph::Graph,
+        repo: &gix::Repository,
+        commit_id: gix::ObjectId,
+        new_message: &BStr,
+    ) -> Result<gix::ObjectId> {
+        let mut editor = graph.to_editor(repo)?;
+        let target_selector = editor.select_commit(commit_id)?;
+
+        let mut commit = editor.find_commit(commit_id)?;
+        commit.message = new_message.to_owned();
+        let new_id = editor.write_commit(commit, DateMode::CommitterUpdateAuthorKeep)?;
+
+        editor.replace(&target_selector, Step::new_pick(new_id));
+
+        let outcome = editor.rebase()?;
+        outcome.materialize()?;
+
+        Ok(new_id)
+    }
+}

crates/but-workspace/src/lib.rs

@@ -47,7 +47,7 @@ pub mod branch;
 mod changeset;
 
 /// Utility types for the [`WorkspaceCommit`].
-pub(crate) mod commit;
+pub mod commit;
 
 /// Types used only when obtaining head-information.
 ///

crates/gitbutler-tauri/src/main.rs

@@ -14,7 +14,7 @@
 use std::sync::Arc;
 
 use anyhow::bail;
-use but_api::{diff, github, legacy};
+use but_api::{commit, diff, github, legacy};
 use but_claude::{Broadcaster, Claude};
 use but_settings::AppSettingsWithDiskSync;
 use gitbutler_tauri::{
@@ -383,6 +383,7 @@ fn main() -> anyhow::Result<()> {
                 claude::claude_cancel_session,
                 claude::claude_is_stack_active,
                 claude::claude_compact_history,
+                commit::tauri_reword_commit::reword_commit
             ])
             .menu(move |handle| menu::build(handle, &app_settings_for_menu))
             .on_window_event(|window, event| match event {