87 lines
3.2 KiB
Python
87 lines
3.2 KiB
Python
|
"""Steal plugin for garden system."""
|
|||
|
|
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
from typing import Optional
|
|||
|
|
|
|||
|
|
from PWF.Convention.Runtime.Architecture import Architecture
|
|||
|
|
|
|||
|
|
from PWF.CoreModules.database import get_db
|
|||
|
|
from Plugins.WPSBackpackSystem import WPSBackpackSystem
|
|||
|
|
|
|||
|
|
from .garden_plugin_base import WPSGardenBase
|
|||
|
|
|
|||
|
|
|
|||
|
|
class WPSGardenSteal(WPSGardenBase):
|
|||
|
|
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"]
|