Updates generate rebase commands to virtual document

Author: d13

src/commands/generateRebase.ts

@@ -2,6 +2,8 @@ import type { CancellationToken, ProgressOptions } from 'vscode';
 import { ProgressLocation } from 'vscode';
 import type { Source } from '../constants.telemetry';
 import type { Container } from '../container';
+import type { MarkdownContentMetadata } from '../documents/markdown';
+import { getMarkdownHeaderContent } from '../documents/markdown';
 import type { GitRepositoryService } from '../git/gitRepositoryService';
 import type { GitReference } from '../git/models/reference';
 import { uncommitted } from '../git/models/revision';
@@ -176,11 +178,15 @@ export async function generateRebase(
 		const diffInfo = extractRebaseDiffInfo(result.commits, result.diff, result.hunkMap);
 
 		// Generate the markdown content that shows each commit and its diffs
-		const markdownContent = generateRebaseMarkdown(result, title);
+		const { content, metadata } = generateRebaseMarkdown(result, title);
 
+		let generateType: 'commits' | 'rebase' = 'rebase';
+		let headRefSlug = head.ref;
 		const shas = await repo.git.patch?.createUnreachableCommitsFromPatches(base.ref, diffInfo);
 		if (shas?.length) {
 			if (head.ref === uncommitted) {
+				generateType = 'commits';
+				headRefSlug = 'uncommitted';
 				// stash the working changes
 				await svc.stash?.saveStash(undefined, undefined, { includeUntracked: true });
 				// await repo.git.checkout?.(shas[shas.length - 1]);
@@ -191,7 +197,14 @@ export async function generateRebase(
 			}
 		}
 
-		showMarkdownPreview(markdownContent);
+		const documentUri = container.markdown.openDocument(
+			content,
+			`/generate/${generateType}/${headRefSlug}/${result.model.id}`,
+			metadata.header.title,
+			metadata,
+		);
+
+		showMarkdownPreview(documentUri);
 	} catch (ex) {
 		Logger.error(ex, 'GenerateRebaseCommand', 'execute');
 		void showGenericErrorMessage('Unable to parse rebase result');
@@ -256,18 +269,28 @@ export function extractRebaseDiffInfo(
 /**
  * Formats the reorganized commits into a readable markdown document with proper git diff format
  */
-function generateRebaseMarkdown(result: AIRebaseResult, title = 'Rebase Commits'): string {
-	let markdown = `# ${title}\n\n> Generated by ${result.model.name}\n\n`;
-
-	const { commits, diff: originalDiff, hunkMap } = result;
-
-	if (commits.length === 0) {
-		markdown += `No commits generated\n\n`;
-		return markdown;
+function generateRebaseMarkdown(
+	result: AIRebaseResult,
+	title = 'Rebase Commits',
+): { content: string; metadata: MarkdownContentMetadata } {
+	const metadata = {
+		header: {
+			title: title,
+			aiModel: result.model.name,
+			subtitle: 'Explanation',
+		},
+	};
+
+	let markdown = '';
+	if (result.commits.length === 0) {
+		markdown = 'No Commits Generated';
+
+		return { content: `${getMarkdownHeaderContent(metadata)}\n\n${markdown}`, metadata: metadata };
 	}
+	const { commits, diff: originalDiff, hunkMap } = result;
 
 	let explanations =
-		"## Explanation\n\nOkay, here's the breakdown of the commits I'd create from the provided diff, along with explanations for each:\n\n";
+		"Okay, here's the breakdown of the commits created from the provided changes, along with explanations for each:\n\n";
 
 	let changes = '## Commits\n\n';
 	for (let i = 0; i < commits.length; i++) {
@@ -329,7 +352,7 @@ function generateRebaseMarkdown(result: AIRebaseResult, title = 'Rebase Commits'
 	// markdown += `\n\n----\n\n## Raw commits\n\n\`\`\`${escapeMarkdownCodeBlocks(JSON.stringify(commits))}\`\`\``;
 	// markdown += `\n\n----\n\n## Original Diff\n\n\`\`\`${escapeMarkdownCodeBlocks(originalDiff)}\`\`\`\n`;
 
-	return markdown;
+	return { content: `${getMarkdownHeaderContent(metadata)}\n\n${markdown}`, metadata: metadata };
 }
 
 /**