#include "QtWidgets/QTreeWidget/qtreewidget_wrap.h"

#include <QWidget>

#include "Extras/Utils/nutils.h"
#include "QtWidgets/QTreeWidgetItem/qtreewidgetitem_wrap.h"
#include "QtWidgets/QWidget/qwidget_wrap.h"

Napi::FunctionReference QTreeWidgetWrap::constructor;

Napi::Object QTreeWidgetWrap::init(Napi::Env env, Napi::Object exports) {
  Napi::HandleScope scope(env);
  char CLASSNAME[] = "QTreeWidget";
  Napi::Function func = DefineClass(
      env, CLASSNAME,
      {InstanceMethod("addTopLevelItem", &QTreeWidgetWrap::addTopLevelItem),
       InstanceMethod("addTopLevelItems", &QTreeWidgetWrap::addTopLevelItems),
       InstanceMethod("insertTopLevelItem",
                      &QTreeWidgetWrap::insertTopLevelItem),
       InstanceMethod("insertTopLevelItems",
                      &QTreeWidgetWrap::insertTopLevelItems),
       InstanceMethod("selectedItems", &QTreeWidgetWrap::selectedItems),
       InstanceMethod("setColumnCount", &QTreeWidgetWrap::setColumnCount),
       InstanceMethod("setColumnWidth", &QTreeWidgetWrap::setColumnWidth),
       InstanceMethod("setHeaderLabel", &QTreeWidgetWrap::setHeaderLabel),
       InstanceMethod("setHeaderLabels", &QTreeWidgetWrap::setHeaderLabels),
       InstanceMethod("setItemWidget", &QTreeWidgetWrap::setItemWidget),
       InstanceMethod("currentItem", &QTreeWidgetWrap::currentItem),
       InstanceMethod("findItems", &QTreeWidgetWrap::findItems),
       InstanceMethod("takeTopLevelItem", &QTreeWidgetWrap::takeTopLevelItem),
       InstanceMethod("clear", &QTreeWidgetWrap::clear),
       QWIDGET_WRAPPED_METHODS_EXPORT_DEFINE(QTreeWidgetWrap)});
  constructor = Napi::Persistent(func);
  exports.Set(CLASSNAME, func);
  QOBJECT_REGISTER_WRAPPER(QTreeWidget, QTreeWidgetWrap);
  return exports;
}

QTreeWidget* QTreeWidgetWrap::getInternalInstance() { return this->instance; }

QTreeWidgetWrap::QTreeWidgetWrap(const Napi::CallbackInfo& info)
    : Napi::ObjectWrap<QTreeWidgetWrap>(info) {
  Napi::Env env = info.Env();
  size_t argCount = info.Length();
  if (argCount == 0) {
    // --- Construct a new instance
    this->instance = new NTreeWidget();
  } else if (argCount == 1) {
    if (info[0].IsExternal()) {
      // --- Wrap a given C++ instance
      this->instance = info[0].As<Napi::External<QTreeWidget>>().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 NTreeWidget(parentWidgetWrap->getInternalInstance());
    }
  } else {
    Napi::TypeError::New(
        env,
        "NodeGui: QTreeWidgetWrap: Wrong number of arguments to constructor")
        .ThrowAsJavaScriptException();
  }
  this->rawData =
      extrautils::configureQWidget(this->getInternalInstance(), true);
}

QTreeWidgetWrap::~QTreeWidgetWrap() { extrautils::safeDelete(this->instance); }

Napi::Value QTreeWidgetWrap::addTopLevelItem(const Napi::CallbackInfo& info) {
  Napi::Env env = info.Env();
  Napi::Object itemObject = info[0].As<Napi::Object>();
  QTreeWidgetItemWrap* itemWrap =
      Napi::ObjectWrap<QTreeWidgetItemWrap>::Unwrap(itemObject);
  QTreeWidgetItem* item = itemWrap->getInternalInstance();
  this->instance->addTopLevelItem(item);
  return env.Null();
}

Napi::Value QTreeWidgetWrap::addTopLevelItems(const Napi::CallbackInfo& info) {
  Napi::Env env = info.Env();
  Napi::Array itemsNapi = info[0].As<Napi::Array>();
  QList<QTreeWidgetItem*> items;
  for (int i = 0; i < itemsNapi.Length(); i++) {
    Napi::Value itemNapi = itemsNapi[i];
    Napi::Object itemObject = itemNapi.As<Napi::Object>();
    QTreeWidgetItemWrap* itemWrap =
        Napi::ObjectWrap<QTreeWidgetItemWrap>::Unwrap(itemObject);
    QTreeWidgetItem* item = itemWrap->getInternalInstance();
    items.append(item);
  }
  this->instance->addTopLevelItems(items);

  return env.Null();
}

Napi::Value QTreeWidgetWrap::insertTopLevelItem(
    const Napi::CallbackInfo& info) {
  Napi::Env env = info.Env();
  int index = info[0].As<Napi::Number>().Int32Value();
  Napi::Object itemObject = info[1].As<Napi::Object>();
  QTreeWidgetItemWrap* itemWrap =
      Napi::ObjectWrap<QTreeWidgetItemWrap>::Unwrap(itemObject);
  QTreeWidgetItem* item = itemWrap->getInternalInstance();
  this->instance->insertTopLevelItem(index, item);

  return env.Null();
}

Napi::Value QTreeWidgetWrap::insertTopLevelItems(
    const Napi::CallbackInfo& info) {
  Napi::Env env = info.Env();
  int index = info[0].As<Napi::Number>().Int32Value();
  Napi::Array itemsNapi = info[1].As<Napi::Array>();
  QList<QTreeWidgetItem*> items;
  for (int i = 0; i < itemsNapi.Length(); i++) {
    Napi::Value itemNapi = itemsNapi[i];
    Napi::Object itemObject = itemNapi.As<Napi::Object>();
    QTreeWidgetItemWrap* itemWrap =
        Napi::ObjectWrap<QTreeWidgetItemWrap>::Unwrap(itemObject);
    QTreeWidgetItem* item = itemWrap->getInternalInstance();
    items.append(item);
  }
  this->instance->insertTopLevelItems(index, items);

  return env.Null();
}

Napi::Value QTreeWidgetWrap::selectedItems(const Napi::CallbackInfo& info) {
  Napi::Env env = info.Env();
  QList<QTreeWidgetItem*> items = this->instance->selectedItems();
  Napi::Array napiItems = Napi::Array::New(env, items.size());
  for (int i = 0; i < items.size(); i++) {
    QTreeWidgetItem* item = items[i];
    // disable deletion of the native instance for these by passing true
    Napi::Object val = QTreeWidgetItemWrap::constructor.New(
        {Napi::External<QTreeWidgetItem>::New(env, item),
         Napi::Boolean::New(env, true)});
    napiItems[i] = val;
  }

  return napiItems;
}

Napi::Value QTreeWidgetWrap::setColumnCount(const Napi::CallbackInfo& info) {
  Napi::Env env = info.Env();
  int columns = info[0].As<Napi::Number>().Int32Value();
  this->instance->setColumnCount(columns);

  return env.Null();
}

Napi::Value QTreeWidgetWrap::setColumnWidth(const Napi::CallbackInfo& info) {
  Napi::Env env = info.Env();
  int columns = info[0].As<Napi::Number>().Int32Value();
  int width = info[1].As<Napi::Number>().Int32Value();
  this->instance->setColumnWidth(columns, width);

  return env.Null();
}

Napi::Value QTreeWidgetWrap::setHeaderLabel(const Napi::CallbackInfo& info) {
  Napi::Env env = info.Env();
  Napi::String napiLabel = info[0].As<Napi::String>();
  if (napiLabel.IsEmpty()) {
    Napi::TypeError::New(env, "Label must be specified")
        .ThrowAsJavaScriptException();
  }

  std::string label = napiLabel.Utf8Value();
  if (QTreeWidgetItem* header = this->instance->headerItem()) {
    header->setText(0, QString::fromUtf8(label.c_str()));
  } else {
    this->instance->setHeaderLabel(QString::fromUtf8(label.c_str()));
  }

  return env.Null();
}

Napi::Value QTreeWidgetWrap::setHeaderLabels(const Napi::CallbackInfo& info) {
  Napi::Env env = info.Env();
  Napi::Array napiLabelArray = info[0].As<Napi::Array>();
  QStringList headerLabels;
  for (int i = 0; i < napiLabelArray.Length(); i++) {
    Napi::Value label = napiLabelArray[i];
    headerLabels.push_back(label.As<Napi::String>().Utf8Value().c_str());
  }

  this->instance->setHeaderLabels(headerLabels);

  return env.Null();
}

Napi::Value QTreeWidgetWrap::setItemWidget(const Napi::CallbackInfo& info) {
  Napi::Env env = info.Env();
  Napi::Object itemObject = info[0].As<Napi::Object>();
  QTreeWidgetItemWrap* itemWidgetWrap =
      Napi::ObjectWrap<QTreeWidgetItemWrap>::Unwrap(itemObject);
  QTreeWidgetItem* item = itemWidgetWrap->getInternalInstance();

  int column = info[1].As<Napi::Number>().Int32Value();

  Napi::Object widgetObject = info[2].As<Napi::Object>();
  NodeWidgetWrap* widgetWrap =
      Napi::ObjectWrap<NodeWidgetWrap>::Unwrap(widgetObject);
  QWidget* widget = widgetWrap->getInternalInstance();

  this->instance->setItemWidget(item, column, widget);

  return env.Null();
}

Napi::Value QTreeWidgetWrap::currentItem(const Napi::CallbackInfo& info) {
  Napi::Env env = info.Env();
  QTreeWidgetItem* currentItem = this->instance->currentItem();
  if (currentItem != nullptr) {
    Napi::Object value = QTreeWidgetItemWrap::constructor.New(
        {Napi::External<QTreeWidgetItem>::New(
            env, new QTreeWidgetItem(*currentItem))});

    return value;
  } else {
    return env.Null();
  }
}

Napi::Value QTreeWidgetWrap::findItems(const Napi::CallbackInfo& info) {
  Napi::Env env = info.Env();
  Napi::String napiText = info[0].As<Napi::String>();
  std::string text = napiText.Utf8Value();
  int flag = info[1].As<Napi::Number>().Int32Value();
  int column = info[2].As<Napi::Number>().Int32Value();
  QList<QTreeWidgetItem*> items = this->instance->findItems(
      QString::fromUtf8(text.c_str()), Qt::MatchFlags(flag), column);
  Napi::Array napiItems = Napi::Array::New(env, items.size());
  for (int i = 0; i < items.size(); i++) {
    QTreeWidgetItem* item = items[i];
    // disable deletion of the native instance for these by passing true
    Napi::Object val = QTreeWidgetItemWrap::constructor.New(
        {Napi::External<QTreeWidgetItem>::New(env, item),
         Napi::Boolean::New(env, true)});
    napiItems[i] = val;
  }
  return napiItems;
}

Napi::Value QTreeWidgetWrap::takeTopLevelItem(const Napi::CallbackInfo& info) {
  Napi::Env env = info.Env();
  int index = info[0].As<Napi::Number>().Int32Value();

  QTreeWidgetItem* itemRemoved = this->instance->takeTopLevelItem(index);

  if (itemRemoved != nullptr) {
    Napi::Object value = QTreeWidgetItemWrap::constructor.New(
        {Napi::External<QTreeWidgetItem>::New(
            env, new QTreeWidgetItem(*itemRemoved))});

    return value;
  } else {
    return env.Null();
  }
}

Napi::Value QTreeWidgetWrap::clear(const Napi::CallbackInfo& info) {
  Napi::Env env = info.Env();
  this->instance->clear();
  return env.Null();
}