""" Weixin platform adapter. Connects Hermes Agent to WeChat personal accounts via Tencent's iLink Bot API. Design notes: - Long-poll ``getupdates`` drives inbound delivery. - Every outbound reply must echo the latest ``context_token`` for the peer. - Media files move through an AES-128-ECB encrypted CDN protocol. - QR login is exposed as a helper for the gateway setup wizard. """ from __future__ import annotations import asyncio import base64 import hashlib import json import logging import mimetypes import os import re import secrets import struct import tempfile import time import uuid from datetime import datetime from pathlib import Path from typing import Any, Dict, List, Optional, Tuple from urllib.parse import quote logger = logging.getLogger(__name__) try: import aiohttp AIOHTTP_AVAILABLE = True except ImportError: # pragma: no cover - dependency gate aiohttp = None # type: ignore[assignment] AIOHTTP_AVAILABLE = False try: from cryptography.hazmat.backends import default_backend from cryptography.hazmat.primitives.ciphers import Cipher, algorithms, modes CRYPTO_AVAILABLE = True except ImportError: # pragma: no cover - dependency gate default_backend = None # type: ignore[assignment] Cipher = None # type: ignore[assignment] algorithms = None # type: ignore[assignment] modes = None # type: ignore[assignment] CRYPTO_AVAILABLE = False from gateway.config import Platform, PlatformConfig from gateway.platforms.helpers import MessageDeduplicator from gateway.platforms.base import ( BasePlatformAdapter, MessageEvent, MessageType, SendResult, cache_audio_from_bytes, cache_document_from_bytes, cache_image_from_bytes, ) from hermes_constants import get_hermes_home from utils import atomic_json_write ILINK_BASE_URL = "https://ilinkai.weixin.qq.com" WEIXIN_CDN_BASE_URL = "https://novac2c.cdn.weixin.qq.com/c2c" ILINK_APP_ID = "bot" CHANNEL_VERSION = "2.2.0" ILINK_APP_CLIENT_VERSION = (2 << 16) | (2 << 8) | 0 EP_GET_UPDATES = "ilink/bot/getupdates" EP_SEND_MESSAGE = "ilink/bot/sendmessage" EP_SEND_TYPING = "ilink/bot/sendtyping" EP_GET_CONFIG = "ilink/bot/getconfig" EP_GET_UPLOAD_URL = "ilink/bot/getuploadurl" EP_GET_BOT_QR = "ilink/bot/get_bot_qrcode" EP_GET_QR_STATUS = "ilink/bot/get_qrcode_status" LONG_POLL_TIMEOUT_MS = 35_000 API_TIMEOUT_MS = 15_000 CONFIG_TIMEOUT_MS = 10_000 QR_TIMEOUT_MS = 35_000 MAX_CONSECUTIVE_FAILURES = 3 RETRY_DELAY_SECONDS = 2 BACKOFF_DELAY_SECONDS = 30 SESSION_EXPIRED_ERRCODE = -14 MESSAGE_DEDUP_TTL_SECONDS = 300 MEDIA_IMAGE = 1 MEDIA_VIDEO = 2 MEDIA_FILE = 3 MEDIA_VOICE = 4 ITEM_TEXT = 1 ITEM_IMAGE = 2 ITEM_VOICE = 3 ITEM_FILE = 4 ITEM_VIDEO = 5 MSG_TYPE_USER = 1 MSG_TYPE_BOT = 2 MSG_STATE_FINISH = 2 TYPING_START = 1 TYPING_STOP = 2 _HEADER_RE = re.compile(r"^(#{1,6})\s+(.+?)\s*$") _TABLE_RULE_RE = re.compile(r"^\s*\|?(?:\s*:?-{3,}:?\s*\|)+\s*:?-{3,}:?\s*\|?\s*$") _FENCE_RE = re.compile(r"^```([^\n`]*)\s*$") def check_weixin_requirements() -> bool: """Return True when runtime dependencies for Weixin are available.""" return AIOHTTP_AVAILABLE and CRYPTO_AVAILABLE def _safe_id(value: Optional[str], keep: int = 8) -> str: raw = str(value or "").strip() if not raw: return "?" if len(raw) <= keep: return raw return raw[:keep] def _json_dumps(payload: Dict[str, Any]) -> str: return json.dumps(payload, ensure_ascii=False, separators=(",", ":")) def _pkcs7_pad(data: bytes, block_size: int = 16) -> bytes: pad_len = block_size - (len(data) % block_size) return data + bytes([pad_len] * pad_len) def _aes128_ecb_encrypt(plaintext: bytes, key: bytes) -> bytes: cipher = Cipher(algorithms.AES(key), modes.ECB(), backend=default_backend()) encryptor = cipher.encryptor() return encryptor.update(_pkcs7_pad(plaintext)) + encryptor.finalize() def _aes128_ecb_decrypt(ciphertext: bytes, key: bytes) -> bytes: cipher = Cipher(algorithms.AES(key), modes.ECB(), backend=default_backend()) decryptor = cipher.decryptor() padded = decryptor.update(ciphertext) + decryptor.finalize() if not padded: return padded pad_len = padded[-1] if 1 <= pad_len <= 16 and padded.endswith(bytes([pad_len]) * pad_len): return padded[:-pad_len] return padded def _aes_padded_size(size: int) -> int: return ((size + 1 + 15) // 16) * 16 def _random_wechat_uin() -> str: value = struct.unpack(">I", secrets.token_bytes(4))[0] return base64.b64encode(str(value).encode("utf-8")).decode("ascii") def _base_info() -> Dict[str, Any]: return {"channel_version": CHANNEL_VERSION} def _headers(token: Optional[str], body: str) -> Dict[str, str]: headers = { "Content-Type": "application/json", "AuthorizationType": "ilink_bot_token", "Content-Length": str(len(body.encode("utf-8"))), "X-WECHAT-UIN": _random_wechat_uin(), "iLink-App-Id": ILINK_APP_ID, "iLink-App-ClientVersion": str(ILINK_APP_CLIENT_VERSION), } if token: headers["Authorization"] = f"Bearer {token}" return headers def _account_dir(hermes_home: str) -> Path: path = Path(hermes_home) / "weixin" / "accounts" path.mkdir(parents=True, exist_ok=True) return path def _account_file(hermes_home: str, account_id: str) -> Path: return _account_dir(hermes_home) / f"{account_id}.json" def save_weixin_account( hermes_home: str, *, account_id: str, token: str, base_url: str, user_id: str = "", ) -> None: """Persist account credentials for later reuse.""" payload = { "token": token, "base_url": base_url, "user_id": user_id, "saved_at": time.strftime("%Y-%m-%dT%H:%M:%SZ", time.gmtime()), } path = _account_file(hermes_home, account_id) atomic_json_write(path, payload) try: path.chmod(0o600) except OSError: pass def load_weixin_account(hermes_home: str, account_id: str) -> Optional[Dict[str, Any]]: """Load persisted account credentials.""" path = _account_file(hermes_home, account_id) if not path.exists(): return None try: return json.loads(path.read_text(encoding="utf-8")) except Exception: return None class ContextTokenStore: """Disk-backed ``context_token`` cache keyed by account + peer.""" def __init__(self, hermes_home: str): self._root = _account_dir(hermes_home) self._cache: Dict[str, str] = {} def _path(self, account_id: str) -> Path: return self._root / f"{account_id}.context-tokens.json" def _key(self, account_id: str, user_id: str) -> str: return f"{account_id}:{user_id}" def restore(self, account_id: str) -> None: path = self._path(account_id) if not path.exists(): return try: data = json.loads(path.read_text(encoding="utf-8")) except Exception as exc: logger.warning("weixin: failed to restore context tokens for %s: %s", _safe_id(account_id), exc) return restored = 0 for user_id, token in data.items(): if isinstance(token, str) and token: self._cache[self._key(account_id, user_id)] = token restored += 1 if restored: logger.info("weixin: restored %d context token(s) for %s", restored, _safe_id(account_id)) def get(self, account_id: str, user_id: str) -> Optional[str]: return self._cache.get(self._key(account_id, user_id)) def set(self, account_id: str, user_id: str, token: str) -> None: self._cache[self._key(account_id, user_id)] = token self._persist(account_id) def _persist(self, account_id: str) -> None: prefix = f"{account_id}:" payload = { key[len(prefix) :]: value for key, value in self._cache.items() if key.startswith(prefix) } try: atomic_json_write(self._path(account_id), payload) except Exception as exc: logger.warning("weixin: failed to persist context tokens for %s: %s", _safe_id(account_id), exc) class TypingTicketCache: """Short-lived typing ticket cache from ``getconfig``.""" def __init__(self, ttl_seconds: float = 600.0): self._ttl_seconds = ttl_seconds self._cache: Dict[str, Tuple[str, float]] = {} def get(self, user_id: str) -> Optional[str]: entry = self._cache.get(user_id) if not entry: return None if time.time() - entry[1] >= self._ttl_seconds: self._cache.pop(user_id, None) return None return entry[0] def set(self, user_id: str, ticket: str) -> None: self._cache[user_id] = (ticket, time.time()) def _cdn_download_url(cdn_base_url: str, encrypted_query_param: str) -> str: return f"{cdn_base_url.rstrip('/')}/download?encrypted_query_param={quote(encrypted_query_param, safe='')}" def _cdn_upload_url(cdn_base_url: str, upload_param: str, filekey: str) -> str: return ( f"{cdn_base_url.rstrip('/')}/upload" f"?encrypted_query_param={quote(upload_param, safe='')}" f"&filekey={quote(filekey, safe='')}" ) def _parse_aes_key(aes_key_b64: str) -> bytes: decoded = base64.b64decode(aes_key_b64) if len(decoded) == 16: return decoded if len(decoded) == 32: text = decoded.decode("ascii", errors="ignore") if text and all(ch in "0123456789abcdefABCDEF" for ch in text): return bytes.fromhex(text) raise ValueError(f"unexpected aes_key format ({len(decoded)} decoded bytes)") def _guess_chat_type(message: Dict[str, Any], account_id: str) -> Tuple[str, str]: room_id = str(message.get("room_id") or message.get("chat_room_id") or "").strip() to_user_id = str(message.get("to_user_id") or "").strip() is_group = bool(room_id) or (to_user_id and account_id and to_user_id != account_id and message.get("msg_type") == 1) if is_group: return "group", room_id or to_user_id or str(message.get("from_user_id") or "") return "dm", str(message.get("from_user_id") or "") async def _api_post( session: "aiohttp.ClientSession", *, base_url: str, endpoint: str, payload: Dict[str, Any], token: Optional[str], timeout_ms: int, ) -> Dict[str, Any]: body = _json_dumps({**payload, "base_info": _base_info()}) url = f"{base_url.rstrip('/')}/{endpoint}" timeout = aiohttp.ClientTimeout(total=timeout_ms / 1000) async with session.post(url, data=body, headers=_headers(token, body), timeout=timeout) as response: raw = await response.text() if not response.ok: raise RuntimeError(f"iLink POST {endpoint} HTTP {response.status}: {raw[:200]}") return json.loads(raw) async def _api_get( session: "aiohttp.ClientSession", *, base_url: str, endpoint: str, timeout_ms: int, ) -> Dict[str, Any]: url = f"{base_url.rstrip('/')}/{endpoint}" headers = { "iLink-App-Id": ILINK_APP_ID, "iLink-App-ClientVersion": str(ILINK_APP_CLIENT_VERSION), } timeout = aiohttp.ClientTimeout(total=timeout_ms / 1000) async with session.get(url, headers=headers, timeout=timeout) as response: raw = await response.text() if not response.ok: raise RuntimeError(f"iLink GET {endpoint} HTTP {response.status}: {raw[:200]}") return json.loads(raw) async def _get_updates( session: "aiohttp.ClientSession", *, base_url: str, token: str, sync_buf: str, timeout_ms: int, ) -> Dict[str, Any]: try: return await _api_post( session, base_url=base_url, endpoint=EP_GET_UPDATES, payload={"get_updates_buf": sync_buf}, token=token, timeout_ms=timeout_ms, ) except asyncio.TimeoutError: return {"ret": 0, "msgs": [], "get_updates_buf": sync_buf} async def _send_message( session: "aiohttp.ClientSession", *, base_url: str, token: str, to: str, text: str, context_token: Optional[str], client_id: str, ) -> None: message: Dict[str, Any] = { "from_user_id": "", "to_user_id": to, "client_id": client_id, "message_type": MSG_TYPE_BOT, "message_state": MSG_STATE_FINISH, } if text: message["item_list"] = [{"type": ITEM_TEXT, "text_item": {"text": text}}] if context_token: message["context_token"] = context_token await _api_post( session, base_url=base_url, endpoint=EP_SEND_MESSAGE, payload={"msg": message}, token=token, timeout_ms=API_TIMEOUT_MS, ) async def _send_typing( session: "aiohttp.ClientSession", *, base_url: str, token: str, to_user_id: str, typing_ticket: str, status: int, ) -> None: await _api_post( session, base_url=base_url, endpoint=EP_SEND_TYPING, payload={ "ilink_user_id": to_user_id, "typing_ticket": typing_ticket, "status": status, }, token=token, timeout_ms=CONFIG_TIMEOUT_MS, ) async def _get_config( session: "aiohttp.ClientSession", *, base_url: str, token: str, user_id: str, context_token: Optional[str], ) -> Dict[str, Any]: payload: Dict[str, Any] = {"ilink_user_id": user_id} if context_token: payload["context_token"] = context_token return await _api_post( session, base_url=base_url, endpoint=EP_GET_CONFIG, payload=payload, token=token, timeout_ms=CONFIG_TIMEOUT_MS, ) async def _get_upload_url( session: "aiohttp.ClientSession", *, base_url: str, token: str, to_user_id: str, media_type: int, filekey: str, rawsize: int, rawfilemd5: str, filesize: int, aeskey_hex: str, ) -> Dict[str, Any]: return await _api_post( session, base_url=base_url, endpoint=EP_GET_UPLOAD_URL, payload={ "filekey": filekey, "media_type": media_type, "to_user_id": to_user_id, "rawsize": rawsize, "rawfilemd5": rawfilemd5, "filesize": filesize, "no_need_thumb": True, "aeskey": aeskey_hex, }, token=token, timeout_ms=API_TIMEOUT_MS, ) async def _upload_ciphertext( session: "aiohttp.ClientSession", *, ciphertext: bytes, cdn_base_url: str, upload_param: str, filekey: str, ) -> str: url = _cdn_upload_url(cdn_base_url, upload_param, filekey) timeout = aiohttp.ClientTimeout(total=120) async with session.post(url, data=ciphertext, headers={"Content-Type": "application/octet-stream"}, timeout=timeout) as response: if response.status == 200: encrypted_param = response.headers.get("x-encrypted-param") if encrypted_param: await response.read() return encrypted_param raw = await response.text() raise RuntimeError(f"CDN upload missing x-encrypted-param header: {raw[:200]}") raw = await response.text() raise RuntimeError(f"CDN upload HTTP {response.status}: {raw[:200]}") async def _download_bytes( session: "aiohttp.ClientSession", *, url: str, timeout_seconds: float = 60.0, ) -> bytes: timeout = aiohttp.ClientTimeout(total=timeout_seconds) async with session.get(url, timeout=timeout) as response: response.raise_for_status() return await response.read() def _media_reference(item: Dict[str, Any], key: str) -> Dict[str, Any]: return (item.get(key) or {}).get("media") or {} async def _download_and_decrypt_media( session: "aiohttp.ClientSession", *, cdn_base_url: str, encrypted_query_param: Optional[str], aes_key_b64: Optional[str], full_url: Optional[str], timeout_seconds: float, ) -> bytes: if encrypted_query_param: raw = await _download_bytes( session, url=_cdn_download_url(cdn_base_url, encrypted_query_param), timeout_seconds=timeout_seconds, ) elif full_url: raw = await _download_bytes(session, url=full_url, timeout_seconds=timeout_seconds) else: raise RuntimeError("media item had neither encrypt_query_param nor full_url") if aes_key_b64: raw = _aes128_ecb_decrypt(raw, _parse_aes_key(aes_key_b64)) return raw def _mime_from_filename(filename: str) -> str: return mimetypes.guess_type(filename)[0] or "application/octet-stream" def _split_table_row(line: str) -> List[str]: row = line.strip() if row.startswith("|"): row = row[1:] if row.endswith("|"): row = row[:-1] return [cell.strip() for cell in row.split("|")] def _rewrite_headers_for_weixin(line: str) -> str: match = _HEADER_RE.match(line) if not match: return line.rstrip() level = len(match.group(1)) title = match.group(2).strip() if level == 1: return f"【{title}】" return f"**{title}**" def _rewrite_table_block_for_weixin(lines: List[str]) -> str: if len(lines) < 2: return "\n".join(lines) headers = _split_table_row(lines[0]) body_rows = [_split_table_row(line) for line in lines[2:] if line.strip()] if not headers or not body_rows: return "\n".join(lines) formatted_rows: List[str] = [] for row in body_rows: pairs = [] for idx, header in enumerate(headers): if idx >= len(row): break label = header or f"Column {idx + 1}" value = row[idx].strip() if value: pairs.append((label, value)) if not pairs: continue if len(pairs) == 1: label, value = pairs[0] formatted_rows.append(f"- {label}: {value}") continue if len(pairs) == 2: label, value = pairs[0] other_label, other_value = pairs[1] formatted_rows.append(f"- {label}: {value}") formatted_rows.append(f" {other_label}: {other_value}") continue summary = " | ".join(f"{label}: {value}" for label, value in pairs) formatted_rows.append(f"- {summary}") return "\n".join(formatted_rows) if formatted_rows else "\n".join(lines) def _normalize_markdown_blocks(content: str) -> str: lines = content.splitlines() result: List[str] = [] i = 0 in_code_block = False while i < len(lines): line = lines[i].rstrip() fence_match = _FENCE_RE.match(line.strip()) if fence_match: in_code_block = not in_code_block result.append(line) i += 1 continue if in_code_block: result.append(line) i += 1 continue if ( i + 1 < len(lines) and "|" in lines[i] and _TABLE_RULE_RE.match(lines[i + 1].rstrip()) ): table_lines = [lines[i].rstrip(), lines[i + 1].rstrip()] i += 2 while i < len(lines) and "|" in lines[i]: table_lines.append(lines[i].rstrip()) i += 1 result.append(_rewrite_table_block_for_weixin(table_lines)) continue result.append(_rewrite_headers_for_weixin(line)) i += 1 normalized = "\n".join(item.rstrip() for item in result) normalized = re.sub(r"\n{3,}", "\n\n", normalized) return normalized.strip() def _split_markdown_blocks(content: str) -> List[str]: if not content: return [] blocks: List[str] = [] lines = content.splitlines() current: List[str] = [] in_code_block = False for raw_line in lines: line = raw_line.rstrip() if _FENCE_RE.match(line.strip()): if not in_code_block and current: blocks.append("\n".join(current).strip()) current = [] current.append(line) in_code_block = not in_code_block if not in_code_block: blocks.append("\n".join(current).strip()) current = [] continue if in_code_block: current.append(line) continue if not line.strip(): if current: blocks.append("\n".join(current).strip()) current = [] continue current.append(line) if current: blocks.append("\n".join(current).strip()) return [block for block in blocks if block] def _split_delivery_units_for_weixin(content: str) -> List[str]: """Split formatted content into chat-friendly delivery units. Weixin can render Markdown, but chat readability is better when top-level line breaks become separate messages. Keep fenced code blocks intact and attach indented continuation lines to the previous top-level line so transformed tables/lists do not get torn apart. """ units: List[str] = [] for block in _split_markdown_blocks(content): if _FENCE_RE.match(block.splitlines()[0].strip()): units.append(block) continue current: List[str] = [] for raw_line in block.splitlines(): line = raw_line.rstrip() if not line.strip(): if current: units.append("\n".join(current).strip()) current = [] continue is_continuation = bool(current) and raw_line.startswith((" ", "\t")) if is_continuation: current.append(line) continue if current: units.append("\n".join(current).strip()) current = [line] if current: units.append("\n".join(current).strip()) return [unit for unit in units if unit] def _pack_markdown_blocks_for_weixin(content: str, max_length: int) -> List[str]: if len(content) <= max_length: return [content] packed: List[str] = [] current = "" for block in _split_markdown_blocks(content): candidate = block if not current else f"{current}\n\n{block}" if len(candidate) <= max_length: current = candidate continue if current: packed.append(current) current = "" if len(block) <= max_length: current = block continue packed.extend(BasePlatformAdapter.truncate_message(block, max_length)) if current: packed.append(current) return packed def _split_text_for_weixin_delivery( content: str, max_length: int, split_per_line: bool = False, ) -> List[str]: """Split content into sequential Weixin messages. *compact* (default): Keep everything in a single message whenever it fits within the platform limit, even when the author used explicit line breaks. Only fall back to block-aware packing when the payload exceeds ``max_length``. *per_line* (``split_per_line=True``): Legacy behavior — top-level line breaks become separate chat messages; oversized units still use block-aware packing. The active mode is controlled via ``config.yaml`` -> ``platforms.weixin.extra.split_multiline_messages`` (``true`` / ``false``) or the env var ``WEIXIN_SPLIT_MULTILINE_MESSAGES``. """ if split_per_line: # Legacy: one message per top-level delivery unit. if len(content) <= max_length and "\n" not in content: return [content] chunks: List[str] = [] for unit in _split_delivery_units_for_weixin(content): if len(unit) <= max_length: chunks.append(unit) continue chunks.extend(_pack_markdown_blocks_for_weixin(unit, max_length)) return chunks or [content] # Compact (default): single message when under the limit. if len(content) <= max_length: return [content] return _pack_markdown_blocks_for_weixin(content, max_length) or [content] def _coerce_bool(value: Any, default: bool = True) -> bool: """Coerce a config value to bool, tolerating strings like ``"true"``.""" if value is None: return default if isinstance(value, bool): return value if isinstance(value, (int, float)): return bool(value) text = str(value).strip().lower() if not text: return default if text in {"1", "true", "yes", "on"}: return True if text in {"0", "false", "no", "off"}: return False return default def _extract_text(item_list: List[Dict[str, Any]]) -> str: for item in item_list: if item.get("type") == ITEM_TEXT: text = str((item.get("text_item") or {}).get("text") or "") ref = item.get("ref_msg") or {} ref_item = ref.get("message_item") or {} ref_type = ref_item.get("type") if ref_type in (ITEM_IMAGE, ITEM_VIDEO, ITEM_FILE, ITEM_VOICE): title = ref.get("title") or "" prefix = f"[引用媒体: {title}]\n" if title else "[引用媒体]\n" return f"{prefix}{text}".strip() if ref_item: parts: List[str] = [] if ref.get("title"): parts.append(str(ref["title"])) ref_text = _extract_text([ref_item]) if ref_text: parts.append(ref_text) if parts: return f"[引用: {' | '.join(parts)}]\n{text}".strip() return text for item in item_list: if item.get("type") == ITEM_VOICE: voice_text = str((item.get("voice_item") or {}).get("text") or "") if voice_text: return voice_text return "" def _message_type_from_media(media_types: List[str], text: str) -> MessageType: if any(m.startswith("image/") for m in media_types): return MessageType.PHOTO if any(m.startswith("video/") for m in media_types): return MessageType.VIDEO if any(m.startswith("audio/") for m in media_types): return MessageType.VOICE if media_types: return MessageType.DOCUMENT if text.startswith("/"): return MessageType.COMMAND return MessageType.TEXT def _sync_buf_path(hermes_home: str, account_id: str) -> Path: return _account_dir(hermes_home) / f"{account_id}.sync.json" def _load_sync_buf(hermes_home: str, account_id: str) -> str: path = _sync_buf_path(hermes_home, account_id) if not path.exists(): return "" try: return json.loads(path.read_text(encoding="utf-8")).get("get_updates_buf", "") except Exception: return "" def _save_sync_buf(hermes_home: str, account_id: str, sync_buf: str) -> None: path = _sync_buf_path(hermes_home, account_id) atomic_json_write(path, {"get_updates_buf": sync_buf}) async def qr_login( hermes_home: str, *, bot_type: str = "3", timeout_seconds: int = 480, ) -> Optional[Dict[str, str]]: """ Run the interactive iLink QR login flow. Returns a credential dict on success, or ``None`` if login fails or times out. """ if not AIOHTTP_AVAILABLE: raise RuntimeError("aiohttp is required for Weixin QR login") async with aiohttp.ClientSession() as session: try: qr_resp = await _api_get( session, base_url=ILINK_BASE_URL, endpoint=f"{EP_GET_BOT_QR}?bot_type={bot_type}", timeout_ms=QR_TIMEOUT_MS, ) except Exception as exc: logger.error("weixin: failed to fetch QR code: %s", exc) return None qrcode_value = str(qr_resp.get("qrcode") or "") qrcode_url = str(qr_resp.get("qrcode_img_content") or "") if not qrcode_value: logger.error("weixin: QR response missing qrcode") return None print("\n请使用微信扫描以下二维码：") if qrcode_url: print(qrcode_url) try: import qrcode qr = qrcode.QRCode() qr.add_data(qrcode_url or qrcode_value) qr.make(fit=True) qr.print_ascii(invert=True) except Exception: print("（终端二维码渲染失败，请直接打开上面的二维码链接）") deadline = time.time() + timeout_seconds current_base_url = ILINK_BASE_URL refresh_count = 0 while time.time() < deadline: try: status_resp = await _api_get( session, base_url=current_base_url, endpoint=f"{EP_GET_QR_STATUS}?qrcode={qrcode_value}", timeout_ms=QR_TIMEOUT_MS, ) except asyncio.TimeoutError: await asyncio.sleep(1) continue except Exception as exc: logger.warning("weixin: QR poll error: %s", exc) await asyncio.sleep(1) continue status = str(status_resp.get("status") or "wait") if status == "wait": print(".", end="", flush=True) elif status == "scaned": print("\n已扫码，请在微信里确认...") elif status == "scaned_but_redirect": redirect_host = str(status_resp.get("redirect_host") or "") if redirect_host: current_base_url = f"https://{redirect_host}" elif status == "expired": refresh_count += 1 if refresh_count > 3: print("\n二维码多次过期，请重新执行登录。") return None print(f"\n二维码已过期，正在刷新... ({refresh_count}/3)") try: qr_resp = await _api_get( session, base_url=ILINK_BASE_URL, endpoint=f"{EP_GET_BOT_QR}?bot_type={bot_type}", timeout_ms=QR_TIMEOUT_MS, ) qrcode_value = str(qr_resp.get("qrcode") or "") qrcode_url = str(qr_resp.get("qrcode_img_content") or "") if qrcode_url: print(qrcode_url) except Exception as exc: logger.error("weixin: QR refresh failed: %s", exc) return None elif status == "confirmed": account_id = str(status_resp.get("ilink_bot_id") or "") token = str(status_resp.get("bot_token") or "") base_url = str(status_resp.get("baseurl") or ILINK_BASE_URL) user_id = str(status_resp.get("ilink_user_id") or "") if not account_id or not token: logger.error("weixin: QR confirmed but credential payload was incomplete") return None save_weixin_account( hermes_home, account_id=account_id, token=token, base_url=base_url, user_id=user_id, ) print(f"\n微信连接成功，account_id={account_id}") return { "account_id": account_id, "token": token, "base_url": base_url, "user_id": user_id, } await asyncio.sleep(1) print("\n微信登录超时。") return None class WeixinAdapter(BasePlatformAdapter): """Native Hermes adapter for Weixin personal accounts.""" MAX_MESSAGE_LENGTH = 4000 def __init__(self, config: PlatformConfig): super().__init__(config, Platform.WEIXIN) extra = config.extra or {} hermes_home = str(get_hermes_home()) self._hermes_home = hermes_home self._token_store = ContextTokenStore(hermes_home) self._typing_cache = TypingTicketCache() self._session: Optional[aiohttp.ClientSession] = None self._poll_task: Optional[asyncio.Task] = None self._dedup = MessageDeduplicator(ttl_seconds=MESSAGE_DEDUP_TTL_SECONDS) self._account_id = str(extra.get("account_id") or os.getenv("WEIXIN_ACCOUNT_ID", "")).strip() self._token = str(config.token or extra.get("token") or os.getenv("WEIXIN_TOKEN", "")).strip() self._base_url = str(extra.get("base_url") or os.getenv("WEIXIN_BASE_URL", ILINK_BASE_URL)).strip().rstrip("/") self._cdn_base_url = str( extra.get("cdn_base_url") or os.getenv("WEIXIN_CDN_BASE_URL", WEIXIN_CDN_BASE_URL) ).strip().rstrip("/") self._dm_policy = str(extra.get("dm_policy") or os.getenv("WEIXIN_DM_POLICY", "open")).strip().lower() self._group_policy = str(extra.get("group_policy") or os.getenv("WEIXIN_GROUP_POLICY", "disabled")).strip().lower() allow_from = extra.get("allow_from") if allow_from is None: allow_from = os.getenv("WEIXIN_ALLOWED_USERS", "") group_allow_from = extra.get("group_allow_from") if group_allow_from is None: group_allow_from = os.getenv("WEIXIN_GROUP_ALLOWED_USERS", "") self._allow_from = self._coerce_list(allow_from) self._group_allow_from = self._coerce_list(group_allow_from) self._split_multiline_messages = _coerce_bool( extra.get("split_multiline_messages") or os.getenv("WEIXIN_SPLIT_MULTILINE_MESSAGES"), default=False, ) if self._account_id and not self._token: persisted = load_weixin_account(hermes_home, self._account_id) if persisted: self._token = str(persisted.get("token") or "").strip() self._base_url = str(persisted.get("base_url") or self._base_url).strip().rstrip("/") @staticmethod def _coerce_list(value: Any) -> List[str]: if value is None: return [] if isinstance(value, str): return [item.strip() for item in value.split(",") if item.strip()] if isinstance(value, (list, tuple, set)): return [str(item).strip() for item in value if str(item).strip()] return [str(value).strip()] if str(value).strip() else [] async def connect(self) -> bool: if not check_weixin_requirements(): message = "Weixin startup failed: aiohttp and cryptography are required" self._set_fatal_error("weixin_missing_dependency", message, retryable=False) logger.warning("[%s] %s", self.name, message) return False if not self._token: message = "Weixin startup failed: WEIXIN_TOKEN is required" self._set_fatal_error("weixin_missing_token", message, retryable=False) logger.warning("[%s] %s", self.name, message) return False if not self._account_id: message = "Weixin startup failed: WEIXIN_ACCOUNT_ID is required" self._set_fatal_error("weixin_missing_account", message, retryable=False) logger.warning("[%s] %s", self.name, message) return False try: if not self._acquire_platform_lock('weixin-bot-token', self._token, 'Weixin bot token'): return False except Exception as exc: logger.debug("[%s] Token lock unavailable (non-fatal): %s", self.name, exc) self._session = aiohttp.ClientSession() self._token_store.restore(self._account_id) self._poll_task = asyncio.create_task(self._poll_loop(), name="weixin-poll") self._mark_connected() logger.info("[%s] Connected account=%s base=%s", self.name, _safe_id(self._account_id), self._base_url) return True async def disconnect(self) -> None: self._running = False if self._poll_task and not self._poll_task.done(): self._poll_task.cancel() try: await self._poll_task except asyncio.CancelledError: pass self._poll_task = None if self._session and not self._session.closed: await self._session.close() self._session = None self._release_platform_lock() self._mark_disconnected() logger.info("[%s] Disconnected", self.name) async def _poll_loop(self) -> None: assert self._session is not None sync_buf = _load_sync_buf(self._hermes_home, self._account_id) timeout_ms = LONG_POLL_TIMEOUT_MS consecutive_failures = 0 while self._running: try: response = await _get_updates( self._session, base_url=self._base_url, token=self._token, sync_buf=sync_buf, timeout_ms=timeout_ms, ) suggested_timeout = response.get("longpolling_timeout_ms") if isinstance(suggested_timeout, int) and suggested_timeout > 0: timeout_ms = suggested_timeout ret = response.get("ret", 0) errcode = response.get("errcode", 0) if ret not in (0, None) or errcode not in (0, None): if ret == SESSION_EXPIRED_ERRCODE or errcode == SESSION_EXPIRED_ERRCODE: logger.error("[%s] Session expired; pausing for 10 minutes", self.name) await asyncio.sleep(600) consecutive_failures = 0 continue consecutive_failures += 1 logger.warning( "[%s] getUpdates failed ret=%s errcode=%s errmsg=%s (%d/%d)", self.name, ret, errcode, response.get("errmsg", ""), consecutive_failures, MAX_CONSECUTIVE_FAILURES, ) await asyncio.sleep(BACKOFF_DELAY_SECONDS if consecutive_failures >= MAX_CONSECUTIVE_FAILURES else RETRY_DELAY_SECONDS) if consecutive_failures >= MAX_CONSECUTIVE_FAILURES: consecutive_failures = 0 continue consecutive_failures = 0 new_sync_buf = str(response.get("get_updates_buf") or "") if new_sync_buf: sync_buf = new_sync_buf _save_sync_buf(self._hermes_home, self._account_id, sync_buf) for message in response.get("msgs") or []: asyncio.create_task(self._process_message_safe(message)) except asyncio.CancelledError: break except Exception as exc: consecutive_failures += 1 logger.error("[%s] poll error (%d/%d): %s", self.name, consecutive_failures, MAX_CONSECUTIVE_FAILURES, exc) await asyncio.sleep(BACKOFF_DELAY_SECONDS if consecutive_failures >= MAX_CONSECUTIVE_FAILURES else RETRY_DELAY_SECONDS) if consecutive_failures >= MAX_CONSECUTIVE_FAILURES: consecutive_failures = 0 async def _process_message_safe(self, message: Dict[str, Any]) -> None: try: await self._process_message(message) except Exception as exc: logger.error("[%s] unhandled inbound error from=%s: %s", self.name, _safe_id(message.get("from_user_id")), exc, exc_info=True) async def _process_message(self, message: Dict[str, Any]) -> None: assert self._session is not None sender_id = str(message.get("from_user_id") or "").strip() if not sender_id: return if sender_id == self._account_id: return message_id = str(message.get("message_id") or "").strip() if message_id and self._dedup.is_duplicate(message_id): return chat_type, effective_chat_id = _guess_chat_type(message, self._account_id) if chat_type == "group": if self._group_policy == "disabled": return if self._group_policy == "allowlist" and effective_chat_id not in self._group_allow_from: return elif not self._is_dm_allowed(sender_id): return context_token = str(message.get("context_token") or "").strip() if context_token: self._token_store.set(self._account_id, sender_id, context_token) asyncio.create_task(self._maybe_fetch_typing_ticket(sender_id, context_token or None)) item_list = message.get("item_list") or [] text = _extract_text(item_list) media_paths: List[str] = [] media_types: List[str] = [] for item in item_list: await self._collect_media(item, media_paths, media_types) ref_message = item.get("ref_msg") or {} ref_item = ref_message.get("message_item") if isinstance(ref_item, dict): await self._collect_media(ref_item, media_paths, media_types) if not text and not media_paths: return source = self.build_source( chat_id=effective_chat_id, chat_type=chat_type, user_id=sender_id, user_name=sender_id, ) event = MessageEvent( text=text, message_type=_message_type_from_media(media_types, text), source=source, raw_message=message, message_id=message_id or None, media_urls=media_paths, media_types=media_types, timestamp=datetime.now(), ) logger.info("[%s] inbound from=%s type=%s media=%d", self.name, _safe_id(sender_id), source.chat_type, len(media_paths)) await self.handle_message(event) def _is_dm_allowed(self, sender_id: str) -> bool: if self._dm_policy == "disabled": return False if self._dm_policy == "allowlist": return sender_id in self._allow_from return True async def _collect_media(self, item: Dict[str, Any], media_paths: List[str], media_types: List[str]) -> None: item_type = item.get("type") if item_type == ITEM_IMAGE: path = await self._download_image(item) if path: media_paths.append(path) media_types.append("image/jpeg") elif item_type == ITEM_VIDEO: path = await self._download_video(item) if path: media_paths.append(path) media_types.append("video/mp4") elif item_type == ITEM_FILE: path, mime = await self._download_file(item) if path: media_paths.append(path) media_types.append(mime) elif item_type == ITEM_VOICE: voice_path = await self._download_voice(item) if voice_path: media_paths.append(voice_path) media_types.append("audio/silk") async def _download_image(self, item: Dict[str, Any]) -> Optional[str]: media = _media_reference(item, "image_item") try: data = await _download_and_decrypt_media( self._session, cdn_base_url=self._cdn_base_url, encrypted_query_param=media.get("encrypt_query_param"), aes_key_b64=(item.get("image_item") or {}).get("aeskey") and base64.b64encode(bytes.fromhex(str((item.get("image_item") or {}).get("aeskey")))).decode("ascii") or media.get("aes_key"), full_url=media.get("full_url"), timeout_seconds=30.0, ) return cache_image_from_bytes(data, ".jpg") except Exception as exc: logger.warning("[%s] image download failed: %s", self.name, exc) return None async def _download_video(self, item: Dict[str, Any]) -> Optional[str]: media = _media_reference(item, "video_item") try: data = await _download_and_decrypt_media( self._session, cdn_base_url=self._cdn_base_url, encrypted_query_param=media.get("encrypt_query_param"), aes_key_b64=media.get("aes_key"), full_url=media.get("full_url"), timeout_seconds=120.0, ) return cache_document_from_bytes(data, "video.mp4") except Exception as exc: logger.warning("[%s] video download failed: %s", self.name, exc) return None async def _download_file(self, item: Dict[str, Any]) -> Tuple[Optional[str], str]: file_item = item.get("file_item") or {} media = file_item.get("media") or {} filename = str(file_item.get("file_name") or "document.bin") mime = _mime_from_filename(filename) try: data = await _download_and_decrypt_media( self._session, cdn_base_url=self._cdn_base_url, encrypted_query_param=media.get("encrypt_query_param"), aes_key_b64=media.get("aes_key"), full_url=media.get("full_url"), timeout_seconds=60.0, ) return cache_document_from_bytes(data, filename), mime except Exception as exc: logger.warning("[%s] file download failed: %s", self.name, exc) return None, mime async def _download_voice(self, item: Dict[str, Any]) -> Optional[str]: voice_item = item.get("voice_item") or {} media = voice_item.get("media") or {} if voice_item.get("text"): return None try: data = await _download_and_decrypt_media( self._session, cdn_base_url=self._cdn_base_url, encrypted_query_param=media.get("encrypt_query_param"), aes_key_b64=media.get("aes_key"), full_url=media.get("full_url"), timeout_seconds=60.0, ) return cache_audio_from_bytes(data, ".silk") except Exception as exc: logger.warning("[%s] voice download failed: %s", self.name, exc) return None async def _maybe_fetch_typing_ticket(self, user_id: str, context_token: Optional[str]) -> None: if not self._session or not self._token: return if self._typing_cache.get(user_id): return try: response = await _get_config( self._session, base_url=self._base_url, token=self._token, user_id=user_id, context_token=context_token, ) typing_ticket = str(response.get("typing_ticket") or "") if typing_ticket: self._typing_cache.set(user_id, typing_ticket) except Exception as exc: logger.debug("[%s] getConfig failed for %s: %s", self.name, _safe_id(user_id), exc) def _split_text(self, content: str) -> List[str]: return _split_text_for_weixin_delivery( content, self.MAX_MESSAGE_LENGTH, self._split_multiline_messages, ) async def send( self, chat_id: str, content: str, reply_to: Optional[str] = None, metadata: Optional[Dict[str, Any]] = None, ) -> SendResult: if not self._session or not self._token: return SendResult(success=False, error="Not connected") context_token = self._token_store.get(self._account_id, chat_id) last_message_id: Optional[str] = None try: chunks = self._split_text(self.format_message(content)) for idx, chunk in enumerate(chunks): if idx > 0: await asyncio.sleep(0.3) client_id = f"hermes-weixin-{uuid.uuid4().hex}" await _send_message( self._session, base_url=self._base_url, token=self._token, to=chat_id, text=chunk, context_token=context_token, client_id=client_id, ) last_message_id = client_id return SendResult(success=True, message_id=last_message_id) except Exception as exc: logger.error("[%s] send failed to=%s: %s", self.name, _safe_id(chat_id), exc) return SendResult(success=False, error=str(exc)) async def send_typing(self, chat_id: str, metadata: Optional[Dict[str, Any]] = None) -> None: if not self._session or not self._token: return typing_ticket = self._typing_cache.get(chat_id) if not typing_ticket: return try: await _send_typing( self._session, base_url=self._base_url, token=self._token, to_user_id=chat_id, typing_ticket=typing_ticket, status=TYPING_START, ) except Exception as exc: logger.debug("[%s] typing start failed for %s: %s", self.name, _safe_id(chat_id), exc) async def stop_typing(self, chat_id: str) -> None: if not self._session or not self._token: return typing_ticket = self._typing_cache.get(chat_id) if not typing_ticket: return try: await _send_typing( self._session, base_url=self._base_url, token=self._token, to_user_id=chat_id, typing_ticket=typing_ticket, status=TYPING_STOP, ) except Exception as exc: logger.debug("[%s] typing stop failed for %s: %s", self.name, _safe_id(chat_id), exc) async def send_image( self, chat_id: str, image_url: str, caption: str, reply_to: Optional[str] = None, metadata: Optional[Dict[str, Any]] = None, ) -> SendResult: if image_url.startswith(("http://", "https://")): file_path = await self._download_remote_media(image_url) cleanup = True else: file_path = image_url.replace("file://", "") if not os.path.isabs(file_path): file_path = os.path.abspath(file_path) cleanup = False try: return await self.send_document(chat_id, file_path, caption=caption, metadata=metadata) finally: if cleanup and file_path and os.path.exists(file_path): try: os.unlink(file_path) except OSError: pass async def send_image_file( self, chat_id: str, path: str, caption: str = "", reply_to: Optional[str] = None, metadata: Optional[Dict[str, Any]] = None, ) -> SendResult: return await self.send_document(chat_id, path, caption=caption, metadata=metadata) async def send_document( self, chat_id: str, path: str, caption: str = "", metadata: Optional[Dict[str, Any]] = None, ) -> SendResult: if not self._session or not self._token: return SendResult(success=False, error="Not connected") try: message_id = await self._send_file(chat_id, path, caption) return SendResult(success=True, message_id=message_id) except Exception as exc: logger.error("[%s] send_document failed to=%s: %s", self.name, _safe_id(chat_id), exc) return SendResult(success=False, error=str(exc)) async def _download_remote_media(self, url: str) -> str: from tools.url_safety import is_safe_url if not is_safe_url(url): raise ValueError(f"Blocked unsafe URL (SSRF protection): {url}") assert self._session is not None async with self._session.get(url, timeout=aiohttp.ClientTimeout(total=30)) as response: response.raise_for_status() data = await response.read() suffix = Path(url.split("?", 1)[0]).suffix or ".bin" with tempfile.NamedTemporaryFile(delete=False, suffix=suffix) as handle: handle.write(data) return handle.name async def _send_file(self, chat_id: str, path: str, caption: str) -> str: assert self._session is not None and self._token is not None plaintext = Path(path).read_bytes() media_type, item_builder = self._outbound_media_builder(path) filekey = secrets.token_hex(16) aes_key = secrets.token_bytes(16) rawsize = len(plaintext) upload_response = await _get_upload_url( self._session, base_url=self._base_url, token=self._token, to_user_id=chat_id, media_type=media_type, filekey=filekey, rawsize=rawsize, rawfilemd5=hashlib.md5(plaintext).hexdigest(), filesize=_aes_padded_size(rawsize), aeskey_hex=aes_key.hex(), ) upload_param = str(upload_response.get("upload_param") or "") upload_full_url = str(upload_response.get("upload_full_url") or "") ciphertext = _aes128_ecb_encrypt(plaintext, aes_key) if upload_param: encrypted_query_param = await _upload_ciphertext( self._session, ciphertext=ciphertext, cdn_base_url=self._cdn_base_url, upload_param=upload_param, filekey=filekey, ) elif upload_full_url: timeout = aiohttp.ClientTimeout(total=120) async with self._session.put( upload_full_url, data=ciphertext, headers={"Content-Type": "application/octet-stream"}, timeout=timeout, ) as response: response.raise_for_status() encrypted_query_param = response.headers.get("x-encrypted-param") or filekey else: raise RuntimeError(f"getUploadUrl returned neither upload_param nor upload_full_url: {upload_response}") context_token = self._token_store.get(self._account_id, chat_id) media_item = item_builder( encrypt_query_param=encrypted_query_param, aes_key_b64=base64.b64encode(aes_key).decode("ascii"), ciphertext_size=len(ciphertext), plaintext_size=rawsize, filename=Path(path).name, ) last_message_id = None if caption: last_message_id = f"hermes-weixin-{uuid.uuid4().hex}" await _send_message( self._session, base_url=self._base_url, token=self._token, to=chat_id, text=self.format_message(caption), context_token=context_token, client_id=last_message_id, ) last_message_id = f"hermes-weixin-{uuid.uuid4().hex}" await _api_post( self._session, base_url=self._base_url, endpoint=EP_SEND_MESSAGE, payload={ "msg": { "from_user_id": "", "to_user_id": chat_id, "client_id": last_message_id, "message_type": MSG_TYPE_BOT, "message_state": MSG_STATE_FINISH, "item_list": [media_item], **({"context_token": context_token} if context_token else {}), } }, token=self._token, timeout_ms=API_TIMEOUT_MS, ) return last_message_id def _outbound_media_builder(self, path: str): mime = mimetypes.guess_type(path)[0] or "application/octet-stream" if mime.startswith("image/"): return MEDIA_IMAGE, lambda **kwargs: { "type": ITEM_IMAGE, "image_item": { "media": { "encrypt_query_param": kwargs["encrypt_query_param"], "aes_key": kwargs["aes_key_b64"], "encrypt_type": 1, }, "mid_size": kwargs["ciphertext_size"], }, } if mime.startswith("video/"): return MEDIA_VIDEO, lambda **kwargs: { "type": ITEM_VIDEO, "video_item": { "media": { "encrypt_query_param": kwargs["encrypt_query_param"], "aes_key": kwargs["aes_key_b64"], "encrypt_type": 1, }, "video_size": kwargs["ciphertext_size"], }, } return MEDIA_FILE, lambda **kwargs: { "type": ITEM_FILE, "file_item": { "media": { "encrypt_query_param": kwargs["encrypt_query_param"], "aes_key": kwargs["aes_key_b64"], "encrypt_type": 1, }, "file_name": kwargs["filename"], "len": str(kwargs["plaintext_size"]), }, } async def get_chat_info(self, chat_id: str) -> Dict[str, Any]: chat_type = "group" if chat_id.endswith("@chatroom") else "dm" return {"name": chat_id, "type": chat_type, "chat_id": chat_id} def format_message(self, content: Optional[str]) -> str: if content is None: return "" return _normalize_markdown_blocks(content) async def send_weixin_direct( *, extra: Dict[str, Any], token: Optional[str], chat_id: str, message: str, media_files: Optional[List[Tuple[str, bool]]] = None, ) -> Dict[str, Any]: """ One-shot send helper for ``send_message`` and cron delivery. This bypasses the long-poll adapter lifecycle and uses the raw API directly. """ account_id = str(extra.get("account_id") or os.getenv("WEIXIN_ACCOUNT_ID", "")).strip() base_url = str(extra.get("base_url") or os.getenv("WEIXIN_BASE_URL", ILINK_BASE_URL)).strip().rstrip("/") cdn_base_url = str(extra.get("cdn_base_url") or os.getenv("WEIXIN_CDN_BASE_URL", WEIXIN_CDN_BASE_URL)).strip().rstrip("/") resolved_token = str(token or extra.get("token") or os.getenv("WEIXIN_TOKEN", "")).strip() if not resolved_token: return {"error": "Weixin token missing. Configure WEIXIN_TOKEN or platforms.weixin.token."} if not account_id: return {"error": "Weixin account ID missing. Configure WEIXIN_ACCOUNT_ID or platforms.weixin.extra.account_id."} token_store = ContextTokenStore(str(get_hermes_home())) token_store.restore(account_id) context_token = token_store.get(account_id, chat_id) async with aiohttp.ClientSession() as session: adapter = WeixinAdapter( PlatformConfig( enabled=True, token=resolved_token, extra={ **dict(extra or {}), "account_id": account_id, "base_url": base_url, "cdn_base_url": cdn_base_url, }, ) ) adapter._session = session adapter._token = resolved_token adapter._account_id = account_id adapter._base_url = base_url adapter._cdn_base_url = cdn_base_url adapter._token_store = token_store last_result: Optional[SendResult] = None cleaned = adapter.format_message(message) if cleaned: last_result = await adapter.send(chat_id, cleaned) if not last_result.success: return {"error": f"Weixin send failed: {last_result.error}"} for media_path, _is_voice in media_files or []: ext = Path(media_path).suffix.lower() if ext in {".jpg", ".jpeg", ".png", ".gif", ".webp", ".bmp"}: last_result = await adapter.send_image_file(chat_id, media_path) else: last_result = await adapter.send_document(chat_id, media_path) if not last_result.success: return {"error": f"Weixin media send failed: {last_result.error}"} return { "success": True, "platform": "weixin", "chat_id": chat_id, "message_id": last_result.message_id if last_result else None, "context_token_used": bool(context_token), }