module ActiveSupport::TaggedLogging

def self.logger(*args, **kwargs)
  new ActiveSupport::Logger.new(*args, **kwargs)
end

def self.new(logger)
  logger = logger.clone
  if logger.formatter
    logger.formatter = logger.formatter.clone
    # Workaround for https://bugs.ruby-lang.org/issues/20250
    # Can be removed when Ruby 3.4 is the least supported version.
    logger.formatter.object_id if logger.formatter.is_a?(Proc)
  else
    # Ensure we set a default formatter so we aren't extending nil!
    logger.formatter = ActiveSupport::Logger::SimpleFormatter.new
  end
  logger.formatter.extend Formatter
  logger.extend(self)
end

def flush
  clear_tags!
  super if defined?(super)
end

def tagged(*tags)
  if block_given?
    formatter.tagged(*tags) { yield self }
  else
    logger = ActiveSupport::TaggedLogging.new(self)
    logger.formatter.extend LocalTagStorage
    logger.push_tags(*formatter.current_tags, *tags)
    logger
  end
end