#include <stdarg.h>
#include <stdio.h>
#include <string.h>
#include <sys/file.h>
#include <unistd.h>

#include <log.h>
#include <util.h>

static const char *colors[] = {
    [0] = ANSI_BLINK ANSI_REVERSE ANSI_BOLD ANSI_RED,
    [1] = ANSI_REVERSE ANSI_BOLD ANSI_RED,
    [2] = ANSI_BOLD ANSI_RED,
    [3] = ANSI_RED,
    [4] = ANSI_BOLD,
    [5] = ANSI_BRIGHT_WHITE,
    [6] = ANSI_RESET,
    [7] = ANSI_ITALIC ANSI_BRIGHT_BLUE,
};

int console_lock = 0;

int print(const char *fmt, ...)
{
    int loglevel = 5;
    if(fmt[0] == LOG_SOH_ASCII) {
        loglevel = (fmt[1] - 0x30) % 10;
        fmt += 2;
    }

    /* we essentially print the user's raw input to its own buffer,
       later we will parse it and print out ANSI colors and what not */
    char buf[512];

    va_list ap;
    va_start(ap, fmt);
    vsnprintf(buf, 512, fmt, ap);
    va_end(ap);
    buf[512 - 1] = '\0';

    size_t colon = 0;
    for(; colon < strlen(buf); ++colon) {
        if(buf[colon] == ':')
            break;
    }

    /* spin lock, at the cost of architecture portability
       concurrency is something that we need to adjust for, and the
       console will be scrambled and unreadable if we allow writing all
       at the same time. I considered simply writing all at once, but
       ended up just not caring enough to the point where spinlocks
       prevail. */
    __asm__(".spin_lock:");
    __asm__("mov rax, 1");
    __asm__("xchg rax, [console_lock]");
    __asm__("test rax, rax");
    __asm__("jnz .spin_lock");

    /* we want to support stuff without colons, but frankly I havent
       tested this at time of writing. will find out later */
    if(buf[colon] == ':') {
        writeputs(ANSI_RESET);
        writeputs(colors[loglevel]);
        writeputs(ANSI_YELLOW);
        write(STDOUT_FILENO, buf, colon);
        writeputs(ANSI_RESET);
    }
    writeputs(colors[loglevel]);
    if(colon && *(buf + colon)) {
        writeputs(buf + colon);
    } else {
        writeputs(buf);
    }
    writeputs(ANSI_RESET);
    write(STDOUT_FILENO, "\n", 1);

    console_lock = 0;
    return 0;
}