require 'rubygems'
require 'bundler/setup'

require 'rack'

module RubyApp

  module Rack

    class Duration
      require 'rubyapp/mixins/duration'

      include RubyApp::Mixins::Duration

      def initialize(application)
        @application = application
      end
  
      def call(environment)
        results = nil
        duration = self.measure_duration do
          results = @application.call(environment)
        end
        results[2].each { |result| result.gsub!(/__DURATION__/, duration.to_s) } if results
        results
      end

    end

  end
 
end