/*
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * 
 * -----------------------------------------------------------------------------
 * File: DynamicArray.c
 * Description:
 *  This file contains the functions for modifying the Dynamic Array. It includes 
 *  functions to append, Insert before, Insert after, Insert at index, 
 *  Get, Remove element, Remove at index, Check if has element and Check if is empty.
 * 
 * Author: Mineplay
 * -----------------------------------------------------------------------------
 */
#include "../../Include/Fledasty/Core/DynamicArray.h"

#include <Hallocy/Core/Allocator.h>
#include <Hallocy/Core/Memory.h>
#include <Hallocy/Utils/Error.h>

FledastyError fledasty_dynamic_array_initialize(FledastyDynamicArray *new_dynamic_array, void *values, const size_t values_size, const size_t element_byte_size) {
    if (new_dynamic_array == NULL) {
        return FLEDASTY_ERROR_INVALID_POINTER;
    }

    new_dynamic_array->element_byte_size = element_byte_size;
    if (values == NULL || values_size == 0) {
        new_dynamic_array->size = 0;
        new_dynamic_array->capacity = 10;

        new_dynamic_array->buffer = (unsigned char*)hallocy_malloc(new_dynamic_array->capacity * element_byte_size);
        if (new_dynamic_array->buffer == NULL) {
            return FLEDASTY_ERROR_FAILED_ALLOCATION;
        }
    } else {
        new_dynamic_array->size = values_size;
        new_dynamic_array->capacity = new_dynamic_array->size + new_dynamic_array->size;

        new_dynamic_array->buffer = (unsigned char*)hallocy_malloc(new_dynamic_array->capacity * element_byte_size);
        if (new_dynamic_array->buffer == NULL) {
            return FLEDASTY_ERROR_FAILED_ALLOCATION;
        }

        hallocy_copy_memory(new_dynamic_array->buffer, values, values_size * element_byte_size);
    }

    return FLEDASTY_ERROR_NONE;
}

FledastyError fledasty_dynamic_array_destroy(FledastyDynamicArray *current_dynamic_array) {
    if (current_dynamic_array == NULL) {
        return FLEDASTY_ERROR_INVALID_POINTER;
    }

    if (hallocy_free(current_dynamic_array->buffer) != HALLOCY_ERROR_NONE) {
        return FLEDASTY_ERROR_FAILED_ALLOCATION;
    }

    current_dynamic_array->buffer = NULL;
    return FLEDASTY_ERROR_NONE;
}

FledastyError fledasty_dynamic_array_append(FledastyDynamicArray *current_dynamic_array, void *value) {
    if (current_dynamic_array == NULL || value == NULL) {
        return FLEDASTY_ERROR_INVALID_POINTER;
    }

    if (current_dynamic_array->size == current_dynamic_array->capacity) {
        current_dynamic_array->capacity += current_dynamic_array->capacity;
        current_dynamic_array->buffer = (unsigned char*)hallocy_realloc(current_dynamic_array->buffer, current_dynamic_array->capacity * current_dynamic_array->element_byte_size);
        
        if (current_dynamic_array->buffer == NULL) {
            return FLEDASTY_ERROR_FAILED_ALLOCATION;
        }
    }

    hallocy_copy_memory(current_dynamic_array->buffer + (current_dynamic_array->size * current_dynamic_array->element_byte_size), value, current_dynamic_array->element_byte_size);    
    current_dynamic_array->size += 1;

    return FLEDASTY_ERROR_NONE;
}

FledastyError fledasty_dynamic_array_insert_at_index(FledastyDynamicArray *current_dynamic_array, const size_t index, void *value) {
    if (current_dynamic_array == NULL || value == NULL) {
        return FLEDASTY_ERROR_FAILED_ALLOCATION;
    }

    if (index >= current_dynamic_array->size) {
        return FLEDASTY_ERROR_INDEX_OUT_OF_RANGE;
    }

    if (current_dynamic_array->size == current_dynamic_array->capacity) {
        current_dynamic_array->capacity += current_dynamic_array->capacity;
        current_dynamic_array->buffer = (unsigned char*)hallocy_realloc(current_dynamic_array->buffer, current_dynamic_array->capacity * current_dynamic_array->element_byte_size);
        
        if (current_dynamic_array->buffer == NULL) {
            return FLEDASTY_ERROR_FAILED_ALLOCATION;
        }
    }

    unsigned char *insert_pointer = current_dynamic_array->buffer + (index * current_dynamic_array->element_byte_size);
    hallocy_move_memory(insert_pointer + current_dynamic_array->element_byte_size, insert_pointer, (current_dynamic_array->size - index) * current_dynamic_array->element_byte_size);
    hallocy_copy_memory(insert_pointer, value, current_dynamic_array->element_byte_size);
    current_dynamic_array->size += 1;

    return FLEDASTY_ERROR_NONE;
}

FledastyError fledasty_dynamic_array_insert_before_value(FledastyDynamicArray *current_dynamic_array, void *before_value, void *value) {
    if (current_dynamic_array == NULL || before_value == NULL || value == NULL) {
        return FLEDASTY_ERROR_INVALID_POINTER;
    }

    size_t byte_index = 0;
    size_t byte_size = current_dynamic_array->size * current_dynamic_array->element_byte_size;
    while (byte_index < byte_size) {
        if (hallocy_compare_memory(current_dynamic_array->buffer + byte_index, before_value, current_dynamic_array->element_byte_size)) {
            if (current_dynamic_array->size == current_dynamic_array->capacity) {
                current_dynamic_array->capacity += current_dynamic_array->capacity;
                current_dynamic_array->buffer = (unsigned char*)hallocy_realloc(current_dynamic_array->buffer, current_dynamic_array->capacity * current_dynamic_array->element_byte_size);
        
                if (current_dynamic_array->buffer == NULL) {
                    return FLEDASTY_ERROR_FAILED_ALLOCATION;
                }
            }
        
            hallocy_move_memory(current_dynamic_array->buffer + (byte_index + current_dynamic_array->element_byte_size), current_dynamic_array->buffer + byte_index, byte_size - byte_index);
            hallocy_copy_memory(current_dynamic_array->buffer + byte_index, value, current_dynamic_array->element_byte_size);
            current_dynamic_array->size += 1;
        
            return FLEDASTY_ERROR_NONE;
        }

        byte_index += current_dynamic_array->element_byte_size;
    }

    return FLEDASTY_ERROR_VALUE_NOT_FOUND;
}

FledastyError fledasty_dynamic_array_insert_after_value(FledastyDynamicArray *current_dynamic_array, void *after_value, void *value) {
    if (current_dynamic_array == NULL || after_value == NULL || value == NULL) {
        return FLEDASTY_ERROR_INVALID_POINTER;
    }

    size_t byte_index = 0;
    size_t byte_size = current_dynamic_array->size * current_dynamic_array->element_byte_size;
    while (byte_index < byte_size) {
        if (hallocy_compare_memory(current_dynamic_array->buffer + byte_index, after_value, current_dynamic_array->element_byte_size)) {
            if (current_dynamic_array->size == current_dynamic_array->capacity) {
                current_dynamic_array->capacity += current_dynamic_array->capacity;
                current_dynamic_array->buffer = (unsigned char*)hallocy_realloc(current_dynamic_array->buffer, current_dynamic_array->capacity * current_dynamic_array->element_byte_size);
        
                if (current_dynamic_array->buffer == NULL) {
                    return FLEDASTY_ERROR_FAILED_ALLOCATION;
                }
            }
        
            byte_index += current_dynamic_array->element_byte_size;
            hallocy_move_memory(current_dynamic_array->buffer + (byte_index + current_dynamic_array->element_byte_size), current_dynamic_array->buffer + byte_index, byte_size - byte_index);
            hallocy_copy_memory(current_dynamic_array->buffer + byte_index, value, current_dynamic_array->element_byte_size);
            current_dynamic_array->size += 1;
        
            return FLEDASTY_ERROR_NONE;
        }

        byte_index += current_dynamic_array->element_byte_size;
    }

    return FLEDASTY_ERROR_VALUE_NOT_FOUND;
}

void *fledasty_dynamic_array_get(const FledastyDynamicArray *current_dynamic_array, const size_t index) {
    if (current_dynamic_array == NULL || index >= current_dynamic_array->size) {
        return NULL;
    }

    return current_dynamic_array->buffer + (index * current_dynamic_array->element_byte_size);
}

FledastyError fledasty_dynamic_array_remove_at_index(FledastyDynamicArray *current_dynamic_array, const size_t index) {
    if (current_dynamic_array == NULL) {
        return FLEDASTY_ERROR_INVALID_POINTER;
    }

    if (index >= current_dynamic_array->size) {
        return FLEDASTY_ERROR_INDEX_OUT_OF_RANGE;
    }

    unsigned char *index_pointer = current_dynamic_array->buffer + (index * current_dynamic_array->element_byte_size);
    hallocy_copy_memory(index_pointer, index_pointer + current_dynamic_array->element_byte_size, (current_dynamic_array->size + index) * current_dynamic_array->element_byte_size);
    current_dynamic_array->size -= 1;

    return FLEDASTY_ERROR_NONE;
}

FledastyError fledasty_dynamic_array_remove_value(FledastyDynamicArray *current_dynamic_array, void *value) {
    if (current_dynamic_array == NULL || value == NULL) {
        return FLEDASTY_ERROR_INVALID_POINTER;
    }

    size_t byte_index = 0;
    size_t byte_size = current_dynamic_array->size * current_dynamic_array->element_byte_size;
    while (byte_index < byte_size) {
        if (hallocy_compare_memory(current_dynamic_array->buffer + byte_index, value, current_dynamic_array->element_byte_size)) {
            hallocy_copy_memory(current_dynamic_array->buffer + byte_index, current_dynamic_array->buffer + (byte_index + current_dynamic_array->element_byte_size), byte_size - (byte_index + current_dynamic_array->element_byte_size));
            current_dynamic_array->size -= 1;

            return FLEDASTY_ERROR_NONE;
        }
        
        byte_index += current_dynamic_array->element_byte_size;
    }

    return FLEDASTY_ERROR_VALUE_NOT_FOUND;
}

FledastyError fledasty_dynamic_array_clear(FledastyDynamicArray *current_dynamic_array) {
    if (current_dynamic_array == NULL) {
        return FLEDASTY_ERROR_INVALID_POINTER;
    }

    current_dynamic_array->size = 0;
    return FLEDASTY_ERROR_NONE;
}

FledastyError fledasty_dynamic_array_shrink_to_fit(FledastyDynamicArray *current_dynamic_array) {
    if (current_dynamic_array == NULL) {
        return FLEDASTY_ERROR_INVALID_POINTER;
    }

    current_dynamic_array->capacity = (current_dynamic_array->size == 0) ? 10 : current_dynamic_array->size;
    unsigned char *shrinked_buffer = (unsigned char*)hallocy_malloc(current_dynamic_array->capacity * current_dynamic_array->element_byte_size);
    if (shrinked_buffer == NULL) {
        return FLEDASTY_ERROR_FAILED_ALLOCATION;
    }

    hallocy_copy_memory(shrinked_buffer, current_dynamic_array->buffer, current_dynamic_array->size * current_dynamic_array->element_byte_size);
    if (hallocy_free(current_dynamic_array->buffer) != HALLOCY_ERROR_NONE) {
        return FLEDASTY_ERROR_FAILED_ALLOCATION;
    }

    current_dynamic_array->buffer = shrinked_buffer;
    return FLEDASTY_ERROR_NONE;
}

bool fledasty_dynamic_array_has_value(const FledastyDynamicArray *current_dynamic_array, void *value) {
    if (current_dynamic_array == NULL) {
        return false;
    }

    for (size_t byte_index = 0; byte_index < current_dynamic_array->size * current_dynamic_array->element_byte_size; byte_index += current_dynamic_array->element_byte_size) {
        if (hallocy_compare_memory(current_dynamic_array->buffer + byte_index, value, current_dynamic_array->element_byte_size)) {
            return true;
        }
    }

    return false;
}