class RorVsWild::Local::Middleware

def call(env) 

def call(env)
  case env["PATH_INFO"]
  when "/rorvswild" then serve_standalone_profiler(env)
  when "/rorvswild.css" then serve_stylesheet
  when "/rorvswild.js" then serve_javascript
  when "/rorvswild.json" then serve_json
  else serve_embed_profiler(env)
  end
end

def concatenate_assets(directory, files) 

def concatenate_assets(directory, files)
  files.map { |file| File.read(File.join(directory, file)) }.join("\n")
end

def concatenate_javascript 

def concatenate_javascript
  js = File.read(File.join(JS_FOLDER, "application.js"))
  js = js.split("// include javascript here")
  js[0] + concatenate_assets(JS_FOLDER, JS_FILES) + js[1]
end

def concatenate_stylesheet 

def concatenate_stylesheet
  concatenate_assets(CSS_FOLDER, CSS_FILES)
end

def empty_html_page 

def empty_html_page
  "<!DOCTYPE html>\n<html><head></head><body></body></html>"
end

def initialize(app, config) 

def initialize(app, config)
  @app, @config = app, config
end

def inject_into(html) 

def inject_into(html)
  if index = html.index("</body>")
    markup = File.read(File.join(LOCAL_FOLDER, "local.html.erb"))
    markup = ERB.new(markup).result(binding)
    markup = markup.html_safe if markup.respond_to?(:html_safe)
    html.insert(index, markup)
  end
  html
rescue Encoding::UndefinedConversionError => ex
  log_incompatible_encoding_warning(ex)
  nil
end

def log_incompatible_encoding_warning(exception) 

def log_incompatible_encoding_warning(exception)
  RorVsWild.logger.warn("RorVsWild::Local cannot be embeded into your HTML page because of incompatible #{exception.message}." +
    " However you can just visit the /rorvswild page to see the profiler.")
end

def log_incompatible_middleware_warning 

def log_incompatible_middleware_warning
  RorVsWild.logger.warn("RorVsWild::Local cannot be embeded into your HTML page because of compression." +
    " Try to disable Rack::Deflater in development only." +
    " In the meantime just visit the /rorvswild page to see the profiler.")
end

def serve_embed_profiler(env) 

def serve_embed_profiler(env)
  status, headers, body = app.call(env)
  status = status.to_i
  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 serve_javascript 

def serve_javascript
  [200, {"Content-Type" => "application/javascript"}, StringIO.new(concatenate_javascript)]
end

def serve_json 

def serve_json
  [200, {"Content-Type" => "application/json"}, StringIO.new(RorVsWild.agent.queue.requests.to_json)]
end

def serve_standalone_profiler(env) 

def serve_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 serve_stylesheet 

def serve_stylesheet
  [200, {"Content-Type" => "text/css"}, StringIO.new(concatenate_stylesheet)]
end

def widget_css 

def widget_css
  config = RorVsWild.agent.config
  config && config[:widget] && "is-#{config[:widget]}"
end