From 7e4650ced45ad952c2f96fd17db050d7703d5201 Mon Sep 17 00:00:00 2001
From: JianJia2018 <39438074+JianJia2018@users.noreply.github.com>
Date: Wed, 10 Apr 2024 17:17:05 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20env=E5=8F=AF=E8=87=AA=E5=AE=9A=E4=B9=89?=
 =?UTF-8?q?UA=20(#371)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Co-authored-by: 简佳 <15779599039@163.com>
---
 env.example.js  | 5 ++++-
 lib/net/http.js | 2 +-
 main.js         | 1 +
 3 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/env.example.js b/env.example.js
index 967ca7a..7deafe1 100644
--- a/env.example.js
+++ b/env.example.js
@@ -5,6 +5,7 @@ module.exports = Object.freeze({
      * - `NOTE` 帐号备注
      * - `NUMBER` 表示是第几个账号
      * - `CLEAR` 是否启用清理功能
+     * - `ACCOUNT_UA` 账号UA, 可在浏览器控制台输入 navigator.userAgent 查看
      * ## 高级功能
      * - `ENABLE_CHAT_CAPTCHA_OCR` 开启评论验证码识别 使用方法见README
      * - `ENABLE_MULTIPLE_ACCOUNT` 是否启用多账号
@@ -25,6 +26,7 @@ module.exports = Object.freeze({
         NOTE: "",
         NUMBER: 1,
         CLEAR: true,
+        ACCOUNT_UA: "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/87.0.4280.88 Safari/537.36",
 
         ENABLE_CHAT_CAPTCHA_OCR: "",
         ENABLE_MULTIPLE_ACCOUNT: false,
@@ -59,7 +61,8 @@ module.exports = Object.freeze({
             NOTE: "",
             NUMBER: 1,
             CLEAR: true,
-            WAIT: 60 * 1000
+            WAIT: 60 * 1000,
+            ACCOUNT_UA: "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/87.0.4280.88 Safari/537.36"
         }
     ],
 
diff --git a/lib/net/http.js b/lib/net/http.js
index 37e8e4e..9b43559 100644
--- a/lib/net/http.js
+++ b/lib/net/http.js
@@ -56,7 +56,7 @@ const DEFAULT_REDIRECT = false;
 /**错误尝试次数 */
 const DEFAULT_RETRY_TIMES = 6;
 /**Google Chrome */
-const DEFAULT_UA = 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/87.0.4280.88 Safari/537.36';
+const DEFAULT_UA = process.env.ACCOUNT_UA || 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/87.0.4280.88 Safari/537.36';
 /**默认url编码 */
 const DEFAULT_CONTENT_TYPE = 'application/x-www-form-urlencoded; charset=UTF-8';
 /**
diff --git a/main.js b/main.js
index 30b8656..989b5c5 100644
--- a/main.js
+++ b/main.js
@@ -37,6 +37,7 @@ async function main() {
             process.env.NUMBER = acco.NUMBER;
             process.env.CLEAR = acco.CLEAR;
             process.env.NOTE = acco.NOTE;
+            process.env.ACCOUNT_UA = acco.ACCOUNT_UA;
             const err_msg = await main();
             if (err_msg) {
                 return err_msg