Allow linking chats

core/clients/signal.py

@@ -11,7 +11,7 @@ from signalbot import Command, Context, SignalBot
 
 from core.clients import ClientBase, signalapi
 from core.messaging import ai, history, media_bridge, natural, replies, utils
-from core.models import Chat, Manipulation, PersonIdentifier, QueuedMessage
+from core.models import Chat, Manipulation, PersonIdentifier, PlatformChatLink, QueuedMessage
 from core.util import logs
 
 log = logs.get_logger("signalF")
@@ -263,6 +263,49 @@ class NewSignalBot(SignalBot):
         await self._resolve_commands()
         await self._produce_consume_messages()
 
+    async def _upsert_groups(self) -> None:
+        groups = getattr(self, "groups", None) or []
+        if not groups:
+            self.log.debug("[Signal] _upsert_groups: no groups to persist")
+            return
+
+        identifiers = await sync_to_async(list)(
+            PersonIdentifier.objects.filter(service="signal").select_related("user")
+        )
+        seen_user_ids: set = set()
+        users = []
+        for pi in identifiers:
+            if pi.user_id not in seen_user_ids:
+                seen_user_ids.add(pi.user_id)
+                users.append(pi.user)
+        if not users:
+            self.log.debug("[Signal] _upsert_groups: no PersonIdentifiers found — skipping")
+            return
+
+        for user in users:
+            for group in groups:
+                group_id = group.get("id") or ""
+                name = group.get("name") or group_id
+                if not group_id:
+                    continue
+                await sync_to_async(PlatformChatLink.objects.update_or_create)(
+                    user=user,
+                    service="signal",
+                    chat_identifier=group_id,
+                    defaults={
+                        "person": None,
+                        "person_identifier": None,
+                        "is_group": True,
+                        "chat_name": name,
+                    },
+                )
+
+        self.log.info("[Signal] upserted %d groups for %d users", len(groups), len(users))
+
+    async def _detect_groups(self):
+        await super()._detect_groups()
+        await self._upsert_groups()
+
     def start(self):
         """Start bot without blocking the caller's event loop."""
         task = self._event_loop.create_task(