#include "../picopass_i.h"
#include <dolphin/dolphin.h>
#include "../picopass_keys.h"

#define PICOPASS_SCENE_DICT_ATTACK_KEYS_BATCH_UPDATE (10)

#define TAG "PicopassSceneEliteDictAttack"

enum {
    PicopassSceneEliteDictAttackDictEliteUser,
    PicopassSceneEliteDictAttackDictStandard,
    PicopassSceneEliteDictAttackDictElite,
};

const char* picopass_dict_name[] = {
    [PicopassSceneEliteDictAttackDictEliteUser] = "Elite User Dictionary",
    [PicopassSceneEliteDictAttackDictStandard] = "Standard System Dictionary",
    [PicopassSceneEliteDictAttackDictElite] = "Elite System Dictionary",
};

static bool picopass_elite_dict_attack_change_dict(Picopass* picopass) {
    bool success = false;

    do {
        uint32_t scene_state =
            scene_manager_get_scene_state(picopass->scene_manager, PicopassSceneEliteDictAttack);
        keys_dict_free(picopass->dict);
        picopass->dict = NULL;
        if(scene_state == PicopassSceneEliteDictAttackDictElite) break;
        if(scene_state == PicopassSceneEliteDictAttackDictEliteUser) {
            if(!keys_dict_check_presence(PICOPASS_ICLASS_STANDARD_DICT_FLIPPER_NAME)) break;
            picopass->dict = keys_dict_alloc(
                PICOPASS_ICLASS_STANDARD_DICT_FLIPPER_NAME,
                KeysDictModeOpenExisting,
                PICOPASS_KEY_LEN);
            scene_state = PicopassSceneEliteDictAttackDictStandard;
        } else if(scene_state == PicopassSceneEliteDictAttackDictStandard) {
            if(!keys_dict_check_presence(PICOPASS_ICLASS_ELITE_DICT_FLIPPER_NAME)) break;
            picopass->dict = keys_dict_alloc(
                PICOPASS_ICLASS_ELITE_DICT_FLIPPER_NAME,
                KeysDictModeOpenExisting,
                PICOPASS_KEY_LEN);
            scene_state = PicopassSceneEliteDictAttackDictElite;
        }
        picopass->dict_attack_ctx.total_keys = keys_dict_get_total_keys(picopass->dict);
        picopass->dict_attack_ctx.current_key = 0;
        picopass->dict_attack_ctx.name = picopass_dict_name[scene_state];
        scene_manager_set_scene_state(
            picopass->scene_manager, PicopassSceneEliteDictAttack, scene_state);
        success = true;
    } while(false);

    return success;
}

NfcCommand picopass_elite_dict_attack_worker_callback(PicopassPollerEvent event, void* context) {
    furi_assert(context);
    NfcCommand command = NfcCommandContinue;

    Picopass* picopass = context;

    if(event.type == PicopassPollerEventTypeRequestMode) {
        event.data->req_mode.mode = PicopassPollerModeRead;
    } else if(event.type == PicopassPollerEventTypeRequestKey) {
        uint8_t key[PICOPASS_KEY_LEN] = {};
        bool is_key_provided = true;
        if(!keys_dict_get_next_key(picopass->dict, key, PICOPASS_KEY_LEN)) {
            if(picopass_elite_dict_attack_change_dict(picopass)) {
                is_key_provided = keys_dict_get_next_key(picopass->dict, key, PICOPASS_KEY_LEN);
                view_dispatcher_send_custom_event(
                    picopass->view_dispatcher, PicopassCustomEventDictAttackUpdateView);
            } else {
                is_key_provided = false;
            }
        }
        uint32_t scene_state =
            scene_manager_get_scene_state(picopass->scene_manager, PicopassSceneEliteDictAttack);
        memcpy(event.data->req_key.key, key, PICOPASS_KEY_LEN);
        event.data->req_key.is_elite_key =
            (scene_state != PicopassSceneEliteDictAttackDictStandard);
        event.data->req_key.is_key_provided = is_key_provided;
        if(is_key_provided) {
            picopass->dict_attack_ctx.current_key++;
            if(picopass->dict_attack_ctx.current_key %
                   PICOPASS_SCENE_DICT_ATTACK_KEYS_BATCH_UPDATE ==
               0) {
                view_dispatcher_send_custom_event(
                    picopass->view_dispatcher, PicopassCustomEventDictAttackUpdateView);
            }
        }
    } else if(
        event.type == PicopassPollerEventTypeSuccess ||
        event.type == PicopassPollerEventTypeFail ||
        event.type == PicopassPollerEventTypeAuthFail) {
        const PicopassDeviceData* data = picopass_poller_get_data(picopass->poller);
        memcpy(&picopass->dev->dev_data, data, sizeof(PicopassDeviceData));
        view_dispatcher_send_custom_event(
            picopass->view_dispatcher, PicopassCustomEventPollerSuccess);
    } else if(event.type == PicopassPollerEventTypeCardLost) {
        picopass->dict_attack_ctx.card_detected = false;
        view_dispatcher_send_custom_event(
            picopass->view_dispatcher, PicopassCustomEventDictAttackUpdateView);
    } else if(event.type == PicopassPollerEventTypeCardDetected) {
        picopass->dict_attack_ctx.card_detected = true;
        view_dispatcher_send_custom_event(
            picopass->view_dispatcher, PicopassCustomEventDictAttackUpdateView);
    }

    return command;
}

static void picopass_scene_elite_dict_attack_update_view(Picopass* instance) {
    if(instance->dict_attack_ctx.card_detected) {
        dict_attack_set_card_detected(instance->dict_attack);
        dict_attack_set_header(instance->dict_attack, instance->dict_attack_ctx.name);
        dict_attack_set_total_dict_keys(
            instance->dict_attack, instance->dict_attack_ctx.total_keys);
        dict_attack_set_current_dict_key(
            instance->dict_attack, instance->dict_attack_ctx.current_key);
    } else {
        dict_attack_set_card_removed(instance->dict_attack);
    }
}

static void picopass_scene_elite_dict_attack_callback(void* context) {
    Picopass* instance = context;

    view_dispatcher_send_custom_event(
        instance->view_dispatcher, PicopassCustomEventDictAttackSkip);
}

void picopass_scene_elite_dict_attack_on_enter(void* context) {
    Picopass* picopass = context;
    dolphin_deed(DolphinDeedNfcRead);

    // Setup dict attack context
    uint32_t state = PicopassSceneEliteDictAttackDictEliteUser;

    bool use_user_dict = keys_dict_check_presence(PICOPASS_ICLASS_ELITE_DICT_USER_NAME);
    if(use_user_dict) {
        picopass->dict = keys_dict_alloc(
            PICOPASS_ICLASS_ELITE_DICT_USER_NAME, KeysDictModeOpenExisting, PICOPASS_KEY_LEN);
        if(keys_dict_get_total_keys(picopass->dict) == 0) {
            keys_dict_free(picopass->dict);
            use_user_dict = false;
        }
    }
    if(use_user_dict) {
        state = PicopassSceneEliteDictAttackDictEliteUser;
    } else {
        picopass->dict = keys_dict_alloc(
            PICOPASS_ICLASS_STANDARD_DICT_FLIPPER_NAME,
            KeysDictModeOpenExisting,
            PICOPASS_KEY_LEN);
        state = PicopassSceneEliteDictAttackDictStandard;
    }
    dict_attack_reset(picopass->dict_attack);
    picopass->dict_attack_ctx.card_detected = false;
    picopass->dict_attack_ctx.total_keys = keys_dict_get_total_keys(picopass->dict);
    picopass->dict_attack_ctx.current_key = 0;
    picopass->dict_attack_ctx.name = picopass_dict_name[state];
    scene_manager_set_scene_state(picopass->scene_manager, PicopassSceneEliteDictAttack, state);

    // Setup view
    picopass_scene_elite_dict_attack_update_view(picopass);
    dict_attack_set_callback(
        picopass->dict_attack, picopass_scene_elite_dict_attack_callback, picopass);

    // Start worker
    picopass->poller = picopass_poller_alloc(picopass->nfc);
    picopass_poller_start(picopass->poller, picopass_elite_dict_attack_worker_callback, picopass);

    view_dispatcher_switch_to_view(picopass->view_dispatcher, PicopassViewDictAttack);
    picopass_blink_start(picopass);
}

bool picopass_scene_elite_dict_attack_on_event(void* context, SceneManagerEvent event) {
    Picopass* picopass = context;
    bool consumed = false;
    PicopassDeviceAuthMethod auth = picopass->dev->dev_data.auth;

    if(event.type == SceneManagerEventTypeCustom) {
        if(event.event == PicopassCustomEventPollerSuccess) {
            if(memcmp(
                   picopass->dev->dev_data.pacs.key,
                   picopass_factory_debit_key,
                   PICOPASS_BLOCK_LEN) == 0) {
                scene_manager_next_scene(picopass->scene_manager, PicopassSceneReadFactorySuccess);
            } else {
                if(auth == PicopassDeviceAuthMethodFailed && picopass->nr_mac_type == AutoNRMAC) {
                    // save partial as <CSN>-partial
                    picopass->dev->format = PicopassDeviceSaveFormatPartial;
                    uint8_t* csn =
                        picopass->dev->dev_data.card_data[PICOPASS_CSN_BLOCK_INDEX].data;
                    for(size_t i = 0; i < PICOPASS_BLOCK_LEN; i++) {
                        snprintf(
                            picopass->text_store + 2 * i,
                            sizeof(picopass->text_store),
                            "%02X",
                            csn[i]);
                    }
                    snprintf(
                        picopass->text_store + 2 * PICOPASS_BLOCK_LEN,
                        sizeof(picopass->text_store),
                        "-partial");
                    strlcpy(
                        picopass->dev->dev_name,
                        picopass->text_store,
                        strlen(picopass->text_store) + 1);
                    FURI_LOG_D(TAG, "Saving name: %s", picopass->text_store);

                    picopass_device_delete(picopass->dev, true);
                    if(picopass_device_save(picopass->dev, picopass->text_store)) {
                        scene_manager_next_scene(picopass->scene_manager, PicopassSceneEmulate);
                    } else {
                        FURI_LOG_W(TAG, "Failed to save partial file");
                        scene_manager_next_scene(
                            picopass->scene_manager, PicopassSceneReadCardSuccess);
                    }
                } else {
                    scene_manager_next_scene(
                        picopass->scene_manager, PicopassSceneReadCardSuccess);
                }
            }
            consumed = true;
        } else if(event.event == PicopassCustomEventDictAttackUpdateView) {
            picopass_scene_elite_dict_attack_update_view(picopass);
            consumed = true;
        } else if(event.event == PicopassCustomEventDictAttackSkip) {
            uint32_t scene_state = scene_manager_get_scene_state(
                picopass->scene_manager, PicopassSceneEliteDictAttack);
            if(scene_state != PicopassSceneEliteDictAttackDictElite) {
                picopass_elite_dict_attack_change_dict(picopass);
                picopass_scene_elite_dict_attack_update_view(picopass);
            } else {
                if(memcmp(
                       picopass->dev->dev_data.pacs.key,
                       picopass_factory_debit_key,
                       PICOPASS_BLOCK_LEN) == 0) {
                    scene_manager_next_scene(
                        picopass->scene_manager, PicopassSceneReadFactorySuccess);
                } else {
                    scene_manager_next_scene(
                        picopass->scene_manager, PicopassSceneReadCardSuccess);
                }
            }
            consumed = true;
        }
    }
    return consumed;
}

void picopass_scene_elite_dict_attack_on_exit(void* context) {
    Picopass* picopass = context;

    if(picopass->dict) {
        keys_dict_free(picopass->dict);
        picopass->dict = NULL;
    }
    picopass->dict_attack_ctx.current_key = 0;
    picopass->dict_attack_ctx.total_keys = 0;

    picopass_poller_stop(picopass->poller);
    picopass_poller_free(picopass->poller);

    // Clear view
    popup_reset(picopass->popup);
    scene_manager_set_scene_state(
        picopass->scene_manager,
        PicopassSceneEliteDictAttack,
        PicopassSceneEliteDictAttackDictEliteUser);

    picopass_blink_stop(picopass);
}