Merge pull request #25 from H3CoF6/feat/dynamic-next

fix (dynamic_next_id): get next action id dynamically

src/wechat_decrypt_tool/key_service.py

@@ -14,8 +14,10 @@ import subprocess
 import hashlib
 import os
 import json
+import re
 import random
 import logging
+import asyncio
 import httpx
 from pathlib import Path
 from typing import Optional, List, Dict, Any
@@ -203,7 +205,56 @@ def get_db_key_workflow():
 
 # 远程 API 配置
 REMOTE_URL = "https://view.free.c3o.re/dashboard"
-NEXT_ACTION_ID = "7c8f99280c70626ccf5960cc4a68f368197e15f8e9"
+BASE_URL = "https://view.free.c3o.re"  # 用于拼接js
+
+# NEXT_ACTION_ID = "7c8f99280c70626ccf5960cc4a68f368197e15f8e9"  # 不可以硬编码
+
+
+async def fetch_js_and_scan(client: httpx.AsyncClient, js_path: str) -> Optional[str]:
+    """
+    异步下载单个 JS 文件并匹配 Action ID
+    """
+    full_url = f"{BASE_URL}{js_path}" if js_path.startswith("/") else js_path
+    try:
+        response = await client.get(full_url)
+        if response.status_code != 200:
+            return None
+
+        content = response.text
+
+        action_id_pattern = re.compile(r'createServerReference.*?["\']([a-f0-9]{42})["\'].*?["\']getUserConfigFromBytes["\']')
+
+        match = action_id_pattern.search(content)
+        if match:
+            found_id = match.group(1)
+            return found_id
+
+    except Exception as e:
+        logger.error(f"Error fetching {js_path}: {e}")
+    return None
+
+
+async def _get_next_action_id_async() -> str:
+    async with httpx.AsyncClient(timeout=10.0) as client:
+
+        resp = await client.get(REMOTE_URL)
+        html = resp.text
+
+        js_file_pattern = re.compile(r'src="(/_next/static/chunks/[^"]+\.js)"')
+        js_files = set(js_file_pattern.findall(html))
+
+        if not js_files:
+            raise Exception("未找到任何 Next.js chunk 文件，可能页面结构已变动。")
+
+        tasks = [fetch_js_and_scan(client, js_path) for js_path in js_files]
+
+        results = await asyncio.gather(*tasks)
+
+        for res in results:
+            if res:
+                return res
+
+        raise Exception("遍历了所有 JS 文件，但未找到匹配的 createServerReference ID。")
 
 
 def get_wechat_internal_global_config(wx_dir: Path, file_name1) -> bytes:
@@ -221,41 +272,20 @@ def get_wechat_internal_global_config(wx_dir: Path, file_name1) -> bytes:
     return Path(target_path).read_bytes()
 
 
-# def get_local_config_sha3_224() -> bytes:
-#     """
-#     不要在意，抽象的实现 哈哈哈
-#     """
-#     content = json.dumps({
-#         "wxfile_dir": "C:\\Users\\17078\\xwechat_files",
-#         "weixin_id_folder": "wxid_lnyf4hdo9csb12_f1c4",
-#         "cache_dir": "C:\\Users\\17078\\Desktop\\wxDBHook\\test\\wx-dat\\wx-dat\\.cache",
-#         "db_key": "",
-#         "port": 8001
-#     }, indent=4).encode("utf-8")
-#
-#     # 计算 SHA3-224
-#     digest = hashlib.sha3_224(content).digest()
-#     return digest
-
-# async def log_request(request):
-#     print(f"--- Request Raw ---")
-#     print(f"{request.method} {request.url} {request.extensions.get('http_version', b'HTTP/1.1').decode()}")
-#     for name, value in request.headers.items():
-#         print(f"{name}: {value}")
-#
-#     print()
-#
-#     body = request.read()
-#     if body:
-#         print(body.decode(errors='replace'))
-#     print(f"-------------------\n")
-
 
 async def fetch_and_save_remote_keys(account: Optional[str] = None) -> Dict[str, Any]:
     account_dir = _resolve_account_dir(account)
     wx_id_dir = _resolve_account_wxid_dir(account_dir)
     wxid = wx_id_dir.name
 
+    logger.info("尝试获取next_action_id")
+    try:
+        next_action_id = await _get_next_action_id_async()
+        logger.info(f"获取next_action_id成功: {next_action_id}")
+    except Exception as e:
+        raise RuntimeError(f"获取next_action_id失败：{e}")
+
+
     logger.info(f"正在为账号 {wxid} 获取密钥...")
 
     try:
@@ -274,7 +304,7 @@ async def fetch_and_save_remote_keys(account: Optional[str] = None) -> Dict[str,
 
     headers = {
         "Accept": "text/x-component",
-        "Next-Action": NEXT_ACTION_ID,
+        "Next-Action": next_action_id,
         "Next-Router-State-Tree": "%5B%22%22%2C%7B%22children%22%3A%5B%22dashboard%22%2C%7B%22children%22%3A%5B%22__PAGE__%22%2C%7B%7D%2Cnull%2Cnull%5D%7D%2Cnull%2Cnull%5D%7D%2Cnull%2Cnull%2Ctrue%5D",
         "Origin": "https://view.free.c3o.re",
         "Referer": "https://view.free.c3o.re/dashboard",