module Rack
  # Rack::Bench shows total request duration for any request.
  class Bench
    def initialize(app)
      @app = app
    end

    def call(env)
      start_time = Time.now
      code, headers, body = @app.call(env)
      end_time = Time.now
      
      contents = <<-EOT
<style>
.kiss_bench {
  text-align: left;
  padding: 3px 7px;
  border: 1px solid #ec4;
  border-top: 1px solid #fff4bb;
  border-bottom: 1px solid #d91;
  background-color: #ffe590;
  font-size: 12px;
  color: #101;
}
.kiss_bench a {
  color: #930;
  text-decoration: none;
}
.kiss_bench a:hover {
  color: #930;
  text-decoration: underline;
}
</style>
<div class="kiss_bench">
<tt><b>TOTAL request duration: #{sprintf("%0.3f",end_time.to_f - start_time.to_f)} s</b></tt>
<br><small>kiss bench request total</small>
</div>
EOT
      
      body.each {|p| contents += p }
      headers['Content-Length'] = contents.length.to_s
    
      [ code, headers, contents ]
    end
  end
end