{{ duration }}
{{ title }}
{{ play_num }}播放 · {{ danmaku_num }}弹幕 · {{ reply_num }}评论
{{ like_num }}点赞 · {{ coin_num }}投币 · {{ favorite_num }}收藏
{{ desc }}
发布于{{ publish_time }}
2 |
3 | # redbot
4 |
5 | 一个以 [Graia Ariadne](https://github.com/GraiaProject/Ariadne) 框架为基础的 QQ 机器人
6 |
7 |
8 |
9 |
10 | 
17 |
18 |
19 | 
24 |
25 |
53 |
54 | # RedBot 状态
55 |
56 |
57 |
58 | ## 基本信息
59 | **PID**: {pid}
60 | **启动时间**：{time.strftime("%Y-%m-%d %p %I:%M:%S", started_time)}
61 | **已运行时长**：{running_time}
62 |
63 | ## 运行环境
64 | **Python 版本**：{python_version}
65 | **系统版本**：{system_version}
66 | **CPU 核心数**：{psutil.cpu_count()}
67 | **CPU 占用率**：{psutil.cpu_percent()}%
68 | **系统内存占用**：{"%.1f" % (psutil.virtual_memory().available / 1073741824)}G / {total_memory}G
69 |
70 | ## 依赖版本
71 | **Mirai Api Http**：{await app.get_version()}
72 | **Graia 相关**：
73 | '''
74 | if extra:
75 | md += ''.join(f' - {name}：{version}\n' for name, version in extra)
76 | md += ''.join(f' - {name}：{version}\n' for name, version in official)
77 | if community:
78 | md += ''.join(f' - {name}：{version}\n' for name, version in community)
79 |
80 | await app.send_message(group, MessageChain(Image(data_bytes=await md2img(md.rstrip()))))
81 |
--------------------------------------------------------------------------------
/core_modules/error_handler.py:
--------------------------------------------------------------------------------
1 | import traceback
2 | from io import StringIO
3 |
4 | from graia.ariadne.app import Ariadne
5 | from graia.ariadne.message.chain import MessageChain
6 | from graia.ariadne.message.element import Image, Plain
7 | from graia.broadcast.builtin.event import ExceptionThrowed
8 | from graia.saya import Channel
9 | from graiax.shortcut.saya import listen
10 |
11 | from libs.config import basic_cfg
12 | from libs.text2img import md2img
13 |
14 | channel = Channel.current()
15 | channel.meta['can_disable'] = False
16 |
17 |
18 | @listen(ExceptionThrowed)
19 | async def except_handle(event: ExceptionThrowed):
20 | if isinstance(event.event, ExceptionThrowed):
21 | return
22 | app = Ariadne.current()
23 | with StringIO() as fp:
24 | traceback.print_tb(event.exception.__traceback__, file=fp)
25 | tb = fp.getvalue()
26 | msg = f'''\
27 | ## 异常事件：
28 |
29 | {str(event.event.__repr__())}
30 |
31 | ## 异常类型：
32 |
33 | `{type(event.exception)}`
34 |
35 | ## 异常内容：
36 |
37 | {str(event.exception)}
38 |
39 | ## 异常追踪：
40 |
41 | ```py
42 | {tb}
43 | ```
44 | '''
45 | img_bytes = await md2img(msg, 1500)
46 | await app.send_friend_message(basic_cfg.admin.masterId, MessageChain(Plain('发生异常\n'), Image(data_bytes=img_bytes)))
47 |
48 |
49 | # from graia.ariadne.event.message import GroupMessage
50 |
51 |
52 | # @listen(GroupMessage)
53 | # async def error_handler_test(msg: MessageChain):
54 | # if str(msg) == '.错误捕捉测试':
55 | # raise ValueError('错误捕捉测试')
56 |
--------------------------------------------------------------------------------
/core_modules/module_manage.py:
--------------------------------------------------------------------------------
1 | """
2 | Bot 管理
3 | 包含 分群配置 插件配置更改 菜单
4 | """
5 |
6 | import os
7 | from pathlib import Path
8 |
9 | import kayaku
10 | from graia.ariadne.app import Ariadne
11 | from graia.ariadne.event.message import GroupMessage
12 | from graia.ariadne.message.chain import MessageChain
13 | from graia.ariadne.message.element import At, Image, Plain
14 | from graia.ariadne.message.parser.twilight import RegexMatch, RegexResult, Twilight
15 | from graia.ariadne.model import Group, Member, MemberPerm
16 | from graia.saya import Channel, Saya
17 | from graiax.shortcut.interrupt import FunctionWaiter
18 | from graiax.shortcut.saya import decorate, dispatch, listen
19 | from loguru import logger
20 |
21 | from libs.config import basic_cfg, modules_cfg
22 | from libs.control import require_disable
23 | from libs.control.permission import GroupPermission
24 | from libs.text2img import md2img
25 |
26 | saya = Saya.current()
27 | channel = Channel.current()
28 |
29 | channel.meta['name'] = 'Bot模块管理'
30 | channel.meta['author'] = ['Red_lnn']
31 | channel.meta['description'] = (
32 | '管理Bot已加载的模块\n'
33 | '[!！.](菜单|menu) —— 获取模块列表\n'
34 | '[!！.]启用 <模块ID> —— 【群管理员】启用某个已禁用的模块（全局禁用除外）\n'
35 | '[!！.]禁用 <模块ID> —— 【群管理员】禁用某个已启用的模块（禁止禁用的模块除外）\n'
36 | '[!！.]全局启用 <模块ID> —— 【Bot管理员】全局启用某个已禁用的模块\n'
37 | '[!！.]全局禁用 <模块ID> —— 【Bot管理员】全局禁用某个已启用的模块（禁止禁用的模块除外）\n'
38 | '[!！.]重载 <模块ID> —— 【Bot管理员，强烈不推荐】重载指定模块\n'
39 | '[!！.]加载 <模块文件名> —— 【Bot管理员，强烈不推荐】加载新模块（xxx.xxx 无需后缀）\n'
40 | '[!！.]卸载 <模块ID> —— 【Bot管理员】卸载指定模块'
41 | )
42 | channel.meta['can_disable'] = False
43 |
44 | # ensp = ' ' # 半角空格
45 | # emsp = ' ' # 全角空格
46 |
47 |
48 | async def get_channel(app: Ariadne, module_id: str, group: Group):
49 | if module_id.isdigit():
50 | target_id = int(module_id) - 1
51 | if target_id >= len(saya.channels):
52 | await app.send_message(group, MessageChain(Plain('你指定的模块不存在')))
53 | return
54 | return saya.channels[str(list(saya.channels.keys())[target_id])]
55 | else:
56 | if module_id not in saya.channels:
57 | await app.send_message(group, MessageChain(Plain('你指定的模块不存在')))
58 | return
59 | return saya.channels[module_id]
60 |
61 |
62 | @listen(GroupMessage)
63 | @dispatch(Twilight(RegexMatch(r'[!！.](菜单|menu)')))
64 | @decorate(GroupPermission.require(), require_disable(channel.module))
65 | async def menu(app: Ariadne, group: Group):
66 | md = f'''\
67 |
68 |
69 | # {basic_cfg.botName} 功能菜单
70 |
71 | for {group.name}({group.id})
72 |
73 | | ID | 模块状态 | 模块名 |
74 | | --- | --- | --- |
75 | '''
76 |
77 | for index, module in enumerate(saya.channels, start=1):
78 | global_disabled = module in modules_cfg.globalDisabledModules
79 | disabled_groups = modules_cfg.disabledGroups[module] if module in modules_cfg.disabledGroups else []
80 | num = f' {index}' if index < 10 else str(index)
81 | status = '❌' if global_disabled or group.id in disabled_groups else '✔️'
82 | md += f'| {num} | {status} | {saya.channels[module]._name if saya.channels[module]._name is not None else saya.channels[module].module} |\n'
83 | md += f'''
84 | ---
85 |
86 | 私は {basic_cfg.admin.masterName} の {basic_cfg.botName} です www
87 | 群管理员要想配置模块开关请发送【.启用/禁用 】
88 | 要想查询某模块的用法和介绍请发送【.用法 】
89 | 若无法触发，请检查前缀符号是否正确如！与!
90 | 或是命令中有无多余空格，除特别说明外均不需要 @bot
91 |
92 |
'''
93 | await app.send_message(group, MessageChain(Image(data_bytes=await md2img(md))))
94 |
95 |
96 | @listen(GroupMessage)
97 | @dispatch(Twilight(RegexMatch(r'[!！.]启用'), 'module_id' @ RegexMatch(r'\d+')))
98 | @decorate(GroupPermission.require(MemberPerm.Administrator), require_disable(channel.module))
99 | async def enable_module(app: Ariadne, group: Group, module_id: RegexResult):
100 | if module_id.result is None:
101 | return
102 | target_id = str(module_id.result)
103 | target_module = await get_channel(app, target_id, group)
104 | if target_module is None:
105 | await app.send_message(group, MessageChain(Plain('你指定的模块不存在')))
106 | return
107 | disabled_groups = (
108 | modules_cfg.disabledGroups[target_module.module] if target_module.module in modules_cfg.disabledGroups else []
109 | )
110 | if target_module.module in modules_cfg.globalDisabledModules:
111 | await app.send_message(group, MessageChain(Plain('模块已全局禁用无法开启')))
112 | elif group.id in disabled_groups:
113 | disabled_groups.remove(group.id)
114 | modules_cfg.disabledGroups[target_module.module] = disabled_groups
115 | kayaku.save(modules_cfg)
116 | await app.send_message(group, MessageChain(Plain('模块已启用')))
117 | else:
118 | await app.send_message(group, MessageChain(Plain('无变化，模块已处于开启状态')))
119 |
120 |
121 | @listen(GroupMessage)
122 | @dispatch(Twilight(RegexMatch(r'[!！.]禁用'), 'module_id' @ RegexMatch(r'\d+')))
123 | @decorate(GroupPermission.require(MemberPerm.Administrator), require_disable(channel.module))
124 | async def disable_module(app: Ariadne, group: Group, module_id: RegexResult):
125 | if module_id.result is None:
126 | return
127 | target_module = await get_channel(app, str(module_id.result), group)
128 | if target_module is None:
129 | await app.send_message(group, MessageChain(Plain('你指定的模块不存在')))
130 | return
131 | disabled_groups = (
132 | modules_cfg.disabledGroups[target_module.module] if target_module.module in modules_cfg.disabledGroups else []
133 | )
134 | if not target_module.meta.get('can_disable', True):
135 | await app.send_message(group, MessageChain(Plain('你指定的模块不允许禁用')))
136 | elif group.id not in disabled_groups:
137 | disabled_groups.append(group.id)
138 | modules_cfg.disabledGroups[target_module.module] = disabled_groups
139 | kayaku.save(modules_cfg)
140 | await app.send_message(group, MessageChain(Plain('模块已禁用')))
141 | else:
142 | await app.send_message(group, MessageChain(Plain('无变化，模块已处于禁用状态')))
143 |
144 |
145 | @listen(GroupMessage)
146 | @dispatch(Twilight(RegexMatch(r'[!！.]全局启用'), 'module_id' @ RegexMatch(r'\d+')))
147 | @decorate(GroupPermission.require(MemberPerm.Administrator), require_disable(channel.module))
148 | async def global_enable_module(app: Ariadne, group: Group, module_id: RegexResult):
149 | if module_id.result is None:
150 | return
151 | target_module = await get_channel(app, str(module_id.result), group)
152 | if target_module is None:
153 | await app.send_message(group, MessageChain(Plain('你指定的模块不存在')))
154 | return
155 | if target_module.module not in modules_cfg.globalDisabledModules:
156 | await app.send_message(group, MessageChain(Plain('无变化，模块已处于全局开启状态')))
157 | else:
158 | modules_cfg.globalDisabledModules.remove(target_module.module)
159 | kayaku.save(modules_cfg)
160 | await app.send_message(group, MessageChain(Plain('模块已全局启用')))
161 |
162 |
163 | @listen(GroupMessage)
164 | @dispatch(Twilight(RegexMatch(r'[!！.]全局禁用'), 'module_id' @ RegexMatch(r'\d+')))
165 | @decorate(GroupPermission.require(MemberPerm.Administrator), require_disable(channel.module))
166 | async def global_disable_module(app: Ariadne, group: Group, module_id: RegexResult):
167 | if module_id.result is None:
168 | return
169 | target_module = await get_channel(app, str(module_id.result), group)
170 | if target_module is None:
171 | await app.send_message(group, MessageChain(Plain('你指定的模块不存在')))
172 | return
173 | if not target_module.meta.get('can_disable', True):
174 | await app.send_message(group, MessageChain(Plain('你指定的模块不允许禁用')))
175 | elif target_module.module not in modules_cfg.globalDisabledModules:
176 | modules_cfg.globalDisabledModules.append(target_module.module)
177 | kayaku.save(modules_cfg)
178 | await app.send_message(group, MessageChain(Plain('模块已全局禁用')))
179 | else:
180 | await app.send_message(group, MessageChain(Plain('无变化，模块已处于全局禁用状态')))
181 |
182 |
183 | @listen(GroupMessage)
184 | @dispatch(Twilight(RegexMatch(r'[!！.]用法'), 'module_id' @ RegexMatch(r'\d+')))
185 | @decorate(GroupPermission.require(), require_disable(channel.module))
186 | async def get_usage(app: Ariadne, group: Group, module_id: RegexResult):
187 | if module_id.result is None:
188 | return
189 | target_module = await get_channel(app, str(module_id.result), group)
190 | if target_module is None:
191 | return
192 | disabled_groups = (
193 | modules_cfg.disabledGroups[target_module.module] if target_module.module in modules_cfg.disabledGroups else []
194 | )
195 | if group.id in disabled_groups:
196 | await app.send_message(group, MessageChain(Plain('该模块已在本群禁用')))
197 | return
198 | authors = ''
199 | if target_module._author:
200 | for author in target_module._author:
201 | authors += f'{author}, '
202 | authors = authors.rstrip(', ')
203 | md = f'## {target_module._name}{f" By {authors}" if authors else ""}\n\n'
204 | if target_module._description:
205 | md += '### 描述\n' + target_module._description
206 | img_bytes = await md2img(md)
207 | await app.send_message(group, MessageChain(Image(data_bytes=img_bytes)))
208 |
209 |
210 | @listen(GroupMessage)
211 | @dispatch(Twilight(RegexMatch(r'[!！.]重载模块'), 'module_id' @ RegexMatch(r'\d+')))
212 | @decorate(GroupPermission.require(GroupPermission.BOT_ADMIN), require_disable(channel.module))
213 | async def reload_module(app: Ariadne, group: Group, member: Member, module_id: RegexResult):
214 | await app.send_message(
215 | group,
216 | MessageChain(At(member.id), Plain(' 重载模块有极大可能会出错，请问你确实要重载吗？\n强制重载请在10s内发送 .force ，取消请发送 .cancel')),
217 | )
218 |
219 | async def waiter(waiter_group: Group, waiter_member: Member, waiter_message: MessageChain) -> bool | None:
220 | if waiter_group.id == group.id and waiter_member.id == member.id:
221 | saying = str(waiter_message)
222 | if saying == '.force':
223 | return True
224 | elif saying == '.cancel':
225 | return False
226 | else:
227 | await app.send_message(group, MessageChain(At(member.id), Plain('请发送 .force 或 .cancel')))
228 |
229 | answer = await FunctionWaiter(waiter, [GroupMessage]).wait(timeout=10)
230 | if answer is None:
231 | await app.send_message(group, MessageChain(Plain('已超时取消')))
232 | return
233 | if not answer:
234 | await app.send_message(group, MessageChain(Plain('已取消操作')))
235 | return
236 |
237 | if module_id.result is None:
238 | return
239 | target_module = await get_channel(app, str(module_id.result), group)
240 | if target_module is None:
241 | return
242 | logger.info(f'重载模块: {target_module._name} —— {target_module.module}')
243 | try:
244 | with saya.module_context():
245 | saya.reload_channel(saya.channels[target_module.module])
246 | except Exception as e:
247 | await app.send_message(group, MessageChain(Plain(f'重载模块 {target_module.module} 时出错')))
248 | logger.error(f'重载模块 {target_module.module} 时出错')
249 | logger.exception(e)
250 | else:
251 | await app.send_message(group, MessageChain(Plain(f'重载模块 {target_module.module} 成功')))
252 |
253 |
254 | @listen(GroupMessage)
255 | @dispatch(Twilight(RegexMatch('[!！.]加载'), 'module_id' @ RegexMatch(r'\d+')))
256 | @decorate(GroupPermission.require(GroupPermission.BOT_ADMIN), require_disable(channel.module))
257 | async def load_module(app: Ariadne, group: Group, member: Member, module_id: RegexResult):
258 | if module_id.result is None:
259 | return
260 | await app.send_message(
261 | group,
262 | MessageChain(At(member.id), Plain(' 加载新模块有极大可能会出错，请问你确实吗？\n强制加载请在10s内发送 .force ，取消请发送 .cancel')),
263 | )
264 |
265 | async def waiter(waiter_group: Group, waiter_member: Member, waiter_message: MessageChain) -> bool | None:
266 | if waiter_group.id == group.id and waiter_member.id == member.id:
267 | saying = str(waiter_message)
268 | if saying == '.force':
269 | return True
270 | elif saying == '.cancel':
271 | return False
272 | else:
273 | await app.send_message(group, MessageChain(At(member.id), Plain('请发送 .force 或 .cancel')))
274 |
275 | answer = await FunctionWaiter(waiter, [GroupMessage]).wait(timeout=600)
276 | if answer is None:
277 | await app.send_message(group, MessageChain(Plain('已超时取消')))
278 | return
279 | if not answer:
280 | await app.send_message(group, MessageChain(Plain('已取消操作')))
281 | return
282 | match_result = str(module_id.result)
283 | target_filename = match_result if match_result[-3:] != '.py' else match_result[:-3]
284 |
285 | modules_dir_list = os.listdir(Path(Path.cwd(), 'modules'))
286 | if f'{target_filename}.py' in modules_dir_list or target_filename in modules_dir_list:
287 | try:
288 | with saya.module_context():
289 | saya.require(f'modules.{target_filename}')
290 | except Exception as e:
291 | await app.send_message(group, MessageChain(Plain(f'加载模块 modules.{target_filename} 时出错')))
292 | logger.error(f'加载模块 modules.{target_filename} 时出错')
293 | logger.exception(e)
294 | return
295 | else:
296 | await app.send_message(group, MessageChain(Plain(f'加载模块 modules.{target_filename} 成功')))
297 | return
298 | else:
299 | await app.send_message(group, MessageChain(Plain(f'模块 modules.{target_filename} 不存在')))
300 |
301 |
302 | @listen(GroupMessage)
303 | @dispatch(Twilight(RegexMatch(r'[!！.]卸载'), 'module_id' @ RegexMatch(r'\d+')))
304 | @decorate(GroupPermission.require(GroupPermission.BOT_ADMIN), require_disable(channel.module))
305 | async def unload_module(app: Ariadne, group: Group, module_id: RegexResult):
306 | if module_id.result is None:
307 | return
308 | target_module = await get_channel(app, str(module_id.result), group)
309 | if target_module is None:
310 | return
311 | if not target_module.meta.get('can_disable', True):
312 | await app.send_message(group, MessageChain(Plain('你指定的模块不允许禁用或卸载')))
313 | return
314 | logger.debug(f'原channels: {saya.channels}')
315 | logger.debug(f'要卸载的channel: {saya.channels["modules.BiliVideoInfo"]}')
316 | logger.info(f'卸载模块: {target_module._name} —— {target_module.module}')
317 | try:
318 | saya.uninstall_channel(saya.channels[target_module.module])
319 | except Exception as e:
320 | await app.send_message(group, MessageChain(Plain(f'卸载模块 {target_module.module} 时出错')))
321 | logger.error(f'卸载模块 {target_module.module} 时出错')
322 | logger.exception(e)
323 | else:
324 | logger.debug(f'卸载后的channels: {saya.channels}')
325 | await app.send_message(group, MessageChain(Plain(f'卸载模块 {target_module.module} 成功')))
326 |
--------------------------------------------------------------------------------
/core_modules/recall.py:
--------------------------------------------------------------------------------
1 | import contextlib
2 | import time
3 | from datetime import datetime
4 |
5 | from graia.amnesia.builtins.memcache import Memcache
6 | from graia.ariadne.app import Ariadne
7 | from graia.ariadne.event.message import ActiveGroupMessage, GroupMessage
8 | from graia.ariadne.exception import (
9 | InvalidArgument,
10 | RemoteException,
11 | UnknownError,
12 | UnknownTarget,
13 | )
14 | from graia.ariadne.message.chain import MessageChain
15 | from graia.ariadne.message.element import Plain, Source
16 | from graia.ariadne.model import Group
17 | from graia.saya import Channel
18 | from graia.scheduler.saya import SchedulerSchema
19 | from graia.scheduler.timers import crontabify
20 | from graiax.shortcut.saya import decorate, listen
21 |
22 | from libs.config import basic_cfg
23 | from libs.control import require_disable
24 |
25 | channel = Channel.current()
26 |
27 | channel.meta['name'] = '撤回自己的消息'
28 | channel.meta['author'] = ['Red_lnn']
29 | channel.meta['description'] = '撤回bot发的消息\n用法：\n 回复bot的某条消息【撤回】，或发送【撤回最近】以撤回最近发送的消息'
30 | channel.meta['can_disable'] = False
31 |
32 |
33 | @listen(GroupMessage)
34 | @decorate(require_disable(channel.module))
35 | async def recall_message(app: Ariadne, group: Group, message: MessageChain, event: GroupMessage, memcache: Memcache):
36 | if event.sender.id not in basic_cfg.admin.admins:
37 | return
38 | if str(message) == '.撤回最近':
39 | recent_msg: list[dict] = await memcache.get('recent_msg', [])
40 | for item in recent_msg:
41 | with contextlib.suppress(UnknownTarget, InvalidArgument, RemoteException, UnknownError):
42 | await app.recall_message(item['id'])
43 | with contextlib.suppress(ValueError):
44 | recent_msg.remove(item)
45 | await memcache.set('recent_msg', recent_msg)
46 | elif event.quote is not None:
47 | if event.quote.sender_id != basic_cfg.miraiApiHttp.account:
48 | return
49 | if str(message.include(Plain).merge()).strip() == '.撤回':
50 | target_id = event.quote.id
51 | recent_msg: list[dict] = await memcache.get('recent_msg', [])
52 | for item in recent_msg:
53 | if item['id'] == target_id:
54 | if item['time'] - time.time() > 115:
55 | await app.send_message(group, MessageChain(Plain('该消息已超过撤回时间。')), quote=event.source)
56 |
57 | return
58 | with contextlib.suppress(UnknownTarget, InvalidArgument, RemoteException, UnknownError):
59 | await app.recall_message(item['id'])
60 | with contextlib.suppress(ValueError):
61 | recent_msg.remove(item)
62 | await memcache.set('recent_msg', recent_msg)
63 | break
64 |
65 |
66 | @listen(ActiveGroupMessage)
67 | @decorate(require_disable(channel.module))
68 | async def listener(source: Source, memcache: Memcache):
69 | recent_msg: list[dict] = await memcache.get('recent_msg', [])
70 | recent_msg.append(
71 | {
72 | 'time': datetime.timestamp(source.time),
73 | 'id': source.id,
74 | }
75 | )
76 | await memcache.set('recent_msg', recent_msg)
77 |
78 |
79 | @channel.use(SchedulerSchema(crontabify('0/2 * * * *')))
80 | async def clear_outdated(app: Ariadne):
81 | time_now = time.time()
82 | memcache = app.launch_manager.get_interface(Memcache)
83 | recent_msg: list[dict] = await memcache.get('recent_msg', [])
84 | for item in recent_msg:
85 | if (time_now - item['time']) > 120:
86 | recent_msg.remove(item)
87 | await memcache.set('recent_msg', recent_msg)
88 |
--------------------------------------------------------------------------------
/libs/__init__.py:
--------------------------------------------------------------------------------
1 | import logging
2 | import sys
3 | from importlib import metadata
4 | from pathlib import Path
5 | from typing import TYPE_CHECKING
6 |
7 | from loguru import logger
8 | from richuru import LoguruHandler, install
9 |
10 | from libs.config import basic_cfg
11 | from libs.database import DatabaseManager
12 | from static.path import logs_path
13 |
14 | db = DatabaseManager(basic_cfg.databaseUrl)
15 |
16 | if TYPE_CHECKING:
17 | from graia.ariadne.event import MiraiEvent
18 |
19 |
20 | def get_graia_version():
21 | extra: list[tuple[str, str]] = []
22 | official: list[tuple[str, str]] = []
23 | community: list[tuple[str, str]] = []
24 |
25 | for dist in metadata.distributions():
26 | name: str = dist.metadata['Name']
27 | version: str = dist.version
28 | if name in {'launart', 'creart', 'creart-graia', 'statv', 'richuru'}:
29 | extra.append((name, version))
30 | elif name.startswith('graia-'):
31 | official.append((name, version))
32 | elif name.startswith('graiax-'):
33 | community.append((name, version))
34 |
35 | return extra, official, community
36 |
37 |
38 | def replace_logger(level: str | int = 'INFO', richuru: bool = False):
39 | if richuru:
40 | install(level=level)
41 | else:
42 | logging.basicConfig(handlers=[LoguruHandler()], level=0)
43 | logger.remove()
44 | logger.add(sys.stderr, level=level, enqueue=True)
45 | logger.add(
46 | Path(logs_path, 'latest.log'),
47 | level=level,
48 | rotation='00:00',
49 | retention='30 days',
50 | compression='zip',
51 | encoding='utf-8',
52 | enqueue=True,
53 | )
54 |
55 |
56 | def log_level_handler(event: 'MiraiEvent'):
57 | if basic_cfg.debug:
58 | return 'DEBUG'
59 | from graia.ariadne.event.message import (
60 | ActiveMessage,
61 | FriendMessage,
62 | GroupMessage,
63 | OtherClientMessage,
64 | StrangerMessage,
65 | TempMessage,
66 | )
67 |
68 | if type(event) in {
69 | ActiveMessage,
70 | FriendMessage,
71 | GroupMessage,
72 | OtherClientMessage,
73 | StrangerMessage,
74 | TempMessage,
75 | }:
76 | return 'INFO' if basic_cfg.logChat else 'DEBUG'
77 | return 'INFO'
78 |
--------------------------------------------------------------------------------
/libs/better_pydantic.py:
--------------------------------------------------------------------------------
1 | import orjson
2 | from pydantic import BaseModel as PyDanticBaseModel
3 |
4 |
5 | def orjson_dumps(v, *, default, **dumps_kwargs):
6 | # orjson.dumps returns bytes, to match standard json.dumps we need to decode
7 | return orjson.dumps(v, default=default, **dumps_kwargs).decode()
8 |
9 |
10 | class BaseModel(PyDanticBaseModel):
11 | class Config:
12 | json_loads = orjson.loads
13 | json_dumps = orjson_dumps
14 |
--------------------------------------------------------------------------------
/libs/config.py:
--------------------------------------------------------------------------------
1 | from dataclasses import dataclass, field
2 |
3 | from kayaku import config, create
4 |
5 |
6 | @dataclass
7 | class MAHConfig:
8 | account: int = 123456789
9 | """Mirai Api Http 已登录的账号"""
10 | host: str = 'http://localhost:8080'
11 | """Mirai Api Http 地址"""
12 | verifyKey: str = 'VerifyKey'
13 | """Mirai Api Http 的 verifyKey"""
14 |
15 |
16 | @dataclass
17 | class AdminConfig:
18 | masterId: int = 731347477
19 | """机器人主人的QQ号"""
20 | masterName: str = 'Red_lnn'
21 | """机器人主人的称呼"""
22 | admins: list[int] = field(default_factory=lambda: [731347477])
23 | """机器人管理员列表"""
24 |
25 |
26 | @config('redbot')
27 | class BasicConfig:
28 | botName: str = 'RedBot'
29 | """机器人的QQ号"""
30 | logChat: bool = True
31 | """是否将聊天信息打印在日志中"""
32 | debug: bool = False
33 | """是否启用调试模式"""
34 | databaseUrl: str = 'sqlite+aiosqlite:///data/database.db'
35 | """数据库地址
36 |
37 | MySQL示例：mysql+asyncmy://user:pass@hostname/dbname?charset=utf8mb4
38 | """
39 | miraiApiHttp: MAHConfig = field(default_factory=lambda: MAHConfig(account=123456789, verifyKey='VerifyKey'))
40 | """Mirai Api Http 配置"""
41 | admin: AdminConfig = field(default_factory=lambda: AdminConfig())
42 | """机器人管理相关配置"""
43 |
44 |
45 | @config('modules')
46 | class ModulesConfig:
47 | enabled: bool = True
48 | """是否允许加载模块"""
49 | globalDisabledModules: list[str] = field(default_factory=lambda: [])
50 | """全局禁用的模块列表"""
51 | disabledGroups: dict[str, list[int]] = field(
52 | default_factory=lambda: {'core_modules.bot_manage': [123456789, 123456780]}
53 | )
54 | """分群禁用模块的列表"""
55 |
56 |
57 | basic_cfg = create(BasicConfig)
58 | modules_cfg = create(ModulesConfig, flush=True)
59 |
--------------------------------------------------------------------------------
/libs/control/__init__.py:
--------------------------------------------------------------------------------
1 | from graia.ariadne.event.message import GroupMessage, MessageEvent
2 | from graia.ariadne.event.mirai import NudgeEvent
3 | from graia.broadcast import ExecutionStop
4 | from graia.broadcast.builtin.decorators import Depend
5 | from graia.broadcast.entities.event import Dispatchable
6 |
7 | from libs.config import modules_cfg
8 |
9 |
10 | def require_disable(module_name: str) -> Depend:
11 | def wrapper(event: Dispatchable):
12 | if module_name in modules_cfg.globalDisabledModules:
13 | raise ExecutionStop
14 | elif module_name in modules_cfg.disabledGroups:
15 | if isinstance(event, MessageEvent):
16 | if isinstance(event, GroupMessage) and event.sender.group.id in modules_cfg.disabledGroups[module_name]:
17 | raise ExecutionStop
18 | elif event.sender.id in modules_cfg.disabledGroups[module_name]:
19 | raise ExecutionStop
20 | elif isinstance(event, NudgeEvent) and event.target in modules_cfg.disabledGroups[module_name]:
21 | raise ExecutionStop
22 | elif hasattr(event, 'group') and getattr(event, 'group') in modules_cfg.disabledGroups[module_name]:
23 | raise ExecutionStop
24 |
25 | return Depend(wrapper)
26 |
--------------------------------------------------------------------------------
/libs/control/interval.py:
--------------------------------------------------------------------------------
1 | """
2 | 群、用户调用频率限制（bot主人与bot管理员可以无视，没有开关）
3 |
4 | Xenon 管理：https://github.com/McZoo/Xenon/blob/master/lib/control.py
5 | """
6 |
7 | import time
8 | from asyncio import Lock
9 | from collections import defaultdict
10 | from typing import DefaultDict, Optional, Set, Tuple
11 |
12 | from graia.ariadne.app import Ariadne
13 | from graia.ariadne.message.chain import MessageChain
14 | from graia.ariadne.message.element import At, Plain
15 | from graia.ariadne.model import Group, Member
16 | from graia.broadcast import ExecutionStop
17 | from graia.broadcast.builtin.decorators import Depend
18 |
19 | from .permission import GroupPermission
20 |
21 |
22 | class GroupInterval:
23 | """用于管理群组调用bot的冷却的类，不应被实例化"""
24 |
25 | last_exec: DefaultDict[int, Tuple[int, float]] = defaultdict(lambda: (1, 0.0))
26 | last_alert: DefaultDict[int, float] = defaultdict(float)
27 | sent_alert: Set[int] = set()
28 | lock: Optional[Lock] = None
29 |
30 | @classmethod
31 | async def get_lock(cls):
32 | if not cls.lock:
33 | cls.lock = Lock()
34 | return cls.lock
35 |
36 | @classmethod
37 | def require(
38 | cls,
39 | suspend_time: float,
40 | max_exec: int = 1,
41 | send_alert: bool = True,
42 | alert_time_interval: int = 5,
43 | override_level: int = GroupPermission.ADMIN,
44 | ) -> Depend:
45 | """
46 | 指示用户每执行 `max_exec` 次后需要至少相隔 `suspend_time` 秒才能再次触发功能
47 | 等级在 `override_level` 以上的可以无视限制
48 |
49 | :param suspend_time: 冷却时间
50 | :param max_exec: 使用n次后进入冷却
51 | :param send_alert: 是否发送冷却提示
52 | :param alert_time_interval: 发送冷却提示时间间隔，在设定时间内不会重复警告
53 | :param override_level: 可超越限制的最小等级，默认为群管理员
54 | """
55 |
56 | async def cd_check(app: Ariadne, group: Group, member: Member):
57 | if await GroupPermission.get(member) >= override_level:
58 | return
59 | current = time.time()
60 | async with (await cls.get_lock()):
61 | last = cls.last_exec[group.id]
62 | if current - last[1] >= suspend_time:
63 | cls.last_exec[group.id] = (1, current)
64 | if group.id in cls.sent_alert:
65 | cls.sent_alert.remove(group.id)
66 | return
67 | elif last[0] < max_exec:
68 | cls.last_exec[group.id] = (last[0] + 1, current)
69 | if group.id in cls.sent_alert:
70 | cls.sent_alert.remove(group.id)
71 | return
72 | if send_alert:
73 | if group.id not in cls.sent_alert:
74 | m, s = divmod(last[1] + suspend_time - current, 60)
75 | await app.send_message(
76 | group, MessageChain(Plain(f'功能冷却中...\n还有{f"{str(m)}分" if m else ""}{"%d" % s}秒结束'))
77 | )
78 | cls.last_alert[group.id] = current
79 | cls.sent_alert.add(group.id)
80 | elif current - cls.last_alert[group.id] > alert_time_interval:
81 | cls.sent_alert.remove(group.id)
82 | raise ExecutionStop()
83 |
84 | return Depend(cd_check)
85 |
86 |
87 | class MemberInterval:
88 | """用于管理群成员调用bot的冷却的类，不应被实例化"""
89 |
90 | last_exec: DefaultDict[str, Tuple[int, float]] = defaultdict(lambda: (1, 0.0))
91 | last_alert: DefaultDict[str, float] = defaultdict(float)
92 | sent_alert: Set[str] = set()
93 | lock: Optional[Lock] = None
94 |
95 | @classmethod
96 | async def get_lock(cls):
97 | if not cls.lock:
98 | cls.lock = Lock()
99 | return cls.lock
100 |
101 | @classmethod
102 | def require(
103 | cls,
104 | suspend_time: float,
105 | max_exec: int = 1,
106 | send_alert: bool = True,
107 | alert_time_interval: int = 5,
108 | override_level: int = GroupPermission.ADMIN,
109 | ) -> Depend:
110 | """
111 | 指示用户每执行 `max_exec` 次后需要至少相隔 `suspend_time` 秒才能再次触发功能
112 | 等级在 `override_level` 以上的可以无视限制
113 |
114 | :param suspend_time: 冷却时间
115 | :param max_exec: 使用n次后进入冷却
116 | :param send_alert: 是否发送冷却提示
117 | :param alert_time_interval: 警告时间间隔，在设定时间内不会重复警告
118 | :param override_level: 可超越限制的最小等级，默认为群管理员
119 | """
120 |
121 | async def cd_check(app: Ariadne, group: Group, member: Member):
122 | if await GroupPermission.get(member) >= override_level:
123 | return
124 | current = time.time()
125 | name = f'{member.id}_{group.id}'
126 | async with (await cls.get_lock()):
127 | last = cls.last_exec[name]
128 | if current - cls.last_exec[name][1] >= suspend_time:
129 | cls.last_exec[name] = (1, current)
130 | if name in cls.sent_alert:
131 | cls.sent_alert.remove(name)
132 | return
133 | elif last[0] < max_exec:
134 | cls.last_exec[name] = (last[0] + 1, current)
135 | if member.id in cls.sent_alert:
136 | cls.sent_alert.remove(name)
137 | return
138 | if send_alert:
139 | if member.id not in cls.sent_alert:
140 | m, s = divmod(last[1] + suspend_time - current, 60)
141 | await app.send_message(
142 | group,
143 | MessageChain(
144 | At(member.id), Plain(f' 你在本群暂时不可调用bot，正在冷却中...\n还有{f"{m}分" if m else ""}{"%d" % s}秒结束')
145 | ),
146 | )
147 | cls.last_alert[name] = current
148 | cls.sent_alert.add(name)
149 | elif current - cls.last_alert[name] > alert_time_interval:
150 | cls.sent_alert.remove(name)
151 | raise ExecutionStop()
152 |
153 | return Depend(cd_check)
154 |
155 |
156 | class ManualInterval:
157 | """用于管理自定义的调用bot的冷却的类，不应被实例化"""
158 |
159 | last_exec: DefaultDict[str, Tuple[int, float]] = defaultdict(lambda: (1, 0.0))
160 |
161 | @classmethod
162 | def require(cls, name: str, suspend_time: float, max_exec: int = 1) -> Tuple[bool, Optional[float]]:
163 | """
164 | 指示用户每执行 `max_exec` 次后需要至少相隔 `suspend_time` 秒才能再次触发功能
165 |
166 | :param name: 需要被冷却的功能或自定义flag
167 | :param suspend_time: 冷却时间
168 | :param max_exec: 使用n次后进入冷却
169 | :return: True 为冷却中，False 反之，若为 False，还会返回剩余时间
170 | """
171 |
172 | current = time.time()
173 | last = cls.last_exec[name]
174 | if current - cls.last_exec[name][1] >= suspend_time:
175 | cls.last_exec[name] = (1, current)
176 | return True, None
177 | elif last[0] < max_exec:
178 | cls.last_exec[name] = (last[0] + 1, current)
179 | return True, None
180 | return False, round(last[1] + suspend_time - current, 2)
181 |
--------------------------------------------------------------------------------
/libs/control/permission.py:
--------------------------------------------------------------------------------
1 | """
2 | 权限即黑名单检查
3 |
4 | 移植自 Xenon：https://github.com/McZoo/Xenon/blob/master/lib/control.py
5 | """
6 |
7 | from dataclasses import field
8 |
9 | from graia.ariadne.app import Ariadne
10 | from graia.ariadne.message.chain import MessageChain
11 | from graia.ariadne.message.element import At, Plain
12 | from graia.ariadne.model import Group, Member, MemberPerm
13 | from graia.broadcast import ExecutionStop
14 | from graia.broadcast.builtin.decorators import Depend
15 | from kayaku import config, create
16 |
17 | from ..config import basic_cfg
18 |
19 |
20 | @config('permission')
21 | class PermConfig:
22 | group_whitelist: list[int] = field(default_factory=lambda: [])
23 | """白名单群组列表"""
24 | user_blacklist: list[int] = field(default_factory=lambda: [])
25 | """用户黑名单列表"""
26 |
27 |
28 | perm_cfg = create(PermConfig)
29 |
30 |
31 | class GroupPermission:
32 | """
33 | 用于管理权限的类，不应被实例化
34 |
35 | 适用于群消息和来自群的临时会话
36 | """
37 |
38 | BOT_MASTER: int = 100 # Bot主人
39 | BOT_ADMIN: int = 90 # Bot管理员
40 | OWNER: int = 30 # 群主
41 | ADMIN: int = 20 # 群管理员
42 | USER: int = 10 # 群成员/好友
43 | BANED: int = 0 # Bot黑名单成员
44 | DEFAULT: int = USER
45 |
46 | _levels = {
47 | MemberPerm.Member: USER,
48 | MemberPerm.Administrator: ADMIN,
49 | MemberPerm.Owner: OWNER,
50 | }
51 |
52 | @classmethod
53 | async def get(cls, target: Member) -> int:
54 | """
55 | 获取用户的权限等级
56 |
57 | :param target: Friend 或 Member 实例
58 | :return: 等级，整数
59 | """
60 | if target.id == basic_cfg.admin.masterId:
61 | return cls.BOT_MASTER
62 | elif target.id in basic_cfg.admin.admins:
63 | return cls.BOT_ADMIN
64 | elif isinstance(target, Member):
65 | return cls._levels[target.permission]
66 | else:
67 | return cls.DEFAULT
68 |
69 | @classmethod
70 | def require(
71 | cls,
72 | perm: MemberPerm | int = MemberPerm.Member,
73 | send_alert: bool = True,
74 | alert_text: str = '你没有权限执行此指令',
75 | ) -> Depend:
76 | """
77 | 群消息权限检查
78 |
79 | 指示需要 `level` 以上等级才能触发
80 |
81 | :param perm: 至少需要什么权限才能调用
82 | :param send_alert: 是否发送无权限消息
83 | :param alert_text: 无权限提示的消息内容
84 | """
85 |
86 | async def wrapper(app: Ariadne, group: Group, member: Member):
87 | if group.id not in perm_cfg.group_whitelist or member.id in perm_cfg.user_blacklist:
88 | raise ExecutionStop()
89 | if isinstance(perm, MemberPerm):
90 | target = cls._levels[perm]
91 | elif isinstance(perm, int):
92 | target = perm
93 | else:
94 | raise ValueError('perm 参数类型错误')
95 | if (await cls.get(member)) < target:
96 | if send_alert:
97 | await app.send_message(group, MessageChain(At(member.id), Plain(f' {alert_text}')))
98 | raise ExecutionStop()
99 |
100 | return Depend(wrapper)
101 |
--------------------------------------------------------------------------------
/libs/database/__init__.py:
--------------------------------------------------------------------------------
1 | import contextlib
2 | from asyncio import current_task
3 | from typing import Any, Sequence
4 |
5 | from sqlalchemy.engine import Row
6 | from sqlalchemy.engine.result import Result
7 | from sqlalchemy.engine.url import URL
8 | from sqlalchemy.ext.asyncio import (
9 | AsyncSession,
10 | async_scoped_session,
11 | async_sessionmaker,
12 | create_async_engine,
13 | )
14 | from sqlalchemy.ext.asyncio.engine import AsyncEngine
15 | from sqlalchemy.sql.base import Executable
16 |
17 | from .models import Base
18 | from .types import EngineOptions
19 |
20 | # sqlite_url = 'sqlite+aiosqlite:///data/redbot.db'
21 | # mysql_url = 'mysql+asyncmy://user:pass@hostname/dbname?charset=utf8mb4
22 |
23 |
24 | class DatabaseManager:
25 | engine: AsyncEngine
26 | session_factory: async_sessionmaker[AsyncSession]
27 |
28 | def __init__(self, url: str | URL, engine_options: EngineOptions | None = None):
29 | if engine_options is None:
30 | engine_options = {'echo': 'debug', 'pool_pre_ping': True}
31 | self.engine = create_async_engine(url, **engine_options)
32 |
33 | @classmethod
34 | def get_engine_url(
35 | cls,
36 | driver: str = 'asnycmy',
37 | db_type: str = 'mysql',
38 | host: str = 'localhost',
39 | port: int = 3306,
40 | username: str | None = None,
41 | passwd: str | None = None,
42 | database_name: str | None = None,
43 | **kwargs: dict[str, str],
44 | ) -> str:
45 | if db_type == 'mysql':
46 | if username is None or passwd is None or database_name is None:
47 | raise RuntimeError('Option `username` or `passwd` or `database_name` must in parameter.')
48 | url = f'mysql+{driver}://{username}:{passwd}@{host}:{port}/{database_name}'
49 | elif db_type == 'sqlite':
50 | url = f'sqlite+{driver}://'
51 | else:
52 | raise RuntimeError('Unsupport database type, please creating URL manually.')
53 | kw = ''.join(f'&{key}={value}' for key, value in kwargs.items()).lstrip('&')
54 | return url + kw if kw else url
55 |
56 | async def initialize(self, session_options: dict[str, Any] | None = None):
57 | if session_options is None:
58 | session_options = {'expire_on_commit': False}
59 | async with self.engine.begin() as conn:
60 | await conn.run_sync(Base.metadata.create_all)
61 | self.session_factory = async_sessionmaker(self.engine, **session_options)
62 |
63 | async def stop(self):
64 | # for AsyncEngine created in function scope, close and
65 | # clean-up pooled connections
66 | await self.engine.dispose()
67 |
68 | async def async_safe_session(self):
69 | """
70 | 生成一个异步安全的session回话
71 | :return:
72 | """
73 | self._scoped_session = async_scoped_session(self.session_factory, scopefunc=current_task)
74 | return self._scoped_session
75 |
76 | @contextlib.asynccontextmanager
77 | async def async_session(self):
78 | """
79 | 异步session上下文管理封装
80 | :return:
81 |
82 | >>> async with db_manager.async_session() as session:
83 | >>> res = await session.execute("SELECT 1")
84 | >>> print(res.scalar())
85 | 1
86 |
87 | """
88 | scoped_session = await self.async_safe_session()
89 | try:
90 | yield scoped_session
91 | await self._scoped_session.commit()
92 | except Exception:
93 | await self._scoped_session.rollback()
94 | raise
95 | finally:
96 | await self._scoped_session.remove()
97 |
98 | async def exec(self, sql: Executable) -> Result:
99 | async with self.async_session() as session:
100 | return await session.execute(sql)
101 |
102 | async def select_all(self, sql: Executable) -> Sequence[Row]:
103 | result = await self.exec(sql)
104 | return result.all()
105 |
106 | async def select_first(self, sql: Executable) -> Row | None:
107 | result = await self.exec(sql)
108 | return result.first()
109 |
110 | async def add(self, row):
111 | scoped_session = await self.async_safe_session()
112 | try:
113 | scoped_session.add(row)
114 | await self._scoped_session.commit()
115 | await self._scoped_session.refresh(row)
116 | except Exception:
117 | await self._scoped_session.rollback()
118 | raise
119 | finally:
120 | await self._scoped_session.remove()
121 |
122 | async def add_many(self, rows: Sequence[Base]):
123 | scoped_session = await self.async_safe_session()
124 | try:
125 | scoped_session.add_all(rows)
126 | await self._scoped_session.commit()
127 | for row in rows:
128 | await self._scoped_session.refresh(row)
129 | except Exception:
130 | await self._scoped_session.rollback()
131 | raise
132 | finally:
133 | await self._scoped_session.remove()
134 |
135 | async def update_exist(self, row):
136 | scoped_session = await self.async_safe_session()
137 | try:
138 | await scoped_session.merge(row)
139 | await self._scoped_session.commit()
140 | await self._scoped_session.refresh(row)
141 | except Exception:
142 | await self._scoped_session.rollback()
143 | raise
144 | finally:
145 | await self._scoped_session.remove()
146 |
147 | async def delete_exist(self, row):
148 | async with self.async_session() as session:
149 | await session.delete(row)
150 |
151 | async def delete_many_exist(self, *rows):
152 | async with self.async_session() as session:
153 | for row in rows:
154 | await session.delete(row)
155 |
--------------------------------------------------------------------------------
/libs/database/models.py:
--------------------------------------------------------------------------------
1 | from sqlalchemy.ext.asyncio import AsyncAttrs
2 | from sqlalchemy.orm import DeclarativeBase
3 |
4 |
5 | class Base(AsyncAttrs, DeclarativeBase):
6 | pass
7 |
--------------------------------------------------------------------------------
/libs/database/types.py:
--------------------------------------------------------------------------------
1 | from typing import Any, Callable, Dict, List, Literal, Optional, Type, Union
2 |
3 | from sqlalchemy.engine.interfaces import IsolationLevel, _ExecuteOptions, _ParamStyle
4 | from sqlalchemy.log import _EchoFlagType
5 | from sqlalchemy.pool import Pool, _CreatorFnType, _CreatorWRecFnType, _ResetStyleArgType
6 | from typing_extensions import TypedDict
7 |
8 |
9 | class EngineOptions(TypedDict, total=False):
10 | connect_args: Dict[Any, Any]
11 | convert_unicode: bool
12 | creator: Union[_CreatorFnType, _CreatorWRecFnType]
13 | echo: _EchoFlagType
14 | echo_pool: _EchoFlagType
15 | enable_from_linting: bool
16 | execution_options: _ExecuteOptions
17 | future: Literal[True]
18 | hide_parameters: bool
19 | implicit_returning: Literal[True]
20 | insertmanyvalues_page_size: int
21 | isolation_level: IsolationLevel
22 | json_deserializer: Callable[..., Any]
23 | json_serializer: Callable[..., Any]
24 | label_length: Optional[int]
25 | logging_name: str
26 | max_identifier_length: Optional[int]
27 | max_overflow: int
28 | module: Optional[Any]
29 | paramstyle: Optional[_ParamStyle]
30 | pool: Optional[Pool]
31 | poolclass: Optional[Type[Pool]]
32 | pool_logging_name: str
33 | pool_pre_ping: bool
34 | pool_size: int
35 | pool_recycle: int
36 | pool_reset_on_return: Optional[_ResetStyleArgType]
37 | pool_timeout: float
38 | pool_use_lifo: bool
39 | plugins: List[str]
40 | query_cache_size: int
41 | use_insertmanyvalues: bool
42 | kwargs: Dict[str, Any]
43 |
--------------------------------------------------------------------------------
/libs/database_services.py:
--------------------------------------------------------------------------------
1 | from typing import Literal
2 |
3 | from launart import ExportInterface, Launchable
4 | from loguru import logger
5 |
6 | from libs import db
7 |
8 |
9 | class DatabaseInitService(Launchable):
10 | id: str = 'database/init'
11 |
12 | @property
13 | def required(self) -> set[str | type[ExportInterface]]:
14 | return set()
15 |
16 | @property
17 | def stages(self) -> set[Literal['preparing', 'blocking', 'cleanup']]:
18 | return set()
19 |
20 | async def launch(self, _):
21 | logger.info('Initializing database...')
22 | await db.initialize()
23 |
--------------------------------------------------------------------------------
/libs/fonts_provider.py:
--------------------------------------------------------------------------------
1 | from playwright.async_api import Request, Route
2 | from yarl import URL
3 |
4 | from static.path import lib_path
5 |
6 | font_path = lib_path / 'fonts'
7 |
8 | font_mime_map = {
9 | 'collection': 'font/collection',
10 | 'otf': 'font/otf',
11 | 'sfnt': 'font/sfnt',
12 | 'ttf': 'font/ttf',
13 | 'woff': 'font/woff',
14 | 'woff2': 'font/woff2',
15 | }
16 |
17 |
18 | async def fill_font(route: Route, request: Request):
19 | url = URL(request.url)
20 | if (font_path / url.name).exists():
21 | await route.fulfill(
22 | path=font_path / url.name,
23 | content_type=font_mime_map.get(url.suffix),
24 | )
25 | return
26 | await route.fallback()
27 |
--------------------------------------------------------------------------------
/libs/send_action.py:
--------------------------------------------------------------------------------
1 | from typing import TypeVar, overload
2 |
3 | from graia.ariadne.app import Ariadne
4 | from graia.ariadne.event.message import ActiveMessage
5 | from graia.ariadne.message.chain import MessageChain
6 | from graia.ariadne.message.element import Plain
7 | from graia.ariadne.typing import SendMessageAction, SendMessageException
8 | from graia.ariadne.util.send import Ignore
9 |
10 | Exc_T = TypeVar('Exc_T', bound=SendMessageException)
11 |
12 |
13 | class Safe(SendMessageAction):
14 | """
15 | 安全发送的 SendMessage action
16 |
17 | 行为:
18 | 在第一次尝试失败后先移除 quote,
19 | 之后每次失败时按顺序替换元素为其asDisplay: AtAll, At, Poke, Forward, MultimediaElement
20 | 若最后还是失败 (AccountMuted 等), 则会引发原始异常 (通过传入 ignore 决定)
21 | """
22 |
23 | def __init__(self, ignore: bool = False) -> None:
24 | self.ignore: bool = ignore
25 |
26 | @overload
27 | @staticmethod
28 | async def exception(item) -> ActiveMessage:
29 | ...
30 |
31 | @overload
32 | async def exception(self, item) -> ActiveMessage:
33 | ...
34 |
35 | @staticmethod
36 | async def _handle(item: SendMessageException, ignore: bool):
37 | chain: MessageChain = item.send_data['message']
38 | ariadne = Ariadne.current()
39 |
40 | def convert(msg_chain: MessageChain, type_: str) -> None:
41 | for ind, elem in enumerate(msg_chain.__root__[:]):
42 | if elem.type == type_:
43 | if elem.type == 'at':
44 | msg_chain.__root__[ind] = (
45 | Plain(f'@{elem}({elem.target})') # type: ignore
46 | if elem.target is not None # type: ignore
47 | else Plain(f'@{elem.target}') # type: ignore
48 | )
49 | else:
50 | msg_chain.__root__[ind] = Plain(str(elem))
51 |
52 | for element_type in ('AtAll', 'At', 'Poke', 'Forward', 'MultimediaElement'):
53 | convert(chain, element_type)
54 | val = await ariadne.send_message(**item.send_data, action=Ignore) # type: ignore # noqa
55 | if val is not None:
56 | return val
57 |
58 | if not ignore:
59 | raise item
60 |
61 | @overload
62 | @staticmethod
63 | async def exception(s, i):
64 | ...
65 |
66 | @overload
67 | async def exception(s, i): # sourcery skip: instance-method-first-arg-name #type: ignore
68 | ...
69 |
70 | async def exception(s: 'Safe' | Exc_T, i: Exc_T | None = None): # type: ignore # noqa
71 | # sourcery skip: instance-method-first-arg-name
72 | if not isinstance(s, Safe):
73 | return await Safe._handle(s, True)
74 | if i:
75 | return await Safe._handle(i, s.ignore)
76 |
--------------------------------------------------------------------------------
/libs/text2img.py:
--------------------------------------------------------------------------------
1 | import re
2 | from datetime import datetime
3 |
4 | from graiax.text2img.playwright import (
5 | HTMLRenderer,
6 | MarkdownConverter,
7 | PageOption,
8 | ScreenshotOption,
9 | convert_text,
10 | )
11 | from graiax.text2img.playwright.renderer import BuiltinCSS
12 | from jinja2 import Template
13 |
14 | from libs.fonts_provider import fill_font
15 |
16 | footer = (
17 | ''
20 | f''
21 | )
22 |
23 | html_render = HTMLRenderer(
24 | page_option=PageOption(device_scale_factor=1.5),
25 | screenshot_option=ScreenshotOption(type='jpeg', quality=80, full_page=True, scale='device'),
26 | css=(
27 | BuiltinCSS.reset,
28 | BuiltinCSS.github,
29 | BuiltinCSS.one_dark,
30 | BuiltinCSS.container,
31 | "@font-face{font-family:'harmo';font-weight:300;"
32 | "src:url('http://static.graiax/fonts/HarmonyOS_Sans_SC_Light.ttf') format('truetype');}"
33 | "@font-face{font-family:'harmo';font-weight:400;"
34 | "src:url('http://static.graiax/fonts/HarmonyOS_Sans_SC_Regular.ttf') format('truetype');}"
35 | "@font-face{font-family:'harmo';font-weight:500;"
36 | "src:url('http://static.graiax/fonts/HarmonyOS_Sans_SC_Medium.ttf') format('truetype');}"
37 | "@font-face{font-family:'harmo';font-weight:600;"
38 | "src:url('http://static.graiax/fonts/HarmonyOS_Sans_SC_Bold.ttf') format('truetype');}"
39 | "*{font-family:'harmo',sans-serif}",
40 | ),
41 | page_modifiers=[
42 | lambda page: page.route(lambda url: bool(re.match('^http://static.graiax/fonts/(.+)$', url)), fill_font)
43 | ],
44 | )
45 |
46 | md_converter = MarkdownConverter()
47 |
48 |
49 | async def text2img(text: str, width: int = 800) -> bytes:
50 | html = convert_text(text)
51 | html += footer
52 |
53 | return await html_render.render(
54 | html,
55 | extra_page_option=PageOption(viewport={'width': width, 'height': 10}),
56 | )
57 |
58 |
59 | async def md2img(text: str, width: int = 800) -> bytes:
60 | html = md_converter.convert(text)
61 | html += footer
62 |
63 | return await html_render.render(
64 | html,
65 | extra_page_option=PageOption(viewport={'width': width, 'height': 10}),
66 | )
67 |
68 |
69 | async def template2img(
70 | template: str,
71 | render_option: dict[str, str],
72 | *,
73 | extra_page_option: PageOption | None = None,
74 | extra_screenshot_option: ScreenshotOption | None = None,
75 | ) -> bytes:
76 | """Jinja2 模板转图片
77 | Args:
78 | template (str): Jinja2 模板
79 | render_option (Dict[str, str]): Jinja2.Template.render 的参数
80 | return_html (bool): 返回生成的 HTML 代码而不是图片生成结果的 bytes
81 | extra_page_option (PageOption, optional): Playwright 浏览器 new_page 方法的参数
82 | extra_screenshot_option (ScreenshotOption, optional): Playwright 浏览器页面截图方法的参数
83 | extra_page_methods (Optional[List[Callable[[Page], Awaitable]]]):
84 | 默认为 None，用于 https://playwright.dev/python/docs/api/class-page 中提到的部分方法，
85 | 如 `page.route(...)` 等
86 | """
87 | html_code: str = Template(template).render(**render_option)
88 | return await html_render.render(
89 | html_code,
90 | extra_page_option=extra_page_option,
91 | extra_screenshot_option=extra_screenshot_option,
92 | )
93 |
--------------------------------------------------------------------------------
/main.py:
--------------------------------------------------------------------------------
1 | import pkgutil
2 | from pathlib import Path
3 |
4 | import kayaku
5 | from creart import create
6 |
7 | # from graia.amnesia.builtins.uvicorn import UvicornService
8 | from graia.ariadne.app import Ariadne
9 | from graia.ariadne.connection.config import (
10 | HttpClientConfig,
11 | WebsocketClientConfig,
12 | config,
13 | )
14 | from graia.ariadne.model import LogConfig
15 | from graia.saya import Saya
16 | from graiax.playwright import PlaywrightService
17 |
18 | kayaku.initialize({"{**}": "./config/{**}"})
19 |
20 | from libs import log_level_handler, replace_logger # noqa: E402
21 | from libs.config import basic_cfg, modules_cfg # noqa: E402
22 |
23 | # from libs.fastapi_service import FastAPIStarletteService # noqa: E402
24 | from libs.database_services import DatabaseInitService # noqa: E402
25 | from libs.send_action import Safe # noqa: E402
26 | from static.path import modules_path, root_path # noqa: E402
27 |
28 | kayaku.bootstrap()
29 |
30 | # loop = create(AbstractEventLoop) # 若不需要 loop.run_until_complete()，则不需要此行
31 | # 若 create Saya 则可以省掉 bcc 和 scheduler 的
32 | # sche = create(GraiaScheduler)
33 | # bcc = create(Broadcast)
34 | saya = create(Saya)
35 |
36 | Ariadne.options['installed_log'] = True
37 | app = Ariadne(
38 | connection=config(
39 | basic_cfg.miraiApiHttp.account, # 你的机器人的 qq 号
40 | basic_cfg.miraiApiHttp.verifyKey, # 填入 verifyKey
41 | HttpClientConfig(host=basic_cfg.miraiApiHttp.host),
42 | WebsocketClientConfig(host=basic_cfg.miraiApiHttp.host),
43 | ),
44 | log_config=LogConfig(log_level_handler),
45 | )
46 | app.default_send_action = Safe
47 |
48 | replace_logger(level=0 if basic_cfg.debug else 20, richuru=False)
49 |
50 | ignore = ('__init__.py', '__pycache__')
51 |
52 | with saya.module_context():
53 | core_modules_path = Path(root_path, 'core_modules')
54 | for module in pkgutil.iter_modules([str(core_modules_path)]):
55 | if module.name in ignore or module.name[0] in ('#', '.', '_'):
56 | continue
57 | saya.require(f'core_modules.{module.name}')
58 |
59 | if modules_cfg.enabled:
60 | with saya.module_context():
61 | for module in pkgutil.iter_modules([str(modules_path)]):
62 | if module.name in ignore or module.name[0] in ('#', '.', '_'):
63 | continue
64 | saya.require(f'modules.{module.name}')
65 |
66 | if basic_cfg.miraiApiHttp.account == 123456789:
67 | raise ValueError('在?¿ 填一下配置文件？')
68 |
69 | app.launch_manager.add_service(DatabaseInitService())
70 | # app.launch_manager.add_service(FastAPIStarletteService())
71 | # app.launch_manager.add_service(UvicornService())
72 | app.launch_manager.add_service(PlaywrightService())
73 |
74 | Ariadne.launch_blocking()
75 | kayaku.save_all()
76 |
--------------------------------------------------------------------------------
/modules/bili_share_resolver.py:
--------------------------------------------------------------------------------
1 | """
2 | 识别群内的B站链接、分享、av号、BV号并获取其对应的视频的信息
3 |
4 | 以下几种消息均可触发
5 |
6 | - 新版B站app分享的两种小程序
7 | - 旧版B站app分享的xml消息
8 | - B站概念版分享的json消息
9 | - 文字消息里含有B站视频地址，如 https://www.bilibili.com/video/{av/bv号} （m.bilibili.com 也可以
10 | - 文字消息里含有B站视频地址，如 https://b23.tv/3V31Ap
11 | - 文字消息里含有B站视频地址，如 https://b23.tv/3V31Ap
12 | - BV1xx411c7mD
13 | - av2
14 | """
15 |
16 | import re
17 | import time
18 | from base64 import b64encode
19 | from dataclasses import dataclass
20 | from io import BytesIO
21 | from pathlib import Path
22 | from typing import Literal
23 |
24 | from graia.amnesia.builtins.aiohttp import AiohttpClientInterface
25 | from graia.ariadne.app import Ariadne
26 | from graia.ariadne.event.message import GroupMessage
27 | from graia.ariadne.message.chain import MessageChain
28 | from graia.ariadne.message.element import Image, Plain
29 | from graia.ariadne.model import Group, Member
30 | from graia.saya import Channel
31 | from graiax.shortcut.saya import decorate, listen
32 | from graiax.text2img.playwright import PageOption
33 | from launart import Launart
34 | from loguru import logger
35 | from PIL.Image import Image as PILImage
36 | from qrcode.main import QRCode
37 |
38 | from libs.control import require_disable
39 | from libs.control.interval import ManualInterval
40 | from libs.control.permission import GroupPermission
41 | from libs.text2img import template2img
42 | from static.path import root_path
43 |
44 | channel = Channel.current()
45 |
46 | channel.meta['name'] = 'B站视频信息获取'
47 | channel.meta['author'] = ['Red_lnn']
48 | channel.meta['description'] = (
49 | '识别群内的B站链接、分享、av号、BV号并获取其对应的视频的信息\n'
50 | '以下几种消息均可触发：\n'
51 | ' - 新版B站app分享的两种小程序\n'
52 | ' - 旧版B站app分享的xml消息\n'
53 | ' - B站概念版分享的json消息\n'
54 | ' - 文字消息里含有B站视频地址，如 https://www.bilibili.com/video/{av/bv号} （m.bilibili.com 也可以）\n'
55 | ' - 文字消息里含有B站视频地址，如 https://b23.tv/3V31Ap\n'
56 | ' - 文字消息里含有BV号，如 BV1xx411c7mD\n'
57 | ' - 文字消息里含有av号，如 av2'
58 | )
59 |
60 | avid_re = '(av|AV)(\\d{1,12})'
61 | bvid_re = '[Bb][Vv]1([0-9a-zA-Z]{2})4[1y]1[0-9a-zA-Z]7([0-9a-zA-Z]{2})'
62 |
63 |
64 | @dataclass
65 | class VideoInfo:
66 | cover_url: str # 封面地址
67 | bvid: str # BV号
68 | avid: int # av号
69 | title: str # 视频标题
70 | sub_count: int # 视频分P数
71 | pub_timestamp: int # 视频发布时间（时间戳）
72 | upload_timestamp: int # 视频上传时间（时间戳，不一定准确）
73 | desc: str # 视频简介
74 | duration: int # 视频长度（单位：秒）
75 | up_mid: int # up主mid
76 | up_name: str # up主名称
77 | up_face: str # up主头像地址
78 | views: int # 播放量
79 | danmu: int # 弹幕数
80 | likes: int # 点赞数
81 | coins: int # 投币数
82 | replys: int # 评论数
83 | favorites: int # 收藏量
84 |
85 |
86 | @listen(GroupMessage)
87 | @decorate(GroupPermission.require(), require_disable(channel.module))
88 | async def main(app: Ariadne, group: Group, message: MessageChain, member: Member):
89 | p = re.compile(f'({avid_re})|({bvid_re})')
90 | msg_str = message.as_persistent_string()
91 | if 'b23.tv/' in msg_str:
92 | msg_str = await b23_url_extract(msg_str)
93 | if not msg_str:
94 | return
95 | video_id = p.search(msg_str)
96 | if not video_id or video_id is None:
97 | return
98 | video_id = video_id.group()
99 |
100 | rate_limit, remaining_time = ManualInterval.require(f'{group.id}_{member.id}_bilibiliVideoInfo', 5, 2)
101 | if not rate_limit:
102 | await app.send_message(group, MessageChain(Plain(f'冷却中，剩余{remaining_time}秒，请稍后再试')))
103 | return
104 |
105 | video_info = await get_video_info(video_id)
106 | if video_info['code'] == -404:
107 | return await app.send_message(group, MessageChain(Plain('视频不存在')))
108 | elif video_info['code'] != 0:
109 | error_text = f'解析B站视频 {video_id} 时出错👇\n错误代码：{video_info["code"]}\n错误信息：{video_info["message"]}'
110 | logger.error(error_text)
111 | return await app.send_message(group, MessageChain(Plain(error_text)))
112 | else:
113 | video_info = await info_json_dump(video_info['data'])
114 | img: bytes = await gen_img(video_info)
115 | await app.send_message(
116 | group,
117 | MessageChain(
118 | Image(data_bytes=img),
119 | Plain(
120 | f'{video_info.title}\n'
121 | '—————————————\n'
122 | f'UP主：{video_info.up_name}\n'
123 | f'{math(video_info.views)}播放 {math(video_info.likes)}赞\n'
124 | f'链接：https://b23.tv/{video_info.bvid}'
125 | ),
126 | ),
127 | )
128 |
129 |
130 | async def b23_url_extract(b23_url: str) -> Literal[False] | str:
131 | url = re.search(r'b23.tv[/\\]+([0-9a-zA-Z]+)', b23_url)
132 | if url is None:
133 | return False
134 |
135 | launart = Launart.current()
136 | session = launart.get_interface(AiohttpClientInterface).service.session
137 |
138 | async with session.get(f'https://{url.group()}', allow_redirects=True) as resp:
139 | target = str(resp.url)
140 | return target if 'www.bilibili.com/video/' in target else False
141 |
142 |
143 | async def get_video_info(video_id: str) -> dict:
144 | launart = Launart.current()
145 | session = launart.get_interface(AiohttpClientInterface).service.session
146 |
147 | if video_id[:2].lower() == 'av':
148 | async with session.get(f'http://api.bilibili.com/x/web-interface/view?aid={video_id[2:]}') as resp:
149 | return await resp.json()
150 | elif video_id[:2].lower() == 'bv':
151 | async with session.get(f'http://api.bilibili.com/x/web-interface/view?bvid={video_id}') as resp:
152 | return await resp.json()
153 | return {}
154 |
155 |
156 | async def info_json_dump(obj: dict) -> VideoInfo:
157 | return VideoInfo(
158 | cover_url=obj['pic'],
159 | bvid=obj['bvid'],
160 | avid=obj['aid'],
161 | title=obj['title'],
162 | sub_count=obj['videos'],
163 | pub_timestamp=obj['pubdate'],
164 | upload_timestamp=obj['ctime'],
165 | desc=obj['desc'].strip(),
166 | duration=obj['duration'],
167 | up_mid=obj['owner']['mid'],
168 | up_name=obj['owner']['name'],
169 | up_face=obj['owner']['face'],
170 | views=obj['stat']['view'],
171 | danmu=obj['stat']['danmaku'],
172 | likes=obj['stat']['like'],
173 | coins=obj['stat']['coin'],
174 | replys=obj['stat']['reply'],
175 | favorites=obj['stat']['favorite'],
176 | )
177 |
178 |
179 | def math(num: int):
180 | if num < 10000:
181 | return str(num)
182 | elif num < 100000000:
183 | return ('%.2f' % (num / 10000)) + '万'
184 | else:
185 | return ('%.2f' % (num / 100000000)) + '亿'
186 |
187 |
188 | template = Path(root_path, 'static', 'jinja2_templates', 'bili_video.html').read_text(encoding='utf-8')
189 |
190 |
191 | async def gen_img(data: VideoInfo) -> bytes:
192 | video_length_m, video_length_s = divmod(data.duration, 60) # 将总的秒数转换为时分秒格式
193 | video_length_h, video_length_m = divmod(video_length_m, 60)
194 | if video_length_h == 0:
195 | video_length = f'{video_length_m}:{video_length_s}'
196 | else:
197 | video_length = f'{video_length_h}:{video_length_m}:{video_length_s}'
198 |
199 | desc = data.desc.strip().replace('\n', '') 200 | 201 | launart = Launart.current() 202 | session = launart.get_interface(AiohttpClientInterface).service.session 203 | 204 | async with session.get(f'https://api.bilibili.com/x/relation/stat?vmid={data.up_mid}') as resp: 205 | result = await resp.json() 206 | fans_num: int = result['data']['follower'] 207 | 208 | qr = QRCode(version=2, box_size=4) 209 | qr.add_data(f'https://b23.tv/{data.bvid}') 210 | qr.make() 211 | qrcode: PILImage | None = qr.make_image()._img 212 | if qrcode is None: 213 | qrcode_src = '' 214 | else: 215 | output_buffer = BytesIO() 216 | qrcode.save(output_buffer, format='png') 217 | base64_str = b64encode(output_buffer.getvalue()).decode('utf-8') 218 | qrcode_src = f'data:image/png;base64,{base64_str}' 219 | 220 | return await template2img( 221 | template, 222 | { 223 | 'cover_src': data.cover_url, 224 | 'duration': video_length, 225 | 'title': data.title, 226 | 'play_num': math(data.views), 227 | 'danmaku_num': math(data.danmu), 228 | 'reply_num': math(data.replys), 229 | 'like_num': math(data.likes), 230 | 'coin_num': math(data.coins), 231 | 'favorite_num': math(data.favorites), 232 | 'desc': desc, 233 | 'publish_time': time.strftime('%Y/%m/%d %p %I:%M:%S', time.localtime(data.upload_timestamp)), 234 | 'profile_src': data.up_face, 235 | 'name': data.up_name, 236 | 'fans_num': math(fans_num), 237 | 'qrcode_src': qrcode_src, 238 | }, 239 | extra_page_option=PageOption(viewport={'width': 960, 'height': 10}), 240 | ) 241 | -------------------------------------------------------------------------------- /modules/dont_nudge_me.py: -------------------------------------------------------------------------------- 1 | import asyncio 2 | import contextlib 3 | import os 4 | from pathlib import Path 5 | from random import choice, randrange, uniform 6 | 7 | from graia.ariadne.app import Ariadne 8 | from graia.ariadne.event.mirai import NudgeEvent 9 | from graia.ariadne.exception import UnknownTarget 10 | from graia.ariadne.message.chain import MessageChain 11 | from graia.ariadne.message.element import Image, Plain 12 | from graia.ariadne.model import Friend, Group 13 | from graia.saya import Channel 14 | from graiax.shortcut.saya import decorate, listen 15 | 16 | from libs.config import basic_cfg 17 | from libs.control import require_disable 18 | from libs.control.interval import ManualInterval 19 | from static.path import data_path 20 | 21 | channel = Channel.current() 22 | 23 | channel.meta['name'] = '别戳我' 24 | channel.meta['author'] = ['Red_lnn'] 25 | channel.meta['description'] = '戳一戳bot' 26 | 27 | msg = ( 28 | '别{}啦别{}啦，无论你再怎么{}，我也不会多说一句话的~', 29 | '你再{}！你再{}！你再{}试试！！', 30 | '那...那里...那里不能{}...绝对...绝对不能（小声）...', 31 | '那里不可以...', 32 | '怎么了怎么了？发生什么了？！纳尼？没事？没事你{}我干哈？', 33 | '气死我了！别{}了别{}了！再{}就坏了呜呜...┭┮﹏┭┮', 34 | '呜…别{}了…', 35 | '呜呜…受不了了', 36 | '别{}了！...把手拿开呜呜..', 37 | 'hentai！八嘎！无路赛！', 38 | '変態！バカ！うるさい！', 39 | '。', 40 | '哼哼╯^╰', 41 | ) 42 | 43 | 44 | async def get_message(event: NudgeEvent): 45 | tmp = randrange(0, len(os.listdir(Path(data_path, 'Nudge'))) + len(msg)) 46 | if tmp < len(msg): 47 | return MessageChain(Plain(msg[tmp].replace('{}', event.msg_action[0]))) 48 | if not Path(data_path, 'Nudge').exists(): 49 | Path(data_path, 'Nudge').mkdir() 50 | elif len(os.listdir(Path(data_path, 'Nudge'))) == 0: 51 | return MessageChain(Plain(choice(msg).replace('{}', event.msg_action[0]))) 52 | return MessageChain(Image(path=Path(data_path, 'Nudge', os.listdir(Path(data_path, 'Nudge'))[tmp - len(msg)]))) 53 | 54 | 55 | @listen(NudgeEvent) 56 | @decorate(require_disable(channel.module)) 57 | async def main(app: Ariadne, event: NudgeEvent): 58 | if event.target != basic_cfg.miraiApiHttp.account: 59 | return 60 | elif not ManualInterval.require(f'{event.supplicant}_{event.target}', 3): 61 | return 62 | await asyncio.sleep(uniform(0.2, 0.6)) 63 | with contextlib.suppress(UnknownTarget): 64 | await app.send_nudge(event.supplicant, event.target) 65 | await asyncio.sleep(uniform(0.2, 0.6)) 66 | if isinstance(event.subject, Group | Friend): 67 | await app.send_message(event.subject, (await get_message(event))) 68 | -------------------------------------------------------------------------------- /modules/eat_what/__init__.py: -------------------------------------------------------------------------------- 1 | import random 2 | from pathlib import Path 3 | 4 | from aiofile import async_open 5 | from graia.ariadne.app import Ariadne 6 | from graia.ariadne.event.message import GroupMessage 7 | from graia.ariadne.message.chain import MessageChain 8 | from graia.ariadne.message.element import Plain, Source 9 | from graia.ariadne.message.parser.twilight import RegexMatch, SpacePolicy, Twilight 10 | from graia.ariadne.model import Group 11 | from graia.saya import Channel 12 | from graiax.shortcut.saya import decorate, dispatch, listen 13 | 14 | from libs.control import require_disable 15 | from libs.control.permission import GroupPermission 16 | 17 | channel = Channel.current() 18 | 19 | channel.meta['name'] = '吃啥' 20 | channel.meta['author'] = ['Red_lnn'] 21 | channel.meta['description'] = '[!！.]吃啥' 22 | 23 | 24 | async def get_food(): 25 | async with async_open(Path(Path(__file__).parent, 'foods.txt')) as afp: 26 | foods = await afp.read() 27 | return random.choice(foods.strip().split('\n')) 28 | 29 | 30 | @listen(GroupMessage) 31 | @dispatch(Twilight(RegexMatch(r'[!！.]吃啥').space(SpacePolicy.NOSPACE))) 32 | @decorate(GroupPermission.require(), require_disable(channel.module)) 33 | async def main(app: Ariadne, group: Group, source: Source): 34 | food = await get_food() 35 | chain = MessageChain(Plain(f'吃{food}')) 36 | await app.send_message(group, chain, quote=source) 37 | -------------------------------------------------------------------------------- /modules/eat_what/foods.txt: -------------------------------------------------------------------------------- 1 | 酸辣土豆丝 2 | 可乐鸡翅 3 | 麻婆豆腐 4 | 红烧肉 5 | 糖醋排骨 6 | 西红柿炒鸡蛋 7 | 青椒炒肉丝 8 | 鱼香肉丝 9 | 水煮肉片 10 | 西红柿炖牛腩 11 | 菠萝咕噜肉 12 | 咖喱鸡翅 13 | 香辣牛肉 14 | 水煮鱼 15 | 酸菜鱼 16 | 虾皮鸡蛋羹 17 | 皮蛋拌豆腐 18 | 清蒸大闸蟹 19 | 微波番茄虾 20 | 麻辣鱼 21 | 宫保鸡丁 22 | 家常豆腐 23 | 皮蛋瘦肉粥 24 | 红烧冬瓜 25 | 湘式小炒五花肉 26 | 剁椒鱼头 27 | 红烧鱼块 28 | 清蒸鲈鱼 29 | 酸辣汤 30 | 秘制红焖羊肉 31 | 酱牛肉 32 | 蒜苔炒腊肉 33 | 杭椒牛柳 34 | 红烧排骨 35 | 泡菜炒年糕 36 | 西红柿鸡蛋 37 | 土豆炖牛肉 38 | 回锅肉 39 | 木须肉 40 | 水晶猪皮冻 41 | 莲藕炖排骨 42 | 鱼香茄子 43 | 地三鲜 44 | 京酱肉丝 45 | 苦瓜炒蛋 46 | 啤酒鸭 47 | 西红柿鸡蛋汤 48 | 红烧茄子 49 | 干煸豆角 50 | 糖醋鱼 51 | 冬瓜排骨汤 52 | 凉拌黄瓜 53 | 鱼头豆腐汤 54 | 银耳莲子汤 55 | 清蒸鱼 56 | 葱油饼 57 | 米汤鸡蛋羹 58 | 蚂蚁上树 59 | 干煸菜花 60 | 香辣干锅土豆片 61 | 黄豆芽炒粉条 62 | 卤牛肉 63 | 苦瓜鸡蛋 64 | 盐烤花生米 65 | 香辣虾 66 | 小鸡炖蘑菇 67 | 蛋黄小米粥 68 | 杏仁瓦片 69 | 蜜汁叉烧排骨 70 | 栗子红烧肉 71 | 酸豆角炒鸡胗 72 | 干煸四季豆 73 | 尖椒土豆丝 74 | 青椒土豆丝 75 | 蒜香排骨 76 | 香辣烤凤爪 77 | 青椒炒肉 78 | 酸萝卜老鸭汤 79 | 葱爆羊肉 80 | 虫草花豆腐汤 81 | 酸辣蕨根粉 82 | 黄豆炖猪蹄 83 | 家常葱油饼 84 | 辣子鸡 85 | 盐水毛豆 86 | 牙签肉 87 | 避风塘炒蟹 88 | 五香毛豆 89 | 腊味煲仔饭 90 | 黑芝麻糊 91 | 炸藕盒 92 | 糯米枣 93 | 肉末香菇豆腐 94 | 洋葱拌木耳 95 | 黄金玉米烙 96 | 黄瓜拌粉皮 97 | 白灼虾 98 | 梅干菜干煸豆角 99 | 小炒圆白菜 100 | 家常红烧牛肉 101 | 炸酱面 102 | 肉末酸豆角 103 | 鲜橙蒸蛋 104 | 辣椒油 105 | 南瓜蒸百合 106 | 酸辣黄瓜 107 | 家常烧茄子 108 | 麦辣鸡腿堡 109 | 板烧鸡腿堡 110 | 炸薯条 111 | 香辣鸡腿堡 112 | 炸鸡翅 113 | 炸鸡腿 114 | 琵琶腿 115 | 鸡米花 116 | 香辣鸡肉卷 117 | 鳕鱼卷 118 | 上校鸡块 119 | 骨肉相连 120 | 黄金脆薯饼 121 | 脆皮全鸡 122 | 蜜汁手扒鸡 123 | M9和牛 124 | M10和牛 125 | M11和牛 126 | M12和牛 127 | 帝皇蟹 128 | 梭子蟹 129 | 鲱鱼罐头 130 | 豆汁 131 | 臭豆腐 132 | 豆浆 133 | 小笼包 134 | 豆浆油条 135 | 热干面 136 | 烙饼 137 | 丸子汤 138 | 煎蛋 139 | 鸡蛋灌饼 140 | 三明治 141 | 茶叶蛋 142 | 饭团 143 | 绿豆汤 144 | AD钙奶 145 | 红烧肉 146 | 芝麻脆皮鸡 147 | 油炸大排 148 | 大盘鸡 149 | 烤肉拌饭 150 | 压缩饼干 151 | 水 152 | 白开水 153 | 空气 154 | 泡面 155 | 肉夹馍 156 | 辣炒菜花 157 | 红豆芋圆奶昔粥 158 | 提拉米苏布丁 159 | 麻辣烫 160 | 可爱多 161 | 鲜肉月饼 162 | 汉堡王 163 | 香菜冰激凌 164 | 鲜芋雪冰 165 | 叫妈烤鸡 166 | 脊骨肉 167 | 芋圆奶茶 168 | 牛肉面 169 | 咖啡长崎蛋糕 170 | 叉烧肉饭团 171 | 汉堡王鱼堡 172 | 菜单 173 | 番茄鱼 174 | 凉皮 175 | 麻辣小面 176 | 麻辣牛肉 177 | 煎饼果子 178 | 粽子 179 | 汤达人 180 | 达人泡面 181 | 口水鸡 182 | 大坂烧烤鸡饭 183 | 大阪烧烤鸡饭 184 | 火锅鸡 185 | 关东煮 186 | 麻辣香锅 187 | 黄瓜元气森林 188 | 调料包 189 | 照烧鸡 190 | 白猫 191 | 酸辣粉 192 | 芝士焗饭 193 | 星冰乐 194 | 二手烟 195 | 瓜子 196 | 炒牛肉 197 | 烤年糕 198 | 元宵 199 | 火锅 200 | 杨掌柜粉面菜蛋 201 | 麻辣串 202 | 烧烤配啤酒 203 | 炸鸡 204 | 麻辣兔头 205 | 羊肉面 206 | 奥利给 207 | 手撕包菜 208 | 低软壳 209 | 洋葱 210 | 洋葱外皮 211 | 肉末蒸蛋 212 | 洋葱肉 213 | 上汤娃娃菜 214 | 野兽仙贝 215 | 粽子 216 | 兔肉粽 217 | 洋葱小笼包 218 | 辣椒炒西芹 219 | 鸡脖 220 | 鸡扒 221 | 喷香滴 222 | 李姐 223 | 朵小花花 224 | 吃牛子 225 | 合味道 226 | 合味道海鲜干拌面 227 | 番茄甜辣孜然香辣酱半鸡 228 | 薯片 229 | 澳洲大龙虾 230 | 绿豆饼 231 | 腐乳饼 232 | 腐乳 233 | 冰淇淋 234 | 雪糕 235 | 阿华田 236 | 盐焗鸡 237 | 白切鸡 238 | 白云猪手 239 | 黄油蟹 240 | 叉烧 241 | 礼云子蒸蛋 242 | 响螺瘦肉靓汤 243 | 花甲粥 244 | 白切鹅 245 | 牛杂 246 | 南乳花生 247 | 猪脚姜 248 | 牛蹄 249 | 番薯粥 250 | 菠萝油 251 | 花生饼 252 | 猪骨鱼蛋面 253 | 胡椒鸡脚汤 254 | 鱼 255 | 培根 256 | 四层培根芝士牛肉汉堡 257 | 四人汉堡炸鸡薯条套餐 258 | 小丸子 259 | 紫菜包饭拼盘 260 | 甜甜圈 261 | 麦旋风 262 | 芝士炸鸡狼牙土豆 263 | 糖豆拼盘 264 | 油爆大虾 265 | 薯条 266 | 炸鸡花 267 | 薯条拼盘 268 | 汉堡热狗薯条套餐 269 | 紫菜包饭火鸡面套餐 270 | 脏脏茶 271 | 鸡盒 272 | 鸡块 273 | 寿司套餐 274 | 青椒炸虾 275 | 炸虾热狗 276 | 鸡块薯条套餐 277 | 六层牛肉芝士汉堡 278 | 培根牛肉汉堡 279 | 焦糖盐巧克力棒配煎饼 280 | 虾炒面 281 | 奥利奥雪糕夹心 282 | 三份牛肉堡 283 | 汉堡小食盒 284 | 紫菜包饭小份 285 | 通心粉 286 | 费列罗甜筒 287 | 鸡块汉堡 288 | 双人餐 289 | 西班牙油条（吉事果） 290 | 小食组合 291 | 炸虾小份 292 | 双人套餐 293 | 炸鸡薯条 294 | 甜筒尖尖 295 | 鸡排意大利面 296 | 鸡腿鸡翅拼盘 297 | 芝士牛肉汉堡和薯条牛肉汉堡 298 | 炸鸡薯条欢乐派对餐 299 | 秋游分享寿司套餐 300 | 生猛海鲜 301 | 曲奇饼干 302 | 普通汉堡 303 | 多彩小食盒 304 | 九层牛肉堡 305 | 巧克力蛋糕 306 | 恒河水泡面 307 | 腐竹炒肉 308 | 寿喜锅 309 | 炒粉利 310 | 地锅鸡 311 | 蒜爆鱼 312 | 锅贴 313 | 鸭血粉丝汤 314 | 小笼包 315 | 饺子 316 | 大杯奥利奥巧克力奶昔 317 | 吊烧资本家 318 | 龙井虾仁 319 | 东坡肉 320 | 螃蟹肉饼蒸蛋 321 | 肉丸粥 322 | 关东煮 323 | 旺仔牛奶 324 | 重庆小面 325 | 黑椒鸡扒饭 326 | 碳烤松茸 327 | 金针番茄肥牛汤 328 | 蚝油笋片杏鲍菇 329 | 凤梨肉粽 330 | 披萨 331 | 桃 🍑 332 | peach 333 | 💊 334 | 药 335 | 屁 336 | 屎 337 | 💩 338 | ？吃什么吃？别吃了 339 | 个锤子 340 | 个桃桃，不过可能会好凉凉 341 | -------------------------------------------------------------------------------- /modules/mc_wiki_searcher.py: -------------------------------------------------------------------------------- 1 | """ 2 | 搜索我的世界中文Wiki 3 | 4 | 用法：在群内发送【!wiki {关键词}】即可 5 | """ 6 | 7 | from asyncio.exceptions import TimeoutError 8 | from urllib.parse import quote 9 | 10 | from graia.amnesia.builtins.aiohttp import AiohttpClientInterface 11 | from graia.ariadne.app import Ariadne 12 | from graia.ariadne.event.message import GroupMessage 13 | from graia.ariadne.message.chain import MessageChain 14 | from graia.ariadne.message.element import Plain 15 | from graia.ariadne.message.parser.twilight import ( 16 | RegexMatch, 17 | RegexResult, 18 | SpacePolicy, 19 | Twilight, 20 | ) 21 | from graia.ariadne.model import Group 22 | from graia.saya import Channel 23 | from graiax.shortcut.saya import decorate, dispatch, listen 24 | from lxml import etree 25 | 26 | from libs.control import require_disable 27 | from libs.control.permission import GroupPermission 28 | 29 | channel = Channel.current() 30 | 31 | channel.meta['name'] = '我的世界中文Wiki搜索' 32 | channel.meta['author'] = ['Red_lnn'] 33 | channel.meta['description'] = '[!！.]wiki <要搜索的关键词>' 34 | 35 | 36 | @listen(GroupMessage) 37 | @dispatch(Twilight(RegexMatch(r'[!！.]wiki').space(SpacePolicy.FORCE), 'keyword' @ RegexMatch(r'\S+'))) 38 | @decorate(GroupPermission.require(), require_disable(channel.module)) 39 | async def main(app: Ariadne, group: Group, keyword: RegexResult, aiohttp: AiohttpClientInterface): 40 | if keyword.result is None: 41 | return 42 | key_word: str = str(keyword.result).strip() 43 | search_parm: str = quote(key_word, encoding='utf-8') 44 | 45 | bili_search_url = f'https://searchwiki.biligame.com/mc/index.php?search={search_parm}' 46 | fandom_search_url = f'https://minecraft.fandom.com/zh/index.php?search={search_parm}' 47 | 48 | bili_url = f'https://wiki.biligame.com/mc/{search_parm}' 49 | fandom_url = f'https://minecraft.fandom.com/zh/wiki/{search_parm}?variant=zh-cn' 50 | 51 | try: 52 | async with aiohttp.service.session.get(bili_url) as resp: 53 | status_code = resp.status 54 | text = await resp.text() 55 | except TimeoutError: 56 | status_code = -1 57 | 58 | match status_code: 59 | case 404: 60 | msg = MessageChain( 61 | Plain( 62 | f'Minecraft Wiki 没有名为【{key_word}】的页面，' 63 | '要继续搜索请点击下面的链接：\n' 64 | f'Bilibili 镜像: {bili_search_url}\n' 65 | f'Fandom: {fandom_search_url}' 66 | ) 67 | ) 68 | case 200: 69 | tree = etree.HTML(text) # type: ignore # 虽然这里缺少参数，但可以运行，文档示例也如此 70 | title = tree.xpath('/html/head/title')[0].text 71 | introduction_list: list = tree.xpath( 72 | '//div[contains(@id,"toc")]/preceding::p[1]/descendant-or-self::*/text()' 73 | ) 74 | introduction = ''.join(introduction_list).strip() 75 | msg = MessageChain(Plain(f'{title}\n{introduction}\nBilibili 镜像: {bili_url}\nFandom: {fandom_url}')) 76 | case _: 77 | msg = MessageChain( 78 | Plain( 79 | f'无法查询 Minecraft Wiki，错误代码：{status_code}\n' 80 | f'要继续搜索【{key_word}】请点击下面的链接：\n' 81 | f'Bilibili 镜像: {bili_search_url}\n' 82 | f'Fandom: {fandom_search_url}' 83 | ) 84 | ) 85 | 86 | await app.send_message(group, msg) 87 | -------------------------------------------------------------------------------- /modules/minecraft_ping/__init__.py: -------------------------------------------------------------------------------- 1 | """ 2 | Ping mc服务器 3 | 4 | 获取指定mc服务器的信息 5 | 6 | > 命令：/ping [mc服务器地址] 7 | """ 8 | 9 | import socket 10 | from dataclasses import field 11 | 12 | from graia.ariadne.app import Ariadne 13 | from graia.ariadne.event.message import GroupMessage 14 | from graia.ariadne.message.chain import MessageChain 15 | from graia.ariadne.message.element import Plain 16 | from graia.ariadne.message.parser.twilight import RegexMatch, RegexResult, Twilight 17 | from graia.ariadne.model import Group 18 | from graia.saya import Channel 19 | from graiax.shortcut.saya import decorate, dispatch, listen 20 | from kayaku import config, create 21 | from loguru import logger 22 | 23 | from libs.control import require_disable 24 | from libs.control.interval import MemberInterval 25 | from libs.control.permission import GroupPermission 26 | 27 | from .ping_client import ping 28 | from .utils import is_domain, is_ip 29 | 30 | channel = Channel.current() 31 | 32 | channel.meta['name'] = 'Ping 我的世界服务器' 33 | channel.meta['author'] = ['Red_lnn'] 34 | channel.meta['description'] = '获取指定mc服务器的信息\n用法：\n[!！.]ping {mc服务器地址}' 35 | 36 | 37 | @config(channel.module) 38 | class McServerPingConfig: 39 | servers: dict[str, str] = field(default_factory=lambda: {'123456789': 'localhost:25565'}) 40 | """指定群组的默认服务器""" 41 | 42 | 43 | ping_cfg = create(McServerPingConfig) 44 | 45 | 46 | @listen(GroupMessage) 47 | @dispatch(Twilight(RegexMatch(r'[!！.]ping'), 'ping_target' @ RegexMatch(r'\S+', optional=True))) 48 | @decorate(GroupPermission.require(), MemberInterval.require(10), require_disable(channel.module)) 49 | async def main(app: Ariadne, group: Group, ping_target: RegexResult): 50 | if ping_target.matched and ping_target.result is not None: 51 | server_address = str(ping_target.result).strip() 52 | else: 53 | if str(group.id) not in ping_cfg.servers: 54 | await app.send_message(group, MessageChain(Plain('该群组没有设置默认服务器地址'))) 55 | return 56 | server_address = ping_cfg.servers[str(group.id)] 57 | 58 | if '://' in server_address: 59 | await app.send_message(group, MessageChain(Plain('不支持带有协议前缀的地址'))) 60 | return 61 | elif '/' in server_address: 62 | await app.send_message(group, MessageChain(Plain('ping目标地址出现意外字符'))) 63 | return 64 | 65 | if is_ip(server_address): 66 | kwargs = {'ip': server_address} 67 | elif ':' in server_address: 68 | host, port = server_address.split(':', 1) 69 | if is_ip(host) or is_domain(host): 70 | if port.isdigit(): 71 | kwargs = {'url': host, 'port': int(port)} 72 | else: 73 | await app.send_message(group, MessageChain(Plain('端口号格式不正确'))) 74 | return 75 | else: 76 | await app.send_message(group, MessageChain(Plain('目标地址不是一个有效的域名或IP（不支持中文域名）'))) 77 | return 78 | elif is_domain(server_address): 79 | kwargs = {'url': server_address} 80 | else: 81 | await app.send_message(group, MessageChain(Plain('目标地址不是一个有效的域名或IP（不支持中文域名）'))) 82 | return 83 | 84 | try: 85 | ping_result = await ping(**kwargs) 86 | except ConnectionRefusedError: 87 | await app.send_message(group, MessageChain(Plain('连接被目标拒绝，该地址（及端口）可能不存在 Minecraft 服务器'))) 88 | logger.warning(f'连接被目标拒绝，该地址（及端口）可能不存在Minecraft服务器，目标地址：{server_address}') 89 | return 90 | except socket.timeout: 91 | await app.send_message(group, MessageChain(Plain('连接超时'))) 92 | logger.warning(f'连接超时，目标地址：{server_address}') 93 | return 94 | except socket.gaierror as e: 95 | await app.send_message(group, MessageChain(Plain('出错了，可能是无法解析目标地址\n' + str(e)))) 96 | logger.exception(e) 97 | return 98 | 99 | if not ping_result: 100 | await app.send_message(group, MessageChain(Plain('无法解析目标地址'))) 101 | return 102 | 103 | if ping_result['motd'] is not None and ping_result['motd'] != '': 104 | motd_list: list[str] = ping_result['motd'].split('\n') 105 | motd = f' | {motd_list[0].strip()}' 106 | if len(motd_list) == 2: 107 | motd += f'\n | {motd_list[1].strip()}' 108 | else: 109 | motd = None 110 | msg_send = f'咕咕咕！🎉\n服务器版本: [{ping_result["protocol"]}] {ping_result["version"]}\n' 111 | msg_send += f'MOTD:\n{motd}\n' if motd is not None else '' 112 | msg_send += f'延迟: {ping_result["delay"]}ms\n在线人数: {ping_result["online_player"]}/{ping_result["max_player"]}' 113 | if ping_result['online_player'] != '0' and ping_result['player_list']: 114 | players_list = ''.join(f' | {_["name"]}\n' for _ in ping_result['player_list']) 115 | if int(ping_result['online_player']) != len(ping_result['player_list']): 116 | msg_send += f'\n在线列表：\n{players_list.rstrip()}\n | ...' 117 | else: 118 | msg_send += f'\n在线列表：\n{players_list.rstrip()}' 119 | 120 | await app.send_message(group, MessageChain(Plain(msg_send))) 121 | -------------------------------------------------------------------------------- /modules/minecraft_ping/aiodns_resolver.py: -------------------------------------------------------------------------------- 1 | import asyncio 2 | from typing import Literal 3 | 4 | import aiodns 5 | from aiodns.error import DNSError 6 | 7 | resolver = aiodns.DNSResolver(loop=asyncio.get_event_loop(), nameservers=['119.29.29.29']) 8 | 9 | 10 | async def dns_resolver(domain: str) -> bool | str: 11 | try: 12 | result = await resolver.query(domain, 'A') 13 | return result[0].host 14 | except DNSError: 15 | return False 16 | 17 | 18 | async def dns_resolver_srv(domain: str) -> tuple[Literal[False] | str, Literal[False] | int]: 19 | try: 20 | result = await resolver.query(f'_minecraft._tcp.{domain}', 'SRV') 21 | return result[0].host, result[0].port 22 | except DNSError: 23 | return False, False 24 | -------------------------------------------------------------------------------- /modules/minecraft_ping/ping_client.py: -------------------------------------------------------------------------------- 1 | import re 2 | import socket 3 | import struct 4 | import time 5 | 6 | import orjson 7 | from graia.ariadne.util.async_exec import io_bound 8 | 9 | from .aiodns_resolver import dns_resolver_srv 10 | 11 | 12 | class PingClient: 13 | """ 14 | 改自 https://github.com/djkcyl/ABot-Graia/blob/MAH-V2/saya/MinecraftPing/ 15 | 16 | 可能会报如下错误： 17 | 1. ConnectionRefusedError: [WinError 10061] 由于目标计算机积极拒绝，无法连接。 18 | 2. socket.timeout: timed out 19 | 3. socket.gaierror 无效的主机名 20 | """ 21 | 22 | def __init__(self, host: str = 'localhost', port: int = 25565, timeout: int = 5): 23 | self._host = host 24 | self._port = port 25 | self._timeout = timeout 26 | 27 | @staticmethod 28 | def _unpack_varint(sock): 29 | data = 0 30 | for i in range(5): 31 | ordinal = sock.recv(1) 32 | if len(ordinal) == 0: 33 | break 34 | 35 | byte = ord(ordinal) 36 | data |= (byte & 0x7F) << 7 * i 37 | 38 | if not byte & 0x80: 39 | break 40 | 41 | return data 42 | 43 | @staticmethod 44 | def _pack_varint(data): 45 | ordinal = b'' 46 | 47 | while True: 48 | byte = data & 0x7F 49 | data >>= 7 50 | ordinal += struct.pack('B', byte | (0x80 if data > 0 else 0)) 51 | 52 | if data == 0: 53 | break 54 | 55 | return ordinal 56 | 57 | def _pack_data(self, data): 58 | if isinstance(data, str): 59 | data = data.encode('utf8') 60 | return self._pack_varint(len(data)) + data 61 | elif isinstance(data, int): 62 | return struct.pack('H', data) 63 | elif isinstance(data, float): 64 | return struct.pack('Q', int(data)) 65 | else: 66 | return data 67 | 68 | def _send_data(self, sock, *args): 69 | data = b'' 70 | 71 | for arg in args: 72 | data += self._pack_data(arg) 73 | 74 | sock.send(self._pack_varint(len(data)) + data) 75 | 76 | def _read_fully(self, sock, extra_varint=False) -> bytes: 77 | packet_length = self._unpack_varint(sock) 78 | packet_id = self._unpack_varint(sock) 79 | byte = b'' 80 | 81 | if extra_varint: 82 | if packet_id > packet_length: 83 | self._unpack_varint(sock) 84 | 85 | extra_length = self._unpack_varint(sock) 86 | 87 | while len(byte) < extra_length: 88 | byte += sock.recv(extra_length) 89 | 90 | else: 91 | byte = sock.recv(packet_length) 92 | 93 | return byte 94 | 95 | @staticmethod 96 | def _format_desc(data: dict) -> str: # type: ignore 97 | if 'extra' in data: 98 | return ''.join(part['text'] for part in data['extra']) 99 | elif 'text' in data: 100 | return re.sub(r'§[0-9a-gk-r]', '', data['text']) 101 | 102 | @io_bound 103 | def get_ping(self, format_: bool = True) -> dict: 104 | with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as sock: 105 | sock.settimeout(5) 106 | sock.connect((self._host, self._port)) 107 | self._send_data(sock, b'\x00\x00', self._host, self._port, b'\x01') 108 | self._send_data(sock, b'\x00') 109 | data = self._read_fully(sock, extra_varint=True) 110 | 111 | self._send_data(sock, b'\x01', time.time() * 1000) 112 | unix = self._read_fully(sock) 113 | 114 | status = orjson.loads(data.decode('utf8')) 115 | if format_: 116 | status['description'] = self._format_desc(status['description']) 117 | status['delay'] = time.time() * 1000 - struct.unpack('Q', unix)[0] 118 | return status 119 | 120 | 121 | async def ping(ip: str | None = None, url: str | None = None, port: int | None = None) -> dict: 122 | if ip is not None and url is not None: 123 | raise ValueError('Both IP and URL exist') 124 | 125 | if ip is not None: 126 | host = ip 127 | elif url is not None: 128 | host = url 129 | if port is None: # url and not port 130 | host, port = await dns_resolver_srv(url) 131 | if not host: 132 | host = url 133 | port = port or 25565 134 | else: 135 | raise ValueError('Neither IP nor URL exists') 136 | 137 | client = PingClient(host=host, port=port or 25565) 138 | stats: dict = await client.get_ping() 139 | 140 | player_list = stats['players'].get('sample') or [] 141 | return { 142 | 'version': stats['version']['name'], 143 | 'protocol': str(stats['version']['protocol']), 144 | 'motd': stats['description'], 145 | 'delay': str(round(stats['delay'], 1)), 146 | 'online_player': str(stats['players']['online']), 147 | 'max_player': str(stats['players']['max']), 148 | 'player_list': player_list, 149 | } 150 | -------------------------------------------------------------------------------- /modules/minecraft_ping/utils.py: -------------------------------------------------------------------------------- 1 | import re 2 | 3 | 4 | def is_domain(value: str) -> bool: 5 | """ 6 | Return whether given value is a valid domain. 7 | If the value is valid domain name this function returns ``True``, otherwise False 8 | :param value: domain string to validate 9 | """ 10 | pattern = re.compile( 11 | r'^(([a-zA-Z]{1})|([a-zA-Z]{1}[a-zA-Z]{1})|' 12 | r'([a-zA-Z]{1}[0-9]{1})|([0-9]{1}[a-zA-Z]{1})|' 13 | r'([a-zA-Z0-9][-_.a-zA-Z0-9]{0,61}[a-zA-Z0-9]))\.' 14 | r'([a-zA-Z]{2,13}|[a-zA-Z0-9-]{2,30}.[a-zA-Z]{2,3})$' 15 | ) 16 | return bool(pattern.match(value)) 17 | 18 | 19 | def is_ip(host: str) -> bool: 20 | return bool( 21 | re.match( 22 | r'^((2(5[0-5]|[0-4]\d))|[0-1]?\d{1,2})(\.((2(5[0-5]|[0-4]\d))|[0-1]?\d{1,2})){3}$', 23 | host, 24 | ) 25 | ) 26 | -------------------------------------------------------------------------------- /modules/msg2img.py: -------------------------------------------------------------------------------- 1 | from base64 import b64encode 2 | 3 | from graia.amnesia.builtins.aiohttp import AiohttpClientInterface 4 | from graia.ariadne.app import Ariadne 5 | from graia.ariadne.event.message import GroupMessage 6 | from graia.ariadne.message.chain import MessageChain 7 | from graia.ariadne.message.element import At, AtAll, Image, Plain, Source 8 | from graia.ariadne.message.parser.twilight import RegexMatch, Twilight 9 | from graia.ariadne.model import Group, Member 10 | from graia.saya import Channel 11 | from graiax.shortcut.interrupt import FunctionWaiter 12 | from graiax.shortcut.saya import decorate, dispatch, listen 13 | 14 | from libs.control import require_disable 15 | from libs.control.interval import GroupInterval 16 | from libs.control.permission import GroupPermission 17 | from libs.text2img import md2img 18 | 19 | channel = Channel.current() 20 | 21 | channel.meta['name'] = '消息转图片' 22 | channel.meta['author'] = ['Red_lnn'] 23 | channel.meta['description'] = '仿锤子便签样式的消息转图片，支持纯文本与图像\n用法：\n [!！.](文本转图片|消息转图片)' 24 | 25 | 26 | @listen(GroupMessage) 27 | @dispatch(Twilight(RegexMatch(r'[!！.](文本转图片|消息转图片)'))) 28 | @decorate(GroupPermission.require(), GroupInterval.require(15), require_disable(channel.module)) 29 | async def main(app: Ariadne, group: Group, member: Member, source: Source, aiohttp: AiohttpClientInterface): 30 | await app.send_message(group, MessageChain(Plain('请发送要转换的内容')), quote=source) 31 | 32 | async def waiter(waiter_group: Group, waiter_member: Member, waiter_message: MessageChain) -> MessageChain | None: 33 | if waiter_group.id == group.id and waiter_member.id == member.id: 34 | return waiter_message.include(Plain, At, Image) 35 | 36 | answer = await FunctionWaiter(waiter, [GroupMessage]).wait(timeout=10) 37 | if answer is None: 38 | await app.send_message(group, MessageChain(Plain('已超时取消')), quote=source) 39 | return 40 | 41 | if len(answer) == 0: 42 | await app.send_message(group, MessageChain(Plain('你所发送的消息的类型错误')), quote=source) 43 | return 44 | 45 | content = '' 46 | 47 | for ind, elem in enumerate(answer[:]): 48 | if type(elem) in {At, AtAll}: 49 | answer.__root__[ind] = Plain(str(elem)) 50 | for i in answer[:]: 51 | if isinstance(i, Image) and i.url: 52 | async with aiohttp.service.session.get(i.url) as resp: 53 | img = b64encode(await resp.content.read()) 54 | content += f'\n\n