flipperzero-firmware/applications/u2f/scenes/u2f_scene_main.c

#include "../u2f_app_i.h"
#include "../views/u2f_view.h"
#include <dolphin/dolphin.h>
#include "furi_hal.h"
#include "../u2f.h"

#define U2F_REQUEST_TIMEOUT 500
#define U2F_SUCCESS_TIMEOUT 3000

static void u2f_scene_main_ok_callback(InputType type, void* context) {
    furi_assert(context);
    U2fApp* app = context;
    view_dispatcher_send_custom_event(app->view_dispatcher, U2fCustomEventConfirm);
}

static void u2f_scene_main_event_callback(U2fNotifyEvent evt, void* context) {
    furi_assert(context);
    U2fApp* app = context;
    if(evt == U2fNotifyRegister)
        view_dispatcher_send_custom_event(app->view_dispatcher, U2fCustomEventRegister);
    else if(evt == U2fNotifyAuth)
        view_dispatcher_send_custom_event(app->view_dispatcher, U2fCustomEventAuth);
    else if(evt == U2fNotifyAuthSuccess)
        view_dispatcher_send_custom_event(app->view_dispatcher, U2fCustomEventAuthSuccess);
    else if(evt == U2fNotifyWink)
        view_dispatcher_send_custom_event(app->view_dispatcher, U2fCustomEventWink);
    else if(evt == U2fNotifyConnect)
        view_dispatcher_send_custom_event(app->view_dispatcher, U2fCustomEventConnect);
    else if(evt == U2fNotifyDisconnect)
        view_dispatcher_send_custom_event(app->view_dispatcher, U2fCustomEventDisconnect);
    else if(evt == U2fNotifyError)
        view_dispatcher_send_custom_event(app->view_dispatcher, U2fCustomEventDataError);
}

static void u2f_scene_main_timer_callback(void* context) {
    furi_assert(context);
    U2fApp* app = context;
    view_dispatcher_send_custom_event(app->view_dispatcher, U2fCustomEventTimeout);
}

bool u2f_scene_main_on_event(void* context, SceneManagerEvent event) {
    furi_assert(context);
    U2fApp* app = context;
    bool consumed = false;

    if(event.type == SceneManagerEventTypeCustom) {
        if(event.event == U2fCustomEventConnect) {
            osTimerStop(app->timer);
            u2f_view_set_state(app->u2f_view, U2fMsgIdle);
        } else if(event.event == U2fCustomEventDisconnect) {
            osTimerStop(app->timer);
            app->event_cur = U2fCustomEventNone;
            u2f_view_set_state(app->u2f_view, U2fMsgNotConnected);
        } else if((event.event == U2fCustomEventRegister) || (event.event == U2fCustomEventAuth)) {
            osTimerStart(app->timer, U2F_REQUEST_TIMEOUT);
            if(app->event_cur == U2fCustomEventNone) {
                app->event_cur = event.event;
                if(event.event == U2fCustomEventRegister)
                    u2f_view_set_state(app->u2f_view, U2fMsgRegister);
                else if(event.event == U2fCustomEventAuth)
                    u2f_view_set_state(app->u2f_view, U2fMsgAuth);
                notification_message(app->notifications, &sequence_display_backlight_on);
                notification_message(app->notifications, &sequence_single_vibro);
            }
            notification_message(app->notifications, &sequence_blink_magenta_10);
        } else if(event.event == U2fCustomEventWink) {
            notification_message(app->notifications, &sequence_blink_magenta_10);
        } else if(event.event == U2fCustomEventAuthSuccess) {
            notification_message_block(app->notifications, &sequence_set_green_255);
            DOLPHIN_DEED(DolphinDeedU2fAuthorized);
            osTimerStart(app->timer, U2F_SUCCESS_TIMEOUT);
            app->event_cur = U2fCustomEventNone;
            u2f_view_set_state(app->u2f_view, U2fMsgSuccess);
        } else if(event.event == U2fCustomEventTimeout) {
            notification_message_block(app->notifications, &sequence_reset_rgb);
            app->event_cur = U2fCustomEventNone;
            u2f_view_set_state(app->u2f_view, U2fMsgIdle);
        } else if(event.event == U2fCustomEventConfirm) {
            if(app->event_cur != U2fCustomEventNone) {
                u2f_confirm_user_present(app->u2f_instance);
            }
        } else if(event.event == U2fCustomEventDataError) {
            notification_message(app->notifications, &sequence_set_red_255);
            osTimerStop(app->timer);
            u2f_view_set_state(app->u2f_view, U2fMsgError);
        }
        consumed = true;
    }

    return consumed;
}

void u2f_scene_main_on_enter(void* context) {
    U2fApp* app = context;

    app->timer = osTimerNew(u2f_scene_main_timer_callback, osTimerOnce, app, NULL);

    app->u2f_instance = u2f_alloc();
    app->u2f_ready = u2f_init(app->u2f_instance);
    if(app->u2f_ready == true) {
        u2f_set_event_callback(app->u2f_instance, u2f_scene_main_event_callback, app);
        app->u2f_hid = u2f_hid_start(app->u2f_instance);
        u2f_view_set_ok_callback(app->u2f_view, u2f_scene_main_ok_callback, app);
    } else {
        u2f_free(app->u2f_instance);
        u2f_view_set_state(app->u2f_view, U2fMsgError);
    }

    view_dispatcher_switch_to_view(app->view_dispatcher, U2fAppViewMain);
}

void u2f_scene_main_on_exit(void* context) {
    U2fApp* app = context;
    notification_message_block(app->notifications, &sequence_reset_rgb);
    osTimerStop(app->timer);
    osTimerDelete(app->timer);
    if(app->u2f_ready == true) {
        u2f_hid_stop(app->u2f_hid);
        u2f_free(app->u2f_instance);
    }
}