From ea5bafaefbafd858389f88e537cb3473b4669802 Mon Sep 17 00:00:00 2001
From: underfin <likui6666666@gmail.com>
Date: Tue, 1 Jun 2021 08:00:25 +0800
Subject: [PATCH] fix(plugin-vue): rewrite default after ts compiled (#3591)

Co-authored-by: Shinigami <chrissi92@hotmail.de>
---
 packages/playground/vue/CustomBlock.vue |  5 +++--
 packages/plugin-vue/package.json        |  2 +-
 packages/plugin-vue/src/main.ts         | 18 ++----------------
 3 files changed, 6 insertions(+), 19 deletions(-)

diff --git a/packages/playground/vue/CustomBlock.vue b/packages/playground/vue/CustomBlock.vue
index cc49cd07242fdd..aedf7f09b4bbc0 100644
--- a/packages/playground/vue/CustomBlock.vue
+++ b/packages/playground/vue/CustomBlock.vue
@@ -3,12 +3,12 @@
   <p class="custom-block">{{ t('hello') }}</p>
 </template>
 
-<script>
+<script lang="ts">
 import { getCurrentInstance } from 'vue'
 
 function useI18n(locale = 'en') {
   const instance = getCurrentInstance()
-  const resources = instance.type.i18n || { en: {} }
+  const resources = (instance.type as any).i18n || { en: {} }
   function t(key) {
     const res = resources[locale] || {}
     return res[key]
@@ -16,6 +16,7 @@ function useI18n(locale = 'en') {
   return { t }
 }
 
+// export default
 export default {
   setup() {
     return { ...useI18n('ja') }
diff --git a/packages/plugin-vue/package.json b/packages/plugin-vue/package.json
index 0e47c6342b249a..07d8e1e031708d 100644
--- a/packages/plugin-vue/package.json
+++ b/packages/plugin-vue/package.json
@@ -29,7 +29,7 @@
   },
   "homepage": "https://github.com/vitejs/vite/tree/main/packages/plugin-vue#readme",
   "peerDependencies": {
-    "@vue/compiler-sfc": "^3.0.6"
+    "@vue/compiler-sfc": "^3.0.8"
   },
   "devDependencies": {
     "@rollup/pluginutils": "^4.1.0",
diff --git a/packages/plugin-vue/src/main.ts b/packages/plugin-vue/src/main.ts
index daa330e85b84ca..98bca03db6f081 100644
--- a/packages/plugin-vue/src/main.ts
+++ b/packages/plugin-vue/src/main.ts
@@ -225,9 +225,6 @@ async function genTemplateCode(
   }
 }
 
-const exportDefaultClassRE =
-  /(?:(?:^|\n|;)\s*)export\s+default\s+class\s+([\w$]+)/
-
 async function genScriptCode(
   descriptor: SFCDescriptor,
   options: ResolvedOptions,
@@ -247,19 +244,7 @@ async function genScriptCode(
       (!script.lang || (script.lang === 'ts' && options.devServer)) &&
       !script.src
     ) {
-      // TODO remove the class check logic after upgrading @vue/compiler-sfc
-      const classMatch = script.content.match(exportDefaultClassRE)
-      if (classMatch) {
-        scriptCode =
-          script.content.replace(exportDefaultClassRE, `\nclass $1`) +
-          `\nconst _sfc_main = ${classMatch[1]}`
-        if (/export\s+default/.test(scriptCode)) {
-          // fallback if there are still export default
-          scriptCode = rewriteDefault(script.content, `_sfc_main`)
-        }
-      } else {
-        scriptCode = rewriteDefault(script.content, `_sfc_main`)
-      }
+      scriptCode = script.content
       map = script.map
       if (script.lang === 'ts') {
         const result = await options.devServer!.transformWithEsbuild(
@@ -271,6 +256,7 @@ async function genScriptCode(
         scriptCode = result.code
         map = result.map
       }
+      scriptCode = rewriteDefault(scriptCode, `_sfc_main`)
     } else {
       if (script.src) {
         await linkSrcToDescriptor(script.src, descriptor, pluginContext)