from .base import BasePage

EDIT_WIDGET_BUTTON = '[data-test-id="widget-edit"]'
WIDGET_DRAG_HANDLE = ".widget-drag"
WIDGET_RESIZE_HANDLE = ".widget-resize"
WIDGET_EDITABLE_TEXT_LABEL = '[data-test-id="editable-text-label"]'
WIDGET_TITLE_FIELD = 'input[aria-label="Widget title"]'


class DashboardDetailPage(BasePage):
    def __init__(self, browser, client, **kwargs):
        super().__init__(browser)
        self.client = client
        self.organization = kwargs.get("organization", None)
        self.dashboard = kwargs.get("dashboard", None)

    def wait_until_loaded(self):
        self.browser.wait_until_not('[data-test-id="events-request-loading"]')
        self.browser.wait_until_not('[data-test-id="loading-indicator"]')
        self.browser.wait_until_not('[data-test-id="loading-placeholder"]')
        self.browser.wait_until_not(".loading")

    def visit_default_overview(self):
        self.browser.get(f"/organizations/{self.organization.slug}/dashboard/default-overview/")
        self.wait_until_loaded()
        self.browser.driver.execute_script("window.scrollTo(0, document.body.scrollHeight)")
        self.wait_until_loaded()

    def visit_create_dashboard(self):
        self.browser.get(f"/organizations/{self.organization.slug}/dashboards/new/")
        self.wait_until_loaded()

    def visit_dashboard_detail(self):
        self.browser.get(f"/organizations/{self.organization.slug}/dashboard/{self.dashboard.id}/")
        self.wait_until_loaded()

    def enter_edit_state(self):
        button = self.browser.element('[data-test-id="dashboard-edit"]')
        self.browser.wait_until_clickable('[data-test-id="dashboard-edit"]')
        button.click()
        self.wait_until_loaded()

    def click_dashboard_add_widget_button(self):
        button = self.browser.element('[data-test-id="widget-add"]')
        # HACK: Use JavaScript to execute click to avoid click intercepted issues
        self.browser.driver.execute_script("arguments[0].click()", button)
        self.wait_until_loaded()

    def click_dashboard_header_add_widget_button(self):
        button = self.browser.element('[data-test-id="add-widget-library"]')
        self.browser.wait_until_clickable('[data-test-id="add-widget-library"]')
        button.click()
        self.wait_until_loaded()

    def click_cancel_button(self):
        button = self.browser.element('[data-test-id="dashboard-cancel"]')
        self.browser.wait_until_clickable('[data-test-id="dashboard-cancel"]')
        button.click()
        self.wait_until_loaded()

    def add_widget_through_dashboard(self, widget_title):
        self.click_dashboard_add_widget_button()
        self.browser.element(WIDGET_EDITABLE_TEXT_LABEL).click()
        title_input = self.browser.element(WIDGET_TITLE_FIELD)
        title_input.clear()
        title_input.send_keys(widget_title)
        button = self.browser.element('[aria-label="Add Widget"]')
        button.click()
        self.wait_until_loaded()

    def save_dashboard(self):
        button = self.browser.element('[data-test-id="dashboard-commit"]')
        self.browser.wait_until_clickable('[data-test-id="dashboard-commit"]')
        button.click()
        # This is a kind of hack.
        # After we click the button, an API call is made and we want to wait
        # until the API call finishes. Since the loading indicator isn't used
        # we can't rely on self.wait_until_loaded(). The UI shows a
        # success toast, however if a previous step of a test shows a success
        # toast, a wait_until([data-test-id="toast-success"]) will return
        # immediately due to the previous toast still being in the DOM.
        # Since clicking the save dasboard button is removed once the API
        # call is complete, we can wait for that as a signal
        # that the API is complete.
        self.browser.wait_until_not('[data-test-id="dashboard-commit"]')
        self.wait_until_loaded()