# 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 ap_spec
import trendnet_ap_configurator


class Trendnet731brAPConfigurator(trendnet_ap_configurator.
                                  TrendnetAPConfigurator):
    """Derived class to control the Trendnet TEW-731BR."""

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

        @param page_number: the page to navigate to.
        """
        self.navigate_to_login_page()
        if page_number == 1:
            page_url = ''.join([self.admin_interface_url,
                                  "/wireless_basic.htm"])
            self.get_url(page_url, page_title='TRENDNET')
        if page_number == 2:
            page_url = ''.join([self.admin_interface_url,
                                  "/wireless_auth.htm"])
            self.get_url(page_url, page_title='TRENDNET')


    def navigate_to_login_page(self):
        """
        We need to login first in order to configure settings.
        """
        self.get_url(self.admin_interface_url, page_title='Login')
        self.wait_for_object_by_id('login_n')
        self.set_content_of_text_field_by_id('admin', 'login_n',
                                                 abort_check=True)
        self.set_content_of_text_field_by_id('password', 'login_pass',
                                                abort_check=True)
        self.click_button_by_xpath('//input[@value="Login"]')


    def _set_ssid(self, ssid):
        self.set_content_of_text_field_by_id(ssid, 'show_ssid')
        self._ssid = ssid


    def save_page(self, page_number):
        """Save the settings of the passed in page.

        @param page_number: the page to save.
        """
        if page_number == 1:
            xpath = ('//input[@type="button" and @value="Apply"]')
        elif page_number == 2:
            xpath = ('//input[@type="button" and @value="Apply"]')
        self.click_button_by_xpath(xpath, alert_handler=self._alert_handler)
        xpath = ('//input[@type="button" and @value="Continue"]')
        self.click_button_by_xpath(xpath, alert_handler=self._alert_handler)


    def _set_security_disabled(self):
        self.wait_for_object_by_xpath('//select[@name="wep_type"]')
        self.select_item_from_popup_by_id(' Disable ','wep_type')


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


    def _set_mode(self, mode, band=None):
        # Different bands are not supported so we ignore.
        # Create the mode to popup item mapping
        mode_mapping = {ap_spec.MODE_B | ap_spec.MODE_G | ap_spec.MODE_N:
                        '2.4Ghz 802.11b/g/n mixed mode',
                        ap_spec.MODE_N: '2.4Ghz 802.11n only mode',
                        ap_spec.MODE_B: '2.4Ghz 802.11b only mode',
                        ap_spec.MODE_G: '2.4Ghz 802.11g only mode',
                        ap_spec.MODE_B | ap_spec.MODE_G:
                        '2.4Ghz 802.11b/g 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, 'dot11_mode')

    def get_supported_bands(self):
        return [{'band': ap_spec.BAND_2GHZ, 'channels': range(1, 12)}]


    def _set_channel(self, channel):
        position = self._get_channel_popup_position(channel)
        channel_choices = ['1', '2', '3', '4', '5',
                           '6', '7', '8', '9', '10', '11']
        self.select_item_from_popup_by_id(channel_choices[position],
                                          'wlan0_channel_t')

    def _set_visibility(self, visible=True):
        # value=1 is visible; value=0 is invisible
        int_value = int(visible)
        xpath = ('//input[@value="%d" and @name="wlan0_ssid_broadcast"]' %
                  int_value)
        self.click_button_by_xpath(xpath, alert_handler=self._alert_handler)


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

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


    def _set_security_wep(self, key_value, authentication):
        self.wait_for_object_by_xpath('//select[@name="wep_type"]')
        self.select_item_from_popup_by_id(' WEP ','wep_type')
        # This router doesn not support ASCII passphrase for
        # generating hex key.Converting ASCII to hex before setting
        self.set_content_of_text_field_by_id(key_value.encode("hex"),
                                             'key1_64_hex')


    def _set_security_wpapsk(self, security, shared_key, update_interval=1800):
        self.wait_for_object_by_xpath('//select[@name="wep_type"]')
        if security == ap_spec.SECURITY_TYPE_WPAPSK:
            self.select_item_from_popup_by_id(' WPA ','wep_type')
        else:
            self.select_item_from_popup_by_id(' WPA2 ','wep_type')
        self.set_content_of_text_field_by_id(shared_key,
                                             'wlan0_psk_pass_phrase')
        self.set_content_of_text_field_by_id(shared_key, 'wpapsk2')


    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.

        Args:
          alert: The modal dialog's contents.
        """
        text = alert.text
        if 'To disable SSID Broadcast will cause WPS not work' in text:
            alert.accept()
        else:
            raise RuntimeError('We have an unhandled alert: %s' % text)