#include <libdragon.h>

#include <stdlib.h>
#include <string.h>
#include "path.h"


#define PATH_CAPACITY_INITIAL   255
#define PATH_CAPACITY_ALIGNMENT 32


static void path_resize (path_t *path, size_t min_length) {
    path->capacity = min_length > PATH_CAPACITY_INITIAL ? min_length : PATH_CAPACITY_INITIAL;
    size_t alignment = path->capacity % PATH_CAPACITY_ALIGNMENT;
    if (alignment != (PATH_CAPACITY_ALIGNMENT - 1)) {
        path->capacity += PATH_CAPACITY_ALIGNMENT - alignment;
    }
    path->buffer = realloc(path->buffer, (path->capacity + 1) * sizeof(char));
}

path_t *path_init (char *string) {
    if (string == NULL) {
        string = "";
    }
    path_t *path = calloc(1, sizeof(path_t));
    path_resize(path, strlen(string));
    memset(path->buffer, 0, path->capacity + 1);
    strcpy(path->buffer, string);
    return path;
}

void path_free (path_t *path) {
    free(path->buffer);
    free(path);
}

path_t *path_clone (path_t *path) {
    return path_init(path->buffer);
}

char *path_get (path_t *path) {
    return path->buffer;
}

char *path_last_get (path_t *path) {
    char *last_slash = strrchr(path->buffer, '/');
    return (last_slash == NULL) ? path->buffer : (last_slash + 1);
}

void path_append (path_t *path, char *string) {
    size_t buffer_length = strlen(path->buffer);
    size_t string_length = strlen(string);
    size_t new_path_length = buffer_length + string_length;
    if (new_path_length > path->capacity) {
        path_resize(path, new_path_length);
    }
    strcat(path->buffer, string);
}

void path_concat (path_t *dst, path_t *src) {
    path_append(dst, src->buffer);
}

void path_push (path_t *path, char *string) {
    if (path->buffer[strlen(path->buffer) - 1] != '/') {
        path_append(path, "/");
    }
    if (string[0] == '/') {
        string += 1;
    }
    path_append(path, string);
}

void path_pop (path_t *path) {
    char *last_slash = strrchr(path->buffer, '/');
    if (last_slash != NULL) {
        *last_slash = '\0';
    }
}

char *path_ext_get (path_t *path) {
    char *buffer = path_last_get(path);
    char *last_dot = strrchr(buffer, '.');
    if (last_dot != NULL) {
        return last_dot + 1;
    }
    return NULL;
}

void path_ext_remove (path_t *path) {
    char *buffer = path_last_get(path);
    char *last_dot = strrchr(buffer, '.');
    if (last_dot != NULL) {
        *last_dot = '\0';
    }
}

void path_ext_replace (path_t *path, char *ext) {
    path_ext_remove(path);
    path_append(path, ".");
    path_append(path, ext);
}