color_operators.h

Go to the documentation of this file.

#ifndef KOUTIL_TERM_COLOR_OPERATORS_H
#define KOUTIL_TERM_COLOR_OPERATORS_H
 
#include "koutil/term/color.h"
#include "koutil/util/utils.h"
#include <cstddef>
#include <ostream>
#include <string_view>
 
namespace koutil::term {
 
namespace color_literals {
 
    consteval Color operator""_color(const char* str, std::size_t size) {
        std::string_view hex(str, size);
        return Color::from_hex(hex);
    }
 
    consteval ColorBG operator""_bg(const char* str, std::size_t size) {
        std::string_view hex(str, size);
        return Color::from_hex(hex).as_bg();
    }
 
    consteval ColorFG operator""_fg(const char* str, std::size_t size) {
        std::string_view hex(str, size);
        return Color::from_hex(hex).as_fg();
    }
 
}
 
std::ostream& operator<<(std::ostream& stream, ColorBG bg) {
 
    switch (bg.color.tag) {
    case Color::Tag::RGB:
        stream << util::ESC << "[48;2;" << +bg.color.red << ';' << +bg.color.green << ';' << +bg.color.blue << 'm';
        break;
    case Color::Tag::ID:
        stream << util::ESC << '[' << +bg.color.id() + 10 << 'm';
        break;
    }
 
    return stream;
}
 
std::ostream& operator<<(std::ostream& stream, ColorFG fg) {
 
    switch (fg.color.tag) {
    case Color::Tag::RGB:
        stream << util::ESC << "[38;2;" << +fg.color.red << ';' << +fg.color.green << ';' << +fg.color.blue << 'm';
        break;
    case Color::Tag::ID:
        stream << util::ESC << '[' << +fg.color.id() << 'm';
        break;
    }
 
    return stream;
}
 
std::ostream& reset_bg(std::ostream& stream) {
    stream << util::ESC << "[49m";
    return stream;
}
 
std::ostream& reset_fg(std::ostream& stream) {
    stream << util::ESC << "[39m";
    return stream;
}
 
std::ostream& reset_color(std::ostream& stream) {
    stream << util::ESC << "[39;49m";
    return stream;
}
 
}
 
#endif