diff --git a/frontend/components/BizMessages.vue b/frontend/components/BizMessages.vue new file mode 100644 index 0000000..92d3d02 --- /dev/null +++ b/frontend/components/BizMessages.vue @@ -0,0 +1,270 @@ + + + + + \ No newline at end of file diff --git a/frontend/components/SidebarRail.vue b/frontend/components/SidebarRail.vue index c20b2f4..c83b4d4 100644 --- a/frontend/components/SidebarRail.vue +++ b/frontend/components/SidebarRail.vue @@ -101,6 +101,21 @@ +
+
+
+ +
+
+
+
route.path?.startsWith('/chat')) const isEditsRoute = computed(() => route.path?.startsWith('/edits')) const isSnsRoute = computed(() => route.path?.startsWith('/sns')) const isContactsRoute = computed(() => route.path?.startsWith('/contacts')) +const isBizRoute = computed(() => route.path?.startsWith('/biz')) // 新增 const isWrappedRoute = computed(() => route.path?.startsWith('/wrapped')) -const goChat = async () => { - await navigateTo('/chat') -} -const goEdits = async () => { - await navigateTo('/edits') -} - -const goSns = async () => { - await navigateTo('/sns') -} - -const goContacts = async () => { - await navigateTo('/contacts') -} - -const goWrapped = async () => { - await navigateTo('/wrapped') -} - -const goGuide = async () => { - await navigateTo('/') -} - -const goSettings = () => { - openSettingsDialog() -} +const goChat = async () => { await navigateTo('/chat') } +const goEdits = async () => { await navigateTo('/edits') } +const goSns = async () => { await navigateTo('/sns') } +const goContacts = async () => { await navigateTo('/contacts') } +const goBiz = async () => { await navigateTo('/biz') } +const goWrapped = async () => { await navigateTo('/wrapped') } +const goGuide = async () => { await navigateTo('/') } +const goSettings = () => { openSettingsDialog() } const onWindowKeydown = (event) => { if (event?.key !== 'Escape') return diff --git a/frontend/composables/useApi.js b/frontend/composables/useApi.js index 5d73eb4..f8338fd 100644 --- a/frontend/composables/useApi.js +++ b/frontend/composables/useApi.js @@ -561,6 +561,44 @@ export const useApi = () => { return await request('/get_image_key') } + // 枚举服务号信息 + const listBizAccounts = async (params = {}) => { + const query = new URLSearchParams() + if (params && params.account) query.set('account', params.account) + const url = '/biz/list' + (query.toString() ? `?${query.toString()}` : '') + return await request(url) + } + + // 获取普通服务号消息 + const listBizMessages = async (params = {}) => { + const query = new URLSearchParams() + if (params && params.account) query.set('account', params.account) + if (params && params.username) query.set('username', params.username) + if (params && params.limit != null) query.set('limit', String(params.limit)) + if (params && params.offset != null) query.set('offset', String(params.offset)) + const url = '/biz/messages' + (query.toString() ? `?${query.toString()}` : '') + return await request(url) + } + + // 获取微信支付记录 + const listBizPayRecords = async (params = {}) => { + const query = new URLSearchParams() + if (params && params.account) query.set('account', params.account) + if (params && params.limit != null) query.set('limit', String(params.limit)) + if (params && params.offset != null) query.set('offset', String(params.offset)) + const url = '/biz/pay_records' + (query.toString() ? `?${query.toString()}` : '') + return await request(url) + } + + const getBizProxyImageUrl = (url) => { + if (!url) return '' + if (url.startsWith('data:')) return url // 如果已经是 base64，不处理 + const query = new URLSearchParams() + query.set('url', url) + const base = baseURL ? baseURL.replace(/\/$/, '') : '' + return `${base}/biz/proxy_image?${query.toString()}` + } + return { detectWechat, detectCurrentAccount, @@ -616,5 +654,9 @@ export const useApi = () => { getKeys, getImageKey, getWxStatus, + listBizAccounts, + listBizMessages, + listBizPayRecords, + getBizProxyImageUrl, } } diff --git a/frontend/pages/biz.vue b/frontend/pages/biz.vue new file mode 100644 index 0000000..4c3d61c --- /dev/null +++ b/frontend/pages/biz.vue @@ -0,0 +1,16 @@ + + + \ No newline at end of file diff --git a/src/wechat_decrypt_tool/api.py b/src/wechat_decrypt_tool/api.py index b8866c5..bc2df4d 100644 --- a/src/wechat_decrypt_tool/api.py +++ b/src/wechat_decrypt_tool/api.py @@ -36,6 +36,7 @@ from .routers.wrapped import router as _wrapped_router from .request_logging import log_server_errors_middleware from .sns_stage_timing import add_sns_stage_timing_headers from .wcdb_realtime import WCDB_REALTIME, shutdown as _wcdb_shutdown +from .routers.biz import router as _biz_router app = FastAPI( title="微信数据库解密工具", @@ -96,6 +97,7 @@ app.include_router(_chat_media_router) app.include_router(_sns_router) app.include_router(_sns_export_router) app.include_router(_wrapped_router) +app.include_router(_biz_router) class _SPAStaticFiles(StaticFiles): diff --git a/src/wechat_decrypt_tool/routers/biz.py b/src/wechat_decrypt_tool/routers/biz.py new file mode 100644 index 0000000..df944f1 --- /dev/null +++ b/src/wechat_decrypt_tool/routers/biz.py @@ -0,0 +1,370 @@ +import hashlib +import sqlite3 +import time +import xml.etree.ElementTree as ET +from pathlib import Path +from typing import Optional, Any, Dict, List +import urllib +from fastapi import APIRouter, HTTPException, Response +from pydantic import BaseModel + +from ..chat_helpers import _resolve_account_dir +from ..path_fix import PathFixRoute +from ..logging_config import get_logger + +try: + import zstandard as zstd +except Exception: + zstd = None + +logger = get_logger(__name__) +router = APIRouter(route_class=PathFixRoute) + + +def decompress_zstd_content(data: bytes, source_id: str, local_id: int) -> Optional[bytes]: + """Zstandard 解压逻辑""" + if not data or not data.startswith(b'\x28\xb5\x2f\xfd'): + return None + try: + if zstd: + dctx = zstd.ZstdDecompressor() + return dctx.decompress(data, max_output_size=10 * 1024 * 1024) + except Exception as e: + error_msg = f"❌ [解压失败] 服务号id: {source_id}, local_id: {local_id} -> {e}" + print(error_msg) + logger.error(error_msg) + return None + + +def extract_xml_from_db_content(content: Any, source_id: str, local_id: int) -> str: + """提取并解压数据库内容""" + if not content: + return "" + + if isinstance(content, memoryview): + content = content.tobytes() + elif isinstance(content, str): + content = content.encode('utf-8', errors='ignore') + + if isinstance(content, bytes): + decompressed = decompress_zstd_content(content, source_id, local_id) + if decompressed: + return decompressed.decode('utf-8', errors='ignore') + + # 若不是 zstd 压缩或解压失败，尝试直接 decode + try: + return content.decode('utf-8', errors='ignore') + except Exception: + return "" + return "" + + +def parse_wechat_xml_to_struct(xml_str: str, source_id: str, local_id: int) -> Optional[Dict[str, Any]]: + """解析微信服务号 XML 到 Dict""" + if not xml_str.strip(): + return None + try: + root = ET.fromstring(xml_str) + + def get_tag_text(element, path, default=""): + node = element.find(path) + return node.text if node is not None and node.text else default + + main_cover = get_tag_text(root, ".//appmsg/thumburl") + if not main_cover: + main_cover = get_tag_text(root, ".//topnew/cover") + + result = { + "title": get_tag_text(root, ".//appmsg/title"), + "des": get_tag_text(root, ".//appmsg/des"), + "url": get_tag_text(root, ".//appmsg/url"), + "cover": main_cover, + "content_list": [] + } + + items = root.findall(".//mmreader/category/item") + for item in items: + item_struct = { + "title": get_tag_text(item, "title"), + "url": get_tag_text(item, "url"), + "cover": get_tag_text(item, "cover"), + "summary": get_tag_text(item, "summary") + } + if item_struct["title"]: + result["content_list"].append(item_struct) + + return result + except Exception as e: + error_msg = f"❌ [解析XML失败] 服务号id: {source_id}, local_id: {local_id} -> {e}" + print(error_msg) + logger.error(error_msg) + return None + + +def parse_pay_xml(xml_str: str, local_id: int) -> Optional[Dict[str, Any]]: + """解析微信支付 XML""" + if not xml_str.strip(): + return None + try: + root = ET.fromstring(xml_str) + + def get_text(path): + node = root.find(path) + return node.text if node is not None else "" + + record = { + "title": get_text(".//appmsg/title"), + "description": get_text(".//appmsg/des"), + "merchant_name": get_text(".//template_header/display_name"), + "merchant_icon": get_text(".//template_header/icon_url"), + "timestamp": int(get_text(".//pub_time") or 0), + "formatted_time": "" + } + return record + except Exception as e: + error_msg = f"❌ [解析微信支付XML失败] 支付id: gh_3dfda90e39d6, local_id: {local_id} -> {e}" + print(error_msg) + logger.error(error_msg) + return None + +@router.get("/api/biz/proxy_image", summary="代理请求微信服务号图片") +def proxy_biz_image(url: str): + if not url: + return Response(status_code=400) + try: + req = urllib.request.Request(url, headers={ + 'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36' + }) + with urllib.request.urlopen(req, timeout=10) as response: + content = response.read() + content_type = response.headers.get('Content-Type', 'image/jpeg') + return Response(content=content, media_type=content_type) + except Exception as e: + logger.error(f"[biz] 代理图片失败: {url} -> {e}") + return Response(status_code=500) + +# 接口 1：获取全部的服务号/公众号的信息 +@router.get("/api/biz/list", summary="获取全部服务号/公众号列表") +def get_biz_account_list(account: Optional[str] = None): + account_dir = _resolve_account_dir(account) + + biz_ids = set() + biz_latest_time = {} + + # 1. 遍历 biz_message_*.db + for db_file in account_dir.glob("biz_message*.db"): + try: + conn = sqlite3.connect(str(db_file)) + cursor = conn.cursor() + + cursor.execute("PRAGMA table_info(Name2Id)") + cols = [row[1].lower() for row in cursor.fetchall()] + user_col = "username" if "username" in cols else "user_name" if "user_name" in cols else "" + + if user_col: + rows = cursor.execute(f"SELECT {user_col} FROM Name2Id").fetchall() + for r in rows: + if r[0]: + uname = r[0] + biz_ids.add(uname) + + # 顺便查询该号的最后一条消息时间 + md5_id = hashlib.md5(uname.encode('utf-8')).hexdigest().lower() + table_name = f"Msg_{md5_id}" + try: + time_res = conn.execute(f"SELECT MAX(create_time) FROM {table_name}").fetchone() + if time_res and time_res[0]: + current_max = biz_latest_time.get(uname, 0) + biz_latest_time[uname] = max(current_max, time_res[0]) + except Exception: + pass + conn.close() + except Exception as e: + logger.warning(f"读取 Name2Id 失败 {db_file}: {e}") + + contact_db_path = account_dir / "contact.db" + contact_info = {} + if contact_db_path.exists() and biz_ids: + try: + conn = sqlite3.connect(str(contact_db_path)) + cursor = conn.cursor() + placeholders = ",".join(["?"] * len(biz_ids)) + + # 先查 contact 表 + query_contact = f"SELECT username, remark, nick_name, alias, big_head_url FROM contact WHERE username IN ({placeholders})" + rows_contact = cursor.execute(query_contact, list(biz_ids)).fetchall() + + for r in rows_contact: + uname = r[0] + name = r[1] or r[2] or r[3] or uname + contact_info[uname] = { + "username": uname, + "name": name, + "avatar": r[4], + "type": 3 # 默认给个 3（未知） + } + + # 再查 biz_info 表获取类型 + try: + query_biz = f"SELECT username, type FROM biz_info WHERE username IN ({placeholders})" + rows_biz = cursor.execute(query_biz, list(biz_ids)).fetchall() + for r in rows_biz: + uname = r[0] + biz_type = r[1] + # 如果查到了且是 0, 1, 2，就更新进去，否则保留 3 + if uname in contact_info: + if biz_type in (0, 1, 2): + contact_info[uname]["type"] = biz_type + else: + contact_info[uname]["type"] = 3 + except Exception as e: + logger.warning(f"读取 biz_info 失败: {e}") + + conn.close() + except Exception as e: + logger.warning(f"读取 contact.db 失败: {e}") + + # 3. 组装结果（不在 contact_info 里的直接丢弃） + result = [] + for uid in biz_ids: + if uid in contact_info: + info = contact_info[uid] + info["last_time"] = biz_latest_time.get(uid, 0) + if info["last_time"]: + # 格式化日期给前端展示用 + info["formatted_last_time"] = time.strftime("%Y-%m-%d", time.localtime(info["last_time"])) + else: + info["formatted_last_time"] = "" + result.append(info) + + # 4. 按最后一条消息的时间降序排列 + result.sort(key=lambda x: x.get("last_time", 0), reverse=True) + + return {"status": "success", "total": len(result), "data": result} + + +# 接口 2：获取普通服务号/公众号的 json 消息 (已修复表名比对 bug) +@router.get("/api/biz/messages", summary="获取指定服务号的消息") +def get_biz_messages(username: str, account: Optional[str] = None, limit: int = 50, offset: int = 0): + if username == "gh_3dfda90e39d6": + raise HTTPException(status_code=400, detail="微信支付记录请请求 /api/biz/pay_records 接口") + + account_dir = _resolve_account_dir(account) + md5_id = hashlib.md5(username.encode('utf-8')).hexdigest().lower() + table_name = f"Msg_{md5_id}" + + target_db = None + for db_file in account_dir.glob("biz_message*.db"): + conn = sqlite3.connect(str(db_file)) + try: + # 必须用 table_name.lower()，否则永远匹配不上 + res = conn.execute("SELECT name FROM sqlite_master WHERE type='table' AND lower(name)=?", + (table_name.lower(),)).fetchone() + if res: + target_db = db_file + break + except Exception: + pass + finally: + conn.close() + + if not target_db: + return {"status": "success", "data": [], "message": f"未找到 {username} 的消息历史"} + + # ... (后续数据库查询逻辑保持不变) ... + messages = [] + try: + conn = sqlite3.connect(str(target_db)) + cursor = conn.cursor() + + query = f""" + SELECT local_id, create_time, message_content + FROM [{table_name}] + WHERE local_type != 1 + ORDER BY create_time DESC + LIMIT ? OFFSET ? + """ + rows = cursor.execute(query, (limit, offset)).fetchall() + + for local_id, c_time, content in rows: + raw_xml = extract_xml_from_db_content(content, username, local_id) + if not raw_xml: + continue + + struct_data = parse_wechat_xml_to_struct(raw_xml, username, local_id) + if struct_data: + struct_data["local_id"] = local_id + struct_data["create_time"] = c_time + messages.append(struct_data) + + conn.close() + except Exception as e: + logger.error(f"[biz] 数据库查询出错: {e}") + return {"status": "error", "message": str(e)} + + return {"status": "success", "data": messages} + + +# 接口 3：返回微信支付的 json 消息 (已修复表名比对 bug) +@router.get("/api/biz/pay_records", summary="获取微信支付记录") +def get_wechat_pay_records(account: Optional[str] = None, limit: int = 50, offset: int = 0): + username = "gh_3dfda90e39d6" + account_dir = _resolve_account_dir(account) + md5_id = hashlib.md5(username.encode('utf-8')).hexdigest().lower() + table_name = f"Msg_{md5_id}" + + target_db = None + for db_file in account_dir.glob("biz_message*.db"): + conn = sqlite3.connect(str(db_file)) + try: + # 必须用 table_name.lower() + res = conn.execute("SELECT name FROM sqlite_master WHERE type='table' AND lower(name)=?", + (table_name.lower(),)).fetchone() + if res: + target_db = db_file + break + except Exception: + pass + finally: + conn.close() + + if not target_db: + return {"status": "success", "data": [], "message": "未找到微信支付的消息历史"} + + messages = [] + try: + conn = sqlite3.connect(str(target_db)) + cursor = conn.cursor() + + query = f""" + SELECT local_id, create_time, message_content + FROM [{table_name}] + WHERE local_type = 21474836529 OR local_type != 1 + ORDER BY create_time DESC + LIMIT ? OFFSET ? + """ + rows = cursor.execute(query, (limit, offset)).fetchall() + + for local_id, c_time, content in rows: + raw_xml = extract_xml_from_db_content(content, username, local_id) + if not raw_xml: + continue + + parsed_data = parse_pay_xml(raw_xml, local_id) + if parsed_data: + parsed_data["local_id"] = local_id + parsed_data["create_time"] = c_time + if not parsed_data["timestamp"]: + parsed_data["timestamp"] = c_time + + parsed_data["formatted_time"] = time.strftime( + "%Y-%m-%d %H:%M:%S", time.localtime(parsed_data["timestamp"]) + ) + messages.append(parsed_data) + + conn.close() + except Exception as e: + logger.error(f"[biz] 查询微信支付数据库出错: {e}") + return {"status": "error", "message": str(e)} + + return {"status": "success", "data": messages} \ No newline at end of file