from selenium.webdriver.common import by
from selenium.webdriver.common import keys


from stacklight_tests.helpers.ui import base_pages


class DashboardPage(base_pages.PageObject):
    _submenu_controls_locator = (
        by.By.CLASS_NAME,
        "submenu-controls"
    )

    def __init__(self, driver, dashboard_name):
        super(DashboardPage, self).__init__(driver)
        self._page_title = "Grafana - {}".format(dashboard_name)

    def is_dashboards_page(self):
        return (self.is_the_current_page() and
                self._is_element_visible(*self._submenu_controls_locator))

    def get_back_to_home(self):
        self.go_to_previous_page()
        return MainPage(self.driver)


class MainPage(base_pages.PageObject):
    _dropdown_menu_locator = (
        by.By.CLASS_NAME,
        'navbar-brand-btn')

    _dashboards_list_locator = (
        by.By.CLASS_NAME,
        'sidemenu'
    )

    _dashboard_locator = (by.By.CLASS_NAME,
                          "search-item-dash-db")

    _dashboard_main_locator = (
        by.By.CLASS_NAME,
        "dropdown"
    )

    def __init__(self, driver):
        super(MainPage, self).__init__(driver)
        self._page_title = "Grafana - Home"

    def is_main_page(self):
        return (self.is_the_current_page() and
                self._is_element_visible(*self._dropdown_menu_locator))

    @property
    def dropdown_menu(self):
        return self._get_element(*self._dropdown_menu_locator)

    @property
    def dashboards_list(self):
        self.open_dropdown_menu()
        return self._get_element(*self._dashboards_list_locator)

    @property
    def main_menu_items(self):
        return self.dashboards_list.find_elements(
            *self._dashboard_main_locator
        )

    @property
    def dashboards(self):
        return self.dashboards_list.find_elements(*self._dashboard_locator)

    def is_dropdown_menu_opened(self):
        return self._is_element_present(*self._dashboards_list_locator)

    def open_dropdown_menu(self):
        if not self.is_dropdown_menu_opened():
            self.dropdown_menu.click()

    def open_dashboard(self, dashboard_name):
        dashboards_mapping = {dashboard.text.lower(): dashboard
                              for dashboard in self.dashboards}
        dashboards_mapping[dashboard_name.lower()].click()
        dashboard_page = DashboardPage(self.driver, dashboard_name)
        dashboard_page.is_dashboards_page()
        return dashboard_page


class LoginPage(base_pages.PageObject):
    _login_username_field_locator = (by.By.NAME, 'username')
    _login_password_field_locator = (by.By.NAME, 'password')
    _login_submit_button_locator = (by.By.CLASS_NAME, "btn")

    def __init__(self, driver):
        super(LoginPage, self).__init__(driver)
        self._page_title = "Grafana"

    def is_login_page(self):
        return (self.is_the_current_page() and
                self._is_element_visible(*self._login_submit_button_locator))

    @property
    def username(self):
        return self._get_element(*self._login_username_field_locator)

    @property
    def password(self):
        return self._get_element(*self._login_password_field_locator)

    @property
    def login_button(self):
        return self._get_element(*self._login_submit_button_locator)

    def _click_on_login_button(self):
        self.login_button.click()

    def _press_enter_on_login_button(self):
        self.login_button.send_keys(keys.Keys.RETURN)

    def login(self, user, password):
        return self.login_with_mouse_click(user, password)

    def login_with_mouse_click(self, user, password):
        return self._do_login(user, password, self._click_on_login_button)

    def login_with_enter_key(self, user, password):
        return self._do_login(user, password,
                              self._press_enter_on_login_button)

    def _do_login(self, user, password, login_method):
        return self.login_as_user(user, password, login_method)

    def login_as_user(self, user, password, login_method):
        self._fill_field_element(user, self.username)
        self._fill_field_element(password, self.password)
        login_method()
        return MainPage(self.driver)