# frozen_string_literal: true

module Appsignal
  module Integrations
    module ActiveSupportNotificationsIntegration
      BANG = "!"

      module InstrumentIntegration
        def instrument(name, payload = {}, &block)
          # Events that start with a bang are internal to Rails
          instrument_this = name[0] != ActiveSupportNotificationsIntegration::BANG

          Appsignal::Transaction.current.start_event if instrument_this

          super
        ensure
          if instrument_this
            title, body, body_format = Appsignal::EventFormatter.format(name, payload)
            Appsignal::Transaction.current.finish_event(
              name.to_s,
              title,
              body,
              body_format
            )
          end
        end
      end

      module StartFinishIntegration
        def start(name, payload = {})
          # Events that start with a bang are internal to Rails
          instrument_this = name[0] != ActiveSupportNotificationsIntegration::BANG

          Appsignal::Transaction.current.start_event if instrument_this

          super
        end

        def finish(name, payload = {})
          # Events that start with a bang are internal to Rails
          instrument_this = name[0] != ActiveSupportNotificationsIntegration::BANG

          if instrument_this
            title, body, body_format = Appsignal::EventFormatter.format(name, payload)
            Appsignal::Transaction.current.finish_event(
              name.to_s,
              title,
              body,
              body_format
            )
          end

          super
        end
      end

      module FinishStateIntegration
        def finish_with_state(listeners_state, name, payload = {})
          # Events that start with a bang are internal to Rails
          instrument_this = name[0] != ActiveSupportNotificationsIntegration::BANG

          if instrument_this
            title, body, body_format = Appsignal::EventFormatter.format(name, payload)
            Appsignal::Transaction.current.finish_event(
              name.to_s,
              title,
              body,
              body_format
            )
          end

          super
        end
      end
    end
  end
end