Files

50 lines
1.5 KiB
Python
Raw Permalink Normal View History

"""掷骰子 —— 触发器(Trigger)层。"""
from __future__ import annotations
import re
from nonebot import on_regex, on_startswith, require
from nonebot.adapters.onebot.v11 import MessageEvent
from nonebot.typing import T_State
from nonebot_plugin_alconna import UniMessage
from ..services.dice import build_dice_message
require("nonebot_plugin_alconna")
dice = on_regex(
re.compile(
r"^\.r\s*(?:(?P<num>\d{0,2})d(?:(?P<min>\d{1,4})~)?(?P<max>\d{0,4})(?:(?P<opr>[+-])(?P<offset>\d{0,5}))?)?\b",
re.I,
)
)
marriage = on_startswith(".qj")
@dice.handle()
async def dice_handle(ev: MessageEvent, state: T_State):
num, min_, max_, opr, offset = 1, 1, 100, 1, 0
matched = state.get("_matched_dict") or {}
if s := matched.get("num"):
num = int(s)
if s := matched.get("min"):
min_ = int(s)
if s := matched.get("max"):
max_ = int(s)
if s := matched.get("opr"):
opr = -1 if s == "-" else 1
if s := matched.get("offset"):
offset = int(s)
await UniMessage.text(build_dice_message(num, min_, max_, opr, offset)).send(
at_sender=True
)
@marriage.handle()
async def marriage_handle(ev: MessageEvent):
text = ev.message.extract_plain_text().strip()
wife = text[len(".qj") :].strip()
tip = f"与{wife}的ケッコンカッコカリ结果是:" if wife else "的ケッコンカッコカリ结果是:"
await UniMessage.text(build_dice_message(1, 3, 6, 1, 0, tip)).send(at_sender=True)