# frozen_string_literal: true

require 'forwardable'
require 'fileutils'
require 'rainbow'
require 'active_support'
require 'active_support/logger'

module GitlabQuality
  module TestTooling
    module Runtime
      class Logger
        extend SingleForwardable

        def_delegators :logger, :debug, :info, :warn, :error, :fatal, :unknown

        TIME_FORMAT = "%Y-%m-%d %H:%M:%S"
        LEVEL_COLORS = {
          "DEBUG" => :magenta,
          "INFO" => :green,
          "WARN" => :yellow,
          "ERROR" => :red,
          "FATAL" => :indianred
        }.freeze

        Rainbow.enabled = Runtime::Env.colorized_logs?

        class << self
          # Combined logger instance
          #
          # @param [String] source
          # @return [ActiveSupport::Logger]
          def logger(source: 'Gitlab QA')
            @logger ||= begin
              log_path = Env.log_path
              ::FileUtils.mkdir_p(log_path)

              console_log = console_logger(source: source, level: Env.log_level)
              file_log = file_logger(source: source, path: log_path)

              ActiveSupport::BroadcastLogger.new(console_log, file_log, file_log)
            end
          end

          private

          # Console logger instance
          #
          # @param [String] source
          # @param [<Symbol, String>] level
          # @return [ActiveSupport::Logger]
          def console_logger(source:, level:)
            ActiveSupport::Logger.new($stdout, level: level, datetime_format: TIME_FORMAT).tap do |logger|
              logger.formatter = proc do |severity, datetime, _progname, msg|
                msg_prefix = message_prefix(datetime, source, severity)

                Rainbow(msg_prefix).public_send(LEVEL_COLORS.fetch(severity, :silver)) + "#{msg}\n" # rubocop:disable GitlabSecurity/PublicSend
              end
            end
          end

          # File logger
          #
          # @param [String] source
          # @param [String] path
          # @return [ActiveSupport::Logger]
          def file_logger(source:, path:)
            log_file = "#{path}/#{source.downcase.tr(' ', '-')}.log"

            ActiveSupport::Logger.new(log_file, level: :debug, datetime_format: TIME_FORMAT).tap do |logger|
              logger.formatter = proc do |severity, datetime, _progname, msg|
                msg_prefix = message_prefix(datetime, source, severity)

                "#{msg_prefix}#{msg}\n".gsub(/\e\[(\d+)(?:;\d+)*m/, "")
              end
            end
          end

          # Log message prefix
          #
          # @note when outputted, the date will be formatted as "Jun 07 2022 11:30:00 UTC"
          # @param [DateTime] date
          # @param [String] source
          # @param [String] severity
          # @return [String]
          def message_prefix(date, source, severity)
            "[#{date.strftime('%h %d %Y %H:%M:%S %Z')} (#{source})] #{severity.ljust(5)} -- "
          end
        end
      end
    end
  end
end