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

4 files changed, 198 insertions, 0 deletions

diff --git a/main/config.c b/main/config.c
index 6753f40..32ae30e 100644
--- a/main/config.c
+++ b/main/config.c
@@ -320,3 +320,60 @@ void tollgate_config_derive_unique(tollgate_config_t *cfg)
     ESP_LOGI(TAG, "Unique config derived from nsec: SSID='%s', AP_IP=%s",
              cfg->ap_ssid, cfg->ap_ip_str);
 }
+
+esp_err_t tollgate_config_add_wifi(const char *ssid, const char *password) {
+    if (!ssid || !password) return ESP_ERR_INVALID_ARG;
+    if (g_config.network_count >= TOLLGATE_MAX_WIFI_NETWORKS) return ESP_ERR_NO_MEM;
+
+    strncpy(g_config.networks[g_config.network_count].ssid, ssid,
+            sizeof(g_config.networks[g_config.network_count].ssid) - 1);
+    strncpy(g_config.networks[g_config.network_count].password, password,
+            sizeof(g_config.networks[g_config.network_count].password) - 1);
+    g_config.network_count++;
+
+    cJSON *root = cJSON_CreateObject();
+    cJSON *nsec = cJSON_CreateString(g_config.nsec);
+    cJSON_AddItemToObject(root, "nsec", nsec);
+
+    cJSON *networks = cJSON_CreateArray();
+    for (int i = 0; i < g_config.network_count; i++) {
+        cJSON *net = cJSON_CreateObject();
+        cJSON_AddStringToObject(net, "ssid", g_config.networks[i].ssid);
+        cJSON_AddStringToObject(net, "password", g_config.networks[i].password);
+        cJSON_AddItemToArray(networks, net);
+    }
+    cJSON_AddItemToObject(root, "wifi_networks", networks);
+
+    if (g_config.ap_password[0])
+        cJSON_AddStringToObject(root, "ap_password", g_config.ap_password);
+    cJSON_AddStringToObject(root, "mint_url", g_config.mint_url);
+    cJSON_AddNumberToObject(root, "price_per_step", g_config.price_per_step);
+    cJSON_AddNumberToObject(root, "step_size_ms", g_config.step_size_ms);
+
+    if (g_config.metric[0])
+        cJSON_AddStringToObject(root, "metric", g_config.metric);
+
+    cJSON_AddStringToObject(root, "nostr_geohash", g_config.nostr_geohash);
+
+    cJSON *relays = cJSON_CreateArray();
+    for (int i = 0; i < g_config.nostr_relay_count; i++) {
+        cJSON_AddItemToArray(relays, cJSON_CreateString(g_config.nostr_relays[i]));
+    }
+    cJSON_AddItemToObject(root, "nostr_relays", relays);
+    cJSON_AddNumberToObject(root, "nostr_publish_interval_s", g_config.nostr_publish_interval_s);
+
+    char *json_str = cJSON_Print(root);
+    cJSON_Delete(root);
+
+    FILE *f = fopen("/spiffs/config.json", "w");
+    if (!f) {
+        free(json_str);
+        return ESP_FAIL;
+    }
+    fputs(json_str, f);
+    fclose(f);
+    free(json_str);
+
+    ESP_LOGI(TAG, "Added WiFi network '%s' to config (total: %d)", ssid, g_config.network_count);
+    return ESP_OK;
+}
diff --git a/main/config.h b/main/config.h
index fa4d95c..dad0ef0 100644
--- a/main/config.h
+++ b/main/config.h
@@ -71,5 +71,6 @@ esp_err_t tollgate_config_init(void);
 const tollgate_config_t *tollgate_config_get(void);
 esp_err_t tollgate_config_get_wifi(wifi_config_t *wifi_config);
 esp_err_t tollgate_config_get_next_wifi(wifi_config_t *wifi_config);
+esp_err_t tollgate_config_add_wifi(const char *ssid, const char *password);
 
 #endif
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;
+}
diff --git a/main/wifi_setup.h b/main/wifi_setup.h
new file mode 100644
index 0000000..17712d5
--- /dev/null
+++ b/main/wifi_setup.h
@@ -0,0 +1,51 @@
+#ifndef WIFI_SETUP_H
+#define WIFI_SETUP_H
+
+#include "esp_err.h"
+#include <stdint.h>
+#include <stdbool.h>
+
+#define WIFI_SETUP_MAX_APS     20
+#define WIFI_SETUP_MAX_VISIBLE  8
+#define WIFI_SETUP_SSID_LEN    33
+#define WIFI_SETUP_PASS_LEN    64
+
+typedef enum {
+    SETUP_SCAN,
+    SETUP_LIST,
+    SETUP_PASSWORD,
+    SETUP_CONNECTING,
+    SETUP_SUCCESS,
+    SETUP_FAILED,
+    SETUP_CANCELLED
+} setup_state_t;
+
+typedef struct {
+    char ssid[WIFI_SETUP_SSID_LEN];
+    int rssi;
+    bool secured;
+} wifi_ap_info_t;
+
+typedef struct {
+    setup_state_t state;
+    wifi_ap_info_t aps[WIFI_SETUP_MAX_APS];
+    int ap_count;
+    int list_scroll;
+    int selected_ap;
+    char selected_ssid[WIFI_SETUP_SSID_LEN];
+    char connect_ip[16];
+    bool connect_failed_auth;
+} wifi_setup_t;
+
+void wifi_setup_init(wifi_setup_t *setup);
+void wifi_setup_set_aps(wifi_setup_t *setup, const wifi_ap_info_t *aps, int count);
+int wifi_setup_visible_count(const wifi_setup_t *setup);
+const wifi_ap_info_t *wifi_setup_get_visible(const wifi_setup_t *setup, int idx);
+setup_state_t wifi_setup_handle_select(wifi_setup_t *setup, int list_idx);
+setup_state_t wifi_setup_handle_connect(wifi_setup_t *setup);
+setup_state_t wifi_setup_handle_connect_result(wifi_setup_t *setup, bool success, const char *ip);
+setup_state_t wifi_setup_handle_cancel(wifi_setup_t *setup);
+setup_state_t wifi_setup_handle_retry(wifi_setup_t *setup);
+setup_state_t wifi_setup_handle_change_network(wifi_setup_t *setup);
+
+#endif