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,137 @@
+from nonebot import require
+
+require("nonebot_plugin_orm")
+
+import asyncio
+import random
+import time
+
+from nonebot import on_message, require, logger, get_adapter
+from nonebot.adapters.onebot.v11 import (
+    GroupMessageEvent,
+    GROUP,
+    Message,
+    ActionFailed,
+    Adapter,
+)
+from nonebot.params import Arg
+from nonebot.plugin import PluginMetadata
+from nonebot.rule import Rule
+from nonebot.typing import T_State
+from nonebot_plugin_orm import get_session
+from sqlalchemy import insert
+from .handler import LearningChat
+from .models import ChatMessage
+from .config import config_manager, NICKNAME
+from . import web_api
+from . import web_hub
+from hexi.web_plugin_registry import register_web_plugin
+
+require("nonebot_plugin_apscheduler")
+from nonebot_plugin_apscheduler import scheduler
+
+# 注册到统一 Web 管理台（/hub）：hub 启动时挂载 /api/learning_chat
+register_web_plugin(
+    "learning_chat",
+    "群聊学习",
+    "chat",
+    lambda: web_hub.build_admin_app(),
+    module_name=__name__,
+)
+
+__plugin_meta__ = PluginMetadata(
+    name="群聊学习",
+    description="学习群友们的发言、复读以及主动发言",
+    usage="详见README",
+    type="application",
+    homepage="https://github.com/CMHopeSunshine/nonebot-plugin-learning-chat",
+    supported_adapters={"~onebot.v11"},
+    extra={"author": "惜月"},
+)
+
+
+async def ChatRule(event: GroupMessageEvent, state: T_State) -> bool:
+    if answers := await LearningChat(event).answer():
+        state["answers"] = answers
+        return True
+    return False
+
+
+learning_chat = on_message(
+    priority=99,
+    block=False,
+    rule=Rule(ChatRule),
+    permission=GROUP,
+    state={
+        "pm_name": "群聊学习",
+        "pm_description": "(被动技能)bot会学习群友们的发言",
+        "pm_usage": "群聊学习",
+        "pm_priority": 1,
+    },
+)
+
+
+@learning_chat.handle()
+async def _(event: GroupMessageEvent, answers=Arg("answers")):
+    for answer in answers:
+        try:
+            logger.info(
+                "群聊学习", f'{NICKNAME}将向群<m>{event.group_id}</m>回复<m>"{answer}"</m>'
+            )
+            msg = await learning_chat.send(Message(answer))
+            async with get_session() as session:
+                session.add(ChatMessage(
+                    group_id=event.group_id,
+                    user_id=event.self_id,
+                    message_id=msg["message_id"],
+                    message=answer,
+                    raw_message=answer,
+                    time=int(time.time()),
+                    plain_text=Message(answer).extract_plain_text(),
+                ))
+                await session.commit()
+            await asyncio.sleep(random.random() + 0.5)
+        except ActionFailed:
+            logger.info(
+                "群聊学习",
+                f'{NICKNAME}向群<m>{event.group_id}</m>的回复<m>"{answer}"</m>发送<r>失败，可能处于风控中</r>',
+            )
+
+
+@scheduler.scheduled_job("interval", minutes=3, misfire_grace_time=5)
+async def speak_up():
+    if not config_manager.config.total_enable:
+        return
+    try:
+        bots = get_adapter(Adapter).bots
+        if len(bots) == 0:
+            return
+        bot = list(bots.values())[0]
+    except ValueError:
+        return
+    if not (speak := await LearningChat.speak(int(bot.self_id))):
+        return
+    group_id, messages = speak
+    for msg in messages:
+        try:
+            logger.info("群聊学习", f'{NICKNAME}向群<m>{group_id}</m>主动发言<m>"{msg}"</m>')
+            send_result = await bot.send_group_msg(
+                group_id=group_id, message=Message(msg)
+            )
+            async with get_session() as session:
+                session.add(ChatMessage(
+                    group_id=group_id,
+                    user_id=int(bot.self_id),
+                    message_id=send_result["message_id"],
+                    message=msg,
+                    raw_message=msg,
+                    time=int(time.time()),
+                    plain_text=Message(msg).extract_plain_text(),
+                ))
+                await session.commit()
+            await asyncio.sleep(random.randint(2, 4))
+        except ActionFailed:
+            logger.info(
+                "群聊学习",
+                f'{NICKNAME}向群<m>{group_id}</m>主动发言<m>"{msg}"</m><r>发送失败，可能处于风控中</r>',
+            )