module Bullet class Rack def initialize(app) @app = app end def call(env) return @app.call(env) unless Bullet.enable? Bullet.start_request status, headers, response = @app.call(env) return [status, headers, response] if empty?(response) response_body = nil if Bullet.notification? if status == 200 && !response.body.frozen? && html_request?(headers, response) response_body = response.body << Bullet.gather_inline_notifications headers['Content-Length'] = response_body.length.to_s end Bullet.perform_out_of_channel_notifications(env) end Bullet.end_request no_browser_cache(headers) if Bullet.disable_browser_cache [status, headers, response_body ? [response_body] : response] end # fix issue if response's body is a Proc def empty?(response) # response may be ["Not Found"], ["Move Permanently"], etc. (response.is_a?(Array) && response.size <= 1) || !response.body.is_a?(String) || response.body.empty? end def html_request?(headers, response) headers['Content-Type'] && headers['Content-Type'].include?('text/html') && response.body.include?("") && response.body.include?("") end def no_browser_cache(headers) headers["Cache-Control"] = "no-cache, no-store, max-age=0, must-revalidate" headers["Pragma"] = "no-cache" headers["Expires"] = "Wed, 09 Sep 2009 09:09:09 GMT" end end end