senny/corner_stones

require 'corner_stones/table/selectable_rows'
require 'corner_stones/table/deletable_rows'
require 'corner_stones/table/whitespace_filter'

module CornerStones

  class Table

    class MissingRowError < StandardError; end

    include Capybara::DSL

    def initialize(scope, options = {})
      @scope = scope
      @data_selector = options.fetch(:data_selector) { 'td' }
      @options = options
    end

    def row(options)
      rows.detect { |row|
        identity = row.attributes.select { |key, value| options.has_key?(key) }
        identity == options
      } or raise MissingRowError, "no row with '#{options.inspect}'\n\ngot:#{rows}"
    end

    def empty?
      rows.empty?
    end

    def hashes
      rows.map{|row| row.attributes}
    end

    # returns an <tt>Array</tt> of values for the given column.
    def values(column_header)
      rows.map { |row| row[column_header] }
    end

    def rows
      raw_rows.map do |row|
        build_row(row)
      end
    end

    def raw_rows
      within @scope do
        all('tbody > tr')
      end
    end

    def build_row(node)
      Row.new(node, attributes_for_row(node))
    end
    protected :build_row

    def headers
      @options[:headers] || detect_table_headers
    end

    def detect_table_headers
      within @scope do
        all('thead th').map(&:text)
      end
    end

    def attributes_for_row(row)
      row_data = {}
      headers.each.with_index.with_object(row_data) do |(header, index), row_data|
        augment_row_with_cell(row_data, row, index, header)
      end
      row_data
    end

    def augment_row_with_cell(row_data, row, index, header)
      data = row.all(@data_selector)
      cell = data[index]
      row_data[header] = value_for_cell(cell)
    end

    def value_for_cell(cell)
      cell.text unless cell.nil?
    end

    Row = Struct.new(:node, :attributes) do
      include Capybara::DSL

      def [](key)
        attributes[key]
      end
    end
  end

end