table-formatter 0.1.1 → 0.1.2

data/lib/table-formatter.rb

@@ -2,10 +2,14 @@
 
 class TableFormatter
 
-  attr_accessor :source, :labels
+  attr_accessor :source, :labels, :border
   
-  def initialize()    
+  def initialize(opt={})    
+    o = {source: nil, labels: nil, border: true}.merge(opt)
     super()
+    @source = o[:source]
+    @labels = o[:labels]
+    @border = o[:border]
   end
   
   def display(width=nil)
@@ -16,13 +20,15 @@ class TableFormatter
 
     records = format_rows(a.clone, column_widths)
 
-    div = '-' * records[0].length + "\n"
+    div =  (border == true ? '-' : ' ') * 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
 
+  alias to_s display
+
   private
   
   def tabulate(a)
@@ -37,10 +43,10 @@ class TableFormatter
   end
 
   def format_cols(row, col_widths)
-    buffer = '|'
+    bar = border == true ? '|' : ' '
+    buffer = bar
     row.each_with_index do |col, i|
-      puts col.inspect
-      buffer += ' ' + col.ljust(col_widths[i] + 2) + '|'    
+      buffer += ' ' + col.ljust(col_widths[i] + 2) + bar    
     end
     buffer
   end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: table-formatter
 version: !ruby/object:Gem::Version 
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors: []