feat: add WiFi setup state machine + config_add_wifi with unit tests

1 files changed, 89 insertions, 0 deletions

diff --git a/main/wifi_setup.c b/main/wifi_setup.c
new file mode 100644
index 0000000..b2669e9
--- /dev/null
+++ b/main/wifi_setup.c
@@ -0,0 +1,89 @@
+#include "wifi_setup.h"
+#include <string.h>
+
+void wifi_setup_init(wifi_setup_t *setup) {
+    if (!setup) return;
+    memset(setup, 0, sizeof(*setup));
+    setup->state = SETUP_SCAN;
+    setup->selected_ap = -1;
+}
+
+void wifi_setup_set_aps(wifi_setup_t *setup, const wifi_ap_info_t *aps, int count) {
+    if (!setup || !aps) return;
+    if (count > WIFI_SETUP_MAX_APS) count = WIFI_SETUP_MAX_APS;
+    memcpy(setup->aps, aps, count * sizeof(wifi_ap_info_t));
+    setup->ap_count = count;
+    setup->list_scroll = 0;
+    setup->state = SETUP_LIST;
+}
+
+int wifi_setup_visible_count(const wifi_setup_t *setup) {
+    if (!setup) return 0;
+    int remaining = setup->ap_count - setup->list_scroll;
+    if (remaining > WIFI_SETUP_MAX_VISIBLE) remaining = WIFI_SETUP_MAX_VISIBLE;
+    return remaining < 0 ? 0 : remaining;
+}
+
+const wifi_ap_info_t *wifi_setup_get_visible(const wifi_setup_t *setup, int idx) {
+    if (!setup || idx < 0 || idx >= wifi_setup_visible_count(setup)) return NULL;
+    return &setup->aps[setup->list_scroll + idx];
+}
+
+setup_state_t wifi_setup_handle_select(wifi_setup_t *setup, int list_idx) {
+    if (!setup || setup->state != SETUP_LIST) return setup ? setup->state : SETUP_CANCELLED;
+    if (list_idx < 0 || list_idx >= wifi_setup_visible_count(setup)) return setup->state;
+
+    int real_idx = setup->list_scroll + list_idx;
+    setup->selected_ap = real_idx;
+    strncpy(setup->selected_ssid, setup->aps[real_idx].ssid, WIFI_SETUP_SSID_LEN - 1);
+    setup->selected_ssid[WIFI_SETUP_SSID_LEN - 1] = '\0';
+    setup->state = SETUP_PASSWORD;
+    return setup->state;
+}
+
+setup_state_t wifi_setup_handle_connect(wifi_setup_t *setup) {
+    if (!setup || setup->state != SETUP_PASSWORD) return setup ? setup->state : SETUP_CANCELLED;
+    setup->state = SETUP_CONNECTING;
+    return setup->state;
+}
+
+setup_state_t wifi_setup_handle_connect_result(wifi_setup_t *setup, bool success, const char *ip) {
+    if (!setup) return SETUP_CANCELLED;
+    if (setup->state != SETUP_CONNECTING) return setup->state;
+
+    if (success) {
+        setup->state = SETUP_SUCCESS;
+        if (ip) {
+            strncpy(setup->connect_ip, ip, sizeof(setup->connect_ip) - 1);
+            setup->connect_ip[sizeof(setup->connect_ip) - 1] = '\0';
+        }
+        setup->connect_failed_auth = false;
+    } else {
+        setup->state = SETUP_FAILED;
+        setup->connect_failed_auth = true;
+        setup->connect_ip[0] = '\0';
+    }
+    return setup->state;
+}
+
+setup_state_t wifi_setup_handle_cancel(wifi_setup_t *setup) {
+    if (!setup) return SETUP_CANCELLED;
+    setup->state = SETUP_CANCELLED;
+    return setup->state;
+}
+
+setup_state_t wifi_setup_handle_retry(wifi_setup_t *setup) {
+    if (!setup) return SETUP_CANCELLED;
+    if (setup->state != SETUP_FAILED) return setup->state;
+    setup->state = SETUP_PASSWORD;
+    setup->connect_failed_auth = false;
+    return setup->state;
+}
+
+setup_state_t wifi_setup_handle_change_network(wifi_setup_t *setup) {
+    if (!setup) return SETUP_CANCELLED;
+    if (setup->state != SETUP_FAILED) return setup->state;
+    setup->state = SETUP_LIST;
+    setup->connect_failed_auth = false;
+    return setup->state;
+}