/*
    SPDX-FileCopyrightText: 2016 Volker Krause <vkrause@kde.org>

    SPDX-License-Identifier: MIT
*/

#include "schemamodel.h"

#include <core/schemaentryelement.h>
#include <core/util.h>

#include <limits>

using namespace KUserFeedback::Console;

static const auto TOPLEVEL = std::numeric_limits<quintptr>::max();

SchemaModel::SchemaModel(QObject *parent) :
    QAbstractItemModel(parent)
{
}

SchemaModel::~SchemaModel() = default;

Product SchemaModel::product() const
{
    return m_product;
}

void SchemaModel::setProduct(const Product &product)
{
    beginResetModel();
    m_product = product;
    endResetModel();
}

void SchemaModel::addEntry(const QString &name)
{
    SchemaEntry entry;
    entry.setName(name);
    addEntry(entry);
}

void SchemaModel::addEntry(const SchemaEntry &entry)
{
    auto schema = m_product.schema();
    beginInsertRows(QModelIndex(), schema.size(), schema.size());
    schema.push_back(entry);
    m_product.setSchema(schema);
    endInsertRows();
}

void SchemaModel::addElement(const QModelIndex& parent, const QString& name)
{
    Q_ASSERT(parent.isValid());
    auto schema = m_product.schema();
    auto &entry = schema[parent.row()];
    auto elements = entry.elements();
    SchemaEntryElement e;
    e.setName(name);
    beginInsertRows(parent, elements.size(), elements.size());
    elements.push_back(e);
    entry.setElements(elements);
    m_product.setSchema(schema);
    endInsertRows();
}

void SchemaModel::deleteRow(const QModelIndex &idx)
{
    if (!idx.isValid())
        return;

    auto schema = m_product.schema();
    beginRemoveRows(idx.parent(), idx.row(), idx.row());
    if (idx.internalId() == TOPLEVEL) {
        schema.removeAt(idx.row());
    } else {
        auto &entry = schema[idx.internalId()];
        auto elements = entry.elements();
        elements.removeAt(idx.row());
        entry.setElements(elements);
    }
    m_product.setSchema(schema);
    endRemoveRows();
}

int SchemaModel::columnCount(const QModelIndex &parent) const
{
    Q_UNUSED(parent);
    return 2;
}

int SchemaModel::rowCount(const QModelIndex& parent) const
{
    if (!parent.isValid())
        return m_product.schema().size();
    if (parent.internalId() == TOPLEVEL && parent.column() == 0)
        return m_product.schema().at(parent.row()).elements().size();
    return 0;
}

QVariant SchemaModel::data(const QModelIndex& index, int role) const
{
    if (!index.isValid())
        return {};

    if (index.internalId() == TOPLEVEL) {
        const auto entry = m_product.schema().at(index.row());
        switch (index.column()) {
            case 0:
                if (role == Qt::DisplayRole || role == Qt::EditRole)
                    return entry.name();
                break;
            case 1:
                if (role == Qt::DisplayRole)
                    return Util::enumToString(entry.dataType());
                else if (role == Qt::EditRole)
                    return QVariant::fromValue(entry.dataType());
                break;
        }
    } else {
        const auto entry = m_product.schema().at(index.internalId());
        const auto elem = entry.elements().at(index.row());
        switch (index.column()) {
            case 0:
                if (role == Qt::DisplayRole || role == Qt::EditRole)
                    return elem.name();
                break;
            case 1:
                if (role == Qt::DisplayRole)
                    return Util::enumToString(elem.type());
                else if (role == Qt::EditRole)
                    return QVariant::fromValue(elem.type());
                break;
        }
    }

    return {};
}

QVariant SchemaModel::headerData(int section, Qt::Orientation orientation, int role) const
{
    if (orientation == Qt::Horizontal && role == Qt::DisplayRole) {
        switch (section) {
            case 0: return tr("Name");
            case 1: return tr("Type");
        }
    }
    return QAbstractItemModel::headerData(section, orientation, role);
}

Qt::ItemFlags SchemaModel::flags(const QModelIndex &index) const
{
    const auto baseFlags = QAbstractItemModel::flags(index);
    if (!index.isValid())
        return baseFlags;
    return baseFlags | Qt::ItemIsEditable;
}

bool SchemaModel::setData(const QModelIndex &index, const QVariant &value, int role)
{
    if (role != Qt::EditRole)
        return false;

    auto schema = m_product.schema();

    if (index.internalId() == TOPLEVEL) {
        auto &entry = schema[index.row()];
        switch (index.column()) {
            case 0:
                entry.setName(value.toString());
                break;
            case 1:
                entry.setDataType(value.value<SchemaEntry::DataType>());
                break;
        }
    } else {
        auto &entry = schema[index.internalId()];
        auto elems = entry.elements();
        auto &elem = elems[index.row()];
        switch (index.column()) {
            case 0:
                elem.setName(value.toString());
                break;
            case 1:
                elem.setType(value.value<SchemaEntryElement::Type>());
                break;
        }
        entry.setElements(elems);
    }

    m_product.setSchema(schema);

    Q_EMIT dataChanged(index, index);
    return false;
}

QModelIndex SchemaModel::index(int row, int column, const QModelIndex &parent) const
{
    if (!hasIndex(row, column, parent))
        return {};
    if (!parent.isValid())
        return createIndex(row, column, TOPLEVEL);
    if (parent.internalId() == TOPLEVEL)
        return createIndex(row, column, parent.row());
    return {};
}

QModelIndex SchemaModel::parent(const QModelIndex &index) const
{
    if (!index.isValid() || index.internalId() == TOPLEVEL)
        return {};
    return createIndex(index.internalId(), 0, TOPLEVEL);
}

#include "moc_schemamodel.cpp"