From 22d4e13fa547bdcbf8f75cd7c7a8838f0cf20db5 Mon Sep 17 00:00:00 2001
From: EGOIST <0x142857@gmail.com>
Date: Tue, 14 Jul 2020 02:27:23 +0800
Subject: [PATCH] feat(vue): add vue 3 support

---
 core/poi/lib/plugins/config-vue.js | 42 ++++++++++++++++++++++++++----
 1 file changed, 37 insertions(+), 5 deletions(-)

diff --git a/core/poi/lib/plugins/config-vue.js b/core/poi/lib/plugins/config-vue.js
index aaed1a2d..d4835b26 100644
--- a/core/poi/lib/plugins/config-vue.js
+++ b/core/poi/lib/plugins/config-vue.js
@@ -1,5 +1,7 @@
 const path = require('path')
 const resolveFrom = require('resolve-from')
+const logger = require('@poi/logger')
+const PoiError = require('../utils/PoiError')
 
 exports.name = 'builtin:config-vue'
 
@@ -7,14 +9,42 @@ exports.apply = api => {
   api.hook('createWebpackChain', config => {
     const rule = config.module.rule('vue').test(/\.vue$/)
 
+    const vueLoaderPath = path.dirname(
+      api.localResolve('vue-loader/package.json') ||
+        require.resolve('vue-loader/package.json')
+    )
+    const vueLoaderPkg = require(path.join(vueLoaderPath, 'package.json'))
+    const vueLoaderMajorVersion = parseInt(vueLoaderPkg.version, 10)
+    const isVue3 = vueLoaderMajorVersion >= 16
+
+    if (isVue3 && vueLoaderPkg.version.includes('beta')) {
+      logger.warn(`You are using the beta version of vue-loader, be aware!`)
+    }
+
     const getCacheOptions = () => {
-      const vueLoaderPath = path.dirname(require.resolve('vue-loader'))
+      if (isVue3) {
+        const hasSFCCompiler = api.localResolve(
+          '@vue/compiler-sfc/package.json'
+        )
+        if (!hasSFCCompiler) {
+          throw new PoiError(
+            `Expect @vue/compiler-sfc to be installed in current project`
+          )
+        }
+        return api.getCacheConfig('vue-loader', {
+          'vue-loader': vueLoaderPkg.version,
+          '@vue/compiler-sfc': hasSFCCompiler
+            ? api.localRequire('@vue/compiler-sfc/package.json').version
+            : null
+        })
+      }
+
       const compilerPkg = require(resolveFrom(
         vueLoaderPath,
         '@vue/component-compiler-utils/package'
       ))
       return api.getCacheConfig('vue-loader', {
-        'vue-loader': require('vue-loader/package').version,
+        'vue-loader': vueLoaderPkg.version,
         '@vue/component-compiler-utils': compilerPkg.version,
         'vue-template-compiler': api.localResolve(
           'vue-template-compiler/package'
@@ -28,18 +58,20 @@ exports.apply = api => {
 
     rule
       .use('vue-loader')
-      .loader(require.resolve('vue-loader'))
+      .loader(require.resolve(vueLoaderPath))
       .options(
         Object.assign(
           {
             // TODO: error with thread-loader
-            compiler: api.localRequire('vue-template-compiler')
+            compiler: isVue3
+              ? undefined
+              : api.localRequire('vue-template-compiler')
           },
           // For Vue templates
           api.config.cache && getCacheOptions()
         )
       )
 
-    config.plugin('vue').use(require('vue-loader/lib/plugin'))
+    config.plugin('vue').use(require(vueLoaderPath).VueLoaderPlugin)
   })
 }