@vue/language-service 2.0.6 → 2.0.10

package/lib/plugins/vue-document-drop.js

@@ -6,45 +6,67 @@ const shared_1 = require("@vue/shared");
 const path = require("path-browserify");
 const vue_extract_file_1 = require("../plugins/vue-extract-file");
 const types_1 = require("../types");
-function create(ts) {
+const getUserPreferences_1 = require("volar-service-typescript/lib/configs/getUserPreferences");
+function create(ts, getTsPluginClient) {
     return {
         name: 'vue-document-drop',
         create(context) {
             let casing = types_1.TagNameCasing.Pascal; // TODO
+            const tsPluginClient = getTsPluginClient?.(context);
             return {
                 async provideDocumentDropEdits(document, _position, dataTransfer) {
-                    if (document.languageId !== 'html')
+                    if (document.languageId !== 'html') {
                         return;
-                    const [virtualCode, sourceFile] = context.documents.getVirtualCodeByUri(document.uri);
-                    const vueVirtualCode = sourceFile?.generated?.code;
-                    if (!virtualCode || !(vueVirtualCode instanceof language_core_1.VueGeneratedCode))
+                    }
+                    const decoded = context.decodeEmbeddedDocumentUri(document.uri);
+                    const sourceScript = decoded && context.language.scripts.get(decoded[0]);
+                    const virtualCode = decoded && sourceScript?.generated?.embeddedCodes.get(decoded[1]);
+                    const vueVirtualCode = sourceScript?.generated?.root;
+                    if (!sourceScript || !virtualCode || !(vueVirtualCode instanceof language_core_1.VueGeneratedCode)) {
                         return;
+                    }
                     let importUri;
                     for (const [mimeType, item] of dataTransfer) {
                         if (mimeType === 'text/uri-list') {
                             importUri = item.value;
                         }
                     }
-                    if (!importUri?.endsWith('.vue'))
+                    if (!importUri?.endsWith('.vue')) {
                         return;
+                    }
                     let baseName = importUri.substring(importUri.lastIndexOf('/') + 1);
                     baseName = baseName.substring(0, baseName.lastIndexOf('.'));
                     const newName = (0, shared_1.capitalize)((0, shared_1.camelize)(baseName));
                     const { sfc } = vueVirtualCode;
                     const script = sfc.scriptSetup ?? sfc.script;
-                    if (!script)
+                    if (!script) {
                         return;
+                    }
                     const additionalEdit = {};
                     const code = [...(0, language_core_1.forEachEmbeddedCode)(vueVirtualCode)].find(code => code.id === (sfc.scriptSetup ? 'scriptSetupFormat' : 'scriptFormat'));
                     const lastImportNode = (0, vue_extract_file_1.getLastImportNode)(ts, script.ast);
-                    let importPath = path.relative(path.dirname(document.uri), importUri)
-                        || importUri.substring(importUri.lastIndexOf('/') + 1);
-                    if (!importPath.startsWith('./') && !importPath.startsWith('../')) {
-                        importPath = './' + importPath;
+                    const incomingFileName = context.env.typescript.uriToFileName(importUri);
+                    let importPath;
+                    const serviceScript = sourceScript.generated?.languagePlugin.typescript?.getServiceScript(vueVirtualCode);
+                    if (tsPluginClient && serviceScript) {
+                        const tsDocumentUri = context.encodeEmbeddedDocumentUri(sourceScript.id, serviceScript.code.id);
+                        const tsDocument = context.documents.get(tsDocumentUri, serviceScript.code.languageId, serviceScript.code.snapshot);
+                        const preferences = await (0, getUserPreferences_1.getUserPreferences)(context, tsDocument);
+                        const importPathRequest = await tsPluginClient.getImportPathForFile(vueVirtualCode.fileName, incomingFileName, preferences);
+                        if (importPathRequest) {
+                            importPath = importPathRequest;
+                        }
+                    }
+                    if (!importPath) {
+                        importPath = path.relative(path.dirname(vueVirtualCode.fileName), incomingFileName)
+                            || importUri.substring(importUri.lastIndexOf('/') + 1);
+                        if (!importPath.startsWith('./') && !importPath.startsWith('../')) {
+                            importPath = './' + importPath;
+                        }
                     }
                     additionalEdit.changes ??= {};
-                    additionalEdit.changes[context.documents.getVirtualCodeUri(sourceFile.id, code.id)] = [];
-                    additionalEdit.changes[context.documents.getVirtualCodeUri(sourceFile.id, code.id)].push({
+                    additionalEdit.changes[context.encodeEmbeddedDocumentUri(sourceScript.id, code.id)] = [];
+                    additionalEdit.changes[context.encodeEmbeddedDocumentUri(sourceScript.id, code.id)].push({
                         range: lastImportNode ? {
                             start: script.ast.getLineAndCharacterOfPosition(lastImportNode.end),
                             end: script.ast.getLineAndCharacterOfPosition(lastImportNode.end),
@@ -58,7 +80,7 @@ function create(ts) {
                     if (sfc.script) {
                         const edit = (0, vue_extract_file_1.createAddComponentToOptionEdit)(ts, sfc.script.ast, newName);
                         if (edit) {
-                            additionalEdit.changes[context.documents.getVirtualCodeUri(sourceFile.id, code.id)].push({
+                            additionalEdit.changes[context.encodeEmbeddedDocumentUri(sourceScript.id, code.id)].push({
                                 range: {
                                     start: document.positionAt(edit.range.start),
                                     end: document.positionAt(edit.range.end),

package/lib/plugins/vue-extract-file.d.ts

@@ -1,6 +1,6 @@
-import type { ServicePlugin } from '@volar/language-service';
+import type { ServiceContext, LanguageServicePlugin } from '@volar/language-service';
 import type * as ts from 'typescript';
-export declare function create(ts: typeof import('typescript'), tsPluginClient?: typeof import('@vue/typescript-plugin/lib/client')): ServicePlugin;
+export declare function create(ts: typeof import('typescript'), getTsPluginClient?: (context: ServiceContext) => typeof import('@vue/typescript-plugin/lib/client') | undefined): LanguageServicePlugin;
 export declare function getLastImportNode(ts: typeof import('typescript'), sourceFile: ts.SourceFile): ts.Node | undefined;
 export declare function createAddComponentToOptionEdit(ts: typeof import('typescript'), ast: ts.SourceFile, componentName: string): {
     range: import("@vue/language-core").TextRange;

package/lib/plugins/vue-extract-file.js

@@ -3,10 +3,11 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.createAddComponentToOptionEdit = exports.getLastImportNode = exports.create = void 0;
 const language_core_1 = require("@vue/language-core");
 const unicodeReg = /\\u/g;
-function create(ts, tsPluginClient) {
+function create(ts, getTsPluginClient) {
     return {
         name: 'vue-extract-file',
         create(context) {
+            const tsPluginClient = getTsPluginClient?.(context);
             return {
                 async provideCodeActions(document, range, _context) {
                     const startOffset = document.offsetAt(range.start);
@@ -14,16 +15,21 @@ function create(ts, tsPluginClient) {
                     if (startOffset === endOffset) {
                         return;
                     }
-                    const [code, vueCode] = context.documents.getVirtualCodeByUri(document.uri);
-                    if (!(vueCode?.generated?.code instanceof language_core_1.VueGeneratedCode) || code?.id !== 'template')
+                    const decoded = context.decodeEmbeddedDocumentUri(document.uri);
+                    const sourceScript = decoded && context.language.scripts.get(decoded[0]);
+                    const virtualCode = decoded && sourceScript?.generated?.embeddedCodes.get(decoded[1]);
+                    if (!(sourceScript?.generated?.root instanceof language_core_1.VueGeneratedCode) || virtualCode?.id !== 'template') {
                         return;
-                    const { sfc } = vueCode.generated.code;
+                    }
+                    const { sfc } = sourceScript.generated.root;
                     const script = sfc.scriptSetup ?? sfc.script;
-                    if (!sfc.template || !script)
+                    if (!sfc.template || !script) {
                         return;
+                    }
                     const templateCodeRange = selectTemplateCode(startOffset, endOffset, sfc.template);
-                    if (!templateCodeRange)
+                    if (!templateCodeRange) {
                         return;
+                    }
                     return [
                         {
                             title: 'Extract into new dumb component',
@@ -39,21 +45,27 @@ function create(ts, tsPluginClient) {
                 async resolveCodeAction(codeAction) {
                     const { uri, range, newName } = codeAction.data;
                     const [startOffset, endOffset] = range;
-                    const [code, sourceFile] = context.documents.getVirtualCodeByUri(uri);
-                    if (!(sourceFile?.generated?.code instanceof language_core_1.VueGeneratedCode) || code?.id !== 'template')
+                    const decoded = context.decodeEmbeddedDocumentUri(uri);
+                    const sourceScript = decoded && context.language.scripts.get(decoded[0]);
+                    const virtualCode = decoded && sourceScript?.generated?.embeddedCodes.get(decoded[1]);
+                    if (!(sourceScript?.generated?.root instanceof language_core_1.VueGeneratedCode) || virtualCode?.id !== 'template') {
                         return codeAction;
-                    const document = context.documents.get(uri, code.languageId, code.snapshot);
-                    const sfcDocument = context.documents.get(sourceFile.id, sourceFile.languageId, sourceFile.snapshot);
-                    const { sfc } = sourceFile.generated.code;
+                    }
+                    const document = context.documents.get(uri, virtualCode.languageId, virtualCode.snapshot);
+                    const sfcDocument = context.documents.get(sourceScript.id, sourceScript.languageId, sourceScript.snapshot);
+                    const { sfc } = sourceScript.generated.root;
                     const script = sfc.scriptSetup ?? sfc.script;
-                    if (!sfc.template || !script)
+                    if (!sfc.template || !script) {
                         return codeAction;
+                    }
                     const templateCodeRange = selectTemplateCode(startOffset, endOffset, sfc.template);
-                    if (!templateCodeRange)
+                    if (!templateCodeRange) {
                         return codeAction;
-                    const toExtract = await tsPluginClient?.collectExtractProps(sourceFile.generated.code.fileName, templateCodeRange) ?? [];
-                    if (!toExtract)
+                    }
+                    const toExtract = await tsPluginClient?.collectExtractProps(sourceScript.generated.root.fileName, templateCodeRange) ?? [];
+                    if (!toExtract) {
                         return codeAction;
+                    }
                     const templateInitialIndent = await context.env.getConfiguration('vue.format.template.initialIndent') ?? true;
                     const scriptInitialIndent = await context.env.getConfiguration('vue.format.script.initialIndent') ?? false;
                     const newUri = sfcDocument.uri.substring(0, sfcDocument.uri.lastIndexOf('/') + 1) + `${newName}.vue`;
@@ -114,7 +126,7 @@ function create(ts, tsPluginClient) {
                                 // editing vue sfc
                                 {
                                     textDocument: {
-                                        uri: sourceFile.id,
+                                        uri: sourceScript.id,
                                         version: null,
                                     },
                                     edits: sfcEdits,
@@ -200,8 +212,9 @@ function selectTemplateCode(startOffset, endOffset, templateBlock) {
     }
 }
 function constructTag(name, attributes, initialIndent, content) {
-    if (initialIndent)
+    if (initialIndent) {
         content = content.split('\n').map(line => `\t${line}`).join('\n');
+    }
     const attributesString = attributes.length ? ` ${attributes.join(' ')}` : '';
     return `<${name}${attributesString}>\n${content}\n</${name}>\n`;
 }
@@ -220,8 +233,9 @@ function getLastImportNode(ts, sourceFile) {
 exports.getLastImportNode = getLastImportNode;
 function createAddComponentToOptionEdit(ts, ast, componentName) {
     const exportDefault = language_core_1.scriptRanges.parseScriptRanges(ts, ast, false, true).exportDefault;
-    if (!exportDefault)
+    if (!exportDefault) {
         return;
+    }
     // https://github.com/microsoft/TypeScript/issues/36174
     const printer = ts.createPrinter();
     if (exportDefault.componentsOption && exportDefault.componentsOptionNode) {

package/lib/plugins/vue-sfc.d.ts

@@ -1,7 +1,7 @@
-import type { ServicePlugin } from '@volar/language-service';
+import type { LanguageServicePlugin } from '@volar/language-service';
 import * as vue from '@vue/language-core';
 import type { TextDocument } from 'vscode-languageserver-textdocument';
 export interface Provide {
     'vue/vueFile': (document: TextDocument) => vue.VueGeneratedCode | undefined;
 }
-export declare function create(): ServicePlugin;
+export declare function create(): LanguageServicePlugin;

package/lib/plugins/vue-sfc.js

@@ -12,34 +12,53 @@ function create() {
         create(context) {
             const htmlPlugin = (0, volar_service_html_1.create)({
                 documentSelector: ['vue'],
-                useCustomDataProviders: false,
+                useDefaultDataProvider: false,
+                getCustomData(context) {
+                    sfcDataProvider ??= html.newHTMLDataProvider('vue', (0, data_1.loadLanguageBlocks)(context.env.locale ?? 'en'));
+                    return [sfcDataProvider];
+                },
+                async getFormattingOptions(document, options, context) {
+                    return await worker(document, async (vueCode) => {
+                        const formatSettings = await context.env.getConfiguration?.('html.format') ?? {};
+                        const blockTypes = ['template', 'script', 'style'];
+                        for (const customBlock of vueCode.sfc.customBlocks) {
+                            blockTypes.push(customBlock.type);
+                        }
+                        return {
+                            ...options,
+                            ...formatSettings,
+                            wrapAttributes: 'auto',
+                            unformatted: '',
+                            contentUnformatted: blockTypes.join(','),
+                            endWithNewline: options.insertFinalNewline ? true
+                                : options.trimFinalNewlines ? false
+                                    : document.getText().endsWith('\n'),
+                        };
+                    }) ?? {};
+                },
             }).create(context);
-            const htmlLanguageService = htmlPlugin.provide['html/languageService']();
-            sfcDataProvider ??= html.newHTMLDataProvider('vue', (0, data_1.loadLanguageBlocks)(context.env.locale ?? 'en'));
-            htmlLanguageService.setDataProviders(false, [sfcDataProvider]);
             return {
                 ...htmlPlugin,
                 provide: {
                     'vue/vueFile': document => {
-                        return worker(document, (vueFile) => {
+                        return worker(document, vueFile => {
                             return vueFile;
                         });
                     },
                 },
-                async resolveEmbeddedCodeFormattingOptions(code, options) {
-                    const sourceFile = context.language.files.getByVirtualCode(code);
-                    if (sourceFile.generated?.code instanceof vue.VueGeneratedCode) {
-                        if (code.id === 'scriptFormat' || code.id === 'scriptSetupFormat') {
+                async resolveEmbeddedCodeFormattingOptions(sourceScript, virtualCode, options) {
+                    if (sourceScript.generated?.root instanceof vue.VueGeneratedCode) {
+                        if (virtualCode.id === 'scriptFormat' || virtualCode.id === 'scriptSetupFormat') {
                             if (await context.env.getConfiguration?.('vue.format.script.initialIndent') ?? false) {
                                 options.initialIndentLevel++;
                             }
                         }
-                        else if (code.id.startsWith('style_')) {
+                        else if (virtualCode.id.startsWith('style_')) {
                             if (await context.env.getConfiguration?.('vue.format.style.initialIndent') ?? false) {
                                 options.initialIndentLevel++;
                             }
                         }
-                        else if (code.id === 'template') {
+                        else if (virtualCode.id === 'template') {
                             if (await context.env.getConfiguration?.('vue.format.template.initialIndent') ?? true) {
                                 options.initialIndentLevel++;
                             }
@@ -49,7 +68,7 @@ function create() {
                 },
                 provideDocumentLinks: undefined,
                 provideDocumentSymbols(document) {
-                    return worker(document, (vueSourceFile) => {
+                    return worker(document, vueSourceFile => {
                         const result = [];
                         const descriptor = vueSourceFile.sfc;
                         if (descriptor.template) {
@@ -96,10 +115,12 @@ function create() {
                         }
                         for (const style of descriptor.styles) {
                             let name = 'style';
-                            if (style.scoped)
+                            if (style.scoped) {
                                 name += ' scoped';
-                            if (style.module)
+                            }
+                            if (style.module) {
                                 name += ' module';
+                            }
                             result.push({
                                 name,
                                 kind: 2,
@@ -130,29 +151,13 @@ function create() {
                         return result;
                     });
                 },
-                provideDocumentFormattingEdits(document, range, options) {
-                    return worker(document, async (vueCode) => {
-                        const formatSettings = await context.env.getConfiguration?.('html.format') ?? {};
-                        const blockTypes = ['template', 'script', 'style'];
-                        for (const customBlock of vueCode.sfc.customBlocks) {
-                            blockTypes.push(customBlock.type);
-                        }
-                        return htmlLanguageService.format(document, range, {
-                            ...options,
-                            ...formatSettings,
-                            unformatted: '',
-                            contentUnformatted: blockTypes.join(','),
-                            endWithNewline: options.insertFinalNewline ? true
-                                : options.trimFinalNewlines ? false
-                                    : document.getText().endsWith('\n'),
-                        });
-                    });
-                },
             };
             function worker(document, callback) {
-                const [vueFile] = context.documents.getVirtualCodeByUri(document.uri);
-                if (vueFile instanceof vue.VueGeneratedCode) {
-                    return callback(vueFile);
+                const decoded = context.decodeEmbeddedDocumentUri(document.uri);
+                const sourceScript = decoded && context.language.scripts.get(decoded[0]);
+                const virtualCode = decoded && sourceScript?.generated?.embeddedCodes.get(decoded[1]);
+                if (virtualCode instanceof vue.VueGeneratedCode) {
+                    return callback(virtualCode);
                 }
             }
         },

package/lib/plugins/vue-template.d.ts

@@ -1,3 +1,3 @@
-import type { ServiceEnvironment } from '@volar/language-service';
-import { ServicePlugin, VueCompilerOptions } from '../types';
-export declare function create(mode: 'html' | 'pug', ts: typeof import('typescript'), getVueOptions: (env: ServiceEnvironment) => VueCompilerOptions, tsPluginClient?: typeof import('@vue/typescript-plugin/lib/client')): ServicePlugin;
+import type { ServiceContext, ServiceEnvironment } from '@volar/language-service';
+import { LanguageServicePlugin, VueCompilerOptions } from '../types';
+export declare function create(mode: 'html' | 'pug', ts: typeof import('typescript'), getVueOptions: (env: ServiceEnvironment) => VueCompilerOptions, getTsPluginClient?: (context: ServiceContext) => typeof import('@vue/typescript-plugin/lib/client') | undefined): LanguageServicePlugin;

package/lib/plugins/vue-template.js

@@ -10,9 +10,10 @@ const nameCasing_1 = require("../ideFeatures/nameCasing");
 const types_1 = require("../types");
 const data_1 = require("./data");
 const vscode_uri_1 = require("vscode-uri");
+const common_1 = require("@vue/typescript-plugin/lib/common");
 let builtInData;
 let modelData;
-function create(mode, ts, getVueOptions, tsPluginClient) {
+function create(mode, ts, getVueOptions, getTsPluginClient) {
     let customData = [];
     let extraCustomData = [];
     const onDidChangeCustomDataListeners = new Set();
@@ -41,6 +42,7 @@ function create(mode, ts, getVueOptions, tsPluginClient) {
             '@', // vue event shorthand
         ],
         create(context) {
+            const tsPluginClient = getTsPluginClient?.(context);
             const baseServiceInstance = baseService.create(context);
             const vueCompilerOptions = getVueOptions(context.env);
             builtInData ??= (0, data_1.loadTemplateData)(context.env.locale ?? 'en');
@@ -82,38 +84,46 @@ function create(mode, ts, getVueOptions, tsPluginClient) {
                     disposable?.dispose();
                 },
                 async provideCompletionItems(document, position, completionContext, token) {
-                    if (!isSupportedDocument(document))
+                    if (!isSupportedDocument(document)) {
                         return;
+                    }
                     let sync;
                     let currentVersion;
-                    const [_, sourceFile] = context.documents.getVirtualCodeByUri(document.uri);
-                    if (sourceFile?.generated?.code instanceof language_core_1.VueGeneratedCode) {
-                        sync = (await provideHtmlData(sourceFile.id, sourceFile.generated.code)).sync;
+                    const decoded = context.decodeEmbeddedDocumentUri(document.uri);
+                    const sourceScript = decoded && context.language.scripts.get(decoded[0]);
+                    if (sourceScript?.generated?.root instanceof language_core_1.VueGeneratedCode) {
+                        sync = (await provideHtmlData(sourceScript.id, sourceScript.generated.root)).sync;
                         currentVersion = await sync();
                     }
                     let htmlComplete = await baseServiceInstance.provideCompletionItems?.(document, position, completionContext, token);
                     while (currentVersion !== (currentVersion = await sync?.())) {
                         htmlComplete = await baseServiceInstance.provideCompletionItems?.(document, position, completionContext, token);
                     }
-                    if (!htmlComplete)
+                    if (!htmlComplete) {
                         return;
-                    if (sourceFile?.generated?.code instanceof language_core_1.VueGeneratedCode) {
-                        await afterHtmlCompletion(htmlComplete, context.documents.get(sourceFile.id, sourceFile.languageId, sourceFile.snapshot), sourceFile.generated.code);
+                    }
+                    if (sourceScript?.generated?.root instanceof language_core_1.VueGeneratedCode) {
+                        await afterHtmlCompletion(htmlComplete, context.documents.get(sourceScript.id, sourceScript.languageId, sourceScript.snapshot), sourceScript.generated.root);
                     }
                     return htmlComplete;
                 },
                 async provideInlayHints(document) {
-                    if (!isSupportedDocument(document))
+                    if (!isSupportedDocument(document)) {
                         return;
+                    }
                     const enabled = await context.env.getConfiguration?.('vue.inlayHints.missingProps') ?? false;
-                    if (!enabled)
+                    if (!enabled) {
                         return;
+                    }
                     const result = [];
-                    const [virtualCode] = context.documents.getVirtualCodeByUri(document.uri);
-                    if (!virtualCode)
+                    const decoded = context.decodeEmbeddedDocumentUri(document.uri);
+                    const sourceScript = decoded && context.language.scripts.get(decoded[0]);
+                    const virtualCode = decoded && sourceScript?.generated?.embeddedCodes.get(decoded[1]);
+                    if (!virtualCode) {
                         return;
+                    }
                     for (const map of context.documents.getMaps(virtualCode)) {
-                        const code = context.language.files.get(map.sourceDocument.uri)?.generated?.code;
+                        const code = context.language.scripts.get(map.sourceDocument.uri)?.generated?.root;
                         const scanner = getScanner(baseServiceInstance, document);
                         if (code instanceof language_core_1.VueGeneratedCode && scanner) {
                             // visualize missing required props
@@ -203,23 +213,30 @@ function create(mode, ts, getVueOptions, tsPluginClient) {
                     return result;
                 },
                 provideHover(document, position, token) {
-                    if (!isSupportedDocument(document))
+                    if (!isSupportedDocument(document)) {
                         return;
-                    if (context.documents.getVirtualCodeByUri(document.uri)[0])
+                    }
+                    if (context.decodeEmbeddedDocumentUri(document.uri)) {
                         updateExtraCustomData([]);
+                    }
                     return baseServiceInstance.provideHover?.(document, position, token);
                 },
                 async provideDiagnostics(document, token) {
-                    if (!isSupportedDocument(document))
+                    if (!isSupportedDocument(document)) {
                         return;
+                    }
                     const originalResult = await baseServiceInstance.provideDiagnostics?.(document, token);
-                    const [virtualCode] = context.documents.getVirtualCodeByUri(document.uri);
-                    if (!virtualCode)
+                    const decoded = context.decodeEmbeddedDocumentUri(document.uri);
+                    const sourceScript = decoded && context.language.scripts.get(decoded[0]);
+                    const virtualCode = decoded && sourceScript?.generated?.embeddedCodes.get(decoded[1]);
+                    if (!virtualCode) {
                         return;
+                    }
                     for (const map of context.documents.getMaps(virtualCode)) {
-                        const code = context.language.files.get(map.sourceDocument.uri)?.generated?.code;
-                        if (!(code instanceof language_core_1.VueGeneratedCode))
+                        const code = context.language.scripts.get(map.sourceDocument.uri)?.generated?.root;
+                        if (!(code instanceof language_core_1.VueGeneratedCode)) {
                             continue;
+                        }
                         const templateErrors = [];
                         const { template } = code.sfc;
                         if (template) {
@@ -253,18 +270,58 @@ function create(mode, ts, getVueOptions, tsPluginClient) {
                         ];
                     }
                 },
+                provideDocumentSemanticTokens(document, range, legend) {
+                    if (!isSupportedDocument(document)) {
+                        return;
+                    }
+                    const languageService = context.inject('typescript/languageService');
+                    if (!languageService) {
+                        return;
+                    }
+                    const decoded = context.decodeEmbeddedDocumentUri(document.uri);
+                    const sourceScript = decoded && context.language.scripts.get(decoded[0]);
+                    if (!sourceScript
+                        || !(sourceScript.generated?.root instanceof language_core_1.VueGeneratedCode)
+                        || !sourceScript.generated.root.sfc.template) {
+                        return [];
+                    }
+                    const { template } = sourceScript.generated.root.sfc;
+                    const spans = common_1.getComponentSpans.call({
+                        files: context.language.scripts,
+                        languageService,
+                        typescript: ts,
+                        vueOptions: getVueOptions(context.env),
+                    }, sourceScript.generated.root, template, {
+                        start: document.offsetAt(range.start),
+                        length: document.offsetAt(range.end) - document.offsetAt(range.start),
+                    });
+                    const classTokenIndex = legend.tokenTypes.indexOf('class');
+                    return spans.map(span => {
+                        const start = document.positionAt(span.start);
+                        return [
+                            start.line,
+                            start.character,
+                            span.length,
+                            classTokenIndex,
+                            0,
+                        ];
+                    });
+                },
             };
             async function provideHtmlData(sourceDocumentUri, vueCode) {
                 await (initializing ??= initialize());
                 const casing = await (0, nameCasing_1.getNameCasing)(context, sourceDocumentUri, tsPluginClient);
                 if (builtInData.tags) {
                     for (const tag of builtInData.tags) {
-                        if (tag.name === 'slot')
+                        if (tag.name === 'slot') {
                             continue;
-                        if (tag.name === 'component')
+                        }
+                        if (tag.name === 'component') {
                             continue;
-                        if (tag.name === 'template')
+                        }
+                        if (tag.name === 'template') {
                             continue;
+                        }
                         if (casing.tag === types_1.TagNameCasing.Kebab) {
                             tag.name = (0, language_core_1.hyphenateTag)(tag.name);
                         }
@@ -324,10 +381,8 @@ function create(mode, ts, getVueOptions, tsPluginClient) {
                             }
                             return tags;
                         },
-                        provideAttributes: (tag) => {
-                            tsPluginClient?.getTemplateContextProps;
-                            let failed = false;
-                            let tagInfo = tagInfos.get(tag);
+                        provideAttributes: tag => {
+                            const tagInfo = tagInfos.get(tag);
                             if (!tagInfo) {
                                 promises.push((async () => {
                                     const attrs = await tsPluginClient?.getElementAttrs(vueCode.fileName, tag) ?? [];
@@ -342,9 +397,6 @@ function create(mode, ts, getVueOptions, tsPluginClient) {
                                 })());
                                 return [];
                             }
-                            if (failed) {
-                                return [];
-                            }
                             const { attrs, props, events } = tagInfo;
                             const attributes = [];
                             const _tsCodegen = language_core_1.tsCodegen.get(vueCode.sfc);
@@ -467,8 +519,9 @@ function create(mode, ts, getVueOptions, tsPluginClient) {
                     const textWithoutModifier = replacement.text.split('.')[0];
                     if (validModifiers && hasModifier) {
                         for (const modifier in validModifiers) {
-                            if (modifiers.includes(modifier))
+                            if (modifiers.includes(modifier)) {
                                 continue;
+                            }
                             const modifierDes = validModifiers[modifier];
                             const insertText = textWithoutModifier + modifiers.slice(0, -1).map(m => '.' + m).join('') + '.' + modifier;
                             const newItem = {
@@ -489,8 +542,9 @@ function create(mode, ts, getVueOptions, tsPluginClient) {
                     }
                     else if (hasModifier && isModel) {
                         for (const modifier of modelData.globalAttributes ?? []) {
-                            if (modifiers.includes(modifier.name))
+                            if (modifiers.includes(modifier.name)) {
                                 continue;
+                            }
                             const insertText = textWithoutModifier + modifiers.slice(0, -1).map(m => '.' + m).join('') + '.' + modifier.name;
                             const newItem = {
                                 label: modifier.name,

package/lib/plugins/vue-toggle-v-bind-codeaction.d.ts

@@ -1,2 +1,2 @@
-import type { ServicePlugin } from '@volar/language-service';
-export declare function create(ts: typeof import('typescript')): ServicePlugin;
+import type { LanguageServicePlugin } from '@volar/language-service';
+export declare function create(ts: typeof import('typescript')): LanguageServicePlugin;

package/lib/plugins/vue-toggle-v-bind-codeaction.js

@@ -10,16 +10,19 @@ function create(ts) {
                 provideCodeActions(document, range, _context) {
                     const startOffset = document.offsetAt(range.start);
                     const endOffset = document.offsetAt(range.end);
-                    const [virtualCode] = context.documents.getVirtualCodeByUri(document.uri);
+                    const decoded = context.decodeEmbeddedDocumentUri(document.uri);
+                    const sourceScript = decoded && context.language.scripts.get(decoded[0]);
+                    const virtualCode = decoded && sourceScript?.generated?.embeddedCodes.get(decoded[1]);
                     if (!(virtualCode instanceof language_core_1.VueGeneratedCode)) {
                         return;
                     }
                     const { template } = virtualCode.sfc;
-                    if (!template?.ast)
+                    if (!template?.ast) {
                         return;
+                    }
                     const templateStartOffset = template.startTagEnd;
                     const result = [];
-                    for (const node of (0, language_core_1.eachElementNode)(template.ast)) {
+                    for (const node of (0, language_core_1.forEachElementNode)(template.ast)) {
                         if (startOffset > templateStartOffset + node.loc.end.offset || endOffset < templateStartOffset + node.loc.start.offset) {
                             return;
                         }

package/lib/plugins/vue-twoslash-queries.d.ts

@@ -1,2 +1,2 @@
-import type { ServicePlugin } from '@volar/language-service';
-export declare function create(ts: typeof import('typescript'), tsPluginClient?: typeof import('@vue/typescript-plugin/lib/client')): ServicePlugin;
+import type { ServiceContext, LanguageServicePlugin } from '@volar/language-service';
+export declare function create(ts: typeof import('typescript'), getTsPluginClient?: (context: ServiceContext) => typeof import('@vue/typescript-plugin/lib/client') | undefined): LanguageServicePlugin;