From cc5e7528ecaa86e163d6727651cb455ed7001b64 Mon Sep 17 00:00:00 2001
From: johnsoncodehk <johnsoncodehk@gmail.com>
Date: Mon, 30 May 2022 20:19:05 +0800
Subject: [PATCH] fix: component export default jsdoc loss when use `<script
 setup>`

close #1327
---
 .../vue-code-gen/src/generators/script.ts     | 21 ++++++++++++++++---
 1 file changed, 18 insertions(+), 3 deletions(-)

diff --git a/packages/vue-code-gen/src/generators/script.ts b/packages/vue-code-gen/src/generators/script.ts
index 0f3e7ffc8..54d4c97ba 100644
--- 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),
 			{