126 lines
4.8 KiB
Python
126 lines
4.8 KiB
Python
"""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("偷取")
|
||
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"]
|