push

WebStreamer/server/__init__.py

@@ -0,0 +1,13 @@
+# Taken from megadlbot_oss <https://github.com/eyaadh/megadlbot_oss/blob/master/mega/webserver/__init__.py>
+# Thanks to Eyaadh <https://github.com/eyaadh>
+# This file is a part of TG-FileStreamBot
+# Coding : Jyothis Jayanth [@EverythingSuckz]
+
+from aiohttp import web
+from .stream_routes import routes
+
+
+async def web_server():
+    web_app = web.Application(client_max_size=30000000)
+    web_app.add_routes(routes)
+    return web_app

WebStreamer/server/stream_routes.py

@@ -0,0 +1,77 @@
+# Taken from megadlbot_oss <https://github.com/eyaadh/megadlbot_oss/blob/master/mega/webserver/routes.py>
+# Thanks to Eyaadh <https://github.com/eyaadh>
+
+import math
+import logging
+import secrets
+import mimetypes
+from ..vars import Var
+from aiohttp import web
+from ..bot import StreamBot
+from ..utils.custom_dl import TGCustomYield, chunk_size, offset_fix
+
+routes = web.RouteTableDef()
+
+
+@routes.get("/")
+async def root_route_handler(request):
+    bot_details = await StreamBot.get_me()
+    return web.json_response({"status": "running",
+                              "server_permission": "Open",
+                              "bot_associated_w": bot_details.username})
+
+
+@routes.get("/{message_id}")
+async def stream_handler(request):
+    try:
+        message_id = int(request.match_info['message_id'])
+        return await media_streamer(request, message_id)
+    except ValueError as e:
+        logging.error(e)
+        raise web.HTTPNotFound
+
+
+async def media_streamer(request, message_id: int):
+    range_header = request.headers.get('Range', 0)
+    media_msg = await StreamBot.get_messages(Var.BIN_CHANNEL, message_id)
+    file_properties = await TGCustomYield().generate_file_properties(media_msg)
+    file_size = file_properties.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
+
+    req_length = until_bytes - from_bytes
+
+    new_chunk_size = await chunk_size(req_length)
+    offset = await offset_fix(from_bytes, new_chunk_size)
+    first_part_cut = from_bytes - offset
+    last_part_cut = (until_bytes % new_chunk_size) + 1
+    part_count = math.ceil(req_length / new_chunk_size)
+    body = TGCustomYield().yield_file(media_msg, offset, first_part_cut, last_part_cut, part_count,
+                                      new_chunk_size)
+
+    file_name = file_properties.file_name if file_properties.file_name \
+        else f"{secrets.token_hex(2)}.jpeg"
+    mime_type = file_properties.mime_type if file_properties.mime_type \
+        else f"{mimetypes.guess_type(file_name)}"
+
+    return_resp = web.Response(
+        status=206 if range_header else 200,
+        body=body,
+        headers={
+            "Content-Type": mime_type,
+            "Content-Range": f"bytes {from_bytes}-{until_bytes}/{file_size}",
+            "Content-Disposition": f'attachment; filename="{file_name}"',
+            "Accept-Ranges": "bytes",
+        }
+    )
+
+    if return_resp.status == 200:
+        return_resp.headers.add("Content-Length", str(file_size))
+
+    return return_resp