fix: component export default jsdoc loss when use <script setup>

close #1327
vuejs · May 30, 2022 · cc5e752 · cc5e752
1 parent 71f8837
commit cc5e752
Showing 1 changed file with 18 additions and 3 deletions.
diff --git a/packages/vue-code-gen/src/generators/script.ts b/packages/vue-code-gen/src/generators/script.ts
@@ -39,6 +39,7 @@ export function generate(
 	}
 
 	writeScriptSrc();
+	writeScriptSetupImports();
 	writeScriptBeforeExportDefault();
 	writeScriptSetup();
 	writeScriptSetupTypes();
@@ -167,7 +168,7 @@ export function generate(
 			return;
 
 		if (!!scriptSetup && scriptRanges?.exportDefault) {
-			addVirtualCode('script', 0, scriptRanges.exportDefault.start);
+			addVirtualCode('script', 0, scriptRanges.exportDefault.expression.start);
 		}
 		else {
 			let isExportRawObject = false;
@@ -228,7 +229,7 @@ export function generate(
 			},
 		);
 	}
-	function writeScriptSetup() {
+	function writeScriptSetupImports() {
 
 		if (!scriptSetup)
 			return;
@@ -256,8 +257,22 @@ export function generate(
 				},
 			},
 		);
+	}
+	function writeScriptSetup() {
+
+		if (!scriptSetup)
+			return;
+
+		if (!scriptSetupRanges)
+			return;
+
+		if (scriptRanges?.exportDefault) {
+			codeGen.addText('await (async () => {\n');
+		}
+		else {
+			codeGen.addText('export default await (async () => {\n');
+		}
 
-		codeGen.addText('export default await (async () => {\n');
 		codeGen.addCode(
 			scriptSetup.content.substring(scriptSetupRanges.importSectionEndOffset),
 			{