@volar/typescript 2.0.0-alpha.0 → 2.0.0-alpha.10

package/lib/node/decorateLanguageService.js

@@ -3,20 +3,44 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.decorateLanguageService = void 0;
 const language_core_1 = require("@volar/language-core");
 const dedupe_1 = require("./dedupe");
-function decorateLanguageService(virtualFiles, languageService, isTsPlugin) {
-    const transformedDiagnostics = new WeakMap();
+const utils_1 = require("./utils");
+const transform_1 = require("./transform");
+function decorateLanguageService(files, languageService) {
+    // ignored methods
+    const { getNavigationTree, getOutliningSpans, } = languageService;
+    languageService.getNavigationTree = (fileName) => {
+        const [virtualFile] = (0, utils_1.getVirtualFileAndMap)(files, fileName);
+        if (virtualFile) {
+            const tree = getNavigationTree(fileName);
+            tree.childItems = undefined;
+            return tree;
+        }
+        else {
+            return getNavigationTree(fileName);
+        }
+    };
+    languageService.getOutliningSpans = (fileName) => {
+        const [virtualFile] = (0, utils_1.getVirtualFileAndMap)(files, fileName);
+        if (virtualFile) {
+            return [];
+        }
+        else {
+            return getOutliningSpans(fileName);
+        }
+    };
+    // 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 [virtualFile, sourceFile, map] = getVirtualFileAndMap(fileName);
+        const [virtualFile, sourceFile, map] = (0, utils_1.getVirtualFileAndMap)(files, fileName);
         if (virtualFile) {
             for (const [generateOffset, mapping] of map.getGeneratedOffsets(position)) {
                 if ((0, language_core_1.isCallHierarchyEnabled)(mapping.data)) {
-                    const item = prepareCallHierarchy(fileName, generateOffset + (isTsPlugin ? sourceFile.snapshot.getLength() : 0));
+                    const item = prepareCallHierarchy(fileName, generateOffset + sourceFile.snapshot.getLength());
                     if (Array.isArray(item)) {
-                        return item.map(item => transformCallHierarchyItem(item, language_core_1.isCallHierarchyEnabled));
+                        return item.map(item => (0, transform_1.transformCallHierarchyItem)(files, item, language_core_1.isCallHierarchyEnabled));
                     }
                     else if (item) {
-                        return transformCallHierarchyItem(item, language_core_1.isCallHierarchyEnabled);
+                        return (0, transform_1.transformCallHierarchyItem)(files, item, language_core_1.isCallHierarchyEnabled);
                     }
                 }
             }
@@ -27,11 +51,11 @@ function decorateLanguageService(virtualFiles, languageService, isTsPlugin) {
     };
     languageService.provideCallHierarchyIncomingCalls = (fileName, position) => {
         let calls = [];
-        const [virtualFile, sourceFile, map] = getVirtualFileAndMap(fileName);
+        const [virtualFile, sourceFile, map] = (0, utils_1.getVirtualFileAndMap)(files, fileName);
         if (virtualFile) {
             for (const [generateOffset, mapping] of map.getGeneratedOffsets(position)) {
                 if ((0, language_core_1.isCallHierarchyEnabled)(mapping.data)) {
-                    calls = provideCallHierarchyIncomingCalls(fileName, generateOffset + (isTsPlugin ? sourceFile.snapshot.getLength() : 0));
+                    calls = provideCallHierarchyIncomingCalls(fileName, generateOffset + sourceFile.snapshot.getLength());
                 }
             }
         }
@@ -40,10 +64,10 @@ function decorateLanguageService(virtualFiles, languageService, isTsPlugin) {
         }
         return calls
             .map(call => {
-            const from = transformCallHierarchyItem(call.from, language_core_1.isCallHierarchyEnabled);
+            const from = (0, transform_1.transformCallHierarchyItem)(files, call.from, language_core_1.isCallHierarchyEnabled);
             const fromSpans = call.fromSpans
-                .map(span => transformSpan(call.from.file, span, language_core_1.isCallHierarchyEnabled)?.textSpan)
-                .filter(notEmpty);
+                .map(span => (0, transform_1.transformSpan)(files, call.from.file, span, language_core_1.isCallHierarchyEnabled)?.textSpan)
+                .filter(utils_1.notEmpty);
             return {
                 from,
                 fromSpans,
@@ -52,11 +76,11 @@ function decorateLanguageService(virtualFiles, languageService, isTsPlugin) {
     };
     languageService.provideCallHierarchyOutgoingCalls = (fileName, position) => {
         let calls = [];
-        const [virtualFile, sourceFile, map] = getVirtualFileAndMap(fileName);
+        const [virtualFile, sourceFile, map] = (0, utils_1.getVirtualFileAndMap)(files, fileName);
         if (virtualFile) {
             for (const [generateOffset, mapping] of map.getGeneratedOffsets(position)) {
                 if ((0, language_core_1.isCallHierarchyEnabled)(mapping.data)) {
-                    calls = provideCallHierarchyOutgoingCalls(fileName, generateOffset + (isTsPlugin ? sourceFile.snapshot.getLength() : 0));
+                    calls = provideCallHierarchyOutgoingCalls(fileName, generateOffset + sourceFile.snapshot.getLength());
                 }
             }
         }
@@ -65,10 +89,10 @@ function decorateLanguageService(virtualFiles, languageService, isTsPlugin) {
         }
         return calls
             .map(call => {
-            const to = transformCallHierarchyItem(call.to, language_core_1.isCallHierarchyEnabled);
+            const to = (0, transform_1.transformCallHierarchyItem)(files, call.to, language_core_1.isCallHierarchyEnabled);
             const fromSpans = call.fromSpans
-                .map(span => transformSpan(fileName, span, language_core_1.isCallHierarchyEnabled)?.textSpan)
-                .filter(notEmpty);
+                .map(span => (0, transform_1.transformSpan)(files, fileName, span, language_core_1.isCallHierarchyEnabled)?.textSpan)
+                .filter(utils_1.notEmpty);
             return {
                 to,
                 fromSpans,
@@ -78,18 +102,18 @@ function decorateLanguageService(virtualFiles, languageService, isTsPlugin) {
     languageService.organizeImports = (args, formatOptions, preferences) => {
         const unresolved = organizeImports(args, formatOptions, preferences);
         const resolved = unresolved
-            .map(changes => transformFileTextChanges(changes, language_core_1.isCodeActionsEnabled))
-            .filter(notEmpty);
+            .map(changes => (0, transform_1.transformFileTextChanges)(files, changes, language_core_1.isCodeActionsEnabled))
+            .filter(utils_1.notEmpty);
         return resolved;
     };
     languageService.getQuickInfoAtPosition = (fileName, position) => {
-        const [virtualFile, sourceFile, map] = getVirtualFileAndMap(fileName);
+        const [virtualFile, sourceFile, map] = (0, utils_1.getVirtualFileAndMap)(files, fileName);
         if (virtualFile) {
             for (const [generateOffset, mapping] of map.getGeneratedOffsets(position)) {
                 if ((0, language_core_1.isHoverEnabled)(mapping.data)) {
-                    const result = getQuickInfoAtPosition(fileName, generateOffset + (isTsPlugin ? sourceFile.snapshot.getLength() : 0));
+                    const result = getQuickInfoAtPosition(fileName, generateOffset + sourceFile.snapshot.getLength());
                     if (result) {
-                        const textSpan = transformSpan(fileName, result.textSpan, language_core_1.isHoverEnabled)?.textSpan;
+                        const textSpan = (0, transform_1.transformSpan)(files, fileName, result.textSpan, language_core_1.isHoverEnabled)?.textSpan;
                         if (textSpan) {
                             return {
                                 ...result,
@@ -119,30 +143,30 @@ function decorateLanguageService(virtualFiles, languageService, isTsPlugin) {
                 ...highlights,
                 highlightSpans: highlights.highlightSpans
                     .map(span => {
-                    const textSpan = transformSpan(span.fileName ?? highlights.fileName, span.textSpan, language_core_1.isHighlightEnabled)?.textSpan;
+                    const textSpan = (0, transform_1.transformSpan)(files, span.fileName ?? highlights.fileName, span.textSpan, language_core_1.isHighlightEnabled)?.textSpan;
                     if (textSpan) {
                         return {
                             ...span,
-                            contextSpan: transformSpan(span.fileName ?? highlights.fileName, span.contextSpan, language_core_1.isHighlightEnabled)?.textSpan,
+                            contextSpan: (0, transform_1.transformSpan)(files, span.fileName ?? highlights.fileName, span.contextSpan, language_core_1.isHighlightEnabled)?.textSpan,
                             textSpan,
                         };
                     }
                 })
-                    .filter(notEmpty),
+                    .filter(utils_1.notEmpty),
             };
         });
         return resolved;
     };
     languageService.getApplicableRefactors = (fileName, positionOrRange, preferences, triggerReason, kind, includeInteractiveActions) => {
-        const [virtualFile, sourceFile, map] = getVirtualFileAndMap(fileName);
+        const [virtualFile, sourceFile, map] = (0, utils_1.getVirtualFileAndMap)(files, fileName);
         if (virtualFile) {
             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 + (isTsPlugin ? sourceFile.snapshot.getLength() : 0)
+                        ? generateOffset + sourceFile.snapshot.getLength()
                         : {
-                            pos: generateOffset + (isTsPlugin ? sourceFile.snapshot.getLength() : 0),
-                            end: generateOffset + positionOrRange.end - positionOrRange.pos + (isTsPlugin ? sourceFile.snapshot.getLength() : 0),
+                            pos: generateOffset + sourceFile.snapshot.getLength(),
+                            end: generateOffset + positionOrRange.end - positionOrRange.pos + sourceFile.snapshot.getLength(),
                         };
                     return getApplicableRefactors(fileName, por, preferences, triggerReason, kind, includeInteractiveActions);
                 }
@@ -155,15 +179,15 @@ function decorateLanguageService(virtualFiles, languageService, isTsPlugin) {
     };
     languageService.getEditsForRefactor = (fileName, formatOptions, positionOrRange, refactorName, actionName, preferences) => {
         let edits;
-        const [virtualFile, sourceFile, map] = getVirtualFileAndMap(fileName);
+        const [virtualFile, sourceFile, map] = (0, utils_1.getVirtualFileAndMap)(files, fileName);
         if (virtualFile) {
             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 + (isTsPlugin ? sourceFile.snapshot.getLength() : 0)
+                        ? generateOffset + sourceFile.snapshot.getLength()
                         : {
-                            pos: generateOffset + (isTsPlugin ? sourceFile.snapshot.getLength() : 0),
-                            end: generateOffset + positionOrRange.end - positionOrRange.pos + (isTsPlugin ? sourceFile.snapshot.getLength() : 0),
+                            pos: generateOffset + sourceFile.snapshot.getLength(),
+                            end: generateOffset + positionOrRange.end - positionOrRange.pos + sourceFile.snapshot.getLength(),
                         };
                     edits = getEditsForRefactor(fileName, formatOptions, por, refactorName, actionName, preferences);
                 }
@@ -174,19 +198,19 @@ function decorateLanguageService(virtualFiles, languageService, isTsPlugin) {
         }
         if (edits) {
             edits.edits = edits.edits
-                .map(edit => transformFileTextChanges(edit, language_core_1.isCodeActionsEnabled))
-                .filter(notEmpty);
+                .map(edit => (0, transform_1.transformFileTextChanges)(files, edit, language_core_1.isCodeActionsEnabled))
+                .filter(utils_1.notEmpty);
             return edits;
         }
     };
     languageService.getRenameInfo = (fileName, position, options) => {
-        const [virtualFile, sourceFile, map] = getVirtualFileAndMap(fileName);
+        const [virtualFile, sourceFile, map] = (0, utils_1.getVirtualFileAndMap)(files, fileName);
         if (virtualFile) {
             for (const [generateOffset, mapping] of map.getGeneratedOffsets(position)) {
                 if ((0, language_core_1.isRenameEnabled)(mapping.data)) {
-                    const info = getRenameInfo(fileName, generateOffset + (isTsPlugin ? sourceFile.snapshot.getLength() : 0), options);
+                    const info = getRenameInfo(fileName, generateOffset + sourceFile.snapshot.getLength(), options);
                     if (info.canRename) {
-                        const span = transformSpan(fileName, info.triggerSpan, language_core_1.isRenameEnabled);
+                        const span = (0, transform_1.transformSpan)(files, fileName, info.triggerSpan, language_core_1.isRenameEnabled);
                         if (span) {
                             info.triggerSpan = span.textSpan;
                             return info;
@@ -208,15 +232,17 @@ function decorateLanguageService(virtualFiles, languageService, isTsPlugin) {
     };
     languageService.getCodeFixesAtPosition = (fileName, start, end, errorCodes, formatOptions, preferences) => {
         let fixes = [];
-        const [virtualFile, sourceFile, map] = getVirtualFileAndMap(fileName);
+        const [virtualFile, sourceFile, map] = (0, utils_1.getVirtualFileAndMap)(files, fileName);
         if (virtualFile) {
             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 + (isTsPlugin ? sourceFile.snapshot.getLength() : 0), generateEnd + (isTsPlugin ? sourceFile.snapshot.getLength() : 0), errorCodes, formatOptions, preferences);
+                            fixes = getCodeFixesAtPosition(fileName, generateStart + sourceFile.snapshot.getLength(), generateEnd + sourceFile.snapshot.getLength(), errorCodes, formatOptions, preferences);
+                            break;
                         }
                     }
+                    break;
                 }
             }
         }
@@ -224,13 +250,13 @@ function decorateLanguageService(virtualFiles, languageService, isTsPlugin) {
             fixes = getCodeFixesAtPosition(fileName, start, end, errorCodes, formatOptions, preferences);
         }
         fixes = fixes.map(fix => {
-            fix.changes = fix.changes.map(edit => transformFileTextChanges(edit, language_core_1.isCodeActionsEnabled)).filter(notEmpty);
+            fix.changes = fix.changes.map(edit => (0, transform_1.transformFileTextChanges)(files, edit, language_core_1.isCodeActionsEnabled)).filter(utils_1.notEmpty);
             return fix;
         });
         return fixes;
     };
     languageService.getEncodedSemanticClassifications = (fileName, span, format) => {
-        const [virtualFile, sourceFile, map] = getVirtualFileAndMap(fileName);
+        const [virtualFile, sourceFile, map] = (0, utils_1.getVirtualFileAndMap)(files, fileName);
         if (virtualFile) {
             let start;
             let end;
@@ -246,16 +272,14 @@ function decorateLanguageService(virtualFiles, languageService, isTsPlugin) {
                 start = 0;
                 end = 0;
             }
-            if (isTsPlugin) {
-                start += sourceFile.snapshot.getLength();
-                end += sourceFile.snapshot.getLength();
-            }
+            start += sourceFile.snapshot.getLength();
+            end += sourceFile.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] - (isTsPlugin ? sourceFile.snapshot.getLength() : 0))) {
+                for (const [sourceStart, mapping] of map.getSourceOffsets(result.spans[i] - sourceFile.snapshot.getLength())) {
                     if ((0, language_core_1.isSemanticTokensEnabled)(mapping.data)) {
-                        for (const [sourceEnd, mapping] of map.getSourceOffsets(result.spans[i] + result.spans[i + 1] - (isTsPlugin ? sourceFile.snapshot.getLength() : 0))) {
+                        for (const [sourceEnd, mapping] of map.getSourceOffsets(result.spans[i] + result.spans[i + 1] - sourceFile.snapshot.getLength())) {
                             if ((0, language_core_1.isSemanticTokensEnabled)(mapping.data)) {
                                 spans.push(sourceStart, sourceEnd - sourceStart, result.spans[i + 2]);
                                 break;
@@ -274,18 +298,18 @@ function decorateLanguageService(virtualFiles, languageService, isTsPlugin) {
     };
     languageService.getSyntacticDiagnostics = (fileName) => {
         return getSyntacticDiagnostics(fileName)
-            .map(diagnostic => transformDiagnostic(diagnostic))
-            .filter(notEmpty);
+            .map(d => (0, transform_1.transformDiagnostic)(files, d))
+            .filter(utils_1.notEmpty);
     };
     languageService.getSemanticDiagnostics = (fileName) => {
         return getSemanticDiagnostics(fileName)
-            .map(diagnostic => transformDiagnostic(diagnostic))
-            .filter(notEmpty);
+            .map(d => (0, transform_1.transformDiagnostic)(files, d))
+            .filter(utils_1.notEmpty);
     };
     languageService.getSuggestionDiagnostics = (fileName) => {
         return getSuggestionDiagnostics(fileName)
-            .map(diagnostic => transformDiagnostic(diagnostic))
-            .filter(notEmpty);
+            .map(d => (0, transform_1.transformDiagnostic)(files, d))
+            .filter(utils_1.notEmpty);
     };
     languageService.getDefinitionAndBoundSpan = (fileName, position) => {
         const unresolved = linkedCodeFeatureWorker(fileName, position, language_core_1.isDefinitionEnabled, position => getDefinitionAndBoundSpan(fileName, position), function* (result) {
@@ -294,15 +318,15 @@ function decorateLanguageService(virtualFiles, languageService, isTsPlugin) {
             }
         });
         const textSpan = unresolved
-            .map(s => transformSpan(fileName, s.textSpan, language_core_1.isDefinitionEnabled)?.textSpan)
-            .filter(notEmpty)[0];
+            .map(s => (0, transform_1.transformSpan)(files, 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 => transformDocumentSpan(s, language_core_1.isDefinitionEnabled, s.fileName !== fileName))
-            .filter(notEmpty))
-            .filter(notEmpty)
+            ?.map(s => (0, transform_1.transformDocumentSpan)(files, s, language_core_1.isDefinitionEnabled, s.fileName !== fileName))
+            .filter(utils_1.notEmpty))
+            .filter(utils_1.notEmpty)
             .flat();
         return {
             textSpan,
@@ -320,17 +344,17 @@ function decorateLanguageService(virtualFiles, languageService, isTsPlugin) {
         const resolved = unresolved
             .flat()
             .map(symbol => {
-            const definition = transformDocumentSpan(symbol.definition, language_core_1.isDefinitionEnabled);
+            const definition = (0, transform_1.transformDocumentSpan)(files, symbol.definition, language_core_1.isDefinitionEnabled);
             if (definition) {
                 return {
                     definition,
                     references: symbol.references
-                        .map(r => transformDocumentSpan(r, language_core_1.isReferencesEnabled))
-                        .filter(notEmpty),
+                        .map(r => (0, transform_1.transformDocumentSpan)(files, r, language_core_1.isReferencesEnabled))
+                        .filter(utils_1.notEmpty),
                 };
             }
         })
-            .filter(notEmpty);
+            .filter(utils_1.notEmpty);
         return (0, dedupe_1.dedupeReferencedSymbols)(resolved);
     };
     languageService.getDefinitionAtPosition = (fileName, position) => {
@@ -341,8 +365,8 @@ function decorateLanguageService(virtualFiles, languageService, isTsPlugin) {
         });
         const resolved = unresolved
             .flat()
-            .map(s => transformDocumentSpan(s, language_core_1.isDefinitionEnabled, s.fileName !== fileName))
-            .filter(notEmpty);
+            .map(s => (0, transform_1.transformDocumentSpan)(files, s, language_core_1.isDefinitionEnabled, s.fileName !== fileName))
+            .filter(utils_1.notEmpty);
         return (0, dedupe_1.dedupeDocumentSpans)(resolved);
     };
     languageService.getTypeDefinitionAtPosition = (fileName, position) => {
@@ -353,8 +377,8 @@ function decorateLanguageService(virtualFiles, languageService, isTsPlugin) {
         });
         const resolved = unresolved
             .flat()
-            .map(s => transformDocumentSpan(s, language_core_1.isTypeDefinitionEnabled))
-            .filter(notEmpty);
+            .map(s => (0, transform_1.transformDocumentSpan)(files, s, language_core_1.isTypeDefinitionEnabled))
+            .filter(utils_1.notEmpty);
         return (0, dedupe_1.dedupeDocumentSpans)(resolved);
     };
     languageService.getImplementationAtPosition = (fileName, position) => {
@@ -365,8 +389,8 @@ function decorateLanguageService(virtualFiles, languageService, isTsPlugin) {
         });
         const resolved = unresolved
             .flat()
-            .map(s => transformDocumentSpan(s, language_core_1.isImplementationEnabled))
-            .filter(notEmpty);
+            .map(s => (0, transform_1.transformDocumentSpan)(files, s, language_core_1.isImplementationEnabled))
+            .filter(utils_1.notEmpty);
         return (0, dedupe_1.dedupeDocumentSpans)(resolved);
     };
     languageService.findRenameLocations = (fileName, position, findInStrings, findInComments, preferences) => {
@@ -377,8 +401,8 @@ function decorateLanguageService(virtualFiles, languageService, isTsPlugin) {
         });
         const resolved = unresolved
             .flat()
-            .map(s => transformDocumentSpan(s, language_core_1.isRenameEnabled))
-            .filter(notEmpty);
+            .map(s => (0, transform_1.transformDocumentSpan)(files, s, language_core_1.isRenameEnabled))
+            .filter(utils_1.notEmpty);
         return (0, dedupe_1.dedupeDocumentSpans)(resolved);
     };
     languageService.getReferencesAtPosition = (fileName, position) => {
@@ -389,23 +413,23 @@ function decorateLanguageService(virtualFiles, languageService, isTsPlugin) {
         });
         const resolved = unresolved
             .flat()
-            .map(s => transformDocumentSpan(s, language_core_1.isReferencesEnabled))
-            .filter(notEmpty);
+            .map(s => (0, transform_1.transformDocumentSpan)(files, s, language_core_1.isReferencesEnabled))
+            .filter(utils_1.notEmpty);
         return (0, dedupe_1.dedupeDocumentSpans)(resolved);
     };
-    // need client patch
     languageService.getCompletionsAtPosition = (fileName, position, options, formattingSettings) => {
-        const [virtualFile, sourceFile, map] = getVirtualFileAndMap(fileName);
+        const [virtualFile, sourceFile, map] = (0, utils_1.getVirtualFileAndMap)(files, fileName);
         if (virtualFile) {
             for (const [generateOffset, mapping] of map.getGeneratedOffsets(position)) {
                 if ((0, language_core_1.isCompletionEnabled)(mapping.data)) {
-                    const result = getCompletionsAtPosition(fileName, generateOffset + (isTsPlugin ? sourceFile.snapshot.getLength() : 0), options, formattingSettings);
+                    const result = getCompletionsAtPosition(fileName, generateOffset + sourceFile.snapshot.getLength(), options, formattingSettings);
                     if (result) {
                         for (const entry of result.entries) {
-                            entry.replacementSpan = transformSpan(fileName, entry.replacementSpan, language_core_1.isCompletionEnabled)?.textSpan;
+                            entry.replacementSpan = (0, transform_1.transformSpan)(files, fileName, entry.replacementSpan, language_core_1.isCompletionEnabled)?.textSpan;
                         }
-                        result.optionalReplacementSpan = transformSpan(fileName, result.optionalReplacementSpan, language_core_1.isCompletionEnabled)?.textSpan;
+                        result.optionalReplacementSpan = (0, transform_1.transformSpan)(files, fileName, result.optionalReplacementSpan, language_core_1.isCompletionEnabled)?.textSpan;
                     }
+                    return result;
                 }
             }
         }
@@ -414,16 +438,28 @@ function decorateLanguageService(virtualFiles, languageService, isTsPlugin) {
         }
     };
     languageService.getCompletionEntryDetails = (fileName, position, entryName, formatOptions, source, preferences, data) => {
-        const details = getCompletionEntryDetails(fileName, position, entryName, formatOptions, source, preferences, data);
+        let details;
+        const [virtualFile, sourceFile, map] = (0, utils_1.getVirtualFileAndMap)(files, fileName);
+        if (virtualFile) {
+            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);
+                    break;
+                }
+            }
+        }
+        else {
+            return getCompletionEntryDetails(fileName, position, entryName, formatOptions, source, preferences, data);
+        }
         if (details?.codeActions) {
             for (const codeAction of details.codeActions) {
-                codeAction.changes = codeAction.changes.map(edit => transformFileTextChanges(edit, language_core_1.isCodeLensEnabled)).filter(notEmpty);
+                codeAction.changes = codeAction.changes.map(edit => (0, transform_1.transformFileTextChanges)(files, edit, language_core_1.isCompletionEnabled)).filter(utils_1.notEmpty);
             }
         }
         return details;
     };
     languageService.provideInlayHints = (fileName, span, preferences) => {
-        const [virtualFile, sourceFile, map] = getVirtualFileAndMap(fileName);
+        const [virtualFile, sourceFile, map] = (0, utils_1.getVirtualFileAndMap)(files, fileName);
         if (virtualFile) {
             let start;
             let end;
@@ -439,14 +475,12 @@ function decorateLanguageService(virtualFiles, languageService, isTsPlugin) {
                 start = 0;
                 end = 0;
             }
-            if (isTsPlugin) {
-                start += sourceFile.snapshot.getLength();
-                end += sourceFile.snapshot.getLength();
-            }
+            start += sourceFile.snapshot.getLength();
+            end += sourceFile.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 - (isTsPlugin ? sourceFile.snapshot.getLength() : 0))) {
+                for (const [sourcePosition, mapping] of map.getSourceOffsets(hint.position - sourceFile.snapshot.getLength())) {
                     if ((0, language_core_1.isInlayHintsEnabled)(mapping.data)) {
                         hints.push({
                             ...hint,
@@ -465,18 +499,18 @@ function decorateLanguageService(virtualFiles, languageService, isTsPlugin) {
     languageService.getFileReferences = (fileName) => {
         const unresolved = getFileReferences(fileName);
         const resolved = unresolved
-            .map(s => transformDocumentSpan(s, language_core_1.isReferencesEnabled))
-            .filter(notEmpty);
+            .map(s => (0, transform_1.transformDocumentSpan)(files, 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 [virtualFile, sourceFile, map] = getVirtualFileAndMap(fileName);
+        const [virtualFile, sourceFile, map] = (0, utils_1.getVirtualFileAndMap)(files, fileName);
         if (virtualFile) {
             for (const [generateOffset, mapping] of map.getGeneratedOffsets(position)) {
                 if (filter(mapping.data)) {
-                    process(fileName, generateOffset + (isTsPlugin ? sourceFile.snapshot.getLength() : 0));
+                    process(fileName, generateOffset + sourceFile.snapshot.getLength());
                 }
             }
         }
@@ -494,167 +528,18 @@ function decorateLanguageService(virtualFiles, languageService, isTsPlugin) {
             results = results.concat(result);
             for (const ref of getLinkedCodes(result)) {
                 processedFilePositions.add(ref[0] + ':' + ref[1]);
-                const [virtualFile, sourceFile] = getVirtualFileAndMap(ref[0]);
+                const [virtualFile, sourceFile] = (0, utils_1.getVirtualFileAndMap)(files, ref[0]);
                 if (!virtualFile)
                     continue;
-                const linkedCodeMap = virtualFiles.getLinkedCodeMap(virtualFile);
+                const linkedCodeMap = files.getLinkedCodeMap(virtualFile);
                 if (!linkedCodeMap)
                     continue;
-                for (const linkedCodeOffset of linkedCodeMap.getLinkedOffsets(ref[1] - (isTsPlugin ? sourceFile.snapshot.getLength() : 0))) {
-                    process(ref[0], linkedCodeOffset + (isTsPlugin ? sourceFile.snapshot.getLength() : 0));
-                }
-            }
-        }
-    }
-    // transforms
-    function transformCallHierarchyItem(item, filter) {
-        const span = transformSpan(item.file, item.span, filter);
-        const selectionSpan = transformSpan(item.file, item.selectionSpan, filter);
-        return {
-            ...item,
-            span: span?.textSpan ?? { start: 0, length: 0 },
-            selectionSpan: selectionSpan?.textSpan ?? { start: 0, length: 0 },
-        };
-    }
-    function transformDiagnostic(diagnostic) {
-        if (!transformedDiagnostics.has(diagnostic)) {
-            if (diagnostic.start !== undefined && diagnostic.file) {
-                transformedDiagnostics.set(diagnostic, undefined);
-                const [virtualFile, sourceFile, map] = getVirtualFileAndMap(diagnostic.file?.fileName);
-                if (virtualFile) {
-                    for (const [sourceOffset, mapping] of map.getSourceOffsets(diagnostic.start - (isTsPlugin ? sourceFile.snapshot.getLength() : 0))) {
-                        if ((0, language_core_1.shouldReportDiagnostics)(mapping.data)) {
-                            transformedDiagnostics.set(diagnostic, {
-                                ...diagnostic,
-                                start: sourceOffset,
-                            });
-                            break;
-                        }
-                    }
-                }
-                else {
-                    transformedDiagnostics.set(diagnostic, diagnostic);
+                for (const linkedCodeOffset of linkedCodeMap.getLinkedOffsets(ref[1] - sourceFile.snapshot.getLength())) {
+                    process(ref[0], linkedCodeOffset + sourceFile.snapshot.getLength());
                 }
             }
-            else {
-                transformedDiagnostics.set(diagnostic, diagnostic);
-            }
-            if (diagnostic.relatedInformation) {
-                diagnostic.relatedInformation = diagnostic.relatedInformation
-                    .map(transformDiagnostic)
-                    .filter(notEmpty);
-            }
         }
-        return transformedDiagnostics.get(diagnostic);
-    }
-    function transformFileTextChanges(changes, filter) {
-        const [_, source] = getVirtualFileAndMap(changes.fileName);
-        if (source) {
-            return {
-                ...changes,
-                fileName: source.id,
-                textChanges: changes.textChanges.map(c => {
-                    const span = transformSpan(changes.fileName, c.span, filter);
-                    if (span) {
-                        return {
-                            ...c,
-                            span: span.textSpan,
-                        };
-                    }
-                }).filter(notEmpty),
-            };
-        }
-        else {
-            return changes;
-        }
-    }
-    function transformDocumentSpan(documentSpan, filter, shouldFallback) {
-        let textSpan = transformSpan(documentSpan.fileName, documentSpan.textSpan, filter);
-        if (!textSpan && shouldFallback) {
-            const [virtualFile, source] = getVirtualFileAndMap(documentSpan.fileName);
-            if (virtualFile) {
-                textSpan = {
-                    fileName: source.id,
-                    textSpan: { start: 0, length: 0 },
-                };
-            }
-        }
-        if (!textSpan)
-            return;
-        const contextSpan = transformSpan(documentSpan.fileName, documentSpan.contextSpan, filter);
-        const originalTextSpan = transformSpan(documentSpan.originalFileName, documentSpan.originalTextSpan, filter);
-        const originalContextSpan = transformSpan(documentSpan.originalFileName, documentSpan.originalContextSpan, filter);
-        return {
-            ...documentSpan,
-            fileName: textSpan.fileName,
-            textSpan: textSpan.textSpan,
-            contextSpan: contextSpan?.textSpan,
-            originalFileName: originalTextSpan?.fileName,
-            originalTextSpan: originalTextSpan?.textSpan,
-            originalContextSpan: originalContextSpan?.textSpan,
-        };
-    }
-    function transformSpan(fileName, textSpan, filter) {
-        if (!fileName)
-            return;
-        if (!textSpan)
-            return;
-        const [virtualFile, sourceFile, map] = getVirtualFileAndMap(fileName);
-        if (virtualFile) {
-            for (const sourceStart of map.getSourceOffsets(textSpan.start - (isTsPlugin ? sourceFile.snapshot.getLength() : 0))) {
-                if (filter(sourceStart[1].data)) {
-                    for (const sourceEnd of map.getSourceOffsets(textSpan.start + textSpan.length - (isTsPlugin ? sourceFile.snapshot.getLength() : 0))) {
-                        if (filter(sourceEnd[1].data)) {
-                            return {
-                                fileName: sourceFile.id,
-                                textSpan: {
-                                    start: sourceStart[0],
-                                    length: sourceEnd[0] - sourceStart[0],
-                                },
-                            };
-                        }
-                    }
-                }
-            }
-        }
-        else {
-            return {
-                fileName,
-                textSpan,
-            };
-        }
-    }
-    function getVirtualFileAndMap(fileName) {
-        if (isTsPlugin) {
-            const sourceFile = virtualFiles.getSourceFile(fileName);
-            if (sourceFile?.virtualFile) {
-                for (const virtualFile of (0, language_core_1.forEachEmbeddedFile)(sourceFile.virtualFile[0])) {
-                    const ext = virtualFile.id.substring(fileName.length);
-                    if (virtualFile.typescript && (ext === '.d.ts' || ext.match(/^\.(js|ts)x?$/))) {
-                        for (const map of virtualFiles.getMaps(virtualFile)) {
-                            if (map[1][0] === sourceFile.snapshot) {
-                                return [virtualFile, sourceFile, map[1][1]];
-                            }
-                        }
-                    }
-                }
-            }
-        }
-        else {
-            const [virtualFile, sourceFile] = virtualFiles.getVirtualFile(fileName);
-            if (virtualFile) {
-                for (const map of virtualFiles.getMaps(virtualFile)) {
-                    if (map[1][0] === sourceFile.snapshot) {
-                        return [virtualFile, sourceFile, map[1][1]];
-                    }
-                }
-            }
-        }
-        return [undefined, undefined, undefined];
     }
 }
 exports.decorateLanguageService = decorateLanguageService;
-function notEmpty(value) {
-    return value !== null && value !== undefined;
-}
 //# sourceMappingURL=decorateLanguageService.js.map