class Selenium::WebDriver::BiDi::BrowsingContext

def close
  @bidi.send_cmd('browsingContext.close', context: @id)
end

def create(type, reference_context)
  @bidi.send_cmd('browsingContext.create', type: type.to_s, referenceContext: reference_context)
end

def get_tree(max_depth: nil)
  result = @bidi.send_cmd('browsingContext.getTree', root: @id, maxDepth: max_depth).dig('contexts', 0)
  BrowsingContextInfo.new(
    id: result['context'],
    url: result['url'],
    children: result['children'],
    parent_context: result['parent']
  )
end

def initialize(driver:, browsing_context_id: nil, type: nil, reference_context: nil)
  unless driver.capabilities.web_socket_url
    raise Error::WebDriverError,
          'WebDriver instance must support BiDi protocol'
  end
  unless type.nil? || %i[window tab].include?(type)
    raise ArgumentError,
          "Valid types are :window & :tab. Received: #{type.inspect}"
  end
  @bidi = driver.bidi
  @id = browsing_context_id.nil? ? create(type, reference_context)['context'] : browsing_context_id
end

def navigate(url:, readiness_state: nil)
  unless readiness_state.nil? || READINESS_STATE.key?(readiness_state)
    raise ArgumentError,
          "Valid readiness states are :none, :interactive & :complete. Received: #{readiness_state.inspect}"
  end
  navigate_result = @bidi.send_cmd('browsingContext.navigate', context: @id, url: url,
                                                               wait: READINESS_STATE[readiness_state])
  NavigateResult.new(
    url: navigate_result['url'],
    navigation_id: navigate_result['navigation']
  )
end