Contents

require "delegate"
      
module Chop
  class Table < SimpleDelegator
    def self.diff! table, &block
      klass = Class.new(self) do
        class_attribute :cell_transformers
        self.cell_transformers = []

        def self.cell index, &block
          cell_transformers[index] = block
        end

        def cell_to_text cell, index
          if transformer = cell_transformers[index]
            transformer.call cell
          else
            super
          end
        end

        instance_eval &block if block_given?
      end

      klass.new.diff! table
    end

    def initialize selector = "table", session: Capybara.current_session
      super(session)
      @selector = selector
    end

    def header_elements
      rows("thead")
    end

    def header
      header_elements.collect do |row|
        row.all(:xpath, "./*").map(&:text)
      end
    end

    def body_elements
      rows("tbody")
    end

    def body
      body_elements.collect do |row|
        row_to_text(row)
      end
    end

    def to_a
      header + body
    end

    def normalized_to_a
      raw = to_a
      max = raw.map(&:count).max
      raw.select { |row| row.count == max }
    end

    def diff! table
      table.diff! normalized_to_a
    end

    private

    def rows parent = nil
      node.all("#{parent} tr")
    end

    def node
      @node ||= find(@selector)
    end

    def row_to_text row
      row.all(:xpath, "./*").map.with_index do |cell, index|
        cell_to_text cell, index
      end
    end

    def cell_to_text cell, index
      text = cell.text
      if text.blank? and image = cell.all("img").first
        text = image["alt"]
      end
      text
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
chop-0.8.0	lib/chop/table.rb
chop-0.7.0	lib/chop/table.rb