From ea09fc9b26fa033c43c7eee0ed738450e5876f23 Mon Sep 17 00:00:00 2001
From: abearxiong <xiongxiao1012@outlook.com>
Date: Fri, 2 May 2025 22:47:53 +0800
Subject: [PATCH] fix: fix cookie error

---
 server.py      | 65 +++++++++++++++++++++++++++++++++++++--------
 xha-api/app.py | 72 ++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 126 insertions(+), 11 deletions(-)
 create mode 100644 xha-api/app.py

diff --git a/server.py b/server.py
index c516eae..0c36b79 100644
--- a/server.py
+++ b/server.py
@@ -20,6 +20,7 @@ context_page = None
 def get_context_page(instance, stealth_js_path):
     chromium = instance.chromium
     browser = chromium.launch(headless=True)
+    # browser = chromium.launch(headless=False)
     context = browser.new_context()
     context.add_init_script(path=stealth_js_path)
     page = context.new_page()
@@ -41,8 +42,45 @@ for cookie in cookies:
     if cookie["name"] == "a1":
         global_a1 = cookie["value"]
         print("当前浏览器 cookie 中 a1 值为：" + cookie["value"] + "，请将需要使用的 a1 设置成一样方可签名成功")
+
+loginModal = context_page.query_selector(".reds-mask")
+if loginModal is not None:
+    loginModal.evaluate("el => el.click()")
+    print("登录弹窗已关闭")
 print("跳转小红书首页成功，等待调用")
 
+def setCookie(a1):
+    global browser_context, context_page  # 声明全局变量
+    global global_a1
+    try:
+        # 确保页面仍然有效，如果页面已关闭则重新初始化
+        if context_page is None or context_page.is_closed():
+            browser_context, context_page = get_context_page(playwright, stealth_js_path)
+            context_page.goto("https://www.xiaohongshu.com")
+            time.sleep(5)
+            context_page.reload()
+            time.sleep(1)
+        if a1 != global_a1:
+            # 删除 a1 cookie
+            browser_context.add_cookies([
+                {'name': 'a1', 'value': '', 'domain': ".xiaohongshu.com", 'path': "/"}
+            ])
+            # 等待一段时间以确保 cookie 被删除
+            time.sleep(1)
+            # 设置新的 a1 cookie
+            browser_context.add_cookies([
+                {'name': 'a1', 'value': a1, 'domain': ".xiaohongshu.com", 'path': "/"}
+            ])
+            context_page.reload()
+            time.sleep(1)
+            cookies = browser_context.cookies()
+            print("页面加载后的 Cookie:", cookies)
+            global_a1 = a1
+        print("设置 cookie 成功", a1)
+        return {"status": "success", "message": "Cookie set successfully"}
+    except Exception as e:
+        print(f"Error during setCookie operation: {e}")
+        return {"error": str(e)}
 
 def sign(uri, data, a1, web_session):
     global browser_context, context_page  # 声明全局变量
@@ -57,15 +95,7 @@ def sign(uri, data, a1, web_session):
             time.sleep(1)
             
         if a1 != global_a1:
-          # 清理浏览器 cookie
-          browser_context.clear_cookies()
-          time.sleep(5)
-          browser_context.add_cookies([
-              {'name': 'a1', 'value': a1, 'domain': ".xiaohongshu.com", 'path': "/"}
-          ])
-          context_page.reload()
-          time.sleep(1)
-          global_a1 = a1
+            setCookie(a1)
         # 执行 JavaScript 函数
         # localStorage.getItem("b1")
         b1 = context_page.evaluate("() => localStorage.getItem('b1')")
@@ -89,13 +119,26 @@ def hello_world():
     data = json["data"]
     a1 = json["a1"]
     web_session = json["web_session"]
-    return sign(uri, data, a1, web_session)
+    me = sign(uri, data, a1, web_session)
+    return {
+        "a1": a1,
+        "sign": me,
+    }
 
 
 @app.route("/a1", methods=["GET"])
 def get_a1():
     global global_a1
-    return {'a1': global_a1}
+    # 获取 params？a1的参数
+    a1 = request.args.get('a1')
+    if a1 and a1 != global_a1:
+        # 如果提供了 a1 参数且与当前的 global_a1 不同，则调用 setCookie 函数
+        # 调用 setCookie 函数
+        result = setCookie(a1)
+        return {'a1': a1, 'result': result}
+    else:
+        # 如果没有提供 a1 参数，则返回当前的 global_a1 值
+        return {'a1': global_a1}
 
 
 if __name__ == '__main__':
diff --git a/xha-api/app.py b/xha-api/app.py
new file mode 100644
index 0000000..b155ea2
--- /dev/null
+++ b/xha-api/app.py
@@ -0,0 +1,72 @@
+import time
+
+from flask import Flask, request
+from gevent import monkey
+from playwright.sync_api import sync_playwright
+
+monkey.patch_all()
+
+app = Flask(__name__)
+
+global_a1 = ""
+
+
+def get_context_page(instance, stealth_js_path):
+    chromium = instance.chromium
+    browser = chromium.launch(headless=True)
+    context = browser.new_context()
+    context.add_init_script(path=stealth_js_path)
+    page = context.new_page()
+    return context, page
+
+
+stealth_js_path = "stealth.min.js"
+print("正在启动 playwright")
+playwright = sync_playwright().start()
+browser_context, context_page = get_context_page(playwright, stealth_js_path)
+context_page.goto("https://www.xiaohongshu.com")
+print("正在跳转至小红书首页")
+time.sleep(5)
+context_page.reload()
+time.sleep(1)
+cookies = browser_context.cookies()
+for cookie in cookies:
+    if cookie["name"] == "a1":
+        global_a1 = cookie["value"]
+        print("当前浏览器中 a1 值为：" + global_a1 + "，请将您的 cookie 中的 a1 也设置成一样，方可签名成功")
+print("跳转小红书首页成功，等待调用")
+
+
+def sign(uri, data, a1, web_session):
+    global global_a1
+    if a1 != global_a1:
+        browser_context.add_cookies([
+            {'name': 'a1', 'value': a1, 'domain': ".xiaohongshu.com", 'path': "/"}
+        ])
+        context_page.reload()
+        time.sleep(1)
+        global_a1 = a1
+    encrypt_params = context_page.evaluate("([url, data]) => window._webmsxyw(url, data)", [uri, data])
+    return {
+        "x-s": encrypt_params["X-s"],
+        "x-t": str(encrypt_params["X-t"])
+    }
+
+
+@app.route("/sign", methods=["POST"])
+def hello_world():
+    json = request.json
+    uri = json["uri"]
+    data = json["data"]
+    a1 = json["a1"]
+    web_session = json["web_session"]
+    return sign(uri, data, a1, web_session)
+
+
+@app.route("/a1", methods=["GET"])
+def get_a1():
+    return {'a1': global_a1}
+
+
+if __name__ == '__main__':
+    app.run(host="0.0.0.0", port=5005)
\ No newline at end of file