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
50
51
52
53
54
55
56
57
58
59
60
61
|
require 'rack'
require 'erb'
module Rack
class GoogleAnalytics
EVENT_TRACKING_KEY = "google_analytics.event_tracking"
DEFAULT = { async: true, enhanced_link_attribution: false, advertising: false }
def initialize(app, options = {})
@app, @options = app, DEFAULT.merge(options)
end
def call(env); dup._call(env); end
def _call(env)
@status, @headers, @body = @app.call(env)
return [@status, @headers, @body] unless html?
response = Rack::Response.new([], @status, @headers)
@options[:tracker_vars] = env["google_analytics.custom_vars"] || []
if response.ok?
# Write out the events now
@options[:tracker_vars] += (env[EVENT_TRACKING_KEY]) unless env[EVENT_TRACKING_KEY].nil?
# Get any stored events from a redirection
session = env["rack.session"]
stored_events = session.delete(EVENT_TRACKING_KEY) if session
@options[:tracker_vars] += stored_events unless stored_events.nil?
elsif response.redirection? && env["rack.session"]
# Store the events until next time
env["rack.session"][EVENT_TRACKING_KEY] = env[EVENT_TRACKING_KEY]
end
@options[:tracker] = expand_tracker(env, @options[:tracker])
@body.each { |fragment| response.write inject(fragment) }
@body.close if @body.respond_to?(:close)
response.finish
end
private
def html?; @headers['Content-Type'] =~ /html/; end
def inject(response)
@tracker_options = { cookieDomain: @options[:domain] }.select{|k,v| v }.to_json
@template ||= ::ERB.new ::File.read ::File.expand_path("../../../../../../usr/share/ruby-rack-google-analytics/templates/async.erb",__FILE__)
response.gsub(%r{</head>}, @template.result(binding) + "</head>")
end
def expand_tracker(env, tracker)
tracker.respond_to?(:call) ? tracker.call(env) : tracker
end
end
end
|
