-
Notifications
You must be signed in to change notification settings - Fork 431
/
rack.rb
49 lines (42 loc) · 1.56 KB
/
rack.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
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 file?(headers) || 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.respond_to?(:body) || response.body.empty?
end
# if send file?
def file?(headers)
headers["Content-Transfer-Encoding"] == "binary"
end
def html_request?(headers, response)
headers['Content-Type'] && headers['Content-Type'].include?('text/html') && response.body.include?("<html")
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