module RorVsWild
module Local
class Middleware
include ERB::Util
attr_reader :app, :config
def initialize(app, config)
@app, @config = app, config
end
def call(env)
env["REQUEST_URI"] == "/rorvswild" ? standalone_profiler(env) : embed_profiler(env)
end
def standalone_profiler(env)
html = inject_into(empty_html_page)
[200, {"Content-Type:" => "text/html; charset=utf-8"}, StringIO.new(html || empty_html_page)]
end
def embed_profiler(env)
status, headers, body = app.call(env)
if status >= 200 && status < 300 && headers["Content-Type"] && headers["Content-Type"].include?("text/html")
if headers["Content-Encoding"]
log_incompatible_middleware_warning
elsif body.respond_to?(:each)
content_length = 0
body.each do |string|
inject_into(string)
content_length += string.size
end
headers["Content-Length"] = content_length.to_s if headers["Content-Length"]
end
end
[status, headers, body]
end
def inject_into(html)
markup = html_markup(RorVsWild.agent.queue.requests).encode(html.encoding)
style = "".encode(html.encoding)
markup = markup.html_safe if markup.respond_to?(:html_safe)
style = style.html_safe if markup.respond_to?(:html_safe)
if index = html.index(" ")
html.insert(index, markup)
html.insert(index, style)
end
html
rescue Encoding::UndefinedConversionError => ex
log_incompatible_encoding_warning(ex)
nil
end
LOCAL_FOLDER = File.expand_path(File.dirname(__FILE__))
JS_FOLDER = File.join(LOCAL_FOLDER, "javascript")
CSS_FOLDER = File.join(LOCAL_FOLDER, "stylesheet")
JS_FILES = ["vendor/mustache.js", "vendor/barber.js", "vendor/prism.js", "local.js"]
CSS_FILES = ["vendor/prism.css", "local.css"]
def html_markup(data)
html = File.read(File.join(LOCAL_FOLDER, "local.html"))
html % {data: html_escape(data.to_json), javascript_source: concatenate_javascript}
end
def concatenate_javascript
concatenate_assets(JS_FOLDER, JS_FILES)
end
def concatenate_stylesheet
concatenate_assets(CSS_FOLDER, CSS_FILES)
end
def concatenate_assets(directory, files)
files.map { |file| File.read(File.join(directory, file)) }.join("\n")
end
def empty_html_page
"\n