# Licensed to the Software Freedom Conservancy (SFC) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The SFC licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License.  You may obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied.  See the License for the
# specific language governing permissions and limitations
# under the License.

module Selenium
  module WebDriver
    module Edge

      #
      # Driver implementation for Microsoft Edge.
      # @api private
      #

      class Driver < WebDriver::Driver
        include DriverExtensions::TakesScreenshot

        def initialize(opts = {})
          opts[:desired_capabilities] ||= Remote::W3C::Capabilities.edge

          unless opts.key?(:url)
            driver_path = opts.delete(:driver_path) || Edge.driver_path
            driver_opts = opts.delete(:driver_opts) || {}
            port = opts.delete(:port) || Service::DEFAULT_PORT

            @service = Service.new(driver_path, port, driver_opts)
            @service.host = 'localhost' if @service.host == '127.0.0.1'
            @service.start
            opts[:url] = @service.uri
          end

          listener = opts.delete(:listener)

          # Edge is mostly using W3C dialect, but a request to
          # create session responds with OSS-like body,
          # so we need to force W3C implementation.
          desired_capabilities = opts.delete(:desired_capabilities)
          bridge = Remote::Bridge.new(opts)
          capabilities = bridge.create_session(desired_capabilities)

          WebDriver.logger.info 'Forcing W3C dialect.'
          @bridge = Remote::W3C::Bridge.new(capabilities, bridge.session_id, opts)
          @bridge.extend Edge::Bridge

          super(@bridge, listener: listener)
        end

        def browser
          :edge
        end

        def quit
          super
        ensure
          @service.stop if @service
        end

      end # Driver
    end # Edge
  end # WebDriver
end # Selenium