@volar/typescript 2.3.0-alpha.1 → 2.3.0-alpha.11

package/lib/node/decorateLanguageService.js

@@ -3,17 +3,17 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.decorateLanguageService = void 0;
 const language_core_1 = require("@volar/language-core");
 const dedupe_1 = require("./dedupe");
-const utils_1 = require("./utils");
 const transform_1 = require("./transform");
+const utils_1 = require("./utils");
 const windowsPathReg = /\\/g;
 function decorateLanguageService(language, languageService) {
     // ignored methods
     const { getNavigationTree, getOutliningSpans, } = languageService;
     languageService.getNavigationTree = filePath => {
         const fileName = filePath.replace(windowsPathReg, '/');
-        const [serviceScript] = (0, utils_1.getServiceScript)(language, fileName);
-        if (serviceScript) {
-            const tree = getNavigationTree(fileName);
+        const [serviceScript, targetScript] = (0, utils_1.getServiceScript)(language, fileName);
+        if (serviceScript || targetScript?.associatedOnly) {
+            const tree = getNavigationTree(targetScript.id);
             tree.childItems = undefined;
             return tree;
         }
@@ -23,8 +23,8 @@ function decorateLanguageService(language, languageService) {
     };
     languageService.getOutliningSpans = filePath => {
         const fileName = filePath.replace(windowsPathReg, '/');
-        const [serviceScript] = (0, utils_1.getServiceScript)(language, fileName);
-        if (serviceScript) {
+        const [serviceScript, targetScript] = (0, utils_1.getServiceScript)(language, fileName);
+        if (serviceScript || targetScript?.associatedOnly) {
             return [];
         }
         else {
@@ -35,14 +35,18 @@ function decorateLanguageService(language, languageService) {
     const { findReferences, findRenameLocations, getCompletionEntryDetails, getCompletionsAtPosition, getDefinitionAndBoundSpan, getDefinitionAtPosition, getFileReferences, getFormattingEditsForDocument, getFormattingEditsForRange, getFormattingEditsAfterKeystroke, getImplementationAtPosition, getLinkedEditingRangeAtPosition, getQuickInfoAtPosition, getSignatureHelpItems, getReferencesAtPosition, getSemanticDiagnostics, getSyntacticDiagnostics, getSuggestionDiagnostics, getTypeDefinitionAtPosition, getEncodedSemanticClassifications, getDocumentHighlights, getApplicableRefactors, getEditsForFileRename, getEditsForRefactor, getRenameInfo, getCodeFixesAtPosition, prepareCallHierarchy, provideCallHierarchyIncomingCalls, provideCallHierarchyOutgoingCalls, provideInlayHints, organizeImports, } = languageService;
     languageService.getFormattingEditsForDocument = (filePath, options) => {
         const fileName = filePath.replace(windowsPathReg, '/');
-        const [serviceScript, sourceScript, map] = (0, utils_1.getServiceScript)(language, fileName);
+        const [serviceScript, targetScript, sourceScript] = (0, utils_1.getServiceScript)(language, fileName);
+        if (targetScript?.associatedOnly) {
+            return [];
+        }
         if (serviceScript) {
+            const map = language.maps.get(serviceScript.code, targetScript);
             if (!map.mappings.some(mapping => (0, language_core_1.isFormattingEnabled)(mapping.data))) {
                 return [];
             }
-            const edits = getFormattingEditsForDocument(fileName, options);
+            const edits = getFormattingEditsForDocument(targetScript.id, options);
             return edits
-                .map(edit => (0, transform_1.transformTextChange)(sourceScript, map, edit, language_core_1.isFormattingEnabled))
+                .map(edit => (0, transform_1.transformTextChange)(sourceScript, language, serviceScript, edit, language_core_1.isFormattingEnabled)?.[1])
                 .filter(utils_1.notEmpty);
         }
         else {
@@ -51,14 +55,17 @@ function decorateLanguageService(language, languageService) {
     };
     languageService.getFormattingEditsForRange = (filePath, start, end, options) => {
         const fileName = filePath.replace(windowsPathReg, '/');
-        const [serviceScript, sourceScript, map] = (0, utils_1.getServiceScript)(language, fileName);
+        const [serviceScript, targetScript, sourceScript] = (0, utils_1.getServiceScript)(language, fileName);
+        if (targetScript?.associatedOnly) {
+            return [];
+        }
         if (serviceScript) {
-            const generateStart = (0, transform_1.toGeneratedOffset)(sourceScript, map, start, language_core_1.isFormattingEnabled);
-            const generateEnd = (0, transform_1.toGeneratedOffset)(sourceScript, map, end, language_core_1.isFormattingEnabled);
+            const generateStart = (0, transform_1.toGeneratedOffset)(language, serviceScript, targetScript, start, language_core_1.isFormattingEnabled);
+            const generateEnd = (0, transform_1.toGeneratedOffset)(language, serviceScript, targetScript, end, language_core_1.isFormattingEnabled);
             if (generateStart !== undefined && generateEnd !== undefined) {
-                const edits = getFormattingEditsForRange(fileName, generateStart, generateEnd, options);
+                const edits = getFormattingEditsForRange(targetScript.id, generateStart, generateEnd, options);
                 return edits
-                    .map(edit => (0, transform_1.transformTextChange)(sourceScript, map, edit, language_core_1.isFormattingEnabled))
+                    .map(edit => (0, transform_1.transformTextChange)(sourceScript, language, serviceScript, edit, language_core_1.isFormattingEnabled)?.[1])
                     .filter(utils_1.notEmpty);
             }
             return [];
@@ -69,13 +76,16 @@ function decorateLanguageService(language, languageService) {
     };
     languageService.getFormattingEditsAfterKeystroke = (filePath, position, key, options) => {
         const fileName = filePath.replace(windowsPathReg, '/');
-        const [serviceScript, sourceScript, map] = (0, utils_1.getServiceScript)(language, fileName);
+        const [serviceScript, targetScript, sourceScript] = (0, utils_1.getServiceScript)(language, fileName);
+        if (targetScript?.associatedOnly) {
+            return [];
+        }
         if (serviceScript) {
-            const generatePosition = (0, transform_1.toGeneratedOffset)(sourceScript, map, position, language_core_1.isFormattingEnabled);
+            const generatePosition = (0, transform_1.toGeneratedOffset)(language, serviceScript, targetScript, position, language_core_1.isFormattingEnabled);
             if (generatePosition !== undefined) {
-                const edits = getFormattingEditsAfterKeystroke(fileName, generatePosition, key, options);
+                const edits = getFormattingEditsAfterKeystroke(targetScript.id, generatePosition, key, options);
                 return edits
-                    .map(edit => (0, transform_1.transformTextChange)(sourceScript, map, edit, language_core_1.isFormattingEnabled))
+                    .map(edit => (0, transform_1.transformTextChange)(sourceScript, language, serviceScript, edit, language_core_1.isFormattingEnabled)?.[1])
                     .filter(utils_1.notEmpty);
             }
             return [];
@@ -86,21 +96,22 @@ function decorateLanguageService(language, languageService) {
     };
     languageService.getEditsForFileRename = (oldFilePath, newFilePath, formatOptions, preferences) => {
         const edits = getEditsForFileRename(oldFilePath, newFilePath, formatOptions, preferences);
-        return edits
-            .map(edit => (0, transform_1.transformFileTextChanges)(language, edit, language_core_1.isRenameEnabled))
-            .filter(utils_1.notEmpty);
+        return (0, transform_1.transformFileTextChanges)(language, edits, language_core_1.isRenameEnabled);
     };
     languageService.getLinkedEditingRangeAtPosition = (filePath, position) => {
         const fileName = filePath.replace(windowsPathReg, '/');
-        const [serviceScript, sourceScript, map] = (0, utils_1.getServiceScript)(language, fileName);
+        const [serviceScript, targetScript, sourceScript] = (0, utils_1.getServiceScript)(language, fileName);
+        if (targetScript?.associatedOnly) {
+            return undefined;
+        }
         if (serviceScript) {
-            const generatePosition = (0, transform_1.toGeneratedOffset)(sourceScript, map, position, language_core_1.isLinkedEditingEnabled);
+            const generatePosition = (0, transform_1.toGeneratedOffset)(language, serviceScript, targetScript, position, language_core_1.isLinkedEditingEnabled);
             if (generatePosition !== undefined) {
-                const info = getLinkedEditingRangeAtPosition(fileName, generatePosition);
+                const info = getLinkedEditingRangeAtPosition(targetScript.id, generatePosition);
                 if (info) {
                     return {
                         ranges: info.ranges
-                            .map(span => (0, transform_1.transformTextSpan)(sourceScript, map, span, language_core_1.isLinkedEditingEnabled))
+                            .map(span => (0, transform_1.transformTextSpan)(sourceScript, language, serviceScript, span, language_core_1.isLinkedEditingEnabled)?.[1])
                             .filter(utils_1.notEmpty),
                         wordPattern: info.wordPattern,
                     };
@@ -113,11 +124,14 @@ function decorateLanguageService(language, languageService) {
     };
     languageService.prepareCallHierarchy = (filePath, position) => {
         const fileName = filePath.replace(windowsPathReg, '/');
-        const [serviceScript, sourceScript, map] = (0, utils_1.getServiceScript)(language, fileName);
+        const [serviceScript, targetScript] = (0, utils_1.getServiceScript)(language, fileName);
+        if (targetScript?.associatedOnly) {
+            return undefined;
+        }
         if (serviceScript) {
-            const generatePosition = (0, transform_1.toGeneratedOffset)(sourceScript, map, position, language_core_1.isCallHierarchyEnabled);
+            const generatePosition = (0, transform_1.toGeneratedOffset)(language, serviceScript, targetScript, position, language_core_1.isCallHierarchyEnabled);
             if (generatePosition !== undefined) {
-                const item = prepareCallHierarchy(fileName, generatePosition);
+                const item = prepareCallHierarchy(targetScript.id, generatePosition);
                 if (Array.isArray(item)) {
                     return item.map(item => (0, transform_1.transformCallHierarchyItem)(language, item, language_core_1.isCallHierarchyEnabled));
                 }
@@ -133,11 +147,14 @@ function decorateLanguageService(language, languageService) {
     languageService.provideCallHierarchyIncomingCalls = (filePath, position) => {
         let calls = [];
         const fileName = filePath.replace(windowsPathReg, '/');
-        const [serviceScript, sourceScript, map] = (0, utils_1.getServiceScript)(language, fileName);
+        const [serviceScript, targetScript] = (0, utils_1.getServiceScript)(language, fileName);
+        if (targetScript?.associatedOnly) {
+            return [];
+        }
         if (serviceScript) {
-            const generatePosition = (0, transform_1.toGeneratedOffset)(sourceScript, map, position, language_core_1.isCallHierarchyEnabled);
+            const generatePosition = (0, transform_1.toGeneratedOffset)(language, serviceScript, targetScript, position, language_core_1.isCallHierarchyEnabled);
             if (generatePosition !== undefined) {
-                calls = provideCallHierarchyIncomingCalls(fileName, generatePosition);
+                calls = provideCallHierarchyIncomingCalls(targetScript.id, generatePosition);
             }
         }
         else {
@@ -158,9 +175,12 @@ function decorateLanguageService(language, languageService) {
     languageService.provideCallHierarchyOutgoingCalls = (filePath, position) => {
         let calls = [];
         const fileName = filePath.replace(windowsPathReg, '/');
-        const [serviceScript, sourceScript, map] = (0, utils_1.getServiceScript)(language, fileName);
+        const [serviceScript, targetScript, sourceScript] = (0, utils_1.getServiceScript)(language, fileName);
+        if (targetScript?.associatedOnly) {
+            return [];
+        }
         if (serviceScript) {
-            const generatePosition = (0, transform_1.toGeneratedOffset)(sourceScript, map, position, language_core_1.isCallHierarchyEnabled);
+            const generatePosition = (0, transform_1.toGeneratedOffset)(language, serviceScript, targetScript, position, language_core_1.isCallHierarchyEnabled);
             if (generatePosition !== undefined) {
                 calls = provideCallHierarchyOutgoingCalls(fileName, generatePosition);
             }
@@ -172,8 +192,8 @@ function decorateLanguageService(language, languageService) {
             .map(call => {
             const to = (0, transform_1.transformCallHierarchyItem)(language, call.to, language_core_1.isCallHierarchyEnabled);
             const fromSpans = call.fromSpans
-                .map(span => sourceScript
-                ? (0, transform_1.transformTextSpan)(sourceScript, map, span, language_core_1.isCallHierarchyEnabled)
+                .map(span => serviceScript
+                ? (0, transform_1.transformTextSpan)(sourceScript, language, serviceScript, span, language_core_1.isCallHierarchyEnabled)?.[1]
                 : span)
                 .filter(utils_1.notEmpty);
             return {
@@ -184,27 +204,24 @@ function decorateLanguageService(language, languageService) {
     };
     languageService.organizeImports = (args, formatOptions, preferences) => {
         const unresolved = organizeImports(args, formatOptions, preferences);
-        const resolved = unresolved
-            .map(changes => (0, transform_1.transformFileTextChanges)(language, changes, language_core_1.isCodeActionsEnabled))
-            .filter(utils_1.notEmpty);
-        return resolved;
+        return (0, transform_1.transformFileTextChanges)(language, unresolved, language_core_1.isCodeActionsEnabled);
     };
     languageService.getQuickInfoAtPosition = (filePath, position) => {
         const fileName = filePath.replace(windowsPathReg, '/');
-        const [serviceScript, sourceScript, map] = (0, utils_1.getServiceScript)(language, fileName);
+        const [serviceScript, targetScript, sourceScript] = (0, utils_1.getServiceScript)(language, fileName);
+        if (targetScript?.associatedOnly) {
+            return undefined;
+        }
         if (serviceScript) {
             const infos = [];
-            for (const [generatePosition, mapping] of (0, transform_1.toGeneratedOffsets)(sourceScript, map, position)) {
-                if (!(0, language_core_1.isHoverEnabled)(mapping.data)) {
-                    continue;
-                }
-                const info = getQuickInfoAtPosition(fileName, generatePosition);
+            for (const [generatePosition] of (0, transform_1.toGeneratedOffsets)(language, serviceScript, targetScript, position, language_core_1.isHoverEnabled)) {
+                const info = getQuickInfoAtPosition(targetScript.id, generatePosition);
                 if (info) {
-                    const textSpan = (0, transform_1.transformTextSpan)(sourceScript, map, info.textSpan, language_core_1.isHoverEnabled);
+                    const textSpan = (0, transform_1.transformTextSpan)(sourceScript, language, serviceScript, info.textSpan, language_core_1.isHoverEnabled)?.[1];
                     if (textSpan) {
                         infos.push({
                             ...info,
-                            textSpan,
+                            textSpan: textSpan,
                         });
                     }
                 }
@@ -254,13 +271,16 @@ function decorateLanguageService(language, languageService) {
     };
     languageService.getSignatureHelpItems = (filePath, position, options) => {
         const fileName = filePath.replace(windowsPathReg, '/');
-        const [serviceScript, sourceScript, map] = (0, utils_1.getServiceScript)(language, fileName);
+        const [serviceScript, targetScript, sourceScript] = (0, utils_1.getServiceScript)(language, fileName);
+        if (targetScript?.associatedOnly) {
+            return undefined;
+        }
         if (serviceScript) {
-            const generatePosition = (0, transform_1.toGeneratedOffset)(sourceScript, map, position, language_core_1.isSignatureHelpEnabled);
+            const generatePosition = (0, transform_1.toGeneratedOffset)(language, serviceScript, targetScript, position, language_core_1.isSignatureHelpEnabled);
             if (generatePosition !== undefined) {
-                const result = getSignatureHelpItems(fileName, generatePosition, options);
+                const result = getSignatureHelpItems(targetScript.id, generatePosition, options);
                 if (result) {
-                    const applicableSpan = (0, transform_1.transformTextSpan)(sourceScript, map, result.applicableSpan, language_core_1.isSignatureHelpEnabled);
+                    const applicableSpan = (0, transform_1.transformTextSpan)(sourceScript, language, serviceScript, result.applicableSpan, language_core_1.isSignatureHelpEnabled)?.[1];
                     if (applicableSpan) {
                         return {
                             ...result,
@@ -276,7 +296,7 @@ function decorateLanguageService(language, languageService) {
     };
     languageService.getDocumentHighlights = (filePath, position, filesToSearch) => {
         const fileName = filePath.replace(windowsPathReg, '/');
-        const unresolved = linkedCodeFeatureWorker(fileName, position, language_core_1.isHighlightEnabled, position => getDocumentHighlights(fileName, position, filesToSearch), function* (result) {
+        const unresolved = linkedCodeFeatureWorker(fileName, position, language_core_1.isHighlightEnabled, (fileName, position) => getDocumentHighlights(fileName, position, filesToSearch), function* (result) {
             for (const ref of result) {
                 for (const reference of ref.highlightSpans) {
                     yield [reference.fileName ?? ref.fileName, reference.textSpan.start];
@@ -290,7 +310,7 @@ function decorateLanguageService(language, languageService) {
                 ...highlights,
                 highlightSpans: highlights.highlightSpans
                     .map(span => {
-                    const textSpan = (0, transform_1.transformSpan)(language, 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) ?? {};
                     if (textSpan) {
                         return {
                             ...span,
@@ -306,9 +326,12 @@ function decorateLanguageService(language, languageService) {
     };
     languageService.getApplicableRefactors = (filePath, positionOrRange, preferences, triggerReason, kind, includeInteractiveActions) => {
         const fileName = filePath.replace(windowsPathReg, '/');
-        const [serviceScript, sourceScript, map] = (0, utils_1.getServiceScript)(language, fileName);
+        const [serviceScript, targetScript] = (0, utils_1.getServiceScript)(language, fileName);
+        if (targetScript?.associatedOnly) {
+            return [];
+        }
         if (serviceScript) {
-            const generatePosition = (0, transform_1.toGeneratedOffset)(sourceScript, map, typeof positionOrRange === 'number' ? positionOrRange : positionOrRange.pos, language_core_1.isCodeActionsEnabled);
+            const generatePosition = (0, transform_1.toGeneratedOffset)(language, serviceScript, targetScript, typeof positionOrRange === 'number' ? positionOrRange : positionOrRange.pos, language_core_1.isCodeActionsEnabled);
             if (generatePosition !== undefined) {
                 const por = typeof positionOrRange === 'number'
                     ? generatePosition
@@ -316,7 +339,7 @@ function decorateLanguageService(language, languageService) {
                         pos: generatePosition,
                         end: generatePosition + positionOrRange.end - positionOrRange.pos,
                     };
-                return getApplicableRefactors(fileName, por, preferences, triggerReason, kind, includeInteractiveActions);
+                return getApplicableRefactors(targetScript.id, por, preferences, triggerReason, kind, includeInteractiveActions);
             }
             return [];
         }
@@ -327,9 +350,12 @@ function decorateLanguageService(language, languageService) {
     languageService.getEditsForRefactor = (filePath, formatOptions, positionOrRange, refactorName, actionName, preferences) => {
         let edits;
         const fileName = filePath.replace(windowsPathReg, '/');
-        const [serviceScript, sourceScript, map] = (0, utils_1.getServiceScript)(language, fileName);
+        const [serviceScript, targetScript] = (0, utils_1.getServiceScript)(language, fileName);
+        if (targetScript?.associatedOnly) {
+            return undefined;
+        }
         if (serviceScript) {
-            const generatePosition = (0, transform_1.toGeneratedOffset)(sourceScript, map, typeof positionOrRange === 'number'
+            const generatePosition = (0, transform_1.toGeneratedOffset)(language, serviceScript, targetScript, typeof positionOrRange === 'number'
                 ? positionOrRange
                 : positionOrRange.pos, language_core_1.isCodeActionsEnabled);
             if (generatePosition !== undefined) {
@@ -339,31 +365,32 @@ function decorateLanguageService(language, languageService) {
                         pos: generatePosition,
                         end: generatePosition + positionOrRange.end - positionOrRange.pos,
                     };
-                edits = getEditsForRefactor(fileName, formatOptions, por, refactorName, actionName, preferences);
+                edits = getEditsForRefactor(targetScript.id, formatOptions, por, refactorName, actionName, preferences);
             }
         }
         else {
             edits = getEditsForRefactor(fileName, formatOptions, positionOrRange, refactorName, actionName, preferences);
         }
         if (edits) {
-            edits.edits = edits.edits
-                .map(edit => (0, transform_1.transformFileTextChanges)(language, edit, language_core_1.isCodeActionsEnabled))
-                .filter(utils_1.notEmpty);
+            edits.edits = (0, transform_1.transformFileTextChanges)(language, edits.edits, language_core_1.isCodeActionsEnabled);
             return edits;
         }
     };
     languageService.getRenameInfo = (filePath, position, options) => {
         const fileName = filePath.replace(windowsPathReg, '/');
-        const [serviceScript, sourceScript, map] = (0, utils_1.getServiceScript)(language, fileName);
+        const [serviceScript, targetScript, sourceScript] = (0, utils_1.getServiceScript)(language, fileName);
+        if (targetScript?.associatedOnly) {
+            return {
+                canRename: false,
+                localizedErrorMessage: "Cannot rename"
+            };
+        }
         if (serviceScript) {
             let failed;
-            for (const [generateOffset, mapping] of (0, transform_1.toGeneratedOffsets)(sourceScript, map, position)) {
-                if (!(0, language_core_1.isRenameEnabled)(mapping.data)) {
-                    continue;
-                }
-                const info = getRenameInfo(fileName, generateOffset, options);
+            for (const [generateOffset] of (0, transform_1.toGeneratedOffsets)(language, serviceScript, targetScript, position, language_core_1.isRenameEnabled)) {
+                const info = getRenameInfo(targetScript.id, generateOffset, options);
                 if (info.canRename) {
-                    const span = (0, transform_1.transformTextSpan)(sourceScript, map, info.triggerSpan, language_core_1.isRenameEnabled);
+                    const span = (0, transform_1.transformTextSpan)(sourceScript, language, serviceScript, info.triggerSpan, language_core_1.isRenameEnabled)?.[1];
                     if (span) {
                         info.triggerSpan = span;
                         return info;
@@ -388,29 +415,39 @@ function decorateLanguageService(language, languageService) {
     languageService.getCodeFixesAtPosition = (filePath, start, end, errorCodes, formatOptions, preferences) => {
         let fixes = [];
         const fileName = filePath.replace(windowsPathReg, '/');
-        const [serviceScript, sourceScript, map] = (0, utils_1.getServiceScript)(language, fileName);
+        const [serviceScript, targetScript] = (0, utils_1.getServiceScript)(language, fileName);
+        if (targetScript?.associatedOnly) {
+            return [];
+        }
         if (serviceScript) {
-            const generateStart = (0, transform_1.toGeneratedOffset)(sourceScript, map, start, language_core_1.isCodeActionsEnabled);
-            const generateEnd = (0, transform_1.toGeneratedOffset)(sourceScript, map, end, language_core_1.isCodeActionsEnabled);
+            const generateStart = (0, transform_1.toGeneratedOffset)(language, serviceScript, targetScript, start, language_core_1.isCodeActionsEnabled);
+            const generateEnd = (0, transform_1.toGeneratedOffset)(language, serviceScript, targetScript, end, language_core_1.isCodeActionsEnabled);
             if (generateStart !== undefined && generateEnd !== undefined) {
-                fixes = getCodeFixesAtPosition(fileName, generateStart, generateEnd, errorCodes, formatOptions, preferences);
+                fixes = getCodeFixesAtPosition(targetScript.id, generateStart, generateEnd, errorCodes, formatOptions, preferences);
             }
         }
         else {
             fixes = getCodeFixesAtPosition(fileName, start, end, errorCodes, formatOptions, preferences);
         }
         fixes = fixes.map(fix => {
-            fix.changes = fix.changes.map(edit => (0, transform_1.transformFileTextChanges)(language, edit, language_core_1.isCodeActionsEnabled)).filter(utils_1.notEmpty);
+            fix.changes = (0, transform_1.transformFileTextChanges)(language, fix.changes, language_core_1.isCodeActionsEnabled);
             return fix;
         });
         return fixes;
     };
     languageService.getEncodedSemanticClassifications = (filePath, span, format) => {
         const fileName = filePath.replace(windowsPathReg, '/');
-        const [serviceScript, sourceScript, map] = (0, utils_1.getServiceScript)(language, fileName);
+        const [serviceScript, targetScript, sourceScript] = (0, utils_1.getServiceScript)(language, fileName);
+        if (targetScript?.associatedOnly) {
+            return {
+                spans: [],
+                endOfLineState: 0
+            };
+        }
         if (serviceScript) {
             let start;
             let end;
+            const map = language.maps.get(serviceScript.code, targetScript);
             for (const mapping of map.mappings) {
                 // TODO reuse the logic from language service
                 if ((0, language_core_1.isSemanticTokensEnabled)(mapping.data) && mapping.sourceOffsets[0] >= span.start && mapping.sourceOffsets[0] <= span.start + span.length) {
@@ -421,16 +458,23 @@ function decorateLanguageService(language, languageService) {
                 }
             }
             start ??= 0;
-            end ??= sourceScript.snapshot.getLength();
-            start += sourceScript.snapshot.getLength();
-            end += sourceScript.snapshot.getLength();
-            const result = getEncodedSemanticClassifications(fileName, { start, length: end - start }, format);
+            end ??= targetScript.snapshot.getLength();
+            const mappingOffset = (0, transform_1.getMappingOffset)(language, serviceScript);
+            start += mappingOffset;
+            end += mappingOffset;
+            const result = getEncodedSemanticClassifications(targetScript.id, { start, length: end - start }, format);
             const spans = [];
             for (let i = 0; i < result.spans.length; i += 3) {
-                const sourceStart = (0, transform_1.toSourceOffset)(sourceScript, map, result.spans[i], language_core_1.isSemanticTokensEnabled);
-                const sourceEnd = (0, transform_1.toSourceOffset)(sourceScript, map, result.spans[i] + result.spans[i + 1], language_core_1.isSemanticTokensEnabled);
-                if (sourceStart !== undefined && sourceEnd !== undefined) {
-                    spans.push(sourceStart, sourceEnd - sourceStart, result.spans[i + 2]);
+                for (const sourceStart of (0, transform_1.toSourceOffsets)(sourceScript, language, serviceScript, result.spans[i], language_core_1.isSemanticTokensEnabled)) {
+                    for (const sourceEnd of (0, transform_1.toSourceOffsets)(sourceScript, language, serviceScript, result.spans[i] + result.spans[i + 1], language_core_1.isSemanticTokensEnabled)) {
+                        if (sourceStart[0] === sourceEnd[0] && sourceEnd[1] >= sourceStart[1]) {
+                            spans.push(sourceStart[1], sourceEnd[1] - sourceStart[1], result.spans[i + 2]);
+                            break;
+                        }
+                    }
+                    if (spans.length) {
+                        break;
+                    }
                 }
             }
             result.spans = spans;
@@ -442,25 +486,37 @@ function decorateLanguageService(language, languageService) {
     };
     languageService.getSyntacticDiagnostics = filePath => {
         const fileName = filePath.replace(windowsPathReg, '/');
+        const [_serviceScript, targetScript, sourceScript] = (0, utils_1.getServiceScript)(language, fileName);
+        if (targetScript?.associatedOnly) {
+            return [];
+        }
         return getSyntacticDiagnostics(fileName)
-            .map(d => (0, transform_1.transformDiagnostic)(language, d, false))
+            .map(d => (0, transform_1.transformDiagnostic)(sourceScript, language, d, languageService.getProgram(), false))
             .filter(utils_1.notEmpty);
     };
     languageService.getSemanticDiagnostics = filePath => {
         const fileName = filePath.replace(windowsPathReg, '/');
+        const [_serviceScript, targetScript, sourceScript] = (0, utils_1.getServiceScript)(language, fileName);
+        if (targetScript?.associatedOnly) {
+            return [];
+        }
         return getSemanticDiagnostics(fileName)
-            .map(d => (0, transform_1.transformDiagnostic)(language, d, false))
+            .map(d => (0, transform_1.transformDiagnostic)(sourceScript, language, d, languageService.getProgram(), false))
             .filter(utils_1.notEmpty);
     };
     languageService.getSuggestionDiagnostics = filePath => {
         const fileName = filePath.replace(windowsPathReg, '/');
+        const [_serviceScript, targetScript, sourceScript] = (0, utils_1.getServiceScript)(language, fileName);
+        if (targetScript?.associatedOnly) {
+            return [];
+        }
         return getSuggestionDiagnostics(fileName)
-            .map(d => (0, transform_1.transformDiagnostic)(language, d, false))
+            .map(d => (0, transform_1.transformDiagnostic)(sourceScript, language, d, languageService.getProgram(), false))
             .filter(utils_1.notEmpty);
     };
     languageService.getDefinitionAndBoundSpan = (filePath, position) => {
         const fileName = filePath.replace(windowsPathReg, '/');
-        const unresolved = linkedCodeFeatureWorker(fileName, position, language_core_1.isDefinitionEnabled, position => getDefinitionAndBoundSpan(fileName, position), function* (result) {
+        const unresolved = linkedCodeFeatureWorker(fileName, position, language_core_1.isDefinitionEnabled, (fileName, position) => getDefinitionAndBoundSpan(fileName, position), function* (result) {
             for (const ref of result.definitions ?? []) {
                 yield [ref.fileName, ref.textSpan.start];
             }
@@ -484,7 +540,7 @@ function decorateLanguageService(language, languageService) {
     };
     languageService.findReferences = (filePath, position) => {
         const fileName = filePath.replace(windowsPathReg, '/');
-        const unresolved = linkedCodeFeatureWorker(fileName, position, language_core_1.isReferencesEnabled, position => findReferences(fileName, position), function* (result) {
+        const unresolved = linkedCodeFeatureWorker(fileName, position, language_core_1.isReferencesEnabled, (fileName, position) => findReferences(fileName, position), function* (result) {
             for (const ref of result) {
                 for (const reference of ref.references) {
                     yield [reference.fileName, reference.textSpan.start];
@@ -506,7 +562,7 @@ function decorateLanguageService(language, languageService) {
     };
     languageService.getDefinitionAtPosition = (filePath, position) => {
         const fileName = filePath.replace(windowsPathReg, '/');
-        const unresolved = linkedCodeFeatureWorker(fileName, position, language_core_1.isDefinitionEnabled, position => getDefinitionAtPosition(fileName, position), function* (result) {
+        const unresolved = linkedCodeFeatureWorker(fileName, position, language_core_1.isDefinitionEnabled, (fileName, position) => getDefinitionAtPosition(fileName, position), function* (result) {
             for (const ref of result) {
                 yield [ref.fileName, ref.textSpan.start];
             }
@@ -519,7 +575,7 @@ function decorateLanguageService(language, languageService) {
     };
     languageService.getTypeDefinitionAtPosition = (filePath, position) => {
         const fileName = filePath.replace(windowsPathReg, '/');
-        const unresolved = linkedCodeFeatureWorker(fileName, position, language_core_1.isTypeDefinitionEnabled, position => getTypeDefinitionAtPosition(fileName, position), function* (result) {
+        const unresolved = linkedCodeFeatureWorker(fileName, position, language_core_1.isTypeDefinitionEnabled, (fileName, position) => getTypeDefinitionAtPosition(fileName, position), function* (result) {
             for (const ref of result) {
                 yield [ref.fileName, ref.textSpan.start];
             }
@@ -532,7 +588,7 @@ function decorateLanguageService(language, languageService) {
     };
     languageService.getImplementationAtPosition = (filePath, position) => {
         const fileName = filePath.replace(windowsPathReg, '/');
-        const unresolved = linkedCodeFeatureWorker(fileName, position, language_core_1.isImplementationEnabled, position => getImplementationAtPosition(fileName, position), function* (result) {
+        const unresolved = linkedCodeFeatureWorker(fileName, position, language_core_1.isImplementationEnabled, (fileName, position) => getImplementationAtPosition(fileName, position), function* (result) {
             for (const ref of result) {
                 yield [ref.fileName, ref.textSpan.start];
             }
@@ -545,7 +601,7 @@ function decorateLanguageService(language, languageService) {
     };
     languageService.findRenameLocations = (filePath, position, findInStrings, findInComments, preferences) => {
         const fileName = filePath.replace(windowsPathReg, '/');
-        const unresolved = linkedCodeFeatureWorker(fileName, position, language_core_1.isRenameEnabled, position => findRenameLocations(fileName, position, findInStrings, findInComments, preferences), function* (result) {
+        const unresolved = linkedCodeFeatureWorker(fileName, position, language_core_1.isRenameEnabled, (fileName, position) => findRenameLocations(fileName, position, findInStrings, findInComments, preferences), function* (result) {
             for (const ref of result) {
                 yield [ref.fileName, ref.textSpan.start];
             }
@@ -558,7 +614,7 @@ function decorateLanguageService(language, languageService) {
     };
     languageService.getReferencesAtPosition = (filePath, position) => {
         const fileName = filePath.replace(windowsPathReg, '/');
-        const unresolved = linkedCodeFeatureWorker(fileName, position, language_core_1.isReferencesEnabled, position => getReferencesAtPosition(fileName, position), function* (result) {
+        const unresolved = linkedCodeFeatureWorker(fileName, position, language_core_1.isReferencesEnabled, (fileName, position) => getReferencesAtPosition(fileName, position), function* (result) {
             for (const ref of result) {
                 yield [ref.fileName, ref.textSpan.start];
             }
@@ -571,14 +627,14 @@ function decorateLanguageService(language, languageService) {
     };
     languageService.getCompletionsAtPosition = (filePath, position, options, formattingSettings) => {
         const fileName = filePath.replace(windowsPathReg, '/');
-        const [serviceScript, sourceScript, map] = (0, utils_1.getServiceScript)(language, fileName);
+        const [serviceScript, targetScript, sourceScript] = (0, utils_1.getServiceScript)(language, fileName);
+        if (targetScript?.associatedOnly) {
+            return undefined;
+        }
         if (serviceScript) {
             const results = [];
-            for (const [generatedOffset, mapping] of (0, transform_1.toGeneratedOffsets)(sourceScript, map, position)) {
-                if (!(0, language_core_1.isCompletionEnabled)(mapping.data)) {
-                    continue;
-                }
-                const result = getCompletionsAtPosition(fileName, generatedOffset, options, formattingSettings);
+            for (const [generatedOffset, mapping] of (0, transform_1.toGeneratedOffsets)(language, serviceScript, targetScript, position, language_core_1.isCompletionEnabled)) {
+                const result = getCompletionsAtPosition(targetScript.id, generatedOffset, options, formattingSettings);
                 if (!result) {
                     continue;
                 }
@@ -586,10 +642,10 @@ function decorateLanguageService(language, languageService) {
                     result.entries = result.entries.filter(entry => !!entry.sourceDisplay);
                 }
                 for (const entry of result.entries) {
-                    entry.replacementSpan = entry.replacementSpan && (0, transform_1.transformTextSpan)(sourceScript, map, entry.replacementSpan, language_core_1.isCompletionEnabled);
+                    entry.replacementSpan = entry.replacementSpan && (0, transform_1.transformTextSpan)(sourceScript, language, serviceScript, entry.replacementSpan, language_core_1.isCompletionEnabled)?.[1];
                 }
                 result.optionalReplacementSpan = result.optionalReplacementSpan
-                    && (0, transform_1.transformTextSpan)(sourceScript, map, result.optionalReplacementSpan, language_core_1.isCompletionEnabled);
+                    && (0, transform_1.transformTextSpan)(sourceScript, language, serviceScript, result.optionalReplacementSpan, language_core_1.isCompletionEnabled)?.[1];
                 const isAdditional = typeof mapping.data.completion === 'object' && mapping.data.completion.isAdditional;
                 if (isAdditional) {
                     results.push(result);
@@ -614,11 +670,14 @@ function decorateLanguageService(language, languageService) {
     languageService.getCompletionEntryDetails = (filePath, position, entryName, formatOptions, source, preferences, data) => {
         let details;
         const fileName = filePath.replace(windowsPathReg, '/');
-        const [serviceScript, sourceScript, map] = (0, utils_1.getServiceScript)(language, fileName);
+        const [serviceScript, targetScript] = (0, utils_1.getServiceScript)(language, fileName);
+        if (targetScript?.associatedOnly) {
+            return undefined;
+        }
         if (serviceScript) {
-            const generatePosition = (0, transform_1.toGeneratedOffset)(sourceScript, map, position, language_core_1.isCompletionEnabled);
+            const generatePosition = (0, transform_1.toGeneratedOffset)(language, serviceScript, targetScript, position, language_core_1.isCompletionEnabled);
             if (generatePosition !== undefined) {
-                details = getCompletionEntryDetails(fileName, generatePosition, entryName, formatOptions, source, preferences, data);
+                details = getCompletionEntryDetails(targetScript.id, generatePosition, entryName, formatOptions, source, preferences, data);
             }
         }
         else {
@@ -626,17 +685,21 @@ function decorateLanguageService(language, languageService) {
         }
         if (details?.codeActions) {
             for (const codeAction of details.codeActions) {
-                codeAction.changes = codeAction.changes.map(edit => (0, transform_1.transformFileTextChanges)(language, edit, language_core_1.isCompletionEnabled)).filter(utils_1.notEmpty);
+                codeAction.changes = (0, transform_1.transformFileTextChanges)(language, codeAction.changes, language_core_1.isCompletionEnabled);
             }
         }
         return details;
     };
     languageService.provideInlayHints = (filePath, span, preferences) => {
         const fileName = filePath.replace(windowsPathReg, '/');
-        const [serviceScript, sourceScript, map] = (0, utils_1.getServiceScript)(language, fileName);
+        const [serviceScript, targetScript, sourceScript] = (0, utils_1.getServiceScript)(language, fileName);
+        if (targetScript?.associatedOnly) {
+            return [];
+        }
         if (serviceScript) {
             let start;
             let end;
+            const map = language.maps.get(serviceScript.code, targetScript);
             for (const mapping of map.mappings) {
                 if ((0, language_core_1.isInlayHintsEnabled)(mapping.data) && mapping.sourceOffsets[0] >= span.start && mapping.sourceOffsets[0] <= span.start + span.length) {
                     start ??= mapping.generatedOffsets[0];
@@ -649,16 +712,17 @@ function decorateLanguageService(language, languageService) {
                 start = 0;
                 end = 0;
             }
-            start += sourceScript.snapshot.getLength();
-            end += sourceScript.snapshot.getLength();
-            const result = provideInlayHints(fileName, { start, length: end - start }, preferences);
+            const mappingOffset = (0, transform_1.getMappingOffset)(language, serviceScript);
+            start += mappingOffset;
+            end += mappingOffset;
+            const result = provideInlayHints(targetScript.id, { start, length: end - start }, preferences);
             const hints = [];
             for (const hint of result) {
-                const sourcePosition = (0, transform_1.toSourceOffset)(sourceScript, map, hint.position, language_core_1.isInlayHintsEnabled);
+                const sourcePosition = (0, transform_1.toSourceOffset)(sourceScript, language, serviceScript, hint.position, language_core_1.isInlayHintsEnabled);
                 if (sourcePosition !== undefined) {
                     hints.push({
                         ...hint,
-                        position: sourcePosition,
+                        position: sourcePosition[1],
                     });
                 }
             }
@@ -679,12 +743,10 @@ function decorateLanguageService(language, languageService) {
     function linkedCodeFeatureWorker(fileName, position, filter, worker, getLinkedCodes) {
         const results = [];
         const processedFilePositions = new Set();
-        const [serviceScript, sourceScript, map] = (0, utils_1.getServiceScript)(language, fileName);
+        const [serviceScript, targetScript] = (0, utils_1.getServiceScript)(language, fileName);
         if (serviceScript) {
-            for (const [generatedOffset, mapping] of map.getGeneratedOffsets(position)) {
-                if (filter(mapping.data)) {
-                    process(fileName, generatedOffset + sourceScript.snapshot.getLength());
-                }
+            for (const [generatedOffset] of (0, transform_1.toGeneratedOffsets)(language, serviceScript, targetScript, position, filter)) {
+                process(targetScript.id, generatedOffset);
             }
         }
         else {
@@ -696,23 +758,24 @@ function decorateLanguageService(language, languageService) {
                 return;
             }
             processedFilePositions.add(fileName + ':' + position);
-            const result = worker(position);
+            const result = worker(fileName, position);
             if (!result) {
                 return;
             }
             results.push(result);
             for (const ref of getLinkedCodes(result)) {
                 processedFilePositions.add(ref[0] + ':' + ref[1]);
-                const [virtualFile, sourceScript] = (0, utils_1.getServiceScript)(language, ref[0]);
-                if (!virtualFile) {
+                const [serviceScript] = (0, utils_1.getServiceScript)(language, ref[0]);
+                if (!serviceScript) {
                     continue;
                 }
-                const linkedCodeMap = language.linkedCodeMaps.get(virtualFile.code);
+                const linkedCodeMap = language.linkedCodeMaps.get(serviceScript.code);
                 if (!linkedCodeMap) {
                     continue;
                 }
-                for (const linkedCodeOffset of linkedCodeMap.getLinkedOffsets(ref[1] - sourceScript.snapshot.getLength())) {
-                    process(ref[0], linkedCodeOffset + sourceScript.snapshot.getLength());
+                const mappingOffset = (0, transform_1.getMappingOffset)(language, serviceScript);
+                for (const linkedCodeOffset of linkedCodeMap.getLinkedOffsets(ref[1] - mappingOffset)) {
+                    process(ref[0], linkedCodeOffset + mappingOffset);
                 }
             }
         }