|
|
|
|
|
import os |
|
|
import time |
|
|
import math |
|
|
import logging |
|
|
import asyncio |
|
|
import traceback |
|
|
from aiohttp import web |
|
|
from pyrogram import raw |
|
|
|
|
|
from aiohttp.http_exceptions import BadStatusLine |
|
|
|
|
|
|
|
|
|
|
|
from FileStream.config import Telegram |
|
|
from FileStream import utils, StartTime, __version__ |
|
|
from FileStream.Tools import mime_identifier, Time_ISTKolNow |
|
|
from FileStream.bot import req_client, FileStream |
|
|
from FileStream.bot import MULTI_CLIENTS, WORK_LOADS, ACTIVE_CLIENTS |
|
|
from FileStream.server.exceptions import FIleNotFound, InvalidHash |
|
|
from FileStream.server.render_template import render_page, render_upload |
|
|
from FileStream.utils.FileProcessors.custom_ul import TeleUploader |
|
|
|
|
|
|
|
|
async def media_streamer(request: web.Request, db_id: str, speed:str): |
|
|
|
|
|
range_header = request.headers.get("Range", 0) |
|
|
|
|
|
|
|
|
|
|
|
client = await req_client() |
|
|
|
|
|
if Telegram.MULTI_CLIENT: |
|
|
logging.info(f"Client {client['index']} is now serving {request.headers.get('X-FORWARDED-FOR',request.remote)}") |
|
|
|
|
|
if client['client'] in ACTIVE_CLIENTS: |
|
|
tg_connect = ACTIVE_CLIENTS[client['client']] |
|
|
logging.debug(f"Using cached ByteStreamer object for client {client['index']}") |
|
|
|
|
|
else: |
|
|
logging.debug(f"Creating new ByteStreamer object for client {client['index']}") |
|
|
tg_connect = utils.ByteStreamer(client['client']) |
|
|
ACTIVE_CLIENTS[client['client']] = tg_connect |
|
|
|
|
|
logging.debug("before calling get_file_properties") |
|
|
file_id = await tg_connect.get_file_properties(db_id, MULTI_CLIENTS) |
|
|
logging.debug("after calling get_file_properties") |
|
|
|
|
|
file_size = file_id.file_size |
|
|
|
|
|
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 |
|
|
|
|
|
if (until_bytes > file_size) or (from_bytes < 0) or (until_bytes < from_bytes): |
|
|
|
|
|
return web.Response( |
|
|
status=416, |
|
|
body="416: Range not satisfiable", |
|
|
headers={"Content-Range": f"bytes */{file_size}"}, |
|
|
) |
|
|
|
|
|
chunk_size = 4 * 1024 * 1024 if speed == "FAST" else 512 * 1024 |
|
|
|
|
|
until_bytes = min(until_bytes, file_size - 1) |
|
|
|
|
|
offset = from_bytes - (from_bytes % chunk_size) |
|
|
first_part_cut = from_bytes - offset |
|
|
last_part_cut = until_bytes % chunk_size + 1 |
|
|
|
|
|
req_length = until_bytes - from_bytes + 1 |
|
|
part_count = math.ceil(until_bytes / chunk_size) - math.floor(offset / chunk_size) |
|
|
|
|
|
body = tg_connect.yield_file(file_id, client['index'], offset, first_part_cut,last_part_cut, part_count, chunk_size) |
|
|
|
|
|
mime_type = file_id.mime_type |
|
|
file_name = utils.get_name(file_id) |
|
|
disposition = "attachment" |
|
|
|
|
|
if not mime_type: |
|
|
mime_type = mimetypes.guess_type(file_name)[0] or "application/octet-stream" |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
return web.Response( |
|
|
status=206 if range_header else 200, |
|
|
body=body, |
|
|
headers={ |
|
|
"Content-Type": f"{mime_type}", |
|
|
"Content-Range": f"bytes {from_bytes}-{until_bytes}/{file_size}", |
|
|
"Content-Length": str(req_length), |
|
|
"Content-Disposition": f'{disposition}; filename="{file_name}"', |
|
|
"Accept-Ranges": "bytes", |
|
|
}, |
|
|
) |
