Add colored text

Author: Pixels67

src/example/example.cpp

@@ -1,8 +1,9 @@
 #include "logger.hpp"
 
 int main() {
-    Logger::log(LogLevel::FATAL)   << "Hello, Fatal!";
-    Logger::log(LogLevel::ERROR)   << "Hello, Error!";
-    Logger::log(LogLevel::WARNING) << "Hello, Warning!";
-    Logger::log(LogLevel::INFO)    << "Hello, Info!";
+    Logger::useColor = true;
+    // Logger::log(LogLevel::FATAL)   << "Hello, Fatal!";
+    Logger::log(LogLevel::Error)   << "Hello, Error!";
+    Logger::log(LogLevel::Warning) << "Hello, Warning!";
+    Logger::log(LogLevel::Info)    << "Hello, Info!";
 }

src/logger.cpp

@@ -4,14 +4,20 @@
 #include <iostream>
 #include <sstream>
 
+#if _WIN32
+
+#include <windows.h>
+
+#endif
+
 std::string Logger::logLevelToString(const LogLevel level) {
     switch (level) {
-        case LogLevel::FATAL:   return "[FTL]";
-        case LogLevel::ERROR:   return "[ERR]";
-        case LogLevel::WARNING: return "[WRN]";
-        case LogLevel::INFO:    return "[INF]";
+        case LogLevel::Fatal:   return "[FTL]";
+        case LogLevel::Error:   return "[ERR]";
+        case LogLevel::Warning: return "[WRN]";
+        case LogLevel::Info:    return "[INF]";
         default:
-            log(LogLevel::ERROR) << "Invalid log level specified!";
+            log(LogLevel::Error) << "Invalid log level specified!";
             return "\0";
     }
 }
@@ -26,6 +32,40 @@ std::string Logger::getTimestamp() {
     return '[' + currentTimeString + ']';
 }
 
+#if _WIN32
+
+int Logger::getColor(const LogLevel level) {
+    switch (level) {
+        case LogLevel::Fatal:   return 0x0C;  // Red
+        case LogLevel::Error:   return 0x0C;  // Red
+        case LogLevel::Warning: return 0x0E;  // Yellow
+        case LogLevel::Info:    return 0x09;  // Blue
+        default:
+            log(LogLevel::Error) << "Invalid log level specified!";
+            return 0x0F;
+    }
+}
+
+void Logger::setWindowsColor(const int color) {
+    SetConsoleTextAttribute(GetStdHandle(STD_OUTPUT_HANDLE), color);
+}
+
+#else
+
+std::string Logger::getColor(const LogLevel level) {
+    switch (level) {
+        case LogLevel::Fatal:   return "\033[1;31m";  // Red
+        case LogLevel::Error:   return "\033[0;31m";  // Light red
+        case LogLevel::Warning: return "\033[0;33m";  // Yellow
+        case LogLevel::Info:    return "\033[0;34m";  // Blue
+        default:
+            log(LogLevel::Error) << "Invalid log level specified!";
+            return "\033[0m";
+    }
+}
+
+#endif
+
 Logger::LoggerStream::LoggerStream(const LogLevel level)
 : messageLogLevel(level) {
 }
@@ -35,16 +75,37 @@ Logger::LoggerStream::~LoggerStream() {
 
     std::ostringstream message;
 
+#if _WIN32
+    if (useColor)
+        setWindowsColor(getColor(messageLogLevel));
+#else
+    if (useColor)
+        message << getColor(messageLogLevel);
+#endif
+
     if (showTimestamp)
         message << getTimestamp() << ' ';
 
     message << logLevelToString(messageLogLevel) << ' ' << strStream.str() << std::endl;
 
-    if (messageLogLevel == LogLevel::FATAL || messageLogLevel == LogLevel::ERROR) {
+#if not _WIN32
+    if (useColor)
+        message << "\033[0m";
+#endif
+
+    if (messageLogLevel == LogLevel::Fatal) {
+        std::cerr << message.str();
+        std::exit(EXIT_FAILURE);
+    } else if (messageLogLevel == LogLevel::Error) {
         std::cerr << message.str();
     } else {
         std::cout << message.str();
     }
+
+#if _WIN32
+    if (useColor)
+        setWindowsColor(0x0F);
+#endif
 }
 
 Logger::LoggerStream Logger::log(const LogLevel level) {

src/logger.hpp

@@ -4,12 +4,12 @@
 
 namespace Logger {
     enum class LogLevel {
-        NONE    = 0,
-        FATAL   = 1,
-        ERROR   = 2,
-        WARNING = 4,
-        INFO    = 8,
-        ALL     = FATAL | ERROR | WARNING | INFO
+        None    = 0,
+        Fatal   = 1,
+        Error   = 2,
+        Warning = 4,
+        Info    = 8,
+        All     = Fatal | Error | Warning | Info
     };
 
     constexpr LogLevel operator|(LogLevel a, LogLevel b) {
@@ -28,12 +28,20 @@ namespace Logger {
         return static_cast<LogLevel>(~static_cast<char>(level));
     }
 
-    inline LogLevel logLevel = LogLevel::ALL;
+    inline LogLevel logLevel = LogLevel::All;
     inline bool showTimestamp = true;
+    inline bool useColor = false;
 
     std::string logLevelToString(LogLevel level);
     std::string getTimestamp();
 
+#if _WIN32
+    int getColor(LogLevel level);
+    void setWindowsColor(int color);
+#else
+    std::string getColor(LogLevel level);
+#endif
+
     class LoggerStream {
     public:
         explicit LoggerStream(LogLevel level);