#include <errno.h>
#include <node.h>
#include <nan.h>
#include "./i2c-dev.h"
#include "./writeblock.h"
#include "./util.h"

#include <stdio.h>

static __s32 WriteBlock(int fd, __u8 cmd, __u8 length, const __u8 *block) {
  return i2c_smbus_write_block_data(fd, cmd, length, block);
}

class WriteBlockWorker : public I2cAsyncWorker {
public:
  WriteBlockWorker(
    Nan::Callback *callback,
    int fd,
    __u8 cmd,
    __u32 length,
    const __u8* block,
    v8::Local<v8::Object> &bufferHandle
  ) : I2cAsyncWorker(callback), fd(fd), cmd(cmd), length(length), block(block) {
    SaveToPersistent("buffer", bufferHandle);
  }

  ~WriteBlockWorker() {}

  void Execute() {
    __s32 ret = WriteBlock(fd, cmd, length, block);
    if (ret == -1) {
      SetErrorNo(errno);
      SetErrorSyscall("writeBlock");
    }
  }

  void HandleOKCallback() {
    Nan::HandleScope scope;

    v8::Local<v8::Value> bufferHandle = GetFromPersistent("buffer");

    v8::Local<v8::Value> argv[] = {
      Nan::Null(),
      Nan::New<v8::Integer>(length),
      bufferHandle
    };

    callback->Call(3, argv, async_resource);
  }

private:
  int fd;
  __u8 cmd;
  __u32 length;
  const __u8* block;
};

NAN_METHOD(WriteBlockAsync) {
  if (info.Length() < 5 ||
      !info[0]->IsInt32() ||
      !info[1]->IsInt32() ||
      !info[2]->IsUint32() ||
      !info[3]->IsObject() ||
      !info[4]->IsFunction()) {
    return Nan::ThrowError(Nan::ErrnoException(EINVAL, "writeBlock",
      "incorrect arguments passed to writeBlock"
      "(int fd, int cmd, int length, Buffer buffer, function cb)"));
  }

  int fd = Nan::To<int32_t>(info[0]).FromJust();
  __u8 cmd = Nan::To<int32_t>(info[1]).FromJust();
  __u32 length = Nan::To<uint32_t>(info[2]).FromJust();
  v8::Local<v8::Object> bufferHandle = info[3].As<v8::Object>();
  Nan::Callback *callback = new Nan::Callback(info[4].As<v8::Function>());

  const __u8* bufferData = (const __u8*) node::Buffer::Data(bufferHandle);
  size_t bufferLength = node::Buffer::Length(bufferHandle);

  if (length > I2C_SMBUS_I2C_BLOCK_MAX) {
    return Nan::ThrowError(Nan::ErrnoException(EINVAL, "writeBlock",
      "writeBlock can't write blocks with more than 32 bytes"));
  }

  if (length > bufferLength) {
    return Nan::ThrowError(Nan::ErrnoException(EINVAL, "writeBlock",
      "buffer passed to writeBlock contains less than 'length' bytes"));
  }

  Nan::AsyncQueueWorker(new WriteBlockWorker(
    callback,
    fd,
    cmd,
    length,
    bufferData,
    bufferHandle
  ));
}

NAN_METHOD(WriteBlockSync) {
  if (info.Length() < 4 ||
      !info[0]->IsInt32() ||
      !info[1]->IsInt32() ||
      !info[2]->IsUint32() ||
      !info[3]->IsObject()) {
    return Nan::ThrowError(Nan::ErrnoException(EINVAL, "writeBlockSync",
      "incorrect arguments passed to writeBlockSync"
      "(int fd, int cmd, int length, Buffer buffer)"));
  }

  int fd = Nan::To<int32_t>(info[0]).FromJust();
  __u8 cmd = Nan::To<int32_t>(info[1]).FromJust();
  __u32 length = Nan::To<uint32_t>(info[2]).FromJust();
  v8::Local<v8::Object> bufferHandle = info[3].As<v8::Object>();

  const __u8* bufferData = (const __u8*) node::Buffer::Data(bufferHandle);
  size_t bufferLength = node::Buffer::Length(bufferHandle);

  if (length > I2C_SMBUS_I2C_BLOCK_MAX) {
    return Nan::ThrowError(Nan::ErrnoException(EINVAL, "writeBlockSync",
      "writeBlockSync can't write blocks with more than 32 bytes"));
  }

  if (length > bufferLength) {
    return Nan::ThrowError(Nan::ErrnoException(EINVAL, "writeBlockSync",
      "buffer passed to writeBlockSync contains less than 'length' bytes"));
  }

  __s32 ret = WriteBlock(fd, cmd, length, bufferData);
  if (ret == -1) {
    return Nan::ThrowError(Nan::ErrnoException(errno, "writeBlockSync", ""));
  }
}