refactor(language-core): transform template code features internally

revert 053aa3c

Author: KazariEX

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

@@ -7,13 +7,11 @@ import { codeFeatures } from '../codeFeatures';
 import type { TemplateCodegenContext } from '../template/context';
 import { endOfLine, generateSfcBlockSection, newLine } from '../utils';
 import { generateComponentSelf } from './componentSelf';
-import { type ScriptCodegenContext } from './context';
+import { createScriptCodegenContext, type ScriptCodegenContext } from './context';
 import { generateScriptSetup, generateScriptSetupImports } from './scriptSetup';
 import { generateSrc } from './src';
 import { generateTemplate } from './template';
 
-export * from './context';
-
 export interface ScriptCodegenOptions {
 	ts: typeof ts;
 	compilerOptions: ts.CompilerOptions;
@@ -28,7 +26,19 @@ export interface ScriptCodegenOptions {
 	templateRefNames: Set<string>;
 }
 
-export function* generateScript(
+export { generate as generateScript };
+
+function generate(options: ScriptCodegenOptions) {
+	const context = createScriptCodegenContext(options);
+	const codegen = generateScript(options, context);
+
+	return {
+		...context,
+		codes: [...codegen],
+	};
+}
+
+function* generateScript(
 	options: ScriptCodegenOptions,
 	ctx: ScriptCodegenContext,
 ): Generator<Code> {

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

@@ -5,13 +5,11 @@ import { getSlotsPropertyName } from '../../utils/shared';
 import { codeFeatures } from '../codeFeatures';
 import { endOfLine, newLine } from '../utils';
 import { wrapWith } from '../utils/wrapWith';
-import type { TemplateCodegenContext } from './context';
+import { createTemplateCodegenContext, type TemplateCodegenContext } from './context';
 import { generateObjectProperty } from './objectProperty';
 import { generateStyleScopedClassReferences } from './styleScopedClasses';
 import { generateTemplateChild, getVForNode } from './templateChild';
 
-export * from './context';
-
 export interface TemplateCodegenOptions {
 	ts: typeof ts;
 	compilerOptions: ts.CompilerOptions;
@@ -28,7 +26,27 @@ export interface TemplateCodegenOptions {
 	selfComponentName?: string;
 }
 
-export function* generateTemplate(
+export { generate as generateTemplate };
+
+function generate(options: TemplateCodegenOptions) {
+	const context = createTemplateCodegenContext(options, options.template.ast);
+	const codegen = generateTemplate(options, context);
+
+	const codes: Code[] = [];
+	for (const code of codegen) {
+		if (typeof code === 'object') {
+			code[3] = context.resolveCodeFeatures(code[3]);
+		}
+		codes.push(code);
+	}
+
+	return {
+		...context,
+		codes,
+	};
+}
+
+function* generateTemplate(
 	options: TemplateCodegenOptions,
 	ctx: TemplateCodegenContext,
 ): Generator<Code> {

packages/language-core/lib/plugins/vue-tsx.ts

@@ -1,12 +1,12 @@
 import { camelize, capitalize } from '@vue/shared';
 import { computed } from 'alien-signals';
 import * as path from 'path-browserify';
-import { createScriptCodegenContext, generateScript, type ScriptCodegenOptions } from '../codegen/script';
-import { createTemplateCodegenContext, generateTemplate, type TemplateCodegenOptions } from '../codegen/template';
+import { generateScript } from '../codegen/script';
+import { generateTemplate } from '../codegen/template';
 import { parseScriptRanges } from '../parsers/scriptRanges';
 import { parseScriptSetupRanges } from '../parsers/scriptSetupRanges';
 import { parseVueCompilerOptions } from '../parsers/vueCompilerOptions';
-import type { Code, Sfc, VueLanguagePlugin } from '../types';
+import type { Sfc, VueLanguagePlugin } from '../types';
 import { computedSet } from '../utils/signals';
 import { CompilerOptionsResolver } from '../utils/ts';
 
@@ -178,8 +178,7 @@ function createTsx(
 		if (getResolvedOptions().skipTemplateCodegen || !sfc.template) {
 			return;
 		}
-
-		const options: TemplateCodegenOptions = {
+		return generateTemplate({
 			ts,
 			compilerOptions: ctx.compilerOptions,
 			vueCompilerOptions: getResolvedOptions(),
@@ -193,26 +192,11 @@ function createTsx(
 			propsAssignName: getSetupPropsAssignName(),
 			inheritAttrs: getSetupInheritAttrs(),
 			selfComponentName: getComponentSelfName(),
-		};
-		const context = createTemplateCodegenContext(options, sfc.template.ast);
-		const codegen = generateTemplate(options, context);
-
-		const codes: Code[] = [];
-		for (const code of codegen) {
-			if (typeof code === 'object') {
-				code[3] = context.resolveCodeFeatures(code[3]);
-			}
-			codes.push(code);
-		}
-
-		return {
-			...context,
-			codes,
-		};
+		});
 	});
 
 	const getGeneratedScript = computed(() => {
-		const options: ScriptCodegenOptions = {
+		return generateScript({
 			ts,
 			compilerOptions: ctx.compilerOptions,
 			vueCompilerOptions: getResolvedOptions(),
@@ -224,14 +208,7 @@ function createTsx(
 			templateCodegen: getGeneratedTemplate(),
 			destructuredPropNames: getSetupDestructuredPropNames(),
 			templateRefNames: getSetupTemplateRefNames(),
-		};
-		const context = createScriptCodegenContext(options);
-		const codegen = generateScript(options, context);
-
-		return {
-			...context,
-			codes: [...codegen],
-		};
+		});
 	});
 
 	return {