jnicklas/turnip

module Turnip
  class Table
    class WidthMismatch < StandardError
      def initialize(expected, actual)
        super("Expected the table to be #{expected} columns wide, got #{actual}")
      end
    end

    class ColumnNotExist < StandardError
      def initialize(column_name)
        super("The column named \"#{column_name}\" does not exist")
      end
    end

    attr_reader :raw
    alias_method :to_a, :raw

    include Enumerable

    def initialize(raw)
      @raw = raw
    end

    def headers
      raw.first
    end

    def rows
      raw.drop(1)
    end

    def hashes
      rows.map { |row| Hash[headers.zip(row)] }
    end

    def rows_hash
      raise WidthMismatch.new(2, width) unless width == 2
      transpose.hashes.first
    end

    def transpose
      self.class.new(raw.transpose)
    end

    def each
      raw.each { |row| yield(row) }
    end

    def map_column!(name, strict = true)
      index = headers.index(name.to_s)
      if index.nil?
        raise ColumnNotExist.new(name) if strict
      else
        rows.each { |row| row[index] = yield(row[index]) }
      end
    end

    private

    def width
      raw[0].size
    end
  end
end