#pragma once

#include <stdexcept>
#include <string>
#include <unordered_map>
#include "../env.hpp"
#include "../types.hpp"
#if OS_FAMILY != OS_FAMILY_WINDOWS
    #include <arpa/inet.h>
    #include <netdb.h>
#else
    #include <winsock2.h>
    #include <ws2tcpip.h>
    int inet_pton(int af, const char *src, void *dst);
    int inet_pton4(const char *src, void *dst);
    int inet_pton6(const char *src, void *dst);
    typedef u_short sa_family_t;
    typedef uint32_t in_addr_t;
#endif

namespace sibs
{
    class InvalidAddressException : public std::runtime_error
    {
    public:
        InvalidAddressException(const std::string &errMsg) : std::runtime_error(errMsg) {}
    };
    
    class Ipv4
    {
    public:
        Ipv4();
        // If @ip is nullptr, then bind to all available sockets (typical for servers)
        // Throws InvalidAddressException on error.
        Ipv4(const char *ip, unsigned short port);
        Ipv4(u16 family, u32 address, u16 port);
        Ipv4(const Ipv4 &other);
        Ipv4& operator = (const Ipv4 &other);
        
        std::string getAddress() const;
        unsigned short getPort() const;

        bool operator == (const Ipv4 &other) const;
        bool operator != (const Ipv4 &other) const;
        
        struct sockaddr_in address;
    };

    struct Ipv4Hasher
    {
        size_t operator()(const Ipv4 &address) const
        {
            return address.address.sin_addr.s_addr ^ address.address.sin_port;
        }
    };
    
    template <typename ValueType>
    using Ipv4Map = std::unordered_map<Ipv4, ValueType, Ipv4Hasher>;
}