from typing import Self

from selenium.webdriver import Chrome
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.remote.webdriver import WebDriver

from pomcorn import ComponentWithBaseLocator, Element, ListComponent, Page, locators


# Prepare base page
class PyPIPage(Page):

    APP_ROOT = "https://pypi.org"

    def check_page_is_loaded(self) -> bool:
        return self.init_element(locators.TagNameLocator("main")).is_displayed

    @property
    def search(self) -> Element[locators.XPathLocator]:
        return self.init_element(locators.IdLocator("search"))


# Prepare components
Package = ComponentWithBaseLocator[PyPIPage]


class PackageList(ListComponent[Package, PyPIPage]):

    item_class = Package

    @property
    def base_item_locator(self) -> locators.XPathLocator:
        return self.base_locator // locators.ClassLocator("snippet__name")

    @property
    def names(self) -> list[str]:
        return [package.body.get_text() for package in self.all]


# Prepare search page
class SearchPage(PyPIPage):

    @classmethod
    def open(cls, webdriver: WebDriver, **kwargs) -> Self:
        pypi_page = super().open(webdriver, **kwargs)
        # Specific logic for PyPI for an open search page
        pypi_page.search.fill("")
        pypi_page.search.send_keys(Keys.ENTER)
        return cls(webdriver, **kwargs)

    @property
    def results(self) -> PackageList:
        return PackageList(
            page=self,
            base_locator=locators.PropertyLocator(
                prop="aria-label",
                value="Search results",
            ),
        )

    def find(self, query: str) -> PackageList:
        self.search.fill(query)
        self.search.send_keys(Keys.ENTER)
        return self.results


search_page = SearchPage.open(webdriver=Chrome())
print(search_page.find("saritasa").names)