feat(electron): add resolveHostname() resolveEnv()

Author: caoxiemeihao

packages/electron/src/config.ts

@@ -1,9 +1,11 @@
 import fs from 'fs'
 import path from 'path'
+import type { AddressInfo } from 'net'
 import {
   type InlineConfig,
   type ResolvedConfig,
   type Plugin,
+  type ViteDevServer,
   mergeConfig,
   normalizePath,
 } from 'vite'
@@ -150,3 +152,43 @@ checkPkgMain.buildElectronMainPlugin = function buildElectronMainPlugin(runtime:
     },
   }
 }
+
+/**
+ * @see https://github.com/vitejs/vite/blob/c3f6731bafeadd310efa4325cb8dcc639636fe48/packages/vite/src/node/constants.ts#L131-L141
+ */
+export function resolveHostname(hostname: string) {
+  const loopbackHosts = new Set([
+    'localhost',
+    '127.0.0.1',
+    '::1',
+    '0000:0000:0000:0000:0000:0000:0000:0001'
+  ])
+  const wildcardHosts = new Set([
+    '0.0.0.0',
+    '::',
+    '0000:0000:0000:0000:0000:0000:0000:0000'
+  ])
+
+  return loopbackHosts.has(hostname) || wildcardHosts.has(hostname) ? 'localhost' : hostname
+}
+
+export function resolveEnv(server: ViteDevServer) {
+  const addressInfo = server.httpServer.address()
+  const isAddressInfo = (x: any): x is AddressInfo => x?.address
+
+  if (isAddressInfo(addressInfo)) {
+    const { address, port } = addressInfo
+    const host = resolveHostname(address)
+
+    const options = server.config.server
+    const protocol = options.https ? 'https' : 'http'
+    const devBase = server.config.base
+
+    const path = typeof options.open === 'string' ? options.open : devBase
+    const url = path.startsWith('http')
+      ? path
+      : `${protocol}://${host}:${port}${path}`
+
+    return { url, host, port }
+  }
+}