# frozen_string_literal: true

require "dry-configurable"
require "dry/core/extensions"

module Dry
  module Monitor
    Notifications.register_event(:sql)

    module SQL
      class Logger
        extend Dry::Core::Extensions
        extend Dry::Configurable

        register_extension(:default_colorizer) do
          require "dry/monitor/sql/colorizers/default"

          module DefaultColorizer
            def colorizer
              @colorizer ||= Colorizers::Default.new(config.theme)
            end
          end

          Logger.include(DefaultColorizer)
        end

        register_extension(:rouge_colorizer) do
          require "dry/monitor/sql/colorizers/rouge"

          module RougeColorizer
            def colorizer
              @colorizer ||= Colorizers::Rouge.new(config.theme)
            end
          end

          Logger.include(RougeColorizer)
        end

        setting :theme
        setting :message_template, default: %(  Loaded %s in %sms %s)

        attr_reader :config, :logger, :template

        load_extensions(:default_colorizer)

        def initialize(logger, config = self.class.config)
          @logger = logger
          @config = config
          @template = config.message_template
        end

        def subscribe(notifications)
          notifications.subscribe(:sql) { |params| log_query(**params) }
        end

        def log_query(time:, name:, query:)
          logger.info template % [name.inspect, time, colorizer.call(query)]
        end
      end
    end
  end
end