class Remnant
  class Rails
    module ClassMethods
      def logger
        ::Rails.logger
      end

      def setup!
        Remnant.configure do
          environment ::Rails.env
        end

        #
        # helper hooks
        #

        # hook into dependency unloading
        ::ActiveSupport::Dependencies.class_eval do
          class << self
            def clear_with_remnant_rediscover(*args, &block)
              clear_without_remnant_rediscover(*args, &block).tap do
                Remnant::Discover.rediscover!
              end
            end
            alias_method_chain :clear, :remnant_rediscover
          end
        end


        #
        # stat collection below
        #

        # hook remnants
        Remnant::Discover.find('request',  ActionController::Dispatcher,            :call)
        Remnant::Discover.find('dispatch', ActionController::Dispatcher,            :_call)
        Remnant::Discover.find('process',  ActionController::Base,                  :process)
        Remnant::Discover.find('filters',  ActionController::Filters::BeforeFilter, :call)
        Remnant::Discover.find('action',   ActionController::Base,                  :perform_action)
        Remnant::Discover.find('view',     ActionController::Base,                  :render)
        Remnant::Discover.find('filters',  ActionController::Filters::AfterFilter,  :call)

        # last hook into request cycle for sending results
        ::ActionController::Dispatcher.class_eval do
          def call_with_remnant_discovery(*args, &block) #:nodoc:
            call_without_remnant_discovery(*args, &block).tap do |status, headers, response|
              ::Remnant.collect
              ::Rails.logger.flush if ::Rails.logger.respond_to? :flush
            end
          end
          alias_method_chain :call, :remnant_discovery
        end

        # hook into perform_action for the extra remnant key
        ::ActionController::Base.class_eval do
          def perform_action_with_remnant_key(*args, &block) #:nodoc:
            ::Remnant::Discover.results[:extra_remnant_key] = "#{params[:controller]}.#{params[:action]}"
            perform_action_without_remnant_key(*args, &block)
          end
          alias_method_chain :perform_action, :remnant_key
        end
      end # setup!
    end
    extend ClassMethods
  end
end