option to set header capitalization

arches · Jul 7, 2014 · e5fc964 · e5fc964
1 parent e382331
commit e5fc964
Show file tree

Hide file tree

Showing 5 changed files with 34 additions and 1 deletion.
diff --git a/features/configuring_output.feature b/features/configuring_output.feature
@@ -55,3 +55,18 @@ Feature: Configuring output
     ------
     Ryan
     """
+
+  Scenario: Setting a lowecase column name
+    Given a class named Blog
+
+    Given Blog has attributes title, author
+
+    When I instantiate a Blog with {:title => "post!", :author => 'Ryan'}
+    And I configure capitalize_headers with false
+    And table_print Blog, {:author => {:display_name => "Wom Bat"}}
+    Then the output should contain
+    """
+    Wom Bat
+    -------
+    Ryan
+    """
diff --git a/features/support/step_definitions/before.rb b/features/support/step_definitions/before.rb
@@ -1,3 +1,4 @@
 Before do
   Sandbox.cleanup!
+  TablePrint::Config.clear(:capitalize_headers)
 end
diff --git a/features/support/step_definitions/steps.rb b/features/support/step_definitions/steps.rb
@@ -62,6 +62,10 @@
   TablePrint::Config.set(:multibyte, [value == "true"])
 end
 
+When /^I configure capitalize_headers with (.*)$/ do |value|
+  TablePrint::Config.set(:capitalize_headers, [value == "true"])
+end
+
 When /^configure (.*) with (.*)$/ do |klass, config|
   klass = Sandbox.const_get_from_string(klass)
   TablePrint::Config.set(klass, eval(config))

diff --git a/lib/table_print/config.rb b/lib/table_print/config.rb
@@ -4,11 +4,13 @@ class Config
     DEFAULT_MAX_WIDTH = 30
     DEFAULT_TIME_FORMAT = "%Y-%m-%d %H:%M:%S"
     DEFAULT_IO = STDOUT
+    DEFAULT_CAPITALIZE_HEADERS = true
 
     @@max_width = DEFAULT_MAX_WIDTH
     @@time_format = DEFAULT_TIME_FORMAT
     @@multibyte = false
     @@io = STDOUT
+    @@capitalize_headers = true
 
     @@klasses = {}
 
@@ -57,6 +59,14 @@ def self.time_format=(format)
       @@time_format = format
     end
 
+    def self.capitalize_headers
+      @@capitalize_headers
+    end
+
+    def self.capitalize_headers=(caps)
+      @@capitalize_headers = caps
+    end
+
     def self.io
       @@io
     end

diff --git a/lib/table_print/row_group.rb b/lib/table_print/row_group.rb
@@ -73,7 +73,10 @@ def header
         f.format(column.name)
       end
 
-      padded_names.join(" | ").upcase
+      header_string = padded_names.join(" | ")
+      header_string.upcase! if TablePrint::Config.capitalize_headers
+
+      header_string
     end
 
     def add_formatter(name, formatter)