Color.hpp

Go to the documentation of this file.

#pragma once
 
#include <string>
 
#include <fmt/color.h>
#include <fmt/format.h>
 
#ifdef _WIN32
#include <windows.h>
#else
#include <unistd.h>
#endif

namespace ppforest2::io::style {
  inline bool& color_enabled() {
    static bool enabled = true;
    return enabled;
  }

  inline void init_color(bool no_color) {
    if (no_color) {
      color_enabled() = false;
      return;
    }
 
#ifdef _WIN32
    HANDLE hOut  = GetStdHandle(STD_OUTPUT_HANDLE);
    DWORD dwMode = 0;
 
    if (hOut != INVALID_HANDLE_VALUE && GetConsoleMode(hOut, &dwMode)) {
      SetConsoleMode(hOut, dwMode | ENABLE_VIRTUAL_TERMINAL_PROCESSING);
      color_enabled() = true;
    } else {
      color_enabled() = false;
    }
 
#else
    color_enabled() = isatty(fileno(stdout));
#endif
  }

  inline std::string styled(std::string const& s, fmt::terminal_color c) {
    if (!color_enabled())
      return s;
 
    return fmt::format("\033[{}m{}\033[39m", static_cast<uint8_t>(c), s);
  }

  inline std::string error(std::string const& s) {
    return styled(s, fmt::terminal_color::red);
  }

  inline std::string success(std::string const& s) {
    return styled(s, fmt::terminal_color::green);
  }

  inline std::string emphasis(std::string const& s) {
    if (!color_enabled())
      return s;
 
    return fmt::format("\033[1m{}\033[22m", s);
  }

  inline std::string muted(std::string const& s) {
    return styled(s, fmt::terminal_color::bright_black);
  }

  inline std::string info(std::string const& s) {
    return styled(s, fmt::terminal_color::cyan);
  }

  inline std::string warning(std::string const& s) {
    return styled(s, fmt::terminal_color::yellow);
  }
}