aboutsummaryrefslogtreecommitdiffstats
path: root/actionpack/lib/action_view/helpers/capture_helper.rb
diff options
context:
space:
mode:
Diffstat (limited to 'actionpack/lib/action_view/helpers/capture_helper.rb')
-rw-r--r--actionpack/lib/action_view/helpers/capture_helper.rb18
1 files changed, 13 insertions, 5 deletions
diff --git a/actionpack/lib/action_view/helpers/capture_helper.rb b/actionpack/lib/action_view/helpers/capture_helper.rb
index f9a73a4c8d..837305d96c 100644
--- a/actionpack/lib/action_view/helpers/capture_helper.rb
+++ b/actionpack/lib/action_view/helpers/capture_helper.rb
@@ -31,11 +31,11 @@ module ActionView
# </body></html>
#
def capture(*args, &block)
- @output_buffer, old_buffer = '', @output_buffer
- yield *args
- @output_buffer
- ensure
- @output_buffer = old_buffer
+ if @output_buffer
+ with_temporary_output_buffer { yield *args }
+ else
+ block.call(*args)
+ end
end
# Calling content_for stores a block of markup in an identifier for later use.
@@ -119,6 +119,14 @@ module ActionView
end
private
+ def with_temporary_output_buffer
+ @output_buffer, old_buffer = '', @output_buffer
+ yield
+ @output_buffer
+ ensure
+ @output_buffer = old_buffer
+ end
+
def erb_content_for(name, &block)
ivar = "@content_for_#{name}"
instance_variable_set(ivar, "#{instance_variable_get(ivar)}#{capture(&block)}")