"""Remove trap plugin for garden system."""

from __future__ import annotations

from typing import Optional, Sequence

from Plugins.WPSAPI import GuideEntry

from .garden_plugin_base import WPSGardenBase


class WPSGardenRemoveTrap(WPSGardenBase):
    def get_guide_subtitle(self) -> str:
        return "移除地块上的防护陷阱"

    def collect_command_entries(self) -> Sequence[GuideEntry]:
        return (
            GuideEntry(
                title="移除陷阱",
                identifier="移除陷阱 <地块序号>",
                description="移除地块上的陷阱。",
                metadata={"别名": "remove_trap"},
                icon="🗑️",
                details=[
                    {
                        "type": "steps",
                        "items": [
                            "输入地块序号。",
                            "系统检查地块是否存在且是否有陷阱。",
                            "成功移除陷阱。",
                        ],
                    },
                ],
            ),
        )

    def collect_guide_entries(self) -> Sequence[GuideEntry]:
        return (
            {
                "title": "指令格式",
                "description": "`移除陷阱 <地块序号>`。",
            },
        )

    def wake_up(self) -> None:
        super().wake_up()
        self.register_plugin("移除陷阱")
        self.register_plugin("remove_trap")

    async def callback(self, message: str, chat_id: int, user_id: int) -> Optional[str]:
        payload = self.parse_message_after_at(message).strip()
        if not payload:
            return await self.send_markdown_message(
                "❌ 指令格式：`移除陷阱 <地块序号>`", 
                chat_id, user_id
            )
        
        tokens = [token.strip() for token in payload.split() if token.strip()]
        if not tokens or not tokens[0].isdigit():
            return await self.send_markdown_message(
                "❌ 指令格式：`移除陷阱 <地块序号>`", 
                chat_id, user_id
            )
        
        plot_index = int(tokens[0])
        try:
            self.service().remove_trap(user_id=user_id, plot_index=plot_index)
            return await self.send_markdown_message(
                f"✅ 已移除地块 {plot_index} 上的陷阱", 
                chat_id, user_id
            )
        except ValueError as exc:
            return await self.send_markdown_message(f"❌ {exc}", chat_id, user_id)


__all__ = ["WPSGardenRemoveTrap"]