mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-05-12 10:57:16 +08:00
2217 lines
86 KiB
Python
2217 lines
86 KiB
Python
import base64
|
||
import hashlib
|
||
import json
|
||
import mimetypes
|
||
import os
|
||
import random
|
||
import re
|
||
import threading
|
||
import time
|
||
import uuid
|
||
from dataclasses import dataclass, field
|
||
from pathlib import Path
|
||
from typing import Any, Dict, List, Optional, Tuple
|
||
from urllib.parse import quote
|
||
|
||
from Crypto.Cipher import AES
|
||
from Crypto.Util.Padding import pad
|
||
|
||
from app.core.cache import FileCache
|
||
from app.core.config import settings
|
||
from app.core.context import Context, MediaInfo
|
||
from app.core.metainfo import MetaInfo
|
||
from app.log import logger
|
||
from app.utils.http import RequestUtils
|
||
from app.utils.string import StringUtils
|
||
|
||
|
||
@dataclass
|
||
class ILinkIncomingMessage:
|
||
"""iLink 归一化后的入站消息。"""
|
||
|
||
user_id: str
|
||
text: Optional[str] = None
|
||
username: Optional[str] = None
|
||
message_id: Optional[str] = None
|
||
chat_id: Optional[str] = None
|
||
context_token: Optional[str] = None
|
||
images: List[Dict[str, Any]] = field(default_factory=list)
|
||
audio_refs: List[str] = field(default_factory=list)
|
||
files: List[Dict[str, Any]] = field(default_factory=list)
|
||
raw: Dict[str, Any] = field(default_factory=dict)
|
||
|
||
def to_message_payload(self) -> Dict[str, Any]:
|
||
payload = {
|
||
"__channel__": "wechatclawbot",
|
||
"userid": self.user_id,
|
||
"username": self.username or self.user_id,
|
||
"text": self.text or "",
|
||
"message_id": self.message_id,
|
||
"chat_id": self.chat_id,
|
||
"context_token": self.context_token,
|
||
"images": self.images or None,
|
||
"audio_refs": self.audio_refs or None,
|
||
"files": self.files or None,
|
||
}
|
||
return {key: value for key, value in payload.items() if value is not None}
|
||
|
||
|
||
class ILinkClient:
|
||
"""iLink HTTP 客户端，负责二维码登录、消息发送与长轮询。"""
|
||
|
||
channel_version = "1.0.2"
|
||
cdn_base_url = "https://novac2c.cdn.weixin.qq.com/c2c"
|
||
|
||
def __init__(
|
||
self,
|
||
base_url: str,
|
||
bot_token: Optional[str] = None,
|
||
account_id: Optional[str] = None,
|
||
sync_buf: Optional[str] = None,
|
||
timeout: int = 20,
|
||
):
|
||
"""保存 iLink 会话参数，供二维码登录、长轮询和消息发送复用。"""
|
||
self.base_url = (base_url or "https://ilinkai.weixin.qq.com").rstrip("/")
|
||
self.bot_token = bot_token
|
||
self.account_id = account_id
|
||
self.sync_buf = sync_buf
|
||
self.timeout = timeout
|
||
|
||
def set_credentials(
|
||
self,
|
||
bot_token: Optional[str],
|
||
account_id: Optional[str] = None,
|
||
sync_buf: Optional[str] = None,
|
||
) -> None:
|
||
"""更新 iLink 登录凭证与会话状态。"""
|
||
self.bot_token = bot_token
|
||
self.account_id = account_id
|
||
if sync_buf is not None:
|
||
self.sync_buf = sync_buf
|
||
|
||
def _headers(self, auth_required: bool = True) -> Dict[str, str]:
|
||
"""构建请求头，包含必要的身份验证信息。"""
|
||
headers = {
|
||
"Content-Type": "application/json",
|
||
"Accept": "application/json, text/plain, */*",
|
||
"User-Agent": "MoviePilot-WechatClawBot/1.0",
|
||
}
|
||
if auth_required and self.bot_token:
|
||
headers["AuthorizationType"] = "ilink_bot_token"
|
||
headers["Authorization"] = f"Bearer {self.bot_token}"
|
||
headers["X-WECHAT-UIN"] = self._build_wechat_uin()
|
||
return headers
|
||
|
||
@staticmethod
|
||
def _build_wechat_uin() -> str:
|
||
"""生成一个随机的微信 UIN，用于标识请求来源。"""
|
||
random_u32 = random.getrandbits(32)
|
||
return base64.b64encode(str(random_u32).encode("utf-8")).decode("ascii")
|
||
|
||
def _with_base_info(self, body: Optional[Dict[str, Any]]) -> Dict[str, Any]:
|
||
"""向请求体中补充基础信息（如版本号）。"""
|
||
payload = dict(body or {})
|
||
base_info = payload.get("base_info")
|
||
if not isinstance(base_info, dict):
|
||
base_info = {}
|
||
base_info.setdefault("channel_version", self.channel_version)
|
||
payload["base_info"] = base_info
|
||
return payload
|
||
|
||
@staticmethod
|
||
def _json(resp: Any) -> Dict[str, Any]:
|
||
"""尝试将响应解析为 JSON，处理异常情况。"""
|
||
if not resp:
|
||
return {}
|
||
try:
|
||
return resp.json() or {}
|
||
except Exception:
|
||
text = (getattr(resp, "text", "") or "").strip()
|
||
if not text:
|
||
return {}
|
||
try:
|
||
return json.loads(text)
|
||
except Exception:
|
||
return {}
|
||
|
||
@staticmethod
|
||
def _short_text(value: Any, max_len: int = 240) -> str:
|
||
"""将对象转换为简短的字符串描述，用于日志记录。"""
|
||
if value is None:
|
||
return ""
|
||
if isinstance(value, (dict, list)):
|
||
try:
|
||
text = json.dumps(value, ensure_ascii=False)
|
||
except Exception:
|
||
text = str(value)
|
||
else:
|
||
text = str(value)
|
||
text = text.replace("\n", " ").replace("\r", " ").strip()
|
||
if len(text) > max_len:
|
||
return f"{text[:max_len]}..."
|
||
return text
|
||
|
||
@staticmethod
|
||
def _normalize_qrcode_url(value: Any) -> Optional[str]:
|
||
"""规范化二维码展示字段，兼容图片 URL、data URL 与裸 base64。"""
|
||
if value is None:
|
||
return None
|
||
raw = str(value).strip()
|
||
if not raw:
|
||
return None
|
||
lowered = raw.lower()
|
||
if lowered.startswith("data:image/"):
|
||
return raw
|
||
if lowered.startswith("//"):
|
||
return f"https:{raw}"
|
||
# 某些实现会直接返回二维码图片的 base64 内容，这里自动补齐 data URL 前缀
|
||
if len(raw) >= 128 and re.fullmatch(r"[A-Za-z0-9+/=_-]+", raw):
|
||
return f"data:image/png;base64,{raw}"
|
||
return raw
|
||
|
||
@staticmethod
|
||
def _pick_value(obj: Dict[str, Any], keys: List[str]) -> Optional[Any]:
|
||
"""从字典中尝试获取指定键中的第一个非空值。"""
|
||
for key in keys:
|
||
if key in obj and obj.get(key) not in (None, ""):
|
||
return obj.get(key)
|
||
return None
|
||
|
||
@staticmethod
|
||
def _pick_present_value(obj: Dict[str, Any], keys: List[str]) -> Optional[Any]:
|
||
"""
|
||
获取已存在的字段值，允许空字符串和 0 作为有效值。
|
||
|
||
sync_buf 这类游标字段可能会被服务端明确置为空串，表示游标已经回到初始状态。
|
||
这里不能复用默认的“忽略空字符串”逻辑，否则会错误沿用上一轮旧游标。
|
||
"""
|
||
for key in keys:
|
||
if key in obj and obj.get(key) is not None:
|
||
return obj.get(key)
|
||
return None
|
||
|
||
@classmethod
|
||
def _find_first_value(
|
||
cls, data: Any, keys: List[str], max_depth: int = 5
|
||
) -> Optional[Any]:
|
||
"""在嵌套结构中递归查找指定键中的第一个非空值。"""
|
||
if max_depth < 0 or data is None:
|
||
return None
|
||
if isinstance(data, dict):
|
||
direct = cls._pick_value(data, keys)
|
||
if direct not in (None, ""):
|
||
return direct
|
||
for value in data.values():
|
||
found = cls._find_first_value(value, keys, max_depth - 1)
|
||
if found not in (None, ""):
|
||
return found
|
||
elif isinstance(data, list):
|
||
for value in data:
|
||
found = cls._find_first_value(value, keys, max_depth - 1)
|
||
if found not in (None, ""):
|
||
return found
|
||
return None
|
||
|
||
@classmethod
|
||
def _find_first_list(
|
||
cls, data: Any, prefer_keys: List[str], max_depth: int = 5
|
||
) -> Optional[List[Any]]:
|
||
"""在嵌套结构中递归查找第一个符合条件的列表。"""
|
||
if max_depth < 0 or data is None:
|
||
return None
|
||
if isinstance(data, dict):
|
||
for key in prefer_keys:
|
||
value = data.get(key)
|
||
if isinstance(value, list):
|
||
return value
|
||
for value in data.values():
|
||
found = cls._find_first_list(value, prefer_keys, max_depth - 1)
|
||
if found is not None:
|
||
return found
|
||
elif isinstance(data, list):
|
||
if data and all(isinstance(item, dict) for item in data):
|
||
return data
|
||
for value in data:
|
||
found = cls._find_first_list(value, prefer_keys, max_depth - 1)
|
||
if found is not None:
|
||
return found
|
||
return None
|
||
|
||
@staticmethod
|
||
def _ok(payload: Dict[str, Any]) -> bool:
|
||
"""检查 API 响应是否表示成功。"""
|
||
if not payload:
|
||
return False
|
||
code = payload.get("errcode")
|
||
if code is None:
|
||
code = payload.get("code")
|
||
if code is None:
|
||
code = payload.get("ret")
|
||
if code is None:
|
||
err = payload.get("errmsg") or payload.get("error") or payload.get(
|
||
"error_msg"
|
||
)
|
||
if err and str(err).strip().lower() not in {"ok", "success", "succeed"}:
|
||
return False
|
||
state = payload.get("status") or payload.get("state")
|
||
if isinstance(state, str) and state.strip().lower() in {
|
||
"error",
|
||
"failed",
|
||
"fail",
|
||
}:
|
||
return False
|
||
return True
|
||
try:
|
||
return int(str(code)) == 0
|
||
except Exception:
|
||
return str(code).strip().lower() in {"0", "ok", "success", "succeed"}
|
||
|
||
def _is_send_success(self, payload: Dict[str, Any]) -> bool:
|
||
"""检查 API 响应中是否包含明确的发送成功信号。"""
|
||
if not payload:
|
||
return False
|
||
code = self._find_first_value(
|
||
payload, ["errcode", "code", "ret", "result_code", "status_code"]
|
||
)
|
||
if code is not None:
|
||
try:
|
||
return int(str(code)) == 0
|
||
except Exception:
|
||
return str(code).strip().lower() in {"0", "ok", "success", "succeed"}
|
||
success_flag = self._find_first_value(
|
||
payload, ["success", "ok", "is_success", "sent"]
|
||
)
|
||
if isinstance(success_flag, bool):
|
||
return success_flag
|
||
if success_flag is not None:
|
||
if str(success_flag).strip().lower() in {
|
||
"1",
|
||
"true",
|
||
"ok",
|
||
"success",
|
||
"succeed",
|
||
"sent",
|
||
}:
|
||
return True
|
||
state = self._find_first_value(payload, ["status", "state", "send_status"])
|
||
if state is not None:
|
||
if str(state).strip().lower() in {"ok", "success", "succeed", "sent", "done"}:
|
||
return True
|
||
if str(state).strip().lower() in {"failed", "error", "denied"}:
|
||
return False
|
||
err_text = self._find_first_value(payload, ["errmsg", "error", "error_msg", "detail"])
|
||
if err_text is not None and str(err_text).strip():
|
||
err_value = str(err_text).strip().lower()
|
||
if err_value not in {"ok", "success", "succeed", "sent"}:
|
||
return False
|
||
return False
|
||
|
||
def _is_send_explicit_failure(self, payload: Dict[str, Any]) -> bool:
|
||
"""检查 API 响应中是否包含明确的发送失败信号。"""
|
||
if not payload:
|
||
return False
|
||
code = self._find_first_value(
|
||
payload, ["errcode", "code", "ret", "result_code", "status_code"]
|
||
)
|
||
if code is not None:
|
||
try:
|
||
return int(str(code)) != 0
|
||
except Exception:
|
||
return str(code).strip().lower() not in {"0", "ok", "success", "succeed"}
|
||
success_flag = self._find_first_value(
|
||
payload, ["success", "ok", "is_success", "sent"]
|
||
)
|
||
if isinstance(success_flag, bool):
|
||
return not success_flag
|
||
if success_flag is not None:
|
||
if str(success_flag).strip().lower() in {
|
||
"0",
|
||
"false",
|
||
"fail",
|
||
"failed",
|
||
"error",
|
||
"denied",
|
||
}:
|
||
return True
|
||
state = self._find_first_value(payload, ["status", "state", "send_status"])
|
||
if state is not None:
|
||
if str(state).strip().lower() in {
|
||
"failed",
|
||
"error",
|
||
"denied",
|
||
"forbidden",
|
||
"blocked",
|
||
}:
|
||
return True
|
||
err_text = self._find_first_value(payload, ["errmsg", "error", "error_msg", "detail"])
|
||
if err_text is not None and str(err_text).strip():
|
||
if str(err_text).strip().lower() not in {"ok", "success", "succeed", "sent"}:
|
||
return True
|
||
return False
|
||
|
||
def _is_send_http_success(self, resp: Any, payload: Dict[str, Any]) -> bool:
|
||
"""根据 HTTP 状态码和解析后的响应判断发送是否成功。"""
|
||
if resp is None:
|
||
return False
|
||
status_code = getattr(resp, "status_code", None)
|
||
if status_code is None:
|
||
return False
|
||
try:
|
||
status_ok = 200 <= int(status_code) < 300
|
||
except Exception:
|
||
status_ok = False
|
||
if not status_ok:
|
||
return False
|
||
if not payload:
|
||
return True
|
||
return not self._is_send_explicit_failure(payload)
|
||
|
||
@staticmethod
|
||
def _build_user_candidates(to_user: str) -> List[str]:
|
||
"""为目标用户 ID 构建多种格式的候选列表以提高匹配成功率。"""
|
||
raw = str(to_user or "").strip()
|
||
if not raw:
|
||
return []
|
||
candidates = [raw]
|
||
if "@" in raw:
|
||
candidates.append(raw.split("@", 1)[0])
|
||
if raw.endswith("@im.wechat"):
|
||
candidates.append(raw[: -len("@im.wechat")])
|
||
else:
|
||
candidates.append(f"{raw}@im.wechat")
|
||
uniq: List[str] = []
|
||
for item in candidates:
|
||
value = str(item or "").strip()
|
||
if value and value not in uniq:
|
||
uniq.append(value)
|
||
return uniq
|
||
|
||
@staticmethod
|
||
def _build_text_payloads(user_id: str, text: str) -> List[Dict[str, Any]]:
|
||
"""构建多种格式的纯文本消息报文以兼容不同接口。"""
|
||
return [
|
||
{"to_user": user_id, "msg_type": "text", "text": {"content": text}},
|
||
{"to_user": user_id, "msg_type": "text", "text": text},
|
||
{"touser": user_id, "msgtype": "text", "text": {"content": text}},
|
||
{"touser": user_id, "msgtype": "text", "text": text},
|
||
{"to": user_id, "type": "text", "content": text},
|
||
{"to_user_id": user_id, "msg_type": "text", "content": text},
|
||
]
|
||
|
||
@staticmethod
|
||
def _aes_ecb_padded_size(plaintext_size: int) -> int:
|
||
"""计算 AES ECB 加密后的填充大小。"""
|
||
return ((int(plaintext_size) + 1 + 15) // 16) * 16
|
||
|
||
@staticmethod
|
||
def _encrypt_aes_ecb(plaintext: bytes, key: bytes) -> bytes:
|
||
"""使用 AES ECB 模式对明文进行加密。"""
|
||
cipher = AES.new(key, AES.MODE_ECB)
|
||
return cipher.encrypt(pad(plaintext, AES.block_size))
|
||
|
||
@staticmethod
|
||
def _encode_media_aes_key(aeskey: bytes) -> str:
|
||
"""将媒体 AES 密钥进行 Base64 编码。"""
|
||
return base64.b64encode(aeskey.hex().encode("ascii")).decode("ascii")
|
||
|
||
def _build_protocol_text_payload(
|
||
self, user_id: str, text: str, context_token: Optional[str]
|
||
) -> Dict[str, Any]:
|
||
"""构建协议层标准文本消息报文。"""
|
||
msg = {
|
||
"from_user_id": str(self.account_id or ""),
|
||
"to_user_id": user_id,
|
||
"client_id": f"mp-{uuid.uuid4()}",
|
||
"message_type": 2,
|
||
"message_state": 2,
|
||
"item_list": [{"type": 1, "text_item": {"text": text}}],
|
||
}
|
||
if context_token:
|
||
msg["context_token"] = context_token
|
||
return {"msg": msg}
|
||
|
||
def _build_protocol_image_payload(
|
||
self,
|
||
user_id: str,
|
||
context_token: Optional[str],
|
||
download_param: str,
|
||
aeskey_b64: str,
|
||
cipher_size: int,
|
||
) -> Dict[str, Any]:
|
||
"""构建协议层标准图片消息报文。"""
|
||
msg: Dict[str, Any] = {
|
||
"from_user_id": str(self.account_id or ""),
|
||
"to_user_id": user_id,
|
||
"client_id": f"mp-{uuid.uuid4()}",
|
||
"message_type": 2,
|
||
"message_state": 2,
|
||
"item_list": [
|
||
{
|
||
"type": 2,
|
||
"image_item": {
|
||
"media": {
|
||
"encrypt_query_param": download_param,
|
||
"aes_key": aeskey_b64,
|
||
"encrypt_type": 1,
|
||
},
|
||
"mid_size": int(cipher_size),
|
||
},
|
||
}
|
||
],
|
||
}
|
||
if context_token:
|
||
msg["context_token"] = context_token
|
||
return {"msg": msg}
|
||
|
||
def _build_protocol_file_payloads(
|
||
self,
|
||
user_id: str,
|
||
context_token: Optional[str],
|
||
download_param: str,
|
||
aeskey_b64: str,
|
||
cipher_size: int,
|
||
raw_size: int,
|
||
file_name: str,
|
||
mime_type: str,
|
||
file_md5: str,
|
||
) -> List[Dict[str, Any]]:
|
||
"""构建协议层标准文件消息报文（含多种格式候选）。"""
|
||
media = {
|
||
"encrypt_query_param": download_param,
|
||
"aes_key": aeskey_b64,
|
||
"encrypt_type": 1,
|
||
}
|
||
file_item = {
|
||
"name": file_name,
|
||
"file_name": file_name,
|
||
"filename": file_name,
|
||
"title": file_name,
|
||
"size": int(raw_size),
|
||
"file_size": int(raw_size),
|
||
"raw_size": int(raw_size),
|
||
"mid_size": int(cipher_size),
|
||
"mime_type": mime_type,
|
||
"content_type": mime_type,
|
||
"md5": file_md5,
|
||
"media": media,
|
||
}
|
||
msg_base: Dict[str, Any] = {
|
||
"from_user_id": str(self.account_id or ""),
|
||
"to_user_id": user_id,
|
||
"client_id": f"mp-{uuid.uuid4()}",
|
||
"message_type": 2,
|
||
"message_state": 2,
|
||
}
|
||
if context_token:
|
||
msg_base["context_token"] = context_token
|
||
protocol_candidates = []
|
||
for item_type in (6, 5, 3):
|
||
msg = dict(msg_base)
|
||
msg["item_list"] = [{"type": item_type, "file_item": dict(file_item)}]
|
||
protocol_candidates.append({"msg": msg})
|
||
simple_candidates = [
|
||
{
|
||
"to_user": user_id,
|
||
"msg_type": "file",
|
||
"file": dict(file_item),
|
||
},
|
||
{
|
||
"touser": user_id,
|
||
"msgtype": "file",
|
||
"file": dict(file_item),
|
||
},
|
||
{"to": user_id, "type": "file", "file": dict(file_item)},
|
||
]
|
||
return [*protocol_candidates, *simple_candidates]
|
||
|
||
def _request_upload_param(
|
||
self,
|
||
to_user: str,
|
||
plaintext: bytes,
|
||
media_types: Optional[List[int]] = None,
|
||
) -> Tuple[Optional[str], Optional[str], Optional[bytes], Optional[int], Optional[str]]:
|
||
"""
|
||
向 iLink 申请 CDN 上传参数。
|
||
|
||
iLink 对同一文件类型的兼容性并不稳定，这里会按候选 media_type 依次尝试，
|
||
成功后返回上传地址、AES 密钥和文件元信息给后续 CDN 上传流程复用。
|
||
"""
|
||
rawsize = len(plaintext)
|
||
rawfilemd5 = hashlib.md5(plaintext).hexdigest()
|
||
filesize = self._aes_ecb_padded_size(rawsize)
|
||
filekey = os.urandom(16).hex()
|
||
aeskey = os.urandom(16)
|
||
last_payload = None
|
||
media_type_candidates = media_types or [1]
|
||
for media_type in media_type_candidates:
|
||
body = self._with_base_info(
|
||
{
|
||
"filekey": filekey,
|
||
"media_type": media_type,
|
||
"to_user_id": to_user,
|
||
"rawsize": rawsize,
|
||
"rawfilemd5": rawfilemd5,
|
||
"filesize": filesize,
|
||
"no_need_thumb": True,
|
||
"aeskey": aeskey.hex(),
|
||
}
|
||
)
|
||
url = f"{self.base_url}/ilink/bot/getuploadurl"
|
||
resp = RequestUtils(
|
||
headers=self._headers(auth_required=True), timeout=self.timeout
|
||
).post(url, json=body)
|
||
payload = self._json(resp)
|
||
last_payload = payload or getattr(resp, "text", None)
|
||
upload_param = (
|
||
self._find_first_value(payload, ["upload_param", "uploadParam"])
|
||
if payload
|
||
else None
|
||
)
|
||
upload_full_url = (
|
||
self._find_first_value(payload, ["upload_full_url", "uploadFullUrl", "full_url"])
|
||
if payload
|
||
else None
|
||
)
|
||
if upload_param or upload_full_url:
|
||
return (
|
||
str(upload_param) if upload_param else None,
|
||
str(upload_full_url) if upload_full_url else None,
|
||
aeskey,
|
||
filesize,
|
||
filekey,
|
||
)
|
||
logger.warning(f"getuploadurl 失败: resp={self._short_text(last_payload)}")
|
||
return None, None, None, None, None
|
||
|
||
def _upload_encrypted_to_cdn(
|
||
self,
|
||
upload_param: Optional[str],
|
||
upload_full_url: Optional[str],
|
||
filekey: str,
|
||
plaintext: bytes,
|
||
aeskey: bytes,
|
||
) -> Tuple[Optional[str], Optional[int]]:
|
||
"""将文件按协议加密后上传到微信 CDN，并返回后续发送消息所需的下载参数。"""
|
||
ciphertext = self._encrypt_aes_ecb(plaintext, aeskey)
|
||
if upload_full_url:
|
||
upload_url = str(upload_full_url).strip()
|
||
elif upload_param:
|
||
upload_url = (
|
||
f"{self.cdn_base_url}/upload?encrypted_query_param={quote(str(upload_param), safe='')}&"
|
||
f"filekey={quote(filekey, safe='')}"
|
||
)
|
||
else:
|
||
logger.warning("CDN 上传失败: 缺少 upload_url 参数")
|
||
return None, None
|
||
resp = RequestUtils(
|
||
headers={"Content-Type": "application/octet-stream"},
|
||
timeout=self.timeout,
|
||
).post(upload_url, data=ciphertext)
|
||
if getattr(resp, "status_code", None) != 200:
|
||
logger.warning(
|
||
f"CDN 上传失败: http={getattr(resp, 'status_code', None)}, "
|
||
f"err={self._short_text(getattr(resp, 'text', ''))}"
|
||
)
|
||
return None, None
|
||
download_param = None
|
||
if resp is not None and getattr(resp, "headers", None):
|
||
download_param = resp.headers.get("x-encrypted-param")
|
||
if not download_param:
|
||
logger.warning("CDN 上传成功但缺少 x-encrypted-param")
|
||
return None, None
|
||
return str(download_param), len(ciphertext)
|
||
|
||
def _send_payload_candidates(
|
||
self,
|
||
to_user: str,
|
||
payload_candidates: List[Dict[str, Any]],
|
||
url_candidates: Optional[List[str]] = None,
|
||
) -> bool:
|
||
"""
|
||
尝试一组候选发送报文，兼容 iLink 不同接口形态。
|
||
|
||
这里不会只依赖单一 URL 或单一报文结构，而是按“用户候选 -> 接口候选 -> 报文候选”
|
||
逐层回退，尽量提高不同账号和不同版本服务端的发送成功率。
|
||
"""
|
||
url_candidates = url_candidates or [
|
||
f"{self.base_url}/ilink/bot/sendmessage",
|
||
f"{self.base_url}/ilink/bot/sendmessage?bot_type=3",
|
||
]
|
||
last_error = ""
|
||
for user_id in self._build_user_candidates(to_user):
|
||
for url in url_candidates:
|
||
for index, body in enumerate(payload_candidates, start=1):
|
||
request_body = self._with_base_info(body)
|
||
resp = RequestUtils(
|
||
headers=self._headers(auth_required=True), timeout=self.timeout
|
||
).post(url, json=request_body)
|
||
payload = self._json(resp)
|
||
if self._is_send_success(payload) or self._is_send_http_success(resp, payload):
|
||
logger.info(f"发送消息成功: to_user={user_id}, variant={index}")
|
||
return True
|
||
http_code = getattr(resp, "status_code", None)
|
||
err_msg = (
|
||
self._find_first_value(payload, ["errmsg", "message", "error", "detail"])
|
||
if payload
|
||
else None
|
||
)
|
||
if not err_msg and resp is not None:
|
||
err_msg = self._short_text(getattr(resp, "text", ""))
|
||
last_error = f"http={http_code}, err={self._short_text(err_msg)}"
|
||
logger.debug(
|
||
f"发送候选失败: to_user={user_id}, variant={index}, "
|
||
f"{last_error}, req={self._short_text(request_body)}, "
|
||
f"resp={self._short_text(payload)}"
|
||
)
|
||
logger.warning(f"发送消息失败: to_user={to_user}, {last_error}")
|
||
return False
|
||
|
||
def get_qrcode(self) -> Dict[str, Any]:
|
||
"""向服务端请求并解析二维码登录信息。"""
|
||
url = f"{self.base_url}/ilink/bot/get_bot_qrcode?bot_type=3"
|
||
logger.debug(f"请求二维码: {url}")
|
||
resp = RequestUtils(
|
||
headers=self._headers(auth_required=False), timeout=self.timeout
|
||
).get_res(url)
|
||
payload = self._json(resp)
|
||
if not payload:
|
||
return {"success": False, "message": "获取二维码失败"}
|
||
data = payload.get("data") or payload.get("result") or payload
|
||
qrcode = data.get("qrcode") or data.get("qr_code") or data.get(
|
||
"qrcode_id"
|
||
) or data.get("ticket")
|
||
qrcode_url = (
|
||
data.get("qrcode_url")
|
||
or data.get("url")
|
||
or data.get("qrcodeUrl")
|
||
or data.get("qr_url")
|
||
or data.get("qrcode_img_content")
|
||
or data.get("qrcode_img_url")
|
||
or data.get("qr_img")
|
||
)
|
||
qrcode_url = self._normalize_qrcode_url(qrcode_url)
|
||
if not qrcode_url and qrcode:
|
||
qrcode_url = f"https://liteapp.weixin.qq.com/q/7GiQu1?qrcode={qrcode}&bot_type=3"
|
||
return {
|
||
"success": self._ok(payload) and bool(qrcode or qrcode_url),
|
||
"qrcode": qrcode,
|
||
"qrcode_url": qrcode_url,
|
||
"raw": payload,
|
||
"message": payload.get("errmsg") or payload.get("message"),
|
||
}
|
||
|
||
def get_qrcode_status(self, qrcode: str) -> Dict[str, Any]:
|
||
"""查询二维码扫描状态并获取登录凭证。"""
|
||
url = f"{self.base_url}/ilink/bot/get_qrcode_status"
|
||
resp = RequestUtils(
|
||
headers=self._headers(auth_required=False), timeout=self.timeout
|
||
).get_res(url, params={"qrcode": qrcode})
|
||
payload = self._json(resp)
|
||
if not payload:
|
||
retry_resp = RequestUtils(
|
||
headers=self._headers(auth_required=False), timeout=self.timeout
|
||
).get_res(f"{url}?qrcode={qrcode}")
|
||
payload = self._json(retry_resp)
|
||
if not payload:
|
||
return {
|
||
"success": False,
|
||
"status": "waiting",
|
||
"token": None,
|
||
"account_id": None,
|
||
"raw": {},
|
||
"message": "二维码状态接口返回空响应",
|
||
}
|
||
data = payload.get("data") or payload.get("result") or payload
|
||
token = (
|
||
data.get("bot_token")
|
||
or data.get("token")
|
||
or data.get("access_token")
|
||
or self._find_first_value(
|
||
data, ["bot_token", "access_token", "token", "jwt", "auth_token"]
|
||
)
|
||
)
|
||
account_id = (
|
||
data.get("account_id")
|
||
or data.get("ilink_bot_id")
|
||
or data.get("wxid")
|
||
or data.get("uid")
|
||
or data.get("user_id")
|
||
or self._find_first_value(
|
||
data,
|
||
[
|
||
"account_id",
|
||
"ilink_bot_id",
|
||
"wxid",
|
||
"uid",
|
||
"user_id",
|
||
"from_user",
|
||
"from_uid",
|
||
],
|
||
)
|
||
)
|
||
base_url = (
|
||
data.get("baseurl")
|
||
or data.get("base_url")
|
||
or payload.get("baseurl")
|
||
or payload.get("base_url")
|
||
)
|
||
if token:
|
||
self.bot_token = token
|
||
if account_id:
|
||
self.account_id = str(account_id)
|
||
state = (
|
||
data.get("status")
|
||
or data.get("state")
|
||
or payload.get("status")
|
||
or payload.get("state")
|
||
or self._find_first_value(data, ["status", "state", "scan_status"])
|
||
or "waiting"
|
||
)
|
||
return {
|
||
"success": self._ok(payload),
|
||
"status": str(state).lower(),
|
||
"token": token,
|
||
"account_id": account_id,
|
||
"base_url": base_url,
|
||
"qrcode_url": self._normalize_qrcode_url(
|
||
data.get("qrcode_url")
|
||
or data.get("url")
|
||
or data.get("qrcodeUrl")
|
||
or data.get("qr_url")
|
||
or data.get("qrcode_img_content")
|
||
or data.get("qrcode_img_url")
|
||
or data.get("qr_img")
|
||
),
|
||
"raw": payload,
|
||
"message": payload.get("errmsg") or payload.get("message"),
|
||
}
|
||
|
||
def send_text(self, to_user: str, text: str, context_token: Optional[str] = None) -> bool:
|
||
"""发送纯文本消息。"""
|
||
if not self.bot_token:
|
||
logger.warning("发送消息失败：bot token 未配置")
|
||
return False
|
||
if not to_user or not text:
|
||
logger.warning("发送消息失败：to_user 或 text 为空")
|
||
return False
|
||
payload_candidates = [
|
||
self._build_protocol_text_payload(
|
||
user_id=str(to_user), text=text, context_token=context_token
|
||
),
|
||
*self._build_text_payloads(str(to_user), text),
|
||
]
|
||
return self._send_payload_candidates(to_user=to_user, payload_candidates=payload_candidates)
|
||
|
||
def send_image_text_png(
|
||
self,
|
||
to_user: str,
|
||
image_bytes: bytes,
|
||
text: str,
|
||
context_token: Optional[str] = None,
|
||
) -> bool:
|
||
"""发送包含图片和文本的消息。"""
|
||
if not self.bot_token:
|
||
logger.warning("发送图文失败：bot token 未配置")
|
||
return False
|
||
if not to_user or not image_bytes or not text:
|
||
logger.warning("发送图文失败：to_user 或 image_bytes 或 text 为空")
|
||
return False
|
||
for user_id in self._build_user_candidates(to_user):
|
||
upload_param, upload_full_url, aeskey, _, filekey = self._request_upload_param(
|
||
user_id, image_bytes, media_types=[1]
|
||
)
|
||
if (not upload_param and not upload_full_url) or not aeskey or not filekey:
|
||
continue
|
||
download_param, cipher_size = self._upload_encrypted_to_cdn(
|
||
upload_param=upload_param,
|
||
upload_full_url=upload_full_url,
|
||
filekey=filekey,
|
||
plaintext=image_bytes,
|
||
aeskey=aeskey,
|
||
)
|
||
if not download_param or not cipher_size:
|
||
continue
|
||
aeskey_b64 = self._encode_media_aes_key(aeskey)
|
||
message_items = [
|
||
{"type": 1, "text_item": {"text": text}},
|
||
{
|
||
"type": 2,
|
||
"image_item": {
|
||
"media": {
|
||
"encrypt_query_param": download_param,
|
||
"aes_key": aeskey_b64,
|
||
"encrypt_type": 1,
|
||
},
|
||
"mid_size": int(cipher_size),
|
||
},
|
||
},
|
||
]
|
||
sent_all = True
|
||
for item in message_items:
|
||
msg = {
|
||
"from_user_id": str(self.account_id or ""),
|
||
"to_user_id": user_id,
|
||
"client_id": f"mp-{uuid.uuid4()}",
|
||
"message_type": 2,
|
||
"message_state": 2,
|
||
"item_list": [item],
|
||
}
|
||
if context_token:
|
||
msg["context_token"] = context_token
|
||
if not self._send_payload_candidates(
|
||
to_user=user_id,
|
||
payload_candidates=[{"msg": msg}],
|
||
):
|
||
sent_all = False
|
||
break
|
||
if sent_all:
|
||
return True
|
||
return False
|
||
|
||
def send_image_png(
|
||
self, to_user: str, image_bytes: bytes, context_token: Optional[str] = None
|
||
) -> bool:
|
||
"""发送图片消息。"""
|
||
if not self.bot_token:
|
||
logger.warning("发送图片失败：bot token 未配置")
|
||
return False
|
||
if not to_user or not image_bytes:
|
||
logger.warning("发送图片失败：to_user 或 image_bytes 为空")
|
||
return False
|
||
for user_id in self._build_user_candidates(to_user):
|
||
upload_param, upload_full_url, aeskey, _, filekey = self._request_upload_param(
|
||
user_id, image_bytes, media_types=[1]
|
||
)
|
||
if (not upload_param and not upload_full_url) or not aeskey or not filekey:
|
||
continue
|
||
download_param, cipher_size = self._upload_encrypted_to_cdn(
|
||
upload_param=upload_param,
|
||
upload_full_url=upload_full_url,
|
||
filekey=filekey,
|
||
plaintext=image_bytes,
|
||
aeskey=aeskey,
|
||
)
|
||
if not download_param or not cipher_size:
|
||
continue
|
||
aeskey_b64 = self._encode_media_aes_key(aeskey)
|
||
body = self._build_protocol_image_payload(
|
||
user_id=user_id,
|
||
context_token=context_token,
|
||
download_param=download_param,
|
||
aeskey_b64=aeskey_b64,
|
||
cipher_size=cipher_size,
|
||
)
|
||
if self._send_payload_candidates(
|
||
to_user=user_id,
|
||
payload_candidates=[body],
|
||
):
|
||
return True
|
||
return False
|
||
|
||
def send_file_bytes(
|
||
self,
|
||
to_user: str,
|
||
file_bytes: bytes,
|
||
file_name: str,
|
||
mime_type: str,
|
||
context_token: Optional[str] = None,
|
||
) -> bool:
|
||
"""发送文件消息。"""
|
||
if not self.bot_token:
|
||
logger.warning("发送文件失败：bot token 未配置")
|
||
return False
|
||
if not to_user or not file_bytes:
|
||
logger.warning("发送文件失败：to_user 或 file_bytes 为空")
|
||
return False
|
||
file_name = file_name or "attachment"
|
||
mime_type = mime_type or "application/octet-stream"
|
||
file_md5 = hashlib.md5(file_bytes).hexdigest()
|
||
for user_id in self._build_user_candidates(to_user):
|
||
upload_param, upload_full_url, aeskey, _, filekey = self._request_upload_param(
|
||
user_id, file_bytes, media_types=[2, 6, 1]
|
||
)
|
||
if (not upload_param and not upload_full_url) or not aeskey or not filekey:
|
||
continue
|
||
download_param, cipher_size = self._upload_encrypted_to_cdn(
|
||
upload_param=upload_param,
|
||
upload_full_url=upload_full_url,
|
||
filekey=filekey,
|
||
plaintext=file_bytes,
|
||
aeskey=aeskey,
|
||
)
|
||
if not download_param or not cipher_size:
|
||
continue
|
||
aeskey_b64 = self._encode_media_aes_key(aeskey)
|
||
payload_candidates = self._build_protocol_file_payloads(
|
||
user_id=user_id,
|
||
context_token=context_token,
|
||
download_param=download_param,
|
||
aeskey_b64=aeskey_b64,
|
||
cipher_size=cipher_size,
|
||
raw_size=len(file_bytes),
|
||
file_name=file_name,
|
||
mime_type=mime_type,
|
||
file_md5=file_md5,
|
||
)
|
||
if self._send_payload_candidates(
|
||
to_user=user_id,
|
||
payload_candidates=payload_candidates,
|
||
):
|
||
return True
|
||
return False
|
||
|
||
@classmethod
|
||
def _encode_ref_payload(cls, kind: str, payload: Dict[str, Any]) -> str:
|
||
"""将附件元信息编码为 wxclaw:// 协议链接。"""
|
||
encoded = base64.urlsafe_b64encode(
|
||
json.dumps(payload, ensure_ascii=False).encode("utf-8")
|
||
).decode("ascii").rstrip("=")
|
||
return f"wxclaw://{kind}/{encoded}"
|
||
|
||
def _build_attachment_ref(
|
||
self,
|
||
kind: str,
|
||
attachment: Dict[str, Any],
|
||
default_name: Optional[str] = None,
|
||
) -> Optional[Tuple[str, Dict[str, Any]]]:
|
||
"""从附件信息构建 wxclaw 协议引用。"""
|
||
if not isinstance(attachment, dict):
|
||
return None
|
||
download_url = (
|
||
attachment.get("download_url")
|
||
or attachment.get("url")
|
||
or attachment.get("cdnurl")
|
||
or attachment.get("cdn_url")
|
||
or attachment.get("file_url")
|
||
)
|
||
if not download_url:
|
||
return None
|
||
payload = {
|
||
"url": download_url,
|
||
"aeskey": attachment.get("aeskey")
|
||
or attachment.get("encoding_aes_key")
|
||
or attachment.get("encrypt_key"),
|
||
"encrypt_type": attachment.get("encrypt_type"),
|
||
"mime_type": attachment.get("mime_type")
|
||
or attachment.get("content_type"),
|
||
"name": attachment.get("name")
|
||
or attachment.get("filename")
|
||
or default_name,
|
||
"size": attachment.get("size") or attachment.get("file_size"),
|
||
}
|
||
ref = self._encode_ref_payload(kind=kind, payload=payload)
|
||
return ref, payload
|
||
|
||
@staticmethod
|
||
def _as_scalar(value: Any) -> Optional[Any]:
|
||
"""将值转换为标量，过滤掉字典和列表。"""
|
||
if value in (None, ""):
|
||
return None
|
||
if isinstance(value, (dict, list, tuple, set)):
|
||
return None
|
||
return value
|
||
|
||
def _build_poll_result(
|
||
self,
|
||
success: bool,
|
||
payload: Optional[Dict[str, Any]] = None,
|
||
message: Optional[str] = None,
|
||
item_count: int = 0,
|
||
parsed_count: int = 0,
|
||
) -> Dict[str, Any]:
|
||
"""构建轮询结果摘要。"""
|
||
payload = payload or {}
|
||
resolved_message = message or self._find_first_value(
|
||
payload, ["errmsg", "message", "error", "error_msg", "detail"]
|
||
)
|
||
return {
|
||
"success": success,
|
||
"raw": payload,
|
||
"message": self._short_text(resolved_message) if resolved_message else None,
|
||
"item_count": item_count,
|
||
"parsed_count": parsed_count,
|
||
}
|
||
|
||
def _parse_incoming(self, item: Dict[str, Any]) -> Optional[ILinkIncomingMessage]:
|
||
"""
|
||
将 getupdates 返回的原始事件归一化为 MoviePilot 可消费的入站消息。
|
||
|
||
iLink 返回结构存在多种嵌套和字段别名，这里集中做兼容处理，统一提取
|
||
用户、消息ID、文本、图片、语音和文件附件。
|
||
"""
|
||
if not isinstance(item, dict):
|
||
return None
|
||
message = item
|
||
for key in ["message", "msg", "event", "payload", "data", "body"]:
|
||
child = item.get(key)
|
||
if isinstance(child, dict):
|
||
message = child
|
||
break
|
||
sender = (
|
||
message.get("from")
|
||
if isinstance(message.get("from"), dict)
|
||
else message.get("sender")
|
||
if isinstance(message.get("sender"), dict)
|
||
else message.get("user")
|
||
if isinstance(message.get("user"), dict)
|
||
else message.get("from_user")
|
||
if isinstance(message.get("from_user"), dict)
|
||
else {}
|
||
)
|
||
user_id = (
|
||
self._pick_value(sender, ["user_id", "id", "wxid", "uid"])
|
||
or self._pick_value(
|
||
message,
|
||
[
|
||
"from_user",
|
||
"from_user_id",
|
||
"user_id",
|
||
"uid",
|
||
"wxid",
|
||
"from_uid",
|
||
"fromUser",
|
||
"fromUserId",
|
||
"openid",
|
||
],
|
||
)
|
||
or self._pick_value(
|
||
item,
|
||
[
|
||
"from_user",
|
||
"from_user_id",
|
||
"user_id",
|
||
"uid",
|
||
"wxid",
|
||
"from_uid",
|
||
"fromUser",
|
||
"fromUserId",
|
||
"openid",
|
||
],
|
||
)
|
||
or self._find_first_value(
|
||
message,
|
||
[
|
||
"from_user",
|
||
"from_user_id",
|
||
"user_id",
|
||
"sender_id",
|
||
"uid",
|
||
"wxid",
|
||
"from_uid",
|
||
"fromUserId",
|
||
"openid",
|
||
],
|
||
)
|
||
)
|
||
user_id = self._as_scalar(user_id)
|
||
if not user_id:
|
||
return None
|
||
username = (
|
||
self._pick_value(sender, ["name", "nickname", "username", "remark"])
|
||
or self._pick_value(
|
||
message, ["username", "nickname", "from_name", "fromNick", "sender_name"]
|
||
)
|
||
or str(user_id)
|
||
)
|
||
message_id = self._pick_value(
|
||
message, ["message_id", "msg_id", "id", "client_msg_id", "msgId", "seq"]
|
||
) or self._pick_value(
|
||
item, ["message_id", "msg_id", "id", "client_msg_id", "msgId", "seq"]
|
||
)
|
||
chat_id = self._pick_value(
|
||
message,
|
||
["chat_id", "conversation_id", "room_id", "chatId", "conversationId", "roomId"],
|
||
) or self._pick_value(
|
||
item,
|
||
["chat_id", "conversation_id", "room_id", "chatId", "conversationId", "roomId"],
|
||
)
|
||
context_token = self._pick_value(message, ["context_token", "contextToken"]) or self._pick_value(
|
||
item, ["context_token", "contextToken"]
|
||
)
|
||
text_parts: List[str] = []
|
||
images: List[Dict[str, Any]] = []
|
||
audio_refs: List[str] = []
|
||
files: List[Dict[str, Any]] = []
|
||
|
||
def append_text(value: Any) -> None:
|
||
if value in (None, ""):
|
||
return
|
||
if isinstance(value, dict):
|
||
value = self._pick_value(value, ["content", "text", "value", "message"])
|
||
if value in (None, ""):
|
||
return
|
||
text_value = str(value).strip()
|
||
if text_value and text_value not in text_parts:
|
||
text_parts.append(text_value)
|
||
|
||
msgtype = str(
|
||
message.get("msgtype") or message.get("msg_type") or message.get("type") or ""
|
||
).lower()
|
||
if msgtype == "text":
|
||
append_text((message.get("text") or {}).get("content") if isinstance(message.get("text"), dict) else None)
|
||
append_text(message.get("content"))
|
||
elif msgtype == "image":
|
||
image_data = message.get("image") or {}
|
||
image_ref = self._build_attachment_ref("image", image_data)
|
||
if image_ref:
|
||
ref, payload = image_ref
|
||
images.append(
|
||
{
|
||
"ref": ref,
|
||
"name": payload.get("name"),
|
||
"mime_type": payload.get("mime_type"),
|
||
"size": payload.get("size"),
|
||
}
|
||
)
|
||
elif msgtype == "file":
|
||
file_data = message.get("file") or {}
|
||
file_ref = self._build_attachment_ref("file", file_data)
|
||
if file_ref:
|
||
ref, payload = file_ref
|
||
files.append(
|
||
{
|
||
"ref": ref,
|
||
"name": payload.get("name"),
|
||
"mime_type": payload.get("mime_type"),
|
||
"size": payload.get("size"),
|
||
}
|
||
)
|
||
elif msgtype == "voice":
|
||
voice_data = message.get("voice") or {}
|
||
append_text((voice_data or {}).get("content"))
|
||
voice_ref = self._build_attachment_ref("voice", voice_data, default_name="voice.amr")
|
||
if voice_ref:
|
||
ref, _ = voice_ref
|
||
audio_refs.append(ref)
|
||
elif msgtype == "mixed":
|
||
for msg_item in ((message.get("mixed") or {}).get("msg_item") or []):
|
||
item_type = str(msg_item.get("msgtype") or msg_item.get("type") or "").lower()
|
||
if item_type == "text":
|
||
append_text((msg_item.get("text") or {}).get("content"))
|
||
elif item_type == "image":
|
||
image_ref = self._build_attachment_ref("image", msg_item.get("image") or {})
|
||
if image_ref:
|
||
ref, payload = image_ref
|
||
images.append(
|
||
{
|
||
"ref": ref,
|
||
"name": payload.get("name"),
|
||
"mime_type": payload.get("mime_type"),
|
||
"size": payload.get("size"),
|
||
}
|
||
)
|
||
elif item_type == "file":
|
||
file_ref = self._build_attachment_ref("file", msg_item.get("file") or {})
|
||
if file_ref:
|
||
ref, payload = file_ref
|
||
files.append(
|
||
{
|
||
"ref": ref,
|
||
"name": payload.get("name"),
|
||
"mime_type": payload.get("mime_type"),
|
||
"size": payload.get("size"),
|
||
}
|
||
)
|
||
elif item_type == "voice":
|
||
append_text(((msg_item.get("voice") or {}).get("content") or "").strip())
|
||
voice_ref = self._build_attachment_ref(
|
||
"voice",
|
||
msg_item.get("voice") or {},
|
||
default_name="voice.amr",
|
||
)
|
||
if voice_ref:
|
||
ref, _ = voice_ref
|
||
audio_refs.append(ref)
|
||
|
||
item_list = message.get("item_list") if isinstance(message.get("item_list"), list) else []
|
||
for one in item_list:
|
||
if not isinstance(one, dict):
|
||
continue
|
||
item_type = one.get("type")
|
||
if item_type == 1 and isinstance(one.get("text_item"), dict):
|
||
append_text((one.get("text_item") or {}).get("text"))
|
||
elif item_type == 2 and isinstance(one.get("image_item"), dict):
|
||
image_ref = self._build_attachment_ref(
|
||
"image", (one.get("image_item") or {}).get("media") or {}
|
||
)
|
||
if image_ref:
|
||
ref, payload = image_ref
|
||
images.append(
|
||
{
|
||
"ref": ref,
|
||
"name": payload.get("name"),
|
||
"mime_type": payload.get("mime_type"),
|
||
"size": payload.get("size"),
|
||
}
|
||
)
|
||
elif item_type in {5, 6}:
|
||
file_ref = self._build_attachment_ref(
|
||
"file", (one.get("file_item") or {}).get("media") or {}
|
||
)
|
||
if file_ref:
|
||
ref, payload = file_ref
|
||
files.append(
|
||
{
|
||
"ref": ref,
|
||
"name": (one.get("file_item") or {}).get("name")
|
||
or (one.get("file_item") or {}).get("file_name")
|
||
or payload.get("name"),
|
||
"mime_type": (one.get("file_item") or {}).get("mime_type")
|
||
or payload.get("mime_type"),
|
||
"size": (one.get("file_item") or {}).get("size")
|
||
or payload.get("size"),
|
||
}
|
||
)
|
||
|
||
if not text_parts:
|
||
append_text(
|
||
self._pick_value(message, ["content", "message", "msg", "text", "body", "msg_content", "msgContent"])
|
||
)
|
||
append_text(
|
||
self._pick_value(item, ["content", "message", "msg", "text", "body", "msg_content", "msgContent"])
|
||
)
|
||
append_text(self._find_first_value(message, ["content", "text", "message", "msg", "body", "cmd"]))
|
||
|
||
text = "\n".join(part for part in text_parts if part).strip() or None
|
||
return ILinkIncomingMessage(
|
||
user_id=str(user_id),
|
||
text=text,
|
||
username=str(username) if username else None,
|
||
message_id=str(message_id) if message_id else None,
|
||
chat_id=str(chat_id) if chat_id else None,
|
||
context_token=str(context_token) if context_token else None,
|
||
images=images,
|
||
audio_refs=audio_refs,
|
||
files=files,
|
||
raw=item,
|
||
)
|
||
|
||
def _extract_updates(
|
||
self, payload: Dict[str, Any]
|
||
) -> Tuple[List[Dict[str, Any]], Optional[str]]:
|
||
"""
|
||
提取轮询结果中的消息列表与游标。
|
||
|
||
线上存在两种等价字段命名：较新的实现返回 `get_updates_buf`，
|
||
部分实例仍然返回 `sync_buf`。两者都表示下一轮轮询应携带的游标。
|
||
"""
|
||
sync_buf = self._pick_present_value(
|
||
payload, ["get_updates_buf", "sync_buf", "syncBuf"]
|
||
)
|
||
items = payload.get("msgs")
|
||
if isinstance(items, list):
|
||
return items, sync_buf
|
||
return [], sync_buf
|
||
|
||
@staticmethod
|
||
def _has_canonical_poll_shape(payload: Dict[str, Any]) -> bool:
|
||
"""官方响应至少应包含顶层 msgs 列表。"""
|
||
return isinstance(payload.get("msgs"), list)
|
||
|
||
def _resolve_poll_success(self, payload: Dict[str, Any]) -> Optional[bool]:
|
||
"""
|
||
判断 getupdates 是否给出了明确的成功/失败信号。
|
||
|
||
轮询接口不能沿用“只要没有明显报错就算成功”的宽松策略，否则服务端返回旧消息列表、
|
||
但状态码其实失败时，会被误判为可消费响应，导致旧消息再次进入业务链路。
|
||
返回 `None` 表示响应里没有显式状态，需要交给协议结构继续判断。
|
||
"""
|
||
if not payload:
|
||
return False
|
||
code = self._find_first_value(
|
||
payload, ["errcode", "code", "ret", "result_code", "status_code"]
|
||
)
|
||
if code is not None:
|
||
try:
|
||
return int(str(code)) == 0
|
||
except Exception:
|
||
return str(code).strip().lower() in {"0", "ok", "success", "succeed"}
|
||
success_flag = self._find_first_value(
|
||
payload, ["success", "ok", "is_success"]
|
||
)
|
||
if isinstance(success_flag, bool):
|
||
return success_flag
|
||
if success_flag is not None:
|
||
return str(success_flag).strip().lower() in {
|
||
"1",
|
||
"true",
|
||
"ok",
|
||
"success",
|
||
"succeed",
|
||
}
|
||
state = self._find_first_value(payload, ["status", "state"])
|
||
if state is not None:
|
||
lowered = str(state).strip().lower()
|
||
if lowered in {"ok", "success", "succeed", "done"}:
|
||
return True
|
||
if lowered in {"failed", "fail", "error", "denied", "blocked"}:
|
||
return False
|
||
return None
|
||
|
||
def poll_updates(
|
||
self, timeout_seconds: int = 25
|
||
) -> Tuple[List[ILinkIncomingMessage], Optional[str], Dict[str, Any]]:
|
||
"""
|
||
执行一次 iLink 长轮询。
|
||
|
||
返回值同时包含归一化后的消息、最新游标和原始结果摘要，
|
||
便于上层在推进 sync_buf 的同时记录调试信息。
|
||
"""
|
||
if not self.bot_token:
|
||
return [], self.sync_buf, {"success": False, "message": "bot token 未配置"}
|
||
url = f"{self.base_url}/ilink/bot/getupdates"
|
||
request_body = self._with_base_info({"get_updates_buf": self.sync_buf or ""})
|
||
resp = RequestUtils(
|
||
headers=self._headers(auth_required=True),
|
||
timeout=timeout_seconds + 10,
|
||
).post(url, json=request_body)
|
||
payload = self._json(resp)
|
||
explicit_success = self._resolve_poll_success(payload)
|
||
has_canonical_shape = self._has_canonical_poll_shape(payload)
|
||
# 某些 iLink 部署不会返回 ret/success，但顶层 msgs + sync_buf 已经足够表明
|
||
# 这是一次有效的轮询响应；只有出现显式失败信号时才应拒绝消费。
|
||
success = bool(payload) and (
|
||
explicit_success is True
|
||
or (explicit_success is None and has_canonical_shape)
|
||
)
|
||
last_message = None
|
||
if payload and explicit_success is False:
|
||
last_message = self._find_first_value(
|
||
payload, ["errmsg", "message", "error", "error_msg", "detail"]
|
||
) or self._short_text(payload)
|
||
if not payload:
|
||
return [], self.sync_buf, self._build_poll_result(
|
||
success=False,
|
||
message="轮询返回空响应",
|
||
)
|
||
if not success:
|
||
return [], self.sync_buf, self._build_poll_result(
|
||
success=False,
|
||
payload=payload,
|
||
message=last_message or "轮询响应未明确成功",
|
||
)
|
||
if not has_canonical_shape:
|
||
logger.warning(
|
||
"getupdates 返回非官方结构，已拒绝消费: %s",
|
||
self._short_text(payload),
|
||
)
|
||
return [], self.sync_buf, self._build_poll_result(
|
||
success=False,
|
||
payload=payload,
|
||
message="轮询响应结构非官方，缺少顶层 msgs 字段",
|
||
)
|
||
items, sync_buf = self._extract_updates(payload)
|
||
parsed: List[ILinkIncomingMessage] = []
|
||
for item in items:
|
||
message = self._parse_incoming(item)
|
||
if message:
|
||
parsed.append(message)
|
||
if sync_buf is not None:
|
||
self.sync_buf = str(sync_buf)
|
||
return parsed, self.sync_buf, self._build_poll_result(
|
||
success=True,
|
||
payload=payload,
|
||
item_count=len(items),
|
||
parsed_count=len(parsed),
|
||
)
|
||
|
||
def test_connection(self) -> Tuple[bool, str]:
|
||
"""测试与 iLink 服务端的连接连通性。"""
|
||
if not self.bot_token:
|
||
return False, "未登录，缺少 bot token"
|
||
url = f"{self.base_url}/ilink/bot/getconfig"
|
||
resp = RequestUtils(
|
||
headers=self._headers(auth_required=True), timeout=self.timeout
|
||
).post(url, json={})
|
||
payload = self._json(resp)
|
||
if self._ok(payload):
|
||
return True, "连接正常"
|
||
return False, payload.get("errmsg") or payload.get("message") or "连接失败"
|
||
|
||
|
||
class WechatClawBot:
|
||
"""微信 ClawBot 渠道客户端。"""
|
||
|
||
_default_base_url = "https://ilinkai.weixin.qq.com"
|
||
_qrcode_ttl_seconds = 240
|
||
_active_target_ttl_seconds = 24 * 60 * 60
|
||
|
||
@classmethod
|
||
def _build_cache_key(cls, config_name: str) -> str:
|
||
"""根据配置名称构建缓存键。"""
|
||
safe_name = hashlib.md5(str(config_name or "wechatclawbot").encode("utf-8")).hexdigest()[:12]
|
||
return f"__wechatclawbot_state_{safe_name}__"
|
||
|
||
def __init__(
|
||
self,
|
||
WECHATCLAWBOT_BASE_URL: Optional[str] = None,
|
||
WECHATCLAWBOT_DEFAULT_TARGET: Optional[str] = None,
|
||
WECHATCLAWBOT_ADMINS: Optional[str] = None,
|
||
WECHATCLAWBOT_POLL_TIMEOUT: Optional[int] = None,
|
||
name: Optional[str] = None,
|
||
auto_start_polling: bool = True,
|
||
**kwargs,
|
||
):
|
||
"""初始化微信 ClawBot 实例及相关参数。"""
|
||
self._config_name = name or "wechatclawbot"
|
||
self._base_url = (WECHATCLAWBOT_BASE_URL or self._default_base_url).rstrip("/")
|
||
self._default_target = (WECHATCLAWBOT_DEFAULT_TARGET or "").strip() or None
|
||
self._auto_start_polling = bool(auto_start_polling)
|
||
self._admins = [
|
||
admin.strip()
|
||
for admin in str(WECHATCLAWBOT_ADMINS or "").split(",")
|
||
if admin.strip()
|
||
]
|
||
try:
|
||
self._poll_timeout = max(10, int(WECHATCLAWBOT_POLL_TIMEOUT or 25))
|
||
except Exception:
|
||
self._poll_timeout = 25
|
||
self._cache_key = self._build_cache_key(self._config_name)
|
||
self._filecache = FileCache()
|
||
self._lock = threading.Lock()
|
||
self._stop_event = threading.Event()
|
||
self._poll_thread: Optional[threading.Thread] = None
|
||
self._state = self._load_state()
|
||
self._message_endpoint = (
|
||
f"http://127.0.0.1:{settings.PORT}/api/v1/message?token={settings.API_TOKEN}&source={quote(self._config_name, safe='')}"
|
||
)
|
||
if self._state.get("bot_token") and self._auto_start_polling:
|
||
self._start_polling()
|
||
|
||
def _load_state(self) -> Dict[str, Any]:
|
||
"""从文件缓存加载登录状态。"""
|
||
content = self._filecache.get(self._cache_key)
|
||
if not content:
|
||
return {
|
||
"bot_token": None,
|
||
"account_id": None,
|
||
"sync_buf": None,
|
||
"qrcode": {},
|
||
"known_targets": {},
|
||
"user_context_tokens": {},
|
||
"base_url": self._base_url,
|
||
}
|
||
try:
|
||
data = json.loads(content.decode("utf-8"))
|
||
if not isinstance(data, dict):
|
||
raise ValueError("invalid state")
|
||
data.setdefault("qrcode", {})
|
||
data.setdefault("known_targets", {})
|
||
data.setdefault("user_context_tokens", {})
|
||
data.setdefault("base_url", self._base_url)
|
||
return data
|
||
except Exception as err:
|
||
logger.warning(f"加载登录状态失败，已重置缓存：{err}")
|
||
return {
|
||
"bot_token": None,
|
||
"account_id": None,
|
||
"sync_buf": None,
|
||
"qrcode": {},
|
||
"known_targets": {},
|
||
"user_context_tokens": {},
|
||
"base_url": self._base_url,
|
||
}
|
||
|
||
def _save_state(self) -> None:
|
||
"""将当前状态持久化到文件缓存。"""
|
||
self._state["base_url"] = self._base_url
|
||
self._filecache.set(
|
||
self._cache_key,
|
||
json.dumps(self._state, ensure_ascii=False).encode("utf-8"),
|
||
)
|
||
|
||
def _build_client(self) -> ILinkClient:
|
||
"""根据当前持久化状态构建一次性的 iLink 客户端。"""
|
||
return ILinkClient(
|
||
base_url=self._state.get("base_url") or self._base_url,
|
||
bot_token=self._state.get("bot_token"),
|
||
account_id=self._state.get("account_id"),
|
||
sync_buf=self._state.get("sync_buf"),
|
||
timeout=max(self._poll_timeout, 20),
|
||
)
|
||
|
||
def _update_state(self, **kwargs) -> None:
|
||
"""更新并保存持久化状态。"""
|
||
with self._lock:
|
||
self._state.update(kwargs)
|
||
self._save_state()
|
||
|
||
def _clear_login_state(self) -> None:
|
||
"""清理当前的登录状态。"""
|
||
with self._lock:
|
||
self._state["bot_token"] = None
|
||
self._state["account_id"] = None
|
||
self._state["sync_buf"] = None
|
||
self._state["user_context_tokens"] = {}
|
||
self._save_state()
|
||
|
||
def _qrcode_expired(self, updated_at: Optional[int]) -> bool:
|
||
"""检查二维码是否过期。"""
|
||
if not updated_at:
|
||
return True
|
||
return int(time.time()) - int(updated_at) > self._qrcode_ttl_seconds
|
||
|
||
def _remember_target(
|
||
self, user_id: str, username: Optional[str], context_token: Optional[str]
|
||
) -> None:
|
||
"""记录已知消息目标，用于消息发送时定位。"""
|
||
if not user_id:
|
||
return
|
||
now_ts = int(time.time())
|
||
with self._lock:
|
||
known_targets = self._state.setdefault("known_targets", {})
|
||
known_targets[str(user_id)] = {
|
||
"username": username or str(user_id),
|
||
"last_active": now_ts,
|
||
}
|
||
if context_token:
|
||
tokens = self._state.setdefault("user_context_tokens", {})
|
||
tokens[str(user_id)] = str(context_token)
|
||
self._save_state()
|
||
|
||
def _get_context_token(self, user_id: str) -> Optional[str]:
|
||
"""获取特定用户的上下文 Token。"""
|
||
tokens = self._state.get("user_context_tokens") or {}
|
||
token = tokens.get(str(user_id))
|
||
return str(token) if token else None
|
||
|
||
def _get_targets(self, userid: Optional[str] = None) -> List[str]:
|
||
"""获取消息发送的目标列表。"""
|
||
if userid:
|
||
return [str(userid)]
|
||
if self._default_target:
|
||
return [self._default_target]
|
||
now_ts = int(time.time())
|
||
known_targets = self._state.get("known_targets") or {}
|
||
active_targets = [
|
||
target
|
||
for target, data in known_targets.items()
|
||
if isinstance(data, dict)
|
||
and now_ts - int(data.get("last_active") or 0) <= self._active_target_ttl_seconds
|
||
]
|
||
if active_targets:
|
||
return sorted(active_targets)
|
||
return sorted(known_targets.keys())
|
||
|
||
@staticmethod
|
||
def _short_text(value: Any, max_len: int = 240) -> str:
|
||
"""将内容缩短用于日志记录。"""
|
||
if value is None:
|
||
return ""
|
||
if isinstance(value, (dict, list)):
|
||
try:
|
||
text = json.dumps(value, ensure_ascii=False)
|
||
except Exception:
|
||
text = str(value)
|
||
else:
|
||
text = str(value)
|
||
text = text.replace("\n", " ").replace("\r", " ").strip()
|
||
if len(text) > max_len:
|
||
return f"{text[:max_len]}..."
|
||
return text
|
||
|
||
@staticmethod
|
||
def _split_content(content: str, max_bytes: int = 3000) -> List[str]:
|
||
"""将长消息内容拆分为符合大小限制的块。"""
|
||
if not content:
|
||
return []
|
||
chunks: List[str] = []
|
||
current = bytearray()
|
||
for line in content.splitlines():
|
||
encoded = (line + "\n").encode("utf-8")
|
||
if len(encoded) > max_bytes:
|
||
if current:
|
||
chunks.append(current.decode("utf-8", errors="replace").strip())
|
||
current = bytearray()
|
||
start = 0
|
||
while start < len(encoded):
|
||
end = min(start + max_bytes, len(encoded))
|
||
while start < end < len(encoded) and (encoded[end] & 0xC0) == 0x80:
|
||
end -= 1
|
||
chunks.append(encoded[start:end].decode("utf-8", errors="replace").strip())
|
||
start = end
|
||
continue
|
||
if len(current) + len(encoded) > max_bytes:
|
||
chunks.append(current.decode("utf-8", errors="replace").strip())
|
||
current = bytearray()
|
||
current += encoded
|
||
if current:
|
||
chunks.append(current.decode("utf-8", errors="replace").strip())
|
||
return [chunk for chunk in chunks if chunk]
|
||
|
||
@staticmethod
|
||
def _compose_text(
|
||
title: Optional[str] = None,
|
||
text: Optional[str] = None,
|
||
link: Optional[str] = None,
|
||
) -> str:
|
||
"""组合标题、文本和链接，生成消息内容。"""
|
||
parts = []
|
||
if title:
|
||
parts.append(str(title).strip())
|
||
if text:
|
||
parts.append(str(text).replace("\n\n", "\n"))
|
||
if link:
|
||
parts.append(f"查看详情：{link}")
|
||
return "\n\n".join(part for part in parts if part).strip()
|
||
|
||
@staticmethod
|
||
def _guess_mime_type(file_path: Path, file_bytes: bytes) -> str:
|
||
"""猜测文件 MIME 类型。"""
|
||
guessed = mimetypes.guess_type(file_path.name)[0]
|
||
if guessed:
|
||
return guessed
|
||
if file_bytes.startswith(b"\x89PNG\r\n\x1a\n"):
|
||
return "image/png"
|
||
if file_bytes.startswith(b"\xff\xd8\xff"):
|
||
return "image/jpeg"
|
||
if file_bytes.startswith((b"GIF87a", b"GIF89a")):
|
||
return "image/gif"
|
||
return "application/octet-stream"
|
||
|
||
@staticmethod
|
||
def _load_remote_image(image: str) -> Optional[bytes]:
|
||
"""加载远程图片并返回二进制数据。"""
|
||
image_url = str(image or "").strip()
|
||
if not image_url:
|
||
return None
|
||
if image_url.startswith("data:image/"):
|
||
try:
|
||
_, raw = image_url.split(",", 1)
|
||
return base64.b64decode(raw)
|
||
except Exception:
|
||
return None
|
||
if image_url.startswith("/"):
|
||
image_url = settings.MP_DOMAIN(image_url)
|
||
if not image_url.lower().startswith("http"):
|
||
return None
|
||
try:
|
||
resp = RequestUtils(
|
||
timeout=20,
|
||
proxies=settings.PROXY,
|
||
ua=settings.USER_AGENT,
|
||
).get_res(image_url)
|
||
if resp and resp.status_code == 200 and resp.content:
|
||
content_type = (resp.headers.get("Content-Type") or "").lower()
|
||
if not content_type or "image" in content_type:
|
||
return resp.content
|
||
except Exception as err:
|
||
logger.warning(f"加载图片失败：{err}")
|
||
return None
|
||
|
||
def get_state(self) -> bool:
|
||
"""获取当前登录状态。"""
|
||
return bool(self._state.get("bot_token"))
|
||
|
||
def stop(self) -> None:
|
||
"""停止消息轮询。"""
|
||
self._stop_event.set()
|
||
if self._poll_thread and self._poll_thread.is_alive():
|
||
self._poll_thread.join(timeout=5)
|
||
self._poll_thread = None
|
||
|
||
def _start_polling(self) -> None:
|
||
"""启动消息轮询线程。"""
|
||
if not self._state.get("bot_token"):
|
||
return
|
||
if self._poll_thread and self._poll_thread.is_alive():
|
||
return
|
||
self._stop_event.clear()
|
||
self._poll_thread = threading.Thread(target=self._poll_loop, daemon=True)
|
||
self._poll_thread.start()
|
||
logger.info("消息轮询线程已启动")
|
||
|
||
def _poll_loop(self) -> None:
|
||
"""
|
||
持续拉取微信消息并转发到本地消息入口。
|
||
|
||
轮询异常时按退避策略重试；本地消息入口失败只记日志，不回滚 sync_buf，
|
||
由上层的消息ID去重兜底，避免单次本地异常导致后续消息全部阻塞。
|
||
"""
|
||
consecutive_failures = 0
|
||
backoff = [1, 2, 5, 10, 30]
|
||
while not self._stop_event.is_set() and self._state.get("bot_token"):
|
||
try:
|
||
client = self._build_client()
|
||
messages, sync_buf, result = client.poll_updates(
|
||
timeout_seconds=self._poll_timeout
|
||
)
|
||
if sync_buf is not None and sync_buf != self._state.get("sync_buf"):
|
||
self._update_state(sync_buf=sync_buf)
|
||
if not result.get("success"):
|
||
raise RuntimeError(result.get("message") or "poll failed")
|
||
for message in messages:
|
||
self._remember_target(
|
||
user_id=message.user_id,
|
||
username=message.username,
|
||
context_token=message.context_token,
|
||
)
|
||
response = None
|
||
try:
|
||
response = RequestUtils(timeout=15).post_res(
|
||
self._message_endpoint,
|
||
json=message.to_message_payload(),
|
||
)
|
||
if response is None:
|
||
logger.error(
|
||
f"转发微信 ClawBot 消息失败：message_id={message.message_id}, "
|
||
"本地消息入口无响应"
|
||
)
|
||
elif response.status_code != 200:
|
||
logger.error(
|
||
"转发微信 ClawBot 消息失败："
|
||
f"message_id={message.message_id}, status={response.status_code}, "
|
||
f"body={self._short_text(response.text)}"
|
||
)
|
||
except Exception as err:
|
||
logger.error(
|
||
f"转发微信 ClawBot 消息失败：message_id={message.message_id}, error={err}"
|
||
)
|
||
finally:
|
||
if response is not None:
|
||
try:
|
||
response.close()
|
||
except Exception:
|
||
pass
|
||
consecutive_failures = 0
|
||
except Exception as err:
|
||
consecutive_failures += 1
|
||
delay = backoff[min(consecutive_failures - 1, len(backoff) - 1)]
|
||
logger.warning(f"轮询异常，{delay}s 后重试：{err}")
|
||
if consecutive_failures >= 10:
|
||
logger.error("轮询连续失败，已清理登录状态")
|
||
self._clear_login_state()
|
||
break
|
||
self._stop_event.wait(delay)
|
||
|
||
def _build_known_targets(self) -> List[Dict[str, Any]]:
|
||
known_targets = self._state.get("known_targets") or {}
|
||
items = []
|
||
for userid, data in known_targets.items():
|
||
if not isinstance(data, dict):
|
||
continue
|
||
items.append(
|
||
{
|
||
"userid": userid,
|
||
"username": data.get("username") or userid,
|
||
"last_active": data.get("last_active"),
|
||
}
|
||
)
|
||
return sorted(items, key=lambda item: item.get("last_active") or 0, reverse=True)
|
||
|
||
def refresh_qrcode(self) -> Dict[str, Any]:
|
||
if self._state.get("bot_token"):
|
||
return self.get_status(refresh_remote=False)
|
||
client = ILinkClient(
|
||
base_url=self._base_url,
|
||
timeout=max(self._poll_timeout, 20),
|
||
)
|
||
result = client.get_qrcode()
|
||
if not result.get("success"):
|
||
return result
|
||
qrcode = {
|
||
"qrcode": result.get("qrcode"),
|
||
"qrcode_url": result.get("qrcode_url"),
|
||
"status": "waiting",
|
||
"updated_at": int(time.time()),
|
||
}
|
||
self._update_state(qrcode=qrcode)
|
||
return self.get_status(refresh_remote=False)
|
||
|
||
def get_status(
|
||
self,
|
||
refresh_remote: bool = True,
|
||
auto_generate_qrcode: bool = False,
|
||
) -> Dict[str, Any]:
|
||
qrcode = self._state.get("qrcode") or {}
|
||
if (
|
||
auto_generate_qrcode
|
||
and not self._state.get("bot_token")
|
||
and (not qrcode.get("qrcode") or self._qrcode_expired(qrcode.get("updated_at")))
|
||
):
|
||
self.refresh_qrcode()
|
||
qrcode = self._state.get("qrcode") or {}
|
||
if refresh_remote and not self._state.get("bot_token") and qrcode.get("qrcode"):
|
||
client = ILinkClient(
|
||
base_url=self._base_url,
|
||
timeout=max(self._poll_timeout, 20),
|
||
)
|
||
result = client.get_qrcode_status(str(qrcode.get("qrcode")))
|
||
updated_qrcode = dict(qrcode)
|
||
updated_qrcode["status"] = result.get("status") or updated_qrcode.get("status") or "waiting"
|
||
updated_qrcode["updated_at"] = int(time.time())
|
||
if result.get("qrcode_url"):
|
||
updated_qrcode["qrcode_url"] = result.get("qrcode_url")
|
||
update_payload: Dict[str, Any] = {"qrcode": updated_qrcode}
|
||
if result.get("token"):
|
||
update_payload.update(
|
||
{
|
||
"bot_token": result.get("token"),
|
||
"account_id": str(result.get("account_id") or "") or None,
|
||
"sync_buf": None,
|
||
"base_url": (result.get("base_url") or self._base_url).rstrip("/"),
|
||
}
|
||
)
|
||
self._base_url = (result.get("base_url") or self._base_url).rstrip("/")
|
||
self._update_state(**update_payload)
|
||
if self._auto_start_polling:
|
||
self._start_polling()
|
||
else:
|
||
self._update_state(**update_payload)
|
||
qrcode = self._state.get("qrcode") or {}
|
||
return {
|
||
"success": True,
|
||
"connected": bool(self._state.get("bot_token")),
|
||
"account_id": self._state.get("account_id"),
|
||
"qrcode": qrcode.get("qrcode"),
|
||
"qrcode_url": qrcode.get("qrcode_url"),
|
||
"qrcode_status": qrcode.get("status"),
|
||
"qrcode_updated_at": qrcode.get("updated_at"),
|
||
"known_targets": self._build_known_targets(),
|
||
"default_target": self._default_target,
|
||
"base_url": self._base_url,
|
||
}
|
||
|
||
def logout(self) -> Dict[str, Any]:
|
||
self.stop()
|
||
with self._lock:
|
||
self._state["bot_token"] = None
|
||
self._state["account_id"] = None
|
||
self._state["sync_buf"] = None
|
||
self._state["qrcode"] = {}
|
||
self._state["user_context_tokens"] = {}
|
||
self._save_state()
|
||
return {"success": True, "message": "已退出微信 ClawBot 登录"}
|
||
|
||
def test_connection(self) -> Tuple[bool, str]:
|
||
if not self._state.get("bot_token"):
|
||
return False, "未登录，请先扫码完成绑定"
|
||
return self._build_client().test_connection()
|
||
|
||
@classmethod
|
||
def migrate_cached_state(
|
||
cls,
|
||
old_name: Optional[str],
|
||
new_name: Optional[str],
|
||
cleanup_old: bool = False,
|
||
overwrite: bool = False,
|
||
) -> Tuple[bool, str]:
|
||
"""迁移配置重命名后的登录缓存，默认复制而不是删除旧缓存。"""
|
||
source_name = str(old_name or "").strip()
|
||
target_name = str(new_name or "").strip()
|
||
if not source_name or not target_name:
|
||
return False, "旧名称或新名称不能为空"
|
||
if source_name == target_name:
|
||
return True, "通知名称未变化，无需迁移登录缓存"
|
||
|
||
cache = FileCache()
|
||
source_key = cls._build_cache_key(source_name)
|
||
target_key = cls._build_cache_key(target_name)
|
||
source_state = cache.get(source_key)
|
||
if not source_state:
|
||
return True, "未找到可迁移的微信 ClawBot 登录缓存"
|
||
if cache.exists(target_key) and not overwrite:
|
||
return True, "新名称下已存在登录缓存，跳过迁移"
|
||
|
||
cache.set(target_key, source_state)
|
||
if cleanup_old:
|
||
cache.delete(source_key)
|
||
return True, f"已将微信 ClawBot 登录缓存从 {source_name} 迁移到 {target_name}"
|
||
|
||
@staticmethod
|
||
def _decode_ref_payload(ref: str, kind: str) -> Optional[Dict[str, Any]]:
|
||
prefix = f"wxclaw://{kind}/"
|
||
if not ref or not ref.startswith(prefix):
|
||
return None
|
||
encoded = ref.replace(prefix, "", 1)
|
||
padding = "=" * (-len(encoded) % 4)
|
||
try:
|
||
return json.loads(
|
||
base64.urlsafe_b64decode((encoded + padding).encode("ascii")).decode(
|
||
"utf-8"
|
||
)
|
||
)
|
||
except Exception:
|
||
return None
|
||
|
||
@staticmethod
|
||
def _decode_aes_key(aeskey: Optional[str]) -> Optional[bytes]:
|
||
if not aeskey:
|
||
return None
|
||
value = str(aeskey).strip()
|
||
if not value:
|
||
return None
|
||
if re.fullmatch(r"[0-9a-fA-F]{32}", value):
|
||
try:
|
||
return bytes.fromhex(value)
|
||
except Exception:
|
||
return None
|
||
for candidate in (value, value + "=" * (-len(value) % 4)):
|
||
try:
|
||
decoded = base64.b64decode(candidate)
|
||
if len(decoded) == 16:
|
||
return decoded
|
||
if len(decoded) == 32 and re.fullmatch(rb"[0-9a-fA-F]{32}", decoded):
|
||
return bytes.fromhex(decoded.decode("ascii"))
|
||
except Exception:
|
||
continue
|
||
return None
|
||
|
||
@staticmethod
|
||
def _unpad_bytes(content: bytes) -> Optional[bytes]:
|
||
if not content:
|
||
return None
|
||
padding_len = content[-1]
|
||
if padding_len <= 0 or padding_len > 16:
|
||
return None
|
||
if content[-padding_len:] != bytes([padding_len]) * padding_len:
|
||
return None
|
||
return content[:-padding_len]
|
||
|
||
@classmethod
|
||
def _decrypt_if_needed(
|
||
cls,
|
||
content: bytes,
|
||
aeskey: Optional[str],
|
||
mime_type: Optional[str] = None,
|
||
) -> bytes:
|
||
aes_bytes = cls._decode_aes_key(aeskey)
|
||
if not aes_bytes or not content:
|
||
return content
|
||
candidates = [content]
|
||
try:
|
||
candidates.append(AES.new(aes_bytes, AES.MODE_ECB).decrypt(content))
|
||
except Exception:
|
||
pass
|
||
try:
|
||
candidates.append(AES.new(aes_bytes, AES.MODE_CBC, aes_bytes[:16]).decrypt(content))
|
||
except Exception:
|
||
pass
|
||
normalized_candidates: List[bytes] = []
|
||
for candidate in candidates:
|
||
normalized_candidates.append(candidate)
|
||
unpadded = cls._unpad_bytes(candidate)
|
||
if unpadded:
|
||
normalized_candidates.append(unpadded)
|
||
preferred_prefixes = []
|
||
mime_value = (mime_type or "").lower()
|
||
if mime_value.startswith("image/png"):
|
||
preferred_prefixes.append(b"\x89PNG\r\n\x1a\n")
|
||
elif mime_value.startswith("image/jpeg"):
|
||
preferred_prefixes.append(b"\xff\xd8\xff")
|
||
elif mime_value.startswith("image/gif"):
|
||
preferred_prefixes.extend([b"GIF87a", b"GIF89a"])
|
||
elif mime_value.startswith("audio/"):
|
||
preferred_prefixes.extend([b"#!AMR", b"ID3", b"OggS", b"RIFF"])
|
||
for candidate in normalized_candidates:
|
||
if any(candidate.startswith(prefix) for prefix in preferred_prefixes):
|
||
return candidate
|
||
for candidate in normalized_candidates:
|
||
if candidate.startswith((b"\x89PNG\r\n\x1a\n", b"\xff\xd8\xff", b"GIF87a", b"GIF89a", b"PK\x03\x04", b"#!AMR", b"ID3", b"OggS", b"RIFF")):
|
||
return candidate
|
||
return content
|
||
|
||
@staticmethod
|
||
def _guess_binary_mime(content: bytes, default: str = "application/octet-stream") -> str:
|
||
if not content:
|
||
return default
|
||
if content.startswith(b"\x89PNG\r\n\x1a\n"):
|
||
return "image/png"
|
||
if content.startswith(b"\xff\xd8\xff"):
|
||
return "image/jpeg"
|
||
if content.startswith((b"GIF87a", b"GIF89a")):
|
||
return "image/gif"
|
||
if content.startswith(b"#!AMR"):
|
||
return "audio/amr"
|
||
if content.startswith(b"OggS"):
|
||
return "audio/ogg"
|
||
if content.startswith(b"ID3"):
|
||
return "audio/mpeg"
|
||
if content.startswith(b"RIFF") and b"WAVE" in content[:16]:
|
||
return "audio/wav"
|
||
if content.startswith(b"PK\x03\x04"):
|
||
return "application/zip"
|
||
return default
|
||
|
||
def _download_ref_bytes(self, ref: str, kind: str) -> Optional[Tuple[bytes, str, Optional[str]]]:
|
||
payload = self._decode_ref_payload(ref=ref, kind=kind)
|
||
if not payload:
|
||
return None
|
||
download_url = payload.get("url")
|
||
if not download_url:
|
||
return None
|
||
try:
|
||
resp = RequestUtils(timeout=30).get_res(download_url)
|
||
except Exception as err:
|
||
logger.error(f"下载 {kind} 失败：{err}")
|
||
return None
|
||
if not resp or not resp.content:
|
||
return None
|
||
content = self._decrypt_if_needed(
|
||
content=resp.content,
|
||
aeskey=payload.get("aeskey"),
|
||
mime_type=payload.get("mime_type"),
|
||
)
|
||
mime_type = payload.get("mime_type") or self._guess_binary_mime(
|
||
content,
|
||
(resp.headers.get("Content-Type") or "application/octet-stream").split(
|
||
";", 1
|
||
)[0],
|
||
)
|
||
return content, mime_type, payload.get("name")
|
||
|
||
def download_image_to_data_url(self, image_ref: str) -> Optional[str]:
|
||
result = self._download_ref_bytes(ref=image_ref, kind="image")
|
||
if not result:
|
||
return None
|
||
content, mime_type, _ = result
|
||
return f"data:{mime_type};base64,{base64.b64encode(content).decode()}"
|
||
|
||
def download_media_bytes(self, media_ref: str) -> Optional[bytes]:
|
||
kind = None
|
||
if media_ref.startswith("wxclaw://file/"):
|
||
kind = "file"
|
||
elif media_ref.startswith("wxclaw://voice/"):
|
||
kind = "voice"
|
||
if not kind:
|
||
return None
|
||
result = self._download_ref_bytes(ref=media_ref, kind=kind)
|
||
return result[0] if result else None
|
||
|
||
def send_msg(
|
||
self,
|
||
title: str,
|
||
text: Optional[str] = None,
|
||
image: Optional[str] = None,
|
||
userid: Optional[str] = None,
|
||
link: Optional[str] = None,
|
||
**kwargs,
|
||
) -> Optional[bool]:
|
||
targets = self._get_targets(userid=userid)
|
||
if not targets:
|
||
logger.warning("未找到可发送的微信 ClawBot 目标")
|
||
return False
|
||
content = self._compose_text(title=title, text=text, link=link)
|
||
# 当前 iLink 发送实现会把图文拆成两条消息发送。
|
||
# 为避免用户看到“同一条通知被拆成文本 + 图片”两次触达，这里约定：
|
||
# 只要通知里已经有文本内容，就优先只发文本；只有纯图片通知才发送图片。
|
||
image_bytes = self._load_remote_image(image) if image and not content else None
|
||
ok = False
|
||
for target in targets:
|
||
context_token = self._get_context_token(target)
|
||
if content:
|
||
client = self._build_client()
|
||
sent = True
|
||
for chunk in self._split_content(content):
|
||
if not client.send_text(
|
||
to_user=target,
|
||
text=chunk,
|
||
context_token=context_token,
|
||
):
|
||
sent = False
|
||
break
|
||
elif image_bytes:
|
||
sent = self._build_client().send_image_png(
|
||
to_user=target,
|
||
image_bytes=image_bytes,
|
||
context_token=context_token,
|
||
)
|
||
else:
|
||
client = self._build_client()
|
||
sent = True
|
||
for chunk in self._split_content(content):
|
||
if not client.send_text(
|
||
to_user=target,
|
||
text=chunk,
|
||
context_token=context_token,
|
||
):
|
||
sent = False
|
||
break
|
||
ok = ok or bool(sent)
|
||
return ok
|
||
|
||
def send_file(
|
||
self,
|
||
file_path: str,
|
||
file_name: Optional[str] = None,
|
||
title: Optional[str] = None,
|
||
text: Optional[str] = None,
|
||
userid: Optional[str] = None,
|
||
**kwargs,
|
||
) -> Optional[bool]:
|
||
path = Path(file_path)
|
||
if not path.exists() or not path.is_file():
|
||
logger.warning(f"待发送文件不存在：{file_path}")
|
||
return False
|
||
file_bytes = path.read_bytes()
|
||
effective_name = file_name or path.name
|
||
mime_type = self._guess_mime_type(path, file_bytes)
|
||
targets = self._get_targets(userid=userid)
|
||
if not targets:
|
||
return False
|
||
ok = False
|
||
for target in targets:
|
||
context_token = self._get_context_token(target)
|
||
# send_file 的主语义是发送附件本体，附加文案可以丢弃，避免出现
|
||
# “先发一段说明，再发一个文件”的重复触达体验。
|
||
if mime_type.startswith("image/"):
|
||
sent = self._build_client().send_image_png(
|
||
to_user=target,
|
||
image_bytes=file_bytes,
|
||
context_token=context_token,
|
||
)
|
||
else:
|
||
sent = self._build_client().send_file_bytes(
|
||
to_user=target,
|
||
file_bytes=file_bytes,
|
||
file_name=effective_name,
|
||
mime_type=mime_type,
|
||
context_token=context_token,
|
||
)
|
||
ok = ok or bool(sent)
|
||
return ok
|
||
|
||
def send_medias_msg(
|
||
self, medias: List[MediaInfo], userid: Optional[str] = None
|
||
) -> Optional[bool]:
|
||
if not medias:
|
||
return False
|
||
lines = []
|
||
for index, media in enumerate(medias, start=1):
|
||
line = f"{index}. {media.title_year}"
|
||
if media.vote_average:
|
||
line += f" 评分：{media.vote_average}"
|
||
if media.detail_link:
|
||
line += f"\n{media.detail_link}"
|
||
lines.append(line)
|
||
return self.send_msg(title="媒体列表", text="\n\n".join(lines), userid=userid)
|
||
|
||
def send_torrents_msg(
|
||
self,
|
||
torrents: List[Context],
|
||
userid: Optional[str] = None,
|
||
title: Optional[str] = None,
|
||
link: Optional[str] = None,
|
||
) -> Optional[bool]:
|
||
if not torrents:
|
||
return False
|
||
lines = []
|
||
for index, context in enumerate(torrents, start=1):
|
||
torrent = context.torrent_info
|
||
meta = MetaInfo(title=torrent.title, subtitle=torrent.description)
|
||
text = (
|
||
f"{index}.【{torrent.site_name}】{meta.season_episode} {meta.resource_term} "
|
||
f"{meta.video_term} {meta.release_group} {StringUtils.str_filesize(torrent.size)} "
|
||
f"{torrent.volume_factor} {torrent.seeders}↑"
|
||
)
|
||
text = re.sub(r"\s+", " ", text).strip()
|
||
if torrent.page_url:
|
||
text += f"\n{torrent.page_url}"
|
||
lines.append(text)
|
||
return self.send_msg(
|
||
title=title or "种子列表",
|
||
text="\n\n".join(lines),
|
||
userid=userid,
|
||
link=link,
|
||
)
|