133 lines
4.6 KiB
Python
133 lines
4.6 KiB
Python
"""
|
||
艾叶 IM Bridge — WhatsApp 通道
|
||
对接方式：
|
||
1. WhatsApp Business API (Cloud API) — Webhook 回调
|
||
2. 可扩展对接 OpenClaw/Moltbot Gateway 的 WebSocket
|
||
|
||
当前实现：WhatsApp Cloud API Webhook 模式
|
||
|
||
配置项：
|
||
phone_number_id: WhatsApp Business 电话号码 ID
|
||
access_token: Meta Graph API 长期令牌
|
||
verify_token: Webhook 验证令牌（自定义字符串）
|
||
api_version: Graph API 版本（默认 v21.0）
|
||
|
||
Webhook: POST /webhook/whatsapp
|
||
验证: GET /webhook/whatsapp
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
import logging
|
||
from typing import Any
|
||
|
||
import httpx
|
||
from fastapi import Request, Response
|
||
|
||
from core.channel_base import ChannelBase, InboundMessage, MessageType, OutboundMessage
|
||
|
||
logger = logging.getLogger("aiye.channel.whatsapp")
|
||
|
||
|
||
class WhatsAppChannel(ChannelBase):
|
||
"""WhatsApp Cloud API 通道"""
|
||
|
||
@property
|
||
def platform(self) -> str:
|
||
return "whatsapp"
|
||
|
||
async def start(self) -> None:
|
||
logger.info("WhatsApp 通道已就绪，Webhook: /webhook/whatsapp")
|
||
|
||
async def stop(self) -> None:
|
||
logger.info("WhatsApp 通道已停止")
|
||
|
||
async def send(self, msg: OutboundMessage) -> bool:
|
||
phone_id = self._config.get("phone_number_id", "")
|
||
token = self._config.get("access_token", "")
|
||
api_ver = self._config.get("api_version", "v21.0")
|
||
if not phone_id or not token:
|
||
logger.warning("WhatsApp 未配置 phone_number_id / access_token")
|
||
return False
|
||
|
||
url = f"https://graph.facebook.com/{api_ver}/{phone_id}/messages"
|
||
payload = {
|
||
"messaging_product": "whatsapp",
|
||
"to": msg.chat_id,
|
||
"type": "text",
|
||
"text": {"body": msg.content},
|
||
}
|
||
try:
|
||
async with httpx.AsyncClient(timeout=15) as client:
|
||
resp = await client.post(
|
||
url,
|
||
json=payload,
|
||
headers={"Authorization": f"Bearer {token}"},
|
||
)
|
||
if resp.status_code == 200:
|
||
return True
|
||
logger.warning("WhatsApp 发送失败: %d %s", resp.status_code, resp.text[:200])
|
||
except Exception as e:
|
||
logger.error("WhatsApp 发送异常: %s", e)
|
||
return False
|
||
|
||
def register_routes(self, app: Any) -> None:
|
||
channel = self
|
||
|
||
@app.get("/webhook/whatsapp")
|
||
async def whatsapp_verify(
|
||
request: Request,
|
||
):
|
||
"""Meta Webhook 验证"""
|
||
params = request.query_params
|
||
mode = params.get("hub.mode", "")
|
||
token = params.get("hub.verify_token", "")
|
||
challenge = params.get("hub.challenge", "")
|
||
if mode == "subscribe" and token == channel._config.get("verify_token", ""):
|
||
return Response(content=challenge, media_type="text/plain")
|
||
return Response(content="forbidden", status_code=403)
|
||
|
||
@app.post("/webhook/whatsapp")
|
||
async def whatsapp_webhook(request: Request):
|
||
try:
|
||
data = await request.json()
|
||
except Exception:
|
||
return {"status": "error"}
|
||
|
||
entries = data.get("entry", [])
|
||
for entry in entries:
|
||
changes = entry.get("changes", [])
|
||
for change in changes:
|
||
value = change.get("value", {})
|
||
messages = value.get("messages", [])
|
||
contacts = value.get("contacts", [])
|
||
contact_map = {
|
||
c.get("wa_id", ""): c.get("profile", {}).get("name", "")
|
||
for c in contacts
|
||
}
|
||
|
||
for wa_msg in messages:
|
||
msg_type = wa_msg.get("type", "")
|
||
from_id = wa_msg.get("from", "")
|
||
text = ""
|
||
if msg_type == "text":
|
||
text = wa_msg.get("text", {}).get("body", "")
|
||
|
||
if not text.strip():
|
||
continue
|
||
|
||
inbound = InboundMessage(
|
||
channel_id=channel.channel_id,
|
||
platform=channel.platform,
|
||
sender_id=from_id,
|
||
sender_name=contact_map.get(from_id, ""),
|
||
chat_id=from_id,
|
||
content=text,
|
||
raw=wa_msg,
|
||
)
|
||
|
||
reply = await channel.dispatch(inbound)
|
||
if reply:
|
||
await channel.send(reply)
|
||
|
||
return {"status": "ok"}
|