# Copyright (c) 2013 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import os

import ap_spec
import trendnet_ap_configurator

from selenium.common.exceptions import WebDriverException

class Trendnet692grAPConfigurator(trendnet_ap_configurator.
                                  TrendnetAPConfigurator):
    """Derived class to control the Trendnet TEW-692GR."""


    def _alert_handler(self, alert):
        """Checks for any modal dialogs which popup to alert the user and
        either raises a RuntimeError or ignores the alert.

        @param alert: The modal dialog's contents.
        """
        text = alert.text
        if 'Please input 5 or 13 characters of WEP key1' in text:
            alert.accept()
            raise RuntimeError(text)


    def get_number_of_pages(self):
        return 2


    def save_page(self, page_number):
        """Save the given page.

        @param page_number: the page to save.
        """
        xpath = ('//input[@class="button_submit" and @value="Apply"]')
        element = self.wait_for_object_by_xpath(xpath)
        if element and element.is_displayed():
            self.click_button_by_xpath(xpath, alert_handler=self._alert_handler)
            # Reboot the device.
            reboot_button = '//input[@value="Reboot the Device"]'
            if self.object_by_xpath_exist(reboot_button):
                self.click_button_by_xpath(reboot_button)
                self.wait_for_progress_bar()
        else:
            pass


    def get_supported_bands(self):
        return [{'band': ap_spec.BAND_2GHZ,
                 'channels': range(1, 12)},
                {'band': ap_spec.BAND_5GHZ,
                 'channels':[36, 40, 44, 48, 149, 153, 157, 161]}]


    def get_supported_modes(self):
        return [{'band': ap_spec.BAND_2GHZ,
                 'modes': [ap_spec.MODE_N,
                           ap_spec.MODE_B | ap_spec.MODE_G,
                           ap_spec.MODE_B | ap_spec.MODE_G | ap_spec.MODE_N]},
                {'band': ap_spec.BAND_5GHZ,
                 'modes': [ap_spec.MODE_A | ap_spec.MODE_N, ap_spec.MODE_A]}]


    def navigate_to_page(self, page_number):
        """Navigates to the given page.

        @param page_number: the page to navigate to.
        """
        # All settings are on the same page, so we always open the config page
        if self.current_band == ap_spec.BAND_2GHZ:
            if page_number == 1:
                page_url = os.path.join(self.admin_interface_url ,
                                        'wireless/basic.asp')
            elif page_number == 2:
                page_url = os.path.join(self.admin_interface_url ,
                                        'wireless/security.asp')
            else:
                raise RuntimeError('Invalid page number passed. Number of pages'
                                   '%d, page value sent was %d' %
                                   (self.get_number_of_pages(), page_number))
        elif self.current_band == ap_spec.BAND_5GHZ:
            if page_number == 1:
                page_url = os.path.join(self.admin_interface_url ,
                                        'wireless2/basic.asp')
            elif page_number == 2:
                page_url = os.path.join(self.admin_interface_url ,
                                        'wireless2/security.asp')
            else:
                raise RuntimeError('Invalid page number passed. Number of pages'
                                   '%d, page value sent was %d' %
                                    (self.get_number_of_pages(), page_number))
        else:
            raise RuntimeError('Incorrect band band = %s' % self.current_band)
        try:
            self.get_url(page_url, page_title='TEW-692GR')
        except WebDriverException, e:
            if "unexpected alert open" in str(e):
                alert = self.driver.switch_to_alert()
                self._alert_handler(alert)
            else:
                raise WebDriverException(str(e))


    def is_update_interval_supported(self):
        """
        Returns True if setting the PSK refresh interval is supported.

        @return True is supported; False otherwise
        """
        return True


    def _set_mode(self, mode, band=None):
        # Create the mode to popup item mapping
        mode_mapping = {ap_spec.MODE_B | ap_spec.MODE_G | ap_spec.MODE_N:
                        '2.4GHz 802.11 b/g/n mixed mode',
                        ap_spec.MODE_B | ap_spec.MODE_G:
                        '2.4GHz 802.11 b/g mixed mode',
                        ap_spec.MODE_N: '2.4GHz 802.11 n only'}

        if self.current_band == ap_spec.BAND_5GHZ:
            mode_mapping = {ap_spec.MODE_A: '5GHz 802.11 a only',
                            ap_spec.MODE_A | ap_spec.MODE_N:
                            '5GHz 802.11 a/n mixed mode'}
        mode_name = ''
        if mode in mode_mapping.keys():
            mode_name = mode_mapping[mode]
        else:
            raise RuntimeError('The mode selected %d is not supported by router'
                               ' %s.', hex(mode), self.name)
        self.select_item_from_popup_by_id(mode_name, 'wirelessmode',
                                          wait_for_xpath='id("wds_mode")')


    def set_band(self, band):
        if band == ap_spec.BAND_5GHZ:
            self.current_band = ap_spec.BAND_5GHZ
        elif band == ap_spec.BAND_2GHZ:
            self.current_band = ap_spec.BAND_2GHZ
        else:
            raise RuntimeError('Invalid band sent %s' % band)


    def set_radio(self, enabled=True):
        # TODO: Implement the RADIO button for this AP
        return None


    def _set_channel(self, channel):
        position = self._get_channel_popup_position(channel)
        channel_choices_2GHZ = ['2412MHz (Channel 1)', '2417MHz (Channel 2)',
                                '2422MHz (Channel 3)', '2427MHz (Channel 4)',
                                '2432MHz (Channel 5)', '2437MHz (Channel 6)',
                                '2442MHz (Channel 7)', '2447MHz (Channel 8)',
                                '2452MHz (Channel 9)', '2457MHz (Channel 10)',
                                '2462MHz (Channel 11)']
        channel_choices_5GHZ = ['5180MHz (Channel 36)', '5200MHz (Channel 40)',
                                '5220MHz (Channel 44)', '5240MHz (Channel 48)',
                                '5745MHz (Channel 149)',
                                '5765MHz (Channel 153)',
                                '5785MHz (Channel 157)',
                                '5805MHz (Channel 161)']
        if self.current_band == ap_spec.BAND_2GHZ:
            self.select_item_from_popup_by_id(channel_choices_2GHZ[position],
                                              'sz11gChannel')
        else:
            self.select_item_from_popup_by_id(channel_choices_5GHZ[position],
                                              'sz11aChannel')


    def set_security_disabled(self):
        self.add_item_to_command_list(self._set_security_disabled, (), 2, 1000)


    def _set_security_disabled(self):
        self.wait_for_object_by_id('security_mode')
        self.select_item_from_popup_by_id('Disable', 'security_mode')


    def _set_security_wpapsk(self, security, shared_key, update_interval=1800):
        self.wait_for_object_by_id('security_mode')
        if security == ap_spec.SECURITY_TYPE_WPAPSK:
            wpa_item = 'WPA-PSK'
        else:
            wpa_item = 'WPA2-PSK'
        self.select_item_from_popup_by_id(wpa_item, 'security_mode',
                                          wait_for_xpath='id("passphrase")')
        self.set_content_of_text_field_by_id(shared_key, 'passphrase')
        self.set_content_of_text_field_by_id(update_interval,
                                             'keyRenewalInterval')


    def set_security_wep(self, key_value, authentication):
        self.add_item_to_command_list(self._set_security_wep,
                                      (key_value, authentication), 2, 900)


    def _set_security_wep(self, key_value, authentication):
        self.wait_for_object_by_id('security_mode')
        self.select_item_from_popup_by_id('WEP-OPEN', 'security_mode')
        self.select_item_from_popup_by_id('ASCII', 'WEP1Select')
        self.set_content_of_text_field_by_id(key_value, 'WEP1',
                                             abort_check=True)