feat:初版

plugins/DogDiary/main.py

@@ -0,0 +1,133 @@
+"""
+舔狗日记插件
+
+用户发送"舔狗"或"舔狗日记"触发，返回随机舔狗日记
+"""
+
+import tomllib
+import aiohttp
+from pathlib import Path
+from loguru import logger
+from utils.plugin_base import PluginBase
+from utils.decorators import on_text_message
+from WechatHook import WechatHookClient
+
+
+class DogDiary(PluginBase):
+    """舔狗日记插件"""
+
+    # 插件元数据
+    description = "舔狗日记插件，发送舔狗日记"
+    author = "ShiHao"
+    version = "1.0.0"
+
+    def __init__(self):
+        super().__init__()
+        self.config = None
+
+    async def async_init(self):
+        """插件异步初始化"""
+        # 读取配置
+        config_path = Path(__file__).parent / "config.toml"
+        with open(config_path, "rb") as f:
+            self.config = tomllib.load(f)
+
+        logger.success("[DogDiary] 舔狗日记插件已加载")
+
+    @on_text_message(priority=50)
+    async def handle_dog_diary(self, bot: WechatHookClient, message: dict):
+        """处理舔狗日记请求"""
+        content = message.get("Content", "").strip()
+        from_wxid = message.get("FromWxid", "")
+        is_group = message.get("IsGroup", False)
+
+        # 检查是否是触发关键词（精确匹配）
+        if content not in self.config["behavior"]["keywords"]:
+            return
+
+        # 检查群聊/私聊过滤
+        if is_group:
+            if not self.config["behavior"]["enable_group"]:
+                return
+            if not self._should_handle_group(from_wxid):
+                return
+        else:
+            if not self.config["behavior"]["enable_private"]:
+                return
+
+        logger.info(f"[DogDiary] 收到舔狗日记请求")
+
+        # 调用 API 获取舔狗日记
+        try:
+            diary_text = await self._fetch_dog_diary()
+            if diary_text:
+                await bot.send_text(from_wxid, diary_text)
+                logger.success(f"[DogDiary] 发送成功: {diary_text[:30]}...")
+            else:
+                await bot.send_text(from_wxid, "❌ 获取舔狗日记失败，请稍后再试")
+                logger.warning("[DogDiary] 获取舔狗日记失败")
+        except Exception as e:
+            logger.error(f"[DogDiary] 处理请求失败: {e}")
+            import traceback
+            logger.error(f"详细错误: {traceback.format_exc()}")
+            await bot.send_text(from_wxid, "❌ 获取舔狗日记失败，请稍后再试")
+
+        # 返回 False 阻止消息继续传播
+        return False
+
+    async def _fetch_dog_diary(self) -> str:
+        """调用 API 获取舔狗日记"""
+        api_url = self.config["api"]["url"]
+        timeout = self.config["api"]["timeout"]
+
+        try:
+            async with aiohttp.ClientSession() as session:
+                async with session.get(
+                    api_url,
+                    timeout=aiohttp.ClientTimeout(total=timeout)
+                ) as response:
+                    if response.status != 200:
+                        logger.error(f"[DogDiary] API 请求失败: HTTP {response.status}")
+                        return None
+
+                    result = await response.json()
+                    logger.debug(f"[DogDiary] API 返回: {result}")
+
+                    # 检查返回状态
+                    code = result.get("code")
+                    if code != 200:
+                        logger.error(f"[DogDiary] API 返回错误: {result.get('msg', '未知错误')}")
+                        return None
+
+                    # 提取舔狗日记内容
+                    diary_text = result.get("data", "")
+                    if not diary_text:
+                        logger.error("[DogDiary] API 返回数据为空")
+                        return None
+
+                    return diary_text
+
+        except aiohttp.ClientError as e:
+            logger.error(f"[DogDiary] 网络请求失败: {e}")
+            return None
+        except Exception as e:
+            logger.error(f"[DogDiary] 获取舔狗日记失败: {e}")
+            import traceback
+            logger.error(f"详细错误: {traceback.format_exc()}")
+            return None
+
+    def _should_handle_group(self, room_wxid: str) -> bool:
+        """判断是否应该在该群处理请求"""
+        enabled_groups = self.config["behavior"]["enabled_groups"]
+        disabled_groups = self.config["behavior"]["disabled_groups"]
+
+        # 如果在禁用列表中，不处理
+        if room_wxid in disabled_groups:
+            return False
+
+        # 如果启用列表为空，对所有群生效
+        if not enabled_groups:
+            return True
+
+        # 否则只对启用列表中的群生效
+        return room_wxid in enabled_groups