#ifndef MEMORY_C
#define MEMORY_C


#include <stddef.h>

#include "terminal.c"
#include "util/printer.c"

#define MEMORY_START 0x200000
//#define MEMORY_END 0x300000

#define PAGE_ALIGNMENT 4

//size_t total_memory = MEMORY_END - MEMORY_START;

typedef struct page_tag {
    size_t size;
    size_t realsize;

    struct page_tag *prev;
    struct page_tag *next;
} page_tag;


page_tag *start = (void *) MEMORY_START;;

size_t calculate_realsize(size_t size) {
    if (size % PAGE_ALIGNMENT == 0) return size;
    return size + PAGE_ALIGNMENT - (size % PAGE_ALIGNMENT);
}

void *alloc(size_t size) {
    page_tag *curr_page = start;

    while (curr_page->next != NULL) {
        page_tag *new_page = ((void *) curr_page) + sizeof(page_tag) + curr_page->realsize;
        size_t realsize = calculate_realsize(size);
        if ((void *) new_page + sizeof(page_tag) + realsize <= (void *) curr_page->next) {

            new_page->size = size;
            new_page->realsize = realsize;

            curr_page->next->prev = new_page;
            new_page->next = curr_page->next;

            curr_page->next = new_page;
            new_page->prev = curr_page;

            return new_page + 1;
        }
        curr_page = curr_page->next;
    }

    page_tag *new_page = ((void *) curr_page) + sizeof(page_tag) + curr_page->realsize;
    new_page->size = size;
    new_page->realsize = calculate_realsize(size);

    curr_page->next = new_page;
    new_page->prev = curr_page;
    new_page->next = NULL;
    return new_page + 1;
}

void free(void *data) {
    page_tag *data_tag = data - sizeof(page_tag);

    data_tag->prev->next = data_tag->next;
    data_tag->next->prev = data_tag->prev;
}

int amount_of_digits(int x) {
    if (x == 0) return 1;
    int n = 0;
    while (x != 0) {
        n += 1;
        x /= 10;
    }
    return n;
}

#define MEM_SIZE_WIDTH 3
#define SPACED_ARRAY {' ', ' ', ' '};

void print_memory() {
    print("____________________\n");
    print(" ===  MEM DUMP  === \n");
    print("--------------------\n");
    print("Page tag size:    %d bytes\n", sizeof(page_tag));
    print("Memory alignment:  %d bytes\n", PAGE_ALIGNMENT);
    print("=> idx: [page_tag_address] [data_address (size|aligment_padding_size)]\n\n");
    page_tag *curr_page = start;
    int i = 0;
    bool left = true;
    while (curr_page != NULL) {
        char padding[MEM_SIZE_WIDTH] = SPACED_ARRAY;
        padding[MEM_SIZE_WIDTH - amount_of_digits(curr_page->size)] = '\0';
        print("%d: [%x] [%x (%s%d|%d)]",
              i,
              curr_page,
              (curr_page + 1), padding, curr_page->size,
              curr_page->realsize - curr_page->size);

        if (left) {
            print(" | ");
        } else {
            print("\n");
        }
        left = !left;

        bool empty_print = false;
        void *empty_start = (void *) curr_page + sizeof(page_tag) + curr_page->realsize;
        if (empty_start + sizeof(page_tag) < (void *) curr_page->next) {
            size_t memory_left = (void *) curr_page->next -
                                 (empty_start + sizeof(page_tag));
            char empty_padding[MEM_SIZE_WIDTH] = SPACED_ARRAY;
            empty_padding[MEM_SIZE_WIDTH - amount_of_digits(memory_left)] = '\0';

            print("_: free space      (%d + %s%d| ) ",
                  sizeof(page_tag),
                  empty_padding,
                  memory_left
            );
            empty_print = true;
        } else if (empty_start < (void *) curr_page->next) {
            size_t memory_left = (void *) curr_page->next - empty_start;

            char empty_padding[MEM_SIZE_WIDTH] = SPACED_ARRAY;
            empty_padding[MEM_SIZE_WIDTH - amount_of_digits(memory_left)] = '\0';

            print("_: not enough room      (%s%d| ) ",
                  empty_padding,
                  memory_left);
            empty_print = true;
        }
        if (empty_print) {
            if (left) {
                print(" | ");
            } else {
                print("\n");
            }
            left = !left;
        }


        curr_page = curr_page->next;
        i += 1;
    }
    if (!left) {
        print("\n");
    }
    print("____________________\n");
}

int command_mem_dump(char *string) {
    print_memory();
}

//
//void test_allocs() {
//    void *ptr0 = __alloc(64);
//    void *ptr1 = __alloc(64);
//    void *ptr4 = __alloc(64);
//    print();
//
//    printf("Free nr 2 -> %p\n", ptr1);
//    __free(ptr1);
//    print();
//
//    printf("Alloc 32 bytes\n");
//    void *ptr2 = __alloc(32);
//    print();
//
//    printf("Alloc 32 bytes\n");
//    __alloc(32);
//    print();
//
//    printf("Alloc 8 bytes\n");
//    void *ptr3 = __alloc(8);
//    print();
//
//    __free(ptr2);
//    print();
//    __free(ptr4);
//    print();
//    __free(ptr3);
//    print();
//
//    __free(ptr0);
//    print();
//
//    printf("#########\nTry almost fill\n");
//    __alloc(160);
//    print();
//    __alloc(50);
//    print();
//
//
//}
//
//int main() {
//    start = malloc(sizeof(size_t) * 1000);
//    start->next = NULL;
//    start->prev = NULL;
//    test_allocs();
//}

#endif // MEMORY_C