feat(language-core): add option to resolve hidden extensions (#190)

volarjs · May 31, 2024 · 3231202 · 3231202
1 parent a2ca6e0
commit 3231202
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git a/packages/language-core/lib/types.ts b/packages/language-core/lib/types.ts
@@ -97,6 +97,10 @@ export interface LanguagePlugin<T, K extends VirtualCode = VirtualCode> {
 		 * LSP + TS Plugin
 		 */
 		extraFileExtensions: ts.FileExtensionInfo[];
+		/**
+		 * LSP + TS Plugin
+		 */
+		resolveHiddenExtensions?: boolean;
 		/**
 		 * LSP + TS Plugin
 		 */

diff --git a/packages/typescript/lib/resolveModuleName.ts b/packages/typescript/lib/resolveModuleName.ts
@@ -52,6 +52,24 @@ export function createResolveModuleName<T>(
 						}
 					}
 				}
+				if (typescript.resolveHiddenExtensions && fileName.endsWith(`.d.ts`)) {
+					for (const { extension } of typescript.extraFileExtensions) {
+						const sourceFileName = fileName.slice(0, -`.d.ts`.length) + `.${extension}`;
+						if (fileExists(sourceFileName)) {
+							const sourceScript = getSourceScript(sourceFileName);
+							if (sourceScript?.generated) {
+								const serviceScript = sourceScript.generated.languagePlugin.typescript?.getServiceScript(sourceScript.generated.root);
+								if (serviceScript) {
+									toSourceFileInfo.set(fileName, {
+										sourceFileName,
+										extension: serviceScript.extension,
+									});
+									return true;
+								}
+							}
+						}
+					}
+				}
 			}
 			return host.fileExists(fileName);
 		},