#include "QtWidgets/QStandardItemModel/qstandarditemmodel_wrap.h"

#include <QDebug>
#include <QWidget>

#include "Extras/Utils/nutils.h"
#include "QtCore/QObject/qobject_wrap.h"
#include "QtGui/QIcon/qicon_wrap.h"
#include "QtWidgets/QMenu/qmenu_wrap.h"
#include "QtWidgets/QStandardItem/qstandarditem_wrap.h"
#include "QtWidgets/QWidget/qwidget_wrap.h"

Napi::FunctionReference QStandardItemModelWrap::constructor;

Napi::Object QStandardItemModelWrap::init(Napi::Env env, Napi::Object exports) {
  Napi::HandleScope scope(env);
  char CLASSNAME[] = "QStandardItemModel";
  Napi::Function func = DefineClass(
      env, CLASSNAME,
      {InstanceMethod("item", &QStandardItemModelWrap::item),

       QOBJECT_WRAPPED_METHODS_EXPORT_DEFINE(QStandardItemModelWrap)});
  constructor = Napi::Persistent(func);
  exports.Set(CLASSNAME, func);
  QOBJECT_REGISTER_WRAPPER(QStandardItemModel, QStandardItemModelWrap);
  return exports;
}

QStandardItemModel* QStandardItemModelWrap::getInternalInstance() {
  return this->instance;
}

QStandardItemModelWrap::QStandardItemModelWrap(const Napi::CallbackInfo& info)
    : Napi::ObjectWrap<QStandardItemModelWrap>(info) {
  Napi::Env env = info.Env();
  size_t argCount = info.Length();
  if (argCount == 0) {
    // --- Construct a new instance
    this->instance = new NStandardItemModel();
  } else if (argCount == 1) {
    if (info[0].IsExternal()) {
      // --- Wrap a given C++ instance
      this->instance = info[0].As<Napi::External<QStandardItemModel>>().Data();
    } else {
      // --- Construct a new instance and pass a parent
      Napi::Object parentObject = info[0].As<Napi::Object>();
      NodeWidgetWrap* parentWidgetWrap =
          Napi::ObjectWrap<NodeWidgetWrap>::Unwrap(parentObject);
      this->instance =
          new NStandardItemModel(parentWidgetWrap->getInternalInstance());
    }
  } else {
    Napi::TypeError::New(env,
                         "NodeGui: QStandardItemModelWrap: Wrong number of "
                         "arguments to constructor")
        .ThrowAsJavaScriptException();
  }
  this->rawData = extrautils::configureQObject(this->getInternalInstance());
}

QStandardItemModelWrap::~QStandardItemModelWrap() {}

Napi::Value QStandardItemModelWrap::item(const Napi::CallbackInfo& info) {
  Napi::Env env = info.Env();
  int row = info[0].As<Napi::Number>().Int32Value();
  int column = info[1].As<Napi::Number>().Int32Value();

  QStandardItem* item = this->instance->item(row, column);
  if (item != nullptr) {
    Napi::Object itemWrap = QStandardItemWrap::constructor.New(
        {Napi::External<QStandardItem>::New(env, item),
         Napi::Boolean::New(env, true)});
    return itemWrap;
  } else {
    return env.Null();
  }
}