Upload 4 files

app.py

@@ -1,6 +1,6 @@
 import os
-import os
 import json
+import asyncio
 
 from fastapi import FastAPI, Request
 from fastapi.responses import Response, StreamingResponse, FileResponse
@@ -12,6 +12,8 @@ UPSTREAM = f"http://127.0.0.1:{os.environ.get('TELEGRAM_UPSTREAM_PORT', '8081')}
 app = FastAPI()
 
 WORK_DIR = os.environ.get("TELEGRAM_WORK_DIR", "/tmp/telegram-bot-api-data")
+DOWNLOAD_WAIT_SECONDS = float(os.environ.get("TELEGRAM_DOWNLOAD_WAIT_SECONDS", "8"))
+DOWNLOAD_POLL_INTERVAL_MS = int(os.environ.get("TELEGRAM_DOWNLOAD_POLL_INTERVAL_MS", "200"))
 
 def _normalize_bot_api_file_path(raw_fp: str | None, token_enc: str | None) -> str:
     if not raw_fp:
@@ -92,6 +94,48 @@ async def proxy(path: str, request: Request):
             except Exception:
                 pass
 
+        # 如果本地没有这个文件（HF 不持久化常见），允许 CloudPaste 通过 query 传入 file_id 来触发回源下载：
+        # - 不新增新的路径接口，仍然是 /tg/file/bot<TOKEN>/<file_path>
+        # - 只是在 query 上多带一个 file_id（例如 ?file_id=xxx）
+        file_id = request.query_params.get("file_id") or request.query_params.get("fid")
+        if token_enc and file_id:
+            async with httpx.AsyncClient(timeout=None, follow_redirects=True) as client:
+                r = await client.get(f"{UPSTREAM}/bot{token_enc}/getFile", params={"file_id": file_id})
+                payload = None
+                try:
+                    payload = r.json()
+                except Exception:
+                    payload = None
+
+            if r.status_code == 200 and isinstance(payload, dict) and payload.get("ok") is True:
+                fp = None
+                try:
+                    fp = (payload.get("result") or {}).get("file_path")
+                except Exception:
+                    fp = None
+
+                rel2 = _normalize_bot_api_file_path(fp if isinstance(fp, str) else None, token_enc)
+                if rel2:
+                    candidates2 = [
+                        os.path.join(WORK_DIR, token_enc, rel2),
+                        os.path.join(WORK_DIR, rel2),
+                    ]
+
+                    waited = 0.0
+                    interval = max(0.05, DOWNLOAD_POLL_INTERVAL_MS / 1000.0)
+                    max_wait = max(0.0, DOWNLOAD_WAIT_SECONDS)
+                    while waited <= max_wait:
+                        for p2 in candidates2:
+                            try:
+                                if p2 and os.path.isfile(p2):
+                                    return FileResponse(p2)
+                            except Exception:
+                                pass
+                        if waited >= max_wait:
+                            break
+                        await asyncio.sleep(interval)
+                        waited += interval
+
     url = f"{UPSTREAM}/{path_for_upstream}"
 
     params = list(request.query_params.multi_items())