from selenium.webdriver import DesiredCapabilities, ChromeOptions, FirefoxOptions

import nerodia


class Capabilities(object):
    DEFAULT_URL = 'http://127.0.0.1:{}/wd/hub'

    def __init__(self, browser, **options):
        self.options = options.copy()
        nerodia.logger.info('Creating Browser instance of {} with user provided options: '
                            '{}'.format(browser, options))

        if browser == 'remote' and 'browser' in self.options:
            self.browser = self.options.pop('browser')
        elif browser == 'remote' and 'desired_capabilities' in self.options:
            self.browser = self.options['desired_capabilities'].get('browserName')
        else:
            self.browser = browser

        if browser == 'remote' or options.get('url'):
            self.selenium_browser = 'remote'
        else:
            self.selenium_browser = browser

        self.selenium_opts = {}

    @property
    def kwargs(self):
        self._process_arguments()
        return self.selenium_opts

    # private

    def _process_arguments(self):
        url = None
        port = self.options.pop('port', None)
        if port:
            url = self.DEFAULT_URL.format(port)

        url = self.options.pop('url', url)
        if url:
            self.selenium_opts['command_executor'] = url

        self._process_browser_options()
        self._process_capabilities()
        nerodia.logger.info('Creating Browser instance with Nerodia processed options: '
                            '{}'.format(self.selenium_opts))

    def _process_browser_options(self):
        browser_options = self.options.pop('options', {})

        if self.selenium_browser == 'chrome':
            self._process_chrome_options(browser_options)
        elif self.selenium_browser == 'firefox':
            self._process_firefox_options(browser_options)
        elif self.selenium_browser == 'safari':
            self._process_safari_options(browser_options)
        elif self.selenium_browser == 'ie':
            self._process_ie_options(browser_options)
        elif self.selenium_browser == 'remote':
            self._process_remote_options(browser_options)

    def _process_chrome_options(self, opts):
        if isinstance(opts, ChromeOptions):
            options = opts
        else:
            options = ChromeOptions()
            if 'args' in opts:
                for arg in opts.pop('args'):
                    options.add_argument(arg)
            if opts.pop('headless', None) or self.options.pop('headless', None):
                options.headless = True
            if 'binary' in opts or 'binary_location' in opts:
                options.binary_location = opts.pop('binary') or opts.pop('binary_location')
            if 'debugger_address' in opts:
                options.debugger_address = opts.pop('debugger_address')
            if 'extensions' in opts:
                for path in opts.pop('extensions'):
                    options.add_extension(path)
            if 'encoded_extensions' in opts:
                for string in opts.pop('encoded_extensions'):
                    options.add_encoded_extension(string)
            if 'experimental_options' in opts:
                for name, value in opts.pop('experimental_options').items():
                    options.add_experimental_option(name, value)
        self.selenium_opts['options'] = options

    def _process_firefox_options(self, opts):
        if isinstance(opts, FirefoxOptions):
            options = opts
        else:
            options = FirefoxOptions()
            if 'args' in opts:
                for arg in opts.pop('args'):
                    options.add_argument(arg)
            if opts.pop('headless', None) or self.options.pop('headless', None):
                options.headless = True
            if 'binary' in opts or 'binary_location' in opts:
                options.binary = opts.pop('binary') or opts.pop('binary_location')
            if 'prefs' in opts:
                for name, value in opts.pop('prefs').items():
                    options.set_preference(name, value)
            if 'proxy' in opts:
                options.proxy = opts.pop('proxy')
            if 'profile' in opts:
                options.profile = opts.pop('profile')
            if 'log_level' in opts:
                options.log.level = opts.pop('log_level')
        self.selenium_opts['options'] = options

    def _process_safari_options(self, opts):
        if 'technology_preview' in opts:
            self.selenium_opts['executable_path'] = \
                '/Applications/Safari Technology Preview.app/Contents/MacOS/safaridriver'

    def _process_ie_options(self, opts):
        from selenium.webdriver.ie.options import Options
        if isinstance(opts, Options):
            options = opts
        else:
            options = Options()
            if 'args' in opts:
                for arg in opts.pop('args'):
                    options.add_argument(arg)
        self.selenium_opts['options'] = options

    def _process_remote_options(self, opts):
        if self.browser == 'chrome':
            caps = opts.pop('desired_capabilities', DesiredCapabilities.CHROME.copy())
            self._process_chrome_options(opts)
            self.selenium_opts['desired_capabilities'] = caps

        if self.browser == 'firefox':
            caps = opts.pop('desired_capabilities', DesiredCapabilities.FIREFOX.copy())
            self._process_firefox_options(opts)
            self.selenium_opts['desired_capabilities'] = caps

        if self.browser == 'safari':
            caps = opts.pop('desired_capabilities', DesiredCapabilities.SAFARI.copy())
            if 'technology_preview' in opts:
                caps['safari.options'] = {'technologyPreview': opts.pop('technology_preview')}
            self.selenium_opts['desired_capabilities'] = caps

        if self.browser == 'ie':
            caps = opts.pop('desired_capabilities', DesiredCapabilities.INTERNETEXPLORER.copy())
            self._process_ie_options(opts)
            self.selenium_opts['desired_capabilities'] = caps

        if self.browser == 'edge':
            caps = opts.pop('desired_capabilities', DesiredCapabilities.EDGE.copy())
            self.selenium_opts['desired_capabilities'] = caps

    def _process_capabilities(self):
        caps = self.options.pop('desired_capabilities', None)

        if caps:
            nerodia.logger.warn('You can now pass values directly into nerodia.browser.Browser '
                                'without needing to use desired_capabilities',
                                ids=['use_capabilities'])
            self.selenium_opts.update(self.options)
        else:
            from selenium.webdriver.common.desired_capabilities import DesiredCapabilities
            browser = 'internetexplorer' if self.browser == 'ie' else self.browser
            caps = getattr(DesiredCapabilities, browser.upper())

        if self.browser in ['firefox', 'ie', 'edge']:
            self.selenium_opts['capabilities'] = caps
        else:
            self.selenium_opts['desired_capabilities'] = caps

@@ -1322,16 +1322,181 @@
  = url%0A%0A
+        executable_path = self.options.pop('executable_path', None)%0A        if executable_path:%0A            self.selenium_opts%5B'executable_path'%5D = executable_path%0A%0A
         

allow executable_path to be passed to Browser