#include <stdint.h>
#include <unistd.h>
#include <string.h>

#include <espressif/esp_common.h>
#include <FreeRTOS.h>
#include <task.h>
#include <dhcpserver.h>
#include <lwip/api.h>

#include "common.h"
#include "network.h"

SemaphoreHandle_t semNetworks;

struct network_info networks[MAX_NETWORKS];

static void scan_done_cb(void *arg, sdk_scan_status_t status) {
    int i;

    if (xSemaphoreTake(semNetworks, TIME_MS(50)) == pdTRUE) {
        if (status != SCAN_OK) {
            networks[0].channel = networks[0].rssi = 0;
        } else {
            struct sdk_bss_info *bss = (struct sdk_bss_info *)arg;
            // first one is invalid
            bss = bss->next.stqe_next;

            for (i=0; NULL != bss && i<MAX_NETWORKS; i++) {
                memcpy(networks[i].bssid, bss->bssid, sizeof(networks[i].bssid));
                memcpy(networks[i].ssid, bss->ssid, sizeof(networks[i].ssid));
                networks[i].channel = bss->channel;
                networks[i].rssi= bss->rssi;
                networks[i].authmode = bss->authmode;
                networks[i].is_hidden = bss->is_hidden;

                bss = bss->next.stqe_next;
            }
            if (i < MAX_NETWORKS) {
                networks[i].channel = networks[i].rssi = 0;
            }
        }

        xSemaphoreGive(semNetworks);
    }
}

void NetworkScanTask(void *p) {
    networks[0].channel = networks[0].rssi = 0;
    semNetworks = xSemaphoreCreateMutex();

    for (;;) {
        sdk_wifi_station_scan(NULL, scan_done_cb);
        vTaskDelay(TIME_MS(5000));
    }
}


/* This task ensures the network connection is up.
 * If the network has not been configured yet,
 * setup an AP with captive portal to allow
 * network configuration.
 */
void NetworkConnectionTask(void *p) {
    struct sdk_softap_config apconf;
    struct sdk_station_config staconf;
    uint8_t mac[6];

    if (1 || !sdk_wifi_station_get_config(&staconf)) {
        /* Station mode not configured: fallback to SoftAP configuration */
        printf("sdk_wifi_station_get_config() failed\n");
        if (!sdk_wifi_softap_get_config(&apconf)) {
            /* SoftAP mode not configured: initialize default configuration */
            printf("sdk_wifi_softap_get_config() failed\n");
            sdk_wifi_get_macaddr(SOFTAP_IF, mac);
            sprintf((char *)apconf.ssid, "UT61e-%02x%02x%02x", mac[3], mac[4], mac[5]);
            apconf.ssid_len = strlen((char *)apconf.ssid);
            apconf.channel = 1; /* TODO: auto channel selection? */
            apconf.authmode = AUTH_OPEN;
            apconf.ssid_hidden = 0;
            apconf.max_connection = 4;
            apconf.beacon_interval = 100;
            sdk_wifi_softap_set_config(&apconf);
        } else {
            struct ip_info ap_ip;
            ip_addr_t first_client_ip;

            IP4_ADDR(&ap_ip.ip, 192, 168, 13, 1);
            IP4_ADDR(&ap_ip.gw, 0, 0, 0, 0);
            IP4_ADDR(&ap_ip.netmask, 255, 255, 255, 0);
            sdk_wifi_set_ip_info(1, &ap_ip);

            IP4_ADDR(&first_client_ip, 192, 168, 13, 2);
            dhcpserver_start(&first_client_ip, 4);

            sdk_wifi_set_opmode(SOFTAP_MODE);

            printf("SoftAP config:\n");
            printf("SSID: %s\n", apconf.ssid);
            printf("Password: %s\n", apconf.password);
            printf("Channel: %d\n", apconf.channel);
            printf("Authmode: %d\n", apconf.authmode);
            for (;;)
                vTaskDelay(TIME_MS(5000));
        }
    } else {
        sdk_wifi_set_opmode(STATION_MODE);
        sdk_wifi_station_connect();
        printf("Station config:\n");
        printf("SSID: %s\n", staconf.ssid);
        printf("Password: %s\n", staconf.password);
        printf("BSSID: %s\n", staconf.bssid);
        printf("BSSID set: %d\n", staconf.bssid_set);
        uint8_t st;
        for (;;) {
            switch (st = sdk_wifi_station_get_connect_status()) {
            case STATION_IDLE:
                printf("IDLE\n");
                break;
            case STATION_CONNECTING:
                printf("CONNECTING\n");
                break;
            case STATION_WRONG_PASSWORD:
                printf("WRONG_PASSWORD\n");
                break;
            case STATION_NO_AP_FOUND:
                printf("NO_AP_FOUND\n");
                break;
            case STATION_CONNECT_FAIL:
                printf("CONNECT_FAIL\n");
                break;
            case STATION_GOT_IP:
                printf("GOT_IP\n");
                break;
            default:
                printf("UNEXPECTED status: %d\n", st);
            }
            vTaskDelay(TIME_MS(750));
        }
    }
}