#include "neonucleus.h"

#ifndef NN_BAREMETAL

#ifdef NN_POSIX
#include <time.h>
#else
#include <time.h>
#include <windows.h>
#endif

#endif

void *nn_alloc(nn_Alloc *alloc, nn_size_t size) {
    if(size == 0) return alloc->proc;
    return alloc->proc(alloc->userdata, NULL, 0, size, NULL);
}

void *nn_resize(nn_Alloc *alloc, void *memory, nn_size_t oldSize, nn_size_t newSize) {
    if(oldSize == newSize) return memory;
    if(newSize == 0) {
        nn_dealloc(alloc, memory, oldSize);
        return alloc->proc;
    }
    if(memory == NULL) {
        return nn_alloc(alloc, newSize);
    }
    if(memory == alloc->proc) {
        if(newSize == 0) return memory;
        return nn_alloc(alloc, newSize);
    }
    return alloc->proc(alloc->userdata, memory, oldSize, newSize, NULL);
}

void nn_dealloc(nn_Alloc *alloc, void *memory, nn_size_t size) {
    if(memory == NULL) return; // matches free()
    if(memory == alloc->proc) return; // 0-sized memory
    alloc->proc(alloc->userdata, memory, size, 0, NULL);
}

#ifndef NN_BAREMETAL

#include <stdlib.h>

static void *nn_libcAllocProc(void *_, void *ptr, nn_size_t oldSize, nn_size_t newSize, void *__) {
    if(newSize == 0) {
        //printf("Freed %lu bytes from %p\n", oldSize, ptr);
        free(ptr);
        return NULL;
    } else {
        void *rptr = realloc(ptr, newSize);
        //printf("Allocated %lu bytes for %p\n", newSize - oldSize, rptr);
        return rptr;
    }
}

nn_Alloc nn_libcAllocator(void) {
    return (nn_Alloc) {
        .userdata = NULL,
        .proc = nn_libcAllocProc,
    };
}

static nn_size_t nni_rand(void *userdata) {
    return rand();
}

nn_Rng nn_libcRng(void) {
    srand(time(NULL));
    return (nn_Rng) {
        .userdata = NULL,
        .maximum = RAND_MAX,
        .proc = nni_rand,
    };
}

nn_Context nn_libcContext(void) {
    return (nn_Context) {
        .allocator = nn_libcAllocator(),
        .clock = nn_libcRealTime(),
        .lockManager = nn_libcMutex(),
        .rng = nn_libcRng(),
    };
}
#endif


// Utilities, both internal and external
char *nn_strdup(nn_Alloc *alloc, const char *s) {
    nn_size_t l = nn_strlen(s);
    char *m = nn_alloc(alloc, l+1);
    if(m == NULL) return m;
    return nn_strcpy(m, s);
}

void *nn_memdup(nn_Alloc *alloc, const void *buf, nn_size_t len) {
    char *m = nn_alloc(alloc, len);
    if(m == NULL) return m;
    nn_memcpy(m, buf, len);
    return m;
}

void nn_deallocStr(nn_Alloc *alloc, char *s) {
    if(s == NULL) return;
    nn_dealloc(alloc, s, nn_strlen(s)+1);
}

static void nni_randomHex(nn_Context *ctx, char *buf, nn_size_t len) {
    const char *hex = "0123456789abcdef";

    for(nn_size_t i = 0; i < len; i++) {
        int r = nn_rand(&ctx->rng) % 16;
        buf[i] = hex[r];
    }
}

nn_address nn_randomUUID(nn_Context *ctx) {
    nn_address addr = nn_alloc(&ctx->allocator, 37);
    if(addr == NULL) return NULL;
    nni_randomHex(ctx, addr + 0, 8);
    addr[8] = '-';
    nni_randomHex(ctx, addr + 9, 4);
    addr[13] = '-';
    nni_randomHex(ctx, addr + 14, 4);
    addr[18] = '-';
    nni_randomHex(ctx, addr + 19, 4);
    addr[23] = '-';
    nni_randomHex(ctx, addr + 24, 12);
    addr[36] = '\0';


    // UUIDv4 variant 1
    addr[14] = '4';
    addr[19] = '1';
    return addr;
}

nn_size_t nn_rand(nn_Rng *rng) {
    return rng->proc(rng->userdata);
}

// returns from 0 to 1 (inclusive)
double nn_randf(nn_Rng *rng) {
    double x = nn_rand(rng);
    return x / rng->maximum;
}

// returns from 0 to 1 (exclusive)
double nn_randfe(nn_Rng *rng) {
    double x = nn_rand(rng);
    if(x >= rng->maximum) return 0;
    return x / rng->maximum;
}

#ifndef NN_BAREMETAL

#ifdef NN_POSIX

static double nni_realTime(void) {
    struct timespec time;
    if(clock_gettime(CLOCK_MONOTONIC, &time) < 0) return 0; // oh no
    return time.tv_sec + ((double)time.tv_nsec) / 1e9;
}

#else

static double nni_realTime(void) {
    LARGE_INTEGER frequency = {0};
    if(!QueryPerformanceFrequency(&frequency)) return 0;

    LARGE_INTEGER now = {0};
    if(!QueryPerformanceCounter(&now)) return 0;

    return (double)now.QuadPart / frequency.QuadPart;
}

#endif

double nni_realTimeClock(void *_) {
    return nni_realTime();
}

nn_Clock nn_libcRealTime(void) {
    return (nn_Clock) {
        .userdata = NULL,
        .proc = nni_realTimeClock,
    };
}

#endif

// TODO: use OKLAB the color space for more accurate results.

typedef struct nn_rgbColor {
    double r, g, b;
} nn_rgbColor;

nn_rgbColor nni_splitColorToRgb(int color) {
    double r = (color & 0xFF0000) >> 16;
    double g = (color & 0x00FF00) >> 8;
    double b = color & 0x0000FF;

    int max = 0xFF;
    return (nn_rgbColor) {
        .r = r / max,
        .g = g / max,
        .b = b / max,
    };
}

double nn_colorDistance(int colorA, int colorB) {
    if(colorA == colorB) return 0;
    nn_rgbColor a = nni_splitColorToRgb(colorA);
    nn_rgbColor b = nni_splitColorToRgb(colorB);

    nn_rgbColor delta;
    delta.r = a.r - b.r;
    delta.g = a.g - b.g;
    delta.b = a.b - b.b;

    return 0.2126 * delta.r*delta.r + 0.7152 * delta.g*delta.g + 0.0722 * delta.b*delta.b;
}

int nn_mapColor(int color, int *palette, int paletteSize) {
    if(paletteSize <= 0) return color;
    int bestColor = palette[0];
    double fitness = nn_colorDistance(color, bestColor);

    for(int i = 1; i < paletteSize; i++) {
        double dist = nn_colorDistance(color, palette[i]);
        if(dist < fitness) {
            bestColor = palette[i];
            fitness = dist;
        }
        if(bestColor == color) return color;
    }
    return bestColor;
}

void nn_memset(void *buf, unsigned char byte, nn_size_t len) {
    if(buf == NULL) return;
    unsigned char *bytes = buf;
    for(nn_size_t i = 0; i < len; i++) bytes[i] = byte;
}

void nn_memcpy(void *dest, const void *src, nn_size_t len) {
    if(dest == NULL) return;
    if(src == NULL) return;
    if(len == 0) return;

    char *destBytes = dest;
    const char *srcBytes = src;
    for(nn_size_t i = 0; i < len; i++) {
        destBytes[i] = srcBytes[i];
    }
}

char *nn_strcpy(char *dest, const char *src) {
    if(dest == NULL) return dest;
    nn_size_t i = 0;
    while(src[i]) {
        dest[i] = src[i];
        i++;
    }
    dest[i] = 0;
    return dest;
}

int nn_strcmp(const char *a, const char *b) {
    nn_size_t i = 0;
    while(NN_TRUE) {
        unsigned char ca = a[i];
        unsigned char cb = b[i];

        if(ca < cb) {
            return -1;
        }
        if(ca > cb) {
            return -1;
        }
        if(ca == 0 && cb == 0) { // reached terminator
            return 0;
        }
        i++;
    }
}

const char *nn_strchr(const char *str, int ch) {
    if(str == NULL) return NULL;
    nn_size_t i = 0;
    while(NN_TRUE) {
        if(str[i] == ch) return str + i;
        if(str[i] == 0) return NULL;
        i++;
    }
}

nn_size_t nn_strlen(const char *a) {
    if(a == NULL) return 0;
    nn_size_t l = 0;
    while(a[l]) l++;
    return l;
}

nn_bool_t nn_strbegin(const char *s, const char *prefix) {
    nn_size_t i = 0;
    while(true) {
        if(prefix[i] == 0) return true; // prefix over, it matched
        if(s[i] == 0) return false; // string over, it didn't match
        if(s[i] != prefix[i]) return false;
        i++;
    }
}

nn_bool_t nn_error_isEmpty(nn_errorbuf_t buf) {
    if(buf == NULL) return true;
    return buf[0] == 0;
}

void nn_error_write(nn_errorbuf_t buf, const char *s) {
    if(buf == NULL) return;
    for(nn_size_t i = 0; i < NN_MAX_ERROR_BUFFER; i++) {
        buf[i] = s[i];
        if(s[i] == 0) break;
    }
    // just to be sure
    buf[NN_MAX_ERROR_BUFFER-1] = 0;
}

void nn_error_clear(nn_errorbuf_t buf) {
    if(buf == NULL) return;
    buf[0] = 0;
}


nn_bool_t nn_path_hasSlash(const char *path) {
    while(*path) {
        if(*path == '/') return true;
        path++;
    }
    return false;
}

nn_size_t nn_path_firstSlash(const char *path) {
    for(nn_size_t i = 0; path[i]; i++) {
        if(path[i] == '/') return i;
    }
    return 0; // should never happen
}

nn_size_t nn_path_lastSlash(const char *path) {
    nn_size_t slash = 0;
    for(nn_size_t i = 0; path[i]; i++) {
        if(path[i] == '/') slash = i;
    }
    return slash;
}

// returns whether it is the last name
nn_bool_t nn_path_firstName(const char *path, char firstDirectory[NN_MAX_PATH], char subpath[NN_MAX_PATH]) {
    if(!nn_path_hasSlash(path)) {
        nn_strcpy(firstDirectory, path);
        nn_strcpy(subpath, "");
        return true; // end
    }
    nn_size_t slash = nn_path_firstSlash(path);

    nn_memcpy(firstDirectory, path, slash);
    firstDirectory[slash] = 0;

    nn_strcpy(subpath, path + slash + 1);
    return false;
}

// returns whether it is the only name
nn_bool_t nn_path_lastName(const char *path, char name[NN_MAX_PATH], char parent[NN_MAX_PATH]) {
    if(!nn_path_hasSlash(path)) {
        nn_strcpy(name, path);
        nn_strcpy(parent, "");
        return true; // end
    }

    nn_size_t slash = nn_path_lastSlash(path);
    nn_strcpy(name, path + slash + 1);

    nn_memcpy(parent, path, slash);
    parent[slash] = 0;
    return false;
}

const char *nn_path_illegal = "\"\\:*?<>|";

nn_bool_t nn_path_isValid(const char *path) {
    // if we don't check for these, we will be FUCKED

    for(nn_size_t i = 0; nn_path_illegal[i] != '\0'; i++) {
        if(nn_strchr(path, nn_path_illegal[i]) != NULL) return false;
    }
    return nn_strlen(path) < NN_MAX_PATH; // less then because we depend on the terminator
}

static nn_bool_t nni_path_isAllDots(const char *path, nn_size_t len) {
    for(nn_size_t i = 0; i < len; i++) {
        if(path[i] != '.') return false;
    }
    return true;
}

nn_bool_t nn_path_canonical(const char *path, char canonical[NN_MAX_PATH]) {
    // attempts to convert a random barely legal path
    // if this shit is ever bugged and a sandbox escape is done
    // by tricking it into sneaking some .. in there
    // !!!! WE WILL BE FUCKED, THE SERVER WILL BE HACKED, AND WE WILL DIE !!!!

    if(!nn_path_isValid(path)) {
        // HELL NO
        return true;
    }

    // 0'd out because it fills it up with terminators, simplifying the rest of the code
    // in theory this is suboptimal, however, I'm lazy
    nn_memset(canonical, 0, NN_MAX_PATH);
    nn_size_t ptr = 0;
    nn_size_t i = 0;

    // TODO: burn it with fire and get banned from programming
    while(true) {
        while(path[i] == '/') i++; // just do not ask
        if(path[i] == 0) break;

        const char *subpath = path + i;
        nn_size_t len = nn_path_firstSlash(subpath);

        if(len == 0) {
            len = nn_strlen(path) - i;
        }

        if(nni_path_isAllDots(subpath, len)) {
            // we don't actually resolve them because they shouldn't be there
            // to begin with
            i += len;
            continue;
        }

        if(ptr == 0) {
            // at the start
            nn_memcpy(canonical, subpath, len);
            ptr = len;
            i += len;
            continue;
        }
        // just append to it
        canonical[ptr] = '/';
        ptr++;
        nn_memcpy(canonical + ptr, subpath, len);
        ptr += len;
        i += len;
        continue;
    }

    return false;
}