ConsoleLogger.hpp

Go to the documentation of this file.

#pragma once
#ifndef _LOGIT_CONSOLE_LOGGER_HPP_INCLUDED
#define _LOGIT_CONSOLE_LOGGER_HPP_INCLUDED
 
#include "ILogger.hpp"
#include "../TaskExecutor.hpp"
#include "../LogMacros.hpp"
#include <iostream>
#if defined(__MINGW32__) || defined(_WIN32)
#include <windows.h>
#endif
#include <mutex>
#include <atomic>
 
namespace logit {
 
    class ConsoleLogger : public ILogger {
    public:
 
        struct Config {
            TextColor default_color = LOGIT_DEFAULT_COLOR; 
            bool async = true; 
        };
 
        ConsoleLogger() {
            reset_color();
        }
 
        ConsoleLogger(const Config& config) : m_config(config) {
            reset_color();
        }
 
        ConsoleLogger(const bool async) {
            m_config.async = async;
            reset_color();
        }
 
        virtual ~ConsoleLogger() = default;
 
        void set_config(const Config& config) {
            std::lock_guard<std::mutex> lock(m_mutex);
            m_config = config;
        }
 
        Config get_config() {
            std::lock_guard<std::mutex> lock(m_mutex);
            return m_config;
        }
 
        void log(const LogRecord& record, const std::string& message) override {
            m_last_log_ts = record.timestamp_ms;
            std::unique_lock<std::mutex> lock(m_mutex);
            if (!m_config.async) {
#               if defined(__MINGW32__) || defined(_WIN32)
                // For Windows, parse the message for ANSI color codes and apply them
                handle_ansi_colors_windows(message);
#               else
                // For other systems, output the message as is
                std::cout << message << std::endl;
#               endif
                return;
            }
            lock.unlock();
            TaskExecutor::get_instance().add_task([this, message](){
                std::lock_guard<std::mutex> lock(m_mutex);
#               if defined(__MINGW32__) || defined(_WIN32)
                // For Windows, parse the message for ANSI color codes and apply them
                handle_ansi_colors_windows(message);
#               else
                // For other systems, output the message as is
                std::cout << message << std::endl;
#               endif
            });
        }
 
        std::string get_string_param(const LoggerParam& param) const override {
            switch (param) {
            case LoggerParam::LastLogTimestamp: return std::to_string(get_last_log_ts());
            case LoggerParam::TimeSinceLastLog: return std::to_string(get_time_since_last_log());
            default:
                break;
            };
            return std::string();
        }
 
        int64_t get_int_param(const LoggerParam& param) const override {
            switch (param) {
            case LoggerParam::LastLogTimestamp: return get_last_log_ts();
            case LoggerParam::TimeSinceLastLog: return get_time_since_last_log();
            default:
                break;
            };
            return 0;
        }
 
        double get_float_param(const LoggerParam& param) const override {
            switch (param) {
            case LoggerParam::LastLogTimestamp: return (double)get_last_log_ts() / 1000.0;
            case LoggerParam::TimeSinceLastLog: return (double)get_time_since_last_log() / 1000.0;
            default:
                break;
            };
            return 0.0;
        }
 
        void wait() override {
            std::unique_lock<std::mutex> lock(m_mutex);
            if (!m_config.async) return;
            lock.unlock();
            TaskExecutor::get_instance().wait();
        }
 
    private:
        mutable std::mutex m_mutex;     
        Config             m_config;    
        std::atomic<int64_t> m_last_log_ts = ATOMIC_VAR_INIT(0);
 
#       if defined(__MINGW32__) || defined(_WIN32)
 
        // Windows console colors
        enum class WinColor {
            Black = 0,
            DarkBlue = 1,
            DarkGreen = 2,
            DarkCyan = 3,
            DarkRed = 4,
            DarkMagenta = 5,
            DarkYellow = 6,
            Gray = 7,
            DarkGray = 8,
            Blue = 9,
            Green = 10,
            Cyan = 11,
            Red = 12,
            Magenta = 13,
            Yellow = 14,
            White = 15,
        };
 
        void handle_ansi_colors_windows(const std::string& message) const {
            std::string::size_type start = 0;
            std::string::size_type pos = 0;
 
            HANDLE handle_stdout = GetStdHandle(STD_OUTPUT_HANDLE);
 
            while ((pos = message.find("\033[", start)) != std::string::npos) {
                // Output the part of the string before the ANSI code
                if (pos > start) {
                    std::cout << message.substr(start, pos - start);
                }
 
                // Find the end of the ANSI code
                std::string::size_type end_pos = message.find('m', pos);
                if (end_pos != std::string::npos) {
                    // Extract the ANSI code
                    std::string ansi_code = message.substr(pos + 2, end_pos - pos - 2);
                    apply_color_from_ansi_code(ansi_code, handle_stdout);
 
                    // Update position
                    start = end_pos + 1;
                } else {
                    break;
                }
            }
 
            // Output any remaining part of the message
            if (start < message.size()) {
                std::cout << message.substr(start);
            }
            if (!message.empty()) std::cout << std::endl;
 
            // Reset the console color to default
            SetConsoleTextAttribute(handle_stdout, static_cast<WORD>(text_color_to_win_color(m_config.default_color)));
        }
 
        void apply_color_from_ansi_code(const std::string& ansi_code, HANDLE handle_stdout) const {
            WORD color_value = static_cast<WORD>(text_color_to_win_color(m_config.default_color)); // Default color
            const int code = std::stoi(ansi_code);
            switch (code) {
                case 30: color_value = static_cast<WORD>(WinColor::Black); break;
                case 31: color_value = static_cast<WORD>(WinColor::DarkRed); break;
                case 32: color_value = static_cast<WORD>(WinColor::DarkGreen); break;
                case 33: color_value = static_cast<WORD>(WinColor::DarkYellow); break;
                case 34: color_value = static_cast<WORD>(WinColor::DarkBlue); break;
                case 35: color_value = static_cast<WORD>(WinColor::DarkMagenta); break;
                case 36: color_value = static_cast<WORD>(WinColor::DarkCyan); break;
                case 37: color_value = static_cast<WORD>(WinColor::Gray); break;
                case 90: color_value = static_cast<WORD>(WinColor::DarkGray); break;
                case 91: color_value = static_cast<WORD>(WinColor::Red); break;
                case 92: color_value = static_cast<WORD>(WinColor::Green); break;
                case 93: color_value = static_cast<WORD>(WinColor::Yellow); break;
                case 94: color_value = static_cast<WORD>(WinColor::Blue); break;
                case 95: color_value = static_cast<WORD>(WinColor::Magenta); break;
                case 96: color_value = static_cast<WORD>(WinColor::Cyan); break;
                case 97: color_value = static_cast<WORD>(WinColor::White); break;
                default:
                    // Unknown code, use default color
                    break;
            };
            // Set the console text attribute to the desired color
            SetConsoleTextAttribute(handle_stdout, color_value);
        }
 
        WinColor text_color_to_win_color(const TextColor& color) const {
            switch (color) {
                case TextColor::Black:       return WinColor::Black;
                case TextColor::DarkRed:     return WinColor::DarkRed;
                case TextColor::DarkGreen:   return WinColor::DarkGreen;
                case TextColor::DarkYellow:  return WinColor::DarkYellow;
                case TextColor::DarkBlue:    return WinColor::DarkBlue;
                case TextColor::DarkMagenta: return WinColor::DarkMagenta;
                case TextColor::DarkCyan:    return WinColor::DarkCyan;
                case TextColor::LightGray:   return WinColor::Gray;
                case TextColor::DarkGray:    return WinColor::DarkGray;
                case TextColor::Red:         return WinColor::Red;
                case TextColor::Green:       return WinColor::Green;
                case TextColor::Yellow:      return WinColor::Yellow;
                case TextColor::Blue:        return WinColor::Blue;
                case TextColor::Magenta:     return WinColor::Magenta;
                case TextColor::Cyan:        return WinColor::Cyan;
                case TextColor::White:       return WinColor::White;
                default:                     return WinColor::White;
            }
        }
#       endif
 
        void reset_color() {
#           if defined(__MINGW32__) || defined(_WIN32)
            handle_ansi_colors_windows(std::string());
#           else
            std::cout << to_string(m_config.default_color);
#           endif
        }
 
        int64_t get_last_log_ts() const {
            return m_last_log_ts;
        }
 
        int64_t get_time_since_last_log() const {
            return LOGIT_CURRENT_TIMESTAMP_MS() - m_last_log_ts;
        }
    }; // ConsoleLogger
 
}; // namespace logit
 
#endif // _LOGIT_CONSOLE_LOGGER_HPP_INCLUDED