# typed: false
module Datadog
  module Contrib
    module ActionCable
      module Instrumentation
        # When a new WebSocket is open, we receive a Rack request resource name "GET -1".
        # This module overrides the current Rack resource name to provide a meaningful name.
        module ActionCableConnection
          def on_open
            Datadog.tracer.trace(Ext::SPAN_ON_OPEN) do |span|
              begin
                span.resource = "#{self.class}#on_open"
                span.span_type = Datadog::Ext::AppTypes::WEB

                span.set_tag(Ext::TAG_ACTION, 'on_open')
                span.set_tag(Ext::TAG_CONNECTION, self.class.to_s)

                # Set the resource name of the Rack request span
                rack_request_span = env[Datadog::Contrib::Rack::TraceMiddleware::RACK_REQUEST_SPAN]
                rack_request_span.resource = span.resource if rack_request_span
              rescue StandardError => e
                Datadog.logger.error("Error preparing span for ActionCable::Connection: #{e}")
              end

              super
            end
          end
        end

        # Instrumentation for when a Channel is subscribed to/unsubscribed from.
        module ActionCableChannel
          def self.included(base)
            base.class_eval do
              set_callback(
                :subscribe,
                :around,
                ->(channel, block) { Tracer.trace(channel, :subscribe, &block) },
                prepend: true
              )

              set_callback(
                :unsubscribe,
                :around,
                ->(channel, block) { Tracer.trace(channel, :unsubscribe, &block) },
                prepend: true
              )
            end
          end

          # Instrumentation for Channel hooks.
          class Tracer
            def self.trace(channel, hook)
              configuration = Datadog.configuration[:action_cable]

              Datadog.tracer.trace("action_cable.#{hook}") do |span|
                span.service = configuration[:service_name]
                span.resource = "#{channel.class}##{hook}"
                span.span_type = Datadog::Ext::AppTypes::WEB

                # Set analytics sample rate
                if Contrib::Analytics.enabled?(configuration[:analytics_enabled])
                  Contrib::Analytics.set_sample_rate(span, configuration[:analytics_sample_rate])
                end

                # Measure service stats
                Contrib::Analytics.set_measured(span)

                span.set_tag(Ext::TAG_CHANNEL_CLASS, channel.class.to_s)

                yield
              end
            end
          end
        end
      end
    end
  end
end