#!/usr/bin/env python3
# -*- coding:utf-8 -*-
'''
File: __init__.py
Created Date: 2021-02-04 21:14:48
Author: Kyomotoi
Email: Kyomotoiowo@gmail.com
License: GPLv3
Project: https://github.com/Kyomotoi/ATRI
--------
Last Modified: Sunday, 7th March 2021 3:14:16 pm
Modified By: Kyomotoi (kyomotoiowo@gmail.com)
--------
Copyright (c) 2021 Kyomotoi
'''

import re

from nonebot.plugin import on_command
from nonebot.adapters.cqhttp import Bot, MessageEvent

from ATRI.rule import (
    is_in_banlist,
    is_in_dormant,
    is_in_service
)
from .data_source import roll_dice


__plugin_name__ = "roll"

roll = on_command(
    "/roll",
    rule=is_in_banlist() & is_in_dormant()
    & is_in_service(__plugin_name__)
)

@roll.handle()
async def _roll(bot: Bot, event: MessageEvent, state: dict) -> None:
    args = str(event.message).strip()
    if args:
        state['resu'] = args

@roll.got("resu", prompt="roll 参数不能为空~！\ndemo：1d10 或 2d10+2d10")
async def _(bot: Bot, event: MessageEvent, state: dict) -> None:
    resu = state['resu']
    match = re.match(r'^([\dd+\s]+?)$', resu)
    
    if not match:
        await roll.finish("请输入正确的参数！！\ndemo：1d10 或 2d10+2d10")
    
    await roll.finish(roll_dice(resu))


# __plugin_name__ = "fakemsg"

# fakemsg = on_command(
#     "/fakemsg",
#     rule=is_in_banlist() & is_in_dormant()
#     & is_in_service(__plugin_name__)
# )

# @fakemsg.handle()
# async def _fakemsg(bot: Bot, event: MessageEvent, state: dict) -> None:
#     ...

# @fakemsg.got()