refactor(language-core): re-implement component references by codegen (#5736)

Author: KazariEX

packages/language-core/lib/codegen/script/index.ts

@@ -167,28 +167,36 @@ function* generateExportDefault(options: ScriptCodegenOptions): Generator<Code>
 		return;
 	}
 
-	let prefix: Code;
-	let suffix: Code;
+	const expression = `{} as typeof __VLS_export`;
+
 	if (options.sfc.script && options.scriptRanges?.exportDefault) {
 		const { exportDefault, componentOptions } = options.scriptRanges;
-		prefix = generateSfcBlockSection(
+		yield generateSfcBlockSection(
 			options.sfc.script,
 			exportDefault.start,
 			(componentOptions ?? exportDefault).expression.start,
 			codeFeatures.all,
 		);
-		suffix = generateSfcBlockSection(
+		yield expression;
+		yield generateSfcBlockSection(
 			options.sfc.script,
 			(componentOptions ?? exportDefault).expression.end,
 			options.sfc.script.content.length,
 			codeFeatures.all,
 		);
 	}
 	else {
-		prefix = `export default `;
-		suffix = endOfLine;
+		yield `export `;
+		if (options.sfc.template) {
+			for (let i = 0; i < 'template'.length + 1; i++) {
+				yield [
+					``,
+					'main',
+					options.sfc.template.start + 1 + i,
+					i ? { __combineOffset: i } : codeFeatures.navigationWithoutRename,
+				];
+			}
+		}
+		yield `default ${expression}${endOfLine}`;
 	}
-	yield prefix;
-	yield `{} as typeof __VLS_export`;
-	yield suffix;
 }

packages/typescript-plugin/lib/common.ts

@@ -14,8 +14,6 @@ export function createVueLanguageServiceProxy<T>(
 	const proxyCache = new Map<string | symbol, Function | undefined>();
 	const getProxyMethod = (target: ts.LanguageService, p: string | symbol): Function | undefined => {
 		switch (p) {
-			case 'findReferences':
-				return findReferences(ts, language, languageService, asScriptId, target[p]);
 			case 'getCompletionsAtPosition':
 				return getCompletionsAtPosition(ts, language, asScriptId, vueOptions, target[p]);
 			case 'getCompletionEntryDetails':
@@ -44,56 +42,6 @@ export function createVueLanguageServiceProxy<T>(
 	});
 }
 
-function findReferences<T>(
-	ts: typeof import('typescript'),
-	language: Language<T>,
-	languageService: ts.LanguageService,
-	asScriptId: (fileName: string) => T,
-	findReferences: ts.LanguageService['findReferences'],
-): ts.LanguageService['findReferences'] {
-	return (fileName, position) => {
-		const result = findReferences(fileName, position);
-
-		const sourceScript = language.scripts.get(asScriptId(fileName));
-		const root = sourceScript?.generated?.root;
-		if (!(root instanceof VueVirtualCode)) {
-			return result;
-		}
-
-		const { template } = root.sfc;
-		if (template) {
-			const textSpan = {
-				start: template.start + 1,
-				length: 'template'.length,
-			};
-			if (position >= textSpan.start && position <= textSpan.start + textSpan.length) {
-				result?.push({
-					definition: {
-						fileName,
-						textSpan,
-						kind: ts.ScriptElementKind.scriptElement,
-						name: fileName,
-						containerKind: ts.ScriptElementKind.unknown,
-						containerName: '',
-						displayParts: [],
-					},
-					references: [
-						{
-							fileName,
-							textSpan,
-							isDefinition: true,
-							isWriteAccess: false,
-						},
-						...languageService.getFileReferences(fileName),
-					],
-				});
-			}
-		}
-
-		return result;
-	};
-}
-
 function getCompletionsAtPosition<T>(
 	ts: typeof import('typescript'),
 	language: Language<T>,