ForcePilot/backend/package/yuxi/channel/extensions/workplace/menu.py
Kris 7215418610 feat(channel): 添加企业微信、微博、WhatsApp 和 Workplace 渠道扩展
新增企业微信、微博、WhatsApp、Workplace 四个渠道扩展。

企业微信渠道扩展主要模块:config, gateway, webhook, webhook_bot, outbound, streaming, pairing, security, crypto, dedupe, persistent_dedupe, card, directory, events, externalcontact, media, mentions, menu, message, oauth, status

微博渠道扩展主要模块:config, gateway, webhook, outbound, streaming, pairing, security, dedupe, passive_reply, broadcast, message, menu, media, subscription, template, status

WhatsApp 渠道扩展主要模块:config, gateway, webhook, outbound, streaming, pairing, security, dedupe, actions, monitor, status

Workplace 渠道扩展主要模块:config, gateway, webhook, outbound, streaming, pairing, security, dedupe, actions, challenge, groups, media, mentions, menu, monitor, persona, quick_reply, signature, subscriptions, template, threading, users, status
2026-05-21 12:01:56 +08:00

115 lines
3.8 KiB
Python

import logging
import httpx
logger = logging.getLogger(__name__)
MENU_MAX_TOP_LEVEL = 3
MENU_MAX_SUB_LEVEL = 5
MENU_TITLE_MAX_CHARS = 30
MENU_PAYLOAD_MAX_CHARS = 1000
def _build_menu_api_url(api_version: str = "v24.0") -> str:
return f"https://graph.facebook.com/{api_version}/me/messenger_profile"
def _compute_appsecret_proof(access_token: str, app_secret: str) -> str:
import hashlib
import hmac
return hmac.new(
app_secret.encode("utf-8"),
access_token.encode("utf-8"),
hashlib.sha256,
).hexdigest()
async def set_persistent_menu(
access_token: str,
menu_items: list[dict],
*,
app_secret: str = "",
api_version: str = "v24.0",
composer_input_disabled: bool = False,
) -> dict:
url = _build_menu_api_url(api_version)
params: dict = {"access_token": access_token}
if app_secret:
params["appsecret_proof"] = _compute_appsecret_proof(access_token, app_secret)
payload = {
"persistent_menu": [
{
"locale": "default",
"composer_input_disabled": composer_input_disabled,
"call_to_actions": menu_items,
}
]
}
async with httpx.AsyncClient(timeout=15.0) as client:
try:
resp = await client.post(url, json=payload, params=params)
if resp.status_code == 200:
logger.info("Persistent menu set successfully")
return {"success": True, "data": resp.json()}
logger.error("Failed to set persistent menu: %d %s", resp.status_code, resp.text[:200])
return {"success": False, "error": resp.text[:200]}
except httpx.RequestError as exc:
logger.error("Persistent menu network error: %s", exc)
return {"success": False, "error": str(exc)}
async def delete_persistent_menu(
access_token: str,
*,
app_secret: str = "",
api_version: str = "v24.0",
) -> dict:
url = _build_menu_api_url(api_version)
params: dict = {"access_token": access_token}
if app_secret:
params["appsecret_proof"] = _compute_appsecret_proof(access_token, app_secret)
payload = {"fields": ["persistent_menu"]}
async with httpx.AsyncClient(timeout=15.0) as client:
try:
resp = await client.delete(url, json=payload, params=params)
if resp.status_code == 200:
logger.info("Persistent menu deleted successfully")
return {"success": True, "data": resp.json()}
logger.error("Failed to delete persistent menu: %d %s", resp.status_code, resp.text[:200])
return {"success": False, "error": resp.text[:200]}
except httpx.RequestError as exc:
logger.error("Persistent menu delete network error: %s", exc)
return {"success": False, "error": str(exc)}
async def get_persistent_menu(
access_token: str,
*,
app_secret: str = "",
api_version: str = "v24.0",
) -> dict:
url = _build_menu_api_url(api_version)
params: dict = {
"access_token": access_token,
"fields": "persistent_menu",
}
if app_secret:
params["appsecret_proof"] = _compute_appsecret_proof(access_token, app_secret)
async with httpx.AsyncClient(timeout=15.0) as client:
try:
resp = await client.get(url, params=params)
if resp.status_code == 200:
data = resp.json()
return {"success": True, "data": data.get("data", [])}
logger.error("Failed to get persistent menu: %d %s", resp.status_code, resp.text[:200])
return {"success": False, "error": resp.text[:200]}
except httpx.RequestError as exc:
logger.error("Persistent menu get network error: %s", exc)
return {"success": False, "error": str(exc)}