From c568348fc9c2e40111143c122969910973e2b5ed Mon Sep 17 00:00:00 2001
From: xiaohui <yanxiaohui@jicesys.com>
Date: Mon, 20 Jun 2022 00:38:05 +0800
Subject: [PATCH] fix: ELECTRON_RENDERER_URL is incorrect

---
 src/server.ts | 4 ++--
 src/utils.ts  | 6 ++++++
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/server.ts b/src/server.ts
index bb2267a..7214636 100644
--- a/src/server.ts
+++ b/src/server.ts
@@ -2,7 +2,7 @@ import { spawn } from 'child_process'
 import { createServer as ViteCreateServer, build as viteBuild, createLogger } from 'vite'
 import colors from 'picocolors'
 import { InlineConfig, resolveConfig } from './config'
-import { ensureElectronEntryFile, getElectronPath } from './utils'
+import { ensureElectronEntryFile, getElectronPath, resolveHostname } from './utils'
 
 export async function createServer(inlineConfig: InlineConfig = {}): Promise<void> {
   const config = await resolveConfig(inlineConfig, 'serve', 'development')
@@ -39,7 +39,7 @@ export async function createServer(inlineConfig: InlineConfig = {}): Promise<voi
       const conf = server.config.server
 
       const protocol = conf.https ? 'https:' : 'http:'
-      const host = conf.host || 'localhost'
+      const host = resolveHostname(conf.host)
       const port = conf.port
       process.env.ELECTRON_RENDERER_URL = `${protocol}//${host}:${port}`
 
diff --git a/src/utils.ts b/src/utils.ts
index c7ad593..93a6ce1 100644
--- a/src/utils.ts
+++ b/src/utils.ts
@@ -37,3 +37,9 @@ export function getElectronPath(): string {
     throw new Error('Electron uninstall')
   }
 }
+
+export const wildcardHosts = new Set(['0.0.0.0', '::', '0000:0000:0000:0000:0000:0000:0000:0000'])
+
+export function resolveHostname(optionsHost: string | boolean | undefined): string {
+  return typeof optionsHost === 'string' && !wildcardHosts.has(optionsHost) ? optionsHost : 'localhost'
+}