2024-07-24 13:57:40 +00:00
|
|
|
from __future__ import annotations
|
|
|
|
|
|
|
|
import time
|
2024-11-24 23:04:19 +00:00
|
|
|
from http import HTTPStatus
|
2025-03-01 19:55:35 +01:00
|
|
|
from typing import Any
|
2024-07-24 13:57:40 +00:00
|
|
|
|
2025-02-17 23:08:14 +00:00
|
|
|
from fastapi import Body
|
2025-03-01 19:55:35 +01:00
|
|
|
from pydantic import BaseModel, Field
|
2024-07-24 13:57:40 +00:00
|
|
|
|
2025-02-17 23:08:14 +00:00
|
|
|
from src import consts
|
2024-11-30 12:42:51 +00:00
|
|
|
|
2024-07-24 13:57:40 +00:00
|
|
|
|
|
|
|
class LinkRequest(BaseModel):
|
2025-03-01 19:55:35 +01:00
|
|
|
cmd: str = Field(
|
|
|
|
default_factory=lambda: "request.get",
|
|
|
|
description="Type of request, currently only supports GET requests. This string is purely for compatibility with FlareSolverr.",
|
|
|
|
)
|
|
|
|
url: str = Field(pattern=r"^https?://", default="https://")
|
|
|
|
max_timeout: int = Field(default=60)
|
2024-07-24 13:57:40 +00:00
|
|
|
|
|
|
|
|
|
|
|
class Solution(BaseModel):
|
|
|
|
url: str
|
|
|
|
status: int
|
2024-07-24 15:53:14 +00:00
|
|
|
cookies: list
|
2024-07-24 13:57:40 +00:00
|
|
|
userAgent: str # noqa: N815 # Ignore to preserve compatibility
|
|
|
|
headers: dict[str, Any]
|
|
|
|
response: str
|
|
|
|
|
2024-11-24 23:04:19 +00:00
|
|
|
@classmethod
|
2024-11-30 12:42:51 +00:00
|
|
|
def invalid(cls, url: str):
|
|
|
|
"""
|
|
|
|
Return an empty Solution with default values.
|
|
|
|
|
|
|
|
Useful for returning an error response.
|
|
|
|
"""
|
2024-11-24 23:04:19 +00:00
|
|
|
return cls(
|
2024-11-30 12:42:51 +00:00
|
|
|
url=url,
|
2024-11-24 23:04:19 +00:00
|
|
|
status=HTTPStatus.INTERNAL_SERVER_ERROR,
|
|
|
|
cookies=[],
|
|
|
|
userAgent="",
|
|
|
|
headers={},
|
|
|
|
response="",
|
|
|
|
)
|
|
|
|
|
2024-07-24 13:57:40 +00:00
|
|
|
|
|
|
|
class LinkResponse(BaseModel):
|
|
|
|
status: str = "ok"
|
|
|
|
message: str
|
|
|
|
solution: Solution
|
2025-03-01 19:55:35 +01:00
|
|
|
start_timestamp: int = Field(alias="startTimestamp", default_factory=lambda: int(time.time() * 1000))
|
|
|
|
end_timestamp: int = Field(alias="endTimestamp", default_factory=lambda: int(time.time() * 1000))
|
2024-11-30 12:42:51 +00:00
|
|
|
version: str = consts.VERSION
|
2024-07-24 13:57:40 +00:00
|
|
|
|
|
|
|
@classmethod
|
2024-11-30 12:42:51 +00:00
|
|
|
def invalid(cls, url: str):
|
|
|
|
"""
|
|
|
|
Return an invalid LinkResponse with default error values.
|
|
|
|
|
|
|
|
This method is used to generate a response indicating an invalid request.
|
|
|
|
"""
|
2024-07-24 13:57:40 +00:00
|
|
|
return cls(
|
2024-11-24 23:04:19 +00:00
|
|
|
status="error",
|
|
|
|
message="Invalid request",
|
2024-11-30 12:42:51 +00:00
|
|
|
solution=Solution.invalid(url),
|
2025-02-17 23:08:14 +00:00
|
|
|
start_timestamp=int(time.time() * 1000),
|
|
|
|
end_timestamp=int(time.time() * 1000),
|
2025-03-01 19:55:35 +01:00
|
|
|
)
|