154 lines
5.8 KiB
Python
154 lines
5.8 KiB
Python
"""Place trap plugin for garden system."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Optional, Sequence
|
|
|
|
from PWF.Convention.Runtime.Architecture import *
|
|
|
|
from Plugins.WPSConfigSystem import WPSConfigAPI
|
|
from Plugins.WPSAPI import GuideEntry
|
|
from Plugins.WPSBackpackSystem import WPSBackpackSystem
|
|
|
|
from .garden_plugin_base import WPSGardenBase
|
|
from .garden_models import GARDEN_TRAPS_DICT
|
|
|
|
|
|
class WPSGardenPlaceTrap(WPSGardenBase):
|
|
#@override
|
|
#def get_webhook_url(self, message: str, chat_id: int, user_id: int) -> str:
|
|
# config : WPSConfigAPI = Architecture.Get(WPSConfigAPI)
|
|
# url = config.get_user_url(user_id)
|
|
# if url:
|
|
# return url
|
|
# else:
|
|
# return super().get_webhook_url(message, chat_id, user_id)
|
|
|
|
def get_guide_subtitle(self) -> str:
|
|
return "在地块上放置防护陷阱"
|
|
|
|
def collect_command_entries(self) -> Sequence[GuideEntry]:
|
|
return (
|
|
GuideEntry(
|
|
title="放置陷阱",
|
|
identifier="放置陷阱 <地块序号> <陷阱物品>",
|
|
description="在地块上放置防护陷阱，当偷盗者触发时会受到惩罚。",
|
|
metadata={"别名": "place_trap"},
|
|
icon="🪤",
|
|
details=[
|
|
{
|
|
"type": "steps",
|
|
"items": [
|
|
"输入地块序号和陷阱物品名称或ID。",
|
|
"系统检查地块是否存在且背包中是否有该陷阱。",
|
|
"成功放置后陷阱会在下次偷盗时生效。",
|
|
],
|
|
},
|
|
"陷阱有不同的触发概率、罚金和禁止时长，等级越高效果越强。",
|
|
],
|
|
),
|
|
)
|
|
|
|
def collect_guide_entries(self) -> Sequence[GuideEntry]:
|
|
return (
|
|
{
|
|
"title": "指令格式",
|
|
"description": "`放置陷阱 <地块序号> <陷阱物品>`。",
|
|
},
|
|
{
|
|
"title": "陷阱效果",
|
|
"description": "不同等级的陷阱具有不同的触发概率、罚金和禁止时长。",
|
|
},
|
|
)
|
|
|
|
def wake_up(self) -> None:
|
|
super().wake_up()
|
|
self.register_plugin("放置陷阱")
|
|
self.register_plugin("place_trap")
|
|
|
|
def _resolve_trap_id(self, keyword: str) -> Optional[str]:
|
|
"""解析陷阱物品ID"""
|
|
key = keyword.strip().lower()
|
|
for trap_item_id, trap in GARDEN_TRAPS_DICT.items():
|
|
if trap_item_id.lower() == key:
|
|
return trap_item_id
|
|
if trap.display_name.lower() == key:
|
|
return trap_item_id
|
|
if trap.display_name.lower().replace("陷阱", "").replace("守护", "").replace("结界", "").replace("网", "") == key:
|
|
return trap_item_id
|
|
return None
|
|
|
|
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
|
|
)
|
|
|
|
if not tokens[0].isdigit():
|
|
return await self.send_markdown_message(
|
|
"❌ 指令格式：`放置陷阱 <地块序号> <陷阱物品>`",
|
|
chat_id, user_id
|
|
)
|
|
|
|
plot_index = int(tokens[0])
|
|
trap_identifier = " ".join(tokens[1:])
|
|
|
|
trap_item_id = self._resolve_trap_id(trap_identifier)
|
|
if not trap_item_id:
|
|
return await self.send_markdown_message(
|
|
"❌ 未找到该陷阱物品，可用陷阱：\n" +
|
|
"\n".join([f"- {trap.display_name} ({trap_item_id})"
|
|
for trap_item_id, trap in GARDEN_TRAPS_DICT.items()]),
|
|
chat_id, user_id
|
|
)
|
|
|
|
# 检查背包中是否有该陷阱
|
|
backpack: WPSBackpackSystem = Architecture.Get(WPSBackpackSystem)
|
|
user_items = backpack.get_user_items(user_id)
|
|
owned_quantity = 0
|
|
for item in user_items:
|
|
if item.item_id == trap_item_id:
|
|
owned_quantity = item.quantity
|
|
break
|
|
|
|
if owned_quantity <= 0:
|
|
trap = GARDEN_TRAPS_DICT[trap_item_id]
|
|
return await self.send_markdown_message(
|
|
f"❌ 背包中没有 {trap.display_name}，需要先合成获取",
|
|
chat_id, user_id
|
|
)
|
|
|
|
try:
|
|
self.service().place_trap(
|
|
user_id=user_id,
|
|
plot_index=plot_index,
|
|
trap_item_id=trap_item_id,
|
|
)
|
|
# 消耗陷阱物品
|
|
backpack.set_item_quantity(user_id, trap_item_id, owned_quantity - 1)
|
|
|
|
trap = GARDEN_TRAPS_DICT[trap_item_id]
|
|
return await self.send_markdown_message(
|
|
f"✅ 已在地块 {plot_index} 上放置 {trap.display_name}\n"
|
|
f"- 触发概率：{trap.trigger_rate * 100:.0f}%\n"
|
|
f"- 罚金：{trap.fine_points} 分\n"
|
|
f"- 禁止时长：{trap.ban_hours} 小时\n"
|
|
f"- 耐久度：{trap.durability} 次",
|
|
chat_id, user_id
|
|
)
|
|
except ValueError as exc:
|
|
return await self.send_markdown_message(f"❌ {exc}", chat_id, user_id)
|
|
|
|
|
|
__all__ = ["WPSGardenPlaceTrap"]
|
|
|