新增关键词回复插件 keyword (#93)

* Update list.json

* Create keyword.py

keyword.py

@@ -0,0 +1,279 @@
+import re, time, uuid
+from base64 import b64encode, b64decode
+from pagermaid import bot, redis, log, redis_status
+from pagermaid.listener import listener
+
+msg_rate = 1
+last_time = time.time()
+
+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(uuid.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 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)
+
+@listener(is_plugin=True, outgoing=True, command="keyword",
+          description="关键词自动回复。",
+          parameters="``new <plain|regex> '<规则>' '<回复信息>'` 或者 `del <plain|regex> '<规则>'` 或者 `list` 或者 `clear <plain|regex>")
+async def reply(context):
+    if not redis_status():
+        await context.edit("出错了呜呜呜 ~ Redis 离线，无法运行。")
+        return
+    chat_id = context.chat_id
+    if chat_id > 0:
+        await context.edit("请在群组中使用")
+        return
+    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", "clear")) or len(parse[0].split()) > 2:
+        await context.edit("[Code: -1] 格式错误，格式为 `-keyword` 加上 `new <plain|regex> '<规则>' '<回复信息>'` 或者 `del <plain|regex> '<规则>'` 或者 `list` 或者 `clear <plain|regex>`")
+        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("格式错误，格式为 `-keyword` 加上 `new <plain|regex> '<规则>' '<回复信息>'` 或者 `del <plain|regex> '<规则>'` 或者 `list` 或者 `clear <plain|regex>`")
+            return
+        await context.edit("设置成功")
+    elif parse[0][0] == "del" and len(parse) == 2:
+        if parse[0][1] == "plain":
+            if parse[1] in plain_dict: 
+                plain_dict.pop(parse[1])
+                redis.set(f"keyword.{chat_id}.plain", save_rules(plain_dict, placeholder))
+            else:
+                await context.edit("规则不存在")
+                return
+        elif parse[0][1] == "regex":
+            if parse[1] in regex_dict: 
+                regex_dict.pop(parse[1])
+                redis.set(f"keyword.{chat_id}.regex", save_rules(regex_dict, placeholder))
+            else:
+                await context.edit("规则不存在")
+                return
+        else:
+            await context.edit("格式错误，格式为 -keyword 加上 new <plain|regex> '<规则>' '<回复信息>' 或者 del <plain|regex> '<规则>' 或者 list 或者 clear <plain|regex>")
+            return
+        await context.edit("删除成功")
+    elif parse[0][0] == "list" and len(parse) == 1:
+        plain_msg = "Plain: \n"
+        for k, v in plain_dict.items():
+            plain_msg += f"`{k}` -> `{v}`\n"
+        regex_msg = "Regex: \n"
+        for k, v in regex_dict.items():
+            regex_msg += f"`{k}` -> `{v}`\n"
+        await context.edit(plain_msg + "\n" + regex_msg)
+    elif parse[0][0] == "clear" and len(parse) == 1:
+        if parse[0][1] == "plain":
+            redis.set(f"keyword.{chat_id}.plain", "")
+        elif parse[0][1] == "regex":
+            redis.set(f"keyword.{chat_id}.regex", "")
+        else:
+            await context.edit("参数错误")
+            return
+        await context.edit("清除成功")
+    else:
+        await context.edit("[Code -2] 格式错误，格式为 `-keyword` 加上 `new <plain|regex> '<规则>' '<回复信息>'` 或者 `del <plain|regex> '<规则>'` 或者 `list` 或者 `clear <plain|regex>`")
+        return
+
+@listener(outgoing=True, command="replyset",
+          description="自动回复设置。",
+          parameters="help")
+async def reply_set(context):
+    if not redis_status():
+        await context.edit("出错了呜呜呜 ~ Redis 离线，无法运行。")
+        return
+    chat_id = context.chat_id
+    if chat_id > 0:
+        await context.edit("请在群组中使用")
+        return
+    params = context.parameter
+    is_global = len(params) >= 1 and params[0] == "global"
+    redis_data = "keyword.settings" if is_global else f"keyword.{chat_id}.settings"
+    if is_global:
+        del params[0]
+    settings_dict = get_redis(redis_data)
+    cmd_list = ["help", "mode", "list", "show", "clear"]
+    cmd_dict = {"help": (1, ), "mode": (2, ), "list": (2, 3), "show": (1, ), "clear": (1, )}
+    if len(params) < 1:
+        await context.edit("参数错误")
+        return
+    if params[0] in cmd_list 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 <add/del/show/clear> [user_id]`。
+在 `-replyset` 后面加上 `global` 即为全局设置''')
+            return
+        elif params[0] == "show":
+            defaults = {"mode": "0", "list": "无"}
+            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("模式已更改为白名单")
+                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("清除成功")
+                return
+            else:
+                await context.edit("参数错误")
+                return
+        elif params[0] == "list":
+            if params[1] == "show" and len(params) == 2:
+                user_list = settings_dict["list"] if "list" in settings_dict else 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("列表为空")
+                    return
+            elif params[1] == "add" and len(params) == 3:
+                if is_num(params[2]):
+                    tmp = settings_dict["list"] if "list" in settings_dict else 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("添加成功")
+                    return
+                else:
+                    await context.edit("user_id 需为整数")
+                    return
+            elif params[1] == "del" and len(params) == 3:
+                if is_num(params[2]):
+                    tmp = settings_dict["list"] if "list" in settings_dict else 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("删除成功")
+                            return
+                        else:
+                            await context.edit("user_id 不在列表")
+                            return
+                    else:
+                        await context.edit("列表为空")
+                        return
+                else:
+                    await context.edit("user_id 需为整数")
+                    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("清除成功")
+                return
+            else:
+                await context.edit("参数错误")
+                return
+        elif params[0] == "clear":
+            redis.delete(redis_data)
+            await context.edit("清除成功")
+            return
+    else:
+        await context.edit("参数错误")
+        return
+    
+
+@listener(incoming=True, ignore_edited=True)
+async def auto_reply(context):
+    global msg_rate, last_time
+    chat_id = context.chat_id
+    sender_id = context.sender_id
+    if chat_id < 0:
+        plain_dict = get_redis(f"keyword.{chat_id}.plain")
+        regex_dict = get_redis(f"keyword.{chat_id}.regex")
+        g_settings = get_redis("keyword.settings")
+        n_settings = get_redis(f"keyword.{chat_id}.settings")
+        g_mode = g_settings["mode"] if "mode" in g_settings else None
+        n_mode = n_settings["mode"] if "mode" in n_settings else None
+        mode = "0"
+        g_list = g_settings["list"] if "list" in g_settings else None
+        n_list = n_settings["list"] if "list" in n_settings else None
+        user_list = []
+        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
+        for k, v in plain_dict.items():
+            if k in context.text and time.time() - last_time > msg_rate:
+                if validate(str(sender_id), int(mode), user_list):
+                    last_time = time.time()
+                    await bot.send_message(chat_id, v)
+        for k, v in regex_dict.items():
+            pattern = re.compile(k)
+            if pattern.search(context.text) and time.time() - last_time > msg_rate:
+                if validate(str(sender_id), int(mode), user_list):
+                    last_time = time.time()
+                    await bot.send_message(chat_id, v)

list.json

@@ -359,6 +359,16 @@
             "supported": true,
             "des-short": "一键跑路（删所有消息并禁言）",
             "des": "paolu。"
+        },
+        {
+            "name": "keyword",
+            "version": "1.00",
+            "section": "chat",
+            "maintainer": "c3b2a",
+            "size": "11.9 kb",
+            "supported": true,
+            "des-short": "群组关键词自动回复插件",
+            "des": "keyword, replyset。"
         }
     ]
 }