From 3c12e0e8d51699abeebb822414cf1ef3cc126f26 Mon Sep 17 00:00:00 2001
From: 简律纯 <hsiangnianian@outlook.com>
Date: Sun, 23 Apr 2023 23:36:31 +0800
Subject: ✨fix
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 tests/plugins/iamai_plugin_reply/__init__.py | 43 ++++++++++++++++++++++++++++
 1 file changed, 43 insertions(+)
 create mode 100644 tests/plugins/iamai_plugin_reply/__init__.py

(limited to 'tests/plugins/iamai_plugin_reply/__init__.py')

diff --git a/tests/plugins/iamai_plugin_reply/__init__.py b/tests/plugins/iamai_plugin_reply/__init__.py
new file mode 100644
index 0000000..6835ca8
--- /dev/null
+++ b/tests/plugins/iamai_plugin_reply/__init__.py
@@ -0,0 +1,43 @@
+import re
+import json
+
+from plugins.iamai_plugin_base import BasePlugin
+
+from .config import Config
+
+
+class Reply(BasePlugin[None, Config]):
+    priority: int = 1
+    Config = Config
+
+    def __post_init__(self):
+        with open(self.config.data_file, "r",encoding="utf-8") as fp:
+            if self.config.data_type == "json":
+                json_data = json.load(fp)
+            else:
+                raise ValueError(f"data_type must be json, not {self.config.data_type}")
+        self.rule_to_message = {
+            item["rule"]: item["message"]
+            for item in json_data
+            if isinstance(item, dict)
+            and "rule" in item.keys()
+            and "message" in item.keys()
+        }
+
+    async def handle(self) -> None:
+        msg = self.rule_to_message[self.msg_match.re.pattern]
+        if isinstance(msg, str):
+            await self.event.reply(self.format_str(msg, self.msg_match.string))
+        else:
+            await self.event.reply(msg)
+
+    def str_match(self, msg_str: str) -> bool:
+        msg_str = msg_str.strip()
+        for rule in self.rule_to_message.keys():
+            msg_match = re.fullmatch(
+                rule, msg_str, flags=re.I if self.config.ignore_case else 0
+            )
+            if msg_match:
+                self.msg_match = msg_match
+                return bool(self.msg_match)
+        return False
-- 
cgit v1.2.3-70-g09d2