/*
 * eth_udp_client.hpp
 *
 *  Created on: 15 ���. 2023 �.
 *      Author: sychev
 */

#ifndef FREE_RTOS_ETHERNET_IP_ETH_UDP_CLIENT_HPP_
#define FREE_RTOS_ETHERNET_IP_ETH_UDP_CLIENT_HPP_

#include "free_rtos/semaphore/semaphore.hpp"
#include "free_rtos/mutex/mutex.hpp"
#include "free_rtos/ethernet/eth_frame.h"
#include "free_rtos/ethernet_ip/eth_ip_iface.hpp"
#include "free_rtos/ethernet/eth_types.h"

#include <array>
#include <vector>
#include <atomic>
#include <iterator>
#include <cstdint>

namespace free_rtos {

enum UDPToggle : uint32_t {
    UDP_BUF_0 = 0,
    UDP_BUF_1,
    UDP_BUF_NUM,
    UDP_BUSY = UDP_BUF_NUM
};

class EthUpdClient {
    friend class EthUdpServer;

public:
    EthUpdClient(EthIpIface& ip_iface, TEthMacPorts port_id,
                 uint16_t port_dst_be, uint16_t port_src_be,
                 bool use_chksum);

    bool capture_buffer();
    size_t get_data_size();

    int32_t read(uint8_t * p_data, uint32_t len);
    int32_t read_async(uint8_t * p_data, uint32_t len);
/*
    template<class T, class Allocator>
    int32_t read_async(std::vector<T, Allocator>& data, uint32_t len)
    {
        if (len == 0) {
            return 0;
        }

        if (buff_[toggle_].empty() == true) {
            if(capture_buffer() == true) {
                return 0;
            }
        }

        return get_data(data, len);
    }
*/
    bool write(uint32_t ip_dst_be, uint8_t * p_data, uint32_t len);

    template<class T, class Allocator>
    bool write(uint32_t ip_dst_be, std::vector<T, Allocator>& data)
    {
        return write(ip_dst_be, data.data(), data.size() * sizeof(T));
    }

    void clear(); ///������� ��������� ������

private:
    void put_data(uint32_t src_ip, uint8_t * p_data, uint32_t len);
    int32_t get_data(uint8_t * p_data, uint32_t len);
/*
    template<class T, class Allocator>
    int32_t get_data(std::vector<T, Allocator>& data, uint32_t len)
    {
        std::vector<uint8_t>& buff = buff_[toggle_];
        uint32_t size = buff.size();

        if (size > len) {
            size = len;
        }

        auto item_begin = buff.begin();
        auto item_end = item_begin + size;

        data.insert(data.end(), item_begin, item_end);
        buff.erase(item_begin, item_end);

        return size;
    }
*/
private:
    const uint32_t max_buf_size = 0x8000; // 0x8000 - 32768, 0x10000 - 65536, 0x11CA - 4554, 0x0BDC - 3036, 0x100000 - 1Mb

    const uint16_t port_dst_be_; /// big endian
    const uint16_t port_src_be_; /// big endian
    const bool use_checksum_;

    EthIpIface& ip_iface_;

    const TEthMacPorts port_id_;

    TEthPkt eth_pkt_;

    uint32_t src_ip_;

    std::array<std::vector<uint8_t>, UDP_BUF_NUM> buff_;
    std::atomic<UDPToggle> in_toggle_{UDPToggle::UDP_BUF_1};
    UDPToggle out_toggle_{UDPToggle::UDP_BUF_0};

    Semaphore rx_sem_;
};

}

#endif /* FREE_RTOS_ETHERNET_IP_ETH_UDP_CLIENT_HPP_ */