#include <string.h>
#include <stdlib.h>
#include "ucore/strvec.h"

void uc_strv_init(struct UCStrv *strv)
{
    strv->cnt = 0;
    strv->allocated = 0;
    strv->strings = NULL;
}

int uc_strv_grow(struct UCStrv *strv, size_t cnt)
{

    size_t allocate = strv->allocated + cnt;
    void *tmp = realloc(strv->strings, sizeof *strv->strings * allocate);

    if (tmp == NULL) {
        return -1;
    }
    strv->strings = tmp;
    strv->allocated = allocate;

    return 0;
}

int uc_strv_check_expand(struct UCStrv *strv)
{
    int rc = 0;

    if (strv->cnt >= strv->allocated) {
        //make some extra room
        rc = uc_strv_grow(strv, 4); 
    }

    return rc;
}

int uc_strv_append(struct UCStrv *strv, const char *restrict str)
{
    int rc;
    char *dup;
    
    rc = uc_strv_check_expand(strv);
    if (rc != 0) {
        return rc;
    }

    dup = strdup(str);
    if (dup == NULL) {
        return -2;
    }

    strv->strings[strv->cnt] = dup;
    strv->cnt++;

    return 0;
}

int uc_strv_append_nocopy(struct UCStrv *strv, char *restrict str)
{
    int rc;
    rc = uc_strv_check_expand(strv);

    if (rc != 0) {
        return rc;
    }

    strv->strings[strv->cnt] = str;
    strv->cnt++;

    return 0;
}

int uc_strv_null_terminate(struct UCStrv *strv)
{
    int rc;

    rc = uc_strv_check_expand(strv);
    if (rc != 0) {
        return rc;
    }

    strv->strings[strv->cnt] = NULL;

    return 0;
}

void uc_strv_clear(struct UCStrv *strv)
{
    size_t i;

    for (i = 0; i < strv->cnt; i++) {
        free(strv->strings[i]);
        strv->strings[i] = NULL;
    }

    strv->cnt = 0;
}

void uc_strv_destroy(struct UCStrv *strv)
{
    uc_strv_clear(strv);
    free(strv->strings);
    strv->strings = NULL;
}

int uc_strv_append_all(struct UCStrv *restrict a, struct UCStrv *restrict b, int copy)
{
    size_t i;
    int rc;

    rc = uc_strv_grow(a, a->allocated + b->cnt);
    if (rc != 0) {
        return rc;
    }

    for (i = 0; i < b->cnt; i++) {
        if (copy) {
            rc = uc_strv_append(a, b->strings[i]);
        } else {
            rc = uc_strv_append_nocopy(a, b->strings[i]);
        }

        if (rc != 0) {
            return rc;
        }
    }

    return 0;
}

void uc_strv_reverse(struct UCStrv *strv)
{
    for (size_t i = 0; i < strv->cnt / 2; i++) {

        size_t k = strv->cnt - i - 1;
        char *tmp = strv->strings[i];

        strv->strings[i] = strv->strings[k];
        strv->strings[k] = tmp;
    }
}