# typed: true
require 'ddtrace/contrib/active_support/notifications/event'
require 'ddtrace/contrib/active_job/ext'

module Datadog
  module Contrib
    module ActiveJob
      # Defines basic behaviors for an ActiveJob event.
      module Event
        def self.included(base)
          base.include(ActiveSupport::Notifications::Event)
          base.extend(ClassMethods)
        end

        # Class methods for ActiveJob events.
        module ClassMethods
          def span_options
            { service: configuration[:service_name] }
          end

          def tracer
            -> { configuration[:tracer] }
          end

          def configuration
            Datadog.configuration[:active_job]
          end

          def set_common_tags(span, payload)
            adapter_name = if payload[:adapter].is_a?(Class)
                             payload[:adapter].name
                           else
                             payload[:adapter].class.name
                           end
            span.set_tag(Ext::TAG_ADAPTER, adapter_name)

            job = payload[:job]
            span.set_tag(Ext::TAG_JOB_ID, job.job_id)
            span.set_tag(Ext::TAG_JOB_QUEUE, job.queue_name)
            span.set_tag(Ext::TAG_JOB_PRIORITY, job.priority) if job.respond_to?(:priority)
            span.set_tag(Ext::TAG_JOB_EXECUTIONS, job.executions) if job.respond_to?(:executions)

            job_scheduled_at = if job.respond_to?(:scheduled_at)
                                 job.scheduled_at
                               elsif job.respond_to?(:enqueued_at)
                                 job.enqueued_at
                               end
            span.set_tag(Ext::TAG_JOB_SCHEDULED_AT, Time.at(job_scheduled_at)) if job_scheduled_at
          end
        end
      end
    end
  end
end