//
// Created by MightyPork on 2017/11/26.
//

#include <utils/avrlibc.h>
#include "platform.h"
#include "utils/hexdump.h"
#include "settings.h"
#include "unit_registry.h"
#include "system_settings.h"
#include "utils/str_utils.h"

// pre-declarations
static void savebuf_flush(PayloadBuilder *pb, bool final);
static bool savebuf_ovhandler(PayloadBuilder *pb, uint32_t more);

// This is the first entry in a valid config.
// Change with each breaking change to force config reset.
#define CONFIG_MARKER 0xA55E

void settings_load(void)
{
    uint8_t *buffer = (uint8_t *) SETTINGS_FLASH_ADDR;

#if DEBUG_FLASH_WRITE
    dbg("reading @ %p", (void*)SETTINGS_FLASH_ADDR);
    hexDump("Flash", buffer, 64);
#endif

    PayloadParser pp = pp_start(buffer, SETTINGS_BLOCK_SIZE, NULL);

    // Check the integrity marker
    if (pp_u16(&pp) != CONFIG_MARKER) {
        dbg("Config not valid!");
        // Save for next run
        settings_save();
        return;
    }

    uint8_t version = pp_u8(&pp); // top level settings format version

    { // Settings
        (void)version; // Conditional choices based on version

        // System section
        if (!systemsettings_load(&pp)) {
            dbg("!! System settings failed to load");
            return;
        }

        if (!ureg_load_units(&pp)) {
            dbg("!! Unit settings failed to load");
            return;
        }
    } // End settings

    dbg("System settings loaded OK");
}


static uint8_t save_buffer[FLASH_SAVE_BUF_LEN];
static uint32_t save_addr;

#if DEBUG_FLASH_WRITE
#define fls_printf(fmt, ...) PRINTF(fmt, ##__VA_ARGS__)
#else
#define fls_printf(fmt, ...) do {} while (0)
#endif

/**
 * Save buffer overflow handler.
 * This should flush whatever is in the buffer and let CWPack continue
 *
 * @param pb - buffer
 * @param more - how many more bytes are needed (this is meant for realloc / buffer expanding)
 * @return - success code
 */
static bool savebuf_ovhandler(PayloadBuilder *pb, uint32_t more)
{
    if (more > FLASH_SAVE_BUF_LEN) return false;
    savebuf_flush(pb, false);
    return true;
}

/** Save settings to flash */
void settings_save(void)
{
    HAL_StatusTypeDef hst;
    PayloadBuilder pb = pb_start(save_buffer, FLASH_SAVE_BUF_LEN, savebuf_ovhandler);

    save_addr = SETTINGS_FLASH_ADDR;

    fls_printf("--- Starting flash write... ---\r\n");
    hst = HAL_FLASH_Unlock();
    assert_param(hst == HAL_OK);
    {
        fls_printf("ERASE flash pages for settings storage...\r\n");
        // We have to first erase the pages
        FLASH_EraseInitTypeDef erase;

#if PLAT_FLASHBANKS
        erase.Banks = FLASH_BANK_1; // TODO ?????
#endif

#if defined(GEX_PLAT_F407_DISCOVERY)
        // specialty for F4 with too much flash
        erase.NbSectors = 1;
        erase.Sector = SETTINGS_FLASH_SECTOR;
        erase.TypeErase = FLASH_TYPEERASE_SECTORS;
        erase.VoltageRange = FLASH_VOLTAGE_RANGE_3;
        erase.Banks = FLASH_BANK_1; // unused for sector erase
#else
        erase.NbPages = SETTINGS_BLOCK_SIZE/FLASH_PAGE_SIZE;
        erase.PageAddress = SETTINGS_FLASH_ADDR;
        erase.TypeErase = FLASH_TYPEERASE_PAGES;
#endif
        uint32_t pgerror = 0;
        hst = HAL_FLASHEx_Erase(&erase, &pgerror);
        assert_param(pgerror == 0xFFFFFFFFU);
        assert_param(hst == HAL_OK);

        // and now we can start writing...

        fls_printf("Beginning settings collect\r\n");

        // Marker that this is a valid save
        pb_u16(&pb, CONFIG_MARKER);
        pb_u8(&pb, 0); // Settings format version

        { // Settings
            fls_printf("Saving system settings\r\n");
            systemsettings_save(&pb);

            fls_printf("Saving units\r\n");
            ureg_save_units(&pb);
        } // End settings

        fls_printf("Final flush\r\n");
        savebuf_flush(&pb, true);
    }

    fls_printf("Locking flash...\r\n");
    hst = HAL_FLASH_Lock();
    assert_param(hst == HAL_OK);
    fls_printf("--- Flash done ---\r\n");

#if DEBUG_FLASH_WRITE
    dbg("written @ %p", (void*)SETTINGS_FLASH_ADDR);
    hexDump("Flash", (void*)SETTINGS_FLASH_ADDR, 64);
#endif
}

/**
 * Flush the save buffer to flash, moving leftovers from uneven half-words
 * to the beginning and adjusting the CWPack curent pointer accordingly.
 *
 * @param ctx - pack context
 * @param final - if true, flush uneven leftovers; else move them to the beginning and keep for next call.
 */
static void savebuf_flush(PayloadBuilder *pb, bool final)
{
    // TODO this might be buggy, was not tested cross-boundary yet
    // TODO remove those printf's after verifying correctness

    uint32_t bytes = (uint32_t) pb_length(pb);

    // Dump what we're flushing
    fls_printf("Flush: ");
    for (uint32_t i = 0; i < bytes; i++) {
        fls_printf("%02X ", save_buffer[i]);
    }
    fls_printf("\r\n");

    uint32_t halfwords = bytes >> 1;
    uint32_t remain = bytes & 1; // how many bytes won't be programmed

    fls_printf("Halfwords: %d, Remain: %d, last? %d\r\n", (int)halfwords, (int)remain, final);

    uint16_t *hwbuf = (void*) &save_buffer[0];

    for (; halfwords > 0; halfwords--) {
        uint16_t hword = *hwbuf++;

        fls_printf("%04X ", hword);

        HAL_StatusTypeDef res = HAL_FLASH_Program(FLASH_TYPEPROGRAM_HALFWORD, save_addr, hword);
        assert_param(HAL_OK == res);
        save_addr += 2; // advance
    }

    // rewind the context buffer
    pb->current = pb->start;

    if (remain) {
        // We have an odd byte left to write
        if (final) {
            // We're done writing, this is the last call. Append the last byte to flash.
            uint16_t hword = save_buffer[bytes-1];
            fls_printf("& %02X ", hword);

            HAL_StatusTypeDef res = HAL_FLASH_Program(FLASH_TYPEPROGRAM_HALFWORD, save_addr, hword);
            assert_param(HAL_OK == res);
        } else {
            // Move the leftover to the beginning of the buffer for next call.
            save_buffer[0] = save_buffer[bytes-1];
            pb->current++;
        }
    }
    fls_printf("\r\n");
}

// ---------------------------------------------------------------

static void ini_preamble(IniWriter *iw, const char *filename)
{
    // File header
    iw_hdr_comment(iw, filename);
    iw_hdr_comment(iw, "GEX v%s on %s", GEX_VERSION, GEX_PLATFORM);
    iw_hdr_comment(iw, "built %s at %s", __DATE__, __TIME__);
    iw_cmt_newline(iw);

    iw_comment(iw, "Overwrite this file to change settings.");
    iw_comment(iw, "Close the LOCK jumper to save them to Flash.");
}

/**
 * Write system settings to INI (without section)
 */
void settings_build_units_ini(IniWriter *iw)
{
    ini_preamble(iw, "UNITS.INI");

    ureg_build_ini(iw);
}

/**
 * Write system settings to INI (without section)
 */
void settings_build_system_ini(IniWriter *iw)
{
    ini_preamble(iw, "SYSTEM.INI");

    systemsettings_build_ini(iw);
}

uint32_t settings_get_units_ini_len(void)
{
    // this writer is configured to skip everything, so each written byte will decrement the skip count
    IniWriter iw = iw_init(NULL, 0xFFFFFFFF, 1); // count is never used, we use 1 because 0 means we're full
    settings_build_units_ini(&iw);
    // now we just check how many bytes were skipped
    return 0xFFFFFFFF - iw.skip;
}

uint32_t settings_get_system_ini_len(void)
{
    // same as above
    IniWriter iw = iw_init(NULL, 0xFFFFFFFF, 1);
    settings_build_system_ini(&iw);
    return 0xFFFFFFFF - iw.skip;
}

// ---------------------------------------------------------------

void settings_load_ini_begin(void)
{
    SystemSettings.modified = true;
    SystemSettings.pristine = true;
}

void settings_load_ini_key(const char *restrict section, const char *restrict key, const char *restrict value)
{
    dbg("[%s] %s = %s", section, key, value);
    static char namebuf[INI_KEY_MAX];

    // SYSTEM and UNITS files must be separate.
    // Init functions are run for first key in the section.

    if (streq(section, "SYSTEM")) {

        if (SystemSettings.pristine) {
            SystemSettings.pristine = false;
            systemsettings_loadDefaults();
        }

        // system is always at the top
        systemsettings_load_ini(key, value);
    }
    else if (streq(section, "UNITS")) {

        if (SystemSettings.pristine) {
            SystemSettings.pristine = false;
            ureg_remove_all_units();
        }

        // this will always come before individual units config
        // install or tear down units as described by the config
        ureg_instantiate_by_ini(key, value);
    }
    else {
        // not a standard section, may be some unit config
        // all unit sections contain the colon character [TYPE:NAME@CALLSIGN]
        const char *nameptr = strchr(section, ':');
        const char *csptr = strchr(section, '@');

        if (nameptr != NULL && csptr != NULL) {
            strncpy(namebuf, nameptr+1, csptr - nameptr - 1);
            namebuf[csptr - nameptr - 1] = 0;
            uint8_t cs = (uint8_t) avr_atoi(csptr + 1);

            error_t rv = ureg_load_unit_ini_key(namebuf, key, value, cs);
            if (rv != E_SUCCESS)
                dbg("!! error loading %s@%d.%s = %s - error %s", namebuf, (int)cs, key, value, error_get_message(rv));
        } else {
            dbg("! Bad config key: [%s] %s = %s", section, key, value);
        }
    }
}

void settings_load_ini_end(void)
{
    if (!ureg_finalize_all_init()) {
        dbg("Some units failed to init!!");
    }
}