# encoding: utf-8
# This file is distributed under New Relic's license terms.
# See https://github.com/newrelic/rpm/blob/master/LICENSE for complete details.

module NewRelic
  module Agent
    module Threading

      class AgentThread < ::Thread
        def initialize(label)
          ::NewRelic::Agent.logger.debug("Creating New Relic thread: #{label}")
          self[:newrelic_label] = label
          super
        end

        def self.bucket_thread(thread, profile_agent_code)
          if thread.key?(:newrelic_label)
            profile_agent_code ? :agent : :ignore
          elsif TransactionState.in_background_transaction?(thread)
            :background
          elsif TransactionState.in_request_transaction?(thread)
            :request
          else
            :other
          end
        end

        def self.scrub_backtrace(thread, profile_agent_code)
          begin
            bt = thread.backtrace
          rescue Exception => e
            ::NewRelic::Agent.logger.debug("Failed to backtrace #{thread.inspect}: #{e.class.name}: #{e.to_s}")
          end
          return nil unless bt
          bt.reject! { |t| t.include?('/newrelic_rpm-') } unless profile_agent_code
          bt
        end
      end

    end
  end
end