From f80e9f05b43a4883e58eaf82f2200d31ac87b23e Mon Sep 17 00:00:00 2001 From: 简律纯 Date: Wed, 19 Apr 2023 17:08:33 +0800 Subject: rename to `HydroRoll` --- ChienDice/plugins/iamai_plugin_base/__init__.py | 85 ---------------------- ChienDice/plugins/iamai_plugin_base/config.py | 30 -------- .../plugins/iamai_plugin_bot_info/__init__.py | 20 ----- ChienDice/plugins/iamai_plugin_bot_info/config.py | 11 --- ChienDice/plugins/iamai_plugin_dice/__init__.py | 50 ------------- ChienDice/plugins/iamai_plugin_dice/config.py | 13 ---- ChienDice/plugins/iamai_plugin_echo/__init__.py | 17 ----- ChienDice/plugins/iamai_plugin_echo/config.py | 11 --- ChienDice/plugins/iamai_plugin_luck/__init__.py | 21 ------ ChienDice/plugins/iamai_plugin_luck/config.py | 15 ---- ChienDice/plugins/iamai_plugin_reply/__init__.py | 43 ----------- ChienDice/plugins/iamai_plugin_reply/config.py | 11 --- ChienDice/plugins/iamai_plugin_send/__init__.py | 28 ------- ChienDice/plugins/iamai_plugin_send/config.py | 15 ---- 14 files changed, 370 deletions(-) delete mode 100644 ChienDice/plugins/iamai_plugin_base/__init__.py delete mode 100644 ChienDice/plugins/iamai_plugin_base/config.py delete mode 100644 ChienDice/plugins/iamai_plugin_bot_info/__init__.py delete mode 100644 ChienDice/plugins/iamai_plugin_bot_info/config.py delete mode 100644 ChienDice/plugins/iamai_plugin_dice/__init__.py delete mode 100644 ChienDice/plugins/iamai_plugin_dice/config.py delete mode 100644 ChienDice/plugins/iamai_plugin_echo/__init__.py delete mode 100644 ChienDice/plugins/iamai_plugin_echo/config.py delete mode 100644 ChienDice/plugins/iamai_plugin_luck/__init__.py delete mode 100644 ChienDice/plugins/iamai_plugin_luck/config.py delete mode 100644 ChienDice/plugins/iamai_plugin_reply/__init__.py delete mode 100644 ChienDice/plugins/iamai_plugin_reply/config.py delete mode 100644 ChienDice/plugins/iamai_plugin_send/__init__.py delete mode 100644 ChienDice/plugins/iamai_plugin_send/config.py (limited to 'ChienDice/plugins') diff --git a/ChienDice/plugins/iamai_plugin_base/__init__.py b/ChienDice/plugins/iamai_plugin_base/__init__.py deleted file mode 100644 index fe0cd32..0000000 --- a/ChienDice/plugins/iamai_plugin_base/__init__.py +++ /dev/null @@ -1,85 +0,0 @@ -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.*)", - 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 deleted file mode 100644 index c36d03e..0000000 --- a/ChienDice/plugins/iamai_plugin_base/config.py +++ /dev/null @@ -1,30 +0,0 @@ -from typing import Set, Optional - -from iamai import ConfigModel - - -class BasePluginConfig(ConfigModel): - __config_name__ = "" - handle_all_message: bool = True - """是否处理所有类型的消息，此配置为 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_bot_info/__init__.py b/ChienDice/plugins/iamai_plugin_bot_info/__init__.py deleted file mode 100644 index 3cc869e..0000000 --- a/ChienDice/plugins/iamai_plugin_bot_info/__init__.py +++ /dev/null @@ -1,20 +0,0 @@ -import re -import sys -import platform -from importlib.metadata import version -from plugins.iamai_plugin_base import CommandPluginBase - -from .config import Config - - -class BotInfo(CommandPluginBase[None, Config]): - Config = Config - - def __post_init__(self): - self.re_pattern = re.compile(r"(?P.*)", flags=re.I) - - async def handle(self) -> None: - await self.event.reply( - self.format_str(self.config.message_str, - f"{self.bot.config.Chien['self']['header']} HydroRoll!{self.bot.config.Chien['version']}({self.bot.config.Chien['svn']}) by {self.bot.config.Chien['author']} on Python {'.'.join(map(str, platform.python_version_tuple()[:3]))} with {' & '.join([adapter + '('+version('iamai-adapter-'+adapter)+')' for adapter in dict(self.bot.config.adapter)])} for iamai({version('iamai')})\n{self.bot.config.Chien['self']['info']}") - ) diff --git a/ChienDice/plugins/iamai_plugin_bot_info/config.py b/ChienDice/plugins/iamai_plugin_bot_info/config.py deleted file mode 100644 index cf92e81..0000000 --- a/ChienDice/plugins/iamai_plugin_bot_info/config.py +++ /dev/null @@ -1,11 +0,0 @@ -from typing import Set - -from plugins.iamai_plugin_base import CommandPluginConfig - - -class Config(CommandPluginConfig): - __config_name__ = "plugin_bot_info" - command: Set[str] = {"bot"} - """命令文本。""" - message_str: str = "{message}" - """最终发送消息的格式。""" diff --git a/ChienDice/plugins/iamai_plugin_dice/__init__.py b/ChienDice/plugins/iamai_plugin_dice/__init__.py deleted file mode 100644 index 6005406..0000000 --- a/ChienDice/plugins/iamai_plugin_dice/__init__.py +++ /dev/null @@ -1,50 +0,0 @@ -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\d+)d(?P\d+)([*x](?P\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 deleted file mode 100644 index a8c6700..0000000 --- a/ChienDice/plugins/iamai_plugin_dice/config.py +++ /dev/null @@ -1,13 +0,0 @@ -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 deleted file mode 100644 index 98cfa39..0000000 --- a/ChienDice/plugins/iamai_plugin_echo/__init__.py +++ /dev/null @@ -1,17 +0,0 @@ -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.*)", 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 deleted file mode 100644 index b1fe445..0000000 --- a/ChienDice/plugins/iamai_plugin_echo/config.py +++ /dev/null @@ -1,11 +0,0 @@ -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 deleted file mode 100644 index f56019b..0000000 --- a/ChienDice/plugins/iamai_plugin_luck/__init__.py +++ /dev/null @@ -1,21 +0,0 @@ -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 deleted file mode 100644 index 3983266..0000000 --- a/ChienDice/plugins/iamai_plugin_luck/config.py +++ /dev/null @@ -1,15 +0,0 @@ -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 deleted file mode 100644 index 6835ca8..0000000 --- a/ChienDice/plugins/iamai_plugin_reply/__init__.py +++ /dev/null @@ -1,43 +0,0 @@ -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",encoding="utf-8") 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 deleted file mode 100644 index e024fb6..0000000 --- a/ChienDice/plugins/iamai_plugin_reply/config.py +++ /dev/null @@ -1,11 +0,0 @@ -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 deleted file mode 100644 index 4e4a516..0000000 --- a/ChienDice/plugins/iamai_plugin_send/__init__.py +++ /dev/null @@ -1,28 +0,0 @@ -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.*)", 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 deleted file mode 100644 index 959822c..0000000 --- a/ChienDice/plugins/iamai_plugin_send/config.py +++ /dev/null @@ -1,15 +0,0 @@ -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 = 2753364619 - """发送消息的对象的 QQ 号码。""" - send_success_msg: Optional[str] = "本大魔王已将消息送出√" - """发送成功时回复的消息，设置为 None 表示不发送任何消息。""" - send_filed_msg: Optional[str] = "发送失败：{message}" - """发送失败时回复的消息。""" -- cgit v1.2.3-70-g09d2