table-formatter 0.1.0 → 0.1.1

data/lib/table-formatter.rb

@@ -5,17 +5,22 @@ class TableFormatter
   attr_accessor :source, :labels
   
   def initialize()    
+    super()
   end
   
-  def display
+  def display(width=nil)
     a = @source
     labels = @labels
     column_widths = fetch_column_widths(a)
+    column_widths[-1] -= column_widths.inject(&:+) - width if width
+
     records = format_rows(a.clone, column_widths)
+
     div = '-' * records[0].length + "\n"
     label_buffer = ''    
     label_buffer = format_cols(labels, column_widths) + "\n" + div if labels
     div + label_buffer + records.join("\n") + "\n" + div
+
   end
 
   private
@@ -34,12 +39,28 @@ class TableFormatter
   def format_cols(row, col_widths)
     buffer = '|'
     row.each_with_index do |col, i|
+      puts col.inspect
       buffer += ' ' + col.ljust(col_widths[i] + 2) + '|'    
     end
     buffer
   end
 
   def format_rows(a, col_widths)
+    col_widths[-1] -= col_widths.inject(&:+) - 80
+    a.each_with_index do |x,i|
+      col_rows = wrap(x[-1], col_widths[-1]).split(/\n/)
+      if col_rows.length > 1 then
+        x[-1] = col_rows[0]
+        col_rows[1..-1].map.with_index {|y,j| a.insert(i+1+j, Array.new(x.length - 1,'') + [y])}
+      end
+    end
+
     a.map {|row| format_cols(row, col_widths)}
   end
-end
+
+  def wrap(s, col=80)
+    s.gsub(/(.{1,#{col}})( +|$\n?)|(.{1,#{col}})/,
+      "\\1\\3\n") 
+  end
+
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: table-formatter
 version: !ruby/object:Gem::Version 
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors: []
 
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-05-02 00:00:00 +01:00
+date: 2010-07-04 00:00:00 +01:00
 default_executable: 
 dependencies: []