wip

Author: johnsoncodehk

packages/typescript-plugin/index.ts

@@ -1,4 +1,5 @@
-import { transformFileTextChanges } from '@volar/typescript/lib/node/transform.js';
+import { toGeneratedRange, toSourceRanges, transformFileTextChanges } from '@volar/typescript/lib/node/transform.js';
+import { getServiceScript } from '@volar/typescript/lib/node/utils.js';
 import { createLanguageServicePlugin } from '@volar/typescript/lib/quickstart/createLanguageServicePlugin';
 import * as core from '@vue/language-core';
 import type * as ts from 'typescript';
@@ -20,6 +21,42 @@ const projectToOriginalLanguageService = new WeakMap<ts.server.Project, ts.Langu
 
 export = createLanguageServicePlugin(
 	(ts, info) => {
+		let _language: core.Language<string> | undefined;
+		const { getCodeFixesAtPosition } = info.languageService;
+		info.languageService.getCodeFixesAtPosition = (fileName, start, end, errorCodes, formatOptions, preferences) => {
+			let fixes = getCodeFixesAtPosition(fileName, start, end, errorCodes, formatOptions, preferences);
+			if (_language) {
+				const [serviceScript, targetScript, sourceScript] = getServiceScript(_language, fileName);
+				if (serviceScript && sourceScript?.generated?.root instanceof core.VueVirtualCode) {
+					for (
+						const sourceRange of toSourceRanges(sourceScript, _language, serviceScript, start, end, true, () => true)
+					) {
+						const generateRange2 = toGeneratedRange(
+							_language,
+							serviceScript,
+							sourceScript,
+							sourceRange[1],
+							sourceRange[2],
+							(data: core.VueCodeInformation) => typeof data.completion === 'object' && !!data.completion.isAdditional,
+						);
+						if (generateRange2 !== undefined) {
+							let importFixes = getCodeFixesAtPosition(
+								targetScript.id,
+								generateRange2[0],
+								generateRange2[1],
+								[2304], // Cannot find name 'xxx'.ts(2304)
+								formatOptions,
+								preferences,
+							);
+							importFixes = importFixes.filter(fix => fix.fixName === 'import');
+							fixes = fixes.concat(importFixes);
+						}
+					}
+				}
+			}
+			return fixes;
+		};
+
 		projectToOriginalLanguageService.set(info.project, info.languageService);
 
 		const vueOptions = getVueCompilerOptions();
@@ -35,6 +72,7 @@ export = createLanguageServicePlugin(
 		return {
 			languagePlugins: [languagePlugin],
 			setup: language => {
+				_language = language;
 				info.languageService = createVueLanguageServiceProxy(
 					ts,
 					language,