Update capture per review

sinatra · Dec 8, 2018 · a3339e0 · a3339e0
1 parent 5913bec
commit a3339e0
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/sinatra-contrib/lib/sinatra/capture.rb b/sinatra-contrib/lib/sinatra/capture.rb
@@ -97,7 +97,9 @@ def capture(*args, &block)
         result = block[*args]
       elsif current_engine == :erb || current_engine == :slim
         @_out_buf, _buf_was = '', @_out_buf
-        result = block.call(*args)
+        raw = block.call(*args)
+        captured = block.binding.eval('@_out_buf')
+        result = captured.empty? ? raw : captured
         @_out_buf = _buf_was
       else
         buffer     = eval '_buf if defined?(_buf)', block.binding