Add storage, delete_all, history, and auto-cleanup

app.py

@@ -2,6 +2,7 @@ import asyncio
 import os
 import time
 import uuid
+import shutil
 import aiohttp
 from fastapi import FastAPI, BackgroundTasks, Request
 from fastapi.responses import FileResponse, StreamingResponse
@@ -9,8 +10,7 @@ from pydantic import BaseModel
 
 app = FastAPI()
 
-# In-memory store for download tasks
-# tasks = { "task_id": { "url": str, "status": "downloading"|"completed"|"error", "total_size": int, "downloaded": int, "speed": float, "file_path": str } }
+# tasks = { "task_id": { ... } }
 tasks = {}
 
 DATA_DIR = "/data"
@@ -19,15 +19,39 @@ os.makedirs(DATA_DIR, exist_ok=True)
 class DownloadRequest(BaseModel):
     url: str
 
+async def cleanup_old_files():
+    while True:
+        try:
+            now = time.time()
+            for filename in os.listdir(DATA_DIR):
+                filepath = os.path.join(DATA_DIR, filename)
+                if os.path.isfile(filepath):
+                    # 24 hours = 86400 seconds
+                    if now - os.path.getmtime(filepath) > 86400:
+                        os.remove(filepath)
+                        # Remove from tasks if exists
+                        for tid, tinfo in list(tasks.items()):
+                            if tinfo.get("file_path") == filepath:
+                                del tasks[tid]
+        except Exception as e:
+            print(f"Cleanup error: {e}")
+        await asyncio.sleep(3600) # Check every hour
+
+@app.on_event("startup")
+async def startup_event():
+    asyncio.create_task(cleanup_old_files())
+
 async def download_file(task_id: str, url: str):
     file_path = os.path.join(DATA_DIR, f"{task_id}.bin")
     tasks[task_id] = {
+        "task_id": task_id,
         "url": url,
         "status": "downloading",
         "total_size": 0,
         "downloaded": 0,
         "speed": 0.0,
-        "file_path": file_path
+        "file_path": file_path,
+        "timestamp": time.time()
     }
     
     try:
@@ -43,7 +67,7 @@ async def download_file(task_id: str, url: str):
                 last_downloaded = 0
                 
                 with open(file_path, 'wb') as f:
-                    async for chunk in response.content.iter_chunked(1024 * 1024): # 1MB chunks to save memory
+                    async for chunk in response.content.iter_chunked(1024 * 1024):
                         if not chunk:
                             break
                         f.write(chunk)
@@ -51,7 +75,7 @@ async def download_file(task_id: str, url: str):
                         tasks[task_id]["downloaded"] = downloaded
                         
                         current_time = time.time()
-                        if current_time - last_time >= 1.0: # Update speed every second
+                        if current_time - last_time >= 1.0:
                             speed = (downloaded - last_downloaded) / (current_time - last_time)
                             tasks[task_id]["speed"] = speed
                             last_time = current_time
@@ -91,7 +115,6 @@ async def stream(task_id: str, request: Request):
     
     range_header = request.headers.get("Range")
     if range_header:
-        # Simple range request handling for streaming
         byte1, byte2 = 0, None
         match = range_header.replace("bytes=", "").split("-")
         if match[0]:
@@ -124,6 +147,27 @@ async def stream(task_id: str, request: Request):
     else:
         return FileResponse(file_path)
 
+@app.get("/storage")
+async def get_storage():
+    total, used, free = shutil.disk_usage(DATA_DIR)
+    return {"total": total, "used": used, "free": free}
+
+@app.post("/delete_all")
+async def delete_all():
+    for filename in os.listdir(DATA_DIR):
+        filepath = os.path.join(DATA_DIR, filename)
+        try:
+            if os.path.isfile(filepath):
+                os.remove(filepath)
+        except Exception as e:
+            print(f"Failed to delete {filepath}: {e}")
+    tasks.clear()
+    return {"status": "success"}
+
+@app.get("/history")
+async def get_history():
+    return {"history": list(tasks.values())}
+
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=7860)