#include "mdns.hpp"

#include <node_buffer.h>

#include "mdns_utils.hpp"
#include "txt_record_ref.hpp"

using namespace v8;
using namespace node;

namespace node_mdns {

size_t length(Handle<Value> v) {
    if (v->IsString()) {
        return v->ToString()->Utf8Length();
    } else if (Buffer::HasInstance(v)) {
        return Buffer::Length(v->ToObject());
    } else {
        return 0;
    }
}

NAN_METHOD(TXTRecordSetValue) {
    Nan::HandleScope scope;
    if (argumentCountMismatch(info, 3)) {
        info.GetReturnValue().Set(throwArgumentCountMismatchException(info, 3));
    }
    if ( ! info[0]->IsObject() || ! TxtRecordRef::HasInstance(info[0]->ToObject())) {
        info.GetReturnValue().Set(throwTypeError("argument 1 must be a TXTRecordRef object"));
    }
    TxtRecordRef * ref = Nan::ObjectWrap::Unwrap<TxtRecordRef>(info[0]->ToObject());

    if ( ! info[1]->IsString()) {
        info.GetReturnValue().Set(throwTypeError("argument 1 must be a string (key)"));
    }
    String::Utf8Value key(info[1]);

    if ( ! (info[2]->IsNull() || info[2]->IsUndefined() ||
        Buffer::HasInstance(info[2]) || info[2]->IsString())) {
        info.GetReturnValue().Set(throwTypeError("argument 1 must be null, undefined, a buffer or a string (value)"));
    }
    DNSServiceErrorType code = TXTRecordSetValue( & ref->GetTxtRecordRef(), *key,
            length(info[2]),
            ((info[2]->IsNull()||info[2]->IsUndefined()) 
                ? NULL : info[2]->IsString() ? *String::Utf8Value(info[2]->ToString()) : Buffer::Data(info[2]->ToObject())));

    if (code != kDNSServiceErr_NoError) {
        info.GetReturnValue().Set(throwMdnsError(code));
    }
    return;
}

} // end of namespace node_mdns