feat: support typed template slots for script setup

johnsoncodehk · johnsoncodehk · commit a6290543e5cc · 2022-06-03T06:54:40.000+08:00
close #1253
diff --git a/packages/vue-code-gen/src/generators/script.ts b/packages/vue-code-gen/src/generators/script.ts
@@ -331,7 +331,7 @@ export function generate(
 			codeGen.addText(`{\n`);
 		}
 		else {
-			codeGen.addText(`return (await import('${vueLibName}')).defineComponent({\n`);
+			codeGen.addText(`const __VLS_Component = (await import('${vueLibName}')).defineComponent({\n`);
 		}
 
 		if (scriptSetup && scriptSetupRanges) {
@@ -512,6 +512,7 @@ export function generate(
 		}
 
 		codeGen.addText(`});\n`);
+		codeGen.addText(`return {} as new () => InstanceType<typeof __VLS_Component> & { $slots: typeof import('./${path.basename(fileName)}.__VLS_template').default };\n`);
 		codeGen.addText(`})();`);
 		exportdefaultEnd = codeGen.getText().length;
 

Original file line number	Diff line number	Diff line change
`@@ -331,7 +331,7 @@ export function generate(`
`331`	`331`	codeGen.addText(`{\n`);
`332`	`332`	`}`
`333`	`333`	`else {`
`334`		- codeGen.addText(`return (await import('${vueLibName}')).defineComponent({\n`);
	`334`	+ codeGen.addText(`const __VLS_Component = (await import('${vueLibName}')).defineComponent({\n`);
`335`	`335`	`}`
`336`	`336`
`337`	`337`	`if (scriptSetup && scriptSetupRanges) {`
`@@ -512,6 +512,7 @@ export function generate(`
`512`	`512`	`}`
`513`	`513`
`514`	`514`	codeGen.addText(`});\n`);
	`515`	+ codeGen.addText(`return {} as new () => InstanceType<typeof __VLS_Component> & { $slots: typeof import('./${path.basename(fileName)}.__VLS_template').default };\n`);
`515`	`516`	codeGen.addText(`})();`);
`516`	`517`	`exportdefaultEnd = codeGen.getText().length;`
`517`	`518`