class Appsignal::Utils::IntegrationMemoryLogger

def add(severity, message, _progname = nil)
  message = formatter.call(severity, Time.now, nil, message) if formatter
  messages[severity] << message
end

def clear
  messages.clear
end

def debug(message)
  add(:DEBUG, message)
end

def error(message)
  add(:ERROR, message)
end

def fatal(message)
  add(:FATAL, message)
end

def info(message)
  add(:INFO, message)
end

def messages
  @messages ||= Hash.new { |hash, key| hash[key] = [] }
end

def messages_for_level(level)
  levels = LEVELS.select { |log_level| log_level >= level }.values
  messages
    .select { |log_level| levels.include?(log_level) }
    .flat_map { |_level, message| message }
end

def unknown(message)
  add(:UNKNOWN, message)
end

def warn(message)
  add(:WARN, message)
end