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)
            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
        end
      end
    end
  end
end