Byparr/main.py

from __future__ import annotations

import asyncio
import logging
import time

import uvicorn
import uvicorn.config
from fastapi import FastAPI, HTTPException
from fastapi.responses import RedirectResponse

from src.models.requests import LinkRequest, LinkResponse
from src.utils import logger
from src.utils.browser import bypass_cloudflare, new_browser
from src.utils.consts import LOG_LEVEL

app = FastAPI(debug=LOG_LEVEL == logging.DEBUG, log_level=LOG_LEVEL)


@app.get("/")
def read_root():
    """Redirect to /docs."""
    logger.info("Redirecting to /docs")
    return RedirectResponse(url="/docs", status_code=301)


@app.get("/health")
async def health_check():
    """Health check endpoint."""
    logger.info("Health check")
    browser = await new_browser()
    await browser.grant_all_permissions()
    page = await browser.get("https://google.com")
    await page.bring_to_front()
    browser.stop()
    return {"status": "ok"}


@app.post("/v1")
async def read_item(request: LinkRequest):
    """Handle POST requests."""
    # request.url = "https://nowsecure.nl"
    logger.info(f"Request: {request}")
    start_time = int(time.time() * 1000)
    browser = await new_browser()
    await browser.grant_all_permissions()
    await asyncio.sleep(1)
    page = await browser.get(request.url)
    await page.bring_to_front()
    timeout = request.maxTimeout
    if timeout == 0:
        timeout = None
    try:
        challenged = await asyncio.wait_for(bypass_cloudflare(page), timeout=timeout)
    except asyncio.TimeoutError as e:
        logger.info("Timed out bypassing Cloudflare")
        browser.stop()
        raise HTTPException(
            detail="Timed out bypassing Cloudflare", status_code=408
        ) from e
    except Exception as e:
        browser.stop()
        raise HTTPException(detail="Couldn't bypass", status_code=500) from e

    logger.info(f"Got webpage: {request.url}")

    response = await LinkResponse.create(
        page=page,
        start_timestamp=start_time,
        challenged=challenged,
    )

    browser.stop()
    return response


if __name__ == "__main__":
    uvicorn.run(app, host="0.0.0.0", port=8191, log_level=LOG_LEVEL)  # noqa: S104