Add HeXi bot codebase: custom plugins, web frontends, tests

- hexi core: message handling, rate limiting, cooldown, plugin manager - Custom plugins: BF stats, daily check-in, quotes, persona cards, etc. - Community plugins vendored under hexi/plugins with local fixes - Web admin frontends (learning-chat, persona-admin), unified hexi/web - Tests for rate_limit/cooldown/memes/persona; poetry.lock Co-Authored-By: Claude <noreply@anthropic.com>
2026-09-01 13:13:40 +08:00
parent 1783c60afa
commit b61d09f09f
3201 changed files with 160436 additions and 171 deletions
@@ -0,0 +1,93 @@
+import asyncio
+from dataclasses import dataclass
+from datetime import datetime
+from typing import TYPE_CHECKING
+
+from nonebot import get_bots, logger
+from nonebot.matcher import current_bot
+
+from ..config import config
+from ..misc_statistics import bot_connect_time, bot_info_cache, recv_num, send_num
+from ..util import format_timedelta
+from . import normal_collector
+
+if TYPE_CHECKING:
+    from nonebot.adapters import Bot as BaseBot
+
+try:
+    from nonebot.adapters.onebot.v11 import Bot as OBV11Bot
+except ImportError:
+    OBV11Bot = None
+
+
+@dataclass
+class BotStatus:
+    self_id: str
+    adapter: str
+    nick: str
+    bot_connected: str
+    msg_rec: str
+    msg_sent: str
+
+
+async def get_ob11_msg_num(bot: "BaseBot") -> tuple[int | None, int | None]:
+    if not (config.ps_ob_v11_use_get_status and OBV11Bot and isinstance(bot, OBV11Bot)):
+        return None, None
+
+    try:
+        bot_stat = (await bot.get_status()).get("stat")
+    except Exception as e:
+        logger.warning(
+            f"Error when getting bot status: {e.__class__.__name__}: {e}",
+        )
+        return None, None
+    if not bot_stat:
+        return None, None
+
+    msg_rec = bot_stat.get("message_received") or bot_stat.get(
+        "MessageReceived",
+    )
+    msg_sent = bot_stat.get("message_sent") or bot_stat.get("MessageSent")
+    return msg_rec, msg_sent
+
+
+async def get_bot_status(bot: "BaseBot", now_time: datetime) -> BotStatus:
+    nick = (
+        ((info := bot_info_cache[bot.self_id]).nick or info.name or info.id)
+        if (not config.ps_use_env_nick) and (bot.self_id in bot_info_cache)
+        else next(iter(config.nickname), None)
+    ) or "Bot"
+    bot_connected = (
+        format_timedelta(now_time - t)
+        if (t := bot_connect_time.get(bot.self_id))
+        else "未知"
+    )
+
+    msg_rec, msg_sent = await get_ob11_msg_num(bot)
+    if msg_rec is None:
+        msg_rec = recv_num.get(bot.self_id)
+    if msg_sent is None:
+        msg_sent = send_num.get(bot.self_id)
+    msg_rec = "未知" if (msg_rec is None) else str(msg_rec)
+    msg_sent = "未知" if (msg_sent is None) else str(msg_sent)
+
+    return BotStatus(
+        self_id=bot.self_id,
+        adapter=bot.adapter.get_name(),
+        nick=nick,
+        bot_connected=bot_connected,
+        msg_rec=msg_rec,
+        msg_sent=msg_sent,
+    )
+
+
+@normal_collector()
+async def bots() -> list[BotStatus]:
+    now_time = datetime.now().astimezone()
+    return (
+        [await get_bot_status(current_bot.get(), now_time)]
+        if config.ps_show_current_bot_only
+        else await asyncio.gather(
+            *(get_bot_status(bot, now_time) for bot in get_bots().values()),
+        )
+    )