90 lines
3.2 KiB
Python
90 lines
3.2 KiB
Python
|
"""Garden overview and selling plugin."""
|
|||
|
|
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
from typing import Optional
|
|||
|
|
|
|||
|
|
from PWF.Convention.Runtime.Architecture import Architecture
|
|||
|
|
|
|||
|
|
from Plugins.WPSBackpackSystem import WPSBackpackSystem
|
|||
|
|
from Plugins.WPSConfigSystem import WPSConfigAPI
|
|||
|
|
|
|||
|
|
from .garden_plugin_base import WPSGardenBase
|
|||
|
|
|
|||
|
|
|
|||
|
|
class WPSGardenView(WPSGardenBase):
|
|||
|
|
def wake_up(self) -> None:
|
|||
|
|
super().wake_up()
|
|||
|
|
self.register_plugin("garden")
|
|||
|
|
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_overview(chat_id, user_id)
|
|||
|
|
|
|||
|
|
tokens = [token.strip() for token in payload.split() if token.strip()]
|
|||
|
|
if tokens and tokens[0] in {"售出", "sell"}:
|
|||
|
|
return await self._handle_sell(tokens[1:], chat_id, user_id)
|
|||
|
|
|
|||
|
|
return await self._send_overview(chat_id, user_id)
|
|||
|
|
|
|||
|
|
async def _send_overview(self, chat_id: int, user_id: int) -> Optional[str]:
|
|||
|
|
overview = self.format_garden_overview(user_id)
|
|||
|
|
return await self.send_markdown_message(overview, chat_id, user_id)
|
|||
|
|
|
|||
|
|
async def _handle_sell(self, args: list[str], chat_id: int, user_id: int) -> Optional[str]:
|
|||
|
|
if len(args) < 2:
|
|||
|
|
return await self.send_markdown_message(
|
|||
|
|
"❌ 指令格式：`菜园 售出 <果实> <数量>`",
|
|||
|
|
chat_id,
|
|||
|
|
user_id,
|
|||
|
|
)
|
|||
|
|
identifier = args[0]
|
|||
|
|
try:
|
|||
|
|
quantity = int(args[1])
|
|||
|
|
except ValueError:
|
|||
|
|
return await self.send_markdown_message("❌ 数量必须是整数", chat_id, user_id)
|
|||
|
|
if quantity <= 0:
|
|||
|
|
return await self.send_markdown_message("❌ 数量必须大于0", chat_id, user_id)
|
|||
|
|
|
|||
|
|
crop = self.resolve_fruit_id(identifier)
|
|||
|
|
if crop is None:
|
|||
|
|
return await self.send_markdown_message("❌ 未找到对应果实", chat_id, user_id)
|
|||
|
|
|
|||
|
|
backpack: WPSBackpackSystem = Architecture.Get(WPSBackpackSystem)
|
|||
|
|
owned = 0
|
|||
|
|
for item in backpack.get_user_items(user_id):
|
|||
|
|
if item.item_id == crop.fruit_id:
|
|||
|
|
owned = item.quantity
|
|||
|
|
break
|
|||
|
|
if owned < quantity:
|
|||
|
|
return await self.send_markdown_message("❌ 果实数量不足", chat_id, user_id)
|
|||
|
|
|
|||
|
|
total_points, price_per = self.service().sell_fruit(
|
|||
|
|
user_id=user_id,
|
|||
|
|
fruit_id=crop.fruit_id,
|
|||
|
|
quantity=quantity,
|
|||
|
|
)
|
|||
|
|
backpack.set_item_quantity(user_id, crop.fruit_id, owned - quantity)
|
|||
|
|
|
|||
|
|
config_api: WPSConfigAPI = Architecture.Get(WPSConfigAPI)
|
|||
|
|
new_points = await config_api.adjust_user_points(
|
|||
|
|
chat_id,
|
|||
|
|
user_id,
|
|||
|
|
total_points,
|
|||
|
|
reason=f"出售 {crop.display_name} 的果实",
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
message = (
|
|||
|
|
"# 🛒 售出成功\n"
|
|||
|
|
f"- 果实：{crop.display_name} × {quantity}\n"
|
|||
|
|
f"- 单价：{price_per} 分\n"
|
|||
|
|
f"- 总计：{total_points} 分\n"
|
|||
|
|
f"- 当前积分：{new_points}"
|
|||
|
|
)
|
|||
|
|
return await self.send_markdown_message(message, chat_id, user_id)
|
|||
|
|
|
|||
|
|
|
|||
|
|
__all__ = ["WPSGardenView"]
|