Add omit_root_container and prefix

Gemfile

@@ -6,6 +6,7 @@ gem 'rake', group: [:development, :test]
 
 group :test do
   gem 'minitest'
+  gem 'minitest-reporters', require: 'minitest/reporters'
   gem 'simplecov'
   gem 'codeclimate-test-reporter'
 end

Readme.markdown

@@ -46,6 +46,8 @@ This will add an ellipses after the first element and truncate values longer tha
 Pizzazz.ify(all_of_the_things, array_limit: 'etc', value_omission: '... (continued)')
 ```
 
+You can also supply `omit_root_container` and `prefix`, a string that's added to the beginning of each line of the output, is really nice for doing custom things. Both options can be used independently as well.
+
 
 ### HTML
 

lib/pizzazz/colorer.rb

@@ -13,11 +13,20 @@ def initialize(object, options = nil)
       @value_limit = options[:value_limit] || 0
       @value_omission = options[:value_omission] || '…'
       @tab = options[:tab] || '  '
+      @prefix = options[:prefix]
+      @omit_root_container = options[:omit_root_container] || false
     end
 
     def ify
       return '' unless @object
-      node(@object)
+
+      # Parse
+      output = node(@object, true)
+      return output unless @prefix
+
+      # Add prefix
+      lines = output.split("\n")
+      @prefix + lines.join("\n#{@prefix}")
     end
 
   private
@@ -33,57 +42,79 @@ def truncate(string)
       (text.length > @value_limit ? text[0...stop] + @value_omission : text).to_s
     end
 
-    def node(object)
+    def node(object, root = false)
+      omit_container = root && @omit_root_container
+
       case object
       when String
         %Q{<span class="string">"#{truncate(::ERB::Util.h(object))}"</span>}
+
       when Time
         %Q{<span class="string">#{object.to_json}</span>}
+
       when TrueClass
         %Q{<span class="constant">true</span>}
+
       when FalseClass
         %Q{<span class="constant">false</span>}
+
       when NilClass
         %Q{<span class="null">null</span>}
+
       when Numeric
         %Q{<span class="number">#{object}</span>}
+
       when Hash
-        if object.length == 0
-          '{}'
+        return omit_container ? '' : '{}' if object.length == 0
+
+        string = if omit_container
+          ''
         else
-          s = "{\n"
           @indent += 1
-          rows = []
-          object.keys.collect(&:to_s).sort.each do |key|
-            value = (object[key] != nil ? object[key] : object[key.to_sym])
-            rows << %Q{#{tab}<span class="string key">"#{key}"</span>: #{node(value)}}
-          end
-          s << rows.join(",\n") + "\n"
+          "{\n"
+        end
+
+        rows = []
+        object.keys.collect(&:to_s).sort.each do |key|
+          value = (object[key] != nil ? object[key] : object[key.to_sym])
+          rows << %Q{#{tab}<span class="string key">"#{key}"</span>: #{node(value)}}
+        end
+        string << rows.join(",\n")
+
+        unless omit_container
           @indent -= 1
-          s << "#{tab}}"
-          s
+          string << "\n#{tab}}"
         end
+
+        string
+
       when Array
-        if object.length == 0
-          '[]'
+        return omit_container ? '' : '[]' if object.length == 0
+        string = if omit_container
+          ''
         else
-          s = "[\n"
           @indent += 1
-          rows = []
-          array = @array_limit > 0 ? object[0...@array_limit] : object
-          array.each do |value|
-            rows << tab + node(value)
-          end
+          "[\n"
+        end
 
-          if @array_limit > 0 and object.length > @array_limit
-            rows << tab + (object[0].is_a?(Hash) ? "{ #{@array_omission} }" : @array_omission)
-          end
+        rows = []
+        array = @array_limit > 0 ? object[0...@array_limit] : object
+        array.each do |value|
+          rows << tab + node(value)
+        end
+
+        if @array_limit > 0 and object.length > @array_limit
+          rows << tab + (object[0].is_a?(Hash) ? "{ #{@array_omission} }" : @array_omission)
+        end
 
-          s << rows.join(",\n") + "\n"
+        string << rows.join(",\n")
+
+        unless omit_container
           @indent -= 1
-          s << "#{tab}]"
-          s
+          string << "\n#{tab}]"
         end
+
+        string
       end
     end
   end

test/test_helper.rb

@@ -5,6 +5,8 @@
 CodeClimate::TestReporter.start
 SimpleCov.start
 
+Minitest::Reporters.use! Minitest::Reporters::DefaultReporter.new
+
 require 'minitest/autorun'
 require 'pizzazz'
 

test/units/colorer_test.rb

@@ -17,6 +17,21 @@ def test_tabs
 }}
   end
 
+  def test_prefix
+    colored = Pizzazz.ify({:foo => 'bar'}, prefix: '**')
+    assert_equal colored, %Q{**{
+**  <span class="string key">"foo"</span>: <span class="string">"bar"</span>
+**}}
+  end
+
+  def test_omit_root_container
+    colored = Pizzazz.ify({:foo => 'bar'}, omit_root_container: true)
+    assert_equal colored, %Q{<span class="string key">"foo"</span>: <span class="string">"bar"</span>}
+
+    colored = Pizzazz.ify([1, 2], omit_root_container: true)
+    assert_equal colored, %Q{<span class="number">1</span>,\n<span class="number">2</span>}
+  end
+
   def test_that_it_truncates_arrays
     colored = Pizzazz.ify({:numbers => [1, 2, 3]}, :array_limit => 2)
     assert_equal colored, %Q{{