"""掷骰子 —— 触发器(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)