1 files changed, 51 insertions, 0 deletions
diff --git a/ATRI/plugins/utils/__init__.py b/ATRI/plugins/utils/__init__.py
index e69de29..1bf1461 100644
--- a/ATRI/plugins/utils/__init__.py
+++ b/ATRI/plugins/utils/__init__.py
@@ -0,0 +1,51 @@
+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__)
+)
+
+[email protected]()
+async def _roll(bot: Bot, event: MessageEvent, state: dict) -> None:
+    args = str(event.message).strip()
+    if args:
+        state['resu'] = args
+
+[email protected]("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()
+\ No newline at end of file