Code Updates & Optimisations :Server Upgrade Test

FileStream/server/Functions/downloader.py

@@ -21,7 +21,7 @@ from FileStream.utils.FileProcessors.custom_ul import  TeleUploader
 
 async def media_streamer(request: web.Request, db_id: str, speed: str):
     # Get the Range header from the request, default to 0 if not present
-    range_header = request.headers.get("Range", "0")
+    range_header = request.headers.get("Range", 0)
     client = await req_client()
     # Log client info if multi-client mode
     if Telegram.MULTI_CLIENT:
@@ -49,7 +49,7 @@ async def media_streamer(request: web.Request, db_id: str, speed: str):
         from_bytes, until_bytes = parse_range(range_header, file_size)
 
         # If range is invalid, return a 416 error
-        if from_bytes is None or until_bytes is None:
+        if (until_bytes > file_size) or (from_bytes < 0) or (until_bytes < from_bytes):
             return web.Response(
                 status=416,
                 body="416: Range not satisfiable",
@@ -58,7 +58,7 @@ async def media_streamer(request: web.Request, db_id: str, speed: str):
 
         # Set chunk size based on speed
         chunk_size =  1024 * 1024 if speed == "FAST" else 512 * 1024
-        
+
 
         # Ensure we don't go past the file size
         until_bytes = min(until_bytes, file_size - 1)
@@ -97,10 +97,13 @@ async def media_streamer(request: web.Request, db_id: str, speed: str):
 def parse_range(range_header: str, file_size: int):
     """Helper function to parse the range header."""
     try:
-        range_str = range_header.replace("bytes=", "")
-        from_bytes, until_bytes = range_str.split("-")
-        from_bytes = int(from_bytes)
-        until_bytes = int(until_bytes) if until_bytes else file_size - 1
+        if range_header:
+            from_bytes, until_bytes = range_header.replace("bytes=", "").split("-")
+            from_bytes = int(from_bytes)
+            until_bytes = int(until_bytes) if until_bytes else file_size - 1
+        else:
+            from_bytes = request.http_range.start or 0
+            until_bytes = (request.http_range.stop or file_size) - 1
         return from_bytes, until_bytes
     except ValueError:
         return None, None