Momentum-Apps/applications/nfc/scenes/nfc_scene_detect_reader.c

#include "../nfc_i.h"
#include <dolphin/dolphin.h>

#define NFC_SCENE_DETECT_READER_LOG_SIZE_MAX (200)

enum {
    NfcSceneDetectReaderStateWidget,
    NfcSceneDetectReaderStateTextBox,
};

bool nfc_detect_reader_worker_callback(NfcWorkerEvent event, void* context) {
    UNUSED(event);
    furi_assert(context);
    Nfc* nfc = context;
    view_dispatcher_send_custom_event(nfc->view_dispatcher, NfcCustomEventWorkerExit);
    return true;
}

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

void nfc_detect_reader_textbox_callback(void* context) {
    furi_assert(context);
    Nfc* nfc = context;
    view_dispatcher_send_custom_event(nfc->view_dispatcher, NfcCustomEventViewExit);
}

// Add widget with device name or inform that data received
static void nfc_scene_detect_reader_widget_config(Nfc* nfc, bool data_received) {
    Widget* widget = nfc->widget;
    widget_reset(widget);

    widget_add_icon_element(widget, 0, 14, &I_Reader_detect);
    widget_add_string_element(
        widget, 64, 3, AlignCenter, AlignTop, FontSecondary, "Hold Near Reader");
    widget_add_string_element(widget, 55, 22, AlignLeft, AlignTop, FontPrimary, "Emulating...");

    if(data_received) {
        widget_add_button_element(
            widget, GuiButtonTypeCenter, "Log", nfc_scene_detect_reader_widget_callback, nfc);
    }
}

void nfc_scene_detect_reader_on_enter(void* context) {
    Nfc* nfc = context;
    DOLPHIN_DEED(DolphinDeedNfcEmulate);
    FuriHalNfcDevData nfc_params = {
        .uid = {0x36, 0x9C, 0xe7, 0xb1, 0x0A, 0xC1, 0x34},
        .uid_len = 7,
        .atqa = {0x44, 0x00},
        .sak = 0x08,
        .type = FuriHalNfcTypeA,
    };
    nfc->dev->dev_data.nfc_data = nfc_params;

    // Setup Widget
    nfc_scene_detect_reader_widget_config(nfc, false);
    // Setup TextBox
    TextBox* text_box = nfc->text_box;
    text_box_set_font(text_box, TextBoxFontHex);
    text_box_set_focus(text_box, TextBoxFocusEnd);
    string_reset(nfc->text_box_store);

    // Set Widget state and view
    scene_manager_set_scene_state(
        nfc->scene_manager, NfcSceneDetectReader, NfcSceneDetectReaderStateWidget);
    view_dispatcher_switch_to_view(nfc->view_dispatcher, NfcViewWidget);
    // Start worker
    memset(&nfc->dev->dev_data.reader_data, 0, sizeof(NfcReaderRequestData));
    nfc_worker_start(
        nfc->worker,
        NfcWorkerStateUidEmulate,
        &nfc->dev->dev_data,
        nfc_detect_reader_worker_callback,
        nfc);

    nfc_blink_start(nfc);
}

bool nfc_scene_detect_reader_on_event(void* context, SceneManagerEvent event) {
    Nfc* nfc = context;
    NfcReaderRequestData* reader_data = &nfc->dev->dev_data.reader_data;
    uint32_t state = scene_manager_get_scene_state(nfc->scene_manager, NfcSceneDetectReader);
    bool consumed = false;

    if(event.type == SceneManagerEventTypeCustom) {
        if(event.event == NfcCustomEventWorkerExit) {
            // Add data button to widget if data is received for the first time
            if(!string_size(nfc->text_box_store)) {
                nfc_scene_detect_reader_widget_config(nfc, true);
            }
            // Update TextBox data
            if(string_size(nfc->text_box_store) < NFC_SCENE_DETECT_READER_LOG_SIZE_MAX) {
                string_cat_printf(nfc->text_box_store, "R:");
                for(uint16_t i = 0; i < reader_data->size; i++) {
                    string_cat_printf(nfc->text_box_store, " %02X", reader_data->data[i]);
                }
                string_push_back(nfc->text_box_store, '\n');
                text_box_set_text(nfc->text_box, string_get_cstr(nfc->text_box_store));
            }
            memset(reader_data, 0, sizeof(NfcReaderRequestData));
            consumed = true;
        } else if(event.event == GuiButtonTypeCenter && state == NfcSceneDetectReaderStateWidget) {
            view_dispatcher_switch_to_view(nfc->view_dispatcher, NfcViewTextBox);
            scene_manager_set_scene_state(
                nfc->scene_manager, NfcSceneDetectReader, NfcSceneDetectReaderStateTextBox);
            consumed = true;
        } else if(event.event == NfcCustomEventViewExit && state == NfcSceneDetectReaderStateTextBox) {
            view_dispatcher_switch_to_view(nfc->view_dispatcher, NfcViewWidget);
            scene_manager_set_scene_state(
                nfc->scene_manager, NfcSceneDetectReader, NfcSceneDetectReaderStateWidget);
            consumed = true;
        }
    } else if(event.type == SceneManagerEventTypeBack) {
        if(state == NfcSceneDetectReaderStateTextBox) {
            view_dispatcher_switch_to_view(nfc->view_dispatcher, NfcViewWidget);
            scene_manager_set_scene_state(
                nfc->scene_manager, NfcSceneDetectReader, NfcSceneDetectReaderStateWidget);
            consumed = true;
        }
    }

    return consumed;
}

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

    // Stop worker
    nfc_worker_stop(nfc->worker);

    // Clear view
    widget_reset(nfc->widget);
    text_box_reset(nfc->text_box);
    string_reset(nfc->text_box_store);

    nfc_blink_stop(nfc);
}