#include "../include/DirectConnection.hpp"
#include <cstdio>
#include <cstring>

#ifndef WIN32
   #include <arpa/inet.h>
   #include <netdb.h>
#else
   #include <winsock2.h>
   #include <ws2tcpip.h>
#endif
#include <udt/udt.h>

namespace sibs
{
    Ipv4::Ipv4(const char *ip, u16 port)
    {        
        struct addrinfo hints = {};
        hints.ai_flags = AI_PASSIVE;
        hints.ai_family = AF_INET;
        hints.ai_socktype = SOCK_STREAM;
        
        char portStr[6];
        sprintf(portStr, "%d", port);
        int addrInfoResult = getaddrinfo(ip, portStr, &hints, &address);
        if(addrInfoResult != 0)
        {
            std::string errMsg = "Ip ";
            errMsg += ip;
            errMsg += " is not a valid ip";
            throw InvalidAddressException(errMsg);
        }
    }
    
    Ipv4::~Ipv4()
    {
        freeaddrinfo(address);
    }
    
    DirectConnections::DirectConnections() : 
        mySocket(0)
    {
        try
        {
            init();
        }
        catch(...)
        {
            cleanup();
        }
    }
    
    DirectConnections::~DirectConnections()
    {
        cleanup();
    }
    
    void DirectConnections::init()
    {
        UDT::startup();
        eid = UDT::epoll_create();
        mySocket = UDT::socket(AI_PASSIVE, AF_INET, SOCK_STREAM);
        bool rendezvous = true;
        UDT::setsockopt(mySocket, 0, UDT_RENDEZVOUS, &rendezvous, sizeof(bool));
        bool reuseAddr = true;
        UDT::setsockopt(mySocket, 0, UDT_REUSEADDR, &reuseAddr, sizeof(bool));
        
        // Windows UDP issue
        // For better performance, modify HKLM\System\CurrentControlSet\Services\Afd\Parameters\FastSendDatagramThreshold
        #ifdef WIN32
        int mss = 1052;
        UDT::setsockopt(socket, 0, UDT_MSS, &mss, sizeof(mss));
        #endif
        
        sockaddr_in myAddr = {};
        myAddr.sin_family = AF_INET;
        myAddr.sin_port = htons(9000);
        myAddr.sin_addr.s_addr = INADDR_ANY;
        memset(&myAddr.sin_zero, '\0', 8);

        if(UDT::bind(mySocket, (sockaddr*)&myAddr, sizeof(myAddr)) == UDT::ERROR)
        {
            // TODO: Add ip and port to error
            std::string errMsg = "UDT: Failed to bind, error: ";
            errMsg += UDT::getlasterror().getErrorMessage();
            throw ConnectionException(errMsg);
        }
    }
    
    void DirectConnections::cleanup()
    {
        UDT::epoll_release(eid);
        
        if(mySocket != 0)
            UDT::close(mySocket);
        
        UDT::cleanup();
    }
    
    void DirectConnections::connect(const Ipv4 &address)
    {
        if(UDT::connect(mySocket, address.address->ai_addr, address.address->ai_addrlen) == UDT::ERROR)
        {
            // TODO: Add ip and port to error
            std::string errMsg = "UDT: Failed to connect, error: ";
            errMsg += UDT::getlasterror().getErrorMessage();
            throw ConnectionException(errMsg);
        }
        
        //UDT::epoll_add_usock(eid, 2);
    } 
}