Momentum-Apps/web_crawler_i.h

/**
 * @brief      Function to allocate resources for the WebCrawlerApp.
 * @return     Pointer to the initialized WebCrawlerApp, or NULL on failure.
 */
// In web_crawler_app_alloc, after allocating and initializing 'app'
static WebCrawlerApp *web_crawler_app_alloc()
{
    WebCrawlerApp *app = (WebCrawlerApp *)malloc(sizeof(WebCrawlerApp));
    if (!app)
    {
        FURI_LOG_E(TAG, "Failed to allocate WebCrawlerApp");
        return NULL;
    }

    // Initialize the entire structure to zero to prevent undefined behavior
    memset(app, 0, sizeof(WebCrawlerApp));

    // Allocate and initialize temp_buffer and path
    app->temp_buffer_size_path = 128;
    app->temp_buffer_path = (char *)malloc(app->temp_buffer_size_path);
    if (!app->temp_buffer_path)
    {
        FURI_LOG_E(TAG, "Failed to allocate temp_buffer_path");
        free(app);
        return NULL;
    }
    app->temp_buffer_path[0] = '\0';

    // Allocate path
    app->path = (char *)malloc(app->temp_buffer_size_path);
    if (!app->path)
    {
        FURI_LOG_E(TAG, "Failed to allocate path");
        free(app->temp_buffer_path);
        free(app);
        return NULL;
    }
    app->path[0] = '\0';

    // Allocate and initialize temp_buffer_ssid
    app->temp_buffer_size_ssid = 128;
    app->temp_buffer_ssid = (char *)malloc(app->temp_buffer_size_ssid);
    if (!app->temp_buffer_ssid)
    {
        FURI_LOG_E(TAG, "Failed to allocate temp_buffer_ssid");
        free_buffers(app);
        return NULL;
    }
    app->temp_buffer_ssid[0] = '\0';

    // Allocate ssid
    app->ssid = (char *)malloc(app->temp_buffer_size_ssid);
    if (!app->ssid)
    {
        FURI_LOG_E(TAG, "Failed to allocate ssid");
        free_buffers(app);
        return NULL;
    }
    app->ssid[0] = '\0';

    // Allocate and initialize temp_buffer_password
    app->temp_buffer_size_password = 128;
    app->temp_buffer_password = (char *)malloc(app->temp_buffer_size_password);
    if (!app->temp_buffer_password)
    {
        FURI_LOG_E(TAG, "Failed to allocate temp_buffer_password");
        free_buffers(app);
        return NULL;
    }
    app->temp_buffer_password[0] = '\0';

    // Allocate password
    app->password = (char *)malloc(app->temp_buffer_size_password);
    if (!app->password)
    {
        FURI_LOG_E(TAG, "Failed to allocate password");
        free_buffers(app);
        return NULL;
    }
    app->password[0] = '\0';

    // Assign to global variable
    app_instance = app;

    // Open GUI
    Gui *gui = furi_record_open(RECORD_GUI);
    if (!gui)
    {
        FURI_LOG_E(TAG, "Failed to open GUI record");
        free_resources(app);
        return NULL;
    }

    // Allocate ViewDispatcher
    app->view_dispatcher = view_dispatcher_alloc();
    if (!app->view_dispatcher)
    {
        FURI_LOG_E(TAG, "Failed to allocate ViewDispatcher");
        furi_record_close(RECORD_GUI);
        free_resources(app);
        return NULL;
    }

    // Attach ViewDispatcher to GUI
    view_dispatcher_attach_to_gui(app->view_dispatcher, gui, ViewDispatcherTypeFullscreen);
    view_dispatcher_set_event_callback_context(app->view_dispatcher, app);

    // Allocate TextInput views
    app->text_input_path = text_input_alloc();
    if (!app->text_input_path)
    {
        free_all(app, "Failed to allocate TextInput for Path");
        return NULL;
    }

    app->text_input_ssid = text_input_alloc();
    if (!app->text_input_ssid)
    {
        free_all(app, "Failed to allocate TextInput for SSID");
        return NULL;
    }

    app->text_input_password = text_input_alloc();
    if (!app->text_input_password)
    {
        free_all(app, "Failed to allocate TextInput for Password");
        return NULL;
    }

    // Add TextInput views with unique view IDs
    view_dispatcher_add_view(
        app->view_dispatcher,
        WebCrawlerViewTextInput,
        text_input_get_view(app->text_input_path));
    view_dispatcher_add_view(
        app->view_dispatcher,
        WebCrawlerViewTextInputSSID,
        text_input_get_view(app->text_input_ssid));
    view_dispatcher_add_view(
        app->view_dispatcher,
        WebCrawlerViewTextInputPassword,
        text_input_get_view(app->text_input_password));

    // Set previous callback for TextInput views to return to Configure screen
    view_set_previous_callback(
        text_input_get_view(app->text_input_path),
        web_crawler_back_to_configure_callback);
    view_set_previous_callback(
        text_input_get_view(app->text_input_ssid),
        web_crawler_back_to_configure_callback);
    view_set_previous_callback(
        text_input_get_view(app->text_input_password),
        web_crawler_back_to_configure_callback);

    // Allocate Configuration screen
    app->variable_item_list_config = variable_item_list_alloc();
    if (!app->variable_item_list_config)
    {
        free_all(app, "Failed to allocate VariableItemList for Configuration");
        return NULL;
    }
    variable_item_list_reset(app->variable_item_list_config);

    // Add "Path" item to the configuration screen
    app->path_item = variable_item_list_add(
        app->variable_item_list_config,
        "Path",
        1,    // Number of possible values (1 for a single text value)
        NULL, // No change callback needed
        NULL  // No context needed
    );
    if (!app->path_item)
    {
        free_all(app, "Failed to add Path item to VariableItemList");
        return NULL;
    }
    variable_item_set_current_value_text(app->path_item, ""); // Initialize

    // Add "SSID" item to the configuration screen
    app->ssid_item = variable_item_list_add(
        app->variable_item_list_config,
        "SSID",
        1,    // Number of possible values (1 for a single text value)
        NULL, // No change callback needed
        NULL  // No context needed
    );
    if (!app->ssid_item)
    {
        free_all(app, "Failed to add SSID item to VariableItemList");
        return NULL;
    }
    variable_item_set_current_value_text(app->ssid_item, ""); // Initialize

    // Add "Password" item to the configuration screen
    app->password_item = variable_item_list_add(
        app->variable_item_list_config,
        "Password",
        1,    // Number of possible values (1 for a single text value)
        NULL, // No change callback needed
        NULL  // No context needed
    );
    if (!app->password_item)
    {
        free_all(app, "Failed to add Password item to VariableItemList");
        return NULL;
    }
    variable_item_set_current_value_text(app->password_item, ""); // Initialize

    // Set a single enter callback for all configuration items
    variable_item_list_set_enter_callback(
        app->variable_item_list_config,
        web_crawler_config_enter_callback,
        app);

    // Set previous callback for configuration screen
    view_set_previous_callback(
        variable_item_list_get_view(app->variable_item_list_config),
        web_crawler_back_to_main_callback);

    // Add Configuration view to ViewDispatcher
    view_dispatcher_add_view(
        app->view_dispatcher,
        WebCrawlerViewConfigure,
        variable_item_list_get_view(app->variable_item_list_config));

    // Allocate Submenu view
    app->submenu = submenu_alloc();
    if (!app->submenu)
    {
        free_all(app, "Failed to allocate Submenu");
        return NULL;
    }

    // Add items to Submenu
    submenu_add_item(app->submenu, "Run", WebCrawlerSubmenuIndexRun, web_crawler_submenu_callback, app);
    submenu_add_item(app->submenu, "About", WebCrawlerSubmenuIndexAbout, web_crawler_submenu_callback, app);
    submenu_add_item(app->submenu, "Configure", WebCrawlerSubmenuIndexSetPath, web_crawler_submenu_callback, app);

    // Set previous callback for Submenu
    view_set_previous_callback(submenu_get_view(app->submenu), web_crawler_exit_app_callback);

    // Initialize UART
    uart_init();

    // Add Submenu view to ViewDispatcher
    view_dispatcher_add_view(
        app->view_dispatcher,
        WebCrawlerViewSubmenu,
        submenu_get_view(app->submenu));

    // Allocate Main view
    app->view_main = view_alloc();
    if (!app->view_main)
    {
        free_all(app, "Failed to allocate Main view");
        return NULL;
    }
    view_set_draw_callback(app->view_main, web_crawler_view_draw_callback);
    view_set_input_callback(app->view_main, web_crawler_view_input_callback);
    view_set_previous_callback(app->view_main, web_crawler_back_to_main_callback);

    // Allocate and initialize the main view's model
    view_allocate_model(app->view_main, ViewModelTypeLockFree, sizeof(WebCrawlerMainModel));
    WebCrawlerMainModel *main_model = (WebCrawlerMainModel *)view_get_model(app->view_main);
    if (main_model)
    {
        strncpy(main_model->path, "", sizeof(main_model->path) - 1);         // Initialize to empty
        strncpy(main_model->ssid, "", sizeof(main_model->ssid) - 1);         // Initialize to empty
        strncpy(main_model->password, "", sizeof(main_model->password) - 1); // Initialize to empty
        main_model->path[sizeof(main_model->path) - 1] = '\0';
        main_model->ssid[sizeof(main_model->ssid) - 1] = '\0';
        main_model->password[sizeof(main_model->password) - 1] = '\0';
    }
    else
    {
        free_all(app, "Failed to allocate main view model");
        return NULL;
    }

    // Add Main view to ViewDispatcher
    view_dispatcher_add_view(app->view_dispatcher, WebCrawlerViewMain, app->view_main);

    // Allocate About view
    app->widget_about = widget_alloc();
    if (!app->widget_about)
    {
        free_all(app, "Failed to allocate About widget");
        return NULL;
    }

    // Add text to About widget
    widget_add_text_scroll_element(
        app->widget_about,
        0,
        0,
        128,
        64,
        "Web Crawler App\n---\nThis is a web crawler app for Flipper Zero.\n---\nVisit github.com/jblanked for more details.\n---\nPress BACK to return.");

    // Load settings
    if (!load_settings(app->path, app->temp_buffer_size_path, app->ssid, app->temp_buffer_size_ssid, app->password, app->temp_buffer_size_password, app))
    {
        FURI_LOG_E(TAG, "Failed to load settings");
    }
    else
    {
        // Update the main view's model
        WebCrawlerMainModel *main_model = (WebCrawlerMainModel *)view_get_model(app->view_main);
        if (main_model)
        {
            strncpy(main_model->path, app->path, sizeof(main_model->path) - 1);
            main_model->path[sizeof(main_model->path) - 1] = '\0';
            strncpy(main_model->ssid, app->ssid, sizeof(main_model->ssid) - 1);
            main_model->ssid[sizeof(main_model->ssid) - 1] = '\0';
            strncpy(main_model->password, app->password, sizeof(main_model->password) - 1);
            main_model->password[sizeof(main_model->password) - 1] = '\0';
        }
    }

    // Set previous callback for About view
    view_set_previous_callback(
        widget_get_view(app->widget_about),
        web_crawler_back_to_main_callback);

    // Add About view to ViewDispatcher
    view_dispatcher_add_view(
        app->view_dispatcher,
        WebCrawlerViewAbout,
        widget_get_view(app->widget_about));

    // Start with the Submenu view
    view_dispatcher_switch_to_view(app->view_dispatcher, WebCrawlerViewSubmenu);

    return app;
}