#include "_network_WLAN.h"
#include "../pikascript-lib/PikaStdDevice/pika_hal.h"
#include "PikaStdData_List.h"
#include "PikaStdData_Tuple.h"

#define check_res(res)                                                       \
    if (res != 0) {                                                          \
        pika_platform_printf("check_res failed: %d, at %s:%d:%s()\r\n", res, \
                             __FILE__, __LINE__, __FUNCTION__);              \
    }

void _network_WLAN___init__(PikaObj* self, int interface_id) {
    pika_dev* hal_wifi = pika_hal_open(PIKA_HAL_WIFI, "WLAN0");
    if (hal_wifi == NULL) {
        return;
    }
    pika_hal_WIFI_config cfg = {0};
    cfg.mode = interface_id;
    check_res(pika_hal_ioctl(hal_wifi, PIKA_HAL_IOCTL_CONFIG, &cfg));
    obj_setPtr(self, "hal_wifi", hal_wifi);
}

void _network_WLAN_active(PikaObj* self, int is_active) {
    pika_dev* hal_wifi = obj_getPtr(self, "hal_wifi");
    if (hal_wifi == NULL) {
        return;
    }
    check_res(pika_hal_ioctl(hal_wifi, PIKA_HAL_IOCTL_ENABLE));
    return;
}

int _network_WLAN_checkActive(PikaObj* self) {
    pika_dev* hal_wifi = obj_getPtr(self, "hal_wifi");
    if (hal_wifi == NULL) {
        return -1;
    }
    int is_active = 0;
    check_res(
        pika_hal_ioctl(hal_wifi, PIKA_HAL_IOCTL_WIFI_GET_ACTIVE, &is_active));
    return is_active;
}

void _network_WLAN_config(PikaObj* self, PikaDict* kwargs) {
    pika_dev* hal_wifi = obj_getPtr(self, "hal_wifi");
    if (hal_wifi == NULL) {
        return;
    }
    pika_hal_WIFI_config cfg = {0};
    if (pikaDict_isArgExist(kwargs, "ssid")) {
        strcpy(cfg.ap_ssid, pikaDict_getStr(kwargs, "ssid"));
    }
    if (pikaDict_isArgExist(kwargs, "password")) {
        strcpy(cfg.ap_password, pikaDict_getStr(kwargs, "password"));
    }
    if (pikaDict_isArgExist(kwargs, "channel")) {
        cfg.channel = pikaDict_getInt(kwargs, "channel") + 1;
    }
    check_res(pika_hal_ioctl(hal_wifi, PIKA_HAL_IOCTL_CONFIG, &cfg));
}

Arg* _network_WLAN_checkConfig(PikaObj* self, char* param) {
    pika_dev* hal_wifi = obj_getPtr(self, "hal_wifi");
    if (hal_wifi == NULL) {
        return NULL;
    }
    pika_hal_WIFI_config* cfg = hal_wifi->ioctl_config;
    if (strcmp(param, "ssid") == 0) {
        return arg_newStr(cfg->ap_ssid);
    } else if (strcmp(param, "password") == 0) {
        return arg_newStr(cfg->ap_password);
    } else if (strcmp(param, "channel") == 0) {
        return arg_newInt(cfg->channel - 1);
    } else {
        return NULL;
    }
}

void _network_WLAN_connect(PikaObj* self, char* ssid, char* key) {
    pika_dev* hal_wifi = obj_getPtr(self, "hal_wifi");
    if (hal_wifi == NULL) {
        return;
    }
    pika_hal_WIFI_connect_config conncfg = {0};
    strcpy(conncfg.ssid, ssid);
    strcpy(conncfg.password, key);
    check_res(pika_hal_ioctl(hal_wifi, PIKA_HAL_IOCTL_WIFI_CONNECT, &conncfg));
    return;
}

void _network_WLAN_connectWithBssid(PikaObj* self,
                                    char* ssid,
                                    char* key,
                                    char* bssid) {
    pika_dev* hal_wifi = obj_getPtr(self, "hal_wifi");
    if (hal_wifi == NULL) {
        return;
    }
    pika_hal_WIFI_connect_config conncfg = {0};
    strcpy(conncfg.ssid, ssid);
    strcpy(conncfg.password, key);
    strcpy(conncfg.bssid, bssid);
    check_res(pika_hal_ioctl(hal_wifi, PIKA_HAL_IOCTL_WIFI_CONNECT, &conncfg));
}

void _network_WLAN_disconnect(PikaObj* self) {
    pika_dev* hal_wifi = obj_getPtr(self, "hal_wifi");
    if (hal_wifi == NULL) {
        return;
    }
    check_res(pika_hal_ioctl(hal_wifi, PIKA_HAL_IOCTL_WIFI_DISCONNECT));
}

int _network_WLAN_status(PikaObj* self) {
    pika_dev* hal_wifi = obj_getPtr(self, "hal_wifi");
    if (hal_wifi == NULL) {
        return -1;
    }
    PIKA_HAL_WIFI_STATUS status = 0;
    check_res(
        pika_hal_ioctl(hal_wifi, PIKA_HAL_IOCTL_WIFI_GET_STATUS, &status));
    return status;
}

int _network_WLAN_isconnected(PikaObj* self) {
    if (_network_WLAN_status(self) == PIKA_HAL_WIFI_STATUS_GOT_IP) {
        return 1;
    } else {
        return 0;
    }
}

int _network_WLAN_statusWithParam(PikaObj* self, char* param) {
    return _network_WLAN_status(self);
}

PikaObj* _network_WLAN_checkIfconfig(PikaObj* self) {
    pika_dev* hal_wifi = obj_getPtr(self, "hal_wifi");
    if (hal_wifi == NULL) {
        return NULL;
    }
    pika_hal_WIFI_ifconfig ifconfig = {0};
    check_res(
        pika_hal_ioctl(hal_wifi, PIKA_HAL_IOCTL_WIFI_GET_IFCONFIG, &ifconfig));
    PikaObj* ifconfig_list = newNormalObj(New_PikaStdData_List);
    PikaStdData_List___init__(ifconfig_list);
    Arg* arg = arg_newStr(ifconfig.ip);
    PikaStdData_List_append(ifconfig_list, arg);
    arg_deinit(arg);
    arg = arg_newStr(ifconfig.netmask);
    PikaStdData_List_append(ifconfig_list, arg);
    arg_deinit(arg);
    arg = arg_newStr(ifconfig.gateway);
    PikaStdData_List_append(ifconfig_list, arg);
    arg_deinit(arg);
    arg = arg_newStr(ifconfig.dns);
    PikaStdData_List_append(ifconfig_list, arg);
    arg_deinit(arg);
    return ifconfig_list;
}

void _network_WLAN_ifconfig(PikaObj* self,
                            char* ip,
                            char* mask,
                            char* gateway,
                            char* dns) {
    pika_dev* hal_wifi = obj_getPtr(self, "hal_wifi");
    if (hal_wifi == NULL) {
        return;
    }
    pika_hal_WIFI_ifconfig ifconfig = {0};
    strcpy(ifconfig.ip, ip);
    strcpy(ifconfig.netmask, mask);
    strcpy(ifconfig.gateway, gateway);
    strcpy(ifconfig.dns, dns);
    check_res(
        pika_hal_ioctl(hal_wifi, PIKA_HAL_IOCTL_WIFI_SET_IFCONFIG, &ifconfig));
    return;
}

PikaObj* _network_WLAN_scan(PikaObj* self) {
    pika_dev* hal_wifi = obj_getPtr(self, "hal_wifi");
    if (hal_wifi == NULL) {
        return NULL;
    }
    pika_hal_WIFI_scan_result* result = NULL;
    check_res(pika_hal_ioctl(hal_wifi, PIKA_HAL_IOCTL_WIFI_SCAN, &result));
    if (NULL == result) {
        return NULL;
    }
    PikaObj* scan_list = newNormalObj(New_PikaStdData_List);
    PikaStdData_List___init__(scan_list);
    for (int i = 0; i < result->count; i++) {
        // (ssid, bssid, channel, RSSI, authmode, hidden)
        PikaObj* record = newNormalObj(New_PikaStdData_Tuple);
        PikaStdData_Tuple___init__(record);
        Arg* arg = arg_newStr(result->records[i].ssid);
        PikaStdData_List_append(record, arg);
        arg_deinit(arg);
        arg = arg_newBytes(result->records[i].bssid,
                           result->records[i].bssid_len);
        PikaStdData_List_append(record, arg);
        arg_deinit(arg);
        arg = arg_newInt(result->records[i].channel);
        PikaStdData_List_append(record, arg);
        arg_deinit(arg);
        arg = arg_newInt(result->records[i].rssi);
        PikaStdData_List_append(record, arg);
        arg_deinit(arg);
        arg = arg_newInt(result->records[i].authmode);
        PikaStdData_List_append(record, arg);
        arg_deinit(arg);
        arg = arg_newInt(0);
        PikaStdData_List_append(record, arg);
        arg_deinit(arg);
        arg = arg_newObj(record);
        PikaStdData_List_append(scan_list, arg);
        arg_deinit(arg);
    }
    pika_platform_free(result);
    return scan_list;
}

void _network_WLAN_close(PikaObj* self) {
    pika_dev* hal_wifi = obj_getPtr(self, "hal_wifi");
    if (hal_wifi == NULL) {
        return;
    }
    check_res(pika_hal_close(hal_wifi));
}