#include "wifi_api.h"
#include "wifi_manager.h"
#include <esp_wifi.h>

void wifi_api_get_ap_info(wifi_api_ap_info_t *ap_info)
{
	wifi_ap_record_t ap_record;
	esp_wifi_sta_get_ap_info(&ap_record);
	strncpy(ap_info->ssid, (char *)ap_record.ssid, sizeof(ap_info->ssid));
	strncpy(ap_info->mac, (char *)ap_record.bssid, sizeof(ap_info->mac));
	ap_info->rssi = ap_record.rssi;

	esp_netif_t *sta_netif = wifi_manager_get_sta_netif();
	esp_netif_ip_info_t ip_info;
	esp_netif_get_ip_info(sta_netif, &ip_info);
	ip4_addr_set(&ap_info->ip, &ip_info.ip);
	ip4_addr_set(&ap_info->gateway, &ip_info.gw);
	ip4_addr_set(&ap_info->netmask, &ip_info.netmask);
}

static int rssi_comp(const void *a, const void *b)
{
	const wifi_api_ap_info_t *r1 = a;
	const wifi_api_ap_info_t *r2 = b;
	return r2->rssi - r1->rssi;
}

/**
 * @brief blocking function
 * @param number
 * @param ap_info
 * @return
 */
int wifi_api_get_scan_list(uint16_t *number, wifi_api_ap_info_t *ap_info)
{
	wifi_ap_record_t *records;
	int err;

	records = malloc(*number * sizeof(wifi_ap_record_t));
	if (records == NULL) {
		return ESP_ERR_NO_MEM;
	}

	err = wifi_manager_get_scan_list(number, records);
	if (err) {
		printf("get scan list err\n");
		free(records);
		return err;
	}

	for (int i = 0; i < *number; ++i) {
		printf("%d %d %s\n", i, records[i].rssi, records[i].ssid);
		strncpy(ap_info[i].ssid, (char *) records[i].ssid, sizeof(ap_info[i].ssid));
		strncpy(ap_info[i].mac, (char *) records[i].bssid, sizeof(ap_info[i].mac));
		ap_info[i].rssi = records[i].rssi;
	}
	free(records);

	qsort(ap_info, *number, sizeof(wifi_api_ap_info_t), rssi_comp);
	return 0;
}

static wifi_api_scan_done_cb scan_done_cb;

static void wifi_manager_scan_done(uint16_t ap_found, wifi_ap_record_t *records, void *arg)
{
	wifi_api_ap_info_t *ap_info;

	ap_info = malloc(ap_found * sizeof(wifi_api_ap_info_t));
	for (int i = 0; i < ap_found; ++i) {
		strncpy(ap_info[i].ssid, (char *) records[i].ssid, sizeof(ap_info[i].ssid));
		strncpy(ap_info[i].mac, (char *) records[i].bssid, sizeof(ap_info[i].mac));
		ap_info[i].rssi = records[i].rssi;
	}
	printf("wifi api scan done\n");
	scan_done_cb(ap_found, ap_info, arg);
}

int wifi_api_trigger_scan(uint16_t *max_ap_count, wifi_api_scan_done_cb cb, void *cb_arg)
{
	int err;
	err = wifi_manager_trigger_scan(wifi_manager_scan_done, cb_arg);
	if (err) {
		return err;
	}
	scan_done_cb = cb;
	return 0;
}