Momentum-Apps/alloc/web_crawler_alloc.c

#include <alloc/web_crawler_alloc.h>

/**
 * @brief      Function to allocate resources for the WebCrawlerApp.
 * @return     Pointer to the initialized WebCrawlerApp, or NULL on failure.
 */
WebCrawlerApp *web_crawler_app_alloc()
{
    // Initialize the entire structure to zero to prevent undefined behavior
    WebCrawlerApp *app = (WebCrawlerApp *)malloc(sizeof(WebCrawlerApp));

    // Open GUI
    Gui *gui = furi_record_open(RECORD_GUI);

    // Initialize UART with the correct callback
    if (!flipper_http_init(flipper_http_rx_callback, app))
    {
        FURI_LOG_E(TAG, "Failed to initialize UART");
        return NULL;
    }

    // Allocate ViewDispatcher
    if (!easy_flipper_set_view_dispatcher(&app->view_dispatcher, gui, app))
    {
        return NULL;
    }
    view_dispatcher_set_custom_event_callback(app->view_dispatcher, web_crawler_custom_event_callback);

    // Allocate and initialize temp_buffer and path
    app->temp_buffer_size_path = 128;
    app->temp_buffer_size_ssid = 64;
    app->temp_buffer_size_password = 64;
    app->temp_buffer_size_file_type = 16;
    app->temp_buffer_size_file_rename = 128;
    app->temp_buffer_size_http_method = 16;
    app->temp_buffer_size_headers = 256;
    app->temp_buffer_size_payload = 256;
    if (!easy_flipper_set_buffer(&app->temp_buffer_path, app->temp_buffer_size_path))
    {
        return NULL;
    }
    if (!easy_flipper_set_buffer(&app->path, app->temp_buffer_size_path))
    {
        return NULL;
    }
    if (!easy_flipper_set_buffer(&app->temp_buffer_ssid, app->temp_buffer_size_ssid))
    {
        return NULL;
    }
    if (!easy_flipper_set_buffer(&app->ssid, app->temp_buffer_size_ssid))
    {
        return NULL;
    }
    if (!easy_flipper_set_buffer(&app->temp_buffer_password, app->temp_buffer_size_password))
    {
        return NULL;
    }
    if (!easy_flipper_set_buffer(&app->password, app->temp_buffer_size_password))
    {
        return NULL;
    }
    if (!easy_flipper_set_buffer(&app->temp_buffer_file_type, app->temp_buffer_size_file_type))
    {
        return NULL;
    }
    if (!easy_flipper_set_buffer(&app->file_type, app->temp_buffer_size_file_type))
    {
        return NULL;
    }
    if (!easy_flipper_set_buffer(&app->temp_buffer_file_rename, app->temp_buffer_size_file_rename))
    {
        return NULL;
    }
    if (!easy_flipper_set_buffer(&app->file_rename, app->temp_buffer_size_file_rename))
    {
        return NULL;
    }
    if (!easy_flipper_set_buffer(&app->temp_buffer_http_method, app->temp_buffer_size_http_method))
    {
        return NULL;
    }
    if (!easy_flipper_set_buffer(&app->http_method, app->temp_buffer_size_http_method))
    {
        return NULL;
    }
    if (!easy_flipper_set_buffer(&app->temp_buffer_headers, app->temp_buffer_size_headers))
    {
        return NULL;
    }
    if (!easy_flipper_set_buffer(&app->headers, app->temp_buffer_size_headers))
    {
        return NULL;
    }
    if (!easy_flipper_set_buffer(&app->temp_buffer_payload, app->temp_buffer_size_payload))
    {
        return NULL;
    }
    if (!easy_flipper_set_buffer(&app->payload, app->temp_buffer_size_payload))
    {
        return NULL;
    }

    // Allocate TextInput views
    if (!easy_flipper_set_uart_text_input(&app->text_input_path, WebCrawlerViewTextInput, "Enter URL", app->temp_buffer_path, app->temp_buffer_size_path, NULL, web_crawler_back_to_request_callback, &app->view_dispatcher, app))
    {
        return NULL;
    }
    if (!easy_flipper_set_uart_text_input(&app->text_input_ssid, WebCrawlerViewTextInputSSID, "Enter SSID", app->temp_buffer_ssid, app->temp_buffer_size_ssid, NULL, web_crawler_back_to_wifi_callback, &app->view_dispatcher, app))
    {
        return NULL;
    }
    if (!easy_flipper_set_uart_text_input(&app->text_input_password, WebCrawlerViewTextInputPassword, "Enter Password", app->temp_buffer_password, app->temp_buffer_size_password, NULL, web_crawler_back_to_wifi_callback, &app->view_dispatcher, app))
    {
        return NULL;
    }
    if (!easy_flipper_set_uart_text_input(&app->text_input_file_type, WebCrawlerViewTextInputFileType, "Enter File Type", app->temp_buffer_file_type, app->temp_buffer_size_file_type, NULL, web_crawler_back_to_file_callback, &app->view_dispatcher, app))
    {
        return NULL;
    }
    if (!easy_flipper_set_uart_text_input(&app->text_input_file_rename, WebCrawlerViewTextInputFileRename, "Enter File Rename", app->temp_buffer_file_rename, app->temp_buffer_size_file_rename, NULL, web_crawler_back_to_file_callback, &app->view_dispatcher, app))
    {
        return NULL;
    }
    if (!easy_flipper_set_uart_text_input(&app->text_input_headers, WebCrawlerViewTextInputHeaders, "Enter Headers", app->temp_buffer_headers, app->temp_buffer_size_headers, NULL, web_crawler_back_to_request_callback, &app->view_dispatcher, app))
    {
        return NULL;
    }
    if (!easy_flipper_set_uart_text_input(&app->text_input_payload, WebCrawlerViewTextInputPayload, "Enter Payload", app->temp_buffer_payload, app->temp_buffer_size_payload, NULL, web_crawler_back_to_request_callback, &app->view_dispatcher, app))
    {
        return NULL;
    }

    // Allocate VariableItemList views
    if (!easy_flipper_set_variable_item_list(&app->variable_item_list_wifi, WebCrawlerViewVariableItemListWifi, web_crawler_wifi_enter_callback, web_crawler_back_to_configure_callback, &app->view_dispatcher, app))
    {
        return NULL;
    }
    if (!easy_flipper_set_variable_item_list(&app->variable_item_list_file, WebCrawlerViewVariableItemListFile, web_crawler_file_enter_callback, web_crawler_back_to_configure_callback, &app->view_dispatcher, app))
    {
        return NULL;
    }
    if (!easy_flipper_set_variable_item_list(&app->variable_item_list_request, WebCrawlerViewVariableItemListRequest, web_crawler_request_enter_callback, web_crawler_back_to_configure_callback, &app->view_dispatcher, app))
    {
        return NULL;
    }

    // set variable items
    app->path_item = variable_item_list_add(app->variable_item_list_request, "Path", 0, NULL, NULL);
    app->http_method_item = variable_item_list_add(app->variable_item_list_request, "HTTP Method", 5, web_crawler_http_method_change, app);
    app->headers_item = variable_item_list_add(app->variable_item_list_request, "Headers", 0, NULL, NULL);
    app->payload_item = variable_item_list_add(app->variable_item_list_request, "Payload", 0, NULL, NULL);
    //
    app->ssid_item = variable_item_list_add(app->variable_item_list_wifi, "SSID", 0, NULL, NULL);         // index 0
    app->password_item = variable_item_list_add(app->variable_item_list_wifi, "Password", 0, NULL, NULL); // index 1
    //
    app->file_read_item = variable_item_list_add(app->variable_item_list_file, "Read File", 0, NULL, NULL);     // index 0
    app->file_type_item = variable_item_list_add(app->variable_item_list_file, "Set File Type", 0, NULL, NULL); // index 1
    app->file_rename_item = variable_item_list_add(app->variable_item_list_file, "Rename File", 0, NULL, NULL); // index 2
    app->file_delete_item = variable_item_list_add(app->variable_item_list_file, "Delete File", 0, NULL, NULL); // index 3

    if (!app->ssid_item || !app->password_item || !app->file_type_item || !app->file_rename_item || !app->path_item || !app->file_read_item || !app->file_delete_item || !app->http_method_item || !app->headers_item || !app->payload_item)
    {
        free_all(app, "Failed to add items to VariableItemList");
        return NULL;
    }

    variable_item_set_current_value_text(app->path_item, "");        // Initialize
    variable_item_set_current_value_text(app->http_method_item, ""); // Initialize
    variable_item_set_current_value_text(app->headers_item, "");     // Initialize
    variable_item_set_current_value_text(app->payload_item, "");     // Initialize
    variable_item_set_current_value_text(app->ssid_item, "");        // Initialize
    variable_item_set_current_value_text(app->password_item, "");    // Initialize
    variable_item_set_current_value_text(app->file_type_item, "");   // Initialize
    variable_item_set_current_value_text(app->file_rename_item, ""); // Initialize
    variable_item_set_current_value_text(app->file_read_item, "");   // Initialize
    variable_item_set_current_value_text(app->file_delete_item, ""); // Initialize

    // Allocate Submenu views
    if (!easy_flipper_set_submenu(&app->submenu_main, WebCrawlerViewSubmenuMain, "Web Crawler v0.8", web_crawler_exit_app_callback, &app->view_dispatcher))
    {
        return NULL;
    }
    if (!easy_flipper_set_submenu(&app->submenu_config, WebCrawlerViewSubmenuConfig, "Settings", web_crawler_back_to_main_callback, &app->view_dispatcher))
    {
        return NULL;
    }

    // Add Submenu items
    submenu_add_item(app->submenu_main, "Run", WebCrawlerSubmenuIndexRun, web_crawler_submenu_callback, app);
    submenu_add_item(app->submenu_main, "About", WebCrawlerSubmenuIndexAbout, web_crawler_submenu_callback, app);
    submenu_add_item(app->submenu_main, "Settings", WebCrawlerSubmenuIndexConfig, web_crawler_submenu_callback, app);

    submenu_add_item(app->submenu_config, "WiFi", WebCrawlerSubmenuIndexWifi, web_crawler_submenu_callback, app);
    submenu_add_item(app->submenu_config, "File", WebCrawlerSubmenuIndexFile, web_crawler_submenu_callback, app);
    submenu_add_item(app->submenu_config, "Request", WebCrawlerSubmenuIndexRequest, web_crawler_submenu_callback, app);

    // Main view
    if (!easy_flipper_set_view(&app->view_loader, WebCrawlerViewLoader, web_crawler_loader_draw_callback, NULL, web_crawler_back_to_main_callback, &app->view_dispatcher, app))
    {
        return NULL;
    }
    web_crawler_loader_init(app->view_loader);

    //-- WIDGET ABOUT VIEW --
    if (!easy_flipper_set_widget(&app->widget_about, WebCrawlerViewAbout, "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.", web_crawler_back_to_main_callback, &app->view_dispatcher))
    {
        return NULL;
    }
    if (!easy_flipper_set_widget(&app->widget_file_read, WebCrawlerViewFileRead, "Data will be displayed here.", web_crawler_back_to_file_callback, &app->view_dispatcher))
    {
        return NULL;
    }
    if (!easy_flipper_set_widget(&app->widget_file_delete, WebCrawlerViewFileDelete, "File deleted.", web_crawler_back_to_file_callback, &app->view_dispatcher))
    {
        return NULL;
    }
    if (!easy_flipper_set_widget(&app->widget_result, WebCrawlerViewWidgetResult, "Error, try again.", web_crawler_back_to_main_callback, &app->view_dispatcher))
    {
        return NULL;
    }

    // Load Settings and Update Views
    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->file_rename,
            app->temp_buffer_size_file_rename,
            app->file_type,
            app->temp_buffer_size_file_type,
            app->http_method,
            app->temp_buffer_size_http_method,
            app->headers,
            app->temp_buffer_size_headers,
            app->payload,
            app->temp_buffer_size_payload,
            app))
    {
        FURI_LOG_E(TAG, "Failed to load settings");
    }
    else
    {
        // Update the configuration items based on loaded settings
        if (app->path_item)
        {
            variable_item_set_current_value_text(app->path_item, app->path);
        }
        else
        {
            variable_item_set_current_value_text(app->path_item, "https://httpbin.org/get"); // Initialize
        }

        if (app->ssid_item)
        {
            variable_item_set_current_value_text(app->ssid_item, app->ssid);
        }
        else
        {
            variable_item_set_current_value_text(app->ssid_item, ""); // Initialize
        }

        if (app->file_type_item)
        {
            variable_item_set_current_value_text(app->file_type_item, app->file_type);
        }
        else
        {
            variable_item_set_current_value_text(app->file_type_item, ".txt"); // Initialize
        }

        if (app->file_rename_item)
        {
            variable_item_set_current_value_text(app->file_rename_item, app->file_rename);
        }
        else
        {
            variable_item_set_current_value_text(app->file_rename_item, "received_data"); // Initialize
        }

        if (app->http_method_item)
        {
            variable_item_set_current_value_text(app->http_method_item, app->http_method);
        }
        else
        {
            variable_item_set_current_value_text(app->http_method_item, "GET"); // Initialize
        }

        if (app->headers_item)
        {
            variable_item_set_current_value_text(app->headers_item, app->headers);
        }
        else
        {
            variable_item_set_current_value_text(app->headers_item, "{\n\t\"Content-Type\": \"application/json\"\n}"); // Initialize
        }

        if (app->payload_item)
        {
            variable_item_set_current_value_text(app->payload_item, app->payload);
        }
        else
        {
            variable_item_set_current_value_text(app->payload_item, "{\n\t\"key\": \"value\"\n}"); // Initialize
        }

        // set the file path for fhttp.file_path
        char file_path[128];
        snprintf(file_path, sizeof(file_path), "%s%s%s", RECEIVED_DATA_PATH, app->file_rename, app->file_type);
        snprintf(fhttp.file_path, sizeof(fhttp.file_path), "%s", file_path);

        // update temp buffers
        strncpy(app->temp_buffer_path, app->path, app->temp_buffer_size_path - 1);
        app->temp_buffer_path[app->temp_buffer_size_path - 1] = '\0';
        strncpy(app->temp_buffer_ssid, app->ssid, app->temp_buffer_size_ssid - 1);
        app->temp_buffer_ssid[app->temp_buffer_size_ssid - 1] = '\0';
        strncpy(app->temp_buffer_file_type, app->file_type, app->temp_buffer_size_file_type - 1);
        app->temp_buffer_file_type[app->temp_buffer_size_file_type - 1] = '\0';
        strncpy(app->temp_buffer_file_rename, app->file_rename, app->temp_buffer_size_file_rename - 1);
        app->temp_buffer_file_rename[app->temp_buffer_size_file_rename - 1] = '\0';
        strncpy(app->temp_buffer_http_method, app->http_method, app->temp_buffer_size_http_method - 1);
        app->temp_buffer_http_method[app->temp_buffer_size_http_method - 1] = '\0';
        strncpy(app->temp_buffer_headers, app->headers, app->temp_buffer_size_headers - 1);
        app->temp_buffer_headers[app->temp_buffer_size_headers - 1] = '\0';
        strncpy(app->temp_buffer_payload, app->payload, app->temp_buffer_size_payload - 1);
        app->temp_buffer_payload[app->temp_buffer_size_payload - 1] = '\0';

        // Password handling can be omitted for security or handled securely
    }

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

    return app;
}