From 647edd7207410d7700c34026ca352bb731681716 Mon Sep 17 00:00:00 2001
From: Hooray Hu <304327508@qq.com>
Date: Wed, 3 Jul 2024 23:01:46 +0800
Subject: [PATCH] =?UTF-8?q?refactor:=20=E7=8E=AF=E5=A2=83=E5=8F=98?=
 =?UTF-8?q?=E9=87=8F=20`VITE=5FAPP=5FDEBUG=5FTOOL`=20=E6=94=B9=E6=88=90?=
 =?UTF-8?q?=E7=94=A8=20vite=20=E6=8F=92=E4=BB=B6=E7=9A=84=E6=96=B9?=
 =?UTF-8?q?=E5=BC=8F=E6=B3=A8=E5=85=A5=E4=BB=A3=E7=A0=81=E4=B8=AD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/App.vue                |  4 ----
 vite/plugins/debug-tool.ts | 28 ++++++++++++++++++++++++++++
 vite/plugins/index.ts      |  2 ++
 3 files changed, 30 insertions(+), 4 deletions(-)
 create mode 100644 vite/plugins/debug-tool.ts

diff --git a/src/App.vue b/src/App.vue
index 8a4d32d9..b665ee0b 100755
--- a/src/App.vue
+++ b/src/App.vue
@@ -1,6 +1,4 @@
 <script setup lang="ts">
-import eruda from 'eruda'
-import VConsole from 'vconsole'
 import DisableDevtool from 'disable-devtool'
 import hotkeys from 'hotkeys-js'
 import eventBus from './utils/eventBus'
@@ -68,8 +66,6 @@ onMounted(() => {
   })
 })
 
-import.meta.env.VITE_APP_DEBUG_TOOL === 'eruda' && eruda.init()
-import.meta.env.VITE_APP_DEBUG_TOOL === 'vconsole' && new VConsole()
 import.meta.env.VITE_APP_DISABLE_DEVTOOL === 'true' && DisableDevtool()
 </script>
 
diff --git a/vite/plugins/debug-tool.ts b/vite/plugins/debug-tool.ts
new file mode 100644
index 00000000..aa41c52a
--- /dev/null
+++ b/vite/plugins/debug-tool.ts
@@ -0,0 +1,28 @@
+import type { Plugin } from 'vite'
+
+export default function createDebugTool(env): Plugin {
+  const { VITE_APP_DEBUG_TOOL } = env
+  return {
+    name: 'debug-plugin',
+    transform: (code, id) => {
+      if (/src\/main.ts$/.test(id)) {
+        if (VITE_APP_DEBUG_TOOL === 'eruda') {
+          code = code.concat(`
+            import eruda from 'eruda'
+            eruda.init()
+          `)
+        }
+        else if (VITE_APP_DEBUG_TOOL === 'vconsole') {
+          code = code.concat(`
+            import VConsole from 'vconsole'
+            new VConsole()
+          `)
+        }
+        return {
+          code,
+          map: null,
+        }
+      }
+    },
+  }
+}
diff --git a/vite/plugins/index.ts b/vite/plugins/index.ts
index a4c0ab42..dcec1497 100755
--- a/vite/plugins/index.ts
+++ b/vite/plugins/index.ts
@@ -16,6 +16,7 @@ import createCompression from './compression'
 import createArchiver from './archiver'
 import createConsole from './console'
 import createBanner from './banner'
+import createDebugTool from './debug-tool'
 
 export default function createVitePlugins(viteEnv, isBuild = false) {
   const vitePlugins: (PluginOption | PluginOption[])[] = [
@@ -42,5 +43,6 @@ export default function createVitePlugins(viteEnv, isBuild = false) {
   vitePlugins.push(createArchiver(viteEnv))
   vitePlugins.push(createConsole())
   vitePlugins.push(createBanner())
+  vitePlugins.push(createDebugTool(viteEnv))
   return vitePlugins
 }