@volar/language-core 2.3.0-alpha.4 → 2.3.0-alpha.5

package/index.d.ts

@@ -3,9 +3,6 @@ export * from './lib/editorFeatures';
 export * from './lib/linkedCodeMap';
 export * from './lib/types';
 export * from './lib/utils';
-import { SourceMap } from '@volar/source-map';
-import type * as ts from 'typescript';
-import type { CodeInformation, Language, LanguagePlugin, SourceScript, VirtualCode } from './lib/types';
+import type { Language, LanguagePlugin, SourceScript, VirtualCode } from './lib/types';
 export declare function createLanguage<T>(plugins: LanguagePlugin<T>[], scriptRegistry: Map<T, SourceScript<T>>, sync: (id: T) => void): Language<T>;
-export declare function updateVirtualCodeMapOfMap<T>(virtualCode: VirtualCode, mapOfMap: Map<T, [ts.IScriptSnapshot, SourceMap<CodeInformation>]>, getSourceSnapshot: (source: string | undefined) => [T, ts.IScriptSnapshot] | undefined): void;
 export declare function forEachEmbeddedCode(virtualCode: VirtualCode): Generator<VirtualCode>;

package/index.js

@@ -14,7 +14,7 @@ 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 });
-exports.forEachEmbeddedCode = exports.updateVirtualCodeMapOfMap = exports.createLanguage = void 0;
+exports.forEachEmbeddedCode = exports.createLanguage = void 0;
 __exportStar(require("@volar/source-map"), exports);
 __exportStar(require("./lib/editorFeatures"), exports);
 __exportStar(require("./lib/linkedCodeMap"), exports);
@@ -23,12 +23,15 @@ __exportStar(require("./lib/utils"), exports);
 const source_map_1 = require("@volar/source-map");
 const linkedCodeMap_1 = require("./lib/linkedCodeMap");
 function createLanguage(plugins, scriptRegistry, sync) {
-    const virtualCodeToSourceFileMap = new WeakMap();
-    const virtualCodeToMaps = new WeakMap();
+    const virtualCodeToSourceScriptMap = new WeakMap();
+    const virtualCodeToSourceMap = new WeakMap();
     const virtualCodeToLinkedCodeMap = new WeakMap();
     return {
         plugins,
         scripts: {
+            fromVirtualCode(virtualCode) {
+                return virtualCodeToSourceScriptMap.get(virtualCode);
+            },
             get(id) {
                 sync(id);
                 return scriptRegistry.get(id);
@@ -57,12 +60,15 @@ function createLanguage(plugins, scriptRegistry, sync) {
                         // snapshot updated
                         sourceScript.snapshot = snapshot;
                         if (sourceScript.generated) {
-                            const newVirtualCode = sourceScript.generated.languagePlugin.updateVirtualCode?.(id, sourceScript.generated.root, snapshot);
+                            const { updateVirtualCode, createVirtualCode } = sourceScript.generated.languagePlugin;
+                            const newVirtualCode = updateVirtualCode
+                                ? updateVirtualCode(id, sourceScript.generated.root, snapshot)
+                                : createVirtualCode?.(id, languageId, snapshot);
                             if (newVirtualCode) {
                                 sourceScript.generated.root = newVirtualCode;
                                 sourceScript.generated.embeddedCodes.clear();
                                 for (const code of forEachEmbeddedCode(sourceScript.generated.root)) {
-                                    virtualCodeToSourceFileMap.set(code, sourceScript);
+                                    virtualCodeToSourceScriptMap.set(code, sourceScript);
                                     sourceScript.generated.embeddedCodes.set(code.id, code);
                                 }
                                 return sourceScript;
@@ -91,7 +97,7 @@ function createLanguage(plugins, scriptRegistry, sync) {
                                 embeddedCodes: new Map(),
                             };
                             for (const code of forEachEmbeddedCode(virtualCode)) {
-                                virtualCodeToSourceFileMap.set(code, sourceScript);
+                                virtualCodeToSourceScriptMap.set(code, sourceScript);
                                 sourceScript.generated.embeddedCodes.set(code.id, code);
                             }
                             break;
@@ -111,76 +117,39 @@ function createLanguage(plugins, scriptRegistry, sync) {
             },
         },
         maps: {
-            get(virtualCode, scriptId) {
-                if (!scriptId) {
-                    const sourceScript = virtualCodeToSourceFileMap.get(virtualCode);
-                    if (!sourceScript) {
-                        return;
-                    }
-                    scriptId = sourceScript.id;
-                }
-                for (const [id, [_snapshot, map]] of this.forEach(virtualCode)) {
-                    if (id === scriptId) {
-                        return map;
-                    }
-                }
-            },
-            forEach(virtualCode) {
-                let map = virtualCodeToMaps.get(virtualCode.snapshot);
-                if (!map) {
-                    map = new Map();
-                    virtualCodeToMaps.set(virtualCode.snapshot, map);
-                }
-                updateVirtualCodeMapOfMap(virtualCode, map, id => {
-                    if (id) {
+            get(virtualCode) {
+                const sourceScript = virtualCodeToSourceScriptMap.get(virtualCode);
+                let mapCache = virtualCodeToSourceMap.get(virtualCode.snapshot);
+                if (mapCache?.[0] !== sourceScript.snapshot) {
+                    if (virtualCode.mappings.some(mapping => mapping.source)) {
                         throw 'not implemented';
-                        // const sourceScript = sourceScripts.get(id)!;
-                        // return [id, sourceScript.snapshot];
-                    }
-                    else {
-                        const sourceScript = virtualCodeToSourceFileMap.get(virtualCode);
-                        return [sourceScript.id, sourceScript.snapshot];
                     }
-                });
-                return map;
+                    virtualCodeToSourceMap.set(virtualCode.snapshot, mapCache = [
+                        sourceScript.snapshot,
+                        new source_map_1.SourceMap(virtualCode.mappings),
+                    ]);
+                }
+                return mapCache[1];
             },
         },
         linkedCodeMaps: {
             get(virtualCode) {
-                if (!virtualCodeToLinkedCodeMap.has(virtualCode.snapshot)) {
-                    virtualCodeToLinkedCodeMap.set(virtualCode.snapshot, virtualCode.linkedCodeMappings
-                        ? new linkedCodeMap_1.LinkedCodeMap(virtualCode.linkedCodeMappings)
-                        : undefined);
+                const sourceScript = virtualCodeToSourceScriptMap.get(virtualCode);
+                let mapCache = virtualCodeToLinkedCodeMap.get(virtualCode.snapshot);
+                if (mapCache?.[0] !== sourceScript.snapshot) {
+                    virtualCodeToLinkedCodeMap.set(virtualCode.snapshot, mapCache = [
+                        sourceScript.snapshot,
+                        virtualCode.linkedCodeMappings
+                            ? new linkedCodeMap_1.LinkedCodeMap(virtualCode.linkedCodeMappings)
+                            : undefined,
+                    ]);
                 }
-                return virtualCodeToLinkedCodeMap.get(virtualCode.snapshot);
+                return mapCache[1];
             },
         },
     };
 }
 exports.createLanguage = createLanguage;
-function updateVirtualCodeMapOfMap(virtualCode, mapOfMap, getSourceSnapshot) {
-    const sources = new Set();
-    if (!virtualCode.mappings.length) {
-        const source = getSourceSnapshot(undefined);
-        if (source) {
-            mapOfMap.set(source[0], [source[1], new source_map_1.SourceMap([])]);
-        }
-    }
-    for (const mapping of virtualCode.mappings) {
-        if (sources.has(mapping.source)) {
-            continue;
-        }
-        sources.add(mapping.source);
-        const source = getSourceSnapshot(mapping.source);
-        if (!source) {
-            continue;
-        }
-        if (!mapOfMap.has(source[0]) || mapOfMap.get(source[0])[0] !== source[1]) {
-            mapOfMap.set(source[0], [source[1], new source_map_1.SourceMap(virtualCode.mappings.filter(mapping2 => mapping2.source === mapping.source))]);
-        }
-    }
-}
-exports.updateVirtualCodeMapOfMap = updateVirtualCodeMapOfMap;
 function* forEachEmbeddedCode(virtualCode) {
     yield virtualCode;
     if (virtualCode.embeddedCodes) {

package/lib/types.d.ts

@@ -7,10 +7,10 @@ export interface Language<T> {
         get(id: T): SourceScript<T> | undefined;
         set(id: T, snapshot: ts.IScriptSnapshot, languageId?: string, plugins?: LanguagePlugin<T>[]): SourceScript<T> | undefined;
         delete(id: T): void;
+        fromVirtualCode(virtualCode: VirtualCode): SourceScript<T>;
     };
     maps: {
-        get(virtualCode: VirtualCode, scriptId?: T): SourceMap<CodeInformation> | undefined;
-        forEach(virtualCode: VirtualCode): Map<T, [ts.IScriptSnapshot, SourceMap<CodeInformation>]>;
+        get(virtualCode: VirtualCode): SourceMap<CodeInformation>;
     };
     linkedCodeMaps: {
         get(virtualCode: VirtualCode): LinkedCodeMap | undefined;
@@ -22,7 +22,7 @@ export interface Language<T> {
             sync?(): Promise<number>;
         };
         languageServiceHost: ts.LanguageServiceHost;
-        getExtraServiceScript(fileName: string): [SourceScript<T>, TypeScriptExtraServiceScript] | undefined;
+        getExtraServiceScript(fileName: string): TypeScriptExtraServiceScript | undefined;
         asScriptId(fileName: string): T;
         asFileName(scriptId: T): string;
     };
@@ -80,30 +80,37 @@ export interface TypeScriptExtraServiceScript extends TypeScriptServiceScript {
     fileName: string;
 }
 export interface LanguagePlugin<T, K extends VirtualCode = VirtualCode> {
+    /**
+     * For files that are not opened in the IDE, the language ID will not be synchronized to the language server, so a hook is needed to parse the language ID of files that are known extension but not opened in the IDE.
+     */
     getLanguageId(scriptId: T): string | undefined;
+    /**
+     * Generate a virtual code.
+     */
     createVirtualCode?(scriptId: T, languageId: string, snapshot: ts.IScriptSnapshot): K | undefined;
+    /**
+     * Incremental update a virtual code. If not provide, call createVirtualCode again.
+     */
     updateVirtualCode?(scriptId: T, virtualCode: K, newSnapshot: ts.IScriptSnapshot): K | undefined;
+    /**
+     * Cleanup a virtual code.
+     */
     disposeVirtualCode?(scriptId: T, virtualCode: K): void;
-    typescript?: {
-        /**
-         * LSP + TS Plugin
-         */
-        extraFileExtensions: ts.FileExtensionInfo[];
-        /**
-         * LSP + TS Plugin
-         */
-        resolveHiddenExtensions?: boolean;
-        /**
-         * LSP + TS Plugin
-         */
-        getServiceScript(rootVirtualCode: K): TypeScriptServiceScript | undefined;
-        /**
-         * LSP only
-         */
-        getExtraServiceScripts?(fileName: string, rootVirtualCode: K): TypeScriptExtraServiceScript[];
-        /**
-         * LSP only
-         */
-        resolveLanguageServiceHost?(host: ts.LanguageServiceHost): ts.LanguageServiceHost;
-    };
+    typescript?: TypeScriptGenericOptions<K> & TypeScriptNonTSPluginOptions<K>;
+}
+/**
+ * The following options available to all situations.
+ */
+interface TypeScriptGenericOptions<T> {
+    extraFileExtensions: ts.FileExtensionInfo[];
+    resolveHiddenExtensions?: boolean;
+    getServiceScript(rootVirtualCode: T): TypeScriptServiceScript | undefined;
+}
+/**
+ * The following options will not be available in TS plugin.
+ */
+interface TypeScriptNonTSPluginOptions<T> {
+    getExtraServiceScripts?(fileName: string, rootVirtualCode: T): TypeScriptExtraServiceScript[];
+    resolveLanguageServiceHost?(host: ts.LanguageServiceHost): ts.LanguageServiceHost;
 }
+export {};

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@volar/language-core",
-	"version": "2.3.0-alpha.4",
+	"version": "2.3.0-alpha.5",
 	"license": "MIT",
 	"files": [
 		"**/*.js",
@@ -12,7 +12,7 @@
 		"directory": "packages/language-core"
 	},
 	"dependencies": {
-		"@volar/source-map": "2.3.0-alpha.4"
+		"@volar/source-map": "2.3.0-alpha.5"
 	},
-	"gitHead": "8ab3ce02b06a410e9321eea3e9f23d36f97fedda"
+	"gitHead": "c3f6867de39095913ceaa2256ad35a9ca1a90487"
 }