feat: add hit someone middleware

adapters/tg.py

@@ -10,6 +10,7 @@ from aiogram.filters import CommandStart, Command
 from aiogram.client.session.aiohttp import AiohttpSession
 from aiogram import F
 
+from core.middleware.rikki import RikkiMiddleware
 from core.post_to_fedi import router as fedi_router
 
 from core.bitflip import handle_bitflip_command
@@ -32,6 +33,7 @@ class TelegramAdapter:
     def __init__(self):
         self.dp = Dispatcher()
         self.stats_middleware = MessageStatsMiddleware()
+        self.hit_middleware = RikkiMiddleware()
         self._setup_middleware()
         self._setup_handlers()
 
@@ -41,6 +43,7 @@ class TelegramAdapter:
         router = Router()
         actions_router = Router()
         repeater_router = Router()
+        unpin_router = Router()
         dummy_router = Router()
 
         # Register handlers on router
@@ -63,7 +66,7 @@ class TelegramAdapter:
         router.message(F.text.contains('http') & ~F.text.contains('/report_broken_links'))(handle_links)
         # unpin 模块
         # 不知道为什么检测不到频道的消息被置顶这个事件，暂时认为所有的频道消息都是被置顶的
-        router.message(F.chat.type.in_({'group', 'supergroup'}) & F.sender_chat & (
+        unpin_router.message(F.chat.type.in_({'group', 'supergroup'}) & F.sender_chat & (
                     F.sender_chat.type == 'channel') & F.is_automatic_forward)(
             handle_unpin_channel_message)
         # repeater 模块
@@ -77,6 +80,7 @@ class TelegramAdapter:
         actions_router.message(F.chat.type.in_({'group', 'supergroup'}) & F.text.startswith('\\'))(handle_reverse_actions)
 
         # Include router in dispatcher
+        self.dp.include_router(unpin_router)
         # 通用的路由
         self.dp.include_router(router)
         self.dp.include_router(actions_router)
@@ -88,7 +92,7 @@ class TelegramAdapter:
     def _setup_middleware(self):
         """注册中间件"""
         self.dp.message.middleware(self.stats_middleware)
-
+        self.dp.message.middleware(self.hit_middleware)
 
     async def start(self):
         """Start the Telegram bot"""

core/middleware/rikki.py

@@ -0,0 +1,91 @@
+import random
+from aiogram import BaseMiddleware
+from aiogram.types import Message
+from typing import Dict, Optional, Callable, Awaitable, Any
+
+
+class RikkiMiddleware(BaseMiddleware):
+    def __init__(self, target_user_id: str = "5545347637"):
+        # 存储每个用户的触发几率，初始值在40-50%之间
+        self.user_probabilities: Dict[str, float] = {}
+        # 触发关键词
+        self.trigger_plus_keywords = ["啃","羡慕", "xm", "xmsl", "羡慕死了", "我菜"]
+        self.trigger_plusplus_keywords = ["kig","电你","被电","⚡","tk","打"]
+        self.target_user_id = target_user_id
+
+    def get_user_probability(self, user_id: str) -> float:
+        """获取用户当前触发几率，如果不存在则初始化"""
+        if user_id != self.target_user_id:
+            return 0.0 # 非目标用户返回默认值
+        if user_id not in self.user_probabilities:
+            self.user_probabilities[user_id] = random.uniform(10.0, 40.0)
+        return self.user_probabilities[user_id]
+
+    def update_probability(self, user_id: str, message: str, hit_by_others = False) -> float:
+        """更新用户触发几率"""
+        if user_id != self.target_user_id:
+            return 0.0 # 非目标用户不更新
+        current_prob = self.get_user_probability(user_id)
+
+        # 基础减少0.1%
+        current_prob -= 0.1
+
+        # 检查是否包含触发关键词
+        for keyword in self.trigger_plus_keywords:
+            if keyword in message:
+                current_prob *= 1.14
+                break
+        for keyword in self.trigger_plusplus_keywords:
+            if keyword in message:
+                current_prob *= 1.91
+                break
+        if hit_by_others:
+            # 如果被其他人触发，增加额外几率
+            current_prob += 20.0
+        self.user_probabilities[user_id] = current_prob
+        return current_prob
+
+    def should_trigger(self, user_id: str) -> bool:
+        """检查是否应该触发回复"""
+        if user_id != self.target_user_id:
+            return False  # 非目标用户不触发
+        probability = self.get_user_probability(user_id)
+        return probability >= 100.0
+
+    def reset_probability(self, user_id: str):
+        """重置用户触发几率到初始随机值"""
+        self.user_probabilities[user_id] = random.uniform(40.0, 50.0)
+
+    async def __call__(
+            self,
+            handler: Callable[[Message, Dict[str, Any]], Awaitable[Any]],
+            event: Message,
+            data: Dict[str, Any]
+    ) -> Any:
+        """
+        处理消息的主要方法
+        """
+        if not event.text:
+            return await handler(event, data)
+        user_id = str(event.from_user.id)
+        if event.chat.type in ['group', 'supergroup'] and user_id == self.target_user_id:
+            # 更新几率
+            self.update_probability(user_id, event.text)
+
+        if event.text and event.text.startswith('/打') and event.reply_to_message and str(event.reply_to_message.from_user.id) == self.target_user_id:
+            self.update_probability(user_id, event.text, hit_by_others=True)
+
+        if self.get_user_probability(user_id) >= 80.0:
+            await event.reply("泥欠打了")
+
+        # 检查是否触发
+        if self.should_trigger(user_id):
+            # 重置几率
+            self.reset_probability(user_id)
+            await event.reply("/打")
+        return await handler(event, data)
+
+    def get_probability_status(self, user_id: str) -> str:
+        """获取用户当前几率状态（用于调试）"""
+        prob = self.get_user_probability(user_id)
+        return f"当前触发几率: {prob:.2f}%"