Move usages tool (microsoft#295139)

* feat: add UsagesTool for finding code symbol usages and implement related tests

* feat: handle late-registered tools in LanguageModelToolsExtensionPointHandler

* feat: enhance UsagesTool to utilize ISearchService for symbol searches and improve reference classification

Author: jrieken

src/vs/editor/common/languageFeatureRegistry.ts

@@ -6,7 +6,7 @@
 import { Emitter } from '../../base/common/event.js';
 import { IDisposable, toDisposable } from '../../base/common/lifecycle.js';
 import { ITextModel, shouldSynchronizeModel } from './model.js';
-import { LanguageFilter, LanguageSelector, score } from './languageSelector.js';
+import { LanguageFilter, LanguageSelector, score, selectLanguageIds } from './languageSelector.js';
 import { URI } from '../../base/common/uri.js';
 
 interface Entry<T> {
@@ -115,6 +115,14 @@ export class LanguageFeatureRegistry<T> {
 		return this._entries.map(entry => entry.provider);
 	}
 
+	get registeredLanguageIds(): ReadonlySet<string> {
+		const result = new Set<string>();
+		for (const entry of this._entries) {
+			selectLanguageIds(entry.selector, result);
+		}
+		return result;
+	}
+
 	ordered(model: ITextModel, recursive = false): T[] {
 		const result: T[] = [];
 		this._orderedForEach(model, recursive, entry => result.push(entry.provider));
@@ -226,4 +234,3 @@ function isBuiltinSelector(selector: LanguageSelector): boolean {
 
 	return Boolean((selector as LanguageFilter).isBuiltin);
 }
-

src/vs/editor/common/languageSelector.ts

@@ -142,3 +142,18 @@ export function targetsNotebooks(selector: LanguageSelector): boolean {
 		return !!(<LanguageFilter>selector).notebookType;
 	}
 }
+
+export function selectLanguageIds(selector: LanguageSelector, into: Set<string>): void {
+	if (typeof selector === 'string') {
+		into.add(selector);
+	} else if (Array.isArray(selector)) {
+		for (const item of selector) {
+			selectLanguageIds(item, into);
+		}
+	} else {
+		const language = (<LanguageFilter>selector).language;
+		if (language) {
+			into.add(language);
+		}
+	}
+}

src/vs/editor/test/common/modes/languageSelector.test.ts

@@ -6,7 +6,7 @@
 import assert from 'assert';
 import { URI } from '../../../../base/common/uri.js';
 import { ensureNoDisposablesAreLeakedInTestSuite } from '../../../../base/test/common/utils.js';
-import { LanguageSelector, score } from '../../../common/languageSelector.js';
+import { LanguageSelector, score, selectLanguageIds } from '../../../common/languageSelector.js';
 
 suite('LanguageSelector', function () {
 
@@ -173,4 +173,27 @@ suite('LanguageSelector', function () {
 		}, obj.uri, obj.langId, true, undefined, undefined);
 		assert.strictEqual(value, 0);
 	});
+
+	test('selectLanguageIds', function () {
+		const result = new Set<string>();
+
+		selectLanguageIds('typescript', result);
+		assert.deepStrictEqual([...result], ['typescript']);
+
+		result.clear();
+		selectLanguageIds({ language: 'python', scheme: 'file' }, result);
+		assert.deepStrictEqual([...result], ['python']);
+
+		result.clear();
+		selectLanguageIds({ scheme: 'file' }, result);
+		assert.deepStrictEqual([...result], []);
+
+		result.clear();
+		selectLanguageIds(['javascript', { language: 'css' }, { scheme: 'untitled' }], result);
+		assert.deepStrictEqual([...result].sort(), ['css', 'javascript']);
+
+		result.clear();
+		selectLanguageIds('*', result);
+		assert.deepStrictEqual([...result], ['*']);
+	});
 });

src/vs/workbench/contrib/chat/browser/chat.contribution.ts

@@ -60,6 +60,7 @@ import { IPromptsService } from '../common/promptSyntax/service/promptsService.j
 import { PromptsService } from '../common/promptSyntax/service/promptsServiceImpl.js';
 import { LanguageModelToolsExtensionPointHandler } from '../common/tools/languageModelToolsContribution.js';
 import { BuiltinToolsContribution } from '../common/tools/builtinTools/tools.js';
+import { UsagesToolContribution } from './tools/usagesTool.js';
 import { IVoiceChatService, VoiceChatService } from '../common/voiceChatService.js';
 import { registerChatAccessibilityActions } from './actions/chatAccessibilityActions.js';
 import { AgentChatAccessibilityHelp, EditsChatAccessibilityHelp, PanelChatAccessibilityHelp, QuickChatAccessibilityHelp } from './actions/chatAccessibilityHelp.js';
@@ -1422,6 +1423,7 @@ registerWorkbenchContribution2(ChatSetupContribution.ID, ChatSetupContribution,
 registerWorkbenchContribution2(ChatTeardownContribution.ID, ChatTeardownContribution, WorkbenchPhase.AfterRestored);
 registerWorkbenchContribution2(ChatStatusBarEntry.ID, ChatStatusBarEntry, WorkbenchPhase.BlockRestore);
 registerWorkbenchContribution2(BuiltinToolsContribution.ID, BuiltinToolsContribution, WorkbenchPhase.Eventually);
+registerWorkbenchContribution2(UsagesToolContribution.ID, UsagesToolContribution, WorkbenchPhase.BlockRestore);
 registerWorkbenchContribution2(ChatAgentSettingContribution.ID, ChatAgentSettingContribution, WorkbenchPhase.AfterRestored);
 registerWorkbenchContribution2(ChatAgentActionsContribution.ID, ChatAgentActionsContribution, WorkbenchPhase.Eventually);
 registerWorkbenchContribution2(ToolReferenceNamesContribution.ID, ToolReferenceNamesContribution, WorkbenchPhase.AfterRestored);