refactor(language-core): #5766 for pug

Author: KazariEX

packages/language-core/lib/plugins/vue-template-html.ts

@@ -16,13 +16,13 @@ type Node =
 const shouldAddSuffix = /(?<=<[^>/]+)$/;
 
 const plugin: VueLanguagePlugin = ({ modules }) => {
+	const CompilerDOM = modules['@vue/compiler-dom'];
+
 	return {
 		version: 2.2,
 
 		compileSFCTemplate(lang, template, options) {
 			if (lang === 'html' || lang === 'md') {
-				const compiler = modules['@vue/compiler-dom'];
-
 				let addedSuffix = false;
 
 				// #4583
@@ -31,29 +31,18 @@ const plugin: VueLanguagePlugin = ({ modules }) => {
 					addedSuffix = true;
 				}
 
-				const ast = compiler.parse(template, {
+				const ast = CompilerDOM.parse(template, {
 					...options,
 					comments: true,
 				});
+				CompilerDOM.transform(ast, options);
 
-				const [nodeTransforms, directiveTransforms] = compiler.getBaseTransformPreset();
-				compiler.transform(ast, {
-					...options,
-					nodeTransforms: [
-						...nodeTransforms,
-						...options.nodeTransforms ?? [],
-					],
-					directiveTransforms: {
-						...directiveTransforms,
-						...options.directiveTransforms,
-					},
-				});
-
-				// @ts-expect-error
 				return {
 					ast,
+					code: '',
+					preamble: '',
 					__addedSuffix: addedSuffix,
-				} as CompilerDOM.CodegenResult;
+				};
 			}
 		},
 
@@ -70,8 +59,6 @@ const plugin: VueLanguagePlugin = ({ modules }) => {
 				}
 			}
 
-			const CompilerDOM = modules['@vue/compiler-dom'];
-
 			const lengthDiff = change.newText.length - (change.end - change.start);
 			let hitNodes: Node[] = [];
 

packages/language-core/lib/virtualFile/computedSfc.ts

@@ -1,4 +1,4 @@
-import type * as CompilerDOM from '@vue/compiler-dom';
+import * as CompilerDOM from '@vue/compiler-dom';
 import type { SFCBlock, SFCParseResult } from '@vue/compiler-sfc';
 import { computed, setActiveSub } from 'alien-signals';
 import type * as ts from 'typescript';
@@ -281,10 +281,13 @@ export function computedSfc(
 
 			const errors: CompilerDOM.CompilerError[] = [];
 			const warnings: CompilerDOM.CompilerError[] = [];
+			const [nodeTransforms, directiveTransforms] = CompilerDOM.getBaseTransformPreset();
 			let options: CompilerDOM.CompilerOptions = {
-				onError: (err: CompilerDOM.CompilerError) => errors.push(err),
-				onWarn: (err: CompilerDOM.CompilerError) => warnings.push(err),
+				onError: err => errors.push(err),
+				onWarn: err => warnings.push(err),
 				expressionPlugins: ['typescript'],
+				nodeTransforms,
+				directiveTransforms,
 			};
 
 			for (const plugin of plugins) {

packages/language-plugin-pug/index.ts

@@ -6,6 +6,8 @@ import * as pug from 'volar-service-pug/lib/languageService';
 const classRegex = /^class\s*=/;
 
 const plugin: VueLanguagePlugin = ({ modules }) => {
+	const CompilerDOM = modules['@vue/compiler-dom'];
+
 	return {
 		name: require('./package.json').name,
 
@@ -61,8 +63,7 @@ const plugin: VueLanguagePlugin = ({ modules }) => {
 				}
 
 				const map = new SourceMap(pugFile.mappings);
-				const compiler = modules['@vue/compiler-dom'];
-				const completed = compiler.compile(pugFile.htmlCode, {
+				const ast = CompilerDOM.parse(pugFile.htmlCode, {
 					...options,
 					comments: true,
 					onWarn(warning) {
@@ -79,8 +80,13 @@ const plugin: VueLanguagePlugin = ({ modules }) => {
 						options.onError?.(createProxyObject(error));
 					},
 				});
+				CompilerDOM.transform(ast, options);
 
-				return createProxyObject(completed);
+				return {
+					ast: createProxyObject(ast),
+					code: '',
+					preamble: '',
+				};
 
 				function createProxyObject(target: any) {
 					const proxys = new WeakMap();