95 lines
3.6 KiB
Python
95 lines
3.6 KiB
Python
from __future__ import annotations
|
|
import asyncio
|
|
from aiohttp import web
|
|
from loguru import logger
|
|
from typing import Any
|
|
import json
|
|
|
|
from nanobot.channels.base import BaseChannel
|
|
from nanobot.bus.events import InboundMessage, OutboundMessage
|
|
|
|
class DashboardChannel(BaseChannel):
|
|
"""
|
|
专门为管理面板设计的渠道。
|
|
它充当机器人内部总线 (Bus) 与宿主机面板之间的桥梁。
|
|
"""
|
|
|
|
def __init__(self, config: Any, bus: Any, host: str = "0.0.0.0", port: int = 9000):
|
|
super().__init__(config, bus)
|
|
self.host = host
|
|
self.port = port
|
|
self.runner = None
|
|
|
|
async def start(self) -> None:
|
|
"""启动 Dashboard HTTP 服务"""
|
|
app = web.Application()
|
|
app.router.add_post("/chat", self._handle_chat)
|
|
|
|
self.runner = web.AppRunner(app)
|
|
await self.runner.setup()
|
|
site = web.TCPSite(self.runner, self.host, self.port)
|
|
|
|
await site.start()
|
|
self._is_running = True
|
|
logger.info(f"🚀 Dashboard Channel 代理已上线，监听端口: {self.port}")
|
|
|
|
async def stop(self) -> None:
|
|
"""停止服务"""
|
|
if self.runner:
|
|
await self.runner.cleanup()
|
|
self._is_running = False
|
|
logger.info("Dashboard Channel 已下线")
|
|
|
|
async def send(self, message: OutboundMessage) -> None:
|
|
"""
|
|
从总线 (Bus) 接收机器人发出的所有消息，并结构化输出到 stdout。
|
|
"""
|
|
media = [str(v).strip().replace("\\", "/") for v in (message.media or []) if str(v).strip()]
|
|
if not message.content and not media:
|
|
return
|
|
|
|
# 核心：从元数据识别消息类型（进度更新 vs 最终回复）
|
|
is_progress = message.metadata.get("_progress", False)
|
|
is_tool_hint = message.metadata.get("_tool_hint", False)
|
|
|
|
payload = {
|
|
"type": "BUS_EVENT",
|
|
"source": "dashboard_channel",
|
|
"is_progress": is_progress,
|
|
"is_tool": is_tool_hint,
|
|
"content": message.content,
|
|
"media": media,
|
|
}
|
|
|
|
# 使用 JSON 格式输出，方便面板后端精准解析，告别正则
|
|
print(f"\n__DASHBOARD_DATA_START__{json.dumps(payload, ensure_ascii=False)}__DASHBOARD_DATA_END__\n", flush=True)
|
|
|
|
async def _handle_chat(self, request: web.Request) -> web.Response:
|
|
"""处理来自面板的指令入站"""
|
|
try:
|
|
data = await request.json()
|
|
user_message = data.get("message", "").strip()
|
|
media = [str(v).strip().replace("\\", "/") for v in (data.get("media") or []) if str(v).strip()]
|
|
|
|
if not user_message and not media:
|
|
return web.json_response({"status": "error", "reason": "empty message and media"}, status=400)
|
|
if not user_message:
|
|
user_message = "[attachment message]"
|
|
|
|
# 调试日志：打印收到的原始消息长度和前 20 个字符，确保中文未乱码
|
|
logger.info(f"📥 [Dashboard Channel] 收到指令 (len={len(user_message)}): {user_message[:20]}...")
|
|
|
|
# 将消息塞入总线
|
|
await self.bus.publish_inbound(InboundMessage(
|
|
channel="dashboard",
|
|
sender_id="user",
|
|
chat_id="direct",
|
|
content=user_message,
|
|
media=media,
|
|
))
|
|
|
|
return web.json_response({"status": "ok"})
|
|
except Exception as e:
|
|
logger.error(f"❌ Dashboard Channel 接收指令失败: {e}")
|
|
return web.json_response({"status": "error", "reason": str(e)}, status=500)
|