From 35f43e8669b72f29cf33fa2c7665eb87a0e6a6a6 Mon Sep 17 00:00:00 2001
From: Vanessa <lly219@gmail.com>
Date: Wed, 16 Jul 2025 18:29:28 +0800
Subject: [PATCH] :art: https://github.com/siyuan-note/siyuan/issues/15308

---
 app/src/util/processMessage.ts | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

diff --git a/app/src/util/processMessage.ts b/app/src/util/processMessage.ts
index 649d3793c..e95e35659 100644
--- a/app/src/util/processMessage.ts
+++ b/app/src/util/processMessage.ts
@@ -33,15 +33,22 @@ export const processMessage = (response: IWebSocketData) => {
         return false;
     }
     if ("reloadui" === response.cmd) {
+        const reloadByMode = (mode: string) => {
+            if (mode) {
+                window.location.pathname = "stage/build/" + mode;
+            } else {
+                window.location.reload();
+            }
+        };
         if (response.data?.resetScroll) {
             window.siyuan.storage[Constants.LOCAL_FILEPOSITION] = {};
             setStorageVal(Constants.LOCAL_FILEPOSITION, window.siyuan.storage[Constants.LOCAL_FILEPOSITION], () => {
                 /// #if MOBILE
-                window.location.reload();
+                reloadByMode(response.data?.mode);
                 /// #else
                 exportLayout({
                     cb() {
-                        window.location.reload();
+                        reloadByMode(response.data?.mode);
                     },
                     errorExit: false,
                 });
@@ -49,11 +56,11 @@ export const processMessage = (response: IWebSocketData) => {
             });
         } else {
             /// #if MOBILE
-            window.location.reload();
+            reloadByMode(response.data?.mode);
             /// #else
             exportLayout({
                 cb() {
-                    window.location.reload();
+                    reloadByMode(response.data?.mode);
                 },
                 errorExit: false,
             });