From 6322163b1034bf06dfc246088442c77bf189f2e4 Mon Sep 17 00:00:00 2001
From: Tanner Collin <git@tannercollin.com>
Date: Wed, 21 Jan 2026 11:49:48 -0700
Subject: [PATCH] feat: Add Telegram bot notification for batch upload
 completion

Co-authored-by: aider (gemini/gemini-2.5-pro) <aider@aider.chat>
---
 app/app.py | 79 +++++++++++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 78 insertions(+), 1 deletion(-)

diff --git a/app/app.py b/app/app.py
index ffb1fb4..556c390 100644
--- a/app/app.py
+++ b/app/app.py
@@ -20,7 +20,8 @@ import sqlite3
 import binascii
 import pytz
 from datetime import datetime
-from typing import Dict, List, Optional
+from typing import Dict, List, Optional, Tuple
+import math
 
 import logging
 import httpx
@@ -173,6 +174,78 @@ hub = SessionHub()
 
 # ---------- Telegram Bot ----------
 
+# Batch upload notifications
+_upload_batch: List[Tuple[str, int]] = []
+_batch_complete_timer: Optional[asyncio.TimerHandle] = None
+_batch_lock = asyncio.Lock()
+
+
+def human_size(bytes_val: int) -> str:
+    """Return a human-readable size string."""
+    if not bytes_val:
+        return "0 B"
+    k = 1024
+    sizes = ['B', 'KB', 'MB', 'GB', 'TB']
+    i = 0
+    if bytes_val > 0:
+        i = int(math.floor(math.log(bytes_val) / math.log(k)))
+    if i >= len(sizes):
+        i = len(sizes) - 1
+    return f"{(bytes_val / (k**i)):.1f} {sizes[i]}"
+
+async def send_batch_notification():
+    """Format and send a summary of the recently completed upload batch."""
+    async with _batch_lock:
+        if not _upload_batch:
+            return
+        
+        batch_copy = list(_upload_batch)
+        _upload_batch.clear()
+        
+        global _batch_complete_timer
+        if _batch_complete_timer:
+            _batch_complete_timer.cancel()
+            _batch_complete_timer = None
+
+    num_files = len(batch_copy)
+    total_size = sum(size for _, size in batch_copy)
+    
+    file_list_str = ""
+    if num_files > 0:
+        filenames = [name or "file" for name, _ in batch_copy]
+        if num_files > 20:
+            file_list_str = "\n".join(f"- {name}" for name in filenames[:20])
+            file_list_str += f"\n... and {num_files - 20} more."
+        else:
+            file_list_str = "\n".join(f"- {name}" for name in filenames)
+    
+    msg = f"✅ Batch upload complete!\n\n- Files: {num_files}\n- Total size: {human_size(total_size)}\n\n{file_list_str}".strip()
+    await send_telegram_message(TELEGRAM_OWNER_ID, msg)
+
+def _schedule_batch_notification():
+    # Helper to run async func from sync context of call_later
+    asyncio.create_task(send_batch_notification())
+
+async def reset_telegram_debounce():
+    """Resets the 30s timer for batch completion notification."""
+    if not SETTINGS.telegram_bot_api_key or not TELEGRAM_OWNER_ID:
+        return
+    
+    global _batch_complete_timer
+    async with _batch_lock:
+        if _batch_complete_timer:
+            _batch_complete_timer.cancel()
+        loop = asyncio.get_event_loop()
+        _batch_complete_timer = loop.call_later(30, _schedule_batch_notification)
+
+async def add_file_to_batch(filename: str, size: int):
+    """Adds a completed file to the batch list."""
+    if not SETTINGS.telegram_bot_api_key or not TELEGRAM_OWNER_ID:
+        return
+    async with _batch_lock:
+        _upload_batch.append((filename, size))
+
+
 TELEGRAM_API_URL = f"https://api.telegram.org/bot{SETTINGS.telegram_bot_api_key}"
 TELEGRAM_OWNER_ID = SETTINGS.telegram_bot_owner_id
 
@@ -620,6 +693,8 @@ async def api_upload(
         with open(save_path, "wb") as f:
             f.write(raw)
         db_insert_upload(checksum, file.filename, size, device_asset_id, None, created_iso)
+        await add_file_to_batch(file.filename, size)
+        await reset_telegram_debounce()
         
         msg = f"Saved to {album_for_saving}/{os.path.basename(save_path)}"
         await send_progress(session_id, item_id, "done", 100, msg)
@@ -761,6 +836,7 @@ async def api_upload_chunk(
     except Exception as e:
         logger.exception("Chunk write failed: %s", e)
         return JSONResponse({"error": "chunk_write_failed"}, status_code=500)
+    await reset_telegram_debounce()
     return JSONResponse({"ok": True})
 
 @app.post("/api/upload/chunk/complete")
@@ -953,6 +1029,7 @@ async def api_upload_chunk_complete(request: Request) -> JSONResponse:
         with open(save_path, "wb") as f:
             f.write(raw)
         db_insert_upload(checksum, file_like_name, file_size, device_asset_id, None, created_iso)
+        await add_file_to_batch(file_like_name, file_size)
         
         msg = f"Saved to {album_for_saving}/{os.path.basename(save_path)}"
         await send_progress(session_id_local, item_id_local, "done", 100, msg)