#include "../nfc_i.h"

void nfc_scene_nfc_data_info_widget_callback(GuiButtonType result, InputType type, void* context) {
    Nfc* nfc = context;
    if(type == InputTypeShort) {
        view_dispatcher_send_custom_event(nfc->view_dispatcher, result);
    }
}

void nfc_scene_slix_build_string(
    FuriString* temp_str,
    NfcVData* nfcv_data,
    SlixTypeFeatures features,
    const char* type) {
    furi_string_cat_printf(temp_str, "Type: %s\n", type);
    furi_string_cat_printf(temp_str, "Keys:\n");
    if(features & SlixFeatureRead) {
        furi_string_cat_printf(
            temp_str,
            " Read     %08llX%s\n",
            nfc_util_bytes2num(nfcv_data->sub_data.slix.key_read, 4),
            (nfcv_data->sub_data.slix.flags & NfcVSlixDataFlagsHasKeyRead) ? "" : " (unset)");
    }
    if(features & SlixFeatureWrite) {
        furi_string_cat_printf(
            temp_str,
            " Write    %08llX%s\n",
            nfc_util_bytes2num(nfcv_data->sub_data.slix.key_write, 4),
            (nfcv_data->sub_data.slix.flags & NfcVSlixDataFlagsHasKeyWrite) ? "" : " (unset)");
    }
    if(features & SlixFeaturePrivacy) {
        furi_string_cat_printf(
            temp_str,
            " Privacy  %08llX%s\n",
            nfc_util_bytes2num(nfcv_data->sub_data.slix.key_privacy, 4),
            (nfcv_data->sub_data.slix.flags & NfcVSlixDataFlagsHasKeyPrivacy) ? "" : " (unset)");
        furi_string_cat_printf(
            temp_str,
            " Privacy mode %s\n",
            (nfcv_data->sub_data.slix.flags & NfcVSlixDataFlagsPrivacy) ? "ENABLED" : "DISABLED");
    }
    if(features & SlixFeatureDestroy) {
        furi_string_cat_printf(
            temp_str,
            " Destroy  %08llX%s\n",
            nfc_util_bytes2num(nfcv_data->sub_data.slix.key_destroy, 4),
            (nfcv_data->sub_data.slix.flags & NfcVSlixDataFlagsHasKeyDestroy) ? "" : " (unset)");
    }
    if(features & SlixFeatureEas) {
        furi_string_cat_printf(
            temp_str,
            " EAS      %08llX%s\n",
            nfc_util_bytes2num(nfcv_data->sub_data.slix.key_eas, 4),
            (nfcv_data->sub_data.slix.flags & NfcVSlixDataFlagsHasKeyEas) ? "" : " (unset)");
    }
    if(features & SlixFeatureSignature) {
        furi_string_cat_printf(
            temp_str,
            "Signature %08llX...\n",
            nfc_util_bytes2num(nfcv_data->sub_data.slix.signature, 4));
    }
    furi_string_cat_printf(
        temp_str,
        "DSFID: %02X %s\n",
        nfcv_data->dsfid,
        (nfcv_data->security_status[0] & NfcVLockBitDsfid) ? "(locked)" : "");
    furi_string_cat_printf(
        temp_str,
        "AFI: %02X %s\n",
        nfcv_data->afi,
        (nfcv_data->security_status[0] & NfcVLockBitAfi) ? "(locked)" : "");
    furi_string_cat_printf(
        temp_str,
        "EAS: %s\n",
        (nfcv_data->security_status[0] & NfcVLockBitEas) ? "locked" : "not locked");

    if(features & SlixFeatureProtection) {
        furi_string_cat_printf(
            temp_str,
            "PPL: %s\n",
            (nfcv_data->security_status[0] & NfcVLockBitPpl) ? "locked" : "not locked");
        furi_string_cat_printf(temp_str, "Prot.ptr %02X\n", nfcv_data->sub_data.slix.pp_pointer);
        furi_string_cat_printf(temp_str, "Prot.con %02X\n", nfcv_data->sub_data.slix.pp_condition);
    }
}

void nfc_scene_nfc_data_info_on_enter(void* context) {
    Nfc* nfc = context;
    Widget* widget = nfc->widget;
    FuriHalNfcDevData* nfc_data = &nfc->dev->dev_data.nfc_data;
    NfcDeviceData* dev_data = &nfc->dev->dev_data;
    NfcProtocol protocol = dev_data->protocol;
    uint8_t text_scroll_height = 0;
    if((protocol == NfcDeviceProtocolMifareDesfire) || (protocol == NfcDeviceProtocolMifareUl) ||
       (protocol == NfcDeviceProtocolMifareClassic)) {
        widget_add_button_element(
            widget, GuiButtonTypeRight, "More", nfc_scene_nfc_data_info_widget_callback, nfc);
        text_scroll_height = 52;
    } else {
        text_scroll_height = 64;
    }

    FuriString* temp_str;
    temp_str = furi_string_alloc();
    // Set name if present
    if(nfc->dev->dev_name[0] != '\0') {
        furi_string_printf(temp_str, "\ec%s\n", nfc->dev->dev_name);
    }

    // Set tag type
    if(protocol == NfcDeviceProtocolEMV) {
        furi_string_cat_printf(temp_str, "\e#EMV Bank Card\n");
    } else if(protocol == NfcDeviceProtocolMifareUl) {
        furi_string_cat_printf(
            temp_str, "\e#%s\n", nfc_mf_ul_type(dev_data->mf_ul_data.type, true));
    } else if(protocol == NfcDeviceProtocolMifareClassic) {
        furi_string_cat_printf(
            temp_str, "\e#%s\n", nfc_mf_classic_type(dev_data->mf_classic_data.type));
    } else if(protocol == NfcDeviceProtocolMifareDesfire) {
        furi_string_cat_printf(temp_str, "\e#MIFARE DESFire\n");
    } else if(protocol == NfcDeviceProtocolNfcV) {
        switch(dev_data->nfcv_data.sub_type) {
        case NfcVTypePlain:
            furi_string_cat_printf(temp_str, "\e#ISO15693\n");
            break;
        case NfcVTypeSlix:
            furi_string_cat_printf(temp_str, "\e#ISO15693 SLIX\n");
            break;
        case NfcVTypeSlixS:
            furi_string_cat_printf(temp_str, "\e#ISO15693 SLIX-S\n");
            break;
        case NfcVTypeSlixL:
            furi_string_cat_printf(temp_str, "\e#ISO15693 SLIX-L\n");
            break;
        case NfcVTypeSlix2:
            furi_string_cat_printf(temp_str, "\e#ISO15693 SLIX2\n");
            break;
        default:
            furi_string_cat_printf(temp_str, "\e#ISO15693 (unknown)\n");
            break;
        }
    } else {
        furi_string_cat_printf(temp_str, "\e#Unknown ISO tag\n");
    }

    // Set tag iso data
    if(protocol == NfcDeviceProtocolNfcV) {
        NfcVData* nfcv_data = &nfc->dev->dev_data.nfcv_data;

        furi_string_cat_printf(temp_str, "UID:\n");
        for(size_t i = 0; i < nfc_data->uid_len; i++) {
            furi_string_cat_printf(temp_str, " %02X", nfc_data->uid[i]);
        }
        furi_string_cat_printf(temp_str, "\n");

        furi_string_cat_printf(temp_str, "IC Ref: %d\n", nfcv_data->ic_ref);
        furi_string_cat_printf(temp_str, "Blocks: %d\n", nfcv_data->block_num);
        furi_string_cat_printf(temp_str, "Blocksize: %d\n", nfcv_data->block_size);

        switch(dev_data->nfcv_data.sub_type) {
        case NfcVTypePlain:
            furi_string_cat_printf(temp_str, "Type: Plain\n");
            break;
        case NfcVTypeSlix:
            nfc_scene_slix_build_string(temp_str, nfcv_data, SlixFeatureSlix, "SLIX");
            break;
        case NfcVTypeSlixS:
            nfc_scene_slix_build_string(temp_str, nfcv_data, SlixFeatureSlixS, "SLIX-S");
            break;
        case NfcVTypeSlixL:
            nfc_scene_slix_build_string(temp_str, nfcv_data, SlixFeatureSlixL, "SLIX-L");
            break;
        case NfcVTypeSlix2:
            nfc_scene_slix_build_string(temp_str, nfcv_data, SlixFeatureSlix2, "SLIX2");
            break;
        default:
            furi_string_cat_printf(temp_str, "\e#ISO15693 (unknown)\n");
            break;
        }

        furi_string_cat_printf(
            temp_str, "Data (%d byte)\n", nfcv_data->block_num * nfcv_data->block_size);

        int maxBlocks = nfcv_data->block_num;
        if(maxBlocks > 32) {
            maxBlocks = 32;
            furi_string_cat_printf(temp_str, "(truncated to %d blocks)\n", maxBlocks);
        }

        for(int block = 0; block < maxBlocks; block++) {
            const char* status = (nfcv_data->security_status[block] & 0x01) ? "(lck)" : "";
            for(int pos = 0; pos < nfcv_data->block_size; pos++) {
                furi_string_cat_printf(
                    temp_str, " %02X", nfcv_data->data[block * nfcv_data->block_size + pos]);
            }
            furi_string_cat_printf(temp_str, " %s\n", status);
        }

    } else {
        char iso_type = FURI_BIT(nfc_data->sak, 5) ? '4' : '3';
        furi_string_cat_printf(temp_str, "ISO 14443-%c (NFC-A)\n", iso_type);
        furi_string_cat_printf(temp_str, "UID:");
        for(size_t i = 0; i < nfc_data->uid_len; i++) {
            furi_string_cat_printf(temp_str, " %02X", nfc_data->uid[i]);
        }
        furi_string_cat_printf(
            temp_str, "\nATQA: %02X %02X ", nfc_data->atqa[1], nfc_data->atqa[0]);
        furi_string_cat_printf(temp_str, " SAK: %02X", nfc_data->sak);
    }

    // Set application specific data
    if(protocol == NfcDeviceProtocolMifareDesfire) {
        MifareDesfireData* data = &dev_data->mf_df_data;
        uint32_t bytes_total = 1UL << (data->version.sw_storage >> 1);
        uint32_t bytes_free = data->free_memory ? data->free_memory->bytes : 0;
        furi_string_cat_printf(temp_str, "\n%lu", bytes_total);
        if(data->version.sw_storage & 1) {
            furi_string_push_back(temp_str, '+');
        }
        furi_string_cat_printf(temp_str, " bytes, %lu bytes free\n", bytes_free);

        uint16_t n_apps = 0;
        uint16_t n_files = 0;
        for(MifareDesfireApplication* app = data->app_head; app; app = app->next) {
            n_apps++;
            for(MifareDesfireFile* file = app->file_head; file; file = file->next) {
                n_files++;
            }
        }
        furi_string_cat_printf(temp_str, "%d Application", n_apps);
        if(n_apps != 1) {
            furi_string_push_back(temp_str, 's');
        }
        furi_string_cat_printf(temp_str, ", %d file", n_files);
        if(n_files != 1) {
            furi_string_push_back(temp_str, 's');
        }
    } else if(protocol == NfcDeviceProtocolMifareUl) {
        MfUltralightData* data = &dev_data->mf_ul_data;
        furi_string_cat_printf(
            temp_str, "\nPages Read %d/%d", data->data_read / 4, data->data_size / 4);
        if(data->data_size > data->data_read) {
            furi_string_cat_printf(temp_str, "\nPassword-protected");
        } else if(data->auth_success) {
            MfUltralightConfigPages* config_pages = mf_ultralight_get_config_pages(data);
            if(config_pages) {
                furi_string_cat_printf(
                    temp_str,
                    "\nPassword: %02X %02X %02X %02X",
                    config_pages->auth_data.pwd.raw[0],
                    config_pages->auth_data.pwd.raw[1],
                    config_pages->auth_data.pwd.raw[2],
                    config_pages->auth_data.pwd.raw[3]);
                furi_string_cat_printf(
                    temp_str,
                    "\nPACK: %02X %02X",
                    config_pages->auth_data.pack.raw[0],
                    config_pages->auth_data.pack.raw[1]);
            }
        }
    } else if(protocol == NfcDeviceProtocolMifareClassic) {
        MfClassicData* data = &dev_data->mf_classic_data;
        uint8_t sectors_total = mf_classic_get_total_sectors_num(data->type);
        uint8_t keys_total = sectors_total * 2;
        uint8_t keys_found = 0;
        uint8_t sectors_read = 0;
        mf_classic_get_read_sectors_and_keys(data, &sectors_read, &keys_found);
        furi_string_cat_printf(temp_str, "\nKeys Found %d/%d", keys_found, keys_total);
        furi_string_cat_printf(temp_str, "\nSectors Read %d/%d", sectors_read, sectors_total);
    }

    // Add text scroll widget
    widget_add_text_scroll_element(
        widget, 0, 0, 128, text_scroll_height, furi_string_get_cstr(temp_str));
    furi_string_free(temp_str);

    view_dispatcher_switch_to_view(nfc->view_dispatcher, NfcViewWidget);
}

bool nfc_scene_nfc_data_info_on_event(void* context, SceneManagerEvent event) {
    Nfc* nfc = context;
    NfcProtocol protocol = nfc->dev->dev_data.protocol;
    bool consumed = false;

    if(event.type == SceneManagerEventTypeCustom) {
        if(event.event == GuiButtonTypeRight) {
            if(protocol == NfcDeviceProtocolMifareDesfire) {
                scene_manager_next_scene(nfc->scene_manager, NfcSceneMfDesfireData);
                consumed = true;
            } else if(protocol == NfcDeviceProtocolMifareUl) {
                scene_manager_next_scene(nfc->scene_manager, NfcSceneMfUltralightData);
                consumed = true;
            } else if(protocol == NfcDeviceProtocolMifareClassic) {
                scene_manager_next_scene(nfc->scene_manager, NfcSceneMfClassicData);
            } else if(protocol == NfcDeviceProtocolNfcV) {
                scene_manager_next_scene(nfc->scene_manager, NfcSceneNfcVMenu);
                consumed = true;
            }
        }
    }

    return consumed;
}

void nfc_scene_nfc_data_info_on_exit(void* context) {
    Nfc* nfc = context;

    widget_reset(nfc->widget);
}