Add code citation message to string representation (#223160)

For accessibility

Author: roblourens

src/vs/workbench/contrib/chat/browser/chatContentParts/chatCodeCitationContentPart.ts

@@ -12,6 +12,7 @@ import { localize } from 'vs/nls';
 import { ITelemetryService } from 'vs/platform/telemetry/common/telemetry';
 import { ChatTreeItem } from 'vs/workbench/contrib/chat/browser/chat';
 import { IChatContentPart, IChatContentPartRenderContext } from 'vs/workbench/contrib/chat/browser/chatContentParts/chatContentParts';
+import { getCodeCitationsMessage } from 'vs/workbench/contrib/chat/common/chatModel';
 import { IChatCodeCitations, IChatRendererContent } from 'vs/workbench/contrib/chat/common/chatViewModel';
 import { IEditorService } from 'vs/workbench/services/editor/common/editorService';
 
@@ -31,10 +32,7 @@ export class ChatCodeCitationContentPart extends Disposable implements IChatCont
 	) {
 		super();
 
-		const licenseTypes = citations.citations.reduce((set, c) => set.add(c.license), new Set<string>());
-		const label = licenseTypes.size === 1 ?
-			localize('codeCitation', "Similar code found with 1 license type", licenseTypes.size) :
-			localize('codeCitations', "Similar code found with {0} license types", licenseTypes.size);
+		const label = getCodeCitationsMessage(citations.citations);
 		const elements = dom.h('.chat-code-citation-message@root', [
 			dom.h('span.chat-code-citation-label@label'),
 			dom.h('.chat-code-citation-button-container@button'),

src/vs/workbench/contrib/chat/common/chatModel.ts

@@ -176,6 +176,7 @@ export class Response implements IResponse {
 	 */
 	private _markdownContent = '';
 
+	private _citations: IChatCodeCitation[] = [];
 
 	get value(): IChatProgressResponseContent[] {
 		return this._responseParts;
@@ -260,6 +261,11 @@ export class Response implements IResponse {
 		}
 	}
 
+	public addCitation(citation: IChatCodeCitation) {
+		this._citations.push(citation);
+		this._updateRepr();
+	}
+
 	private _updateRepr(quiet?: boolean) {
 		this._responseRepr = this._responseParts.map(part => {
 			if (part.kind === 'treeData') {
@@ -281,6 +287,8 @@ export class Response implements IResponse {
 			.filter(s => s.length > 0)
 			.join('\n\n');
 
+		this._responseRepr += this._citations.length ? '\n\n' + getCodeCitationsMessage(this._citations) : '';
+
 		this._markdownContent = this._responseParts.map(part => {
 			if (part.kind === 'inlineReference') {
 				return basename('uri' in part.inlineReference ? part.inlineReference.uri : part.inlineReference);
@@ -428,6 +436,7 @@ export class ChatResponseModel extends Disposable implements IChatResponseModel
 
 	applyCodeCitation(progress: IChatCodeCitation) {
 		this._codeCitations.push(progress);
+		this._response.addCitation(progress);
 		this._onDidChange.fire();
 	}
 
@@ -1140,3 +1149,15 @@ export function appendMarkdownString(md1: IMarkdownString, md2: IMarkdownString
 		baseUri: md1.baseUri
 	};
 }
+
+export function getCodeCitationsMessage(citations: ReadonlyArray<IChatCodeCitation>): string {
+	if (citations.length === 0) {
+		return '';
+	}
+
+	const licenseTypes = citations.reduce((set, c) => set.add(c.license), new Set<string>());
+	const label = licenseTypes.size === 1 ?
+		localize('codeCitation', "Similar code found with 1 license type", licenseTypes.size) :
+		localize('codeCitations', "Similar code found with {0} license types", licenseTypes.size);
+	return label;
+}