#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].path != NULL)
            free(list->items[i].filename);
        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
 */
bool arraylist_includes(ArrayList *list, char* filename, char *path)
{
    char *current_whole_path = strcat(path, filename);
    for (int i = 0; i < list->length; i++) {
        char *list_whole_path = strcat(list->items[i].path, list->items[i].filename);
        if (strcmp(list_whole_path, current_whole_path) == 0) {
            return true;
        }
    }

    return false;
}

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

    free(list->items[index].filename);
    free(list->items[index].path);
    free(list->items[index].stats);
    free(list->items[index].type);
    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 *filename, char *path, char *stats, char *type, wchar_t *icon, int color, bool marked, bool force)
{
    char *filename_cp = NULL;
    char *path_cp = NULL;
    char *stats_cp = NULL;
    char *type_cp = NULL;
    wchar_t *icon_cp = NULL;

    if (filename_cp != NULL) {
        filename_cp = strdup(filename_cp);
        if (filename_cp == NULL)
            perror("can't add filename to arraylist");
    }
    if (path_cp != NULL) {
        path_cp = strdup(path_cp);
        if (path_cp == NULL)
            perror("can't add path to arraylist");
    }
    if (stats != NULL) {
        stats_cp = strdup(stats);
        if (stats_cp == NULL)
            perror("can't add stats to arraylist");
    }
    if (type != NULL) {
        type_cp = strdup(type);
        if (type_cp == NULL)
            perror("can't add type to arraylist");
    }
    if (icon != NULL) {
        icon_cp = wcsdup(icon);
        if (icon_cp == NULL)
            perror("can't add icon to arraylist");
    }

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

    if (list->capacity != list->length) {
        if (marked) {
            for (int i = 0; i < list->length; i++) {
                char *list_whole_path = strcat(list->items[i].path, list->items[i].filename);
                char *current_whole_path = strcat(new_file.path, new_file.filename);
                if (strcmp(list_whole_path, current_whole_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)
{
    file file = list->items[index];
    char *name = strdup(file.path);
    name = basename(name);
    if (name == NULL)
        perror("ccc");

    wchar_t *icon = wcsdup(file.icon);

    char *stats = NULL;
    size_t length;
    if (detail) {
        stats = strdup(file.stats);
        length = strlen(name) + strlen(stats) + 7;   /* 4 for icon, 2 for space and 1 for null */
        if (stats == NULL) {
            perror("ccc");
        }
    } else {
        length = strlen(name) + 6;   /* 4 for icon, 1 for space and 1 for null */
    }
    char *line = memalloc(length * sizeof(char));
   
    if (detail) {
        snprintf(line, length, "%s %ls %s", stats, icon, name);
    } else {
        snprintf(line, length, "%ls %s", icon, name);
    }

    return line;
}