Write to template buffer for erubi and erubis

sinatra · Dec 8, 2018 · ac36e38 · ac36e38
1 parent a3339e0
commit ac36e38
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/sinatra-contrib/lib/sinatra/content_for.rb b/sinatra-contrib/lib/sinatra/content_for.rb
@@ -177,7 +177,9 @@ def yield_content(key, *args, &block)
       else
         content = content_blocks[key.to_sym].map { |b| capture(*args, &b) }
         content.join.tap do |c|
-          @_out_buf << c if block_given? && erb?
+          if block_given? && (erb? || erubi? || erubis?)
+            @_out_buf << c
+          end
         end
       end
     end