From 4baf5aad839750e246a2a909ff616948ebae1bfc Mon Sep 17 00:00:00 2001
From: xCyanGrizzly <sam.agsteribbe@hotmail.com>
Date: Wed, 25 Mar 2026 22:23:47 +0100
Subject: [PATCH] feat: capture media_album_id from TDLib messages during
 scanning

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
---
 worker/src/tdlib/download.ts | 3 +++
 worker/src/tdlib/topics.ts   | 3 +++
 2 files changed, 6 insertions(+)

diff --git a/worker/src/tdlib/download.ts b/worker/src/tdlib/download.ts
index e96e07d..b267df5 100644
--- a/worker/src/tdlib/download.ts
+++ b/worker/src/tdlib/download.ts
@@ -35,6 +35,7 @@ interface TdPhotoSize {
 interface TdMessage {
   id: number;
   date: number;
+  media_album_id?: string;
   content: {
     _: string;
     document?: {
@@ -211,6 +212,7 @@ export async function getChannelMessages(
             fileId: String(doc.document.id),
             fileSize: BigInt(doc.document.size),
             date: new Date(msg.date * 1000),
+            mediaAlbumId: msg.media_album_id && msg.media_album_id !== "0" ? msg.media_album_id : undefined,
           });
           continue;
         }
@@ -227,6 +229,7 @@ export async function getChannelMessages(
             caption,
             fileId: String(smallest.photo.id),
             fileSize: smallest.photo.size || smallest.photo.expected_size,
+            mediaAlbumId: msg.media_album_id && msg.media_album_id !== "0" ? msg.media_album_id : undefined,
           });
         }
       }
diff --git a/worker/src/tdlib/topics.ts b/worker/src/tdlib/topics.ts
index 3a83974..cc6b4b1 100644
--- a/worker/src/tdlib/topics.ts
+++ b/worker/src/tdlib/topics.ts
@@ -201,6 +201,7 @@ export async function getTopicMessages(
       messages?: {
         id: number;
         date: number;
+        media_album_id?: string;
         content: {
           _: string;
           document?: {
@@ -248,6 +249,7 @@ export async function getTopicMessages(
           fileId: String(doc.document.id),
           fileSize: BigInt(doc.document.size),
           date: new Date(msg.date * 1000),
+          mediaAlbumId: msg.media_album_id && msg.media_album_id !== "0" ? msg.media_album_id : undefined,
         });
         continue;
       }
@@ -263,6 +265,7 @@ export async function getTopicMessages(
           caption,
           fileId: String(smallest.photo.id),
           fileSize: smallest.photo.size || smallest.photo.expected_size,
+          mediaAlbumId: msg.media_album_id && msg.media_album_id !== "0" ? msg.media_album_id : undefined,
         });
       }
     }