import re, time, asyncio, requests, os, json, random from io import BytesIO from os import path, remove, makedirs, chdir from shutil import copyfile, move, rmtree from uuid import uuid4 from base64 import b64encode, b64decode from importlib import import_module from pagermaid import bot, redis, log, redis_status, working_dir, version from pagermaid import user_id as me_id from pagermaid.listener import listener from pagermaid.utils import alias_command from telethon.errors.rpcerrorlist import StickersetInvalidError from telethon.tl.functions.messages import GetStickerSetRequest from telethon.tl.types import InputStickerSetID, Channel msg_freq = 1 group_last_time = {} read_context = {} def is_num(x: str): try: x = int(x) return isinstance(x, int) except ValueError: return False def encode(s: str): return str(b64encode(s.encode("utf-8")), "utf-8") def decode(s: str): return str(b64decode(s.encode("utf-8")), "utf-8") def random_str(): return str(uuid4()).replace("-", "") def parse_rules(rules: str): n_rules = {} rules_parse = rules.split(";") for p in rules_parse: d = p.split(":") if len(d) == 2: key = decode(d[0]) value = decode(d[1]) n_rules[key] = value return n_rules def save_rules(rules: dict, placeholder: str): n_rules = "" for k, v in rules.items(): if placeholder: k = k.replace(placeholder, "'") v = v.replace(placeholder, "'") n_rules += encode(k) + ":" + encode(v) + ";" return n_rules def validate(user_id: str, mode: int, user_list: list): if mode == 0: return user_id not in user_list elif mode == 1: return user_id in user_list else: return False def validsent(trig: int, tmp): if tmp: return int(tmp.get("trig", "0")) else: return trig def get_redis(db_key: str): byte_data = redis.get(db_key) byte_data = byte_data if byte_data else b"" byte_data = str(byte_data, "ascii") return parse_rules(byte_data) def parse_multi(rule: str): sep_ph = random_str() col_ph = random_str() rule = rule.replace(r"\||", sep_ph) rule = rule.replace(r"\::", col_ph) rule = rule.split("||") n_rule = [] for r in rule: p = r.split("::") p = [i.replace(sep_ph, "||") for i in p] p = [i.replace(col_ph, "::") for i in p] data = ["plain", ""] if len(p) == 2: data = p else: data[1] = p[0] n_rule.append(data) return n_rule def get_capture(search_data, group_name: str): try: capture_data = search_data.group(group_name) return capture_data except: return None def get_rule(chat_id, rule_type, rule_index): rule_index = int(rule_index) rule_data = get_redis(f"keyword.{chat_id}.{rule_type}") index = 0 for k in rule_data.keys(): if index == rule_index: return encode(k) index += 1 return None def valid_time(chat_id): global msg_freq, group_last_time cus_freq = get_redis(f"keyword.{chat_id}.settings").get("freq", msg_freq) try: cus_freq = float(cus_freq) except: cus_freq = msg_freq n_time = time.time() chat_id = int(chat_id) if chat_id in group_last_time: if n_time - group_last_time[chat_id] >= cus_freq: return True else: return False else: return True def has_cache(chat_id, mode, trigger, filename): basepath = f"data/keyword_cache/{chat_id}/{mode}:{encode(trigger)}" filepath = f"{basepath}/{filename}" if not path.exists(basepath): makedirs(basepath) return (False, filepath) if not path.exists(filepath): return (False, filepath) return (True, filepath) def cache_opened(chat_id, mode, trigger): rule_data = get_redis(f"keyword.{chat_id}.single" f".{mode}.{encode(trigger)}").get("cache", None) chat_data = get_redis(f"keyword.{chat_id}.settings").get("cache", None) global_data = get_redis("keyword.settings").get("cache", None) if rule_data: return True if rule_data == "1" else False elif chat_data: return True if chat_data == "1" else False elif global_data: return True if global_data == "1" else False return False def getsetting(chat_id, mode, trigger, name, default): g_settings = get_redis("keyword.settings") n_settings = get_redis(f"keyword.{chat_id}.settings") s_settings = get_redis(f"keyword.{chat_id}.single.{mode}.{encode(trigger)}") final = default if s_settings.get(name, None): final = s_settings[name] elif n_settings.get(name, None): final = n_settings[name] elif g_settings.get(name, None): final = g_settings[name] return final async def aexec(code, *args, **kwargs): exec( f"async def func(*args, **kwargs):" + "".join(f"\n {p}" for p in code.split("\n")) ) return await locals()["func"](*args, **kwargs) async def del_msg(context, t_lim): await asyncio.sleep(t_lim) try: await context.delete() except: pass async def send_reply(chat_id, trigger, mode, reply_msg, context): try: # 拷贝一份数据 real_chat_id = chat_id chat = context.chat sender = context.sender sender_channel = True if isinstance(sender, Channel) else False replace_data = {} if chat_id < 0: replace_data = { "chat_id": chat.id, "chat_name": chat.title } if sender and not sender_channel: replace_data["user_id"] = sender.id replace_data["first_name"] = sender.first_name if sender.first_name else "" replace_data["last_name"] = sender.last_name if sender.last_name else "" if sender and sender_channel: replace_data["user_id"] = sender.id replace_data["first_name"] = sender.title else: replace_data["user_id"] = chat_id if sender and not sender_channel: replace_data["first_name"] = sender.first_name if sender.first_name else "" replace_data["last_name"] = sender.last_name if sender.last_name else "" if chat: replace_data["chat_id"] = chat.id last_name = chat.last_name if not last_name: last_name = "" replace_data["chat_name"] = f"{chat.first_name} {last_name}" # 校验能否发送消息 could_send_msg = valid_time(chat_id) if not could_send_msg: return message_list = [] for re_type, re_msg in reply_msg: try: update_last_time = False catch_pattern = r"\$\{func_(?P((?!\}).)+)\}" count = 0 bracket_str = random_str() re_msg = re_msg.replace(r"\}", bracket_str) # 函数功能相关 while re.search(catch_pattern, re_msg) and count < 20: func_exec = re.search(catch_pattern, re_msg).group("str") try: func_name = func_exec func_args = None if func_exec.strip().endswith(")"): arg_index = func_exec.find("(") func_name = func_exec[0:arg_index].replace(bracket_str, "}") func_args = func_exec[arg_index + 1:-1].replace(bracket_str, "}") module = f"import_module('data.keyword_func.{func_name}').main" parameter = f"context{', %s' % func_args if func_args else ''}" func_data = await eval(f"{module}({parameter})") except: func_data = "[RE]" chdir(working_dir) re_msg = re_msg.replace("${func_%s}" % func_exec, str(func_data)) count += 1 re_msg = re_msg.replace(bracket_str, "}") for k, v in replace_data.items(): re_type = re_type.replace(f"${k}", str(v)) re_msg = re_msg.replace(f"${k}", str(v)) type_parse = re_type.split(",") edit_id = -1 for s in type_parse: if len(s) >= 5 and "ext_" == s[0:4] and is_num(s[4:]): chat_id = int(s[4:]) type_parse.remove(s) elif len(s) >= 6 and "edit_" == s[0:5] and is_num(s[5:]): edit_id = int(s[5:]) type_parse.remove(s) # 处理file和photo if ("file" in type_parse or "photo" in type_parse or "sticker" in type_parse) and len(re_msg.split()) >= 2: update_last_time = True re_data = re_msg.split(" ") cache_exists, filename = has_cache(chat_id, mode, trigger, re_data[0]) is_opened = cache_opened(chat_id, mode, trigger) if "sticker" not in type_parse: if is_opened: # 开启缓存，处理缓存 if not cache_exists: if re_data[1][0:7] == "file://": re_data[1] = re_data[1][7:] copyfile(" ".join(re_data[1:]), filename) else: fileget = requests.get(" ".join(re_data[1:])) with open(filename, "wb") as f: f.write(fileget.content) else: # 未开启缓存 if re_data[1][0:7] == "file://": re_data[1] = re_data[1][7:] copyfile(" ".join(re_data[1:]), filename) else: fileget = requests.get(" ".join(re_data[1:])) with open(filename, "wb") as f: f.write(fileget.content) reply_to = None if "reply" in type_parse: reply_to = context.id redir = getsetting(chat_id, mode, trigger, "redir", "0") reply = await context.get_reply_message() if redir == "1" and reply: reply_to = reply.id if "sticker" in type_parse: # 单独处理回复贴纸 sticker_params = re_data[0].split(",") sticker_idx = re_data[1].split(",") try: stickers = await context.client( GetStickerSetRequest( stickerset=InputStickerSetID( id=int(sticker_params[0]), access_hash=int(sticker_params[1])))) except StickersetInvalidError: await log('贴纸配置错误。') return try: i = random.randint(0, len(sticker_idx) - 1) message_list.append(await bot.send_file( chat_id, stickers.documents[int(sticker_idx[i])], reply_to=reply_to )) except: pass else: if edit_id == -1: message_list.append(await bot.send_file( chat_id, filename, reply_to=reply_to, force_document=("file" in type_parse) )) else: edit_file = await bot.upload_file(filename) message_list[edit_id] = await message_list[edit_id].edit( file=edit_file, force_document=("file" in type_parse) ) # 未开启缓存，删除文件 if not is_opened: remove(filename) # 处理tgfile和tgphoto elif ("tgfile" in type_parse or "tgphoto" in type_parse) and len(re_msg.split()) >= 2: update_last_time = True re_data = re_msg.split(" ") re_data[0] = " ".join(re_data[0:-1]) re_data[1] = re_data[-1:][0].split("/")[-2:] cache_exists, filename = has_cache(chat_id, mode, trigger, re_data[0]) is_opened = cache_opened(chat_id, mode, trigger) _data = BytesIO() try: msg_chat_id = int(re_data[1][0]) except: async with bot.conversation(re_data[1][0]) as conversation: msg_chat_id = conversation.chat_id msg_id_inchat = int(re_data[1][1]) if is_opened: if not cache_exists: media_msg = await bot.get_messages(msg_chat_id, ids=msg_id_inchat, offset_id=0) if media_msg and media_msg.media: try: await bot.download_file(media_msg.media.document, _data) except: await bot.download_file(media_msg.photo, _data) with open(filename, "wb") as f: f.write(_data.getvalue()) else: media_msg = await bot.get_messages(msg_chat_id, ids=msg_id_inchat, offset_id=0) if media_msg and media_msg.media: try: await bot.download_file(media_msg.media.document, _data) except: await bot.download_file(media_msg.photo, _data) with open(filename, "wb") as f: f.write(_data.getvalue()) reply_to = None if "reply" in type_parse: reply_to = context.id redir = getsetting(chat_id, mode, trigger, "redir", "0") reply = await context.get_reply_message() if redir == "1" and reply: reply_to = reply.id if edit_id == -1: message_list.append(await bot.send_file( chat_id, filename, reply_to=reply_to, force_document=("tgfile" in type_parse) )) else: edit_file = await bot.upload_file(filename) message_list[edit_id] = await message_list[edit_id].edit( file=edit_file, force_document=("tgfile" in type_parse) ) if not is_opened: remove(filename) elif "plain" in type_parse: update_last_time = True if edit_id == -1: message_list.append(await bot.send_message( chat_id, re_msg, link_preview=("nopreview" not in type_parse) )) else: message_list[edit_id] = await message_list[edit_id].edit( re_msg, link_preview=("nopreview" not in type_parse) ) elif "reply" in type_parse and chat_id == real_chat_id: update_last_time = True if edit_id == -1: reply_to = context.id redir = getsetting(chat_id, mode, trigger, "redir", "0") reply = await context.get_reply_message() if redir == "1" and reply: reply_to = reply.id message_list.append(await bot.send_message( chat_id, re_msg, reply_to=reply_to, link_preview=("nopreview" not in type_parse) )) else: message_list[edit_id] = await message_list[edit_id].edit( re_msg, link_preview=("nopreview" not in type_parse) ) elif "op" in type_parse: if re_msg == "delete": await context.delete() elif re_msg.split()[0] == "sleep" and len(re_msg.split()) == 2: sleep_time = re_msg.split()[1] await asyncio.sleep(float(sleep_time)) elif re_msg.split()[0] == "delself" and len(re_msg.split()) == 2: await message_list[int(re_msg.split()[1])].delete() elif re_msg.split()[0] == "trigger" and len(re_msg.split()) == 2: await auto_reply(message_list[int(re_msg.split()[1])]) elif re_msg.split("\n")[0].startswith("exec") and len(re_msg.split("\n")) >= 2: args = [ "\n".join(re_msg.split("\n")[1:]), " ".join(re_msg.split("\n")[0].split(" ")[1:]) ] await eval(f"aexec(args[0]{f', {args[1]}' if args[1] else ''})") chdir(working_dir) if update_last_time: global group_last_time group_last_time[int(chat_id)] = time.time() except Exception as e: pass chat_id = real_chat_id except Exception as e: pass @listener(is_plugin=True, outgoing=True, command=alias_command("keyword"), description="关键词自动回复 [教程](https://telegra.ph/Keyword-插件使用教程-02-07)", parameters="``new '<规则>' '<回复信息>'` 或者 `del '<规则>'` 或者 `list` 或者 " "`clear ") async def reply(context): if not redis_status(): await context.edit("出错了呜呜呜 ~ Redis 离线，无法运行") await del_msg(context, 5) return chat_id = context.chat_id plain_dict = get_redis(f"keyword.{chat_id}.plain") regex_dict = get_redis(f"keyword.{chat_id}.regex") params = context.parameter params = " ".join(params) placeholder = random_str() params = params.replace(r"\'", placeholder) tmp_parse = params.split("'") parse = [] for i in range(len(tmp_parse)): if len(tmp_parse[i].split()) != 0: parse.append(tmp_parse[i]) if len(parse) == 0 or ( len(parse[0].split()) == 1 and parse[0].split()[0] in ("new", "del", "delid", "clear")) or len( parse[0].split()) > 2: await context.edit( "[Code: -1] 格式错误，格式为 `-keyword` 加上 `new '<规则>' '<回复信息>'` 或者 " "`del '<规则>'` 或者 `list` 或者 `clear `") await del_msg(context, 10) return else: parse[0] = parse[0].split() if parse[0][0] == "new" and len(parse) == 3: if parse[0][1] == "plain": plain_dict[parse[1]] = parse[2] redis.set(f"keyword.{chat_id}.plain", save_rules(plain_dict, placeholder)) elif parse[0][1] == "regex": regex_dict[parse[1]] = parse[2] redis.set(f"keyword.{chat_id}.regex", save_rules(regex_dict, placeholder)) else: await context.edit( "[Code: -1] 格式错误，格式为 `-keyword` 加上 `new '<规则>' '<回复信息>'` 或者 " "`del '<规则>'` 或者 `list` 或者 `clear `") await del_msg(context, 10) return await context.edit("设置成功") await del_msg(context, 5) elif parse[0][0] in ("del", "delid") and len(parse) == 2: if parse[0][0] == "delid": parse[1] = get_rule(chat_id, parse[0][1], parse[1]) if parse[1]: parse[1] = decode(parse[1]) if parse[0][1] == "plain": if parse[1] and parse[1] in plain_dict: redis.delete(f"keyword.{chat_id}.single.plain.{encode(parse[1])}") plain_dict.pop(parse[1]) redis.set(f"keyword.{chat_id}.plain", save_rules(plain_dict, placeholder)) else: await context.edit("规则不存在") await del_msg(context, 5) return elif parse[0][1] == "regex": if parse[1] and parse[1] in regex_dict: redis.delete(f"keyword.{chat_id}.single.regex.{encode(parse[1])}") regex_dict.pop(parse[1]) redis.set(f"keyword.{chat_id}.regex", save_rules(regex_dict, placeholder)) else: await context.edit("规则不存在") await del_msg(context, 5) return else: await context.edit( "[Code: -1] 格式错误，格式为 `-keyword` 加上 `new '<规则>' '<回复信息>'` 或者 " "`del '<规则>'` 或者 `list` 或者 `clear `") await del_msg(context, 10) return await context.edit("删除成功") await del_msg(context, 5) elif parse[0][0] == "list" and len(parse) == 1: plain_msg = "Plain: \n" index = 0 for k, v in plain_dict.items(): plain_msg += f"`{index}`: `{k}` -> `{v}`\n" index += 1 regex_msg = "Regex: \n" index = 0 for k, v in regex_dict.items(): regex_msg += f"`{index}`: `{k}` -> `{v}`\n" index += 1 await context.edit(plain_msg + "\n" + regex_msg) elif parse[0][0] == "clear" and len(parse) == 1: if parse[0][1] == "plain": for k in plain_dict.keys(): redis.delete(f"keyword.{chat_id}.single.plain.{encode(k)}") redis.set(f"keyword.{chat_id}.plain", "") elif parse[0][1] == "regex": for k in regex_dict.keys(): redis.delete(f"keyword.{chat_id}.single.regex.{encode(k)}") redis.set(f"keyword.{chat_id}.regex", "") else: await context.edit("参数错误") await del_msg(context, 5) return await context.edit("清除成功") await del_msg(context, 5) else: await context.edit( "[Code: -1] 格式错误，格式为 `-keyword` 加上 `new '<规则>' '<回复信息>'` 或者 " "`del '<规则>'` 或者 `list` 或者 `clear `") await del_msg(context, 10) return @listener(outgoing=True, command=alias_command("replyset"), description="自动回复设置", parameters="help") async def reply_set(context): if not redis_status(): await context.edit("出错了呜呜呜 ~ Redis 离线，无法运行") await del_msg(context, 5) return chat_id = context.chat_id params = context.parameter redis_data = f"keyword.{chat_id}.settings" if len(params) >= 1 and params[0] == "global": redis_data = "keyword.settings" del params[0] elif len(params) >= 2 and params[0] in ("plain", "regex") and is_num(params[1]): rule_data = get_rule(chat_id, params[0], params[1]) if rule_data: redis_data = f"keyword.{chat_id}.single.{params[0]}.{rule_data}" del params[0:2] settings_dict = get_redis(redis_data) cmd_dict = { "help": (1,), "mode": (2,), "list": (2, 3), "freq": (2,), "trig": (2,), "show": (1,), "cache": (2,), "redir": (2,), "status": (2,), "clear": (1,) } if len(params) < 1: await context.edit("参数错误") await del_msg(context, 5) return if params[0] in cmd_dict and len(params) in cmd_dict[params[0]]: if params[0] == "help": await context.edit(''' `-replyset show` 或 `-replyset clear` 或 `-replyset mode <0/1/clear>` ( 0 表示黑名单，1 表示白名单 ) 或 `-replyset list [user_id]` 或 `-replyset freq ` ( float 表示一个正的浮点数，clear 为清除 ) 或 `-replyset trig <0/1/clear>` ( 0 为关闭，1 为开启，clear 为清除 ) 或 `-replyset cache <0/1/clear>` ( 0 为关闭，1 为开启 ) 或 `-replyset status <0/1/clear>` ( 0 为关闭，1 为开启 ) 。 在 `-replyset` 后面加上 `global` 即为全局设置。 在 `-replyset` 后面加上 `plain/regex` 规则序号 可以单独设置一条规则。''') await del_msg(context, 15) return elif params[0] == "show": defaults = { "mode": "未设置 (默认黑名单)", "list": "未设置 (默认为空)", "freq": "未设置 (默认为 1)", "trig": "未设置 (默认关闭)", "cache": "未设置 (默认关闭)", "redir": "未设置 (默认关闭)", "status": "未设置 (默认开启)" } msg = "Settings: \n" for k, v in defaults.items(): msg += f"`{k}` -> `{settings_dict[k] if k in settings_dict else v}`\n" await context.edit(msg) return elif params[0] == "mode": if params[1] in ("0", "1"): settings_dict["mode"] = params[1] redis.set(redis_data, save_rules(settings_dict, None)) if params[1] == "0": await context.edit("模式已更改为黑名单") elif params[1] == "1": await context.edit("模式已更改为白名单") await del_msg(context, 5) return elif params[1] == "clear": if "mode" in settings_dict: del settings_dict["mode"] redis.set(redis_data, save_rules(settings_dict, None)) await context.edit("清除成功") await del_msg(context, 5) return else: await context.edit("参数错误") await del_msg(context, 5) return elif params[0] == "list": if params[1] == "show" and len(params) == 2: user_list = settings_dict.get("list", None) if user_list: msg = "List: \n" for p in user_list.split(","): msg += f"`{p}`\n" await context.edit(msg) return else: await context.edit("列表为空") await del_msg(context, 5) return elif params[1] == "add" and len(params) == 3: if is_num(params[2]): tmp = settings_dict.get("list", None) if not tmp: settings_dict["list"] = params[2] else: settings_dict["list"] += f",{params[2]}" redis.set(redis_data, save_rules(settings_dict, None)) await context.edit("添加成功") await del_msg(context, 5) return else: await context.edit("user_id 需为整数") await del_msg(context, 5) return elif params[1] == "del" and len(params) == 3: if is_num(params[2]): tmp = settings_dict.get("list", None) if tmp: user_list = settings_dict["list"].split(",") if params[2] in user_list: user_list.remove(params[2]) settings_dict["list"] = ",".join(user_list) redis.set(redis_data, save_rules(settings_dict, None)) await context.edit("删除成功") await del_msg(context, 5) return else: await context.edit("user_id 不在列表") await del_msg(context, 5) return else: await context.edit("列表为空") await del_msg(context, 5) return else: await context.edit("user_id 需为整数") await del_msg(context, 5) return elif params[1] == "clear" and len(params) == 2: if "list" in settings_dict: del settings_dict["list"] redis.set(redis_data, save_rules(settings_dict, None)) await context.edit("清除成功") await del_msg(context, 5) return else: await context.edit("参数错误") await del_msg(context, 5) return elif params[0] == "freq": if redis_data == f"keyword.{chat_id}.settings": if params[1] == "clear": if "freq" in settings_dict: del settings_dict["freq"] redis.set(redis_data, save_rules(settings_dict, None)) await context.edit("清除成功") await del_msg(context, 5) return else: try: tmp = float(params[1]) if tmp > 0: settings_dict["freq"] = params[1] redis.set(redis_data, save_rules(settings_dict, None)) await context.edit("设置成功") await del_msg(context, 5) return else: await context.edit("频率需为正数") await del_msg(context, 5) return except: await context.edit("频率需为正数") await del_msg(context, 5) return else: await context.edit("此项无法使用全局设置和单独设置") return elif params[0] == "trig": if params[1] == "0": settings_dict["trig"] = "0" redis.set(redis_data, save_rules(settings_dict, None)) await context.edit("已关闭自我触发") await del_msg(context, 5) return elif params[1] == "1": settings_dict["trig"] = "1" redis.set(redis_data, save_rules(settings_dict, None)) await context.edit("已开启自我触发") await del_msg(context, 5) return elif params[1] == "clear": if "trig" in settings_dict: del settings_dict["trig"] redis.set(redis_data, save_rules(settings_dict, None)) await context.edit("已清除自我触发设置") await del_msg(context, 5) return else: await context.edit("参数错误") await del_msg(context, 5) return elif params[0] == "cache": if params[1] == "0": settings_dict["cache"] = "0" redis.set(redis_data, save_rules(settings_dict, None)) await context.edit("已关闭缓存功能") await del_msg(context, 5) return elif params[1] == "1": settings_dict["cache"] = "1" redis.set(redis_data, save_rules(settings_dict, None)) await context.edit("已开启缓存功能") await del_msg(context, 5) return elif params[1] == "remove": if redis_data == "keyword.settings": rmtree("data/keyword_cache") elif redis_data.split(".")[2] == "single": rmtree(f"data/keyword_cache/{chat_id}/" f"{redis_data.split('.')[3]}:{redis_data.split('.')[4]}") else: rmtree(f"data/keyword_cache/{chat_id}") await context.edit("已删除缓存") await del_msg(context, 5) return elif params[1] == "clear": if "cache" in settings_dict: del settings_dict["cache"] redis.set(redis_data, save_rules(settings_dict, None)) await context.edit("清除成功") await del_msg(context, 5) return else: await context.edit("参数错误") await del_msg(context, 5) return elif params[0] == "redir": if params[1] == "0": settings_dict["redir"] = "0" redis.set(redis_data, save_rules(settings_dict, None)) await context.edit("已关闭回复穿透") await del_msg(context, 5) return elif params[1] == "1": settings_dict["redir"] = "1" redis.set(redis_data, save_rules(settings_dict, None)) await context.edit("已开启回复穿透") await del_msg(context, 5) return elif params[1] == "clear": if "redir" in settings_dict: del settings_dict["redir"] redis.set(redis_data, save_rules(settings_dict, None)) await context.edit("已清除回复穿透设置") await del_msg(context, 5) return else: await context.edit("参数错误") await del_msg(context, 5) return elif params[0] == "status": if redis_data == f"keyword.{chat_id}.settings": if params[1] == "0": settings_dict["status"] = "0" redis.set(redis_data, save_rules(settings_dict, None)) await context.edit("已关闭此聊天的关键词回复") await del_msg(context, 5) return elif params[1] == "1": settings_dict["status"] = "1" redis.set(redis_data, save_rules(settings_dict, None)) await context.edit("已开启此聊天的关键词回复") await del_msg(context, 5) return elif params[1] == "clear": if "status" in settings_dict: del settings_dict["status"] redis.set(redis_data, save_rules(settings_dict, None)) await context.edit("已清除此设置") await del_msg(context, 5) return else: await context.edit("参数错误") await del_msg(context, 5) return else: await context.edit("此项无法使用全局设置和单独设置") return elif params[0] == "clear": redis.delete(redis_data) await context.edit("清除成功") await del_msg(context, 5) return else: await context.edit("参数错误") await del_msg(context, 5) return @listener(outgoing=True, command=alias_command("funcset"), description="设置自定义函数", parameters="help") async def funcset(context): if not path.exists("data/keyword_func"): makedirs("data/keyword_func") try: params = context.parameter params = " ".join(params).split("\n") cmd = [] if len(params) >= 1: cmd = params[0].split() if len(cmd) > 0: if len(cmd) == 1 and cmd[0] == "ls": send_msg = "Functions:\n" count = 1 for p in os.listdir("data/keyword_func"): if path.isfile(f"data/keyword_func/{p}"): try: send_msg += f"{count}: `{p[:-3]}`\n" count += 1 except: pass await context.edit(send_msg) return elif len(cmd) == 2 and cmd[0] == "show": file_path = f"data/keyword_func/{cmd[1]}.py" if path.exists(file_path) and path.isfile(file_path): await bot.send_file(context.chat_id, file_path) await context.edit("发送成功") await del_msg(context, 5) else: await context.edit("函数不存在") await del_msg(context, 5) return elif len(cmd) == 2 and cmd[0] == "del": file_path = f"data/keyword_func/{cmd[1]}.py" if path.exists(file_path) and path.isfile(file_path): remove(file_path) await context.edit("删除成功，PagerMaid-Modify 正在重新启动。") await bot.disconnect() else: await context.edit("函数不存在") await del_msg(context, 5) return elif len(cmd) == 2 and cmd[0] == "new": message = await context.get_reply_message() if context.media: message = context cmd[1] = cmd[1].replace(".py", "") if message and message.media: try: data = BytesIO() await bot.download_file(message.media.document, data) with open(f"data/keyword_func/{cmd[1]}.py", "wb") as f: f.write(data.getvalue()) await context.edit(f"函数 {cmd[1]} 已添加，PagerMaid-Modify 正在重新启动。") await bot.disconnect() except: await context.edit("函数添加失败") await del_msg(context, 5) else: await context.edit("未回复消息或回复的消息中不包含文件") await del_msg(context, 5) return elif len(cmd) == 2 and cmd[0] == "install": func_name = cmd[1] func_online = \ json.loads( requests.get("https://gitlab.com/Xtao-Labs/PagerMaid_Plugins/-/raw/master" "/keyword_func/list.json").content)["list"] if func_name in func_online: func_directory = f"data/keyword_func/" file_path = func_name + ".py" func_content = requests.get( f"https://gitlab.com/Xtao-Labs/PagerMaid_Plugins/-/raw/master" f"/keyword_func/{func_name}.py").content with open(file_path, "wb") as f: f.write(func_content) if path.exists(f"{func_directory}{file_path}"): remove(f"{func_directory}{file_path}") move(file_path, func_directory) else: move(file_path, func_directory) await context.edit(f"函数 {path.basename(file_path)[:-3]} 已添加，PagerMaid-Modify 正在重新启动。") await log(f"成功安装函数 {path.basename(file_path)[:-3]}.") await bot.disconnect() else: await context.edit(f"{func_name} 函数不存在") await del_msg(context, 5) return elif len(cmd) == 1 and cmd[0] == "help": await context.edit(""" `-funcset new ` (要回复带有文件的信息或自己附带文件) `-funcset install ` （云端获取函数文件） `-funcset del ` `-funcset show ` (发送文件) `-funcset ls` (列出所有函数)""") else: await context.edit("参数错误") await del_msg(context, 5) return else: await context.edit("参数错误") await del_msg(context, 5) return except: pass @listener(outgoing=True, command=alias_command("keydata"), description="设置规则数据", parameters="dump / load") async def setdata(context): try: chat_id = context.chat_id params = context.parameter if params[0] == "dump": data = redis.get(f"keyword.{chat_id}.{params[1]}") if not data: await context.edit("无规则数据") await del_msg(context, 5) return data = str(data, "ascii") await context.edit(data) return elif params[0] == "load": redis.set(f"keyword.{chat_id}.{params[1]}", params[2]) await context.edit("设置成功") await del_msg(context, 5) return else: await context.edit("参数错误") await del_msg(context, 5) return except: await context.edit("运行错误") await del_msg(context, 5) return @listener(incoming=True, outgoing=True, ignore_edited=True) async def auto_reply(context): global read_context await asyncio.sleep(random.randint(0, 100) / 1000) # 解决重复回复，不稳定修复 if (context.chat_id, context.id) in read_context: return read_context[(context.chat_id, context.id)] = True # 判断redis状态 if not redis_status(): return try: chat_id = context.chat_id sender_id = context.sender_id # 读取keyword当前会话的设置 n_settings = get_redis(f"keyword.{chat_id}.settings") if n_settings.get("status", "1") == "0": return # 设置本人id self_id = me_id # 获取全局设置 g_settings = get_redis("keyword.settings") # 获取当前会话的2种回复配置 plain_dict = get_redis(f"keyword.{chat_id}.plain") regex_dict = get_redis(f"keyword.{chat_id}.regex") # 获取全局和当前会话设置的mode g_mode = g_settings.get("mode", None) n_mode = n_settings.get("mode", None) mode = "0" # 获取全局和当前会话设置的list g_list = g_settings.get("list", None) n_list = n_settings.get("list", None) user_list = [] # 获取全局和当前会话设置的trig g_trig = g_settings.get("trig", None) n_trig = n_settings.get("trig", None) trig = "0" # 初始化各种设置 if g_mode and n_mode: mode = n_mode elif g_mode or n_mode: mode = g_mode if g_mode else n_mode if g_list and n_list: user_list = n_list elif g_list or n_list: user_list = g_list if g_list else n_list if g_trig and n_trig: trig = n_trig elif g_trig or n_trig: trig = g_trig if g_trig else n_trig send_text = context.text if not send_text: send_text = "" # 是否自我触发 self_sent = (self_id == sender_id) # 处理文本配置 for k, v in plain_dict.items(): # 接收到的文本符合配置 if k in send_text: # 获取要回复的内容配置 tmp = get_redis(f"keyword.{chat_id}.single.plain.{encode(k)}") # 判断是否要回复 could_reply = validate(str(sender_id), int(mode), user_list) if tmp: could_reply = validate(str(sender_id), int(tmp.get("mode", "0")), tmp.get("list", [])) if could_reply and (not self_sent or validsent(int(trig), tmp)): read_context[(chat_id, context.id)] = None # 发送回复 await send_reply(chat_id, k, "plain", parse_multi(v), context) # 处理正则配置 for k, v in regex_dict.items(): pattern = re.compile(k) if pattern.search(send_text): tmp = get_redis(f"keyword.{chat_id}.single.regex.{encode(k)}") could_reply = validate(str(sender_id), int(mode), user_list) if tmp: could_reply = validate(str(sender_id), int(tmp.get("mode", "0")), tmp.get("list", [])) if could_reply and (not self_sent or validsent(int(trig), tmp)): read_context[(chat_id, context.id)] = None catch_pattern = r"\$\{regex_(?P((?!\}).)+)\}" count = 0 while re.search(catch_pattern, v) and count < 20: search_data = re.search(k, send_text) group_name = re.search(catch_pattern, v).group("str") capture_data = get_capture(search_data, group_name) if not capture_data: capture_data = "" if re.search(catch_pattern, capture_data): capture_data = "" v = v.replace("${regex_%s}" % group_name, capture_data) count += 1 await send_reply(chat_id, k, "regex", parse_multi(v), context) except: pass