upd airmouse

https://github.com/flipperdevices/flipperzero-good-faps/pull/115/files

+ use different define and file name to make better visual separation from original ble_hid app

Author: xMasterX

non_catalog_apps/air_mouse_ofw/air_mouse_app.c

@@ -1,129 +1,240 @@
 #include <furi.h>
 #include <furi_hal.h>
+#include <furi_hal_bt_hid.h>
+#include <bt/bt_service/bt.h>
 #include <gui/gui.h>
-#include <dialogs/dialogs.h>
+#include <gui/view_dispatcher.h>
+#include <gui/modules/submenu.h>
+#include <gui/modules/dialog_ex.h>
 #include "imu_mouse.h"
-#include "air_mouse_ofw_icons.h"
+#include "views/air_mouse_view.h"
+#include <furi_hal_usb_hid.h>
+#include <storage/storage.h>
 
 #define TAG "SensorModule"
 
+#define AIRMOUSE_BT_KEYS_STORAGE_NAME ".airm_bt_hid.keys"
+
 typedef struct {
     Gui* gui;
-    ViewPort* view_port;
-    FuriMessageQueue* input_queue;
-
+    ViewDispatcher* view_dispatcher;
+    Submenu* start_submenu;
+    DialogEx* error_dialog;
+    AirMouseView* air_mouse_view;
     FuriHalSpiBusHandle* icm42688p_device;
     ICM42688P* icm42688p;
-    bool icm42688p_valid;
+    FuriHalUsbInterface* usb_mode_prev;
+    Bt* bt;
+} AirMouseApp;
+
+typedef enum {
+    AirMouseViewError,
+    AirMouseViewStartSubmenu,
+    AirMouseViewMain,
+} AirMouseViews;
+
+enum StertSubmenuIndex {
+    StartSubmenuIndexUsb,
+    StartSubmenuIndexBle,
+    StartSubmenuIndexBleReset,
+};
+
+static const ImuHidApi hid_api_usb = {
+    .mouse_move = furi_hal_hid_mouse_move,
+    .mouse_key_press = furi_hal_hid_mouse_press,
+    .mouse_key_release = furi_hal_hid_mouse_release,
+    .mouse_scroll = furi_hal_hid_mouse_scroll,
+    .report_rate_max = 200,
+};
+
+static const ImuHidApi hid_api_ble = {
+    .mouse_move = furi_hal_bt_hid_mouse_move,
+    .mouse_key_press = furi_hal_bt_hid_mouse_press,
+    .mouse_key_release = furi_hal_bt_hid_mouse_release,
+    .mouse_scroll = furi_hal_bt_hid_mouse_scroll,
+    .report_rate_max = 30,
+};
+
+static void ble_hid_remove_pairing(void) {
+    Bt* bt = furi_record_open(RECORD_BT);
+    bt_disconnect(bt);
+
+    // Wait 2nd core to update nvm storage
+    furi_delay_ms(200);
+
+    bt_keys_storage_set_storage_path(bt, APP_DATA_PATH(AIRMOUSE_BT_KEYS_STORAGE_NAME));
+    bt_forget_bonded_devices(bt);
+
+    // Wait 2nd core to update nvm storage
+    furi_delay_ms(200);
+    bt_keys_storage_set_default_path(bt);
+
+    furi_check(bt_set_profile(bt, BtProfileSerial));
+    furi_record_close(RECORD_BT);
+}
+
+static void ble_hid_connection_status_callback(BtStatus status, void* context) {
+    furi_assert(context);
+    AirMouseApp* app = context;
+    bool connected = (status == BtStatusConnected);
+    air_mouse_view_set_connected_status(app->air_mouse_view, connected);
+}
+
+static Bt* ble_hid_init(AirMouseApp* app) {
+    Bt* bt = furi_record_open(RECORD_BT);
+    bt_disconnect(bt);
 
-    ImuThread* imu_thread;
-} SensorModuleApp;
+    // Wait 2nd core to update nvm storage
+    furi_delay_ms(200);
 
-static void render_callback(Canvas* canvas, void* ctx) {
-    UNUSED(ctx);
-    canvas_clear(canvas);
-    canvas_set_color(canvas, ColorBlack);
+    bt_keys_storage_set_storage_path(bt, APP_DATA_PATH(AIRMOUSE_BT_KEYS_STORAGE_NAME));
 
-    canvas_draw_icon(canvas, 64 + 14, 8, &I_Circles_47x47);
-    canvas_draw_icon(canvas, 83 + 14, 27, &I_Left_mouse_icon_9x9);
-    canvas_draw_icon(canvas, 83 + 14, 11, &I_Right_mouse_icon_9x9);
+    furi_check(bt_set_profile(bt, BtProfileHidKeyboard));
 
-    canvas_set_font(canvas, FontPrimary);
-    canvas_draw_str(canvas, 0, 14, "Air Mouse");
-    canvas_set_font(canvas, FontSecondary);
-    canvas_draw_str(canvas, 0, 56, "Press Back to exit");
+    furi_hal_bt_start_advertising();
+    bt_set_status_changed_callback(bt, ble_hid_connection_status_callback, app);
+    return bt;
 }
 
-static void input_callback(InputEvent* input_event, void* ctx) {
-    SensorModuleApp* app = ctx;
-    furi_message_queue_put(app->input_queue, input_event, 0);
+static void ble_hid_deinit(Bt* bt) {
+    bt_set_status_changed_callback(bt, NULL, NULL);
+    bt_disconnect(bt);
+
+    // Wait 2nd core to update nvm storage
+    furi_delay_ms(200);
+    bt_keys_storage_set_default_path(bt);
+
+    furi_check(bt_set_profile(bt, BtProfileSerial));
+    furi_record_close(RECORD_BT);
 }
 
-static SensorModuleApp* sensor_module_alloc(void) {
-    SensorModuleApp* app = malloc(sizeof(SensorModuleApp));
+static uint32_t air_mouse_exit(void* context) {
+    UNUSED(context);
+    return VIEW_NONE;
+}
 
-    app->icm42688p_device = malloc(sizeof(FuriHalSpiBusHandle));
-    memcpy(app->icm42688p_device, &furi_hal_spi_bus_handle_external, sizeof(FuriHalSpiBusHandle));
-    app->icm42688p_device->cs = &gpio_ext_pc3;
-    app->icm42688p = icm42688p_alloc(app->icm42688p_device, &gpio_ext_pb2);
-    app->icm42688p_valid = icm42688p_init(app->icm42688p);
+static uint32_t air_mouse_return_to_menu(void* context) {
+    UNUSED(context);
+    return AirMouseViewStartSubmenu;
+}
+
+static void air_mouse_hid_deinit(void* context) {
+    furi_assert(context);
+    AirMouseApp* app = context;
+
+    if(app->bt) {
+        ble_hid_deinit(app->bt);
+        app->bt = NULL;
+    } else if(app->usb_mode_prev) {
+        furi_hal_usb_set_config(app->usb_mode_prev, NULL);
+        app->usb_mode_prev = NULL;
+    }
+}
 
-    app->input_queue = furi_message_queue_alloc(8, sizeof(InputEvent));
+static void air_mouse_submenu_callback(void* context, uint32_t index) {
+    furi_assert(context);
+    AirMouseApp* app = context;
+    if(index == StartSubmenuIndexUsb) {
+        app->usb_mode_prev = furi_hal_usb_get_config();
+        furi_hal_usb_unlock();
+        furi_hal_usb_set_config(&usb_hid, NULL);
+
+        air_mouse_view_set_hid_api(app->air_mouse_view, &hid_api_usb, false);
+        view_dispatcher_switch_to_view(app->view_dispatcher, AirMouseViewMain);
+    } else if(index == StartSubmenuIndexBle) {
+        app->bt = ble_hid_init(app);
+
+        air_mouse_view_set_hid_api(app->air_mouse_view, &hid_api_ble, true);
+        view_dispatcher_switch_to_view(app->view_dispatcher, AirMouseViewMain);
+    } else if(index == StartSubmenuIndexBleReset) {
+        ble_hid_remove_pairing();
+    }
+}
 
-    app->view_port = view_port_alloc();
-    view_port_draw_callback_set(app->view_port, render_callback, app);
-    view_port_input_callback_set(app->view_port, input_callback, app);
+static AirMouseApp* air_mouse_alloc(void) {
+    AirMouseApp* app = malloc(sizeof(AirMouseApp));
 
     app->gui = furi_record_open(RECORD_GUI);
-    gui_add_view_port(app->gui, app->view_port, GuiLayerFullscreen);
+    app->view_dispatcher = view_dispatcher_alloc();
+    view_dispatcher_enable_queue(app->view_dispatcher);
+    view_dispatcher_attach_to_gui(app->view_dispatcher, app->gui, ViewDispatcherTypeFullscreen);
+
+    app->air_mouse_view = air_mouse_view_alloc(air_mouse_hid_deinit, app);
+    view_set_previous_callback(
+        air_mouse_view_get_view(app->air_mouse_view), air_mouse_return_to_menu);
+    view_dispatcher_add_view(
+        app->view_dispatcher, AirMouseViewMain, air_mouse_view_get_view(app->air_mouse_view));
+
+    app->start_submenu = submenu_alloc();
+    submenu_add_item(
+        app->start_submenu, "USB Remote", StartSubmenuIndexUsb, air_mouse_submenu_callback, app);
+    submenu_add_item(
+        app->start_submenu,
+        "Bluetooth Remote",
+        StartSubmenuIndexBle,
+        air_mouse_submenu_callback,
+        app);
+    submenu_add_item(
+        app->start_submenu,
+        "Remove Pairing",
+        StartSubmenuIndexBleReset,
+        air_mouse_submenu_callback,
+        app);
+    view_set_previous_callback(submenu_get_view(app->start_submenu), air_mouse_exit);
+    view_dispatcher_add_view(
+        app->view_dispatcher, AirMouseViewStartSubmenu, submenu_get_view(app->start_submenu));
+
+    app->error_dialog = dialog_ex_alloc();
+    dialog_ex_set_header(app->error_dialog, "Sensor Module error", 63, 0, AlignCenter, AlignTop);
+    dialog_ex_set_text(app->error_dialog, "Module not conntected", 63, 30, AlignCenter, AlignTop);
+    view_set_previous_callback(dialog_ex_get_view(app->error_dialog), air_mouse_exit);
+    view_dispatcher_add_view(
+        app->view_dispatcher, AirMouseViewError, dialog_ex_get_view(app->error_dialog));
 
     return app;
 }
 
-static void sensor_module_free(SensorModuleApp* app) {
-    gui_remove_view_port(app->gui, app->view_port);
-    furi_record_close(RECORD_GUI);
-    view_port_free(app->view_port);
+static void air_mouse_free(AirMouseApp* app) {
+    view_dispatcher_remove_view(app->view_dispatcher, AirMouseViewStartSubmenu);
+    submenu_free(app->start_submenu);
+    view_dispatcher_remove_view(app->view_dispatcher, AirMouseViewError);
+    dialog_ex_free(app->error_dialog);
+    view_dispatcher_remove_view(app->view_dispatcher, AirMouseViewMain);
+    air_mouse_view_free(app->air_mouse_view);
 
-    furi_message_queue_free(app->input_queue);
+    view_dispatcher_free(app->view_dispatcher);
 
-    if(app->imu_thread) {
-        imu_stop(app->imu_thread);
-        app->imu_thread = NULL;
-    }
-
-    if(!icm42688p_deinit(app->icm42688p)) {
-        FURI_LOG_E(TAG, "Failed to deinitialize ICM42688P");
-    }
-
-    icm42688p_free(app->icm42688p);
-    free(app->icm42688p_device);
+    furi_record_close(RECORD_GUI);
 
     free(app);
 }
 
 int32_t air_mouse_app(void* arg) {
     UNUSED(arg);
-    SensorModuleApp* app = sensor_module_alloc();
-
-    if(!app->icm42688p_valid) {
-        DialogsApp* dialogs = furi_record_open(RECORD_DIALOGS);
-        DialogMessage* message = dialog_message_alloc();
-        dialog_message_set_header(message, "Sensor Module error", 63, 0, AlignCenter, AlignTop);
+    AirMouseApp* app = air_mouse_alloc();
 
-        dialog_message_set_text(message, "Module not conntected", 63, 30, AlignCenter, AlignTop);
-        dialog_message_show(dialogs, message);
-        dialog_message_free(message);
-        furi_record_close(RECORD_DIALOGS);
+    app->icm42688p_device = malloc(sizeof(FuriHalSpiBusHandle));
+    memcpy(app->icm42688p_device, &furi_hal_spi_bus_handle_external, sizeof(FuriHalSpiBusHandle));
+    app->icm42688p_device->cs = &gpio_ext_pc3;
+    app->icm42688p = icm42688p_alloc(app->icm42688p_device, &gpio_ext_pb2);
+    bool icm42688p_valid = icm42688p_init(app->icm42688p);
 
-        sensor_module_free(app);
-        return 0;
+    if(icm42688p_valid) {
+        air_mouse_view_set_device(app->air_mouse_view, app->icm42688p);
+        view_dispatcher_switch_to_view(app->view_dispatcher, AirMouseViewStartSubmenu);
+    } else {
+        view_dispatcher_switch_to_view(app->view_dispatcher, AirMouseViewError);
     }
 
-    view_port_update(app->view_port);
-    app->imu_thread = imu_start(app->icm42688p);
-
-    while(1) {
-        InputEvent input;
-        if(furi_message_queue_get(app->input_queue, &input, FuriWaitForever) == FuriStatusOk) {
-            if((input.key == InputKeyBack) && (input.type == InputTypeShort)) {
-                break;
-            } else if(input.key == InputKeyOk) {
-                if(input.type == InputTypePress) {
-                    imu_mouse_key_press(app->imu_thread, ImuMouseKeyLeft, true);
-                } else if(input.type == InputTypeRelease) {
-                    imu_mouse_key_press(app->imu_thread, ImuMouseKeyLeft, false);
-                }
-            } else if(input.key == InputKeyUp) {
-                if(input.type == InputTypePress) {
-                    imu_mouse_key_press(app->imu_thread, ImuMouseKeyRight, true);
-                } else if(input.type == InputTypeRelease) {
-                    imu_mouse_key_press(app->imu_thread, ImuMouseKeyRight, false);
-                }
-            }
-        }
+    view_dispatcher_run(app->view_dispatcher);
+
+    if(!icm42688p_deinit(app->icm42688p)) {
+        FURI_LOG_E(TAG, "Failed to deinitialize ICM42688P");
     }
 
-    sensor_module_free(app);
+    icm42688p_free(app->icm42688p);
+    free(app->icm42688p_device);
+
+    air_mouse_free(app);
     return 0;
 }