NewWPSBot/Plugins/WPSCombatSystem/combat_plugin_equipment.py

"""装备管理插件 - 装备和卸下物品"""

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"]