Momentum-Firmware/applications/main/bad_ble/bad_ble_app.c

#include "bad_ble_app_i.h"
#include "bad_ble_settings_filename.h"
#include <furi.h>
#include <furi_hal.h>
#include <storage/storage.h>
#include <lib/toolbox/path.h>

#include <bt/bt_service/bt_i.h>
#include <bt/bt_service/bt.h>

#define BAD_BLE_SETTINGS_PATH BAD_BLE_APP_BASE_FOLDER "/" BAD_BLE_SETTINGS_FILE_NAME

static bool bad_ble_app_custom_event_callback(void* context, uint32_t event) {
    furi_assert(context);
    BadBleApp* app = context;
    return scene_manager_handle_custom_event(app->scene_manager, event);
}

static bool bad_ble_app_back_event_callback(void* context) {
    furi_assert(context);
    BadBleApp* app = context;
    return scene_manager_handle_back_event(app->scene_manager);
}

static void bad_ble_app_tick_event_callback(void* context) {
    furi_assert(context);
    BadBleApp* app = context;
    scene_manager_handle_tick_event(app->scene_manager);
}

static void bad_ble_load_settings(BadBleApp* app) {
    File* settings_file = storage_file_alloc(furi_record_open(RECORD_STORAGE));
    if(storage_file_open(settings_file, BAD_BLE_SETTINGS_PATH, FSAM_READ, FSOM_OPEN_EXISTING)) {
        char chr;
        while((storage_file_read(settings_file, &chr, 1) == 1) &&
              !storage_file_eof(settings_file) && !isspace(chr)) {
            furi_string_push_back(app->keyboard_layout, chr);
        }
    }
    storage_file_close(settings_file);
    storage_file_free(settings_file);
}

static void bad_ble_save_settings(BadBleApp* app) {
    File* settings_file = storage_file_alloc(furi_record_open(RECORD_STORAGE));
    if(storage_file_open(settings_file, BAD_BLE_SETTINGS_PATH, FSAM_WRITE, FSOM_OPEN_ALWAYS)) {
        storage_file_write(
            settings_file,
            furi_string_get_cstr(app->keyboard_layout),
            furi_string_size(app->keyboard_layout));
        storage_file_write(settings_file, "\n", 1);
    }
    storage_file_close(settings_file);
    storage_file_free(settings_file);
}

void bad_ble_set_name(BadBleApp* app, const char* fmt, ...) {
    furi_assert(app);

    va_list args;
    va_start(args, fmt);

    vsnprintf(app->name, BAD_BLE_ADV_NAME_MAX_LEN, fmt, args);

    va_end(args);
}

BadBleApp* bad_ble_app_alloc(char* arg) {
    BadBleApp* app = malloc(sizeof(BadBleApp));

    app->bad_ble_script = NULL;

    app->file_path = furi_string_alloc();
    app->keyboard_layout = furi_string_alloc();
    if(arg && strlen(arg)) {
        furi_string_set(app->file_path, arg);
    }

    bad_ble_load_settings(app);

    app->gui = furi_record_open(RECORD_GUI);
    app->notifications = furi_record_open(RECORD_NOTIFICATION);
    app->dialogs = furi_record_open(RECORD_DIALOGS);

    app->view_dispatcher = view_dispatcher_alloc();
    view_dispatcher_enable_queue(app->view_dispatcher);

    app->scene_manager = scene_manager_alloc(&bad_ble_scene_handlers, app);

    view_dispatcher_set_event_callback_context(app->view_dispatcher, app);
    view_dispatcher_set_tick_event_callback(
        app->view_dispatcher, bad_ble_app_tick_event_callback, 500);
    view_dispatcher_set_custom_event_callback(
        app->view_dispatcher, bad_ble_app_custom_event_callback);
    view_dispatcher_set_navigation_event_callback(
        app->view_dispatcher, bad_ble_app_back_event_callback);

    Bt* bt = furi_record_open(RECORD_BT);
    app->bt = bt;
    const char* adv_name = bt_get_profile_adv_name(bt);
    memcpy(app->name, adv_name, BAD_BLE_ADV_NAME_MAX_LEN);

    const uint8_t* mac_addr = bt_get_profile_mac_address(bt);
    memcpy(app->mac, mac_addr, BAD_BLE_MAC_ADDRESS_LEN);

    // Custom Widget
    app->widget = widget_alloc();
    view_dispatcher_add_view(
        app->view_dispatcher, BadBleAppViewError, widget_get_view(app->widget));

    app->submenu = submenu_alloc();
    view_dispatcher_add_view(
        app->view_dispatcher, BadBleAppViewConfig, submenu_get_view(app->submenu));

    app->bad_ble_view = bad_ble_alloc();
    view_dispatcher_add_view(
        app->view_dispatcher, BadBleAppViewWork, bad_ble_get_view(app->bad_ble_view));

    app->text_input = text_input_alloc();
    view_dispatcher_add_view(
        app->view_dispatcher, BadBleAppViewConfigName, text_input_get_view(app->text_input));

    app->byte_input = byte_input_alloc();
    view_dispatcher_add_view(
        app->view_dispatcher, BadBleAppViewConfigMac, byte_input_get_view(app->byte_input));

    view_dispatcher_attach_to_gui(app->view_dispatcher, app->gui, ViewDispatcherTypeFullscreen);

    if(!furi_string_empty(app->file_path)) {
        app->bad_ble_script = bad_ble_script_open(app->file_path, bt);
        bad_ble_script_set_keyboard_layout(app->bad_ble_script, app->keyboard_layout);
        scene_manager_next_scene(app->scene_manager, BadBleSceneWork);
    } else {
        furi_string_set(app->file_path, BAD_BLE_APP_BASE_FOLDER);
        scene_manager_next_scene(app->scene_manager, BadBleSceneFileSelect);
    }

    return app;
}

void bad_ble_app_free(BadBleApp* app) {
    furi_assert(app);

    if(app->bad_ble_script) {
        bad_ble_script_close(app->bad_ble_script);
        app->bad_ble_script = NULL;
    }

    // Views
    view_dispatcher_remove_view(app->view_dispatcher, BadBleAppViewWork);
    bad_ble_free(app->bad_ble_view);

    // Custom Widget
    view_dispatcher_remove_view(app->view_dispatcher, BadBleAppViewError);
    widget_free(app->widget);

    // Submenu
    view_dispatcher_remove_view(app->view_dispatcher, BadBleAppViewConfig);
    submenu_free(app->submenu);

    // Text Input
    view_dispatcher_remove_view(app->view_dispatcher, BadBleAppViewConfigName);
    text_input_free(app->text_input);

    // Byte Input
    view_dispatcher_remove_view(app->view_dispatcher, BadBleAppViewConfigMac);
    byte_input_free(app->byte_input);

    // View dispatcher
    view_dispatcher_free(app->view_dispatcher);
    scene_manager_free(app->scene_manager);

    // Close records
    furi_record_close(RECORD_GUI);
    furi_record_close(RECORD_NOTIFICATION);
    furi_record_close(RECORD_DIALOGS);
    furi_record_close(RECORD_BT);

    bad_ble_save_settings(app);

    furi_string_free(app->file_path);
    furi_string_free(app->keyboard_layout);

    free(app);
}

int32_t bad_ble_app(void* p) {
    BadBleApp* bad_ble_app = bad_ble_app_alloc((char*)p);

    view_dispatcher_run(bad_ble_app->view_dispatcher);

    bad_ble_app_free(bad_ble_app);
    return 0;
}