engine/lib/locomotive/hosting/bushido/middleware.rb

49 lines
1.4 KiB
Ruby
Raw Normal View History

require 'rack/utils'
module Locomotive
module Hosting
module Bushido
class Middleware
2011-05-27 22:39:48 +00:00
# BUSHIDO_JS_URL = 'http://localhost:4567/javascripts/bushido.js'
BUSHIDO_JS_URL = 'http://bushi.do/api/bushido.js'
include Rack::Utils
def initialize(app, opts = {})
@app = app
@bushido_app_name = ENV['BUSHIDO_APP']
@bushido_claimed = ::Locomotive.bushido_app_claimed?
end
def call(env)
status, headers, response = @app.call(env)
2011-05-27 23:24:11 +00:00
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};
(function() {
var bushido = document.createElement('script'); bushido.type = 'text/javascript'; bushido.async = true;
bushido.src = '#{BUSHIDO_JS_URL}?#{::Bushido::VERSION.gsub('.', '')}';
2011-05-27 23:24:11 +00:00
var s = document.getElementsByTagName('script')[0]; s.parentNode.insertBefore(bushido, s);
})();
</script>
</body>
STR
)
headers['content-length'] = bytesize(content).to_s
[status, headers, [content]]
end
end
end
end
end