log4r.rb

# Copyright (c) HashiCorp, Inc.
# SPDX-License-Identifier: BUSL-1.1

# This adds trace level support to log4r. Since log4r
# loggers use the trace method for checking if trace
# information should be included in the output, we
# make some modifications to allow the trace check to
# still work while also supporting trace as a valid level
require "log4r/loggerfactory"

if !Log4r::Logger::LoggerFactory.respond_to?(:fake_define_methods)
  class Log4r::Logger::LoggerFactory
    class << self
      def fake_set_log(logger, lname)
        real_set_log(logger, lname)
        if lname == "TRACE"
          logger.instance_eval do
            alias :trace_as_level :trace
            def trace(*args)
              return @trace if args.empty?
              trace_as_level(*args)
            end
          end
        end
      end

      def fake_undefine_methods(logger)
        real_undefine_methods(logger)
        logger.instance_eval do
          def trace(*_)
            @trace
          end
        end
      end

      alias_method :real_undefine_methods, :undefine_methods
      alias_method :undefine_methods, :fake_undefine_methods
      alias_method :real_set_log, :set_log
      alias_method :set_log, :fake_set_log
    end
  end

  class Log4r::Logger
    # The factory allows using a previously created logger
    # instance if it exists. Doing this prevents knocking
    # out configuration that may have already been applied
    # to the logger instance (like log level)
    def self.factory(name, *args)
      l = Log4r::Logger::Repository[name]
      return l unless l.nil?
      Log4r::Logger.new(name, *args)
    end
  end
end