Class: Bushido::Middleware

Inherits:
Object
  • Object
show all
Includes:
Rack::Utils
Defined in:
lib/bushido/middleware.rb

Constant Summary collapse

BUSHIDO_JS_URL = 
"#{Bushido::Platform.host}/api/bushido.js"

Instance Method Summary collapse

Constructor Details

#initialize(app, opts = {}) ⇒ Middleware

Returns a new instance of Middleware.



10
11
12
13
14
15
 
# File 'lib/bushido/middleware.rb', line 10

def initialize(app, opts = {})
  @app = app
  @bushido_app_name       = ENV['BUSHIDO_APP']
  @bushido_metrics_token  = ENV['BUSHIDO_METRICS_TOKEN']
  @bushido_claimed        = (ENV['BUSHIDO_CLAIMED'].nil? or ENV['BUSHIDO_CLAIMED'].blank?) ? false : true
end

Instance Method Details

#call(env) ⇒ Object 



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
 
# File 'lib/bushido/middleware.rb', line 17

def call(env)
  @bushido_claimed        = (ENV['BUSHIDO_CLAIMED'].nil? or ENV['BUSHIDO_CLAIMED'].blank?) ? false : true

  status, headers, response = @app.call(env)
  
  unless @bushido_app_name.empty?
    content = ""
    response.each { |part| content += part }

    # "claiming" bar + stats ?
    content.gsub!(/<\/body>/i, <<-STR
        <script type="text/javascript">
          var _bushido_app = '#{@bushido_app_name}';
          var _bushido_claimed = #{@bushido_claimed.to_s};
          var _bushido_metrics_token = '#{@bushido_metrics_token}';
          (function() {
            var bushido = document.createElement('script'); bushido.type = 'text/javascript'; bushido.async = true;
            bushido.src = '#{BUSHIDO_JS_URL}?#{::Bushido::VERSION.gsub('.', '')}';
            var s = document.getElementsByTagName('script')[0]; s.parentNode.insertBefore(bushido, s);
          })();
        </script>     
      </body>
    STR
    )

    headers['content-length'] = bytesize(content).to_s
  end
  [status, headers, [content]]
end