refactor(language-core): simplify bindings iteration

Author: KazariEX

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

@@ -1,3 +1,4 @@
+import { camelize, capitalize } from '@vue/shared';
 import * as path from 'path-browserify';
 import type { Code } from '../../types';
 import { codeFeatures } from '../codeFeatures';
@@ -6,7 +7,6 @@ import { endOfLine, generateSfcBlockSection, newLine } from '../utils';
 import { generateComponentSetupReturns, generateEmitsOption, generatePropsOption } from './component';
 import type { ScriptCodegenContext } from './context';
 import type { ScriptCodegenOptions } from './index';
-import { getTemplateUsageVars } from './template';
 
 export function* generateComponentSelf(
 	options: ScriptCodegenOptions,
@@ -21,27 +21,20 @@ export function* generateComponentSelf(
 			yield* generateComponentSetupReturns(options.scriptSetupRanges);
 		}
 		// bindings
-		const templateUsageVars = getTemplateUsageVars(options, ctx);
-		for (
-			const [content, bindings] of [
-				[options.sfc.scriptSetup.content, options.scriptSetupRanges.bindings] as const,
-				options.sfc.script && options.scriptRanges
-					? [options.sfc.script.content, options.scriptRanges.bindings] as const
-					: ['', []] as const,
-			]
-		) {
-			for (const { range } of bindings) {
-				const varName = content.slice(range.start, range.end);
-				if (!templateUsageVars.has(varName) && !templateCodegenCtx.accessExternalVariables.has(varName)) {
-					continue;
-				}
-
-				const token = Symbol(varName.length);
-				yield ['', undefined, 0, { __linkedToken: token }];
-				yield `${varName}: ${varName} as typeof `;
-				yield ['', undefined, 0, { __linkedToken: token }];
-				yield `${varName},${newLine}`;
+		const templateUsageVars = new Set([
+			...options.sfc.template?.ast?.components.flatMap(c => [camelize(c), capitalize(camelize(c))]) ?? [],
+			...options.templateCodegen?.accessExternalVariables.keys() ?? [],
+			...templateCodegenCtx.accessExternalVariables.keys(),
+		]);
+		for (const varName of ctx.bindingNames) {
+			if (!templateUsageVars.has(varName)) {
+				continue;
 			}
+			const token = Symbol(varName.length);
+			yield ['', undefined, 0, { __linkedToken: token }];
+			yield `${varName}: ${varName} as typeof `;
+			yield ['', undefined, 0, { __linkedToken: token }];
+			yield `${varName},${newLine}`;
 		}
 		yield `}${endOfLine}`; // return {
 		yield `},${newLine}`; // setup() {

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

@@ -1,5 +1,4 @@
 import type { Code } from '../../types';
-import { hyphenateTag } from '../../utils/shared';
 import { codeFeatures } from '../codeFeatures';
 import { generateStyleModules } from '../style/modules';
 import { generateStyleScopedClasses } from '../style/scopedClasses';
@@ -137,27 +136,3 @@ function* generateCssVars(options: ScriptCodegenOptions, ctx: TemplateCodegenCon
 	}
 	yield `// CSS variable injection end ${newLine}`;
 }
-
-export function getTemplateUsageVars(options: ScriptCodegenOptions, ctx: ScriptCodegenContext) {
-	const usageVars = new Set<string>();
-	const components = new Set(options.sfc.template?.ast?.components);
-
-	if (options.templateCodegen) {
-		// fix import components unused report
-		for (const varName of ctx.bindingNames) {
-			if (components.has(varName) || components.has(hyphenateTag(varName))) {
-				usageVars.add(varName);
-			}
-		}
-		for (const component of components) {
-			if (component.includes('.')) {
-				usageVars.add(component.split('.')[0]);
-			}
-		}
-		for (const [varName] of options.templateCodegen.accessExternalVariables) {
-			usageVars.add(varName);
-		}
-	}
-
-	return usageVars;
-}