Split into many files

reactbot/bot.py

@@ -0,0 +1,54 @@
+# reminder - A maubot plugin that reacts to messages that match predefined rules.
+# Copyright (C) 2019 Tulir Asokan
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with this program.  If not, see <https://www.gnu.org/licenses/>.
+from typing import Type
+
+from mautrix.types import EventType
+from mautrix.util.config import BaseProxyConfig
+
+from maubot import Plugin, MessageEvent
+from maubot.handlers import event
+
+from .config import Config, ConfigError
+
+
+class ReactBot(Plugin):
+    @classmethod
+    def get_config_class(cls) -> Type[BaseProxyConfig]:
+        return Config
+
+    async def start(self) -> None:
+        await super().start()
+        self.on_external_config_update()
+
+    def on_external_config_update(self) -> None:
+        self.config.load_and_update()
+        try:
+            self.config.parse_data()
+        except ConfigError:
+            self.log.exception("Failed to load config")
+
+    @event.on(EventType.ROOM_MESSAGE)
+    async def event_handler(self, evt: MessageEvent) -> None:
+        if evt.sender == self.client.mxid:
+            return
+        for name, rule in self.config.rules.items():
+            match = rule.match(evt)
+            if match is not None:
+                try:
+                    await rule.execute(evt, match)
+                except Exception:
+                    self.log.exception(f"Failed to execute {name}")
+                return