#include "bt_hid.h"
#include <furi_hal_bt.h>
#include <notification/notification_messages.h>
#include <dolphin/dolphin.h>

#define TAG "BtHidApp"

enum BtDebugSubmenuIndex {
    BtHidSubmenuIndexKeynote,
    BtHidSubmenuIndexKeyboard,
    BtHidSubmenuIndexMedia,
    BtHidSubmenuIndexTikTok,
    BtHidSubmenuIndexMouse,
};

void bt_hid_submenu_callback(void* context, uint32_t index) {
    furi_assert(context);
    BtHid* app = context;
    if(index == BtHidSubmenuIndexKeynote) {
        app->view_id = BtHidViewKeynote;
        view_dispatcher_switch_to_view(app->view_dispatcher, BtHidViewKeynote);
    } else if(index == BtHidSubmenuIndexKeyboard) {
        app->view_id = BtHidViewKeyboard;
        view_dispatcher_switch_to_view(app->view_dispatcher, BtHidViewKeyboard);
    } else if(index == BtHidSubmenuIndexMedia) {
        app->view_id = BtHidViewMedia;
        view_dispatcher_switch_to_view(app->view_dispatcher, BtHidViewMedia);
    } else if(index == BtHidSubmenuIndexMouse) {
        app->view_id = BtHidViewMouse;
        view_dispatcher_switch_to_view(app->view_dispatcher, BtHidViewMouse);
    } else if(index == BtHidSubmenuIndexTikTok) {
        app->view_id = BtHidViewTikTok;
        view_dispatcher_switch_to_view(app->view_dispatcher, BtHidViewTikTok);
    }
}

void bt_hid_dialog_callback(DialogExResult result, void* context) {
    furi_assert(context);
    BtHid* app = context;
    if(result == DialogExResultLeft) {
        view_dispatcher_stop(app->view_dispatcher);
    } else if(result == DialogExResultRight) {
        view_dispatcher_switch_to_view(app->view_dispatcher, app->view_id); // Show last view
    } else if(result == DialogExResultCenter) {
        view_dispatcher_switch_to_view(app->view_dispatcher, BtHidViewSubmenu);
    }
}

uint32_t bt_hid_exit_confirm_view(void* context) {
    UNUSED(context);
    return BtHidViewExitConfirm;
}

uint32_t bt_hid_exit(void* context) {
    UNUSED(context);
    return VIEW_NONE;
}

void bt_hid_connection_status_changed_callback(BtStatus status, void* context) {
    furi_assert(context);
    BtHid* bt_hid = context;
    bool connected = (status == BtStatusConnected);
    if(connected) {
        notification_internal_message(bt_hid->notifications, &sequence_set_blue_255);
    } else {
        notification_internal_message(bt_hid->notifications, &sequence_reset_blue);
    }
    bt_hid_keynote_set_connected_status(bt_hid->bt_hid_keynote, connected);
    bt_hid_keyboard_set_connected_status(bt_hid->bt_hid_keyboard, connected);
    bt_hid_media_set_connected_status(bt_hid->bt_hid_media, connected);
    bt_hid_mouse_set_connected_status(bt_hid->bt_hid_mouse, connected);
    bt_hid_tiktok_set_connected_status(bt_hid->bt_hid_tiktok, connected);
}

BtHid* bt_hid_app_alloc() {
    BtHid* app = malloc(sizeof(BtHid));

    // Gui
    app->gui = furi_record_open(RECORD_GUI);

    // Bt
    app->bt = furi_record_open(RECORD_BT);

    // Notifications
    app->notifications = furi_record_open(RECORD_NOTIFICATION);

    // View dispatcher
    app->view_dispatcher = view_dispatcher_alloc();
    view_dispatcher_enable_queue(app->view_dispatcher);
    view_dispatcher_attach_to_gui(app->view_dispatcher, app->gui, ViewDispatcherTypeFullscreen);

    // Submenu view
    app->submenu = submenu_alloc();
    submenu_add_item(
        app->submenu, "Keynote", BtHidSubmenuIndexKeynote, bt_hid_submenu_callback, app);
    submenu_add_item(
        app->submenu, "Keyboard", BtHidSubmenuIndexKeyboard, bt_hid_submenu_callback, app);
    submenu_add_item(app->submenu, "Media", BtHidSubmenuIndexMedia, bt_hid_submenu_callback, app);
    submenu_add_item(
        app->submenu, "TikTok Controller", BtHidSubmenuIndexTikTok, bt_hid_submenu_callback, app);
    submenu_add_item(app->submenu, "Mouse", BtHidSubmenuIndexMouse, bt_hid_submenu_callback, app);
    view_set_previous_callback(submenu_get_view(app->submenu), bt_hid_exit);
    view_dispatcher_add_view(
        app->view_dispatcher, BtHidViewSubmenu, submenu_get_view(app->submenu));

    // Dialog view
    app->dialog = dialog_ex_alloc();
    dialog_ex_set_result_callback(app->dialog, bt_hid_dialog_callback);
    dialog_ex_set_context(app->dialog, app);
    dialog_ex_set_left_button_text(app->dialog, "Exit");
    dialog_ex_set_right_button_text(app->dialog, "Stay");
    dialog_ex_set_center_button_text(app->dialog, "Menu");
    dialog_ex_set_header(app->dialog, "Close Current App?", 16, 12, AlignLeft, AlignTop);
    view_dispatcher_add_view(
        app->view_dispatcher, BtHidViewExitConfirm, dialog_ex_get_view(app->dialog));

    // Keynote view
    app->bt_hid_keynote = bt_hid_keynote_alloc();
    view_set_previous_callback(
        bt_hid_keynote_get_view(app->bt_hid_keynote), bt_hid_exit_confirm_view);
    view_dispatcher_add_view(
        app->view_dispatcher, BtHidViewKeynote, bt_hid_keynote_get_view(app->bt_hid_keynote));

    // Keyboard view
    app->bt_hid_keyboard = bt_hid_keyboard_alloc();
    view_set_previous_callback(
        bt_hid_keyboard_get_view(app->bt_hid_keyboard), bt_hid_exit_confirm_view);
    view_dispatcher_add_view(
        app->view_dispatcher, BtHidViewKeyboard, bt_hid_keyboard_get_view(app->bt_hid_keyboard));

    // Media view
    app->bt_hid_media = bt_hid_media_alloc();
    view_set_previous_callback(bt_hid_media_get_view(app->bt_hid_media), bt_hid_exit_confirm_view);
    view_dispatcher_add_view(
        app->view_dispatcher, BtHidViewMedia, bt_hid_media_get_view(app->bt_hid_media));

    // TikTok view
    app->bt_hid_tiktok = bt_hid_tiktok_alloc();
    view_set_previous_callback(
        bt_hid_tiktok_get_view(app->bt_hid_tiktok), bt_hid_exit_confirm_view);
    view_dispatcher_add_view(
        app->view_dispatcher, BtHidViewTikTok, bt_hid_tiktok_get_view(app->bt_hid_tiktok));

    // Mouse view
    app->bt_hid_mouse = bt_hid_mouse_alloc();
    view_set_previous_callback(bt_hid_mouse_get_view(app->bt_hid_mouse), bt_hid_exit_confirm_view);
    view_dispatcher_add_view(
        app->view_dispatcher, BtHidViewMouse, bt_hid_mouse_get_view(app->bt_hid_mouse));

    // TODO switch to menu after Media is done
    app->view_id = BtHidViewSubmenu;
    view_dispatcher_switch_to_view(app->view_dispatcher, app->view_id);

    return app;
}

void bt_hid_app_free(BtHid* app) {
    furi_assert(app);

    // Reset notification
    notification_internal_message(app->notifications, &sequence_reset_blue);

    // Free views
    view_dispatcher_remove_view(app->view_dispatcher, BtHidViewSubmenu);
    submenu_free(app->submenu);
    view_dispatcher_remove_view(app->view_dispatcher, BtHidViewExitConfirm);
    dialog_ex_free(app->dialog);
    view_dispatcher_remove_view(app->view_dispatcher, BtHidViewKeynote);
    bt_hid_keynote_free(app->bt_hid_keynote);
    view_dispatcher_remove_view(app->view_dispatcher, BtHidViewKeyboard);
    bt_hid_keyboard_free(app->bt_hid_keyboard);
    view_dispatcher_remove_view(app->view_dispatcher, BtHidViewMedia);
    bt_hid_media_free(app->bt_hid_media);
    view_dispatcher_remove_view(app->view_dispatcher, BtHidViewMouse);
    bt_hid_mouse_free(app->bt_hid_mouse);
    view_dispatcher_remove_view(app->view_dispatcher, BtHidViewTikTok);
    bt_hid_tiktok_free(app->bt_hid_tiktok);
    view_dispatcher_free(app->view_dispatcher);

    // Close records
    furi_record_close(RECORD_GUI);
    app->gui = NULL;
    furi_record_close(RECORD_NOTIFICATION);
    app->notifications = NULL;
    furi_record_close(RECORD_BT);
    app->bt = NULL;

    // Free rest
    free(app);
}

int32_t bt_hid_app(void* p) {
    UNUSED(p);
    // Switch profile to Hid
    BtHid* app = bt_hid_app_alloc();
    bt_set_status_changed_callback(app->bt, bt_hid_connection_status_changed_callback, app);
    // Change profile
    if(!bt_set_profile(app->bt, BtProfileHidKeyboard)) {
        FURI_LOG_E(TAG, "Failed to switch profile");
        bt_hid_app_free(app);
        return -1;
    }
    furi_hal_bt_start_advertising();

    DOLPHIN_DEED(DolphinDeedPluginStart);

    view_dispatcher_run(app->view_dispatcher);

    bt_set_status_changed_callback(app->bt, NULL, NULL);
    // Change back profile to Serial
    bt_set_profile(app->bt, BtProfileSerial);

    bt_hid_app_free(app);

    return 0;
}