#include "../wiegand.h"

void wiegand_add_info_4bit_8bit(FuriString *buffer)
{
    if (bit_count == 8)
    {
        for (int i = 0; i < 4; i++)
        {
            furi_string_cat_printf(
                buffer, "\nbit %d: %d %d (bit %d)", i, data[i], data[i + 4], i + 4);
            if (data[i] == data[i + 4])
            {
                furi_string_cat_printf(buffer, " - ERROR");
            }
            else
            {
                furi_string_cat_printf(buffer, " - OK");
            }
        }
    }
    if (bit_count == 4 || bit_count == 8)
    {
        int code = 0;
        int offset = bit_count == 4 ? 0 : 4;
        for (int i = 0; i < 4; i++)
        {
            code = code << 1;
            code += data[i + offset] ? 1 : 0;
        }
        if (code <= 9)
        {
            furi_string_cat_printf(buffer, "\nButton: %d", code);
        }
        else if (code == 10)
        {
            furi_string_cat_printf(buffer, "\nButton: Escape");
        }
        else if (code == 11)
        {
            furi_string_cat_printf(buffer, "\nButton: Enter");
        }
    }
}

void wiegand_add_info_26bit(FuriString *buffer)
{
    // 26 bit wiegand, the first bit is the even parity bit, which is
    // based on the next 12 bits.  The number of bits that are a 1 should
    // be even.
    // After the parity bit, the next 8 bits are the facility code.
    // Then the next 16 bits are the card id .
    furi_string_cat_printf(buffer, "\nFacility: ");
    int code = 0;
    int count = 0;
    uint32_t dec = 0;
    for (int i = 1; i < 25; i++)
    {
        code = code << 1;
        dec = dec << 1;
        code |= data[i] ? 1 : 0;
        dec |= data[i] ? 1 : 0;
        if (++count % 4 == 0)
        {
            furi_string_cat_printf(buffer, "%X", code);
            code = 0;
        }

        if (i == 8)
        {
            furi_string_cat_printf(buffer, " (%ld)", dec);
            dec = 0;
        }
        // Parity, then 8 bit facility code, then id.
        if (i == 9)
        {
            furi_string_cat_printf(buffer, "\nCard: ");
        }
    }
    furi_string_cat_printf(buffer, " (%ld)", dec);

    int parity = 0;
    if (data[0])
    {
        parity = 1;
    }
    for (int i = 1; i < 13; i++)
    {
        if (data[i])
        {
            parity++;
        }
    }
    if (parity % 2 == 0)
    {
        furi_string_cat_printf(buffer, "\nEven Parity (%d): OK", parity);
    }
    else
    {
        furi_string_cat_printf(buffer, "\nEven Parity (%d): ERROR", parity);
    }

    if (data[13])
    {
        parity = 1;
    }
    else
    {
        parity = 0;
    }
    for (int i = 14; i < 26; i++)
    {
        if (data[i])
        {
            parity++;
        }
    }
    if (parity % 2 == 0)
    {
        furi_string_cat_printf(buffer, "\nOdd Parity (%d): ERROR", parity);
    }
    else
    {
        furi_string_cat_printf(buffer, "\nOdd Parity (%d): OK", parity);
    }
}

void wiegand_add_info_24bit(FuriString *buffer)
{
    // 24 bit wiegand (no parity info).

    // The First 8 bits are the facility code.
    // Then the next 16 bits are the card id.
    furi_string_cat_printf(buffer, "\nFacility: 0x");
    int code = 0;
    int count = 0;
    uint32_t dec = 0;
    for (int i = 0; i < 24; i++)
    {
        code = code << 1;
        dec = dec << 1;
        code |= data[i] ? 1 : 0;
        dec |= data[i] ? 1 : 0;
        if (++count % 4 == 0)
        {
            furi_string_cat_printf(buffer, "%X", code);
            code = 0;
        }
        // The first 8 bits are facility code, then comes id.
        if (i == 8)
        {
            furi_string_cat_printf(buffer, " (%ld)", dec);
            dec = 0;
            furi_string_cat_printf(buffer, "\nCard: 0x");
        }
    }
    furi_string_cat_printf(buffer, " (%ld)", dec);
}

void wiegand_add_info_48bit(FuriString *buffer)
{
    // We assume this is HID 48 bit Corporate 1000 - H2004064 format.
    // The first bit is odd parity 2 (based on bits 2-48).
    // The next bit is even parity (based on 4-5,7-8,10-11,...,46-47).
    // Then 22 bit company code.
    // Then 23 bit card id.
    /// Then odd parity 1 (based on 3-4,6-7,9-10,...,45-46).

    // 22 bits company code (bits 3-24; data[2..23])
    uint32_t code = 0;
    for (int i = 2; i <= 23; i++)
    {
        code = code << 1;
        code |= data[i] ? 1 : 0;
    }
    furi_string_cat_printf(buffer, "\nFacility: %lX (%ld)", code, code);

    // 23 bit card id (bits 25-47; data[24..46]).
    code = 0;
    for (int i = 24; i <= 46; i++)
    {
        code = code << 1;
        code |= data[i] ? 1 : 0;
    }
    furi_string_cat_printf(buffer, "\nCard: %lX (%ld)", code, code);

    // TODO: Add the 3 parity checks.
}

void wiegand_add_info_35bit(FuriString *buffer)
{
    // We assume this is HID 35 bit Corporate 1000 - C1k35s format.

    // 12 bits company code
    uint32_t code = 0;
    for (int i = 2; i <= 13; i++)
    {
        code = code << 1;
        code |= data[i] ? 1 : 0;
    }
    furi_string_cat_printf(buffer, "\nFacility: %lX (%ld)", code, code);

    // 20 bit card id
    code = 0;
    for (int i = 14; i <= 33; i++)
    {
        code = code << 1;
        code |= data[i] ? 1 : 0;
    }
    furi_string_cat_printf(buffer, "\nCard: %lX (%ld)", code, code);
}

void wiegand_add_info_36bit(FuriString *buffer)
{
    // We assume this is HID 36 bit Keyscan - C15001 format.

    // 10 bits OEM
    uint32_t oem = 0;
    for (int i = 1; i <= 10; i++)
    {
        oem = (oem << 1) | (data[i] ? 1 : 0);
    }
    furi_string_cat_printf(buffer, "\nOEM: %lX (%ld)", oem, oem);

    // 8 bits facility code
    uint32_t facilityCode = 0;
    for (int i = 11; i <= 18; i++)
    {
        facilityCode = (facilityCode << 1) | (data[i] ? 1 : 0);
    }
    furi_string_cat_printf(buffer, "\nFacility: %lX (%ld)", facilityCode, facilityCode);

    // 16 bits card ID
    uint32_t cardID = 0;
    for (int i = 19; i <= 34; i++)
    {
        cardID = (cardID << 1) | (data[i] ? 1 : 0);
    }
    furi_string_cat_printf(buffer, "\nCard: %lX (%ld)", cardID, cardID);
}

void wiegand_add_info(FuriString *buffer)
{
    furi_string_push_back(buffer, '\n');
    if (bit_count == 4 || bit_count == 8)
    {
        wiegand_add_info_4bit_8bit(buffer);
    }
    else if (bit_count == 26)
    {
        wiegand_add_info_26bit(buffer);
    }
    else if (bit_count == 24)
    {
        wiegand_add_info_24bit(buffer);
    }
    else if (bit_count == 35)
    {
        wiegand_add_info_35bit(buffer);
    }
    else if (bit_count == 36)
    {
        wiegand_add_info_36bit(buffer);
    }
    else if (bit_count == 48)
    {
        wiegand_add_info_48bit(buffer);
    }
    furi_string_push_back(buffer, '\n');
}

void wiegand_button_callback(GuiButtonType result, InputType type, void *context)
{
    App *app = context;
    if (type == InputTypeShort && result == GuiButtonTypeLeft)
    {
        view_dispatcher_send_custom_event(app->view_dispatcher, WiegandDataSceneSaveButtonEvent);
    }
    else if (type == InputTypeShort && result == GuiButtonTypeCenter)
    {
        view_dispatcher_send_custom_event(app->view_dispatcher, WiegandDataScenePlayButtonEvent);
    }
}

void wiegand_data_scene_on_enter(void *context)
{
    App *app = context;
    widget_reset(app->widget);
    widget_add_string_element(app->widget, 0, 0, AlignLeft, AlignTop, FontPrimary, "Wiegand Data");
    FuriString *buffer = furi_string_alloc(1024);
    furi_string_printf(buffer, "Bits: %d\n", bit_count);
    for (int i = 0; i < bit_count; i++)
    {
        furi_string_push_back(buffer, data[i] ? '1' : '0');
        if ((bit_count - i - 1) % 22 == 21)
        {
            furi_string_push_back(buffer, '\n');
        }
    }
    // furi_string_cat_printf(buffer, "\nPulse: %ld us", (data_rise[0] - data_fall[0]) / 64);
    // furi_string_cat_printf(buffer, "\nPeriod: %ld us", (data_fall[1] - data_fall[0]) / 64);
    wiegand_add_info(buffer);
    for (int i = 0; i < bit_count;)
    {
        uint32_t pulse = (data_rise[i] - data_fall[i]) / 64;
        i++;
        uint32_t period = (i < bit_count) ? (data_fall[i] - data_fall[i - 1]) / 64 : 0;
        furi_string_cat_printf(
            buffer, "\n%c : %ld us, %ld us", data[i] ? '1' : '0', pulse, period);
    }
    widget_add_text_scroll_element(app->widget, 0, 12, 128, 34, furi_string_get_cstr(buffer));
    if (!data_saved)
    {
        widget_add_button_element(
            app->widget, GuiButtonTypeLeft, "Save", wiegand_button_callback, app);
    }
    widget_add_button_element(
        app->widget, GuiButtonTypeCenter, "Play", wiegand_button_callback, app);
    view_dispatcher_switch_to_view(app->view_dispatcher, WiegandWidgetView);
}

bool wiegand_data_scene_on_event(void *context, SceneManagerEvent event)
{
    App *app = context;
    bool consumed = false;
    switch (event.type)
    {
    case SceneManagerEventTypeCustom:
        switch (event.event)
        {
        case WiegandDataScenePlayButtonEvent:
            wiegand_play();
            consumed = true;
            break;
        case WiegandDataSceneSaveButtonEvent:
            scene_manager_next_scene(app->scene_manager, WiegandSaveScene);
            consumed = true;
            break;
        default:
            consumed = false;
            break;
        }
        break;
    default:
        break;
    }
    return consumed;
}