#include "list_ops.h"
#include <stdlib.h>

// Funkcia na vytvorenie noveho zoznamu
list_t *new_list(size_t length, list_element_t elements[]) {
    list_t *list = malloc(sizeof(list_t) + sizeof(list_element_t) * length);
    if (list == NULL) {
        return NULL;
    }

    list->length = length;

    // Skopírujeme prvky zo vstupného poľa
    for (size_t i = 0; i < length; i++) {
        list->elements[i] = elements[i];
    }

    return list;
}

// Funkcia na spojenie dvoch zoznamov
list_t *append_list(list_t *list1, list_t *list2) {
    size_t new_length = list1->length + list2->length;

    list_t *result = malloc(sizeof(list_t) + sizeof(list_element_t) * new_length);
    if (result == NULL) {
        return NULL;
    }

    result->length = new_length;

    // Skopírujeme prvky z prvého zoznamu
    for (size_t i = 0; i < list1->length; i++) {
        result->elements[i] = list1->elements[i];
    }

    // Potom skopírujeme prvky z druhého zoznamu
    for (size_t i = 0; i < list2->length; i++) {
        result->elements[list1->length + i] = list2->elements[i];
    }

    return result;
}

// Funkcia na filtrovanie prvkov podľa podmienky
list_t *filter_list(list_t *list, bool (*filter)(list_element_t)) {
    // Najprv zistíme, koľko prvkov bude vo výsledku
    size_t count = 0;
    for (size_t i = 0; i < list->length; i++) {
        if (filter(list->elements[i])) {
            count++;
        }
    }

    list_t *result = malloc(sizeof(list_t) + sizeof(list_element_t) * count);
    if (result == NULL) {
        return NULL;
    }

    result->length = count;

    // Pridáme len tie prvky, ktoré prejdú filtrom
    size_t index = 0;
    for (size_t i = 0; i < list->length; i++) {
        if (filter(list->elements[i])) {
            result->elements[index] = list->elements[i];
            index++;
        }
    }

    return result;
}

// Vracia dĺžku zoznamu
size_t length_list(list_t *list) {
    return list->length;
}

// Funkcia na mapovanie - aplikuje funkciu na každý prvok
list_t *map_list(list_t *list, list_element_t (*map)(list_element_t)) {
    list_t *result = malloc(sizeof(list_t) + sizeof(list_element_t) * list->length);
    if (result == NULL) {
        return NULL;
    }

    result->length = list->length;

    for (size_t i = 0; i < list->length; i++) {
        result->elements[i] = map(list->elements[i]);
    }

    return result;
}

// Funkcia foldl - zľava doprava
list_element_t foldl_list(list_t *list, list_element_t initial,
                          list_element_t (*foldl)(list_element_t, list_element_t)) {
    list_element_t result = initial;

    for (size_t i = 0; i < list->length; i++) {
        result = foldl(result, list->elements[i]);
    }

    return result;
}

// Funkcia foldr - sprava doľava
list_element_t foldr_list(list_t *list, list_element_t initial,
                          list_element_t (*foldr)(list_element_t, list_element_t)) {
    list_element_t result = initial;

    // Používame int namiesto size_t, aby sme vedeli ísť dozadu
    int i = (int)list->length - 1;

    while (i >= 0) {
        result = foldr(list->elements[i], result);
        i--;
    }

    return result;
}

// Funkcia na otočenie zoznamu
list_t *reverse_list(list_t *list) {
    list_t *result = malloc(sizeof(list_t) + sizeof(list_element_t) * list->length);
    if (result == NULL) {
        return NULL;
    }

    result->length = list->length;

    // Prekopírujeme prvky v opačnom poradí
    for (size_t i = 0; i < list->length; i++) {
        result->elements[i] = list->elements[list->length - 1 - i];
    }

    return result;
}

// Uvoľníme pamäť zoznamu
void delete_list(list_t *list) {
    free(list);
}

}