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;
    }
    // clang-format off
    #ifdef _WIN32
    // clang-format on
    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;
    }
    // clang-format off
    #else
    color_enabled() = static_cast<bool>(isatty(fileno(stdout)));
    #endif
    // clang-format on
  }

  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);
  }

  inline std::string link(std::string const& url, std::string const& text) {
    if (!color_enabled()) {
      return text;
    }
    // BEL terminator (rather than ST `ESC \`) — more widely accepted by
    // terminals that implement only a subset of the OSC 8 spec.
    return fmt::format("\x1b]8;;{}\x07{}\x1b]8;;\x07", url, text);
  }
}