# -*- coding: utf-8 -*-
from selenium.webdriver import ActionChains
from selenium.webdriver.common.keys import Keys
from component import Component
from classes.util.scripts import GetScripts
from classes.util.scripts import SetScripts
from time import time
import math
from time import sleep


class Map(Component):
    selectors = {
        'self': '#map',
        'info': '#info'
    }

    def zoom_selection(self):
        center = self.driver.execute_script(GetScripts.getContainerCenter)
        actions = ActionChains(self.driver)
        actions.move_to_element_with_offset(self.element, int(center['x']), int(center['y']))
        actions.key_down(Keys.SHIFT)
        actions.click_and_hold()
        actions.move_by_offset(300, 300)
        actions.release()
        actions.perform()

    def center_click(self):
        center = self.driver.execute_script(GetScripts.getContainerCenter)
        actions = ActionChains(self.driver)
        actions.move_to_element_with_offset(self.element, int(center['x']), int(center['y']))
        actions.click().perform()

    def center_dbclick(self):
        center = self.driver.execute_script(GetScripts.getContainerCenter)
        actions = ActionChains(self.driver)
        actions.move_to_element_with_offset(self.element, int(center['x']), int(center['y']))
        actions.double_click().perform()

    def set_zoom(self, level):
        self.driver.execute_script(SetScripts.set_zoom(level))

    def wait_init(self, timeout=3, polling=0.5):
        """
        :param timeout: timeout to rise exception
        :param polling: how often check map init
        :return: None
        """
        request_num = math.ceil(timeout / polling)
        tries = 0
        tries_out = False
        map_inited = False
        start = time()
        while not tries_out and not map_inited:
            if time() > (start + tries * polling):
                map_inited = self.driver.execute_script('return !!map.getZoom')
                tries_out = tries >= request_num
                tries += 1

    def drag(self, x_offset, y_offset):
        center = self.driver.execute_script(GetScripts.getContainerCenter)
        actions = ActionChains(self.driver)
        actions.move_to_element_with_offset(self.element, int(center['x']), int(center['y']))
        actions.click_and_hold()
        actions.move_by_offset(x_offset, y_offset)
        actions.release().perform()

    def info_elem(self):
        return self.driver.find_element_by_css_selector(self.selectors['info'])