Store AI result context in memory for markdown documents.

Thats because we cannot modify URI of document to review context after
the result is rendered.
(#4328, #4489)

Author: sergeibbb

src/commands/explainBase.ts

@@ -135,10 +135,14 @@ export abstract class ExplainCommandBase extends GlCommandBase {
 		result: AISummarizeResult,
 		metadata: MarkdownContentMetadata,
 	): void {
-		const metadataWithContext: MarkdownContentMetadata = { ...metadata, context: getAIResultContext(result) };
+		const context = getAIResultContext(result);
+		const metadataWithContext: MarkdownContentMetadata = { ...metadata, context: context };
 		const headerContent = getMarkdownHeaderContent(metadataWithContext, this.container.telemetry.enabled);
 		const content = `${headerContent}\n\n${result.parsed.summary}\n\n${result.parsed.body}`;
 
+		// Store the AI result context in the feedback provider for documents that cannot store it in their URI
+		this.container.aiFeedback.setMarkdownDocument(documentUri.toString(), context);
+
 		this.container.markdown.updateDocument(documentUri, content);
 	}
 

src/plus/ai/utils/-webview/ai.utils.ts

@@ -288,8 +288,9 @@ export function extractAIResultContext(container: Container, uri: Uri | undefine
 		if (!authority) return undefined;
 
 		try {
+			const context: AIResultContext | undefined = container.aiFeedback.getMarkdownDocument(uri.toString());
 			const metadata = decodeGitLensRevisionUriAuthority<MarkdownContentMetadata>(authority);
-			return metadata.context;
+			return context ?? metadata.context;
 		} catch (ex) {
 			Logger.error(ex, 'extractResultContext');
 			return undefined;

src/telemetry/aiFeedbackProvider.ts

@@ -12,7 +12,9 @@ export class AIFeedbackProvider implements Disposable {
 	constructor() {
 		// Listen for document close events to clean up contexts
 		this._disposables.push(
-			workspace.onDidCloseTextDocument(document => this.removeChangelogDocument(document.uri)),
+			workspace.onDidCloseTextDocument(document => {
+				this.removeDocument(document.uri);
+			}),
 		);
 	}
 
@@ -21,16 +23,19 @@ export class AIFeedbackProvider implements Disposable {
 		this.addChangelogUri(uri);
 	}
 
-	private removeChangelogDocument(uri: Uri): void {
-		this.deleteChangelogDocument(uri.toString());
+	private removeDocument(uri: Uri): void {
+		const uriString = uri.toString();
+		this.deleteChangelogDocument(uriString);
 		this.removeChangelogUri(uri);
+		this.deleteMarkdownDocument(uriString);
 	}
 
 	private readonly _disposables: Disposable[] = [];
 	dispose(): void {
 		this._disposables.forEach(d => void d.dispose());
 		this._uriResponses.clear();
 		this._changelogDocuments.clear();
+		this._markdownDocuments.clear();
 		this._changelogUris.clear();
 		this._updateFeedbackContextDebounced = undefined;
 		this._updateChangelogContextDebounced = undefined;
@@ -70,6 +75,18 @@ export class AIFeedbackProvider implements Disposable {
 		this._changelogDocuments.delete(documentUri);
 	}
 
+	// Storage for AI feedback context associated with any document
+	private readonly _markdownDocuments = new Map<string, AIResultContext>();
+	getMarkdownDocument(documentUri: string): AIResultContext | undefined {
+		return this._markdownDocuments.get(documentUri);
+	}
+	setMarkdownDocument(documentUri: string, context: AIResultContext): void {
+		this._markdownDocuments.set(documentUri, context);
+	}
+	private deleteMarkdownDocument(documentUri: string): void {
+		this._markdownDocuments.delete(documentUri);
+	}
+
 	// Storage for AI feedback responses by URI
 	private readonly _uriResponses = new UriMap<AIFeedbackEvent['sentiment']>();
 	private _updateFeedbackContextDebounced: Deferrable<() => void> | undefined;