"""Base class for web classes."""

from abc import ABC, abstractmethod
from modules.interfaces.interfaceChatModel import ActionDocument, ActionResult
from pydantic import BaseModel, Field
from typing import List


# --- Web search ---

# query -> list of URLs


class WebSearchRequest(BaseModel):
    query: str
    max_results: int


class WebSearchResultItem(BaseModel):
    """Individual search result"""

    title: str
    url: str


class WebSearchDocumentData(BaseModel):
    """Complete search results document"""

    query: str
    results: List[WebSearchResultItem]
    total_count: int


class WebSearchActionDocument(ActionDocument):
    documentData: WebSearchDocumentData


class WebSearchActionResult(ActionResult):
    documents: List[WebSearchActionDocument] = Field(default_factory=list)


class WebSearchBase(ABC):
    @abstractmethod
    async def search_urls(self, request: WebSearchRequest) -> WebSearchActionResult: ...


# --- Web crawl ---

# list of URLs -> list of extracted HTML content


class WebCrawlRequest(BaseModel):
    urls: List[str]


class WebCrawlResultItem(BaseModel):
    """Individual crawl result"""

    url: str
    content: str


class WebCrawlDocumentData(BaseModel):
    """Complete crawl results document"""

    urls: List[str]
    results: List[WebCrawlResultItem]
    total_count: int


class WebCrawlActionDocument(ActionDocument):
    documentData: WebCrawlDocumentData = Field(
        description="The data extracted from crawled URLs"
    )


class WebCrawlActionResult(ActionResult):
    documents: List[WebCrawlActionDocument] = Field(default_factory=list)


class WebCrawlBase(ABC):
    @abstractmethod
    async def crawl_urls(self, request: WebCrawlRequest) -> WebCrawlActionResult: ...


# --- Web scrape ---

# scrape -> list of extracted text; combines web search and crawl in one step


class WebScrapeRequest(BaseModel):
    query: str
    max_results: int


class WebScrapeResultItem(BaseModel):
    """Individual scrape result"""

    url: str
    content: str


class WebScrapeDocumentData(BaseModel):
    """Complete scrape results document"""

    query: str
    results: List[WebScrapeResultItem]
    total_count: int


class WebScrapeActionDocument(ActionDocument):
    documentData: WebScrapeDocumentData = Field(
        description="The data extracted from scraped URLs"
    )


class WebScrapeActionResult(ActionResult):
    documents: List[WebScrapeActionDocument] = Field(default_factory=list)


class WebScrapeBase(ABC):
    @abstractmethod
    async def scrape(self, request: WebScrapeRequest) -> WebScrapeActionResult: ...