Contents

module RenderAsMarkdown
  class Table

    attr_accessor :columns, :rows

    def initialize column_titles = []
      @columns = column_titles.map{|title| Column.new title}
      @rows = []
    end

    def add_row row
      # TODO: ensure element count of row is == element count of columns

      # add row to rows
      @rows << row

      # iterate through columns and row, add each row to their column
      @columns.zip(row).each {|col, string| col.add_row string}
    end

    alias_method  '<<', :add_row

    def render
      # join all column headers
      table = @columns.map(&:render_title).join( '|' ) << "\n"

      # join all column lines
      table << @columns.map(&:render_line).join( '|' ) << "\n"

      # join all columns for all rows
      @rows.each_with_index do |row, i|
        table << @columns.map {|col| col.render_row i}.join( '|' ) << "\n"
      end

      # return table
      table
    end
  end


  class Column

    attr_accessor :title, :width, :rows

    def initialize title
      @rows = []
      @title = title
      @width = title.to_s.length
    end

    def render_title
      @title.ljust @width
    end

    def render_line
      '-'.ljust @width, '-'
    end

    def render_row row_number
      @rows[row_number].ljust @width
    end

    def add_row string
      string ||= ''
      @rows << string
      update_width string.to_s.length
    end

    def update_width length
      @width = length if @width < length
    end

  end

end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
render-as-markdown-0.0.3	lib/render-as-markdown/markdown-table.rb