#pragma once

#include "radiotap_fields.h"
#include "rssi_measurement.h"
#include <array>
#include <chrono>
#include <vector>

/**
 * @brief The length of a MAC address, in bytes.
 *
 */
constexpr size_t ETH_ALEN = 6;

class station {
    radiotap_fields m_rt_fields;
    std::array<uint8_t, ETH_ALEN> m_mac;
    int8_t m_rssi_wma;
    time_t m_last_seen_time;
    // the end of this vector will contain the most recent instantaneous measurement.
    std::vector<rssi_measurement> m_rssi_measurements;
    uint8_t m_bandwidth;

public:
    explicit station(uint8_t mac[ETH_ALEN]);
    explicit station(const uint8_t mac[ETH_ALEN]);
    explicit station(const std::array<uint8_t, ETH_ALEN> &mac);
    virtual ~station() = default;
    /**
     * @brief Get this station's MAC address.
     *
     * @return std::array<uint8_t, ETH_ALEN> - the MAC.
     */
    std::array<uint8_t, ETH_ALEN> get_mac() const;
    /**
     * @brief Get the most recent RSSI measurement that has been made for this station.
     *
     * @return int8_t the RSSI
     */
    int8_t get_rssi() const;

    /**
     * @brief Update this station's parsed radiotap fields
     *
     * @param rt_f The radiotap fields to read from.
     */
    void update_rt_fields(const radiotap_fields &rt_f);

    int16_t get_frequency() const;
    uint16_t get_channel() const;
    int8_t get_wma_rssi() const;
    /**
     * @brief Determine if a station is alive still.
     *
     * @param timeout_ms The timeout value (milliseconds) threshold. If there has not been a new RSSI measurement made for this station in `timeout_ms`, then it is considered timedout.
     * @return true if the station is timed out.
     * @return false if the station is NOT timed out, OR if we've never seen an RSSI measurement for this station.
     */
    bool is_timed_out_ms(std::chrono::milliseconds timeout_ms) const;
    bool operator==(const station &other) const;
    /**
     * @brief Computes the weighted average of RSSI values over this station's RSSI measurement list.
     *
     * More recent measurements are weighed higher.
     *
     * (x_1*w_1) + ... + (x_n*w_n) / sum(w_i) -- where x_i is an RSSI measurement value in dBm, and w_i is it's weight.
     */
    void calculate_wma();
    /**
     * @brief Unix timestamp of the last time a packet from this station was seen, in seconds.
     *
     * @return time_t Unix timestamp of this station's last seen packet, in seconds.
     */
    time_t get_last_seen_seconds() const;

    /**
     * @brief Update this station's last_seen field - the timestamp of the last captured packet from this station.
     *
     * @param time_seconds the unix time of the last packet received for this station, in seconds.
     */
    void update_last_seen(time_t time_seconds);

    /**
     * @brief Determine if this station's MAC address is multicast.
     *
     * @return true if this station's MAC is ff:ff:ff:ff:ff:ff, otherwise false.
     */
    bool addr_is_multicast() const;

    /**
     * @brief Set the bandwidth of the radio that last made a measurement for this station.
     *
     * @param bw The bandwidth of the radio.
     */
    void set_bandwidth(uint8_t bw);

    /**
     * @brief The bandwidth of the radio that made this station's last RSSI measurement.
     *
     * @return uint8_t The bandwidth of the radio making measurements for this station.
     */
    uint8_t get_bandwidth() const;

    /**
     * @brief Get this station's last known MCS rate;
     * Note: only available if station is VHT-capable and tx/rx-ing on a VHT channel.
     *
     * @return uint8_t The MCS index if available, else -1.
     */
    uint8_t get_vht_mcs_rate() const;

    /**
     * @brief Get this station's last known number of spatial streams.
     * Note: only available if station is VHT-capable and tx/rx-ing on a VHT channel.
     *
     * @return uint8_t The number of spatial streams if available, else -1.
     */
    uint8_t get_vht_nss() const;

    /**
     * @brief Get this station's last known antenna noise value, defined as:
     * RF noise power at the antenna, decibel difference from an arbitrary, fixed reference.
     *
     * @return uint8_t The antenna noise value.
     */
    uint8_t get_ant_noise() const;
};