@volar/typescript 2.1.4 → 2.2.0-alpha.0

package/index.d.ts

@@ -1,4 +1,3 @@
-export * from './lib/documentRegistry';
 export * from './lib/node/decorateLanguageService';
 export * from './lib/node/decorateLanguageServiceHost';
 export * from './lib/node/decorateProgram';

package/index.js

@@ -14,7 +14,6 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-__exportStar(require("./lib/documentRegistry"), exports);
 __exportStar(require("./lib/node/decorateLanguageService"), exports);
 __exportStar(require("./lib/node/decorateLanguageServiceHost"), exports);
 __exportStar(require("./lib/node/decorateProgram"), exports);

package/lib/node/decorateLanguageService.d.ts

@@ -1,3 +1,3 @@
-import { FileRegistry } from '@volar/language-core';
+import { Language } from '@volar/language-core';
 import type * as ts from 'typescript';
-export declare function decorateLanguageService(files: FileRegistry, languageService: ts.LanguageService): void;
+export declare function decorateLanguageService(language: Language, languageService: ts.LanguageService): void;

package/lib/node/decorateLanguageService.js

@@ -5,12 +5,12 @@ const language_core_1 = require("@volar/language-core");
 const dedupe_1 = require("./dedupe");
 const utils_1 = require("./utils");
 const transform_1 = require("./transform");
-function decorateLanguageService(files, languageService) {
+function decorateLanguageService(language, languageService) {
     // ignored methods
     const { getNavigationTree, getOutliningSpans, } = languageService;
     languageService.getNavigationTree = fileName => {
-        const [virtualCode] = (0, utils_1.getVirtualFileAndMap)(files, fileName);
-        if (virtualCode) {
+        const [serviceScript] = (0, utils_1.getServiceScript)(language, fileName);
+        if (serviceScript) {
             const tree = getNavigationTree(fileName);
             tree.childItems = undefined;
             return tree;
@@ -20,8 +20,8 @@ function decorateLanguageService(files, languageService) {
         }
     };
     languageService.getOutliningSpans = fileName => {
-        const [virtualCode] = (0, utils_1.getVirtualFileAndMap)(files, fileName);
-        if (virtualCode) {
+        const [serviceScript] = (0, utils_1.getServiceScript)(language, fileName);
+        if (serviceScript) {
             return [];
         }
         else {
@@ -31,16 +31,16 @@ function decorateLanguageService(files, languageService) {
     // methods
     const { findReferences, findRenameLocations, getCompletionEntryDetails, getCompletionsAtPosition, getDefinitionAndBoundSpan, getDefinitionAtPosition, getFileReferences, getImplementationAtPosition, getQuickInfoAtPosition, getReferencesAtPosition, getSemanticDiagnostics, getSyntacticDiagnostics, getSuggestionDiagnostics, getTypeDefinitionAtPosition, getEncodedSemanticClassifications, getDocumentHighlights, getApplicableRefactors, getEditsForRefactor, getRenameInfo, getCodeFixesAtPosition, prepareCallHierarchy, provideCallHierarchyIncomingCalls, provideCallHierarchyOutgoingCalls, provideInlayHints, organizeImports, } = languageService;
     languageService.prepareCallHierarchy = (fileName, position) => {
-        const [virtualCode, sourceFile, map] = (0, utils_1.getVirtualFileAndMap)(files, fileName);
-        if (virtualCode) {
+        const [serviceScript, sourceScript, map] = (0, utils_1.getServiceScript)(language, fileName);
+        if (serviceScript) {
             for (const [generateOffset, mapping] of map.getGeneratedOffsets(position)) {
                 if ((0, language_core_1.isCallHierarchyEnabled)(mapping.data)) {
-                    const item = prepareCallHierarchy(fileName, generateOffset + sourceFile.snapshot.getLength());
+                    const item = prepareCallHierarchy(fileName, generateOffset + sourceScript.snapshot.getLength());
                     if (Array.isArray(item)) {
-                        return item.map(item => (0, transform_1.transformCallHierarchyItem)(files, item, language_core_1.isCallHierarchyEnabled));
+                        return item.map(item => (0, transform_1.transformCallHierarchyItem)(language, item, language_core_1.isCallHierarchyEnabled));
                     }
                     else if (item) {
-                        return (0, transform_1.transformCallHierarchyItem)(files, item, language_core_1.isCallHierarchyEnabled);
+                        return (0, transform_1.transformCallHierarchyItem)(language, item, language_core_1.isCallHierarchyEnabled);
                     }
                 }
             }
@@ -51,11 +51,11 @@ function decorateLanguageService(files, languageService) {
     };
     languageService.provideCallHierarchyIncomingCalls = (fileName, position) => {
         let calls = [];
-        const [virtualCode, sourceFile, map] = (0, utils_1.getVirtualFileAndMap)(files, fileName);
-        if (virtualCode) {
+        const [serviceScript, sourceScript, map] = (0, utils_1.getServiceScript)(language, fileName);
+        if (serviceScript) {
             for (const [generateOffset, mapping] of map.getGeneratedOffsets(position)) {
                 if ((0, language_core_1.isCallHierarchyEnabled)(mapping.data)) {
-                    calls = provideCallHierarchyIncomingCalls(fileName, generateOffset + sourceFile.snapshot.getLength());
+                    calls = provideCallHierarchyIncomingCalls(fileName, generateOffset + sourceScript.snapshot.getLength());
                 }
             }
         }
@@ -64,9 +64,9 @@ function decorateLanguageService(files, languageService) {
         }
         return calls
             .map(call => {
-            const from = (0, transform_1.transformCallHierarchyItem)(files, call.from, language_core_1.isCallHierarchyEnabled);
+            const from = (0, transform_1.transformCallHierarchyItem)(language, call.from, language_core_1.isCallHierarchyEnabled);
             const fromSpans = call.fromSpans
-                .map(span => (0, transform_1.transformSpan)(files, call.from.file, span, language_core_1.isCallHierarchyEnabled)?.textSpan)
+                .map(span => (0, transform_1.transformSpan)(language, call.from.file, span, language_core_1.isCallHierarchyEnabled)?.textSpan)
                 .filter(utils_1.notEmpty);
             return {
                 from,
@@ -76,11 +76,11 @@ function decorateLanguageService(files, languageService) {
     };
     languageService.provideCallHierarchyOutgoingCalls = (fileName, position) => {
         let calls = [];
-        const [virtualCode, sourceFile, map] = (0, utils_1.getVirtualFileAndMap)(files, fileName);
-        if (virtualCode) {
+        const [serviceScript, sourceScript, map] = (0, utils_1.getServiceScript)(language, fileName);
+        if (serviceScript) {
             for (const [generateOffset, mapping] of map.getGeneratedOffsets(position)) {
                 if ((0, language_core_1.isCallHierarchyEnabled)(mapping.data)) {
-                    calls = provideCallHierarchyOutgoingCalls(fileName, generateOffset + sourceFile.snapshot.getLength());
+                    calls = provideCallHierarchyOutgoingCalls(fileName, generateOffset + sourceScript.snapshot.getLength());
                 }
             }
         }
@@ -89,9 +89,9 @@ function decorateLanguageService(files, languageService) {
         }
         return calls
             .map(call => {
-            const to = (0, transform_1.transformCallHierarchyItem)(files, call.to, language_core_1.isCallHierarchyEnabled);
+            const to = (0, transform_1.transformCallHierarchyItem)(language, call.to, language_core_1.isCallHierarchyEnabled);
             const fromSpans = call.fromSpans
-                .map(span => (0, transform_1.transformSpan)(files, fileName, span, language_core_1.isCallHierarchyEnabled)?.textSpan)
+                .map(span => (0, transform_1.transformSpan)(language, fileName, span, language_core_1.isCallHierarchyEnabled)?.textSpan)
                 .filter(utils_1.notEmpty);
             return {
                 to,
@@ -102,18 +102,18 @@ function decorateLanguageService(files, languageService) {
     languageService.organizeImports = (args, formatOptions, preferences) => {
         const unresolved = organizeImports(args, formatOptions, preferences);
         const resolved = unresolved
-            .map(changes => (0, transform_1.transformFileTextChanges)(files, changes, language_core_1.isCodeActionsEnabled))
+            .map(changes => (0, transform_1.transformFileTextChanges)(language, changes, language_core_1.isCodeActionsEnabled))
             .filter(utils_1.notEmpty);
         return resolved;
     };
     languageService.getQuickInfoAtPosition = (fileName, position) => {
-        const [virtualCode, sourceFile, map] = (0, utils_1.getVirtualFileAndMap)(files, fileName);
-        if (virtualCode) {
+        const [serviceScript, sourceScript, map] = (0, utils_1.getServiceScript)(language, fileName);
+        if (serviceScript) {
             for (const [generateOffset, mapping] of map.getGeneratedOffsets(position)) {
                 if ((0, language_core_1.isHoverEnabled)(mapping.data)) {
-                    const result = getQuickInfoAtPosition(fileName, generateOffset + sourceFile.snapshot.getLength());
+                    const result = getQuickInfoAtPosition(fileName, generateOffset + sourceScript.snapshot.getLength());
                     if (result) {
-                        const textSpan = (0, transform_1.transformSpan)(files, fileName, result.textSpan, language_core_1.isHoverEnabled)?.textSpan;
+                        const textSpan = (0, transform_1.transformSpan)(language, fileName, result.textSpan, language_core_1.isHoverEnabled)?.textSpan;
                         if (textSpan) {
                             return {
                                 ...result,
@@ -143,11 +143,11 @@ function decorateLanguageService(files, languageService) {
                 ...highlights,
                 highlightSpans: highlights.highlightSpans
                     .map(span => {
-                    const textSpan = (0, transform_1.transformSpan)(files, span.fileName ?? highlights.fileName, span.textSpan, language_core_1.isHighlightEnabled)?.textSpan;
+                    const textSpan = (0, transform_1.transformSpan)(language, span.fileName ?? highlights.fileName, span.textSpan, language_core_1.isHighlightEnabled)?.textSpan;
                     if (textSpan) {
                         return {
                             ...span,
-                            contextSpan: (0, transform_1.transformSpan)(files, span.fileName ?? highlights.fileName, span.contextSpan, language_core_1.isHighlightEnabled)?.textSpan,
+                            contextSpan: (0, transform_1.transformSpan)(language, span.fileName ?? highlights.fileName, span.contextSpan, language_core_1.isHighlightEnabled)?.textSpan,
                             textSpan,
                         };
                     }
@@ -158,15 +158,15 @@ function decorateLanguageService(files, languageService) {
         return resolved;
     };
     languageService.getApplicableRefactors = (fileName, positionOrRange, preferences, triggerReason, kind, includeInteractiveActions) => {
-        const [virtualCode, sourceFile, map] = (0, utils_1.getVirtualFileAndMap)(files, fileName);
-        if (virtualCode) {
+        const [serviceScript, sourceScript, map] = (0, utils_1.getServiceScript)(language, fileName);
+        if (serviceScript) {
             for (const [generateOffset, mapping] of map.getGeneratedOffsets(typeof positionOrRange === 'number' ? positionOrRange : positionOrRange.pos)) {
                 if ((0, language_core_1.isCodeActionsEnabled)(mapping.data)) {
                     const por = typeof positionOrRange === 'number'
-                        ? generateOffset + sourceFile.snapshot.getLength()
+                        ? generateOffset + sourceScript.snapshot.getLength()
                         : {
-                            pos: generateOffset + sourceFile.snapshot.getLength(),
-                            end: generateOffset + positionOrRange.end - positionOrRange.pos + sourceFile.snapshot.getLength(),
+                            pos: generateOffset + sourceScript.snapshot.getLength(),
+                            end: generateOffset + positionOrRange.end - positionOrRange.pos + sourceScript.snapshot.getLength(),
                         };
                     return getApplicableRefactors(fileName, por, preferences, triggerReason, kind, includeInteractiveActions);
                 }
@@ -179,15 +179,15 @@ function decorateLanguageService(files, languageService) {
     };
     languageService.getEditsForRefactor = (fileName, formatOptions, positionOrRange, refactorName, actionName, preferences) => {
         let edits;
-        const [virtualCode, sourceFile, map] = (0, utils_1.getVirtualFileAndMap)(files, fileName);
-        if (virtualCode) {
+        const [serviceScript, sourceScript, map] = (0, utils_1.getServiceScript)(language, fileName);
+        if (serviceScript) {
             for (const [generateOffset, mapping] of map.getGeneratedOffsets(typeof positionOrRange === 'number' ? positionOrRange : positionOrRange.pos)) {
                 if ((0, language_core_1.isCodeActionsEnabled)(mapping.data)) {
                     const por = typeof positionOrRange === 'number'
-                        ? generateOffset + sourceFile.snapshot.getLength()
+                        ? generateOffset + sourceScript.snapshot.getLength()
                         : {
-                            pos: generateOffset + sourceFile.snapshot.getLength(),
-                            end: generateOffset + positionOrRange.end - positionOrRange.pos + sourceFile.snapshot.getLength(),
+                            pos: generateOffset + sourceScript.snapshot.getLength(),
+                            end: generateOffset + positionOrRange.end - positionOrRange.pos + sourceScript.snapshot.getLength(),
                         };
                     edits = getEditsForRefactor(fileName, formatOptions, por, refactorName, actionName, preferences);
                 }
@@ -198,19 +198,19 @@ function decorateLanguageService(files, languageService) {
         }
         if (edits) {
             edits.edits = edits.edits
-                .map(edit => (0, transform_1.transformFileTextChanges)(files, edit, language_core_1.isCodeActionsEnabled))
+                .map(edit => (0, transform_1.transformFileTextChanges)(language, edit, language_core_1.isCodeActionsEnabled))
                 .filter(utils_1.notEmpty);
             return edits;
         }
     };
     languageService.getRenameInfo = (fileName, position, options) => {
-        const [virtualCode, sourceFile, map] = (0, utils_1.getVirtualFileAndMap)(files, fileName);
-        if (virtualCode) {
+        const [serviceScript, sourceScript, map] = (0, utils_1.getServiceScript)(language, fileName);
+        if (serviceScript) {
             for (const [generateOffset, mapping] of map.getGeneratedOffsets(position)) {
                 if ((0, language_core_1.isRenameEnabled)(mapping.data)) {
-                    const info = getRenameInfo(fileName, generateOffset + sourceFile.snapshot.getLength(), options);
+                    const info = getRenameInfo(fileName, generateOffset + sourceScript.snapshot.getLength(), options);
                     if (info.canRename) {
-                        const span = (0, transform_1.transformSpan)(files, fileName, info.triggerSpan, language_core_1.isRenameEnabled);
+                        const span = (0, transform_1.transformSpan)(language, fileName, info.triggerSpan, language_core_1.isRenameEnabled);
                         if (span) {
                             info.triggerSpan = span.textSpan;
                             return info;
@@ -232,13 +232,13 @@ function decorateLanguageService(files, languageService) {
     };
     languageService.getCodeFixesAtPosition = (fileName, start, end, errorCodes, formatOptions, preferences) => {
         let fixes = [];
-        const [virtualCode, sourceFile, map] = (0, utils_1.getVirtualFileAndMap)(files, fileName);
-        if (virtualCode) {
+        const [serviceScript, sourceScript, map] = (0, utils_1.getServiceScript)(language, fileName);
+        if (serviceScript) {
             for (const [generateStart, mapping] of map.getGeneratedOffsets(start)) {
                 if ((0, language_core_1.isCodeActionsEnabled)(mapping.data)) {
                     for (const [generateEnd, mapping] of map.getGeneratedOffsets(end)) {
                         if ((0, language_core_1.isCodeActionsEnabled)(mapping.data)) {
-                            fixes = getCodeFixesAtPosition(fileName, generateStart + sourceFile.snapshot.getLength(), generateEnd + sourceFile.snapshot.getLength(), errorCodes, formatOptions, preferences);
+                            fixes = getCodeFixesAtPosition(fileName, generateStart + sourceScript.snapshot.getLength(), generateEnd + sourceScript.snapshot.getLength(), errorCodes, formatOptions, preferences);
                             break;
                         }
                     }
@@ -250,14 +250,14 @@ function decorateLanguageService(files, languageService) {
             fixes = getCodeFixesAtPosition(fileName, start, end, errorCodes, formatOptions, preferences);
         }
         fixes = fixes.map(fix => {
-            fix.changes = fix.changes.map(edit => (0, transform_1.transformFileTextChanges)(files, edit, language_core_1.isCodeActionsEnabled)).filter(utils_1.notEmpty);
+            fix.changes = fix.changes.map(edit => (0, transform_1.transformFileTextChanges)(language, edit, language_core_1.isCodeActionsEnabled)).filter(utils_1.notEmpty);
             return fix;
         });
         return fixes;
     };
     languageService.getEncodedSemanticClassifications = (fileName, span, format) => {
-        const [virtualCode, sourceFile, map] = (0, utils_1.getVirtualFileAndMap)(files, fileName);
-        if (virtualCode) {
+        const [serviceScript, sourceScript, map] = (0, utils_1.getServiceScript)(language, fileName);
+        if (serviceScript) {
             let start;
             let end;
             for (const mapping of map.mappings) {
@@ -272,14 +272,14 @@ function decorateLanguageService(files, languageService) {
                 start = 0;
                 end = 0;
             }
-            start += sourceFile.snapshot.getLength();
-            end += sourceFile.snapshot.getLength();
+            start += sourceScript.snapshot.getLength();
+            end += sourceScript.snapshot.getLength();
             const result = getEncodedSemanticClassifications(fileName, { start, length: end - start }, format);
             const spans = [];
             for (let i = 0; i < result.spans.length; i += 3) {
-                for (const [sourceStart, mapping] of map.getSourceOffsets(result.spans[i] - sourceFile.snapshot.getLength())) {
+                for (const [sourceStart, mapping] of map.getSourceOffsets(result.spans[i] - sourceScript.snapshot.getLength())) {
                     if ((0, language_core_1.isSemanticTokensEnabled)(mapping.data)) {
-                        for (const [sourceEnd, mapping] of map.getSourceOffsets(result.spans[i] + result.spans[i + 1] - sourceFile.snapshot.getLength())) {
+                        for (const [sourceEnd, mapping] of map.getSourceOffsets(result.spans[i] + result.spans[i + 1] - sourceScript.snapshot.getLength())) {
                             if ((0, language_core_1.isSemanticTokensEnabled)(mapping.data)) {
                                 spans.push(sourceStart, sourceEnd - sourceStart, result.spans[i + 2]);
                                 break;
@@ -298,17 +298,17 @@ function decorateLanguageService(files, languageService) {
     };
     languageService.getSyntacticDiagnostics = fileName => {
         return getSyntacticDiagnostics(fileName)
-            .map(d => (0, transform_1.transformDiagnostic)(files, d))
+            .map(d => (0, transform_1.transformDiagnostic)(language, d))
             .filter(utils_1.notEmpty);
     };
     languageService.getSemanticDiagnostics = fileName => {
         return getSemanticDiagnostics(fileName)
-            .map(d => (0, transform_1.transformDiagnostic)(files, d))
+            .map(d => (0, transform_1.transformDiagnostic)(language, d))
             .filter(utils_1.notEmpty);
     };
     languageService.getSuggestionDiagnostics = fileName => {
         return getSuggestionDiagnostics(fileName)
-            .map(d => (0, transform_1.transformDiagnostic)(files, d))
+            .map(d => (0, transform_1.transformDiagnostic)(language, d))
             .filter(utils_1.notEmpty);
     };
     languageService.getDefinitionAndBoundSpan = (fileName, position) => {
@@ -318,14 +318,14 @@ function decorateLanguageService(files, languageService) {
             }
         });
         const textSpan = unresolved
-            .map(s => (0, transform_1.transformSpan)(files, fileName, s.textSpan, language_core_1.isDefinitionEnabled)?.textSpan)
+            .map(s => (0, transform_1.transformSpan)(language, fileName, s.textSpan, language_core_1.isDefinitionEnabled)?.textSpan)
             .filter(utils_1.notEmpty)[0];
         if (!textSpan) {
             return;
         }
         const definitions = unresolved
             .map(s => s.definitions
-            ?.map(s => (0, transform_1.transformDocumentSpan)(files, s, language_core_1.isDefinitionEnabled, s.fileName !== fileName))
+            ?.map(s => (0, transform_1.transformDocumentSpan)(language, s, language_core_1.isDefinitionEnabled, s.fileName !== fileName))
             .filter(utils_1.notEmpty))
             .filter(utils_1.notEmpty)
             .flat();
@@ -345,12 +345,12 @@ function decorateLanguageService(files, languageService) {
         const resolved = unresolved
             .flat()
             .map(symbol => {
-            const definition = (0, transform_1.transformDocumentSpan)(files, symbol.definition, language_core_1.isDefinitionEnabled);
+            const definition = (0, transform_1.transformDocumentSpan)(language, symbol.definition, language_core_1.isDefinitionEnabled);
             if (definition) {
                 return {
                     definition,
                     references: symbol.references
-                        .map(r => (0, transform_1.transformDocumentSpan)(files, r, language_core_1.isReferencesEnabled))
+                        .map(r => (0, transform_1.transformDocumentSpan)(language, r, language_core_1.isReferencesEnabled))
                         .filter(utils_1.notEmpty),
                 };
             }
@@ -366,7 +366,7 @@ function decorateLanguageService(files, languageService) {
         });
         const resolved = unresolved
             .flat()
-            .map(s => (0, transform_1.transformDocumentSpan)(files, s, language_core_1.isDefinitionEnabled, s.fileName !== fileName))
+            .map(s => (0, transform_1.transformDocumentSpan)(language, s, language_core_1.isDefinitionEnabled, s.fileName !== fileName))
             .filter(utils_1.notEmpty);
         return (0, dedupe_1.dedupeDocumentSpans)(resolved);
     };
@@ -378,7 +378,7 @@ function decorateLanguageService(files, languageService) {
         });
         const resolved = unresolved
             .flat()
-            .map(s => (0, transform_1.transformDocumentSpan)(files, s, language_core_1.isTypeDefinitionEnabled))
+            .map(s => (0, transform_1.transformDocumentSpan)(language, s, language_core_1.isTypeDefinitionEnabled))
             .filter(utils_1.notEmpty);
         return (0, dedupe_1.dedupeDocumentSpans)(resolved);
     };
@@ -390,7 +390,7 @@ function decorateLanguageService(files, languageService) {
         });
         const resolved = unresolved
             .flat()
-            .map(s => (0, transform_1.transformDocumentSpan)(files, s, language_core_1.isImplementationEnabled))
+            .map(s => (0, transform_1.transformDocumentSpan)(language, s, language_core_1.isImplementationEnabled))
             .filter(utils_1.notEmpty);
         return (0, dedupe_1.dedupeDocumentSpans)(resolved);
     };
@@ -402,7 +402,7 @@ function decorateLanguageService(files, languageService) {
         });
         const resolved = unresolved
             .flat()
-            .map(s => (0, transform_1.transformDocumentSpan)(files, s, language_core_1.isRenameEnabled))
+            .map(s => (0, transform_1.transformDocumentSpan)(language, s, language_core_1.isRenameEnabled))
             .filter(utils_1.notEmpty);
         return (0, dedupe_1.dedupeDocumentSpans)(resolved);
     };
@@ -414,13 +414,13 @@ function decorateLanguageService(files, languageService) {
         });
         const resolved = unresolved
             .flat()
-            .map(s => (0, transform_1.transformDocumentSpan)(files, s, language_core_1.isReferencesEnabled))
+            .map(s => (0, transform_1.transformDocumentSpan)(language, s, language_core_1.isReferencesEnabled))
             .filter(utils_1.notEmpty);
         return (0, dedupe_1.dedupeDocumentSpans)(resolved);
     };
     languageService.getCompletionsAtPosition = (fileName, position, options, formattingSettings) => {
-        const [virtualCode, sourceFile, map] = (0, utils_1.getVirtualFileAndMap)(files, fileName);
-        if (virtualCode) {
+        const [serviceScript, sourceScript, map] = (0, utils_1.getServiceScript)(language, fileName);
+        if (serviceScript) {
             let mainResult;
             let additionalResults = [];
             for (const [generateOffset, mapping] of map.getGeneratedOffsets(position)) {
@@ -429,12 +429,12 @@ function decorateLanguageService(files, languageService) {
                     if (!isAdditional && mainResult) {
                         continue;
                     }
-                    const result = getCompletionsAtPosition(fileName, generateOffset + sourceFile.snapshot.getLength(), options, formattingSettings);
+                    const result = getCompletionsAtPosition(fileName, generateOffset + sourceScript.snapshot.getLength(), options, formattingSettings);
                     if (result) {
                         for (const entry of result.entries) {
-                            entry.replacementSpan = (0, transform_1.transformSpan)(files, fileName, entry.replacementSpan, language_core_1.isCompletionEnabled)?.textSpan;
+                            entry.replacementSpan = (0, transform_1.transformSpan)(language, fileName, entry.replacementSpan, language_core_1.isCompletionEnabled)?.textSpan;
                         }
-                        result.optionalReplacementSpan = (0, transform_1.transformSpan)(files, fileName, result.optionalReplacementSpan, language_core_1.isCompletionEnabled)?.textSpan;
+                        result.optionalReplacementSpan = (0, transform_1.transformSpan)(language, fileName, result.optionalReplacementSpan, language_core_1.isCompletionEnabled)?.textSpan;
                         if (isAdditional) {
                             additionalResults.push(result);
                         }
@@ -463,11 +463,11 @@ function decorateLanguageService(files, languageService) {
     };
     languageService.getCompletionEntryDetails = (fileName, position, entryName, formatOptions, source, preferences, data) => {
         let details;
-        const [virtualCode, sourceFile, map] = (0, utils_1.getVirtualFileAndMap)(files, fileName);
-        if (virtualCode) {
+        const [serviceScript, sourceScript, map] = (0, utils_1.getServiceScript)(language, fileName);
+        if (serviceScript) {
             for (const [generateOffset, mapping] of map.getGeneratedOffsets(position)) {
                 if ((0, language_core_1.isCompletionEnabled)(mapping.data)) {
-                    details = getCompletionEntryDetails(fileName, generateOffset + sourceFile.snapshot.getLength(), entryName, formatOptions, source, preferences, data);
+                    details = getCompletionEntryDetails(fileName, generateOffset + sourceScript.snapshot.getLength(), entryName, formatOptions, source, preferences, data);
                     break;
                 }
             }
@@ -477,14 +477,14 @@ function decorateLanguageService(files, languageService) {
         }
         if (details?.codeActions) {
             for (const codeAction of details.codeActions) {
-                codeAction.changes = codeAction.changes.map(edit => (0, transform_1.transformFileTextChanges)(files, edit, language_core_1.isCompletionEnabled)).filter(utils_1.notEmpty);
+                codeAction.changes = codeAction.changes.map(edit => (0, transform_1.transformFileTextChanges)(language, edit, language_core_1.isCompletionEnabled)).filter(utils_1.notEmpty);
             }
         }
         return details;
     };
     languageService.provideInlayHints = (fileName, span, preferences) => {
-        const [virtualCode, sourceFile, map] = (0, utils_1.getVirtualFileAndMap)(files, fileName);
-        if (virtualCode) {
+        const [serviceScript, sourceScript, map] = (0, utils_1.getServiceScript)(language, fileName);
+        if (serviceScript) {
             let start;
             let end;
             for (const mapping of map.mappings) {
@@ -499,12 +499,12 @@ function decorateLanguageService(files, languageService) {
                 start = 0;
                 end = 0;
             }
-            start += sourceFile.snapshot.getLength();
-            end += sourceFile.snapshot.getLength();
+            start += sourceScript.snapshot.getLength();
+            end += sourceScript.snapshot.getLength();
             const result = provideInlayHints(fileName, { start, length: end - start }, preferences);
             const hints = [];
             for (const hint of result) {
-                for (const [sourcePosition, mapping] of map.getSourceOffsets(hint.position - sourceFile.snapshot.getLength())) {
+                for (const [sourcePosition, mapping] of map.getSourceOffsets(hint.position - sourceScript.snapshot.getLength())) {
                     if ((0, language_core_1.isInlayHintsEnabled)(mapping.data)) {
                         hints.push({
                             ...hint,
@@ -523,18 +523,18 @@ function decorateLanguageService(files, languageService) {
     languageService.getFileReferences = fileName => {
         const unresolved = getFileReferences(fileName);
         const resolved = unresolved
-            .map(s => (0, transform_1.transformDocumentSpan)(files, s, language_core_1.isReferencesEnabled))
+            .map(s => (0, transform_1.transformDocumentSpan)(language, s, language_core_1.isReferencesEnabled))
             .filter(utils_1.notEmpty);
         return (0, dedupe_1.dedupeDocumentSpans)(resolved);
     };
     function linkedCodeFeatureWorker(fileName, position, filter, worker, getLinkedCodes) {
         let results = [];
         const processedFilePositions = new Set();
-        const [virtualCode, sourceFile, map] = (0, utils_1.getVirtualFileAndMap)(files, fileName);
-        if (virtualCode) {
+        const [serviceScript, sourceScript, map] = (0, utils_1.getServiceScript)(language, fileName);
+        if (serviceScript) {
             for (const [generateOffset, mapping] of map.getGeneratedOffsets(position)) {
                 if (filter(mapping.data)) {
-                    process(fileName, generateOffset + sourceFile.snapshot.getLength());
+                    process(fileName, generateOffset + sourceScript.snapshot.getLength());
                 }
             }
         }
@@ -554,16 +554,16 @@ function decorateLanguageService(files, languageService) {
             results = results.concat(result);
             for (const ref of getLinkedCodes(result)) {
                 processedFilePositions.add(ref[0] + ':' + ref[1]);
-                const [virtualFile, sourceFile] = (0, utils_1.getVirtualFileAndMap)(files, ref[0]);
-                if (!virtualFile) {
+                const [serviceScript, sourceScript] = (0, utils_1.getServiceScript)(language, ref[0]);
+                if (!serviceScript) {
                     continue;
                 }
-                const linkedCodeMap = files.getLinkedCodeMap(virtualFile.code);
+                const linkedCodeMap = language.linkedCodeMaps.get(serviceScript.code);
                 if (!linkedCodeMap) {
                     continue;
                 }
-                for (const linkedCodeOffset of linkedCodeMap.getLinkedOffsets(ref[1] - sourceFile.snapshot.getLength())) {
-                    process(ref[0], linkedCodeOffset + sourceFile.snapshot.getLength());
+                for (const linkedCodeOffset of linkedCodeMap.getLinkedOffsets(ref[1] - sourceScript.snapshot.getLength())) {
+                    process(ref[0], linkedCodeOffset + sourceScript.snapshot.getLength());
                 }
             }
         }

package/lib/node/decorateLanguageServiceHost.d.ts

@@ -1,4 +1,4 @@
-import { type FileRegistry } from '@volar/language-core';
+import { type Language } from '@volar/language-core';
 import type * as ts from 'typescript';
-export declare function decorateLanguageServiceHost(virtualFiles: FileRegistry, languageServiceHost: ts.LanguageServiceHost, ts: typeof import('typescript')): void;
+export declare function decorateLanguageServiceHost(language: Language, languageServiceHost: ts.LanguageServiceHost, ts: typeof import('typescript')): void;
 export declare function searchExternalFiles(ts: typeof import('typescript'), project: ts.server.Project, exts: string[]): string[];

package/lib/node/decorateLanguageServiceHost.js

@@ -3,10 +3,9 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.searchExternalFiles = exports.decorateLanguageServiceHost = void 0;
 const language_core_1 = require("@volar/language-core");
 const resolveModuleName_1 = require("../resolveModuleName");
-function decorateLanguageServiceHost(virtualFiles, languageServiceHost, ts) {
+function decorateLanguageServiceHost(language, languageServiceHost, ts) {
     let extraProjectVersion = 0;
-    const { languagePlugins } = virtualFiles;
-    const exts = languagePlugins
+    const exts = language.plugins
         .map(plugin => plugin.typescript?.extraFileExtensions.map(ext => '.' + ext.extension) ?? [])
         .flat();
     const scripts = new Map();
@@ -29,8 +28,8 @@ function decorateLanguageServiceHost(virtualFiles, languageServiceHost, ts) {
             return readDirectory(path, extensions, exclude, include, depth);
         };
     }
-    if (languagePlugins.some(language => language.typescript?.extraFileExtensions.length)) {
-        const resolveModuleName = (0, resolveModuleName_1.createResolveModuleName)(ts, languageServiceHost, languagePlugins, fileName => virtualFiles.get(fileName));
+    if (language.plugins.some(language => language.typescript?.extraFileExtensions.length)) {
+        const resolveModuleName = (0, resolveModuleName_1.createResolveModuleName)(ts, languageServiceHost, language.plugins, fileName => language.scripts.get(fileName));
         if (resolveModuleNameLiterals) {
             languageServiceHost.resolveModuleNameLiterals = (moduleLiterals, containingFile, redirectedReference, options, ...rest) => {
                 if (moduleLiterals.every(name => !exts.some(ext => name.text.endsWith(ext)))) {
@@ -86,25 +85,25 @@ function decorateLanguageServiceHost(virtualFiles, languageServiceHost, ts) {
             const snapshot = getScriptSnapshot(fileName);
             if (snapshot) {
                 extraProjectVersion++;
-                const sourceFile = virtualFiles.set(fileName, (0, language_core_1.resolveCommonLanguageId)(fileName), snapshot);
-                if (sourceFile.generated) {
+                const sourceScript = language.scripts.set(fileName, (0, language_core_1.resolveCommonLanguageId)(fileName), snapshot);
+                if (sourceScript.generated) {
                     const text = snapshot.getText(0, snapshot.getLength());
                     let patchedText = text.split('\n').map(line => ' '.repeat(line.length)).join('\n');
-                    const script = sourceFile.generated.languagePlugin.typescript?.getScript(sourceFile.generated.code);
-                    if (script) {
-                        extension = script.extension;
-                        scriptKind = script.scriptKind;
-                        patchedText += script.code.snapshot.getText(0, script.code.snapshot.getLength());
+                    const serviceScript = sourceScript.generated.languagePlugin.typescript?.getServiceScript(sourceScript.generated.root);
+                    if (serviceScript) {
+                        extension = serviceScript.extension;
+                        scriptKind = serviceScript.scriptKind;
+                        patchedText += serviceScript.code.snapshot.getText(0, serviceScript.code.snapshot.getLength());
                     }
                     snapshotSnapshot = ts.ScriptSnapshot.fromString(patchedText);
-                    if (sourceFile.generated.languagePlugin.typescript?.getExtraScripts) {
+                    if (sourceScript.generated.languagePlugin.typescript?.getExtraServiceScripts) {
                         console.warn('getExtraScripts() is not available in this use case.');
                     }
                 }
             }
-            else if (virtualFiles.get(fileName)) {
+            else if (language.scripts.get(fileName)) {
                 extraProjectVersion++;
-                virtualFiles.delete(fileName);
+                language.scripts.delete(fileName);
             }
             scripts.set(fileName, {
                 version,

package/lib/node/decorateProgram.d.ts

@@ -1,3 +1,3 @@
-import type { FileRegistry } from '@volar/language-core';
+import type { Language } from '@volar/language-core';
 import type * as ts from 'typescript';
-export declare function decorateProgram(files: FileRegistry, program: ts.Program): void;
+export declare function decorateProgram(language: Language, program: ts.Program): void;

package/lib/node/decorateProgram.js

@@ -3,7 +3,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.decorateProgram = void 0;
 const utils_1 = require("./utils");
 const transform_1 = require("./transform");
-function decorateProgram(files, program) {
+function decorateProgram(language, program) {
     const emit = program.emit;
     // for tsc --noEmit
     const getSyntacticDiagnostics = program.getSyntacticDiagnostics;
@@ -17,29 +17,29 @@ function decorateProgram(files, program) {
         return {
             ...result,
             diagnostics: result.diagnostics
-                .map(d => (0, transform_1.transformDiagnostic)(files, d))
+                .map(d => (0, transform_1.transformDiagnostic)(language, d))
                 .filter(utils_1.notEmpty),
         };
     };
     program.getSyntacticDiagnostics = (sourceFile, cancellationToken) => {
         return getSyntacticDiagnostics(sourceFile, cancellationToken)
-            .map(d => (0, transform_1.transformDiagnostic)(files, d))
+            .map(d => (0, transform_1.transformDiagnostic)(language, d))
             .filter(utils_1.notEmpty);
     };
     program.getSemanticDiagnostics = (sourceFile, cancellationToken) => {
         return getSemanticDiagnostics(sourceFile, cancellationToken)
-            .map(d => (0, transform_1.transformDiagnostic)(files, d))
+            .map(d => (0, transform_1.transformDiagnostic)(language, d))
             .filter(utils_1.notEmpty);
     };
     program.getGlobalDiagnostics = cancellationToken => {
         return getGlobalDiagnostics(cancellationToken)
-            .map(d => (0, transform_1.transformDiagnostic)(files, d))
+            .map(d => (0, transform_1.transformDiagnostic)(language, d))
             .filter(utils_1.notEmpty);
     };
     // @ts-ignore
     program.getBindAndCheckDiagnostics = (sourceFile, cancellationToken) => {
         return getBindAndCheckDiagnostics(sourceFile, cancellationToken)
-            .map(d => (0, transform_1.transformDiagnostic)(files, d))
+            .map(d => (0, transform_1.transformDiagnostic)(language, d))
             .filter(utils_1.notEmpty);
     };
 }