logging.hpp

 
#pragma once
 
#include <algorithm>
#include <cctype>
#include <iostream>
#include <string>
#include <unordered_map>
 
#define FMT_HEADER_ONLY
#include <fmt/format.h>
#include <fmt/ranges.h>
 
#include "vipra/macros/performance.hpp"
 
namespace VIPRA {
 
class Log {
 public:
  enum class Level {
    DEBUG,
    INFO,
    WARN,
    ERROR,
  };
 
  static void set_level(Level lvl) { level = lvl; }
  static void set_level(std::string lvl)
  {
    static const std::unordered_map<std::string, Level> LEVELS = {
        {"debug", Level::DEBUG},
        {"info", Level::INFO},
        {"warn", Level::WARN},
        {"error", Level::ERROR},
    };
 
    std::transform(lvl.begin(), lvl.end(), lvl.begin(),
                   [](char letter) { return tolower(letter); });
 
    level = LEVELS.at(lvl);
  }

  template <typename... param_ts>
  VIPRA_INLINE static void warn(fmt::format_string<param_ts...> message,
                                param_ts&&... params)
  {
    if ( level <= Level::WARN ) {
      std::cout << "[WARN] " << fmt::format(message, std::forward<param_ts>(params)...)
                << '\n';
    }
  }

  template <typename... param_ts>
  VIPRA_INLINE static void info(fmt::format_string<param_ts...> message,
                                param_ts&&... params)
  {
    if ( level <= Level::INFO ) {
      std::cout << "[INFO] " << fmt::format(message, std::forward<param_ts>(params)...)
                << '\n';
    }
  }

  template <typename... param_ts>
  VIPRA_INLINE static void debug(fmt::format_string<param_ts...> message,
                                 param_ts&&... params)
  {
    if ( level <= Level::DEBUG ) {
      std::cout << "[DEBUG] " << fmt::format(message, std::forward<param_ts>(params)...)
                << '\n';
    }
  }

  template <typename... param_ts>
  VIPRA_INLINE static void error(fmt::format_string<param_ts...> message,
                                 param_ts&&... params)
  {
    std::cout << "[ERROR] " << fmt::format(message, std::forward<param_ts>(params)...)
              << '\n';
  }
 
 private:
  // NOLINTNEXTLINE(cppcoreguidelines-avoid-non-const-global-variables)
  static Level level;
};
}  // namespace VIPRA