From 562d85153ef03f1e2208cf40696b0c648ad88f52 Mon Sep 17 00:00:00 2001
From: Dr Serge Victor <ser@users.noreply.github.com>
Date: Tue, 3 Oct 2023 11:34:48 +0000
Subject: [PATCH] - added config to limit rooms where bot works.

---
 base-config.yaml |  2 ++
 hateheif.py      | 23 +++++++++++++++++++++++
 maubot.yaml      |  5 ++++-
 3 files changed, 29 insertions(+), 1 deletion(-)
 create mode 100644 base-config.yaml

diff --git a/base-config.yaml b/base-config.yaml
new file mode 100644
index 0000000..a61fd8d
--- /dev/null
+++ b/base-config.yaml
@@ -0,0 +1,2 @@
+# If not empty, bot will serve only rooms on the list
+rooms: []
diff --git a/hateheif.py b/hateheif.py
index 2253360..336204f 100644
--- a/hateheif.py
+++ b/hateheif.py
@@ -3,12 +3,19 @@ from io import BytesIO
 from typing import Tuple
 from PIL import Image
 from pillow_heif import HeifImagePlugin
+from typing import Type
 
 from maubot import Plugin, MessageEvent
 from maubot.handlers import command
 from mautrix.client import Client as MatrixClient
 from mautrix.crypto import attachments
 from mautrix.types import EncryptedFile, ImageInfo, MediaMessageEventContent, MessageType
+from mautrix.util.config import BaseProxyConfig, ConfigUpdateHelper
+
+class Config(BaseProxyConfig):
+    def do_update(self, helper: ConfigUpdateHelper) -> None:
+        helper.copy("rooms")
+
 
 
 async def download_encrypted_media(
@@ -108,6 +115,13 @@ async def send_unencrypted_message(
 
 # BOT
 class HateHeifBot(Plugin):
+
+    async def start(self) -> None:
+        await super().start()
+        self.config.load_and_update()
+        self.rooms = self.config['rooms'] if self.config['rooms'] else None
+
+
     @command.passive("", msgtypes=(MessageType.IMAGE,))
     async def hate_heif_message(
             self,
@@ -116,6 +130,12 @@ class HateHeifBot(Plugin):
         """
         If heif = make it jpg.
         """
+        # If there are rooms specified in config, serve only these rooms
+        if self.rooms:
+            if evt.room_id not in self.rooms:
+                self.log.debug(f"Current room {evt.room_id} is not listed in {self.rooms} so I ignore it.")
+                return
+
         # Double check if it is an image message
         if evt.content.msgtype != MessageType.IMAGE:
             return
@@ -161,5 +181,8 @@ class HateHeifBot(Plugin):
                     content.info,
                     evt.client
             )
+    @classmethod
+    def get_config_class(cls) -> Type[BaseProxyConfig]:
+        return Config
 
 # the end.
diff --git a/maubot.yaml b/maubot.yaml
index 25753a0..2b7dd52 100644
--- a/maubot.yaml
+++ b/maubot.yaml
@@ -1,8 +1,11 @@
 id: eu.sergevictor.hateheif
-version: 0.99.99
+version: 0.99.100
 license: MIT
 modules:
 - hateheif
+config: true
+extra_files:
+- base-config.yaml
 main_class: HateHeifBot
 dependencies:
 - pillow-heif