reverse bridge

src/storage.py

@@ -22,6 +22,25 @@ class BridgeStorage:
                 )
                 """
             )
+            conn.execute(
+                """
+                CREATE TABLE IF NOT EXISTS message_mapping (
+                    telegram_chat_id TEXT NOT NULL,
+                    telegram_message_id TEXT NOT NULL,
+                    max_chat_id TEXT NOT NULL,
+                    max_message_id TEXT NOT NULL,
+                    media_group_id TEXT,
+                    created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
+                    PRIMARY KEY (telegram_chat_id, telegram_message_id)
+                )
+                """
+            )
+            conn.execute(
+                "CREATE INDEX IF NOT EXISTS idx_message_mapping_max ON message_mapping (max_chat_id, max_message_id)"
+            )
+            conn.execute(
+                "CREATE INDEX IF NOT EXISTS idx_message_mapping_media_group ON message_mapping (telegram_chat_id, media_group_id)"
+            )
             conn.commit()
 
     def was_forwarded(self, message_id: str, chat_id: str) -> bool:
@@ -39,3 +58,39 @@ class BridgeStorage:
                 (message_id, chat_id),
             )
             conn.commit()
+
+    def save_mapping(
+        self,
+        *,
+        telegram_chat_id: str,
+        telegram_message_id: str,
+        max_chat_id: str,
+        max_message_id: str,
+        media_group_id: str | None = None,
+    ) -> None:
+        with closing(self._connect()) as conn:
+            conn.execute(
+                """
+                INSERT OR REPLACE INTO message_mapping
+                    (telegram_chat_id, telegram_message_id, max_chat_id, max_message_id, media_group_id)
+                VALUES (?, ?, ?, ?, ?)
+                """,
+                (telegram_chat_id, telegram_message_id, max_chat_id, max_message_id, media_group_id),
+            )
+            conn.commit()
+
+    def get_max_message_id_for_telegram(
+        self, *, telegram_chat_id: str, telegram_message_id: str
+    ) -> str | None:
+        with closing(self._connect()) as conn:
+            row = conn.execute(
+                """
+                SELECT max_message_id
+                FROM message_mapping
+                WHERE telegram_chat_id = ? AND telegram_message_id = ?
+                """,
+                (telegram_chat_id, telegram_message_id),
+            ).fetchone()
+            if not row:
+                return None
+            return str(row[0])