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

#include "util.h"

/* files in a link list data structure */
typedef struct file {
    char *path;
    char *stats;
    char *type;
    int color;
    struct file *next;
} file;

file *files = NULL;
file *marked = NULL;

/*
 * Get length of files linked list
 */
long files_len()
{
    file *current = files;
    int count = 0;
    while (current != NULL) {
        count++;
        current = current->next;
    }
    return count;
}

/*
 * Get length of marked files
 */
long marked_len()
{
    file *current = marked;
    int count = 0;
    while (current != NULL) {
        count++;
        current = current->next;
    }
    return count;

}

void free_file(file *toremove)
{
    if (toremove->type != NULL)
        free(toremove->type);
    if (toremove->path != NULL)
        free(toremove->path);
    if (toremove->stats != NULL)
        free(toremove->stats);
    free(toremove);
}

void clear_files()
{
    file *tmp;
    while (files != NULL) {
        tmp = files;
        files = files->next;
        free_file(tmp);
    }
}

void clear_marked()
{
    file *tmp;
    while (marked != NULL) {
        tmp = marked;
        files = marked->next;
        free_file(tmp);
    }
}

long add_file(char *filepath, char *stats, char *type, int color)
{
    file *current = files;
    file *new_file = memalloc(sizeof(file));
    char *buf = strdup(filepath);
    char *buf2 = strdup(stats);
    char *buf3 = strdup(type);
    int buf4 = color;

    if (buf == NULL || buf2 == NULL || buf3 == NULL)
        perror("ccc");

    new_file->path = buf;
    new_file->stats = buf2; 
    new_file->type = buf3;
    new_file->color = buf4;
    new_file->next = NULL;

    if (current == NULL) {
        files = new_file;
        return 0;
    }
    long index = 1;
    while (current->next != NULL) {
        current = current->next;
        index++;
    }
    current->next = new_file;
    return index;
}

void remove_marked(file *marked_file)
{
    /* If the head node itself is marked for removal */
    if (marked == marked_file) {
        marked = marked->next;
        free_file(marked_file);
        return;
    }
    

    /* Search for the marked file node in the list */
    file* temp = marked;
    while (temp != NULL && temp->next != marked_file) {
        temp = temp->next;
    }

    /* If the marked file node is found, remove it from the list */
    if (temp != NULL) {
        temp->next = marked_file->next;
        free_file(marked_file);
    }
}

/*
 * force will not remove duplicate marked files, instead it just skip adding
 */
long add_marked(char *filepath, char *type, bool force)
{
    file *current = marked;
    file *new_file = memalloc(sizeof(file));
    char *buf = strdup(filepath);
    char *buf2 = strdup(type);
    if (buf == NULL || buf2 == NULL) {
        perror("ccc");
    }
    new_file->path = buf;
    new_file->type = buf2;
    new_file->stats = NULL;
    new_file->color = 0;
    new_file->next = NULL;
    if (current == NULL) {
        marked = new_file;
        return 0;
    }
    
    long index = 1;
    while (current->next != NULL) {
        if (strcmp(current->path, new_file->path) == 0) {
            if (force) {
                return index;
            } else {
                remove_marked(current);
                free_file(new_file);
                return -1;
            }
        }
        current = current->next;
        index++;
    }
    if (strcmp(current->path, new_file->path) == 0){
        if (force) {
            return 0;
        } else {
            remove_marked(current);
            free_file(new_file);
            return -1;
        }
    }
    current->next = new_file;
    return index;
}

file *get_marked(long index)
{
    file *current = marked;
    if (index == 0) {
        return current;
    }
    if (index > files_len()) {
        return NULL;
    }
    for (long i = 0; i < index; i++) {
        current = current->next;
    }
    return current;
}

bool in_marked(char *path)
{
    file *tmp = marked;
    if (tmp == NULL)
        return false;

    while (tmp != NULL) {
        if (strcmp(path, tmp->path) == 0) {
            return true;
        }
        tmp = tmp->next;
    }
    return false;
}

file *get_file(long index)
{
    file *current = files;
    if (index == 0) {
        return current;
    }
    if (index > files_len()) {
        return NULL;
    }
    for (long i = 0; i < index; i++) {
        current = current->next;
    }
    return current;
}

char *get_filepath(long index)
{
    file *file = get_file(index);
    if (file != NULL) {
        char *name = strdup(file->path);
        if (!name) {
            perror("ccc");
        }
        return name;
    } else {
        return NULL;
    }
}

int get_color(long index)
{
    file *file = get_file(index);
    if (file != NULL) {
        int color = file->color;
        if (!color) {
            perror("ccc");
        }
        return color;
    } else {
        return 8; /* white */
    }
}

/*
 * Construct a formatted line for display
 */
char *get_line(long index)
{
    file *file = get_file(index);
    if (file != NULL) {
        char *name = strdup(file->path);
        char *stats = strdup(file->stats);
        size_t length = strlen(name) + strlen(stats) + 2;   /* one for space and one for null */
        char *line = memalloc(length * sizeof(char));

        name = basename(name);
        if (name == NULL || stats == NULL)
            perror("ccc");

        snprintf(line, length, "%s %s", stats, name);

        return line;
    } else {
        return NULL;
    }
}