From a96991ac38c1a558234498b6430ddf43311da28d 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: Tue, 9 Nov 2021 08:59:37 +0800
Subject: [PATCH] refactor: improve ws code

---
 scripts/ws.ts        | 24 +++++++++++++++---------
 src/preload/index.ts |  5 +++--
 src/preload/utils.ts | 34 ----------------------------------
 src/preload/ws.ts    | 37 +++++++++++++++++++++++++++++++++++++
 4 files changed, 55 insertions(+), 45 deletions(-)
 create mode 100644 src/preload/ws.ts

diff --git a/scripts/ws.ts b/scripts/ws.ts
index 520a81d..f5dfef4 100644
--- a/scripts/ws.ts
+++ b/scripts/ws.ts
@@ -1,27 +1,33 @@
 /**
+ * Ws server side
  * Hot reload from preload script during development
  */
 import WebSocket from 'ws'
 import chalk from 'chalk'
 import pkg from '../package.json'
 
-export interface CreateWsServerOptions {
-  TAG: string
+
+export interface CreateWsServerOptions { }
+
+export interface WssServer {
+  wss: WebSocket.Server
+  instance: WebSocket | null
 }
 
-export function createWsServer(options: CreateWsServerOptions) {
-  const { TAG } = options
+const TAG = '[ws.ts]'
+
+export function createWsServer(options: CreateWsServerOptions = {}): WssServer {
   const port = pkg.env.PORT_WS
-  const host = '127.0.0.1'
+  const host = pkg.env.HOST // '127.0.0.1'
   const wss = new WebSocket.Server({ host, port })
-  const wssObj: { wss: WebSocket.Server; instance: WebSocket | null } = { wss, instance: null }
+  const wssInstance: WssServer = { wss, instance: null }
 
   console.log(TAG, 'Wss run at - ' + chalk.yellow(`ws://${host}:${port}`))
 
   wss.on('connection', ws => {
     console.log(TAG, chalk.yellow(`wss.on('connection')`))
 
-    wssObj.instance = ws
+    wssInstance.instance = ws
     ws.on('message', message => {
       console.log(TAG, `ws.on('message'):`, message.toString())
     })
@@ -32,10 +38,10 @@ export function createWsServer(options: CreateWsServerOptions) {
   wss.on('close', () => {
     console.log(TAG, chalk.gray(`wss.on('close')`))
 
-    wssObj.instance = null
+    wssInstance.instance = null
   })
 
-  return wssObj
+  return wssInstance
 }
 
 export function formatWsSendData(json: { cmd: string, data?: any }) {
diff --git a/src/preload/index.ts b/src/preload/index.ts
index 05d9c6a..0096f8e 100644
--- a/src/preload/index.ts
+++ b/src/preload/index.ts
@@ -1,6 +1,7 @@
 import fs from 'fs'
 import { contextBridge, ipcRenderer } from 'electron'
-import { domReady, injectWsCode } from './utils'
+import { domReady } from './utils'
+import { injectWsCode } from './ws'
 import { useLoading } from './loading'
 
 const isDev = process.env.NODE_ENV === 'development'
@@ -9,7 +10,7 @@ const { removeLoading, appendLoading } = useLoading()
 domReady().then(() => {
   appendLoading()
   isDev && injectWsCode({
-    host: '127.0.0.1',
+    host: process.env.HOST, // '127.0.0.1'
     port: process.env.PORT_WS as string, // process.env.npm_package_env_PORT_WS
   })
 })
diff --git a/src/preload/utils.ts b/src/preload/utils.ts
index 5405abf..7dd3280 100644
--- a/src/preload/utils.ts
+++ b/src/preload/utils.ts
@@ -13,37 +13,3 @@ export function domReady(condition: DocumentReadyState[] = ['complete', 'interac
     }
   })
 }
-
-/** Inject ws related code */
-export function injectWsCode(options: {
-  host: string
-  port: string | number
-}) {
-  const oScript = document.createElement('script')
-  oScript.id = 'ws-preload-hot-reload'
-
-  oScript.innerHTML = `
-${__ws_hot_reload_for_preload.toString()}
-${__ws_hot_reload_for_preload.name}(${JSON.stringify(options)})
-`
-
-  document.body.appendChild(oScript)
-}
-
-function __ws_hot_reload_for_preload(options: { host: string; port: string | number }) {
-  const ws = new WebSocket(`ws://${options.host}:${options.port}`)
-  ws.onmessage = function (ev) {
-    try {
-      console.log('[preload] ws.onmessage:', ev.data)
-
-      const data = JSON.parse(ev.data) // { "cmd": "string", data: "string|number" }
-
-      if (data.cmd === 'reload') {
-        setTimeout(() => window.location.reload(), 999)
-      }
-    } catch (error) {
-      console.warn(`ws.onmessage should be accept "JSON.string" formatted string.`)
-      console.error(error)
-    }
-  }
-}
diff --git a/src/preload/ws.ts b/src/preload/ws.ts
new file mode 100644
index 0000000..2f817d5
--- /dev/null
+++ b/src/preload/ws.ts
@@ -0,0 +1,37 @@
+/**
+ * Ws client side
+ */
+
+/** Inject ws client-side code */
+export function injectWsCode(options: {
+  host: string
+  port: string | number
+}) {
+  const oScript = document.createElement('script')
+  oScript.id = 'ws-preload-hot-reload'
+
+  oScript.innerHTML = `
+${__ws_hot_reload_for_preload.toString()}
+${__ws_hot_reload_for_preload.name}(${JSON.stringify(options)})
+`
+
+  document.body.appendChild(oScript)
+}
+
+function __ws_hot_reload_for_preload(options: { host: string; port: string | number }) {
+  const ws = new WebSocket(`ws://${options.host}:${options.port}`)
+  ws.onmessage = function (ev) {
+    try {
+      console.log('[preload] ws.onmessage:', ev.data)
+
+      const data = JSON.parse(ev.data) // { "cmd": "string", data: "string|number" }
+
+      if (data.cmd === 'reload') {
+        setTimeout(() => window.location.reload(), 999)
+      }
+    } catch (error) {
+      console.warn(`ws.onmessage should be accept "JSON.string" formatted string.`)
+      console.error(error)
+    }
+  }
+}