#ifndef NODE_MDNS_NETWORK_INTERFACE_INCLUDED
#define NODE_MDNS_NETWORK_INTERFACE_INCLUDED

#include "mdns.hpp"

#include <v8.h>

#include "mdns_utils.hpp"

#ifdef NODE_MDNS_HAVE_INTERFACE_NAME_CONVERSION

# ifdef WIN32
#  include <netioapi.h>
# else
#  include <sys/types.h>
#  include <sys/socket.h>
#  include <net/if.h> // if_nametoindex()
# endif

using namespace v8;

namespace node_mdns {

NAN_METHOD(if_nametoindex) {
    Nan::HandleScope scope;
    if (argumentCountMismatch(info, 1)) {
        info.GetReturnValue().Set(throwArgumentCountMismatchException(info, 1));
    }
    if ( ! info[0]->IsString()) {
        info.GetReturnValue().Set(throwTypeError("argument 1 must be a string (interface name)"));
    }
    String::Utf8Value interfaceName(info[0]->ToString());

#ifdef WIN32
    DWORD aliasLength = MultiByteToWideChar(CP_UTF8, 0, *interfaceName, -1,
            NULL, 0);
    if (aliasLength == 0) {
        info.GetReturnValue().Set(throwError("failed to determine buffer size"));
    }

    wchar_t * alias = new wchar_t[aliasLength];
    if ( ! alias) {
        info.GetReturnValue().Set(throwError("failed to allocate alias buffer"));
    }

    if (MultiByteToWideChar(CP_UTF8, 0, *interfaceName, -1, alias,
                aliasLength) == 0)
    {
        delete [] alias;
        info.GetReturnValue().Set(throwError("failed to convert utf8 to unicode"));
    }

    NET_LUID luid;
    if (ConvertInterfaceAliasToLuid(alias, &luid) != NO_ERROR) {
        delete [] alias;
        info.GetReturnValue().Set(throwError("failed to convert interface alias to luid"));
    }

    delete [] alias;

    NET_IFINDEX index = 0;
    if (ConvertInterfaceLuidToIndex(&luid, &index) != NO_ERROR) {
        info.GetReturnValue().Set(throwError("failed to convert interface luid to index"));
    }
#else
    unsigned int index = ::if_nametoindex(*interfaceName);
#endif
    if (index == 0) {
        info.GetReturnValue().Set(throwError((std::string("interface '") + *interfaceName +
                    "' does not exist").c_str()));
    }
    info.GetReturnValue().Set( Nan::New<Integer>(static_cast<uint32_t>(index)));
}

NAN_METHOD(if_indextoname) {
    Nan::HandleScope scope;
    if (argumentCountMismatch(info, 1)) {
        info.GetReturnValue().Set(throwArgumentCountMismatchException(info, 1));
    }
    if ( ! info[0]->IsUint32()) {
        info.GetReturnValue().Set(throwTypeError("argument 1 must be a positive integer "
                "(interface index)"));
    }
#ifdef WIN32
    NET_LUID luid;
    if (ConvertInterfaceIndexToLuid(Nan::To<uint32_t>(info[0]), &luid) != NO_ERROR)
    {
        info.GetReturnValue().Set(throwError("failed to convert interface index to luid"));
    }
    enum { size = NDIS_IF_MAX_STRING_SIZE + 1 };
    wchar_t alias[size];
    if (ConvertInterfaceLuidToAlias(&luid, alias, size) != NO_ERROR) {
        info.GetReturnValue().Set(throwError("failed to convert interface luid to alias"));
    }
    int utf8Length = WideCharToMultiByte(CP_UTF8, 0, alias, -1,
            NULL, 0, NULL, NULL);
    if (utf8Length == 0) {
        info.GetReturnValue().Set(throwError("failed to determine buffer size"));
    }
    char * nameBuffer = new char[utf8Length];

    if (WideCharToMultiByte(CP_UTF8, 0, alias, -1, nameBuffer, utf8Length,
                NULL, NULL) == 0)
    {
        delete [] nameBuffer;
        info.GetReturnValue().Set(throwError("failed to convert unicode to utf8"));
    }
    Local<String> name = Nan::New(nameBuffer).ToLocalChecked();
    delete [] nameBuffer;
#else
    char nameBuffer[IFNAMSIZ];
    if ( ! ::if_indextoname(Nan::To<uint32_t>(info[0]).FromJust(), nameBuffer)) {
        info.GetReturnValue().Set(throwError("index has no corresponding interface"));
    }
    Local<String> name = Nan::New(nameBuffer).ToLocalChecked();
#endif
    info.GetReturnValue().Set(name);
}

} // end of namespace node_mdns

#endif // NODE_MDNS_HAVE_INTERFACE_NAME_CONVERSION

#endif // NODE_MDNS_NETWORK_INTERFACE_INCLUDED