#define _CRT_SECURE_NO_WARNINGS

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

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

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

    // Check if entry exists already.
    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_to_replace) == 0) {
            entry_exists = true;
            break;
        }
        zip_entry_close(zip_read);
    } 
    zip_close(zip_read);

    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); // Need to close to flush removed entry.

        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 = false;
    zip_entry_close(zip_write);
    
    zip_close(zip_write);

    return result;
}

bool administration_writer_save_all_cost_centers()
{
    //// Get all data.
    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);
    
    int buf_length = num_costcenters * 500;  // Ballpark file content size.
    char* settings_content = (char*)malloc(buf_length);
    memset(settings_content, 0, buf_length);

    char* orig_content = settings_content;
    
    //// Cost centers.
    settings_content += sprintf(settings_content, "<cost_centers>\n");

    for (u32 i = 0; i < num_costcenters; i++) {
        cost_center c = costcenter_buffer[i];

        settings_content += sprintf(settings_content, "\t<cost_center>\n");

        settings_content += sprintf(settings_content, "\t\t<id>%s</id>\n", c.id);
        settings_content += sprintf(settings_content, "\t\t<code>%s</code>\n", c.code);
        settings_content += sprintf(settings_content, "\t\t<description>%s</description>\n", c.description);

        settings_content += sprintf(settings_content, "\t</cost_center>\n");
    }

    settings_content += sprintf(settings_content, "</cost_centers>");

    //// Write to Disk.
    int final_length = (int)strlen(orig_content);
    bool result = true;

    if (!xml_parse_document((uint8_t*)orig_content, final_length)) {
        result = false;
    }
    else {
        administration_writer_write_to_zip(ADMIN_FILE_COSTCENTERS, orig_content, final_length);
    }

    free(costcenter_buffer);
    free(orig_content);

    return result;
}

bool administration_writer_save_all_tax_brackets()
{
    //// 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);
    
    int buf_length = num_brackets * 500;  // Ballpark file content size.
    char* settings_content = (char*)malloc(buf_length);
    memset(settings_content, 0, buf_length);

    char* orig_content = settings_content;

    //// Tax brackets.
    settings_content += sprintf(settings_content, "<country_tax_brackets>\n");

    for (u32 i = 0; i < num_brackets; i++) {
        country_tax_bracket c = bracket_buffer[i];

        settings_content += sprintf(settings_content, "\t<country_tax_bracket>\n");

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

        settings_content += sprintf(settings_content, "\t</country_tax_bracket>\n");
    }

    settings_content += sprintf(settings_content, "</country_tax_brackets>");

    //// Write to Disk.
    int final_length = (int)strlen(orig_content);
    bool result = true;

    if (!xml_parse_document((uint8_t*)orig_content, final_length)) {
        result = false;
    }
    else {
        administration_writer_write_to_zip(ADMIN_FILE_BRACKETS, orig_content, final_length);
    }

    free(bracket_buffer);
    free(orig_content);

    return result;
}

bool administration_writer_save_contact(contact c, char* content_buffer = NULL)
{
    bool result = true;
    bool do_free = false;
    if (!content_buffer) {
        int buf_length = 2000;  // Ballpark contact content size.
        content_buffer = (char*)malloc(buf_length);
        memset(content_buffer, 0, buf_length);
        do_free = true;
    }

    char* orig_content = content_buffer;

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

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

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

    content_buffer += sprintf(content_buffer, "</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 = false;
    }
    else {
        administration_writer_write_to_zip(final_path, orig_content, final_length);
    }

    if (do_free) free(content_buffer);
    return result;
}


bool administration_writer_save_all_contacts()
{
    //// 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);


    int buf_length = 2000;  // Ballpark contact content size.
    char* content_buffer = (char*)malloc(buf_length);
    memset(content_buffer, 0, buf_length);
    char* orig_content = content_buffer;
    bool result = true;

    if (!administration_writer_save_contact(administration_company_info_get(), content_buffer)) result = false;

    // Contacts.
    for (u32 i = 0; i < num_contacts; i++) {
        contact c = contact_buffer[i];

        if (!administration_writer_save_contact(c, content_buffer)) result = false;

        content_buffer = orig_content;
        memset(content_buffer, 0, buf_length);
    }

    free(contact_buffer);
    free(orig_content);

    return result;
}

static int administration_writer_write_all_t(void *arg) {
    int result = 1;
    if (!administration_writer_save_all_tax_brackets()) result = 0;
    if (!administration_writer_save_all_cost_centers()) result = 0;
    if (!administration_writer_save_all_contacts()) result = 0;

    ui_set_status_loading(false);
    return result; // thread exit code
}

bool administration_writer_save_all()
{
    ui_set_status_loading(true);

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