Momentum-Apps/flip_wifi/callback/flip_wifi_callback.c

#include <callback/flip_wifi_callback.h>

char* ssid_list[64];
uint32_t ssid_index = 0;

void flip_wifi_redraw_submenu_saved(FlipWiFiApp* app) {
    // re draw the saved submenu
    submenu_reset(app->submenu_wifi_saved);
    submenu_set_header(app->submenu_wifi_saved, "Saved APs");
    submenu_add_item(
        app->submenu_wifi_saved,
        "[Add Network]",
        FlipWiFiSubmenuIndexWiFiSavedAddSSID,
        callback_submenu_choices,
        app);
    for(uint32_t i = 0; i < app->wifi_playlist.count; i++) {
        submenu_add_item(
            app->submenu_wifi_saved,
            app->wifi_playlist.ssids[i],
            FlipWiFiSubmenuIndexWiFiSavedStart + i,
            callback_submenu_choices,
            app);
    }
}

uint32_t callback_to_submenu_main(void* context) {
    if(!context) {
        FURI_LOG_E(TAG, "Context is NULL");
        return VIEW_NONE;
    }
    UNUSED(context);
    ssid_index = 0;
    return FlipWiFiViewSubmenuMain;
}
uint32_t callback_to_submenu_scan(void* context) {
    if(!context) {
        FURI_LOG_E(TAG, "Context is NULL");
        return VIEW_NONE;
    }
    UNUSED(context);
    ssid_index = 0;
    return FlipWiFiViewSubmenuScan;
}
uint32_t callback_to_submenu_saved(void* context) {
    if(!context) {
        FURI_LOG_E(TAG, "Context is NULL");
        return VIEW_NONE;
    }
    UNUSED(context);
    ssid_index = 0;
    return FlipWiFiViewSubmenuSaved;
}
void popup_callback_saved(void* context) {
    FlipWiFiApp* app = (FlipWiFiApp*)context;
    if(!app) {
        FURI_LOG_E(TAG, "HelloWorldApp is NULL");
        return;
    }
    view_dispatcher_switch_to_view(app->view_dispatcher, FlipWiFiViewSubmenuSaved);
}
void popup_callback_main(void* context) {
    FlipWiFiApp* app = (FlipWiFiApp*)context;
    if(!app) {
        FURI_LOG_E(TAG, "HelloWorldApp is NULL");
        return;
    }
    view_dispatcher_switch_to_view(app->view_dispatcher, FlipWiFiViewSubmenuMain);
}

// Callback for drawing the main screen
void flip_wifi_view_draw_callback_scan(Canvas* canvas, void* model) {
    UNUSED(model);
    canvas_clear(canvas);
    canvas_set_font(canvas, FontPrimary);
    canvas_draw_str(canvas, 0, 10, ssid_list[ssid_index]);
    canvas_draw_icon(canvas, 0, 53, &I_ButtonBACK_10x8);
    canvas_draw_str_aligned(canvas, 12, 54, AlignLeft, AlignTop, "Back");
    canvas_draw_icon(canvas, 96, 53, &I_ButtonRight_4x7);
    canvas_draw_str_aligned(canvas, 103, 54, AlignLeft, AlignTop, "Add");
}
void flip_wifi_view_draw_callback_saved(Canvas* canvas, void* model) {
    UNUSED(model);
    canvas_clear(canvas);
    canvas_set_font(canvas, FontPrimary);
    canvas_draw_str(canvas, 0, 10, app_instance->wifi_playlist.ssids[ssid_index]);
    canvas_set_font(canvas, FontSecondary);
    char password[64];
    snprintf(
        password, sizeof(password), "Pass: %s", app_instance->wifi_playlist.passwords[ssid_index]);
    canvas_draw_str(canvas, 0, 20, password);
    canvas_draw_icon(canvas, 0, 54, &I_ButtonLeft_4x7);
    canvas_draw_str_aligned(canvas, 7, 54, AlignLeft, AlignTop, "Delete");
    canvas_draw_icon(canvas, 37, 53, &I_ButtonBACK_10x8);
    canvas_draw_str_aligned(canvas, 49, 54, AlignLeft, AlignTop, "Back");
    canvas_draw_icon(canvas, 73, 54, &I_ButtonOK_7x7);
    canvas_draw_str_aligned(canvas, 81, 54, AlignLeft, AlignTop, "Set");
    canvas_draw_icon(canvas, 100, 54, &I_ButtonRight_4x7);
    canvas_draw_str_aligned(canvas, 107, 54, AlignLeft, AlignTop, "Edit");
}

// Input callback for the view (async input handling)
bool flip_wifi_view_input_callback_scan(InputEvent* event, void* context) {
    FlipWiFiApp* app = (FlipWiFiApp*)context;
    if(event->type == InputTypePress && event->key == InputKeyRight) {
        // switch to text input to set password
        view_dispatcher_switch_to_view(app->view_dispatcher, FlipWiFiViewTextInputScan);
        return true;
    }
    return false;
}
// Input callback for the view (async input handling)
bool flip_wifi_view_input_callback_saved(InputEvent* event, void* context) {
    FlipWiFiApp* app = (FlipWiFiApp*)context;
    if(!app) {
        FURI_LOG_E(TAG, "FlipWiFiApp is NULL");
        return false;
    }
    if(event->type == InputTypePress && event->key == InputKeyRight) {
        // set text input buffer as the selected password
        strncpy(
            app->uart_text_input_temp_buffer_password_saved,
            app->wifi_playlist.passwords[ssid_index],
            app->uart_text_input_buffer_size_password_saved);
        // switch to text input to set password
        view_dispatcher_switch_to_view(app->view_dispatcher, FlipWiFiViewTextInputSaved);
        return true;
    } else if(event->type == InputTypePress && event->key == InputKeyOk) {
        // save the settings
        if(app->wifi_playlist.ssids[ssid_index] == NULL ||
           app->wifi_playlist.passwords[ssid_index] == NULL) {
            return false;
        }
        save_settings(
            app->wifi_playlist.ssids[ssid_index], app->wifi_playlist.passwords[ssid_index]);

        flipper_http_save_wifi(
            app->wifi_playlist.ssids[ssid_index], app->wifi_playlist.passwords[ssid_index]);

        flipper_http_connect_wifi();

        popup_set_header(app->popup, "[SUCCESS]", 0, 0, AlignLeft, AlignTop);
        popup_set_text(
            app->popup,
            "All FlipperHTTP apps will now\nuse the selected network.",
            0,
            40,
            AlignLeft,
            AlignTop);
        view_set_previous_callback(popup_get_view(app->popup), callback_to_submenu_saved);
        popup_set_callback(app->popup, popup_callback_saved);

        view_dispatcher_switch_to_view(app->view_dispatcher, FlipWiFiViewPopup);
        return true;
    } else if(event->type == InputTypePress && event->key == InputKeyLeft) {
        // delete the selected ssid and password
        free(app->wifi_playlist.ssids[ssid_index]);
        free(app->wifi_playlist.passwords[ssid_index]);
        free(ssid_list[ssid_index]);
        // shift the remaining ssids and passwords
        for(uint32_t i = ssid_index; i < app->wifi_playlist.count - 1; i++) {
            app->wifi_playlist.ssids[i] = app->wifi_playlist.ssids[i + 1];
            app->wifi_playlist.passwords[i] = app->wifi_playlist.passwords[i + 1];
            ssid_list[i] = ssid_list[i + 1];
        }
        app->wifi_playlist.count--;

        // save the playlist to storage
        save_playlist(&app->wifi_playlist);

        // re draw the saved submenu
        flip_wifi_redraw_submenu_saved(app);
        // switch back to the saved view
        view_dispatcher_switch_to_view(app->view_dispatcher, FlipWiFiViewSubmenuSaved);
        return true;
    }
    return false;
}

// Function to trim leading and trailing whitespace
// Returns the trimmed start pointer and updates the length
char* trim_whitespace(char* start, size_t* length) {
    // Trim leading whitespace
    while(*length > 0 && isspace((unsigned char)*start)) {
        start++;
        (*length)--;
    }

    // Trim trailing whitespace
    while(*length > 0 && isspace((unsigned char)start[*length - 1])) {
        (*length)--;
    }

    return start;
}

bool flip_wifi_handle_scan(FlipWiFiApp* app) {
    // load the received data from the saved file
    FuriString* scan_data = flipper_http_load_from_file(fhttp.file_path);
    if(scan_data == NULL) {
        FURI_LOG_E(TAG, "Failed to load received data from file.");
        fhttp.state = ISSUE;
        return false;
    }
    char* data_cstr = (char*)furi_string_get_cstr(scan_data);
    if(data_cstr == NULL) {
        FURI_LOG_E(TAG, "Failed to get C-string from FuriString.");
        furi_string_free(scan_data);
        fhttp.state = ISSUE;
        free(data_cstr);
        return false;
    }

    uint32_t ssid_count = 0;

    char* current_position = data_cstr;
    char* next_comma = NULL;

    // Manually split the string on commas
    while((next_comma = strchr(current_position, ',')) != NULL) {
        // Calculate length of the SSID
        size_t ssid_length = next_comma - current_position;

        // Trim leading and trailing whitespace
        size_t trimmed_length = ssid_length;
        char* trim_start = trim_whitespace(current_position, &trimmed_length);

        // Handle empty SSIDs resulting from consecutive commas
        if(trimmed_length == 0) {
            current_position = next_comma + 1; // Move past the comma
            continue;
        }

        // Allocate memory for the SSID and copy it
        ssid_list[ssid_count] = malloc(trimmed_length + 1);
        if(ssid_list[ssid_count] == NULL) {
            FURI_LOG_E(TAG, "Memory allocation failed");
            free(data_cstr);
            furi_string_free(scan_data);
            return false;
        }
        strncpy(ssid_list[ssid_count], trim_start, trimmed_length);
        ssid_list[ssid_count][trimmed_length] = '\0'; // Null-terminate the string

        ssid_count++;
        if(ssid_count >= MAX_WIFI_NETWORKS) {
            FURI_LOG_E(TAG, "Maximum SSID limit reached");
            break;
        }

        current_position = next_comma + 1; // Move past the comma
    }

    // Handle the last SSID after the last comma (if any)
    if(*current_position != '\0' && ssid_count < MAX_WIFI_NETWORKS) {
        size_t ssid_length = strlen(current_position);

        // Trim leading and trailing whitespace
        size_t trimmed_length = ssid_length;
        char* trim_start = trim_whitespace(current_position, &trimmed_length);

        // Handle empty SSIDs
        if(trimmed_length > 0) {
            ssid_list[ssid_count] = malloc(trimmed_length + 1);
            if(ssid_list[ssid_count] == NULL) {
                FURI_LOG_E(TAG, "Memory allocation failed for the last SSID");
                return false;
            }
            strncpy(ssid_list[ssid_count], trim_start, trimmed_length);
            ssid_list[ssid_count][trimmed_length] = '\0'; // Null-terminate the string
            ssid_count++;
        }
    }

    // Add each SSID as a submenu item
    submenu_set_header(app->submenu_wifi_scan, "WiFi Nearby");
    for(uint32_t i = 0; i < ssid_count; i++) {
        char* ssid_item = ssid_list[i];
        if(ssid_item == NULL) {
            // skip any NULL entries
            continue;
        }
        char ssid[64];
        snprintf(ssid, sizeof(ssid), "%s", ssid_item);
        submenu_add_item(
            app->submenu_wifi_scan,
            ssid,
            FlipWiFiSubmenuIndexWiFiScanStart + i,
            callback_submenu_choices,
            app);
    }
    free(data_cstr);
    furi_string_free(scan_data);
    return true;
}
void callback_submenu_choices(void* context, uint32_t index) {
    FlipWiFiApp* app = (FlipWiFiApp*)context;
    if(!app) {
        FURI_LOG_E(TAG, "FlipWiFiApp is NULL");
        return;
    }
    switch(index) {
    case FlipWiFiSubmenuIndexWiFiScan:
        if(fhttp.state == INACTIVE) {
            popup_set_header(app->popup, "[ERROR]", 0, 0, AlignLeft, AlignTop);
            popup_set_text(
                app->popup,
                "WiFi Devboard Disconnected.\nPlease reconnect the board.",
                0,
                40,
                AlignLeft,
                AlignTop);
            view_set_previous_callback(popup_get_view(app->popup), callback_to_submenu_main);
            popup_set_callback(app->popup, popup_callback_main);
            // switch to the popup view
            view_dispatcher_switch_to_view(app->view_dispatcher, FlipWiFiViewPopup);
        }
        // scan for wifi
        if(!flipper_http_scan_wifi()) {
            FURI_LOG_E(TAG, "Failed to scan for WiFi");
            fhttp.state = ISSUE;
            return;
        } else // start the async feed request
        {
            furi_timer_start(fhttp.get_timeout_timer, TIMEOUT_DURATION_TICKS);
            fhttp.state = RECEIVING;
        }
        while(fhttp.state == RECEIVING && furi_timer_is_running(fhttp.get_timeout_timer) > 0) {
            // Wait for the feed to be received
            furi_delay_ms(100);
        }
        furi_timer_stop(fhttp.get_timeout_timer);
        // set each SSID as a submenu item
        if(fhttp.state != IDLE) {
            fhttp.state = ISSUE;
            FURI_LOG_E(TAG, "Failed to receive WiFi scan");
            popup_set_header(app->popup, "[ERROR]", 0, 0, AlignLeft, AlignTop);
            popup_set_text(
                app->popup,
                "Failed to scan...\nTry reconnecting the board!",
                0,
                40,
                AlignLeft,
                AlignTop);
            view_set_previous_callback(popup_get_view(app->popup), callback_to_submenu_main);
            popup_set_callback(app->popup, popup_callback_main);
            // switch to the popup view
            view_dispatcher_switch_to_view(app->view_dispatcher, FlipWiFiViewPopup);
        } else {
            submenu_reset(app->submenu_wifi_scan);
            submenu_set_header(app->submenu_wifi_scan, "WiFi Nearby");
            if(flip_wifi_handle_scan(app)) {
                // switch to the submenu
                view_dispatcher_switch_to_view(app->view_dispatcher, FlipWiFiViewSubmenuScan);
            } else {
                fhttp.state = ISSUE;
                FURI_LOG_E(TAG, "Failed to handle WiFi scan");
                popup_set_header(app->popup, "[ERROR]", 0, 0, AlignLeft, AlignTop);
                popup_set_text(
                    app->popup,
                    "Failed to scan...\nTry reconnecting the board!",
                    0,
                    40,
                    AlignLeft,
                    AlignTop);
                view_set_previous_callback(popup_get_view(app->popup), callback_to_submenu_main);
                popup_set_callback(app->popup, popup_callback_main);
                // switch to the popup view
                view_dispatcher_switch_to_view(app->view_dispatcher, FlipWiFiViewPopup);
            }
            return;
        }
        break;
    case FlipWiFiSubmenuIndexWiFiSaved:
        view_dispatcher_switch_to_view(app->view_dispatcher, FlipWiFiViewSubmenuSaved);
        break;
    case FlipWiFiSubmenuIndexAbout:
        view_dispatcher_switch_to_view(app->view_dispatcher, FlipWiFiViewAbout);
        break;
    case FlipWiFiSubmenuIndexWiFiSavedAddSSID:
        view_dispatcher_switch_to_view(app->view_dispatcher, FlipWiFiViewTextInputSavedAddSSID);
        break;
    case 100 ... 163:
        ssid_index = index - FlipWiFiSubmenuIndexWiFiScanStart;
        view_dispatcher_switch_to_view(app->view_dispatcher, FlipWiFiViewWiFiScan);
        break;
    case 200 ... 263:
        ssid_index = index - FlipWiFiSubmenuIndexWiFiSavedStart;
        view_dispatcher_switch_to_view(app->view_dispatcher, FlipWiFiViewWiFiSaved);
        break;
    default:
        break;
    }
}

void flip_wifi_text_updated_password_scan(void* context) {
    FlipWiFiApp* app = (FlipWiFiApp*)context;
    if(!app) {
        FURI_LOG_E(TAG, "FlipWiFiApp is NULL");
        return;
    }

    // store the entered text
    strncpy(
        app->uart_text_input_buffer_password_scan,
        app->uart_text_input_temp_buffer_password_scan,
        app->uart_text_input_buffer_size_password_scan);

    // Ensure null-termination
    app->uart_text_input_buffer_password_scan[app->uart_text_input_buffer_size_password_scan - 1] =
        '\0';

    // add the SSID and password_scan to the playlist
    app->wifi_playlist.ssids[app->wifi_playlist.count] = strdup(ssid_list[ssid_index]);
    app->wifi_playlist.passwords[app->wifi_playlist.count] =
        strdup(app->uart_text_input_buffer_password_scan);
    app->wifi_playlist.count++;

    // save the playlist to storage
    save_playlist(&app->wifi_playlist);

    flip_wifi_redraw_submenu_saved(app);

    // switch to back to the scan view
    view_dispatcher_switch_to_view(app->view_dispatcher, FlipWiFiViewSubmenuScan);
}
void flip_wifi_text_updated_password_saved(void* context) {
    FlipWiFiApp* app = (FlipWiFiApp*)context;
    if(!app) {
        FURI_LOG_E(TAG, "FlipWiFiApp is NULL");
        return;
    }

    // store the entered text
    strncpy(
        app->uart_text_input_buffer_password_saved,
        app->uart_text_input_temp_buffer_password_saved,
        app->uart_text_input_buffer_size_password_saved);

    // Ensure null-termination
    app->uart_text_input_buffer_password_saved[app->uart_text_input_buffer_size_password_saved - 1] =
        '\0';

    // update the password_saved in the playlist
    app->wifi_playlist.passwords[ssid_index] = strdup(app->uart_text_input_buffer_password_saved);

    // save the playlist to storage
    save_playlist(&app->wifi_playlist);

    // switch to back to the saved view
    view_dispatcher_switch_to_view(app->view_dispatcher, FlipWiFiViewSubmenuSaved);
}

void flip_wifi_text_updated_add_ssid(void* context) {
    FlipWiFiApp* app = (FlipWiFiApp*)context;
    if(!app) {
        FURI_LOG_E(TAG, "FlipWiFiApp is NULL");
        return;
    }

    // store the entered text
    strncpy(
        app->uart_text_input_buffer_add_ssid,
        app->uart_text_input_temp_buffer_add_ssid,
        app->uart_text_input_buffer_size_add_ssid);

    // Ensure null-termination
    app->uart_text_input_buffer_add_ssid[app->uart_text_input_buffer_size_add_ssid - 1] = '\0';

    // do nothing for now, go to the next text input to set the password
    view_dispatcher_switch_to_view(app->view_dispatcher, FlipWiFiViewTextInputSavedAddPassword);
}
void flip_wifi_text_updated_add_password(void* context) {
    FlipWiFiApp* app = (FlipWiFiApp*)context;
    if(!app) {
        FURI_LOG_E(TAG, "FlipWiFiApp is NULL");
        return;
    }

    // store the entered text
    strncpy(
        app->uart_text_input_buffer_add_password,
        app->uart_text_input_temp_buffer_add_password,
        app->uart_text_input_buffer_size_add_password);

    // Ensure null-termination
    app->uart_text_input_buffer_add_password[app->uart_text_input_buffer_size_add_password - 1] =
        '\0';

    // add the SSID and password_scan to the playlist
    app->wifi_playlist.ssids[app->wifi_playlist.count] =
        strdup(app->uart_text_input_buffer_add_ssid);
    app->wifi_playlist.passwords[app->wifi_playlist.count] =
        strdup(app->uart_text_input_buffer_add_password);
    app->wifi_playlist.count++;

    // save the playlist to storage
    save_playlist(&app->wifi_playlist);

    flip_wifi_redraw_submenu_saved(app);

    // switch to back to the saved view
    view_dispatcher_switch_to_view(app->view_dispatcher, FlipWiFiViewSubmenuSaved);
}