NewWPSBot/Plugins/WPSGardenSystem/garden_plugin_steal.py

"""Steal plugin for garden system."""

from __future__ import annotations

from typing import Optional, Sequence

from PWF.Convention.Runtime.Architecture import Architecture

from PWF.CoreModules.database import get_db
from Plugins.WPSAPI import GuideEntry
from Plugins.WPSBackpackSystem import WPSBackpackSystem

from .garden_plugin_base import WPSGardenBase


class WPSGardenSteal(WPSGardenBase):
    def get_guide_subtitle(self) -> str:
        return "偷取其他用户成熟果实的互动指令"

    def collect_command_entries(self) -> Sequence[GuideEntry]:
        return (
            GuideEntry(
                title="偷取",
                identifier="偷取 <用户> <地块序号>",
                description="从其他用户的成熟作物中偷取果实。",
                metadata={"别名": "steal"},
                icon="🕵️",
                details=[
                    {
                        "type": "steps",
                        "items": [
                            "输入目标用户 ID 或昵称，以及成熟地块序号。",
                            "系统校验目标用户菜园与成熟状态。",
                            "成功后获得 1 个果实并通知对方被偷记录。",
                        ],
                    },
                    "同一地块可多次被不同用户偷取，超限后将提示果实不足。",
                ],
            ),
        )

    def collect_guide_entries(self) -> Sequence[GuideEntry]:
        return (
            {
                "title": "指令格式",
                "description": "`偷取 <用户ID|昵称> <地块序号>`，不可针对自己。",
            },
            {
                "title": "通知机制",
                "description": "成功偷取后会向目标用户推送警报消息，包含剩余果实数量。",
            },
        )

    def wake_up(self) -> None:
        super().wake_up()
        self.register_plugin("steal")
        self.register_plugin("偷取")

    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 len(tokens) < 2:
            return await self.send_markdown_message("❌ 指令格式：`偷取 <用户> <地块序号>`", chat_id, user_id)
        target_identifier = tokens[0]
        if not tokens[1].isdigit():
            return await self.send_markdown_message("❌ 指令格式：`偷取 <用户> <地块序号>`", chat_id, user_id)
        plot_index = int(tokens[1])

        owner_id = self._resolve_user_identifier(target_identifier)
        if owner_id is None:
            return await self.send_markdown_message("❌ 未找到目标用户", chat_id, user_id)
        if owner_id == user_id:
            return await self.send_markdown_message("❌ 不能偷取自己的菜园", chat_id, user_id)

        try:
            result = self.service().steal(
                thief_id=user_id,
                owner_id=owner_id,
                plot_index=plot_index,
            )
        except ValueError as exc:
            return await self.send_markdown_message(f"❌ {exc}", chat_id, user_id)

        crop = result["crop"]
        backpack: WPSBackpackSystem = Architecture.Get(WPSBackpackSystem)
        backpack.add_item(user_id, crop.fruit_id, result["stolen_quantity"])

        remaining = result["remaining"]
        message = (
            "# 🕵️ 偷取成功\n"
            f"- 目标：{crop.display_name}\n"
            f"- 获得：{crop.display_name}的果实 × {result['stolen_quantity']}\n"
            f"- 目标剩余果实：{remaining}"
        )
        await self.send_markdown_message(message, chat_id, user_id)

        owner_chat = result.get("chat_id")
        if owner_chat:
            owner_message = (
                "# ⚠️ 菜园警报\n"
                f"- 你的 {crop.display_name} 被偷走了 1 个果实\n"
                f"- 当前剩余果实：{remaining}"
            )
            await self.send_markdown_message(owner_message, owner_chat, owner_id)
        return None

    def _resolve_user_identifier(self, identifier: str) -> Optional[int]:
        text = identifier.strip()
        if text.isdigit():
            return int(text)
        cursor = get_db().conn.cursor()
        cursor.execute(
            "SELECT user_id FROM user_info WHERE username = ? COLLATE NOCASE",
            (text,),
        )
        row = cursor.fetchone()
        if row:
            return int(row["user_id"])
        return None


__all__ = ["WPSGardenSteal"]