Byparr/main.py

from __future__ import annotations

import logging
import time

import uvicorn.config
from fastapi import FastAPI
from fastapi.responses import RedirectResponse
from sbase import SB, BaseCase

from src.models.requests import LinkRequest, LinkResponse, Solution
from src.utils import logger
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: Chrome = await new_browser()
    # browser.get("https://google.com")
    # browser.stop()
    return {"status": "ok"}


@app.post("/v1")
def read_item(request: LinkRequest):
    """Handle POST requests."""
    start_time = int(time.time() * 1000)
    # request.url = "https://nowsecure.nl"
    logger.info(f"Request: {request}")
    response: LinkResponse

    # start_time = int(time.time() * 1000)
    with SB(uc=True, locale_code="en", test=False, xvfb=True, ad_block=True) as sb:
        sb: BaseCase
        sb.uc_open_with_reconnect(request.url)
        sb.uc_gui_click_captcha()
        logger.info(f"Got webpage: {request.url}")
        sb.save_screenshot("screenshot.png")
        logger.info(f"Got webpage: {request.url}")

        response = LinkResponse(
            message="Success",
            solution=Solution(
                userAgent=sb.get_user_agent(),
                url=sb.get_current_url(),
                status=200,
                cookies=sb.get_cookies(),
                headers={},
                response=sb.get_page_source(),
            ),
            startTimestamp=start_time,
        )

    return response


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