Logger.cpp

#include "Logger.h"

#include <iostream>
#include <iomanip>
#include <ctime>
#include <mutex>

using namespace std;

ofstream Logger::stream;
mutex Logger::logMutex;

void Logger::init(const string& file)
{
    stream.open(file, ios::app);
}

string Logger::timestamp()
{
    time_t now = time(nullptr);
    tm localTime = *localtime(&now);

    stringstream ss;
    ss << put_time(&localTime, "%Y-%m-%d %H:%M:%S");

    return ss.str();
}

string Logger::levelToString(Level level)
{
    switch (level)
    {
        case Level::INFO:
            return "INFO";

        case Level::WARN:
            return "WARN";

        case Level::ERROR:
            return "ERROR";

        case Level::DEBUG:
            return "DEBUG";
    }

    return "";
}

void Logger::log(Level level, const string& message)
{
    lock_guard<std::mutex> lock(logMutex);

    string line =
        "[" +
        timestamp() +
        "] [" +
        levelToString(level) +
        "] " +
        message;

    cout << line << endl;

    if (stream.is_open())
    {
        stream << line << endl;
        stream.flush();
    }
}

void Logger::info(const string& message)
{
    log(Level::INFO, message);
}

void Logger::warn(const string& message)
{
    log(Level::WARN, message);
}

void Logger::warning(const string& message)
{
    log(Level::WARN, message);
}

void Logger::error(const string& message)
{
    log(Level::ERROR, message);
}

void Logger::debug(const string& message)
{
    log(Level::DEBUG, message);
}

void Logger::logRaw(const string& rawText)
{
    lock_guard<std::mutex> lock(logMutex);

    cout << rawText << endl;

    if (stream.is_open())
    {
        stream << rawText << endl;
        stream.flush();
    }
}