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

#include "eos.h"
#include "event.h"
#include "net.h"

#include "wifi.h"

static eos_evt_handler_t evt_handler[EOS_WIFI_MAX_MTYPE];
static uint16_t evt_handler_flags_buf_free = 0;
static uint16_t evt_handler_flags_buf_acq = 0;

static void wifi_handle_evt(unsigned char type, unsigned char *buffer, uint16_t len) {
    if ((buffer == NULL) || (len < 1)) {
        eos_evtq_bad_handler(type, buffer, len);
        eos_net_free(buffer, 0);
        return;
    }

    unsigned char mtype = buffer[0];
    if (mtype < EOS_WIFI_MAX_MTYPE) {
        evt_handler[mtype](type, buffer, len);
    } else {
        eos_evtq_bad_handler(type, buffer, len);
        eos_net_free(buffer, 0);
        return;
    }
}

void eos_wifi_init(void) {
    int i;

    for (i=0; i<EOS_WIFI_MAX_MTYPE; i++) {
        evt_handler[i] = eos_evtq_bad_handler;
    }
    eos_net_set_handler(EOS_NET_MTYPE_WIFI, wifi_handle_evt);
}

void eos_wifi_set_handler(unsigned char mtype, eos_evt_handler_t handler) {
    if (handler == NULL) handler = eos_evtq_bad_handler;
    if (mtype < EOS_WIFI_MAX_MTYPE) evt_handler[mtype] = handler;
}

void eos_wifi_connect(const char *ssid, const char *pass) {
    int ssid_len = strlen(ssid);
    int pass_len = strlen(pass);
    unsigned char *buffer = eos_net_alloc();

    buffer[0] = EOS_WIFI_MTYPE_CONNECT;
    strcpy(buffer+1, ssid);
    buffer[ssid_len+1] = 0;
    strcpy(buffer+ssid_len+2, pass);
    buffer[ssid_len+pass_len+2] = 0;
    eos_net_send(EOS_NET_MTYPE_WIFI, buffer, ssid_len+pass_len+3, 0);
}

void eos_wifi_disconnect(void) {
    unsigned char *buffer = eos_net_alloc();
    buffer[0] = EOS_WIFI_MTYPE_DISCONNECT;
    eos_net_send(EOS_NET_MTYPE_WIFI, buffer, 1, 0);
}