refactor(component-meta): read node directly instead of creating sub ast

Author: KazariEX

Diff for: packages/component-meta/lib/base.ts

@@ -328,9 +328,10 @@ ${commandLine.vueOptions.target < 3 ? vue2TypeHelpersCode : typeHelpersCode}
 
 			// fill defaults
 			const printer = ts.createPrinter(checkerOptions.printer);
-			const snapshot = language.scripts.get(componentPath)?.snapshot!;
+			const sourceScript = language.scripts.get(componentPath)!;
+			const { snapshot } = sourceScript;
 
-			const vueFile = language.scripts.get(componentPath)?.generated?.root;
+			const vueFile = sourceScript.generated?.root;
 			const vueDefaults = vueFile && exportName === 'default'
 				? (vueFile instanceof vue.VueVirtualCode ? readVueComponentDefaultProps(vueFile, printer, ts) : {})
 				: {};
@@ -674,9 +675,9 @@ function createSchemaResolvers(
 	}
 	function getDeclaration(declaration: ts.Declaration): Declaration | undefined {
 		const fileName = declaration.getSourceFile().fileName;
-		const sourceFile = language.scripts.get(fileName);
-		if (sourceFile?.generated) {
-			const script = sourceFile.generated.languagePlugin.typescript?.getServiceScript(sourceFile.generated.root);
+		const sourceScript = language.scripts.get(fileName);
+		if (sourceScript?.generated) {
+			const script = sourceScript.generated.languagePlugin.typescript?.getServiceScript(sourceScript.generated.root);
 			if (script) {
 				for (const [sourceScript, map] of language.maps.forEach(script.code)) {
 					for (const [start] of map.toSourceLocation(declaration.getStart())) {
@@ -711,7 +712,10 @@ function readVueComponentDefaultProps(
 	printer: ts.Printer | undefined,
 	ts: typeof import('typescript')
 ) {
-	let result: Record<string, { default?: string, required?: boolean; }> = {};
+	let result: Record<string, {
+		default?: string;
+		required?: boolean;
+	}> = {};
 
 	scriptSetupWorker();
 	scriptWorker();
@@ -720,16 +724,16 @@ function readVueComponentDefaultProps(
 
 	function scriptSetupWorker() {
 
-		const sfc = root._sfc;
-		const codegen = vue.tsCodegen.get(sfc);
-		const scriptSetupRanges = codegen?.scriptSetupRanges.get();
-
-		if (sfc.scriptSetup && scriptSetupRanges?.withDefaults?.arg) {
+		const ast = root._sfc.scriptSetup?.ast;
+		if (!ast) {
+			return;
+		}
 
-			const defaultsText = sfc.scriptSetup.content.slice(scriptSetupRanges.withDefaults.arg.start, scriptSetupRanges.withDefaults.arg.end);
-			const ast = ts.createSourceFile('/tmp.' + sfc.scriptSetup.lang, '(' + defaultsText + ')', ts.ScriptTarget.Latest);
-			const obj = findObjectLiteralExpression(ast);
+		const codegen = vue.tsCodegen.get(root._sfc);
+		const scriptSetupRanges = codegen?.scriptSetupRanges.get();
 
+		if (scriptSetupRanges?.withDefaults?.argNode) {
+			const obj = findObjectLiteralExpression(scriptSetupRanges.withDefaults.argNode);
 			if (obj) {
 				for (const prop of obj.properties) {
 					if (ts.isPropertyAssignment(prop)) {
@@ -743,19 +747,17 @@ function readVueComponentDefaultProps(
 					}
 				}
 			}
-		} else if (sfc.scriptSetup && scriptSetupRanges?.defineProps?.arg) {
-			const defaultsText = sfc.scriptSetup.content.slice(scriptSetupRanges.defineProps.arg.start, scriptSetupRanges.defineProps.arg.end);
-			const ast = ts.createSourceFile('/tmp.' + sfc.scriptSetup.lang, '(' + defaultsText + ')', ts.ScriptTarget.Latest);
-			const obj = findObjectLiteralExpression(ast);
-
+		}
+		else if (scriptSetupRanges?.defineProps?.argNode) {
+			const obj = findObjectLiteralExpression(scriptSetupRanges.defineProps.argNode);
 			if (obj) {
 				result = {
 					...result,
 					...resolvePropsOption(ast, obj, printer, ts),
 				};
 			}
-		} else if (sfc.scriptSetup && scriptSetupRanges?.defineProps?.destructured) {
-			const ast = sfc.scriptSetup.ast;
+		}
+		else if (scriptSetupRanges?.defineProps?.destructured) {
 			for (const [prop, initializer] of scriptSetupRanges.defineProps.destructured) {
 				if (initializer) {
 					const expText = printer?.printNode(ts.EmitHint.Expression, initializer, ast) ?? initializer.getText(ast);
@@ -780,10 +782,10 @@ function readVueComponentDefaultProps(
 
 	function scriptWorker() {
 
-		const descriptor = root._sfc;
+		const sfc = root._sfc;
 
-		if (descriptor.script) {
-			const scriptResult = readTsComponentDefaultProps(descriptor.script.lang, descriptor.script.content, 'default', printer, ts);
+		if (sfc.script) {
+			const scriptResult = readTsComponentDefaultProps(sfc.script.lang, sfc.script.content, 'default', printer, ts);
 			for (const [key, value] of Object.entries(scriptResult)) {
 				result[key] = value;
 			}

Diff for: packages/language-core/lib/codegen/utils/index.ts

@@ -41,7 +41,11 @@ export function collectVars(
 export function collectIdentifiers(
 	ts: typeof import('typescript'),
 	node: ts.Node,
-	results: Array<{ id: ts.Identifier, isRest: boolean, initializer: ts.Expression | undefined; }> = [],
+	results: {
+		id: ts.Identifier,
+		isRest: boolean,
+		initializer: ts.Expression | undefined;
+	}[] = [],
 	isRest = false,
 	initializer: ts.Expression | undefined = undefined
 ) {

Diff for: packages/language-core/lib/parsers/scriptSetupRanges.ts

@@ -27,9 +27,14 @@ type DefineProps = CallExpressionRange & {
 	destructured?: Map<string, ts.Expression | undefined>;
 	destructuredRest?: string;
 	statement: TextRange;
+	// used by component-meta
+	argNode?: ts.Expression;
 };
 
-type WithDefaults = Pick<CallExpressionRange, 'callExp' | 'exp' | 'arg'>;
+type WithDefaults = Omit<CallExpressionRange, 'typeArg'> & {
+	// used by component-meta
+	argNode?: ts.Expression;
+};
 
 type DefineEmits = CallExpressionRange & {
 	name?: string;
@@ -281,7 +286,8 @@ export function parseScriptSetupRanges(
 			else if (vueCompilerOptions.macros.defineProps.includes(callText)) {
 				defineProps = {
 					...parseCallExpression(node),
-					statement: getStatementRange(ts, parents, node, ast)
+					statement: getStatementRange(ts, parents, node, ast),
+					argNode: node.arguments[0]
 				};
 				if (ts.isVariableDeclaration(parent)) {
 					if (ts.isObjectBindingPattern(parent.name)) {
@@ -316,7 +322,8 @@ export function parseScriptSetupRanges(
 				withDefaults = {
 					callExp: _getStartEnd(node),
 					exp: _getStartEnd(node.expression),
-					arg: arg ? _getStartEnd(arg) : undefined
+					arg: arg ? _getStartEnd(arg) : undefined,
+					argNode: arg
 				};
 			}
 			else if (vueCompilerOptions.macros.defineEmits.includes(callText)) {