diff --git a/extensions/vscode-vue-language-features/src/features/fileReferences.ts b/extensions/vscode-vue-language-features/src/features/fileReferences.ts index 190e7ab99..d30a7fe6a 100644 --- a/extensions/vscode-vue-language-features/src/features/fileReferences.ts +++ b/extensions/vscode-vue-language-features/src/features/fileReferences.ts @@ -6,7 +6,7 @@ import * as nls from 'vscode-nls'; const localize = nls.loadMessageBundle(); export async function register(context: vscode.ExtensionContext, client: BaseLanguageClient) { - vscode.commands.registerCommand('vue.findAllFileReferences', async (uri: vscode.Uri) => { + vscode.commands.registerCommand('vue.findAllFileReferences', async (uri?: vscode.Uri) => { // https://github.com/microsoft/vscode/blob/main/extensions/typescript-language-features/src/languageFeatures/fileReferences.ts await vscode.window.withProgress({ @@ -14,6 +14,13 @@ export async function register(context: vscode.ExtensionContext, client: BaseLan title: localize('progress.title', "Finding file references") }, async (_progress, token) => { + if (!uri) { + const editor = vscode.window.activeTextEditor; + if (!editor) return; + + uri = editor.document.uri; + } + const response = await client.sendRequest(shared.FindFileReferenceRequest.type, { textDocument: { uri: uri.toString() } }); if (!response) { return;