From 4779bcf911d3d55fd9806a49ef76d2cc88ff148b Mon Sep 17 00:00:00 2001
From: alex8088 <244096523@qq.com>
Date: Sun, 18 Sep 2022 17:06:06 +0800
Subject: [PATCH] chore: add electron-vite mode node env

---
 src/build.ts  | 1 +
 src/server.ts | 1 +
 2 files changed, 2 insertions(+)

diff --git a/src/build.ts b/src/build.ts
index 0a5b7c0..550e5e9 100644
--- a/src/build.ts
+++ b/src/build.ts
@@ -5,6 +5,7 @@ import { InlineConfig, resolveConfig } from './config'
  * Bundles the electron app for production.
  */
 export async function build(inlineConfig: InlineConfig = {}): Promise<void> {
+  process.env.NODE_ENV_ELECTRON_VITE = 'production'
   const config = await resolveConfig(inlineConfig, 'build', 'production')
   if (config.config) {
     const mainViteConfig = config.config?.main
diff --git a/src/server.ts b/src/server.ts
index 9e324f3..85d4ac0 100644
--- a/src/server.ts
+++ b/src/server.ts
@@ -13,6 +13,7 @@ import { resolveHostname } from './utils'
 import { startElectron } from './electron'
 
 export async function createServer(inlineConfig: InlineConfig = {}): Promise<void> {
+  process.env.NODE_ENV_ELECTRON_VITE = 'development'
   const config = await resolveConfig(inlineConfig, 'serve', 'development')
   if (config.config) {
     const logger = createLogger(inlineConfig.logLevel)