class Selenium::WebDriver::TargetLocator

def active_element
  @bridge.switch_to_active_element
end

def alert
  Alert.new(@bridge)
end

def default_content
  @bridge.switch_to_default_content
end

def frame(id)
  @bridge.switch_to_frame id
end

def initialize(bridge)
  @bridge = bridge
end

def parent_frame
  @bridge.switch_to_parent_frame
end

def window(id)
  if block_given?
    original = begin
      @bridge.window_handle
    rescue Error::NoSuchWindowError
      nil
    end
    unless @bridge.window_handles.include? id
      raise Error::NoSuchWindowError, "The specified identifier '#{id}' is not found in the window handle list"
    end
    @bridge.switch_to_window id
    begin
      returned = yield
    ensure
      current_handles = @bridge.window_handles
      original = current_handles.first unless current_handles.include? original
      @bridge.switch_to_window original
      returned
    end
  else
    @bridge.switch_to_window id
  end
end