UdpCommunication.hpp

/*
** EPITECH PROJECT, 2022
** B-CPP-500-PAR-5-1-rtype-martin.vanaud
** File description:
** UdpCommunication
*/
 
#ifndef UDPCOMMUNICATION_HPP_
    #define UDPCOMMUNICATION_HPP_
 
    #include "IUdpCommunication.hpp"
 
class UdpCommunication : public IUdpCommunication {
    public:
        UdpCommunication() = default;
        ~UdpCommunication() {
            _socket.close();
        };
 
        UdpCommunication(asio::io_context &context, asio::ip::port_type const &port) : _context(context),
        _socket(_context, asio::ip::udp::endpoint(asio::ip::udp::v4(), port)) {};
 
        UdpCommunication(asio::io_context &context, asio::ip::port_type const &port, std::string const &connectionPort, std::string const &ip) : _context(context),
        _socket(_context, asio::ip::udp::endpoint(asio::ip::udp::v4(), port)) {
            asio::ip::udp::resolver resolver(context);
            asio::ip::udp::resolver::query query(asio::ip::udp::v4(), ip, connectionPort);
            asio::ip::udp::resolver::iterator iter = resolver.resolve(query);
            _endpoint = *iter;
        };
 
        void send(std::vector<byte> const &data) override {
            _socket.send_to(asio::buffer(data.data(), data.size()), _endpoint);
        };
 
        void send(std::vector<byte> const &data, asio::ip::address const &address, unsigned short const &port) override {
            setEnpointInfo(address, port);
            _socket.send_to(asio::buffer(data.data(), data.size()), _endpoint);
        };
 
        void async_send(std::vector<byte> const &data, std::function<void(std::error_code const &, std::size_t)> callBack) override {
            _socket.async_send_to(asio::buffer(data.data(), data.size()), _endpoint, callBack);
        };
 
        void async_send(std::vector<byte> const &data, std::function<void(std::error_code const &, std::size_t)> callBack, asio::ip::address const &address, unsigned short const &port) override {
            setEnpointInfo(address, port);
            _socket.async_send_to(asio::buffer(data.data(), data.size()), _endpoint, callBack);
        };
 
        std::pair<asio::ip::address, unsigned short> receive(std::vector<byte> &data) override {
            data.clear();
            data.resize(1500);
            _socket.receive_from(asio::buffer(data.data(), 1500), _endpoint);
            return getEnpointInfo();
        };
 
        void async_receive(std::vector<byte> &data, std::function<void(std::error_code const &, std::size_t)> callBack) override {
            data.clear();
            data.resize(1500);
            _socket.async_receive_from(asio::buffer(data.data(), 1500), _endpoint, callBack);
        };
 
        std::pair<asio::ip::address, unsigned short> getEnpointInfo() const override {
            return std::make_pair<asio::ip::address, unsigned short>(_endpoint.address(), _endpoint.port());
        }
 
        void setEnpointInfo(std::pair<asio::ip::address, unsigned short> const &endpointInfo) override {
            _endpoint.address(endpointInfo.first);
            _endpoint.port(endpointInfo.second);
        }
 
        void setEnpointInfo(asio::ip::address const &address, unsigned short const &port) override {
            _endpoint.address(address);
            _endpoint.port(port);
        }
 
    protected:
    private:
        asio::io_context &_context; 
        asio::ip::udp::socket _socket; 
        asio::ip::udp::endpoint _endpoint; 
};
 
 
#endif /* !UDPCOMMUNICATION_HPP_ */