🏷bump

15 files changed, 376 insertions, 0 deletions

diff --git a/ChienDice/config.toml b/ChienDice/config.toml
new file mode 100644
index 0000000..0c4aa7b
--- /dev/null
+++ b/ChienDice/config.toml
@@ -0,0 +1,14 @@
+[bot]
+plugins = []
+plugin_dirs = ["plugins"]
+adapters = ["iamai.adapter.cqhttp"]
+
+[bot.log]
+level = "INFO"
+verbose_exception = true
+
+[adapter.cqhttp]
+adapter_type = "reverse-ws"
+host = "127.0.0.1"
+port = 8080
+url = "/cqhttp/ws"
\ No newline at end of file
diff --git a/ChienDice/data/reply_data.json b/ChienDice/data/reply_data.json
new file mode 100644
index 0000000..e4977ee
--- /dev/null
+++ b/ChienDice/data/reply_data.json
@@ -0,0 +1,17 @@
+[
+  {
+    "rule": "hello",
+    "message": "你好！"
+  },
+  {
+    "rule": "test",
+    "message": [
+      {
+        "type": "text",
+        "data": {
+          "text": "测试"
+        }
+      }
+    ]
+  }
+]
\ No newline at end of file
diff --git a/ChienDice/main.py b/ChienDice/main.py
new file mode 100644
index 0000000..bb95e79
--- /dev/null
+++ b/ChienDice/main.py
@@ -0,0 +1,6 @@
+from iamai import Bot
+
+bot = Bot(hot_reload=True)
+
+if __name__ == "__main__":
+    bot.run()
diff --git a/ChienDice/plugins/iamai_plugin_base/__init__.py b/ChienDice/plugins/iamai_plugin_base/__init__.py
new file mode 100644
index 0000000..e99ce15
--- /dev/null
+++ b/ChienDice/plugins/iamai_plugin_base/__init__.py
@@ -0,0 +1,85 @@
+import re
+from abc import ABC, abstractmethod
+from typing import Type, Union, Generic, TypeVar
+
+from iamai import Plugin
+from iamai.typing import T_State
+from iamai.adapter.cqhttp.event import GroupMessageEvent, PrivateMessageEvent
+
+from .config import BasePluginConfig, RegexPluginConfig, CommandPluginConfig
+
+T_Config = TypeVar("T_Config", bound=BasePluginConfig)
+T_RegexPluginConfig = TypeVar("T_RegexPluginConfig", bound=RegexPluginConfig)
+T_CommandPluginConfig = TypeVar("T_CommandPluginConfig", bound=CommandPluginConfig)
+
+
+class BasePlugin(
+    Plugin[Union[PrivateMessageEvent, GroupMessageEvent], T_State, T_Config],
+    ABC,
+    Generic[T_State, T_Config],
+):
+    Config: Type[T_Config] = BasePluginConfig
+
+    def format_str(self, format_str: str, message_str: str = "") -> str:
+        return format_str.format(
+            message=message_str,
+            user_name=self.event.sender.nickname,
+            user_id=self.event.sender.user_id,
+        )
+
+    async def rule(self) -> bool:
+        if self.event.adapter.name != "cqhttp":
+            return False
+        if self.event.type != "message":
+            return False
+        if self.config.handle_all_message:
+            return self.str_match(self.event.message.get_plain_text())
+        elif self.config.handle_friend_message:
+            if self.event.message_type == "private":
+                return self.str_match(self.event.message.get_plain_text())
+        elif self.config.handle_group_message:
+            if self.event.message_type == "group":
+                if (
+                    self.config.accept_group is None
+                    or self.event.group_id in self.config.accept_group
+                ):
+                    return self.str_match(self.event.message.get_plain_text())
+        return False
+
+    @abstractmethod
+    def str_match(self, msg_str: str) -> bool:
+        raise NotImplemented
+
+
+class RegexPluginBase(BasePlugin[T_State, T_RegexPluginConfig], ABC):
+    msg_match: re.Match
+    re_pattern: re.Pattern
+    Config: Type[T_RegexPluginConfig] = RegexPluginConfig
+
+    def str_match(self, msg_str: str) -> bool:
+        msg_str = msg_str.strip()
+        self.msg_match = self.re_pattern.fullmatch(msg_str)
+        return bool(self.msg_match)
+
+
+class CommandPluginBase(RegexPluginBase[T_State, T_CommandPluginConfig], ABC):
+    command_match: re.Match
+    command_re_pattern: re.Pattern
+    Config: Type[T_CommandPluginConfig] = CommandPluginConfig
+
+    def str_match(self, msg_str: str) -> bool:
+        if not hasattr(self, "command_re_pattern"):
+            self.command_re_pattern = re.compile(
+                f'({"|".join(self.config.command_prefix)})'
+                f'({"|".join(self.config.command)})'
+                r"\s*(?P<command_args>.*)",
+                flags=re.I if self.config.ignore_case else 0,
+            )
+        msg_str = msg_str.strip()
+        self.command_match = self.command_re_pattern.fullmatch(msg_str)
+        if not self.command_match:
+            return False
+        self.msg_match = self.re_pattern.fullmatch(
+            self.command_match.group("command_args")
+        )
+        return bool(self.msg_match)
diff --git a/ChienDice/plugins/iamai_plugin_base/config.py b/ChienDice/plugins/iamai_plugin_base/config.py
new file mode 100644
index 0000000..b410888
--- /dev/null
+++ b/ChienDice/plugins/iamai_plugin_base/config.py
@@ -0,0 +1,30 @@
+from typing import Set, Optional
+
+from iamai import ConfigModel
+
+
+class BasePluginConfig(ConfigModel):
+    __config_name__ = ""
+    handle_all_message: bool = False
+    """是否处理所有类型的消息，此配置为 True 时会覆盖 handle_friend_message 和 handle_group_message。"""
+    handle_friend_message: bool = True
+    """是否处理好友消息。"""
+    handle_group_message: bool = True
+    """是否处理群消息。"""
+    accept_group: Optional[Set[int]] = None
+    """处理消息的群号，仅当 handle_group_message 为 True 时生效，留空表示处理所有群。"""
+    message_str: str = "{user_name}: {message}"
+    """最终发送消息的格式。"""
+
+
+class RegexPluginConfig(BasePluginConfig):
+    pass
+
+
+class CommandPluginConfig(RegexPluginConfig):
+    command_prefix: Set[str] = {".", "。"}
+    """命令前缀。"""
+    command: Set[str] = {}
+    """命令文本。"""
+    ignore_case: bool = True
+    """忽略大小写。"""
diff --git a/ChienDice/plugins/iamai_plugin_dice/__init__.py b/ChienDice/plugins/iamai_plugin_dice/__init__.py
new file mode 100644
index 0000000..ef2ca79
--- /dev/null
+++ b/ChienDice/plugins/iamai_plugin_dice/__init__.py
@@ -0,0 +1,50 @@
+import re
+import random
+
+from iamai.log import logger
+
+from plugins.iamai_plugin_base import CommandPluginBase
+
+from .config import Config
+
+
+class Dice(CommandPluginBase[None, Config]):
+    Config = Config
+
+    def __post_init__(self):
+        self.re_pattern = re.compile(
+            r"\s*(?P<dice_times>\d+)d(?P<dice_faces>\d+)([*x](?P<dice_multiply>\d+))?",
+            flags=re.I,
+        )
+
+    async def handle(self) -> None:
+        dice_times = int(self.msg_match.group("dice_times"))
+        dice_faces = int(self.msg_match.group("dice_faces"))
+        if self.msg_match.group("dice_multiply") is None:
+            dice_multiply = None
+        else:
+            dice_multiply = int(self.msg_match.group("dice_multiply"))
+
+        if dice_times > self.config.max_dice_times:
+            await self.event.reply(
+                self.format_str(self.config.exceed_max_dice_times_str)
+            )
+            return
+
+        dice = [random.randint(1, dice_faces) for _ in range(dice_times)]
+        dice_sum = sum(dice)
+        if dice_multiply is None:
+            result_str = f"{dice_times}D{dice_faces}="
+            if dice_times != 1:
+                result_str += f"{'+'.join(map(lambda x: str(x), dice))}="
+            result_str += str(dice_sum)
+        else:
+            result_str = f"{dice_times}D{dice_faces}X{dice_multiply}="
+            if dice_times != 1:
+                result_str += (
+                    f"({'+'.join(map(lambda x: str(x), dice))})X{dice_multiply}="
+                )
+            result_str += f"{dice_sum}X{dice_multiply}={dice_sum * dice_multiply}"
+
+        logger.info(f"Dice Plugin: {result_str}")
+        await self.event.reply(self.format_str(self.config.message_str, result_str))
diff --git a/ChienDice/plugins/iamai_plugin_dice/config.py b/ChienDice/plugins/iamai_plugin_dice/config.py
new file mode 100644
index 0000000..a8c6700
--- /dev/null
+++ b/ChienDice/plugins/iamai_plugin_dice/config.py
@@ -0,0 +1,13 @@
+from typing import Set
+
+from plugins.iamai_plugin_base import CommandPluginConfig
+
+
+class Config(CommandPluginConfig):
+    __config_name__ = "plugin_dice"
+    command: Set[str] = {"r", "roll", "dice"}
+    """命令文本。"""
+    max_dice_times: int = 1000
+    """最大单次投掷次数。"""
+    exceed_max_dice_times_str: str = "错误：超过最大投掷次数。"
+    """超过最大单次投掷次数时的提示语。"""
diff --git a/ChienDice/plugins/iamai_plugin_echo/__init__.py b/ChienDice/plugins/iamai_plugin_echo/__init__.py
new file mode 100644
index 0000000..98cfa39
--- /dev/null
+++ b/ChienDice/plugins/iamai_plugin_echo/__init__.py
@@ -0,0 +1,17 @@
+import re
+
+from plugins.iamai_plugin_base import CommandPluginBase
+
+from .config import Config
+
+
+class Echo(CommandPluginBase[None, Config]):
+    Config = Config
+
+    def __post_init__(self):
+        self.re_pattern = re.compile(r"(?P<echo_str>.*)", flags=re.I)
+
+    async def handle(self) -> None:
+        await self.event.reply(
+            self.format_str(self.config.message_str, self.msg_match.group("echo_str"))
+        )
diff --git a/ChienDice/plugins/iamai_plugin_echo/config.py b/ChienDice/plugins/iamai_plugin_echo/config.py
new file mode 100644
index 0000000..b1fe445
--- /dev/null
+++ b/ChienDice/plugins/iamai_plugin_echo/config.py
@@ -0,0 +1,11 @@
+from typing import Set
+
+from plugins.iamai_plugin_base import CommandPluginConfig
+
+
+class Config(CommandPluginConfig):
+    __config_name__ = "plugin_echo"
+    command: Set[str] = {"echo"}
+    """命令文本。"""
+    message_str: str = "复读{user_name}: {message}"
+    """最终发送消息的格式。"""
diff --git a/ChienDice/plugins/iamai_plugin_luck/__init__.py b/ChienDice/plugins/iamai_plugin_luck/__init__.py
new file mode 100644
index 0000000..f56019b
--- /dev/null
+++ b/ChienDice/plugins/iamai_plugin_luck/__init__.py
@@ -0,0 +1,21 @@
+import re
+import time
+import random
+
+from plugins.iamai_plugin_base import CommandPluginBase
+
+from .config import Config
+
+
+class Luck(CommandPluginBase[None, Config]):
+    Config = Config
+
+    def __post_init__(self):
+        self.re_pattern = re.compile(r".*", flags=re.I)
+
+    async def handle(self) -> None:
+        random.seed(
+            time.strftime("%Y%j", time.localtime()) + self.format_str("{user_id}")
+        )
+        lucy = random.randint(self.config.min_int, self.config.max_int)
+        await self.event.reply(self.format_str(self.config.message_str, str(lucy)))
diff --git a/ChienDice/plugins/iamai_plugin_luck/config.py b/ChienDice/plugins/iamai_plugin_luck/config.py
new file mode 100644
index 0000000..3983266
--- /dev/null
+++ b/ChienDice/plugins/iamai_plugin_luck/config.py
@@ -0,0 +1,15 @@
+from typing import Set
+
+from plugins.iamai_plugin_base import CommandPluginConfig
+
+
+class Config(CommandPluginConfig):
+    __config_name__ = "plugin_luck"
+    command: Set[str] = {"luck"}
+    """命令文本。"""
+    min_int: int = 0
+    """最小随机整数。"""
+    max_int: int = 100
+    """最大随机整数。"""
+    message_str: str = "{user_name}今天的运气是: {message}"
+    """最终发送消息的格式。"""
diff --git a/ChienDice/plugins/iamai_plugin_reply/__init__.py b/ChienDice/plugins/iamai_plugin_reply/__init__.py
new file mode 100644
index 0000000..3be8abd
--- /dev/null
+++ b/ChienDice/plugins/iamai_plugin_reply/__init__.py
@@ -0,0 +1,43 @@
+import re
+import json
+
+from plugins.iamai_plugin_base import BasePlugin
+
+from .config import Config
+
+
+class Reply(BasePlugin[None, Config]):
+    priority: int = 1
+    Config = Config
+
+    def __post_init__(self):
+        with open(self.config.data_file, "r") as fp:
+            if self.config.data_type == "json":
+                json_data = json.load(fp)
+            else:
+                raise ValueError(f"data_type must be json, not {self.config.data_type}")
+        self.rule_to_message = {
+            item["rule"]: item["message"]
+            for item in json_data
+            if isinstance(item, dict)
+            and "rule" in item.keys()
+            and "message" in item.keys()
+        }
+
+    async def handle(self) -> None:
+        msg = self.rule_to_message[self.msg_match.re.pattern]
+        if isinstance(msg, str):
+            await self.event.reply(self.format_str(msg, self.msg_match.string))
+        else:
+            await self.event.reply(msg)
+
+    def str_match(self, msg_str: str) -> bool:
+        msg_str = msg_str.strip()
+        for rule in self.rule_to_message.keys():
+            msg_match = re.fullmatch(
+                rule, msg_str, flags=re.I if self.config.ignore_case else 0
+            )
+            if msg_match:
+                self.msg_match = msg_match
+                return bool(self.msg_match)
+        return False
diff --git a/ChienDice/plugins/iamai_plugin_reply/config.py b/ChienDice/plugins/iamai_plugin_reply/config.py
new file mode 100644
index 0000000..e024fb6
--- /dev/null
+++ b/ChienDice/plugins/iamai_plugin_reply/config.py
@@ -0,0 +1,11 @@
+from plugins.iamai_plugin_base import CommandPluginConfig
+
+
+class Config(CommandPluginConfig):
+    __config_name__ = "plugin_reply"
+    data_type: str = "json"
+    """数据类型，目前只支持 json。"""
+    data_file: str = "data/reply_data.json"
+    """数据文件位置。"""
+    ignore_case: bool = True
+    """是否忽略大小写，默认为 True。"""
diff --git a/ChienDice/plugins/iamai_plugin_send/__init__.py b/ChienDice/plugins/iamai_plugin_send/__init__.py
new file mode 100644
index 0000000..4e4a516
--- /dev/null
+++ b/ChienDice/plugins/iamai_plugin_send/__init__.py
@@ -0,0 +1,28 @@
+import re
+
+from plugins.iamai_plugin_base import CommandPluginBase
+
+from .config import Config
+
+
+class Send(CommandPluginBase[None, Config]):
+    Config = Config
+
+    def __post_init__(self):
+        self.re_pattern = re.compile(r"\s*(?P<message>.*)", flags=re.I)
+
+    async def handle(self) -> None:
+        try:
+            await self.event.adapter.send(
+                self.msg_match.group("message"),
+                "private",
+                self.config.send_user_id,
+            )
+        except Exception as e:
+            if self.config.send_filed_msg is not None:
+                await self.event.reply(
+                    self.format_str(self.config.send_filed_msg, repr(e))
+                )
+        else:
+            if self.config.send_success_msg is not None:
+                await self.event.reply(self.format_str(self.config.send_success_msg))
diff --git a/ChienDice/plugins/iamai_plugin_send/config.py b/ChienDice/plugins/iamai_plugin_send/config.py
new file mode 100644
index 0000000..b60c276
--- /dev/null
+++ b/ChienDice/plugins/iamai_plugin_send/config.py
@@ -0,0 +1,15 @@
+from typing import Set, Optional
+
+from plugins.iamai_plugin_base import CommandPluginConfig
+
+
+class Config(CommandPluginConfig):
+    __config_name__ = "plugin_send"
+    command: Set[str] = {"send"}
+    """命令文本。"""
+    send_user_id: int = 0
+    """发送消息的对象的 QQ 号码。"""
+    send_success_msg: Optional[str] = None
+    """发送成功时回复的消息，设置为 None 表示不发送任何消息。"""
+    send_filed_msg: Optional[str] = "发送失败：{message}"
+    """发送失败时回复的消息。"""