Kris
5946478772
新增小红书、XMPP、元宝、Zalo 四个渠道扩展。 小红书渠道扩展主要模块：config, gateway, webhook, outbound, streaming, pairing, security, dedupe, media, status, window XMPP 渠道扩展主要模块：plugin, config, gateway, outbound, streaming, pairing, security, dedupe, accounts, commands, muc, rate_limiter, stanza_utils, status, monitor 元宝渠道扩展主要模块：plugin, client, config_schema, gateway, outbound(chunk/queue/transport), inbound(dispatcher), streaming, pairing, security, accounts, actions, commands, codec(biz/conn), session, shared, utils Zalo 渠道扩展主要模块：api, config, gateway, webhook, outbound, pairing, security, session, polling, monitor, status
95 lines
3.3 KiB
Python
95 lines
3.3 KiB
Python
import logging
|
|
|
|
import httpx
|
|
|
|
from yuxi.channel.protocols import ChannelAccountSnapshot
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
PROBE_URL = "https://open.xiaohongshu.com/oauth/access_token"
|
|
|
|
|
|
class XiaohongshuStatus:
|
|
default_runtime = ChannelAccountSnapshot(account_id="default")
|
|
|
|
def __init__(self, gateway=None):
|
|
self._gateway = gateway
|
|
|
|
async def probe(self, account: dict | None = None) -> bool:
|
|
gateway = self._gateway
|
|
if gateway is None:
|
|
return False
|
|
|
|
token = gateway.access_token
|
|
if not token:
|
|
return False
|
|
|
|
url = f"{PROBE_URL}?access_token={token}"
|
|
try:
|
|
async with httpx.AsyncClient(timeout=10.0) as client:
|
|
resp = await client.get(url)
|
|
data = resp.json()
|
|
|
|
if data.get("code") == 0:
|
|
return True
|
|
|
|
logger.warning("Xiaohongshu probe failed: %s", data.get("msg", "unknown error"))
|
|
return False
|
|
except Exception:
|
|
logger.exception("Xiaohongshu probe failed")
|
|
return False
|
|
|
|
def build_summary(self, snapshot: object) -> dict:
|
|
if not isinstance(snapshot, ChannelAccountSnapshot):
|
|
return {}
|
|
return {
|
|
"account_id": snapshot.account_id,
|
|
"running": snapshot.running,
|
|
"connected": snapshot.connected,
|
|
"health_state": snapshot.health_state,
|
|
}
|
|
|
|
def build_account_snapshot(
|
|
self,
|
|
account: dict,
|
|
config: dict,
|
|
runtime: ChannelAccountSnapshot | None = None,
|
|
probe_result: object | None = None,
|
|
audit: object | None = None,
|
|
) -> ChannelAccountSnapshot:
|
|
gateway = self._gateway
|
|
token = gateway.access_token if gateway else None
|
|
return ChannelAccountSnapshot(
|
|
account_id=account.get("account_id", "default"),
|
|
name=account.get("name", ""),
|
|
enabled=account.get("enabled", True),
|
|
configured=bool(account.get("app_key") and account.get("app_secret")),
|
|
status_state="linked" if token else "not-linked",
|
|
running=getattr(runtime, "running", False) if runtime else False,
|
|
connected=bool(token),
|
|
last_message_at=getattr(runtime, "last_message_at", None) if runtime else None,
|
|
health_state="ok" if probe_result else "unknown",
|
|
dm_policy=account.get("dm_policy", "open"),
|
|
)
|
|
|
|
def collect_status_issues(self, accounts: list[ChannelAccountSnapshot]) -> list:
|
|
issues = []
|
|
for acc in accounts:
|
|
if not acc.configured:
|
|
issues.append({
|
|
"channel": "xiaohongshu",
|
|
"account_id": acc.account_id,
|
|
"kind": "not-configured",
|
|
"message": "Xiaohongshu credentials not configured",
|
|
"fix": "Set XIAOHONGSHU_APP_KEY and XIAOHONGSHU_APP_SECRET",
|
|
})
|
|
elif not acc.connected:
|
|
issues.append({
|
|
"channel": "xiaohongshu",
|
|
"account_id": acc.account_id,
|
|
"kind": "not-connected",
|
|
"message": "Xiaohongshu gateway not running",
|
|
"fix": "Check environment variables and gateway status",
|
|
})
|
|
return issues
|