#include <string.h>
#include <stdlib.h>
#include <stdbool.h>
#include <libgen.h>
#include <wchar.h>

#include "util.h"
#include "file.h"

ArrayList *arraylist_init(size_t capacity)
{
    ArrayList *list = memalloc(sizeof(ArrayList));
    list->length = 0;
    list->capacity = capacity;
    list->items = memalloc(capacity * sizeof(file));

    return list;
}

void arraylist_free(ArrayList *list)
{
    for (size_t i = 0; i < list->length; i++) {
        if (list->items[i].name != NULL)
            free(list->items[i].name);
        if (list->items[i].path != NULL)
            free(list->items[i].path);
        if (list->items[i].type != NULL)
            free(list->items[i].type);
        if (list->items[i].stats != NULL)
            free(list->items[i].stats);
        if (list->items[i].icon != NULL)
            free(list->items[i].icon);
    }

    free(list->items);
    list->length = 0;
}

/*
 * Check if the file is in the arraylist
 * Treat filepath as base name if bname is true
 */
long arraylist_search(ArrayList *list, char *filepath, bool bname)
{
    for (long i = 0; i < list->length; i++) {
        if (!bname && strcmp(list->items[i].path, filepath) == 0) {
            return i;
        }
        if (bname) {
            if (strcmp(list->items[i].name, filepath) == 0) {
                return i;
            }
        }
    }
    return -1;
}

void arraylist_remove(ArrayList *list, long index)
{
    if (index >= list->length)
        return;

    free(list->items[index].name);
    free(list->items[index].path);
    free(list->items[index].type);
    free(list->items[index].stats);
    free(list->items[index].icon);

    for (long i = index; i < list->length - 1; i++)
        list->items[i] = list->items[i + 1];

    list->length--;
}

/*
 * Force will not remove duplicate marked files, instead it just skip adding
 */
void arraylist_add(ArrayList *list, char *name, char *path, char *stats, char *type, wchar_t *icon, int color, bool marked, bool force)
{
    char *name_cp = NULL;
    char *path_cp = NULL;
    char *type_cp = NULL;
    char *stats_cp = NULL;
    wchar_t *icon_cp = NULL;

    if (name != NULL)
        name_cp = estrdup(name);
    if (path != NULL)
        path_cp = estrdup(path);
    if (type != NULL)
        type_cp = estrdup(type);
    if (stats != NULL)
        stats_cp = estrdup(stats);
    if (icon != NULL)
        icon_cp = ewcsdup(icon);

    /* name, path, stats, type, icon, color */
    file new_file = { name_cp, path_cp, type_cp, stats_cp, icon_cp, color };

    if (list->capacity != list->length) {
        if (marked) {
            for (int i = 0; i < list->length; i++) {
                if (strcmp(list->items[i].path, new_file.path) == 0) {
                    if (!force)
                        arraylist_remove(list, i);
                    return;
                }
            }
        }
        list->items[list->length] = new_file;
    } else {
        int new_cap = list->capacity * 2;
        file *new_items = memalloc(new_cap * sizeof(file));
        file *old_items = list->items;
        list->capacity = new_cap;
        list->items = new_items;

        for (int i = 0; i < list->length; i++)
            new_items[i] = old_items[i];

        free(old_items);
        list->items[list->length] = new_file;
    }
    list->length++;
}

/*
 * Construct a formatted line for display
 */
char *get_line(ArrayList *list, long index, bool detail, bool icons)
{
    file file = list->items[index];
    char *name = estrdup(file.name);
    wchar_t *icon = ewcsdup(file.icon);

    size_t name_len = strlen(name);
    char *stats = NULL;
    size_t length;
    if (detail) {
        stats = estrdup(file.stats);
        length = name_len + strlen(stats) + 7;   /* 4 for icon, 2 for space and 1 for null */
    } else {
        length = name_len + 6;   /* 4 for icon, 1 for space and 1 for null */
    }

    char *line = memalloc(length * sizeof(char));
    line[0] = '\0';
    if (detail) {
        strcat(line, stats);
        strcat(line, " ");
    }
    if (icons) {
        char *tmp = memalloc(8 * sizeof(char));
        snprintf(tmp, 8, "%ls", icon);
        strcat(line, tmp);
        strcat(line, " ");
        free(tmp);
    }
    strcat(line, name);
 
    return line;
}