#define _CRT_SECURE_NO_WARNINGS

#include <zip.h>
#include <xml.h>
#include <stdlib.h>
#include <threads.h>

#include "ui.hpp"
#include "administration_writer.hpp"

mtx_t _save_file_mutex;

bool administration_writer_create()
{
    return mtx_init(&_save_file_mutex, mtx_plain) == thrd_success;
}

void administration_writer_destroy()
{
    mtx_destroy(&_save_file_mutex);
}

static bool administration_writer_entry_exists(char* entry)
{
    bool entry_exists = false;
    struct zip_t *zip_read = zip_open(administration_file_path_get(), 0, 'r');  
    int i, n = (int)zip_entries_total(zip_read);
    for (i = 0; i < n; ++i) {
        zip_entry_openbyindex(zip_read, i);
        const char *entry_name = zip_entry_name(zip_read);
        if (strcmp(entry_name, entry) == 0) {
            entry_exists = true;
            break;
        }
        zip_entry_close(zip_read);
    } 
    zip_close(zip_read);
    return entry_exists;
}

bool administration_writer_delete_entry(char* id)
{
    bool result = 1;
    struct zip_t *zip_write = zip_open(administration_file_path_get(), 0, 'a');
    if (!zip_write) zip_write = zip_open(administration_file_path_get(), 0, 'w');

    char final_path[50];
    snprintf(final_path, 50, "%s.xml", id);

    char* indices[1] = {final_path};
    if (zip_entries_delete(zip_write, indices, 1) < 0) result = 0;
    zip_close(zip_write);
    return result;
}

static bool administration_writer_write_to_zip(char* entry_to_replace, char* orig_content, int final_length)
{
    bool result = 1;
    bool entry_exists = administration_writer_entry_exists(entry_to_replace);

    struct zip_t *zip_write = zip_open(administration_file_path_get(), 0, 'a');
    if (!zip_write) zip_write = zip_open(administration_file_path_get(), 0, 'w');

    if (entry_exists)
    {
        char* indices[1] = {entry_to_replace};
        zip_entries_delete(zip_write, indices, 1);
        zip_close(zip_write);

        zip_write = zip_open(administration_file_path_get(), 0, 'a');
    }

    zip_entry_open(zip_write, entry_to_replace);
    if (zip_entry_write(zip_write, orig_content, final_length) < 0) result = 0;
    zip_entry_close(zip_write);
    
    zip_close(zip_write);

    return result;
}

/////////////////////////////
//// Projects
/////////////////////////////
bool administration_writer_save_project_blocking(project project)
{
    bool result = 1;
    int buf_length = 1000;  // Ballpark file content size.
    char* file_content = (char*)malloc(buf_length);
    memset(file_content, 0, buf_length);

    char* orig_content = file_content;
    
    file_content += sprintf(file_content, "<project>\n");
    file_content += sprintf(file_content, "\t<id>%s</id>\n", project.id);
    file_content += sprintf(file_content, "\t<description>%s</description>\n", project.description);
    file_content += sprintf(file_content, "\t<state>%d</state>\n", project.state);
    file_content += sprintf(file_content, "\t<start_date>%lld</start_date>\n", (long long)project.start_date);
    file_content += sprintf(file_content, "\t<end_date>%lld</end_date>\n", (long long)project.end_date);
    file_content += sprintf(file_content, "</project>\n");

    //// Write to Disk.
    char final_path[50];
    snprintf(final_path, 50, "%s.xml", project.id);
    
    int final_length = (int)strlen(orig_content);
    if (!xml_parse_document((uint8_t*)orig_content, final_length)) result = 0;
    else if (!administration_writer_write_to_zip(final_path, orig_content, final_length)) result = 0;

    free(orig_content);
    return result;
}

static bool administration_writer_save_all_projects_blocking()
{
    bool result = 1;
    u32 num_projects = administration_project_count();
    u32 buffer_size = sizeof(project) * num_projects;
    project* project_buffer = (project*)malloc(buffer_size);
    num_projects = administration_project_get_all(project_buffer);
    
    for (u32 i = 0; i < num_projects; i++) {
        project c = project_buffer[i];
        if (!administration_writer_save_project_blocking(c)) result = 0;
    }
  
    free(project_buffer);
    return result;
}

/////////////////////////////
//// Cost centers
/////////////////////////////
bool administration_writer_save_cost_center_blocking(cost_center cost)
{
    bool result = 1;
    int buf_length = 1000;  // Ballpark file content size.
    char* file_content = (char*)malloc(buf_length);
    memset(file_content, 0, buf_length);

    char* orig_content = file_content;
    
    file_content += sprintf(file_content, "<cost_center>\n");
    file_content += sprintf(file_content, "\t<id>%s</id>\n", cost.id);
    file_content += sprintf(file_content, "\t<code>%s</code>\n", cost.code);
    file_content += sprintf(file_content, "\t<description>%s</description>\n", cost.description);
    file_content += sprintf(file_content, "</cost_center>\n");

    //// Write to Disk.
    char final_path[50];
    snprintf(final_path, 50, "%s.xml", cost.id);
    
    int final_length = (int)strlen(orig_content);
    if (!xml_parse_document((uint8_t*)orig_content, final_length)) result = 0;
    else if (!administration_writer_write_to_zip(final_path, orig_content, final_length)) result = 0;

    free(orig_content);
    return result;
}

static bool administration_writer_save_all_cost_centers_blocking()
{
    bool result = 1;
    u32 num_costcenters = administration_cost_center_count();
    u32 buffer_size = sizeof(cost_center) * num_costcenters;
    cost_center* costcenter_buffer = (cost_center*)malloc(buffer_size);
    num_costcenters = administration_cost_center_get_all(costcenter_buffer);
    
    for (u32 i = 0; i < num_costcenters; i++) {
        cost_center c = costcenter_buffer[i];
        if (!administration_writer_save_cost_center_blocking(c)) result = 0;
    }
  
    free(costcenter_buffer);
    return result;
}

/////////////////////////////
//// Tax brackets
/////////////////////////////
bool administration_writer_save_tax_bracket_blocking(country_tax_bracket bracket)
{
    bool result = 1;
    int buf_length = 1000; // Ballpark file content size.
    char* file_content = (char*)malloc(buf_length);
    memset(file_content, 0, buf_length);
    char* orig_content = file_content;

    file_content += sprintf(file_content, "<country_tax_bracket>\n");
    file_content += sprintf(file_content, "\t<id>%s</id>\n", bracket.id);
    file_content += sprintf(file_content, "\t<country_code>%s</country_code>\n", bracket.country_code);
    file_content += sprintf(file_content, "\t<rate>%.2f</rate>\n", bracket.rate);
    file_content += sprintf(file_content, "\t<description>%s</description>\n", bracket.description);
    file_content += sprintf(file_content, "</country_tax_bracket>\n");

    //// Write to Disk.
    char final_path[50];
    snprintf(final_path, 50, "%s.xml", bracket.id);

    int final_length = (int)strlen(orig_content);
    if (!xml_parse_document((uint8_t*)orig_content, final_length)) result = 0;
    else if (!administration_writer_write_to_zip(final_path, orig_content, final_length)) result = 0;

    free(orig_content);
    return result;
}

static bool administration_writer_save_all_tax_brackets_blocking()
{
    //// Get all data.
    u32 num_brackets = administration_tax_bracket_count();
    u32 buffer_size = sizeof(country_tax_bracket) * num_brackets;
    country_tax_bracket* bracket_buffer = (country_tax_bracket*)malloc(buffer_size);
    num_brackets = administration_tax_bracket_get_all(bracket_buffer);
    
    bool result = 1;

    for (u32 i = 0; i < num_brackets; i++) {
        country_tax_bracket c = bracket_buffer[i];
        if (!administration_writer_save_tax_bracket_blocking(c)) result = 0;
    }

    free(bracket_buffer);
    return result;
}

/////////////////////////////
//// Contacts
/////////////////////////////
bool administration_writer_save_contact_blocking(contact c)
{
    bool result = 1;
    int buf_length = 2000; // Ballpark contact content size.
    char* file_content = (char*)malloc(buf_length);
    memset(file_content, 0, buf_length);

    char* orig_content = file_content;

    file_content += sprintf(file_content, "<contact>\n");

    file_content += sprintf(file_content, "\t<id>%s</id>\n", c.id);
    file_content += sprintf(file_content, "\t<name>%s</name>\n", c.name);
    file_content += sprintf(file_content, "\t<type>%d</type>\n", c.type);
    file_content += sprintf(file_content, "\t<taxid>%s</taxid>\n", c.taxid);
    file_content += sprintf(file_content, "\t<businessid>%s</businessid>\n", c.businessid);
    file_content += sprintf(file_content, "\t<email>%s</email>\n", c.email);
    file_content += sprintf(file_content, "\t<phone_number>%s</phone_number>\n", c.phone_number);
    file_content += sprintf(file_content, "\t<bank_account>%s</bank_account>\n", c.bank_account);

    file_content += sprintf(file_content, "\t<address>\n");
    file_content += sprintf(file_content, "\t\t<address1>%s</address1>\n", c.address.address1);
    file_content += sprintf(file_content, "\t\t<address2>%s</address2>\n", c.address.address2);
    file_content += sprintf(file_content, "\t\t<country_code>%s</country_code>\n", c.address.country_code);
    file_content += sprintf(file_content, "\t</address>\n");

    file_content += sprintf(file_content, "</contact>\n");

    char final_path[50];
    snprintf(final_path, 50, "%s.xml", c.id);

    int final_length = (int)strlen(orig_content);
    if (!xml_parse_document((uint8_t*)orig_content, final_length)) result = 0;
    else if (!administration_writer_write_to_zip(final_path, orig_content, final_length)) result = 0;

    free(orig_content);
    return result;
}

static bool administration_writer_save_all_contacts_blocking()
{
    //// Get all data.
    u32 num_contacts = administration_contact_count();
    u32 buffer_size = sizeof(contact) * num_contacts;
    contact* contact_buffer = (contact*)malloc(buffer_size);
    num_contacts = administration_contact_get_all(contact_buffer);

    bool result = 1;

    // Save company info.
    if (!administration_writer_save_contact_blocking(administration_company_info_get())) result = 0;

    // Save contacts.
    for (u32 i = 0; i < num_contacts; i++) {
        contact c = ((contact*)contact_buffer)[i];
        if (!administration_writer_save_contact_blocking(c)) result = 0;
    }

    free(contact_buffer);
    return result;
}

/////////////////////////////
//// Administration info
/////////////////////////////
bool administration_writer_save_all_administration_info_blocking()
{
    bool result = 1;
    int buf_length = 1000;  // Ballpark file content size.
    char* file_content = (char*)malloc(buf_length);
    memset(file_content, 0, buf_length);
    char* orig_content = file_content;
    
    //// Cost centers.
    file_content += sprintf(file_content, "<administration>\n");
    file_content += sprintf(file_content, "\t<next_id>%d</next_id>\n", administation_get()->next_id);
    file_content += sprintf(file_content, "\t<next_sequence_number>%d</next_sequence_number>\n", administation_get()->next_sequence_number);
    file_content += sprintf(file_content, "\t<program_version>%s</program_version>\n", administation_get()->program_version);
    file_content += sprintf(file_content, "</administration>");

    //// Write to Disk.
    int final_length = (int)strlen(orig_content);
    if (!xml_parse_document((uint8_t*)orig_content, final_length)) result = 0;
    else if (!administration_writer_write_to_zip(ADMIN_FILE_INFO, orig_content, final_length)) result = 0;

    free(orig_content);

    return result;
}

/////////////////////////////
//// Other
/////////////////////////////
static int administration_writer_write_all_t(void *arg) {
    (void)arg;
    int result = 1;

    mtx_lock(&_save_file_mutex);

    if (!administration_writer_save_all_projects_blocking()) result = 0;
    if (!administration_writer_save_all_administration_info_blocking()) result = 0;
    if (!administration_writer_save_all_tax_brackets_blocking()) result = 0;
    if (!administration_writer_save_all_cost_centers_blocking()) result = 0;
    if (!administration_writer_save_all_contacts_blocking()) result = 0;

    mtx_unlock(&_save_file_mutex);

    ui_set_status_loading(false);
    return result;
}

bool administration_writer_save_all_async()
{
    ui_set_status_loading(true);

    thrd_t thr;
    if (thrd_create(&thr, administration_writer_write_all_t, 0) != thrd_success) {
        return false;
    }
    
    return true;
}