#include "CSV_Utilities.hpp"

extern "C"
{
#include "stdlib.h"
#include "string.h"
#include "stdio.h"
}

CSV_Table::CSV_Table()
{
    this->column_count = 0;
    this->mem_use = 0;
    this->Column_List_End = this->Column_List_Head = nullptr;
}

CSV_Table::~CSV_Table()
{
    this->reset();
}

void CSV_Table::reset()
{
    // 清理尾指针
    this->Column_List_End = nullptr;
    this->column_count = 0;
    this->mem_use = 0;

    // 清理头指针及其内容
    if (this->Column_List_Head != nullptr)
    {
        // 释放列
        while (this->Column_List_Head != nullptr)
        {
            decltype(this->Column_List_Head) p_column = this->Column_List_Head;

            // 释放单元格列表
            while (p_column->Cell_List_Head != nullptr)
            {
                decltype(p_column->Cell_List_Head) p_cell = p_column->Cell_List_Head;

                // 释放单元格内容
                if (p_cell->content_str != nullptr)
                    free(p_cell->content_str);

                p_column->Cell_List_Head = p_cell->p_next_cell;
                free(p_cell->content_str);
            }

            // 释放标题
            if (p_column->column_title_str != nullptr)
                free(p_column->column_title_str);

            this->Column_List_Head = p_column->p_next_column;
            free(p_column);
        }
    }
}

CSV_Type::CSV_Err CSV_Table::append_column(const char *title_name_str)
{

    // 开辟新节点
    decltype(this->Column_List_Head) p_new_column = (decltype(this->Column_List_Head))malloc(sizeof(*(this->Column_List_Head)));
    if (p_new_column == nullptr)
        return CSV_Type::CSV_ERROR_MEM_ALLOCATE_FAILED;
    this->mem_use += sizeof(*p_new_column);
    // 节点初始化
    p_new_column->Cell_List_Head = p_new_column->Cell_List_End = nullptr;
    p_new_column->p_next_column = nullptr;
    p_new_column->element_count = 0;

    // 存储列标题名
    if (title_name_str != nullptr)
    {
        char *p_title = (char *)malloc(strlen(title_name_str) + 1); // +1 for '\0'
        if (p_title == nullptr)
        {
            free(p_new_column);
            return CSV_Type::CSV_ERROR_MEM_ALLOCATE_FAILED;
        }

        this->mem_use += strlen(title_name_str) + 1;
        strcpy(p_title, title_name_str);
        p_new_column->column_title_str = p_title;
    }
    else
        p_new_column->column_title_str = nullptr;

    // 插入列链表
    if (this->column_count == 0)
        this->Column_List_End = this->Column_List_Head = p_new_column;
    else
    {
        this->Column_List_End->p_next_column = p_new_column;
        this->Column_List_End = p_new_column;
    }

    this->column_count++;
    return CSV_Type::CSV_ERROR_NONE;
}

CSV_Type::CSV_Err CSV_Table::insert_column_after(const char *target_column_title_str, const char *title_name_str)
{
    size_t count = 0;
    decltype(this->Column_List_Head) p_column = this->Column_List_Head;
    while (p_column != nullptr)
    {
        if (!strcmp(p_column->column_title_str, target_column_title_str))
            return this->insert_column_after(count, title_name_str);

        p_column = p_column->p_next_column;
        count++;
    }
    return CSV_Type::CSV_ERROR_COLUMN_NOT_FOUND;
}

CSV_Type::CSV_Err CSV_Table::insert_column_after(size_t target_column_position, const char *title_name_str)
{
    if ((target_column_position + 1) > this->column_count)
        return CSV_Type::CSV_ERROR_COLUMN_NOT_FOUND;

    decltype(this->Column_List_Head) p_column = this->Column_List_Head;
    for (size_t count = 0; count < target_column_position; count++)
        p_column = p_column->p_next_column;

    // 开辟新节点
    decltype(p_column) p_new_column = (decltype(p_column))malloc(sizeof(*(p_column)));
    if (p_new_column == nullptr)
        return CSV_Type::CSV_ERROR_MEM_ALLOCATE_FAILED;
    this->mem_use += sizeof(*p_new_column);

    // 节点初始化
    p_new_column->Cell_List_Head = p_new_column->Cell_List_End = nullptr;
    p_new_column->p_next_column = nullptr;
    p_new_column->element_count = 0;

    // 存储列标题名
    if (title_name_str != nullptr)
    {
        char *p_title = (char *)malloc(strlen(title_name_str) + 1); // +1 for '\0'
        if (p_title == nullptr)
        {
            free(p_new_column);
            return CSV_Type::CSV_ERROR_MEM_ALLOCATE_FAILED;
        }

        this->mem_use += strlen(title_name_str) + 1;
        strcpy(p_title, title_name_str);
        p_new_column->column_title_str = p_title;
    }
    else
        p_new_column->column_title_str = nullptr;

    // 链接节点
    p_new_column->p_next_column = p_column->p_next_column;
    p_column->p_next_column = p_new_column;
    this->column_count++;

    // 目标位置在末尾时
    if (target_column_position == this->column_count - 1)
        this->Column_List_End = p_new_column;

    return CSV_Type::CSV_ERROR_NONE;
}

CSV_Type::CSV_Err CSV_Table::append_cell(const char *column_title_str, const char *cell_content_str)
{
    size_t count = 0;
    decltype(this->Column_List_Head) p_column = this->Column_List_Head;
    while (p_column != nullptr)
    {
        if (!strcmp(p_column->column_title_str, column_title_str))
            return this->append_cell(count, cell_content_str);

        p_column = p_column->p_next_column;
        count++;
    }
    return CSV_Type::CSV_ERROR_COLUMN_NOT_FOUND;
}

CSV_Type::CSV_Err CSV_Table::append_cell(size_t column_position, const char *cell_content_str)
{
    if ((column_position + 1) > this->column_count)
        return CSV_Type::CSV_ERROR_COLUMN_NOT_FOUND;

    decltype(this->Column_List_Head) p_column = this->Column_List_Head;
    for (size_t count = 0; count < column_position; count++)
        p_column = p_column->p_next_column;

    decltype(p_column->Cell_List_Head) p_new_cell = (decltype(p_column->Cell_List_Head))malloc(sizeof(*(p_column->Cell_List_Head)));
    if (p_new_cell == nullptr)
        return CSV_Type::CSV_ERROR_MEM_ALLOCATE_FAILED;
    this->mem_use += sizeof(*p_new_cell);

    if (cell_content_str != nullptr)
    {
        char *p_content = (char *)malloc(strlen(cell_content_str) + 1); // +1 for '\0'
        if (p_content == nullptr)
        {
            free(p_new_cell);
            return CSV_Type::CSV_ERROR_MEM_ALLOCATE_FAILED;
        }

        this->mem_use += strlen(cell_content_str) + 1;
        strcpy(p_content, cell_content_str);
        p_new_cell->content_str = p_content;
    }
    else
        p_new_cell->content_str = nullptr;
    p_new_cell->p_next_cell = nullptr;

    if (p_column->element_count == 0)
        p_column->Cell_List_Head = p_column->Cell_List_End = p_new_cell;
    else
    {
        p_column->Cell_List_End->p_next_cell = p_new_cell;
        p_column->Cell_List_End = p_new_cell;
    }

    p_column->element_count++;
    return CSV_Type::CSV_ERROR_NONE;
}