From fc81008ad11345314d8933a4714ca2f54274ebef Mon Sep 17 00:00:00 2001
From: DIYgod <i@diygod.me>
Date: Sun, 9 Mar 2025 09:53:43 +0800
Subject: [PATCH] fix: set header before init window

---
 apps/desktop/src/main/src/index.ts | 30 +++++++++++++++---------------
 1 file changed, 15 insertions(+), 15 deletions(-)

diff --git a/apps/desktop/src/main/src/index.ts b/apps/desktop/src/main/src/index.ts
index d6eb2aa01..4e4082ce1 100644
--- a/apps/desktop/src/main/src/index.ts
+++ b/apps/desktop/src/main/src/index.ts
@@ -35,6 +35,12 @@ if (squirrelStartup) {
   app.quit()
 }
 
+const buildSafeHeaders = createBuildSafeHeaders(env.VITE_WEB_URL, [
+  env.VITE_OPENPANEL_API_URL || "",
+  IMAGE_PROXY_URL,
+  env.VITE_API_URL,
+])
+
 function bootstrap() {
   initializeAppStage0()
   const gotTheLock = app.requestSingleInstanceLock()
@@ -92,6 +98,15 @@ function bootstrap() {
     // Set app user model id for windows
     electronApp.setAppUserModelId(`re.${APP_PROTOCOL}`)
 
+    session.defaultSession.webRequest.onBeforeSendHeaders((details, callback) => {
+      details.requestHeaders = buildSafeHeaders({
+        url: details.url,
+        headers: details.requestHeaders,
+      })
+
+      callback({ cancel: false, requestHeaders: details.requestHeaders })
+    })
+
     mainWindow = createMainWindow()
 
     // restore cookies
@@ -120,21 +135,6 @@ function bootstrap() {
     registerUpdater()
     registerAppTray()
 
-    const buildSafeHeaders = createBuildSafeHeaders(env.VITE_WEB_URL, [
-      env.VITE_OPENPANEL_API_URL || "",
-      IMAGE_PROXY_URL,
-      env.VITE_API_URL,
-    ])
-
-    session.defaultSession.webRequest.onBeforeSendHeaders((details, callback) => {
-      details.requestHeaders = buildSafeHeaders({
-        url: details.url,
-        headers: details.requestHeaders,
-      })
-
-      callback({ cancel: false, requestHeaders: details.requestHeaders })
-    })
-
     // handle session cookie when sign in with email in electron
     session.defaultSession.webRequest.onHeadersReceived(
       {