Support emoji completions in review comments

Author: kycutler

src/common/emoji.ts

@@ -36,3 +36,10 @@ export function emojify(message: string) {
 		return emojiMap?.[code] || s;
 	});
 }
+
+export function getEmojis(): Record<string, string> {
+	if (emojiMap === undefined) {
+		return {};
+	}
+	return emojiMap;
+}

src/extension.ts

@@ -43,6 +43,7 @@ import { UriHandler } from './uriHandler';
 import { CommentDecorationProvider } from './view/commentDecorationProvider';
 import { CompareChanges } from './view/compareChangesTreeDataProvider';
 import { CreatePullRequestHelper } from './view/createPullRequestHelper';
+import { EmojiCompletionProvider } from './view/emojiCompletionProvider';
 import { FileTypeDecorationProvider } from './view/fileTypeDecorationProvider';
 import { GitHubCommitFileSystemProvider } from './view/githubFileContentProvider';
 import { getInMemPRFileSystemProvider } from './view/inMemPRContentProvider';
@@ -181,6 +182,12 @@ async function init(
 	const reviewsManager = new ReviewsManager(context, reposManager, reviewManagers, prsTreeModel, tree, changesTree, telemetry, credentialStore, git, copilotRemoteAgentManager, notificationsManager);
 	context.subscriptions.push(reviewsManager);
 
+	context.subscriptions.push(vscode.languages.registerCompletionItemProvider(
+		{ scheme: Schemes.Comment },
+		new EmojiCompletionProvider(context),
+		':'
+	));
+
 	git.onDidChangeState(() => {
 		Logger.appendLine(`Git initialization state changed: state=${git.state}`, ACTIVATION);
 		reviewsManager.reviewManagers.forEach(reviewManager => reviewManager.updateState(true));

src/view/emojiCompletionProvider.ts

@@ -0,0 +1,65 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import * as vscode from 'vscode';
+import { ensureEmojis, getEmojis } from '../common/emoji';
+import { Schemes } from '../common/uri';
+
+export class EmojiCompletionProvider implements vscode.CompletionItemProvider {
+	private static readonly ID: string = 'EmojiCompletionProvider';
+	private _emojiCompletions: vscode.CompletionItem[] = [];
+
+	constructor(private context: vscode.ExtensionContext) {
+		void this.buildEmojiCompletions();
+	}
+
+	private async buildEmojiCompletions(): Promise<void> {
+		await ensureEmojis(this.context);
+		const emojis = getEmojis();
+
+		for (const [name, emoji] of Object.entries(emojis)) {
+			const completionItem = new vscode.CompletionItem({ label: emoji, description: `:${name}:` }, vscode.CompletionItemKind.Text);
+			completionItem.filterText = `:${name}:`;
+			completionItem.sortText = name;
+			this._emojiCompletions.push(completionItem);
+		}
+	}
+
+	provideCompletionItems(
+		document: vscode.TextDocument,
+		position: vscode.Position,
+		_token: vscode.CancellationToken,
+		_context: vscode.CompletionContext
+	): vscode.ProviderResult<vscode.CompletionItem[] | vscode.CompletionList> {
+		// Only provide completions for comment documents
+		if (document.uri.scheme !== Schemes.Comment) {
+			return [];
+		}
+
+		let wordRange = document.getWordRangeAtPosition(position);
+		let wordAtPos = wordRange ? document.getText(wordRange) : undefined;
+		if (!wordRange || wordAtPos?.charAt(0) !== ':') {
+			const start = wordRange?.start ?? position;
+			const testWordRange = new vscode.Range(start.translate(undefined, start.character ? -1 : 0), position);
+			const testWord = document.getText(testWordRange);
+			if (testWord.charAt(0) === ':') {
+				wordRange = testWordRange;
+				wordAtPos = testWord;
+			}
+		}
+
+		// Only provide completions if we're in an emoji context (don't clutter Ctrl+Space results)
+		if (!wordRange) {
+			return [];
+		}
+
+		// Update the range on cached items directly
+		for (const item of this._emojiCompletions) {
+			item.range = wordRange;
+		}
+
+		return new vscode.CompletionList(this._emojiCompletions, false);
+	}
+}