#pragma once

#include "Pubsub.h"
#include "WifiGlobals.h"
#include "WifiSettings.h"
#include <cstdio>
#include <string>
#include <vector>

#ifdef ESP_PLATFORM
#include "esp_wifi.h"
#include "WifiSettings.h"
#else
#include <cstdint>
// From esp_wifi_types.h in ESP-IDF 5.2
typedef enum {
    WIFI_AUTH_OPEN = 0,         /**< authenticate mode : open */
    WIFI_AUTH_WEP,              /**< authenticate mode : WEP */
    WIFI_AUTH_WPA_PSK,          /**< authenticate mode : WPA_PSK */
    WIFI_AUTH_WPA2_PSK,         /**< authenticate mode : WPA2_PSK */
    WIFI_AUTH_WPA_WPA2_PSK,     /**< authenticate mode : WPA_WPA2_PSK */
    WIFI_AUTH_ENTERPRISE,       /**< authenticate mode : WiFi EAP security */
    WIFI_AUTH_WPA2_ENTERPRISE = WIFI_AUTH_ENTERPRISE,  /**< authenticate mode : WiFi EAP security */
    WIFI_AUTH_WPA3_PSK,         /**< authenticate mode : WPA3_PSK */
    WIFI_AUTH_WPA2_WPA3_PSK,    /**< authenticate mode : WPA2_WPA3_PSK */
    WIFI_AUTH_WAPI_PSK,         /**< authenticate mode : WAPI_PSK */
    WIFI_AUTH_OWE,              /**< authenticate mode : OWE */
    WIFI_AUTH_WPA3_ENT_192,     /**< authenticate mode : WPA3_ENT_SUITE_B_192_BIT */
    WIFI_AUTH_WPA3_EXT_PSK,     /**< authenticate mode : WPA3_PSK_EXT_KEY */
    WIFI_AUTH_WPA3_EXT_PSK_MIXED_MODE, /**< authenticate mode: WPA3_PSK + WPA3_PSK_EXT_KEY */
    WIFI_AUTH_MAX
} wifi_auth_mode_t;
#endif

namespace tt::service::wifi {

enum WifiEventType {
    /** Radio was turned on */
    WifiEventTypeRadioStateOn,
    /** Radio is turning on. */
    WifiEventTypeRadioStateOnPending,
    /** Radio is turned off */
    WifiEventTypeRadioStateOff,
    /** Radio is turning off */
    WifiEventTypeRadioStateOffPending,
    /** Started scanning for access points */
    WifiEventTypeScanStarted,
    /** Finished scanning for access points */ // TODO: 1 second validity
    WifiEventTypeScanFinished,
    WifiEventTypeDisconnected,
    WifiEventTypeConnectionPending,
    WifiEventTypeConnectionSuccess,
    WifiEventTypeConnectionFailed
};

enum WifiRadioState {
    WIFI_RADIO_ON_PENDING,
    WIFI_RADIO_ON,
    WIFI_RADIO_CONNECTION_PENDING,
    WIFI_RADIO_CONNECTION_ACTIVE,
    WIFI_RADIO_OFF_PENDING,
    WIFI_RADIO_OFF
};

struct WifiEvent {
    WifiEventType type;
};

struct WifiApRecord {
    std::string ssid;
    int8_t rssi;
    wifi_auth_mode_t auth_mode;
};

/**
 * @brief Get wifi pubsub
 * @return PubSub*
 */
PubSub* getPubsub();

WifiRadioState getRadioState();
/**
 * @brief Request scanning update. Returns immediately. Results are through pubsub.
 */
void scan();

/**
 * @return true if wifi is actively scanning
 */
bool isScanning();

/**
 * @brief Returns the access points from the last scan (if any). It only contains public APs.
 */
std::vector<WifiApRecord> getScanResults();

/**
 * @brief Overrides the default scan result size of 16.
 * @param records the record limit for the scan result (84 bytes per record!)
 */
void setScanRecords(uint16_t records);

/**
 * @brief Enable/disable the radio. Ignores input if desired state matches current state.
 * @param enabled
 */
void setEnabled(bool enabled);

/**
 * @brief Connect to a network. Disconnects any existing connection.
 * Returns immediately but runs in the background. Results are through pubsub.
 * @param ap
 */
void connect(const settings::WifiApSettings* ap, bool remember);

/**
 * @brief Disconnect from the access point. Doesn't have any effect when not connected.
 */
void disconnect();

/**
 * Return true if the connection isn't unencrypted.
 */
bool isConnectionSecure();

/**
 * Returns the RSSI value (negative number) or return 1 when not connected
 */
int getRssi();

} // namespace