#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, "Vixen: 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();
}