From c8af08d7ede410132a2eb699f688ae58b3269dd0 Mon Sep 17 00:00:00 2001
From: daiwei <daiwei521@126.com>
Date: Mon, 6 Sep 2021 09:01:39 +0800
Subject: [PATCH] chore: improve code

---
 .../__snapshots__/compileScript.spec.ts.snap     | 15 ++++++++-------
 .../compiler-sfc/__tests__/compileScript.spec.ts | 16 +++++++++-------
 packages/compiler-sfc/src/compileScript.ts       | 13 +++++++++----
 3 files changed, 26 insertions(+), 18 deletions(-)

diff --git a/packages/compiler-sfc/__tests__/__snapshots__/compileScript.spec.ts.snap b/packages/compiler-sfc/__tests__/__snapshots__/compileScript.spec.ts.snap
index 009a16a8851..befdf2f2797 100644
--- a/packages/compiler-sfc/__tests__/__snapshots__/compileScript.spec.ts.snap
+++ b/packages/compiler-sfc/__tests__/__snapshots__/compileScript.spec.ts.snap
@@ -913,12 +913,13 @@ return {  }
 
 exports[`SFC compile <script setup> with TypeScript defineProps w/ extends interface 1`] = `
 "import { defineComponent as _defineComponent } from 'vue'
-interface Foo { x?: number }
-      interface Bar extends Foo { y?: number }
-      interface Props extends Bar {
-        z: number
-        y: string
-      }
+interface Bar extends Foo { y?: number }
+        interface Props extends Bar {
+          z: number
+          y: string
+        }
+        
+        interface Foo { x?: number }
       
 export default /*#__PURE__*/_defineComponent({
   props: {
@@ -929,7 +930,7 @@ export default /*#__PURE__*/_defineComponent({
   setup(__props: any, { expose }) {
   expose()
 
-      
+        
       
 return {  }
 }
diff --git a/packages/compiler-sfc/__tests__/compileScript.spec.ts b/packages/compiler-sfc/__tests__/compileScript.spec.ts
index aeb544e8a29..32a085ae163 100644
--- a/packages/compiler-sfc/__tests__/compileScript.spec.ts
+++ b/packages/compiler-sfc/__tests__/compileScript.spec.ts
@@ -775,14 +775,16 @@ const emit = defineEmits(['a', 'b'])
 
     test('defineProps w/ extends interface', () => {
       const { content, bindings } = compile(`
+      <script lang="ts">
+        interface Foo { x?: number }
+      </script>
       <script setup lang="ts">
-      interface Foo { x?: number }
-      interface Bar extends Foo { y?: number }
-      interface Props extends Bar {
-        z: number
-        y: string
-      }
-      defineProps<Props>()
+        interface Bar extends Foo { y?: number }
+        interface Props extends Bar {
+          z: number
+          y: string
+        }
+        defineProps<Props>()
       </script>
       `)
       assertCode(content)
diff --git a/packages/compiler-sfc/src/compileScript.ts b/packages/compiler-sfc/src/compileScript.ts
index e57c37bfa74..fe7f2222d89 100644
--- a/packages/compiler-sfc/src/compileScript.ts
+++ b/packages/compiler-sfc/src/compileScript.ts
@@ -440,6 +440,12 @@ export function compileScript(
     return true
   }
 
+  function getAstBody(): Statement[] {
+    return scriptAst
+      ? [...scriptSetupAst.body, ...scriptAst.body]
+      : scriptSetupAst.body
+  }
+
   function resolveExtendsType(
     node: Node,
     qualifier: (node: Node) => boolean,
@@ -451,7 +457,8 @@ export function compileScript(
           extend.type === 'TSExpressionWithTypeArguments' &&
           extend.expression.type === 'Identifier'
         ) {
-          for (const node of scriptSetupAst.body) {
+          const body = getAstBody()
+          for (const node of body) {
             const qualified = isQualifiedType(
               node,
               qualifier,
@@ -521,9 +528,7 @@ export function compileScript(
       node.typeName.type === 'Identifier'
     ) {
       const refName = node.typeName.name
-      const body = scriptAst
-        ? [...scriptSetupAst.body, ...scriptAst.body]
-        : scriptSetupAst.body
+      const body = getAstBody()
       for (const node of body) {
         let qualified = isQualifiedType(
           node,