#include "hid.h"
#include <extra_profiles/hid_profile.h>
#include <profiles/serial_profile.h>
#include "views.h"
#include <notification/notification_messages.h>

#define TAG "CookieClickerApp"

enum HidDebugSubmenuIndex {
    HidSubmenuIndexInstructions,
    HidSubmenuConfigure,
    HidSubmenuIndexClicker,
    HidSubmenuIndexRemovePairing,
    HidSubmenuIndexCredits,
};

static void bt_hid_remove_pairing(Hid* hid) {
    bt_disconnect(hid->bt);
    furi_delay_ms(200);
    furi_hal_bt_stop_advertising();
    bt_forget_bonded_devices(hid->bt);
    furi_hal_bt_start_advertising();
    hid->ble = NULL;
    notification_message(hid->notifications, &sequence_success);
}

static void hid_submenu_callback(void* context, uint32_t index) {
    furi_assert(context);
    Hid* app = context;
    if(index == HidSubmenuIndexInstructions) {
        app->view_id = BtHidViewInstructions;
        view_dispatcher_switch_to_view(app->view_dispatcher, app->view_id);
    } else if(index == HidSubmenuIndexClicker) {
        app->view_id = BtHidViewClicker;
        view_dispatcher_switch_to_view(app->view_dispatcher, app->view_id);
    } else if(index == HidSubmenuIndexCredits) {
        app->view_id = BtHidViewCredits;
        view_dispatcher_switch_to_view(app->view_dispatcher, app->view_id);
    } else if(index == HidSubmenuIndexRemovePairing) {
        bt_hid_remove_pairing(app);
    } else if(index == HidSubmenuConfigure) {
        app->view_id = BtHidViewConfigure;
        view_dispatcher_switch_to_view(app->view_dispatcher, app->view_id);
    }
}

static void bt_hid_connection_status_changed_callback(BtStatus status, void* context) {
    furi_assert(context);
    Hid* hid = context;
    bool connected = (status == BtStatusConnected);
    if(connected) {
        notification_internal_message(hid->notifications, &sequence_set_blue_255);
    } else {
        notification_internal_message(hid->notifications, &sequence_reset_blue);
    }
    hid_cc_set_connected_status(hid->hid_cc, connected);
}

static uint32_t hid_submenu_view(void* context) {
    UNUSED(context);
    return HidViewSubmenu;
}

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

Hid* hid_alloc() {
    Hid* app = malloc(sizeof(Hid));

    // 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_set_header(app->submenu, "Cookie clicker");
    submenu_add_item(
        app->submenu, "Instructions", HidSubmenuIndexInstructions, hid_submenu_callback, app);
    submenu_add_item(
        app->submenu, "Configuration", HidSubmenuConfigure, hid_submenu_callback, app);
    submenu_add_item(
        app->submenu, "BT Phone Clicker", HidSubmenuIndexClicker, hid_submenu_callback, app);
    submenu_add_item(
        app->submenu, "Remove Pairing", HidSubmenuIndexRemovePairing, hid_submenu_callback, app);
    submenu_add_item(app->submenu, "Credits", HidSubmenuIndexCredits, hid_submenu_callback, app);
    view_set_previous_callback(submenu_get_view(app->submenu), hid_exit);
    view_dispatcher_add_view(app->view_dispatcher, HidViewSubmenu, submenu_get_view(app->submenu));
    app->view_id = HidViewSubmenu;
    view_dispatcher_switch_to_view(app->view_dispatcher, app->view_id);
    return app;
}

void hid_setting_changed(Hid* instance) {
    hid_cc_set_cursor_position(
        instance->hid_cc,
        instance->offset_x,
        instance->offset_y,
        instance->offset_repeat,
        instance->offset_speed);
}

void hid_setting_change_x(VariableItem* item) {
    FuriString* str = furi_string_alloc();
    Hid* instance = variable_item_get_context(item);
    uint8_t index = variable_item_get_current_value_index(item);
    instance->offset_x = index * 10;
    furi_string_cat_printf(str, "%d", instance->offset_x);
    variable_item_set_current_value_text(item, furi_string_get_cstr(str));
    furi_string_free(str);
    hid_setting_changed(instance);
}

void hid_setting_change_y(VariableItem* item) {
    FuriString* str = furi_string_alloc();
    Hid* instance = variable_item_get_context(item);
    uint8_t index = variable_item_get_current_value_index(item);
    instance->offset_y = index * 10;
    furi_string_cat_printf(str, "%d", instance->offset_y);
    variable_item_set_current_value_text(item, furi_string_get_cstr(str));
    furi_string_free(str);
    hid_setting_changed(instance);
}

void hid_setting_change_repeat(VariableItem* item) {
    FuriString* str = furi_string_alloc();
    Hid* instance = variable_item_get_context(item);
    uint8_t index = variable_item_get_current_value_index(item);
    instance->offset_repeat = index + 1;
    furi_string_cat_printf(str, "%d", instance->offset_repeat);
    variable_item_set_current_value_text(item, furi_string_get_cstr(str));
    furi_string_free(str);
    hid_setting_changed(instance);
}

void hid_setting_change_speed(VariableItem* item) {
    FuriString* str = furi_string_alloc();
    Hid* instance = variable_item_get_context(item);
    uint8_t index = variable_item_get_current_value_index(item);
    instance->offset_speed = index * 10 + 1;
    furi_string_cat_printf(str, "%d", instance->offset_speed);
    variable_item_set_current_value_text(item, furi_string_get_cstr(str));
    furi_string_free(str);
    hid_setting_changed(instance);
}

Hid* hid_app_alloc_view(void* context) {
    furi_assert(context);
    Hid* app = context;

    // Instructions view
    app->widget_instructions = widget_alloc();
    widget_add_text_scroll_element(
        app->widget_instructions,
        0,
        0,
        128,
        64,
        "iOS17:Settings/Accessibility\n"
        "/AssistiveTouch=ON. Single-\n"
        "Tap=Pass Through to App.\n\n"
        "While this app is running, go to\n"
        "your phone's bluetooth\n"
        "settings and discover devices.\n"
        "You should see 'Control\n"
        "<flipper>' in the list. Click pair\n"
        "on your phone. On the Flipper\n"
        "Zero you should see a pin #.\n"
        "If the pin matches, click OK on\n"
        "your phone and the Flipper.\n"
        "You should now be connected!\n"
        "Launch the Cookie clicker app\n"
        "on your phone. Tap your\n"
        "phone's screen to click on the\n"
        "first cookie, then on the\n"
        "Flipper Zero, choose\n"
        "'Phone clicker'. Click the\n"
        "UP/DOWN buttons to set the\n"
        "speed. Click the OK button on\n"
        "the Flipper to enable/\n"
        "disable the clicker.\n"
        "Use the configuration setting\n"
        "to change the location of the\n"
        "clicking!\n"
        "iPhone 14: x=40,y=80, m=3\n"
        "Samsung A53: x=40, y=120, m=10\n"
        "Enjoy!\n");
    view_set_previous_callback(widget_get_view(app->widget_instructions), hid_submenu_view);
    view_dispatcher_add_view(
        app->view_dispatcher, BtHidViewInstructions, widget_get_view(app->widget_instructions));

    app->variable_item_list = variable_item_list_alloc();
    variable_item_list_reset(app->variable_item_list);
    VariableItem* item =
        variable_item_list_add(app->variable_item_list, "X offset", 13, hid_setting_change_x, app);
    variable_item_set_current_value_index(item, 5); // 0,10,20,30,...
    variable_item_set_current_value_text(item, "40");
    app->offset_x = 40;
    item =
        variable_item_list_add(app->variable_item_list, "Y offset", 13, hid_setting_change_y, app);
    variable_item_set_current_value_index(item, 8); // 0,10,20,30,...
    variable_item_set_current_value_text(item, "80");
    app->offset_y = 80;
    item = variable_item_list_add(
        app->variable_item_list, "Multiplier", 20, hid_setting_change_repeat, app);
    variable_item_set_current_value_index(item, 0); // 1,2,3,4,...
    variable_item_set_current_value_text(item, "1");
    app->offset_repeat = 3;
    item = variable_item_list_add(
        app->variable_item_list, "CursorSpeed", 50, hid_setting_change_speed, app);
    variable_item_set_current_value_index(item, 5); // 1,2,3,4,...
    variable_item_set_current_value_text(item, "50");
    app->offset_speed = 50;
    view_set_previous_callback(
        variable_item_list_get_view(app->variable_item_list), hid_submenu_view);
    view_dispatcher_add_view(
        app->view_dispatcher,
        BtHidViewConfigure,
        variable_item_list_get_view(app->variable_item_list));

    // Clicker view
    app->hid_cc = hid_cc_alloc(app);
    view_set_previous_callback(hid_cc_get_view(app->hid_cc), hid_submenu_view);
    view_dispatcher_add_view(app->view_dispatcher, BtHidViewClicker, hid_cc_get_view(app->hid_cc));
    hid_setting_changed(app);

    // Credits view
    app->widget_credits = widget_alloc();
    widget_add_text_scroll_element(
        app->widget_credits,
        0,
        0,
        128,
        64,
        "Original app was from\n"
        "external/hid_app. \n\n"
        "Modified by CodeAllNight.\n"
        "https://youtube.com/\n"
        "@MrDerekJamison\n\n"
        "Thanks to all the\n"
        "contributors of the original\n"
        "app!\n\n"
        "Flip the World!\n");
    view_set_previous_callback(widget_get_view(app->widget_credits), hid_submenu_view);
    view_dispatcher_add_view(
        app->view_dispatcher, BtHidViewCredits, widget_get_view(app->widget_credits));

    return app;
}

void hid_free(Hid* app) {
    furi_assert(app);

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

    // Free views
    view_dispatcher_remove_view(app->view_dispatcher, BtHidViewClicker);
    hid_cc_free(app->hid_cc);
    view_dispatcher_remove_view(app->view_dispatcher, BtHidViewCredits);
    widget_free(app->widget_credits);
    view_dispatcher_remove_view(app->view_dispatcher, BtHidViewConfigure);
    variable_item_list_free(app->variable_item_list);
    view_dispatcher_remove_view(app->view_dispatcher, BtHidViewInstructions);
    widget_free(app->widget_instructions);
    view_dispatcher_remove_view(app->view_dispatcher, HidViewSubmenu);
    submenu_free(app->submenu);
    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);
}

void hid_hal_mouse_move(Hid* instance, int8_t dx, int8_t dy) {
    furi_assert(instance);
    if(instance->ble) {
        ble_profile_hid_mouse_move(instance->ble, dx, dy);
    }
}

void hid_hal_mouse_press(Hid* instance, uint16_t event) {
    furi_assert(instance);
    if(instance->ble) {
        ble_profile_hid_mouse_press(instance->ble, event);
    }
}

void hid_hal_mouse_release(Hid* instance, uint16_t event) {
    furi_assert(instance);
    if(instance->ble) {
        ble_profile_hid_mouse_release(instance->ble, event);
    }
}

void hid_hal_mouse_release_all(Hid* instance) {
    furi_assert(instance);
    if(instance->ble) {
        ble_profile_hid_mouse_release_all(instance->ble);
    }
}

int32_t hid_cookie_ble_app(void* p) {
    UNUSED(p);
    Hid* app = hid_alloc();
    app = hid_app_alloc_view(app);

    bt_disconnect(app->bt);

    // Wait 2nd core to update nvm storage
    furi_delay_ms(200);

    // Migrate data from old sd-card folder
    Storage* storage = furi_record_open(RECORD_STORAGE);

    storage_common_migrate(
        storage,
        EXT_PATH("apps/Bluetooth/" HID_BT_KEYS_STORAGE_NAME),
        APP_DATA_PATH(HID_BT_KEYS_STORAGE_NAME));

    bt_keys_storage_set_storage_path(app->bt, APP_DATA_PATH(HID_BT_KEYS_STORAGE_NAME));

    furi_record_close(RECORD_STORAGE);

    app->ble = bt_profile_start(app->bt, ble_profile_hid, NULL);
    if(!app->ble) {
        FURI_LOG_E(TAG, "Failed to start BLE profile");
    }

    furi_hal_bt_start_advertising();
    bt_set_status_changed_callback(app->bt, bt_hid_connection_status_changed_callback, app);

    view_dispatcher_run(app->view_dispatcher);

    bt_set_status_changed_callback(app->bt, NULL, NULL);

    bt_disconnect(app->bt);

    // Wait 2nd core to update nvm storage
    furi_delay_ms(200);

    bt_keys_storage_set_default_path(app->bt);

    if(!bt_profile_restore_default(app->bt)) {
        FURI_LOG_E(TAG, "Failed to switch to Serial profile");
    }

    hid_free(app);

    return 0;
}