From c711938f70d59b265481b3e05826b383b6995cab Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=8D=89=E9=9E=8B=E6=B2=A1=E5=8F=B7?= <308487730@qq.com>
Date: Fri, 4 Nov 2022 09:59:03 +0800
Subject: [PATCH] feat: Allow use `import.meta.env.VITE_SOME_KEY` in
 Electron-Main

---
 vite.config.ts | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

diff --git a/vite.config.ts b/vite.config.ts
index 98de3a7..5b50ed4 100644
--- a/vite.config.ts
+++ b/vite.config.ts
@@ -3,7 +3,7 @@ import path from 'path'
 import { defineConfig } from 'vite'
 import react from '@vitejs/plugin-react'
 import electron from 'vite-electron-plugin'
-import { customStart } from 'vite-electron-plugin/plugin'
+import { customStart, loadViteEnv } from 'vite-electron-plugin/plugin'
 import renderer from 'vite-plugin-electron-renderer'
 import pkg from './package.json'
 
@@ -27,10 +27,16 @@ export default defineConfig({
       transformOptions: {
         sourcemap: !!process.env.VSCODE_DEBUG,
       },
-      // Will start Electron via VSCode Debug
-      plugins: process.env.VSCODE_DEBUG
-        ? [customStart(debounce(() => console.log(/* For `.vscode/.debug.script.mjs` */'[startup] Electron App')))]
-        : undefined,
+      plugins: [
+        ...(process.env.VSCODE_DEBUG
+          ? [
+            // Will start Electron via VSCode Debug
+            customStart(debounce(() => console.log(/* For `.vscode/.debug.script.mjs` */'[startup] Electron App'))),
+          ]
+          : []),
+        // Allow use `import.meta.env.VITE_SOME_KEY` in Electron-Main
+        loadViteEnv(),
+      ],
     }),
     renderer({
       nodeIntegration: true,