Contents

require 'pilfer/logger'
require 'pilfer/profiler'

module Pilfer
  class Middleware
    attr_accessor :app, :profiler, :file_matcher, :profile_guard

    def initialize(app, options = {}, &profile_guard)
      @app           = app
      @profiler      = options[:profiler] || default_profiler
      @file_matcher  = options[:file_matcher]
      @profile_guard = profile_guard
    end

    def call(env)
      if profile_guard.call(env)
        run_profiler { app.call(env) }
      else
        app.call(env)
      end
    end

    def run_profiler(&downstream)
      if file_matcher
        profiler.profile_files_matching(file_matcher, &downstream)
      else
        profiler.profile(&downstream)
      end
    end

    def default_profiler
      reporter = Pilfer::Logger.new($stdout, :app_root => ENV['PWD'])
      Pilfer::Profiler.new(reporter)
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
pilfer-0.0.1.pre4	lib/pilfer/middleware.rb