新增战斗/冒险系统

Plugins/WPSCombatSystem/combat_plugin_equipment.py

@@ -0,0 +1,159 @@
+"""装备管理插件 - 装备和卸下物品"""
+
+from __future__ import annotations
+
+from typing import Optional
+
+from PWF.Convention.Runtime.GlobalConfig import ConsoleFrontColor, ProjectConfig
+
+from .combat_plugin_base import WPSCombatBase
+
+
+logger: ProjectConfig = ProjectConfig()
+
+
+class WPSCombatEquipment(WPSCombatBase):
+    """装备管理插件"""
+    
+    def is_enable_plugin(self) -> bool:
+        return True
+    
+    def wake_up(self) -> None:
+        super().wake_up()
+        logger.Log(
+            "Info",
+            f"{ConsoleFrontColor.GREEN}WPSCombatEquipment 插件已加载{ConsoleFrontColor.RESET}"
+        )
+        self.register_plugin("装备")
+        self.register_plugin("equip")
+        self.register_plugin("卸下")
+        self.register_plugin("unequip")
+    
+    async def callback(self, message: str, chat_id: int, user_id: int) -> Optional[str]:
+        """
+        处理装备相关命令
+        
+        命令格式：
+        - 装备 <物品名>
+        - 卸下 <槽位>
+        """
+        message = self.parse_message_after_at(message).strip()
+        
+        if not message:
+            return await self.send_markdown_message(
+                self._help_message(),
+                chat_id,
+                user_id
+            )
+        
+        tokens = message.split(maxsplit=1)
+        if not tokens:
+            return await self.send_markdown_message(
+                self._help_message(),
+                chat_id,
+                user_id
+            )
+        
+        # 判断是装备还是卸下
+        # 注意：由于命令已经通过register_plugin匹配，这里tokens[0]可能为空
+        # 实际命令词已经被消费了
+        
+        # 如果消息为空，说明只输入了命令词
+        if not tokens or len(tokens) == 0:
+            return await self.send_markdown_message(
+                self._help_message(),
+                chat_id,
+                user_id
+            )
+        
+        # tokens[0] 应该是物品名或槽位
+        target = tokens[0] if len(tokens) > 0 else ""
+        
+        # 通过检查self被哪个命令触发来判断操作
+        # 但这里我们无法直接知道，所以通过参数判断
+        # 如果是槽位名称（weapon/helmet等），则是卸下操作
+        # 否则尝试装备
+        
+        slot_names = ["weapon", "helmet", "armor", "boots", "accessory",
+                     "武器", "头盔", "护甲", "鞋子", "饰品"]
+        
+        # 简化：检查是否包含"卸下"或"unequip"
+        # 由于命令注册了这些词，我们可以假设如果达到这里，就是对应的操作
+        
+        # 实际上，由于我们注册了"装备"和"卸下"两个命令，
+        # message已经去掉了命令词，只剩参数
+        # 所以我们需要重新解析原始消息来判断
+        
+        # 更简单的方法：检查target是否是有效槽位
+        slot_map = {
+            "weapon": "weapon", "武器": "weapon",
+            "helmet": "helmet", "头盔": "helmet",
+            "armor": "armor", "护甲": "armor",
+            "boots": "boots", "鞋子": "boots",
+            "accessory": "accessory", "饰品": "accessory",
+        }
+        
+        if target.lower() in slot_map:
+            # 卸下操作
+            return await self._handle_unequip(chat_id, user_id, slot_map[target.lower()])
+        else:
+            # 装备操作
+            return await self._handle_equip(chat_id, user_id, target)
+    
+    async def _handle_equip(self, chat_id: int, user_id: int, item_name: str) -> Optional[str]:
+        """处理装备命令"""
+        if not item_name:
+            return await self.send_markdown_message(
+                "❌ 请指定要装备的物品名称\n用法：`装备 <物品名>`",
+                chat_id,
+                user_id
+            )
+        
+        service = self.service()
+        
+        # 尝试通过名称查找装备item_id
+        from .combat_models import EQUIPMENT_REGISTRY
+        
+        item_id = None
+        for eq_id, eq_def in EQUIPMENT_REGISTRY.items():
+            if eq_def.name.lower() == item_name.lower() or eq_id.lower() == item_name.lower():
+                item_id = eq_id
+                break
+        
+        if not item_id:
+            return await self.send_markdown_message(
+                f"❌ 未找到装备：{item_name}",
+                chat_id,
+                user_id
+            )
+        
+        success, msg = service.equip_item(user_id, item_id)
+        return await self.send_markdown_message(msg, chat_id, user_id)
+    
+    async def _handle_unequip(self, chat_id: int, user_id: int, slot: str) -> Optional[str]:
+        """处理卸下命令"""
+        service = self.service()
+        success, msg = service.unequip_item(user_id, slot)
+        return await self.send_markdown_message(msg, chat_id, user_id)
+    
+    def _help_message(self) -> str:
+        """帮助信息"""
+        return """# ⚔️ 装备管理
+**命令格式：**
+- `装备 <物品名>`：装备指定物品
+- `卸下 <槽位>`：卸下指定槽位的装备
+
+**槽位名称：**
+- weapon/武器
+- helmet/头盔
+- armor/护甲
+- boots/鞋子
+- accessory/饰品
+
+**示例：**
+- `装备 木剑`
+- `卸下 weapon`
+"""
+
+
+__all__ = ["WPSCombatEquipment"]