diff options
| author | 2023-04-28 03:04:40 +0800 | |
|---|---|---|
| committer | 2023-04-28 03:04:40 +0800 | |
| commit | c8c76cb2293001a3f5a8122e581648002033eb15 (patch) | |
| tree | 26a4ded54d50d44ed8d0804f0109d4c38ac511d2 | |
| parent | 57a8f67b27aa617b7a993477231b3d6978f4f842 (diff) | |
| download | HydroRoll-c8c76cb2293001a3f5a8122e581648002033eb15.tar.gz HydroRoll-c8c76cb2293001a3f5a8122e581648002033eb15.zip | |
18 files changed, 613 insertions, 0 deletions
diff --git a/hydroroll/__init__.py b/hydroroll/__init__.py new file mode 100644 index 0000000..c72c991 --- /dev/null +++ b/hydroroll/__init__.py @@ -0,0 +1,4 @@ +name = "hydroroll" + +from hydroroll.bot import Bot +from hydroroll.config import GlobalConfig
\ No newline at end of file diff --git a/hydroroll/bot.py b/hydroroll/bot.py new file mode 100644 index 0000000..c2b99e6 --- /dev/null +++ b/hydroroll/bot.py @@ -0,0 +1,64 @@ +from iamai import Bot as _Bot +from typing import Optional, Dict, List, Type, Any, Union +from pathlib import Path +import os +# 获取当前目录路径 +current_dir = Path.cwd() + +# 获取当前脚本文件路径 +script_file = current_dir.resolve() / __file__ + +# 获取当前脚本文件所在目录路径 +script_dir = script_file.parent + +__all__ = ["Bot"] + +class Bot: + def __init__( + self, + *, + config_file: Optional[str] = "config.toml", + config_dict: Optional[Dict] = None, + hot_reload: bool = False, + ) -> None: + self.bot = _Bot(hot_reload=hot_reload, + config_file=config_file, + config_dict=config_dict + ) + self.bot.load_plugins_from_dirs(Path(f"{script_dir}/plugins")) + self.create_folders() + self.init_data() + self.init_conf() + self.init_webui() + + def run(self) -> None: + self.bot.run() + + def restart(self) -> None: + self.bot.restart() + + def create_folders(self): + folder_path = os.path.dirname(os.path.abspath('__file__')) # 获取main.py所在文件夹路径 + if not os.path.isdir(os.path.join(folder_path, 'user')): + os.mkdir(os.path.join(folder_path, 'user')) + if not os.path.isdir(os.path.join(folder_path, 'data')): + os.mkdir(os.path.join(folder_path, 'data')) + if not os.path.isdir(os.path.join(folder_path, 'models')): + os.mkdir(os.path.join(folder_path, 'models')) + if not os.path.isdir(os.path.join(folder_path, 'web')): + os.mkdir(os.path.join(folder_path, 'web')) + if not os.path.isdir(os.path.join(folder_path, 'config')): + os.mkdir(os.path.join(folder_path, 'config')) + if not os.path.isdir(os.path.join(folder_path, 'logs')): + os.mkdir(os.path.join(folder_path, 'logs')) + if not os.path.isdir(os.path.join(folder_path, 'rules')): + os.mkdir(os.path.join(folder_path, 'rules')) + + def init_data(self): + ... + + def init_webui(self): + ... + + def init_conf(self): + ...
\ No newline at end of file diff --git a/hydroroll/config.py b/hydroroll/config.py new file mode 100644 index 0000000..6739249 --- /dev/null +++ b/hydroroll/config.py @@ -0,0 +1,124 @@ +import argparse +import sys +import platform +from importlib.metadata import version +from iamai import Plugin +import os +import pickle +import threading + + +# 创建全局 ArgumentParser 对象 +global_parser = argparse.ArgumentParser(description='hydroroll[水系] 全局命令参数') + +# 定义全局配置类 +class GlobalConfig: + _name = "hydroroll" + _version = "0.1.0" + _svn = "2" + _author = "简律纯" + _iamai_version = version('iamai') + _python_ver = sys.version + _python_ver_raw= '.'.join(map(str, platform.python_version_tuple()[:3])) + current_path = os.path.dirname(os.path.abspath('__file__')) + + # 定义系统组件 + class HydroSystem: + def __init__(self): + self.parser = argparse.ArgumentParser(description='hydroroll[水系].system 系统命令参数') + self.subparsers = self.parser.add_subparsers() + self.status_parser = self.subparsers.add_parser('status', aliases=['s'], help='系统状态') + self.reload_parser = self.subparsers.add_parser('reload', aliases=['rld'], help='重新加载系统') + self.restart_parser = self.subparsers.add_parser('restart', aliases=['rst'], help='重启系统') + self.help = '\n'.join(self.parser.format_help().replace('\r\n', '\n').replace('\r', '').split('\n')[2:-3]) + class HydroBot: + def __init__(self) -> None: + self.parser = argparse.ArgumentParser(description="Bot命令") + + +class ConfigManager: + def __init__(self, directory): + self.directory = directory + self.configs = {} + self.locks = {} + + def get(self, filename, section, key, default=None): + if not self._check_file_exists(filename): + return default + config = self._get_config(filename) + if section not in config: + return default + return config[section].get(key, default) + + def set(self, filename, section, key, value): + config = self._get_config(filename) + if section not in config: + config[section] = {} + config[section][key] = value + self._save_config(filename, config) + + def delete(self, filename, section, key): + config = self._get_config(filename) + if section not in config: + return + if key not in config[section]: + return + del config[section][key] + self._save_config(filename, config) + + def sections(self, filename): + if not self._check_file_exists(filename): + return [] + config = self._get_config(filename) + return list(config.keys()) + + def section_items(self, filename, section): + if not self._check_file_exists(filename): + return [] + config = self._get_config(filename) + if section not in config: + return [] + return list(config[section].items()) + + def files(self): + return [filename for filename in os.listdir(self.directory) if filename.endswith('.dat')] + + def _get_lock(self, filename): + if filename not in self.locks: + self.locks[filename] = threading.Lock() + return self.locks[filename] + + def _get_config(self, filename): + with self._get_lock(filename): + if filename not in self.configs: + filepath = os.path.join(self.directory, filename) + if os.path.exists(filepath): + try: + with open(filepath, 'rb') as f: + self.configs[filename] = pickle.load(f) + except: + pass + if filename not in self.configs: + self.configs[filename] = {} + return self.configs[filename] + + def _save_config(self, filename, config): + with self._get_lock(filename): + try: + filepath = os.path.join(self.directory, filename) + backuppath = os.path.join(self.directory, filename + '.bak') + lockpath = os.path.join(self.directory, filename + '.lock') + with open(lockpath, 'wb') as f: + pass + if os.path.exists(filepath): + os.replace(filepath, backuppath) + with open(filepath, 'wb') as f: + pickle.dump(config, f) + os.remove(backuppath) + os.remove(lockpath) + except: + pass + + def _check_file_exists(self, filename): + filepath = os.path.join(self.directory, filename) + return os.path.exists(filepath) and filename.endswith('.dat') diff --git a/hydroroll/plugins/HydroRoll_plugin_base/__init__.py b/hydroroll/plugins/HydroRoll_plugin_base/__init__.py new file mode 100644 index 0000000..a050ae1 --- /dev/null +++ b/hydroroll/plugins/HydroRoll_plugin_base/__init__.py @@ -0,0 +1,93 @@ +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: + is_bot_off = True + + if self.event.adapter.name != "cqhttp": + return False + if self.event.type != "message": + return False + match_str = self.event.message.get_plain_text() + if is_bot_off: + if self.event.message.startswith(f'[CQ:at,qq={self.event.self_id}]'): + match_str = re.sub(fr'^\[CQ:at,qq={self.event.self_id}\]', '', match_str) + else: + return False + if self.config.handle_all_message: + return self.str_match(match_str) + elif self.config.handle_friend_message: + if self.event.message_type == "private": + return self.str_match(match_str) + 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(match_str) + 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)
\ No newline at end of file diff --git a/hydroroll/plugins/HydroRoll_plugin_base/config.py b/hydroroll/plugins/HydroRoll_plugin_base/config.py new file mode 100644 index 0000000..311874c --- /dev/null +++ b/hydroroll/plugins/HydroRoll_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 = 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/hydroroll/plugins/HydroRoll_plugin_bot/__init__.py b/hydroroll/plugins/HydroRoll_plugin_bot/__init__.py new file mode 100644 index 0000000..2b3eec8 --- /dev/null +++ b/hydroroll/plugins/HydroRoll_plugin_bot/__init__.py @@ -0,0 +1,30 @@ +import re +from importlib.metadata import version +from plugins.hydroroll_plugin_base import CommandPluginBase +from hydroroll.config import GlobalConfig + +from .config import Config + + +class HydroBot(CommandPluginBase[None, Config]): + Config = Config + CurrentConfig = GlobalConfig + priority = 0 + + def __post_init__(self): + self.re_pattern = re.compile(r"(?P<bot_info_str>.*)", flags=re.I) + + def bot_info(self): + info_str = f'{self.CurrentConfig._name} '\ + f'{self.CurrentConfig._version}({self.CurrentConfig._svn}) '\ + f'by {self.CurrentConfig._author} '\ + f'on Python {self.CurrentConfig._python_ver_raw} '\ + f'with {" & ".join([adapter + "("+version("iamai-adapter-"+adapter) +")" for adapter in dict(self.bot.config.adapter)])} '\ + f'for iamai({self.CurrentConfig._iamai_version})' + + return info_str + + async def handle(self) -> None: + await self.event.reply( + self.format_str(self.config.message_str, self.bot_info()) + ) diff --git a/hydroroll/plugins/HydroRoll_plugin_bot/config.py b/hydroroll/plugins/HydroRoll_plugin_bot/config.py new file mode 100644 index 0000000..717b7b2 --- /dev/null +++ b/hydroroll/plugins/HydroRoll_plugin_bot/config.py @@ -0,0 +1,11 @@ +from typing import Set + +from plugins.hydroroll_plugin_base import CommandPluginConfig + + +class Config(CommandPluginConfig): + __config_name__ = "plugin_bot_info" + command: Set[str] = {"bot"} + """命令文本。""" + message_str: str = "{message}" + """最终发送消息的格式。""" diff --git a/hydroroll/plugins/HydroRoll_plugin_dice/__init__.py b/hydroroll/plugins/HydroRoll_plugin_dice/__init__.py new file mode 100644 index 0000000..2c9d2b4 --- /dev/null +++ b/hydroroll/plugins/HydroRoll_plugin_dice/__init__.py @@ -0,0 +1,51 @@ +import re +import random + +from iamai.log import logger + +from plugins.hydroroll_plugin_base import CommandPluginBase + +from .config import Config + +__all__ = ["Dice"] + +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/hydroroll/plugins/HydroRoll_plugin_dice/config.py b/hydroroll/plugins/HydroRoll_plugin_dice/config.py new file mode 100644 index 0000000..4846409 --- /dev/null +++ b/hydroroll/plugins/HydroRoll_plugin_dice/config.py @@ -0,0 +1,13 @@ +from typing import Set + +from plugins.hydroroll_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/hydroroll/plugins/HydroRoll_plugin_echo/__init__.py b/hydroroll/plugins/HydroRoll_plugin_echo/__init__.py new file mode 100644 index 0000000..e800384 --- /dev/null +++ b/hydroroll/plugins/HydroRoll_plugin_echo/__init__.py @@ -0,0 +1,17 @@ +import re + +from plugins.hydroroll_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/hydroroll/plugins/HydroRoll_plugin_echo/config.py b/hydroroll/plugins/HydroRoll_plugin_echo/config.py new file mode 100644 index 0000000..030bbec --- /dev/null +++ b/hydroroll/plugins/HydroRoll_plugin_echo/config.py @@ -0,0 +1,11 @@ +from typing import Set + +from plugins.hydroroll_plugin_base import CommandPluginConfig + + +class Config(CommandPluginConfig): + __config_name__ = "plugin_echo" + command: Set[str] = {"echo"} + """命令文本。""" + message_str: str = "*{user_name} {message}" + """最终发送消息的格式。""" diff --git a/hydroroll/plugins/HydroRoll_plugin_luck/__init__.py b/hydroroll/plugins/HydroRoll_plugin_luck/__init__.py new file mode 100644 index 0000000..a30c964 --- /dev/null +++ b/hydroroll/plugins/HydroRoll_plugin_luck/__init__.py @@ -0,0 +1,21 @@ +import re +import time +import random + +from plugins.hydroroll_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/hydroroll/plugins/HydroRoll_plugin_luck/config.py b/hydroroll/plugins/HydroRoll_plugin_luck/config.py new file mode 100644 index 0000000..eeef14b --- /dev/null +++ b/hydroroll/plugins/HydroRoll_plugin_luck/config.py @@ -0,0 +1,15 @@ +from typing import Set + +from plugins.hydroroll_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/hydroroll/plugins/HydroRoll_plugin_send/__init__.py b/hydroroll/plugins/HydroRoll_plugin_send/__init__.py new file mode 100644 index 0000000..234e5b3 --- /dev/null +++ b/hydroroll/plugins/HydroRoll_plugin_send/__init__.py @@ -0,0 +1,28 @@ +import re + +from plugins.hydroroll_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/hydroroll/plugins/HydroRoll_plugin_send/config.py b/hydroroll/plugins/HydroRoll_plugin_send/config.py new file mode 100644 index 0000000..6903c0f --- /dev/null +++ b/hydroroll/plugins/HydroRoll_plugin_send/config.py @@ -0,0 +1,15 @@ +from typing import Set, Optional + +from plugins.hydroroll_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}" + """发送失败时回复的消息。""" diff --git a/hydroroll/plugins/HydroRoll_plugin_system/__init__.py b/hydroroll/plugins/HydroRoll_plugin_system/__init__.py new file mode 100644 index 0000000..6e4f232 --- /dev/null +++ b/hydroroll/plugins/HydroRoll_plugin_system/__init__.py @@ -0,0 +1,75 @@ +import re +from plugins.hydroroll_plugin_base import CommandPluginBase +from .config import Config +import psutil +import time +from hydroroll.config import GlobalConfig +from iamai.adapter.cqhttp.message import CQHTTPMessageSegment + +class System(CommandPluginBase[None, Config]): + priority: int = 0 + block: bool = True + Config = Config + CurrentConfig = GlobalConfig + + def __post_init__(self): + self.re_pattern = re.compile(r"(?P<system_info_str>.*)", flags=re.I) + + def eventReply(self, message: str): + return self.event.reply( + self.format_str(self.config.message_str, message) + ) + + def get_system_status(self) -> str: + cpu_usage = psutil.cpu_percent() + memory_usage = psutil.virtual_memory().percent + disk_usage = psutil.disk_usage('/').percent + + current_time = time.time() + start_time = psutil.Process().create_time() + + uptime_seconds = int(current_time - start_time) + uptime_str = time.strftime("%H:%M:%S", time.gmtime(uptime_seconds)) + + info_str = f"{self.CurrentConfig._name} Ver.{self.CurrentConfig._version}" + info_str += f"({self.CurrentConfig._svn}) built in Python {self.CurrentConfig._python_ver}\n" + info_str += f"本地时间: {time.strftime('%Y-%m-%d %H:%M:%S', time.localtime())}\n" + info_str += f"已启动时间:{uptime_str}\n" + info_str += f"CPU使用率:{cpu_usage}%\n" + info_str += f"内存占用率:{memory_usage}%\n" + info_str += f"磁盘使用率:{disk_usage}%" + + return info_str + + async def handle(self) -> None: + system = self.CurrentConfig.HydroSystem() + try: + sub_command = self.event.get_plain_text().split()[1] + except IndexError: + sub_command = "" + + if sub_command in ["status", "s"]: + await self.event.reply( + self.format_str(self.config.message_str, + self.get_system_status()) + ) + elif sub_command in ["restart", "rst"]: + await self.event.reply( + self.format_str(self.config.message_str, "正在重启系统...") + ) + self.bot.restart() + + elif sub_command in ["reload", "rld"]: + await self.event.reply( + self.format_str(self.config.message_str, "正在重载...") + ) + self.bot.reload_plugins() + await self.event.reply( + self.format_str(self.config.message_str, + f"已加载{len(self.bot.plugins)}枚插件") + ) + else: + await self.event.reply( + CQHTTPMessageSegment.reply(self.event.message_id) + + self.format_str(self.config.message_str, system.help) + ) diff --git a/hydroroll/plugins/HydroRoll_plugin_system/config.py b/hydroroll/plugins/HydroRoll_plugin_system/config.py new file mode 100644 index 0000000..416fcbe --- /dev/null +++ b/hydroroll/plugins/HydroRoll_plugin_system/config.py @@ -0,0 +1,11 @@ +from typing import Set + +from plugins.hydroroll_plugin_base import CommandPluginConfig + + +class Config(CommandPluginConfig): + __config_name__ = "plugin_system_info" + command: Set[str] = {"system"} + """命令文本。""" + message_str: str = "{message}" + """最终发送消息的格式。""" diff --git a/hydroroll/test.py b/hydroroll/test.py new file mode 100644 index 0000000..e69de29 --- /dev/null +++ b/hydroroll/test.py |
