From e60dd0a93ebd67aa344600a04ff1ecd98e10a4f5 Mon Sep 17 00:00:00 2001
From: kuaifan <aipaw@live.cn>
Date: Fri, 8 Jul 2022 18:44:06 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E6=B5=8F=E8=A7=88=E5=99=A8ua?=
 =?UTF-8?q?=E9=94=99=E8=AF=AF=E7=9A=84=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 electron/electron.js | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/electron/electron.js b/electron/electron.js
index f98672e76..9a1ff8e9f 100644
--- a/electron/electron.js
+++ b/electron/electron.js
@@ -39,7 +39,8 @@ function createMainWindow() {
             nativeWindowOpen: true
         }
     })
-    mainWindow.webContents.setUserAgent(mainWindow.webContents.getUserAgent() + " MainTaskWindow/" + process.platform + "/" + os.arch() + "/1.0");
+    const originalUA = mainWindow.webContents.session.getUserAgent() || mainWindow.webContents.getUserAgent()
+    mainWindow.webContents.setUserAgent(originalUA + " MainTaskWindow/" + process.platform + "/" + os.arch() + "/1.0");
     mainWindow.webContents.setWindowOpenHandler(({url}) => {
         openExternal(url)
         return {action: 'deny'}
@@ -138,7 +139,8 @@ function createSubWindow(args) {
 
         subWindow.push({ name, browser })
     }
-    browser.webContents.setUserAgent(browser.webContents.getUserAgent() + " SubTaskWindow/" + process.platform + "/" + os.arch() + "/1.0" + (args.userAgent ? (" " + args.userAgent) : ""));
+    const originalUA = browser.webContents.session.getUserAgent() || browser.webContents.getUserAgent()
+    browser.webContents.setUserAgent(originalUA + " SubTaskWindow/" + process.platform + "/" + os.arch() + "/1.0" + (args.userAgent ? (" " + args.userAgent) : ""));
     browser.webContents.setWindowOpenHandler(({url}) => {
         openExternal(url)
         return {action: 'deny'}