Invidious-quic-proxy/main.py

import pathlib
import logging

import pytomlpp
import appdirs
from multidict import CIMultiDict
from aiohttp import web

import quicclient

logging.basicConfig(filename="test.log", level=logging.INFO)

APP_NAME = "QUICProxy"
APP_AUTHOR = "syeopite"

CONFIG_DIRECTORY = pathlib.Path(f"{appdirs.user_config_dir(APP_NAME, APP_AUTHOR)}")
CONFIG_DIRECTORY.mkdir(parents=True, exist_ok=True)

CONFIG_FILE = pathlib.Path(f"{appdirs.user_config_dir(APP_NAME, APP_AUTHOR)}/config.toml")
CONFIG_FILE.touch(exist_ok=True)


with open(f"{CONFIG_FILE}") as config:
    config = pytomlpp.loads(config.read())
if not config:
    config = {"port": 8080, "host": "0.0.0.0"}
routes = web.RouteTableDef()


@routes.post("/")
async def post(request):
    arguments = await request.json()
    post_data = str(arguments.get("data", ""))
    method = arguments["method"]

    # Create heders
    intermediate_header_processing = [(k, v) for k, v in arguments.get("headers", {}).items()]
    processed_headers = CIMultiDict(intermediate_header_processing)

    result = await quicclient.request(arguments["url"], method, processed_headers, post_data if post_data else None)

    if result["headers"][":status"] == "304":
        return web.Response(body=b"", headers=result["headers"], status=304)
    else:
        return web.Response(body=result["response"], headers=result["headers"])


app = web.Application()
app.add_routes(routes)
if __name__ == '__main__':
    web.run_app(app, **config)