#include "user/lib/Logger.h"
#include "kernel/Globals.h"

bool Logger::kout_enabled = false;
bool Logger::serial_enabled = true;
const Semaphore Logger::sem = Semaphore(1);

Logger::LogLevel Logger::level = Logger::TRACE;

void Logger::log(char* message, CGA::color col) const {
    if (Logger::kout_enabled) {
        CGA::color old_col = kout.color_fg;
        kout << fgc(col)
             << Logger::level_to_string(this->current_message_level) << "::"
             << this->name << ":: "
             << message << fgc(old_col);
        kout.flush();  // Don't add newline, Logger already does that
    }
    if (Logger::serial_enabled) {
        serial.write(Logger::level_to_string(this->current_message_level));
        serial.write("::");
        serial.write(this->name);
        serial.write(":: ");
        serial.write(message);
        serial.write('\r');
        // serial.write("\r\n");
    }
}

void Logger::flush() {
    this->buffer[this->pos] = '\0';

    switch (this->current_message_level) {
    case Logger::TRACE:
        this->trace(this->buffer);
        break;
    case Logger::DEBUG:
        this->debug(this->buffer);
        break;
    case Logger::ERROR:
        this->error(this->buffer);
        break;
    case Logger::INFO:
        this->info(this->buffer);
        break;
    }

    this->current_message_level = Logger::INFO;
    this->pos = 0;
}

void Logger::trace(char* message) const {
    if (Logger::level <= Logger::TRACE) {
        this->log(message, CGA::GREEN);
    }
}

void Logger::debug(char* message) const {
    if (Logger::level <= Logger::DEBUG) {
        this->log(message, CGA::LIGHT_BLUE);
    }
}

void Logger::error(char* message) const {
    if (Logger::level <= Logger::ERROR) {
        this->log(message, CGA::RED);
    }
}

void Logger::info(char* message) const {
    if (Logger::level <= Logger::TRACE) {
        this->log(message, CGA::CYAN);
    }
}

// Manipulatoren
Logger& TRACE(Logger& log) {
    log.current_message_level = Logger::TRACE;
    return log;
}

Logger& DEBUG(Logger& log) {
    log.current_message_level = Logger::DEBUG;
    return log;
}

Logger& ERROR(Logger& log) {
    log.current_message_level = Logger::ERROR;
    return log;
}

Logger& INFO(Logger& log) {
    log.current_message_level = Logger::INFO;
    return log;
}