feat: support typed template slots for script setup

close #1253
vuejs · Jun 2, 2022 · a629054 · a629054
1 parent 912f69d
commit a629054
Showing 1 changed file with 2 additions and 1 deletion.
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;