Adds if statement to code

nesquena · Apr 13, 2011 · 8136205 · 8136205
1 parent f1ddd6a
commit 8136205
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/lib/rabl/builder.rb b/lib/rabl/builder.rb
@@ -52,8 +52,10 @@ def attribute(*args)
 
     # Creates an arbitrary code node that is included in the json output
     # code(:foo) { "bar" }
-    def code(name, &block)
-      @_result[name] = block.call(@_object)
+    # code(:foo, :if => lambda { |m| m.foo.present? })
+    def code(name, options={}, &block)
+      append_node = options[:if].nil? || (options[:if].respond_to?(:call) && options[:if].call(@_object))
+      @_result[name] = block.call(@_object) if append_node
     end
 
     # Creates a child node that is included in json output