# We require all of ActiveRecord because the #logger method in ::ActiveRecord::LogSubscriber
# uses ActiveRecord::Base. We can't require active_record/base directly because ActiveRecord
# does not require files properly and we receive unintialized constant errors.
require "active_record"
require "active_record/log_subscriber"

require "timber/integrator"

module Timber
  module Integrations
    module ActiveRecord
      class LogSubscriber < Integrator
        # The log subscriber that replaces the default `ActiveRecord::LogSubscriber`.
        # The intent of this subscriber is to, as transparently as possible, properly
        # track events that are being logged here. This LogSubscriber will never change
        # default behavior / log messages.
        #
        # @private
        class TimberLogSubscriber < ::ActiveRecord::LogSubscriber
          def sql(event)
            return true if silence?

            r = super(event)

            if @message
              payload = event.payload
              event = Events::SQLQuery.new(
                sql: payload[:sql],
                time_ms: event.duration,
                message: @message
              )

              logger.debug event

              @message = nil
            end

            r
          end

          private
            def debug(message)
              @message = message
            end

            def silence?
              ActiveRecord.silence?
            end
        end
      end
    end
  end
end