From 3448eec28187e10a9bc719fede3d2bad5cdb925c Mon Sep 17 00:00:00 2001
From: missytake <missytake@systemli.org>
Date: Sun, 8 Oct 2023 10:59:49 +0200
Subject: [PATCH] store relay mappings in pickle.db. closes #2

---
 src/teams_bot/bot.py | 69 +++++++++++++++++++++++++++-----------------
 1 file changed, 43 insertions(+), 26 deletions(-)

diff --git a/src/teams_bot/bot.py b/src/teams_bot/bot.py
index 9003cea..efde135 100644
--- a/src/teams_bot/bot.py
+++ b/src/teams_bot/bot.py
@@ -34,6 +34,8 @@ class RelayPlugin:
         self.account = account
         self.kvstore = kvstore
         self.crew = account.get_chat_by_id(kvstore.get("crew_id"))
+        if not kvstore.get("relays"):
+            kvstore.set("relays", list())
 
     @account_hookimpl
     def ac_incoming_message(self, message: deltachat.Message):
@@ -99,23 +101,12 @@ class RelayPlugin:
 
     def forward_to_outside(self, message: deltachat.Message):
         """forward an answer to an outsider."""
-        bot_localpart = self.account.get_config("addr").split("@")[0]
-        title_prefix = f"[{bot_localpart}] "
-        chat_title = message.chat.get_name().split(title_prefix)[1]
-        logging.debug("stripped %s to %s", message.chat.get_name(), chat_title)
-        for chat in self.account.get_chats():
-            if chat_title == chat.get_name():
-                if message.quote.text in [msg.text for msg in chat.get_messages()]:
-                    outside_chat = chat
-                    break
-                else:
-                    logging.debug(
-                        "No corresponding message in chat %s with name: %s",
-                        chat.id,
-                        chat.get_name(),
-                    )
-        else:
-            logging.error("Couldn't find the chat with the title: %s", chat_title)
+        outside_chat = self.get_outside_chat(message.chat.id)
+        if not outside_chat:
+            logging.error(
+                "Couldn't find the corresponding outside chat for relay group %s",
+                message.chat.id,
+            )
             return
         outside_chat.send_msg(message)
 
@@ -124,15 +115,13 @@ class RelayPlugin:
         outsider = message.get_sender_contact().addr
         crew_members = self.crew.get_contacts()
         crew_members.remove(self.account.get_self_contact())
-        group_name = "[%s] %s" % (
-            self.account.get_config("addr").split("@")[0],
-            message.chat.get_name(),
-        )
-        for chat in self.account.get_chats():
-            if chat.get_name() == group_name:
-                relay_group = chat
-                break
-        else:
+        relay_group = self.get_relay_group(message.chat.id)
+
+        if not relay_group:
+            group_name = "[%s] %s" % (
+                self.account.get_config("addr").split("@")[0],
+                message.chat.get_name(),
+            )
             logging.info("creating new relay group: '%s'", group_name)
             relay_group = self.account.create_group_chat(
                 group_name, crew_members, verified=False
@@ -142,6 +131,10 @@ class RelayPlugin:
                 "This is the relay group for %s; I'll only forward 'direct replies' to the outside."
                 % (message.chat.get_name())
             )
+            relay_mappings = self.kvstore.get("relays")
+            relay_mappings.append(tuple([message.chat.id, relay_group.id]))
+            self.kvstore.set("relays", relay_mappings)
+
         message.set_override_sender_name(outsider)
         relay_group.send_msg(message)
 
@@ -162,3 +155,27 @@ class RelayPlugin:
             if crew_member not in chat.get_contacts():
                 return False  # all crew members have to be in any relay group
         return True
+
+    def get_outside_chat(self, relay_group_id: int) -> deltachat.Chat:
+        """Get the corresponding outside chat for the ID of a relay group.
+
+        :param relay_group_id: the chat.id of the relay group
+        :return: the outside chat
+        """
+        relay_mappings = self.kvstore.get("relays")
+        for mapping in relay_mappings:
+            if mapping[1] == relay_group_id:
+                return self.account.get_chat_by_id(mapping[0])
+        return None
+
+    def get_relay_group(self, outside_id: int) -> deltachat.Chat:
+        """Get the corresponding relay group for the ID of the outside chat.
+
+        :param outside_id: the chat.id of the outside chat
+        :return: the relay group
+        """
+        relay_mappings = self.kvstore.get("relays")
+        for mapping in relay_mappings:
+            if mapping[0] == outside_id:
+                return self.account.get_chat_by_id(mapping[1])
+        return None