Merge pull request #28 from rjocoleman/nil

convert nil to empty string
jm · Feb 17, 2014 · deeb624 · deeb624
2 parents b47f4dc + 3bf22ea
commit deeb624
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 4 deletions.
diff --git a/lib/toml/generator.rb b/lib/toml/generator.rb
@@ -51,14 +51,20 @@ def visit(hash, path = "")
         if key.include? '.'
           raise SyntaxError, "Periods are not allowed in keys (failed on key: #{key.inspect})"
         end
-        @body += "#{key} = #{format(val)}\n"
+        unless val.nil?
+          @body += "#{key} = #{format(val)}\n"
+        end
       end
       @body += "\n" unless other_pairs.empty?
 
       # Then deal with sub-hashes
       hash_pairs.each do |pair|
         key, hash = pair
-        visit(hash, (path.empty? ? key : [path, key].join(".")))
+        if hash.empty?
+          @body += "[#{path.empty? ? key : [path, key].join(".")}]\n"
+        else
+          visit(hash, (path.empty? ? key : [path, key].join(".")))
+        end
       end
     end#visit
 

diff --git a/test/test_generator.rb b/test/test_generator.rb
@@ -17,9 +17,11 @@ def setup
       "key" => {
         "group" => {
           "value" => "lol"
-        }
+        },
+        "nil_table" => {}
       },
-      "date" => DateTime.now
+      "date" => DateTime.now,
+      "nil" => nil
     }
 
   end
@@ -34,6 +36,9 @@ def test_generator
     original_date = doc.delete "date"
     parsed_date = doc_parsed.delete "date"
 
+    # removing the nil value
+    remove_nil = doc.delete "nil"
+
     assert_equal doc, doc_parsed
     assert_equal original_date.to_time.to_s, parsed_date.to_time.to_s
   end