#include <string.h>
#include <stdint.h>

// Source: http://www.strudel.org.uk/itoa/
char* itoa(uint32_t val, uint8_t base) {
    static char buf[32] = { 0 };
    
    if (base == 0 || base > 16) { return NULL; }

    if (val == 0) {
        buf[0] = '0';
        return buf;
    }
 
    int i;
    for (i = 30; val && i; i--, val /= base) {
        buf[i] = "0123456789abcdef"[val % base];
    }

    return &buf[i + 1];
}

uint32_t atoi(const char* str) {
    uint32_t res = 0;

    // TODO: Add error checking

    while (*str >= '0' && *str <= '9') {
        res = res * 10 + (*str++ - '0');
    }

    return res;
}

int isSpace(char c) {
    if (c == ' ' || c == '\t' || c == '\n') return 1;
    else return 0;
}

// Some implementations taken from https://en.wikibooks.org/wiki/C_Programming/String_manipulation
char* strcpy(char* str1, const char* str2) {
    char* dst = str1;
    const char* src = str2;
    while ((*dst++ = *src++) != '\0');
    return str1;
}

char* strncpy(char* str1, const char* str2, size_t n) {
    char* dst = str1;
    const char* src = str2;
    while (n > 0) {
        n--;
        if ((*dst++ = *src++) == '\0') {
            memset(dst, '\0', n);
            break;
        }
    }
    return str1;
}

char* strcat(char* str1, const char* str2) {
    char* s = str1;
    while (*s != '\0') { s++; }
    strcpy(s, str2);
    return str1;
}

char* strncat(char* str1, const char* str2, size_t n) {
    char* s = str1;
    while (*s != '\0') { s++; }
    strncpy(s, str2, n);
    return str1;
}

char* strchr(const char* str, char c) {
    const char* s = str;
    while (*s != '\0' && *s != c) { s++; }
    return (*s == c) ? (char*)s : NULL;
}

char* strrchr(const char* str, char c) { return NULL; } // TODO

int strcmp(const char* str1, const char* str2) {
    while (*str1 != '\0' && *str1 == *str2) {
        str1++;
        str2++;
    }

    uint8_t c1 = (*(uint8_t*)str1);
    uint8_t c2 = (*(uint8_t*)str2);

    return ((c1 < c2) ? -1 : (c1 > c2));
}

int strncmp(const char* str1, const char* str2, size_t n) {
    if (n == 0) { return 0; }
    while (n-- > 0 && *str1 == *str2) {
        if (n == 0 || *str1 == '\0') { return 0; }
        str1++;
        str2++;
    }

    uint8_t c1 = (*(uint8_t*)str1);
    uint8_t c2 = (*(uint8_t*)str2);

    return ((c1 < c2) ? -1 : (c1 > c2));
}

uint32_t strlen(const char* str) {
    const char* p = str;
    while (*++p != '\0');
    return (uint32_t)(p - str);
}

// Memory functions
void* memcpy(void* src, void* dst, uint32_t length) {
    for (uint32_t i = 0; i < length; i++) {
        *((uint8_t*)dst + i) = *((uint8_t*)src + i);
    }
    return dst;
}

void* memcpy32(void* src, void* dst, uint32_t length) {
    for (uint32_t i = 0; i * 4 < length; i++) {
        *((uint32_t*)dst + i) = *((uint32_t*)src + i);
    }
    return dst;
}

void* memset(void* dst, uint8_t value, uint32_t length) {
    for (uint32_t i = 0; i < length; i++) {
        *(uint8_t*)(dst + i) = value;
    }
    return dst;
}