Update bot/telegram/media.py

bot/telegram/media.py

@@ -1,19 +1,38 @@
 # PATH: bot/telegram/media.py
 import os
+import tempfile
+from typing import Callable, Awaitable, Optional
+
 from hydrogram import Client
 from hydrogram.types import Message
-from bot.temp.files import make_temp_file_path
 
-async def download_to_temp(app: Client, m: Message) -> tuple[str, int, str]:
+ProgressCB = Callable[[int, int], Awaitable[None]]
+
+async def download_to_temp(app: Client, m: Message, progress_cb: Optional[ProgressCB] = None):
     media = m.video or m.document
     if not media:
-        raise RuntimeError("No media to download")
+        return "", 0, ""
+
+    file_name = getattr(media, "file_name", None) or "video.mp4"
+    suffix = "." + file_name.rsplit(".", 1)[1] if "." in file_name else ""
+
+    fd, path = tempfile.mkstemp(prefix="studytube_", suffix=suffix)
+    os.close(fd)
+
+    try:
+        await app.download_media(message=m, file_name=path, progress=progress_cb)
+    except TypeError:
+        await app.download_media(message=m, file_name=path)
+
+    if not os.path.exists(path):
+        return "", 0, file_name
 
-    file_name = getattr(media, "file_name", None) or "video.bin"
-    file_size = int(getattr(media, "file_size", 0) or 0)
+    size = os.path.getsize(path)
+    expected = int(getattr(media, "file_size", 0) or 0)
 
-    out_path = make_temp_file_path(file_name)
-    os.makedirs(os.path.dirname(out_path), exist_ok=True)
+    if expected and size and expected != size:
+        try: os.remove(path)
+        except: pass
+        raise RuntimeError(f"download_size_mismatch expected={expected} got={size}")
 
-    await app.download_media(message=m, file_name=out_path)
-    return out_path, file_size, file_name
+    return path, size, file_name