Fixed the issue that the TCP connection was interrupted when setting a proxy

app/electron/main.js

@@ -752,7 +752,7 @@ app.whenReady().then(() => {
             event.sender.send("siyuan-event", "leave-full-screen");
         });
     });
-    ipcMain.on("siyuan-cmd", (event, data) => {
+    ipcMain.on("siyuan-cmd", async (event, data) => {
         let cmd = data;
         let webContentsId = event.sender.id;
         if (typeof data !== "string") {
@@ -820,17 +820,29 @@ app.whenReady().then(() => {
                 }
                 break;
             case "setProxy":
-                event.sender.session.closeAllConnections().then(() => {
+                try {
+                    let proxy, log;
                     if (data.proxyURL.startsWith("://")) {
-                        event.sender.session.setProxy({mode: "system"}).then(() => {
-                            console.log("network proxy [system]");
-                        });
-                        return;
+                        proxy = {mode: "system"};
+                        log = "network proxy [system]";
+                    } else {
+                        proxy = {proxyRules: data.proxyURL};
+                        log = `network proxy [${data.proxyURL}]`;
                     }
-                    event.sender.session.setProxy({proxyRules: data.proxyURL}).then(() => {
-                        console.log("network proxy [" + data.proxyURL + "]");
+                    await event.sender.session.closeAllConnections();
+                    await event.sender.session.setProxy(proxy);
+                    console.log(log);
+                    event.reply("siyuan-proxy-reply", {
+                        state: "fulfilled",
+                        proxy,
                     });
-                });
+                } catch (error) {
+                    event.reply("siyuan-proxy-reply", {
+                        state: "rejected",
+                        proxy,
+                        error,
+                    });
+                }
                 break;
         }
     });

app/src/boot/onGetConfig.ts

@@ -22,7 +22,6 @@ import {showMessage} from "../dialog/message";
 import {replaceLocalPath} from "../editor/rename";
 import {setTabPosition} from "../window/setHeader";
 import {initBar} from "../layout/topBar";
-import {setProxy} from "../config/util/about";
 import {openChangelog} from "./openChangelog";
 import {getIdFromSYProtocol, isSYProtocol} from "../util/pathName";
 import {App} from "../index";
@@ -135,7 +134,6 @@ export const onGetConfig = (isStart: boolean, app: App) => {
         }
     });
     initBar(app);
-    setProxy();
     initStatus();
     initWindow(app);
     appearance.onSetappearance(window.siyuan.config.appearance);

app/src/config/util/about.ts

@@ -8,9 +8,28 @@ import {Constants} from "../../constants";
 
 export const setProxy = () => {
     /// #if !BROWSER
-    ipcRenderer.send(Constants.SIYUAN_CMD, {
-        cmd: "setProxy",
-        proxyURL: `${window.siyuan.config.system.networkProxy.scheme}://${window.siyuan.config.system.networkProxy.host}:${window.siyuan.config.system.networkProxy.port}`
+    return new Promise((resolve, reject) => {
+        ipcRenderer.once(Constants.SIYUAN_PROXY_REPLY, (event, data: {
+            state: "fulfilled" | "rejected",
+            proxy: {mode: "system"} | {proxyRules: string},
+            error?: any,
+        }) => {
+            switch (data.state) {
+                case "fulfilled":
+                    resolve(data.proxy);
+                    break;
+                case "rejected":
+                    reject(data.error);
+                    break;
+                default:
+                    reject(new Error(`Unknown state: ${data.state}`));
+                    break;
+            }
+        });
+        ipcRenderer.send(Constants.SIYUAN_CMD, {
+            cmd: "setProxy",
+            proxyURL: `${window.siyuan.config.system.networkProxy.scheme}://${window.siyuan.config.system.networkProxy.host}:${window.siyuan.config.system.networkProxy.port}`
+        });
     });
     /// #endif
 };

app/src/constants.ts

@@ -28,6 +28,8 @@ export abstract class Constants {
     public static readonly SIYUAN_GET: string = "siyuan-get";
     public static readonly SIYUAN_EVENT: string = "siyuan-event";
 
+    public static readonly SIYUAN_PROXY_REPLY: string = "siyuan-proxy-reply";
+
     public static readonly SIYUAN_CONFIG_TRAY: string = "siyuan-config-tray";
     public static readonly SIYUAN_QUIT: string = "siyuan-quit";
     public static readonly SIYUAN_HOTKEY: string = "siyuan-hotkey";

app/src/index.ts

@@ -4,6 +4,7 @@ import {Model} from "./layout/Model";
 import {onGetConfig} from "./boot/onGetConfig";
 import {initBlockPopover} from "./block/popover";
 import {account} from "./config/account";
+import {setProxy} from "./config/util/about";
 import {addScript, addScriptSync} from "./protyle/util/addScript";
 import {genUUID} from "./util/genID";
 import {fetchGet, fetchPost} from "./util/fetch";
@@ -33,11 +34,6 @@ export class App {
     public appId: string;
 
     constructor() {
-        /// #if BROWSER
-        registerServiceWorker(`${Constants.SERVICE_WORKER_PATH}?v=${Constants.SIYUAN_VERSION}`);
-        /// #endif
-        addScriptSync(`${Constants.PROTYLE_CDN}/js/lute/lute.min.js?v=${Constants.SIYUAN_VERSION}`, "protyleLuteScript");
-        addScript(`${Constants.PROTYLE_CDN}/js/protyle-html.js?v=${Constants.SIYUAN_VERSION}`, "protyleWcHtmlScript");
         addBaseURL();
 
         this.appId = Constants.SIYUAN_APPID;
@@ -161,6 +157,18 @@ export class App {
                     data: response.data.conf.uiLayout.bottom
                 };
             }
+
+            await setProxy();
+
+            /// #if BROWSER
+            await registerServiceWorker(`${Constants.SERVICE_WORKER_PATH}?v=${Constants.SIYUAN_VERSION}`);
+            /// #endif
+            addScriptSync(`${Constants.PROTYLE_CDN}/js/lute/lute.min.js?v=${Constants.SIYUAN_VERSION}`, "protyleLuteScript");
+            addScript(`${Constants.PROTYLE_CDN}/js/protyle-html.js?v=${Constants.SIYUAN_VERSION}`, "protyleWcHtmlScript");
+
+            setNoteBook();
+            initBlockPopover(this);
+
             await loadPlugins(this);
             getLocalStorage(() => {
                 fetchGet(`/appearance/langs/${window.siyuan.config.appearance.lang}.json?v=${Constants.SIYUAN_VERSION}`, (lauguages) => {
@@ -177,8 +185,6 @@ export class App {
                 });
             });
         });
-        setNoteBook();
-        initBlockPopover(this);
     }
 }
 

app/src/mobile/index.ts

@@ -32,12 +32,8 @@ class App {
     public appId: string;
 
     constructor() {
-        if (!window.webkit?.messageHandlers && !window.JSAndroid) {
-            registerServiceWorker(`${Constants.SERVICE_WORKER_PATH}?v=${Constants.SIYUAN_VERSION}`);
-        }
-        addScriptSync(`${Constants.PROTYLE_CDN}/js/lute/lute.min.js?v=${Constants.SIYUAN_VERSION}`, "protyleLuteScript");
-        addScript(`${Constants.PROTYLE_CDN}/js/protyle-html.js?v=${Constants.SIYUAN_VERSION}`, "protyleWcHtmlScript");
         addBaseURL();
+
         this.appId = Constants.SIYUAN_APPID;
         window.siyuan = {
             zIndex: 10,
@@ -60,6 +56,7 @@ class App {
                 }
             })
         };
+
         // 不能使用 touchstart，否则会被 event.stopImmediatePropagation() 阻塞
         window.addEventListener("click", (event: MouseEvent & { target: HTMLElement }) => {
             if (!window.siyuan.menus.menu.element.contains(event.target) && !hasClosestByAttribute(event.target, "data-menu", "true")) {
@@ -80,9 +77,17 @@ class App {
         window.addEventListener("pagehide", () => {
             saveScroll(window.siyuan.mobile.editor.protyle);
         }, false);
+
         fetchPost("/api/system/getConf", {}, async (confResponse) => {
             confResponse.data.conf.keymap = Constants.SIYUAN_KEYMAP;
             window.siyuan.config = confResponse.data.conf;
+
+            if (!window.webkit?.messageHandlers && !window.JSAndroid) {
+                await registerServiceWorker(`${Constants.SERVICE_WORKER_PATH}?v=${Constants.SIYUAN_VERSION}`);
+            }
+            addScriptSync(`${Constants.PROTYLE_CDN}/js/lute/lute.min.js?v=${Constants.SIYUAN_VERSION}`, "protyleLuteScript");
+            addScript(`${Constants.PROTYLE_CDN}/js/protyle-html.js?v=${Constants.SIYUAN_VERSION}`, "protyleWcHtmlScript");    
+
             await loadPlugins(this);
             getLocalStorage(() => {
                 fetchGet(`/appearance/langs/${window.siyuan.config.appearance.lang}.json?v=${Constants.SIYUAN_VERSION}`, (lauguages) => {

app/src/util/serviceWorker.ts

@@ -1,17 +1,18 @@
 // https://github.com/siyuan-note/siyuan/pull/8012
-export const registerServiceWorker = (scriptURL: string, scope = "/", workerType: WorkerType = "module") => {
-    if (!("serviceWorker" in navigator) || typeof (navigator.serviceWorker) === "undefined" ||
-        !("caches" in window) || !("fetch" in window)) {
-        return;
+export const registerServiceWorker = async (scriptURL: string, scope = "/", workerType: WorkerType = "module") => {
+    if (("serviceWorker" in window.navigator) && ("caches" in window) && ("fetch" in window)) {
+        try {
+            // REF https://developer.mozilla.org/en-US/docs/Web/API/ServiceWorkerRegistration
+            const registration = await window.navigator.serviceWorker.register(scriptURL, {
+                scope,
+                type: workerType,
+            });
+            await registration.update();
+            return true;
+        } catch (error) {
+            console.debug(`Registration failed with ${error}`);
+            return false;
+        }
     }
-    // REF https://developer.mozilla.org/en-US/docs/Web/API/ServiceWorkerRegistration
-    window.navigator.serviceWorker
-        .register(scriptURL, {
-            scope,
-            type: workerType,
-        }).then(registration => {
-        registration.update();
-    }).catch(e => {
-        console.debug(`Registration failed with ${e}`);
-    });
+    return false;
 };