Allow conditionnal blocks within template

This commit is contained in:
ccocchi 2012-09-05 18:11:23 +02:00
parent cf39a3f748
commit 3999737daf
6 changed files with 65 additions and 17 deletions

View File

@ -5,6 +5,7 @@ require 'active_support/core_ext/class/attribute_accessors'
require 'rabl-rails/version'
require 'rabl-rails/template'
require 'rabl-rails/condition'
require 'rabl-rails/compiler'
require 'rabl-rails/renderer'

View File

@ -5,7 +5,7 @@ module RablRails
#
class Compiler
def initialize
@glue_count = 0
@i = 0
end
#
@ -76,8 +76,8 @@ module RablRails
#
def glue(data)
return unless block_given?
name = :"_glue#{@glue_count}"
@glue_count += 1
name = :"_glue#{@i}"
@i += 1
@template[name] = sub_compile(data) { yield }
end
@ -114,6 +114,20 @@ module RablRails
@template.merge!(t.source)
end
#
# Provide a conditionnal block
#
# condition(->(u) { u.is_a?(Admin) }) do
# attributes :secret
# end
#
def condition(proc)
return unless block_given?
name = :"_if#{@i}"
@i += 1
@template[name] = Condition.new(proc, sub_compile(nil) { yield })
end
protected
#
@ -138,7 +152,7 @@ module RablRails
return {} unless block_given?
old_template, @template = @template, {}
yield
@template.merge!(:_data => data)
data ? @template.merge!(:_data => data) : @template
ensure
@template = old_template
end

View File

@ -0,0 +1,10 @@
module RablRails
class Condition
attr_reader :proc, :source
def initialize(proc, source)
@proc = proc
@source = source
end
end
end

View File

@ -65,6 +65,11 @@ module RablRails
else # child
object.respond_to?(:each) ? render_collection(object, current_value) : render_resource(object, current_value)
end
when Condition
if instance_exec data, &(value.proc)
output.merge!(render_resource(data, value.source))
end
next output
end
output[key] = out
output

View File

@ -143,6 +143,12 @@ class CompilerTest < ActiveSupport::TestCase
assert_equal 2, t.source[:foo].size
end
test "conditionnal block compile nicely" do
t = @compiler.compile_source(%{ condition(->(u) {}) do attributes :secret end })
assert_instance_of RablRails::Condition, t.source[:_if0]
assert_equal({ :secret => :secret }, t.source[:_if0].source)
end
test "compile with no object" do
t = @compiler.compile_source(%{
object false

View File

@ -115,6 +115,18 @@ class TestJsonRenderer < ActiveSupport::TestCase
assert_equal %q({"users":[]}), render_json_output
end
test "condition blocks are transparent if the condition passed" do
c = RablRails::Condition.new(->(u) { true }, { :name => :name })
@template.source = { :_if0 => c }
assert_equal %q({"name":"foobar"}), render_json_output
end
test "condition blocks are ignored if the condition is not met" do
c = RablRails::Condition.new(->(u) { false }, { :name => :name })
@template.source = { :_if0 => c }
assert_equal %q({}), render_json_output
end
test "render object with root node" do
@template.root_name = :author
@template.source = { :id => :id, :name => :name }