271 lines
9.8 KiB
Python
271 lines
9.8 KiB
Python
"""消息发送处理器：JSON 解析、参数校验、QQ API 超时与重试、批量发送。"""
|
||
|
||
import asyncio
|
||
from pathlib import Path
|
||
|
||
from aiohttp import web
|
||
|
||
from ..config import QQ_API_MAX_RETRIES, QQ_API_TIMEOUT, UPLOAD_DIR
|
||
from ..response import error, ok
|
||
|
||
VALID_MSG_TYPES = {"text", "image", "file", "video"}
|
||
|
||
|
||
def _resolve_url(url: str) -> str:
|
||
"""如果 url 是本地已上传的文件（相对路径），补全为绝对路径。"""
|
||
file_path = Path(url)
|
||
if not file_path.is_absolute() and (UPLOAD_DIR / url).exists():
|
||
return str((UPLOAD_DIR / url).resolve())
|
||
return url
|
||
|
||
|
||
def _validate_message(msg: dict) -> str | None:
|
||
"""校验单条消息，返回 None 或错误信息。"""
|
||
msg_type = msg.get("type", "text")
|
||
if msg_type not in VALID_MSG_TYPES:
|
||
return f"invalid type: {msg_type}, must be one of {VALID_MSG_TYPES}"
|
||
|
||
# text 必须有 msg，image/file/video 必须有 url
|
||
if msg_type == "text" and not msg.get("msg"):
|
||
return "missing required field: msg"
|
||
if msg_type in {"image", "file", "video"} and not msg.get("url"):
|
||
return f"missing required field: url"
|
||
return None
|
||
|
||
|
||
async def _call_with_retry(coro_factory, logger, rid: str) -> None:
|
||
"""带超时和重试的 API 调用。"""
|
||
last_exc: Exception | None = None
|
||
for attempt in range(1, QQ_API_MAX_RETRIES + 1):
|
||
try:
|
||
await asyncio.wait_for(coro_factory(), timeout=QQ_API_TIMEOUT)
|
||
return
|
||
except asyncio.TimeoutError:
|
||
last_exc = asyncio.TimeoutError()
|
||
logger.warning(f"[{rid}] QQ API timeout, attempt {attempt}/{QQ_API_MAX_RETRIES}")
|
||
except Exception as exc:
|
||
last_exc = exc
|
||
logger.error(f"[{rid}] QQ API error: {exc}, attempt {attempt}/{QQ_API_MAX_RETRIES}")
|
||
|
||
raise Exception(f"qq api failed after {QQ_API_MAX_RETRIES} retries: {last_exc}")
|
||
|
||
|
||
async def webhook_handler(request: web.Request) -> web.Response:
|
||
"""处理消息发送请求，支持单条和批量发送。
|
||
|
||
单条格式：
|
||
{
|
||
"group_id": "123",
|
||
"type": "text",
|
||
"msg": "hello"
|
||
}
|
||
|
||
批量格式（使用 post_group_msg 组合消息段）：
|
||
{
|
||
"group_id": "123",
|
||
"messages": [
|
||
{"type": "text", "msg": "hello"},
|
||
{"type": "image", "url": "photo.jpg"}
|
||
]
|
||
}
|
||
|
||
批量发送时，text/image/video 会在一条消息里组合发送（框架自动拆分冲突段），
|
||
file 单独发送（因为 file 是独占段）。
|
||
"""
|
||
try:
|
||
data = await request.json()
|
||
except Exception:
|
||
return error("invalid json")
|
||
|
||
if not isinstance(data, dict):
|
||
return error("request body must be a json object")
|
||
|
||
group_id = data.get("group_id")
|
||
user_id = data.get("user_id")
|
||
|
||
if not group_id and not user_id:
|
||
return error("need group_id or user_id")
|
||
|
||
api = request.app["qq_api"]
|
||
logger = request.app["logger"]
|
||
rid = request.get("request_id", "-")
|
||
|
||
messages = data.get("messages")
|
||
|
||
if messages is not None:
|
||
# 批量模式
|
||
if not isinstance(messages, list):
|
||
return error("messages must be an array")
|
||
|
||
if not messages:
|
||
return error("messages cannot be empty")
|
||
|
||
# 校验所有消息
|
||
for i, msg in enumerate(messages):
|
||
if not isinstance(msg, dict):
|
||
return error(f"messages[{i}] must be an object")
|
||
err = _validate_message(msg)
|
||
if err:
|
||
return error(f"messages[{i}]: {err}")
|
||
|
||
# 分组：text/image/video 可以组合，file 必须单独发
|
||
combined_msgs: list[dict] = [] # text/image/video 组合
|
||
file_msgs: list[dict] = [] # file 单独发
|
||
|
||
for msg in messages:
|
||
if msg.get("type") == "file":
|
||
file_msgs.append(msg)
|
||
else:
|
||
combined_msgs.append(msg)
|
||
|
||
results: list[dict] = []
|
||
|
||
# 发送组合消息（text + image + video）
|
||
if combined_msgs:
|
||
text_parts: list[str] = []
|
||
image_urls: list[str] = []
|
||
video_url: str | None = None
|
||
|
||
for msg in combined_msgs:
|
||
msg_type = msg.get("type", "text")
|
||
if msg_type == "text":
|
||
text_parts.append(msg.get("msg", ""))
|
||
elif msg_type == "image":
|
||
image_urls.append(_resolve_url(msg.get("url", "")))
|
||
elif msg_type == "video":
|
||
video_url = _resolve_url(msg.get("url", ""))
|
||
|
||
text = "\n".join(text_parts) if text_parts else None
|
||
image = image_urls[0] if image_urls else None # post_group_msg 只支持单张图片
|
||
|
||
try:
|
||
if group_id:
|
||
await _call_with_retry(
|
||
lambda: api.qq.post_group_msg(
|
||
group_id=group_id,
|
||
text=text,
|
||
image=image,
|
||
video=video_url,
|
||
),
|
||
logger, rid
|
||
)
|
||
else:
|
||
await _call_with_retry(
|
||
lambda: api.qq.post_private_msg(
|
||
user_id=user_id,
|
||
text=text,
|
||
image=image,
|
||
video=video_url,
|
||
),
|
||
logger, rid
|
||
)
|
||
|
||
# 所有组合消息标记为成功
|
||
for i, msg in enumerate(messages):
|
||
if msg.get("type") != "file":
|
||
results.append({"index": i, "success": True})
|
||
except Exception as exc:
|
||
logger.error(f"[{rid}] Combined message failed: {exc}")
|
||
for i, msg in enumerate(messages):
|
||
if msg.get("type") != "file":
|
||
results.append({"index": i, "success": False, "error": str(exc)})
|
||
|
||
# 发送文件消息（每个文件单独一条）
|
||
for msg in file_msgs:
|
||
idx = messages.index(msg)
|
||
url = _resolve_url(msg.get("url", ""))
|
||
filename = url.split("/")[-1]
|
||
|
||
try:
|
||
if group_id:
|
||
await _call_with_retry(
|
||
lambda u=url, n=filename: api.qq.send_group_file(
|
||
group_id=group_id, file=u, name=n
|
||
),
|
||
logger, rid
|
||
)
|
||
else:
|
||
await _call_with_retry(
|
||
lambda u=url, n=filename: api.qq.send_private_file(
|
||
user_id=user_id, file=u, name=n
|
||
),
|
||
logger, rid
|
||
)
|
||
results.append({"index": idx, "success": True})
|
||
except Exception as exc:
|
||
logger.error(f"[{rid}] File message failed: {exc}")
|
||
results.append({"index": idx, "success": False, "error": str(exc)})
|
||
|
||
# 按 index 排序结果
|
||
results.sort(key=lambda x: x["index"])
|
||
success_count = sum(1 for r in results if r["success"])
|
||
|
||
return ok(data={
|
||
"total": len(results),
|
||
"success": success_count,
|
||
"failed": len(results) - success_count,
|
||
"results": results,
|
||
})
|
||
|
||
else:
|
||
# 单条模式（兼容旧格式）
|
||
err = _validate_message(data)
|
||
if err:
|
||
return error(err)
|
||
|
||
msg_type = data.get("type", "text")
|
||
url = _resolve_url(data.get("url", "")) if msg_type != "text" else None
|
||
msg_text = data.get("msg", "")
|
||
filename = url.split("/")[-1] if url else None
|
||
|
||
try:
|
||
if group_id:
|
||
match msg_type:
|
||
case "text":
|
||
await _call_with_retry(
|
||
lambda: api.qq.send_group_text(group_id=group_id, text=msg_text),
|
||
logger, rid
|
||
)
|
||
case "image":
|
||
await _call_with_retry(
|
||
lambda: api.qq.send_group_image(group_id=group_id, image=url),
|
||
logger, rid
|
||
)
|
||
case "file":
|
||
await _call_with_retry(
|
||
lambda: api.qq.send_group_file(group_id=group_id, file=url, name=filename),
|
||
logger, rid
|
||
)
|
||
case "video":
|
||
await _call_with_retry(
|
||
lambda: api.qq.send_group_video(group_id=group_id, video=url),
|
||
logger, rid
|
||
)
|
||
else:
|
||
match msg_type:
|
||
case "text":
|
||
await _call_with_retry(
|
||
lambda: api.qq.send_private_text(user_id=user_id, text=msg_text),
|
||
logger, rid
|
||
)
|
||
case "image":
|
||
await _call_with_retry(
|
||
lambda: api.qq.send_private_image(user_id=user_id, image=url),
|
||
logger, rid
|
||
)
|
||
case "file":
|
||
await _call_with_retry(
|
||
lambda: api.qq.send_private_file(user_id=user_id, file=url, name=filename),
|
||
logger, rid
|
||
)
|
||
case "video":
|
||
await _call_with_retry(
|
||
lambda: api.qq.send_private_video(user_id=user_id, video=url),
|
||
logger, rid
|
||
)
|
||
|
||
return ok()
|
||
except Exception as exc:
|
||
logger.error(f"[{rid}] QQ API error: {exc}")
|
||
return error(f"qq api failed: {exc}", code=502, status=502)
|