# Workling instrumentation contributed by Chad Ingram of Aurora Feint
#
DependencyDetection.defer do
  @name = :workling

  depends_on do
    defined?(::Workling) and not ::NewRelic::Control.instance['disable_workling']
  end

  executes do
    NewRelic::Agent.logger.debug 'Installing Workling instrumentation'
  end

  executes do
    ::Workling::Base.class_eval do
      include ::NewRelic::Agent::Instrumentation::ControllerInstrumentation

      alias_method :origin_dispatch_to_worker_method, :dispatch_to_worker_method
      def dispatch_to_worker_method(method, options)
        NewRelic::Agent.manual_start
        origin_dispatch_to_worker_method(method, options)
      ensure
        NewRelic::Agent.shutdown
      end
    end

    ::Workling::Discovery.discovered.each do |clazz|
      (clazz.public_instance_methods - ::Workling::Base.public_instance_methods).each do |method|
        clazz.class_eval do
          NewRelic::Agent.logger.debug "added method tracer Workling/#{clazz.name}/#{method}"
          clazz.send(:add_transaction_tracer, method, :category => "OtherTransaction/Workling")
        end
      end
    end
  end
end