Sha256: e8f3554e7856cf60406ac0d2649ce0f349a64605dde69b350cb17d5134124025

Contents?: true

Size: 1.62 KB

Versions: 12

Compression:

Stored size: 1.62 KB

Contents

class Puppeteer::CustomQueryHandler
  # @param query_one [String] JS function (element: Element | Document, selector: string) => Element | null;
  # @param query_all [String] JS function (element: Element | Document, selector: string) => Element[] | NodeListOf<Element>;
  def initialize(query_one: nil, query_all: nil)
    @query_one = query_one
    @query_all = query_all
  end

  def query_one(element, selector)
    unless @query_one
      raise NotImplementedError.new("#{self.class}##{__method__} is not implemented.")
    end

    handle = element.evaluate_handle(@query_one, selector)
    element = handle.as_element

    if element
      return element
    end
    handle.dispose
    nil
  end

  def wait_for(dom_world, selector, visible: nil, hidden: nil, timeout: nil, root: nil)
    unless @query_one
      raise NotImplementedError.new("#{self.class}##{__method__} is not implemented.")
    end

    dom_world.send(:wait_for_selector_in_page, @query_one, selector, visible: visible, hidden: hidden, timeout: timeout, root: root)
  end

  def query_all(element, selector)
    unless @query_all
      raise NotImplementedError.new("#{self.class}##{__method__} is not implemented.")
    end

    handles = element.evaluate_handle(@query_all, selector)
    properties = handles.properties
    handles.dispose
    properties.values.map(&:as_element).compact
  end

  def query_all_array(element, selector)
    unless @query_all
      raise NotImplementedError.new("#{self.class}##{__method__} is not implemented.")
    end

    handles = element.evaluate_handle(@query_all, selector)
    handles.evaluate_handle('(res) => Array.from(res)')
  end
end

Version data entries

12 entries across 12 versions & 1 rubygems

Version Path
puppeteer-ruby-0.43.1 lib/puppeteer/custom_query_handler.rb
puppeteer-ruby-0.43.0 lib/puppeteer/custom_query_handler.rb
puppeteer-ruby-0.42.0 lib/puppeteer/custom_query_handler.rb
puppeteer-ruby-0.41.0 lib/puppeteer/custom_query_handler.rb
puppeteer-ruby-0.40.7 lib/puppeteer/custom_query_handler.rb
puppeteer-ruby-0.40.6 lib/puppeteer/custom_query_handler.rb
puppeteer-ruby-0.40.5 lib/puppeteer/custom_query_handler.rb
puppeteer-ruby-0.40.4 lib/puppeteer/custom_query_handler.rb
puppeteer-ruby-0.40.3 lib/puppeteer/custom_query_handler.rb
puppeteer-ruby-0.40.2 lib/puppeteer/custom_query_handler.rb
puppeteer-ruby-0.40.1 lib/puppeteer/custom_query_handler.rb
puppeteer-ruby-0.40.0 lib/puppeteer/custom_query_handler.rb