from typing import Any, Literal, Optional
from pydantic import BaseModel, Field, HttpUrl


WaitUntilValue = Literal["load", "domcontentloaded", "networkidle", "commit"]


class VisitRequest(BaseModel):
    url: HttpUrl
    screenshot: bool = False
    waitUntil: WaitUntilValue = Field(default="load")
    proxy: Optional[str] = None
    proxyAuth: Optional[dict[str, Any]] = None
    incognito: bool = False


class ArasResponse(BaseModel):
    status: str
    message: str
    data: Any = None
    extra: Any = None
    code: str = "UNKNOWN_CODE"