#include "lrcompletermodel.h"

#include <QDebug>

CompleterModel::CompleterModel(QObject* parent): QAbstractItemModel(parent)
{
    m_root.setModel(this);
}

QModelIndex CompleterModel::index(int row, int column, const QModelIndex& parent) const
{
    CompleterItem const* parentItem;
    if (!parent.isValid())
        parentItem = &m_root;
    else
        parentItem = static_cast<CompleterItem*>(parent.internalPointer());

    if ((parentItem == nullptr) || (row < 0) || (column < 0) || (row >= parentItem->rowCount())
        || (column >= 1)) {
        return QModelIndex();
    }
    return createIndex(row, column, parentItem->child(row));
}

QModelIndex CompleterModel::parent(const QModelIndex& child) const
{
    if (child.isValid()) {
        if (CompleterItem* childItem = static_cast<CompleterItem*>(child.internalPointer());
            childItem) {
            CompleterItem* parentItem = childItem->parent();
            if (parentItem != &m_root) {
                return indexFromItem(parentItem);
            }
        }
    }
    return QModelIndex();
}

int CompleterModel::rowCount(const QModelIndex& parent) const
{
    if (parent.isValid()) {
        CompleterItem* parentItem = static_cast<CompleterItem*>(parent.internalPointer());
        return parentItem->rowCount();
    }
    return m_root.rowCount();
}

int CompleterModel::columnCount(const QModelIndex& parent) const
{
    Q_UNUSED(parent)
    return 1;
}

QVariant CompleterModel::data(const QModelIndex& index, int role) const
{
    if (index.isValid()) {
        CompleterItem* item = static_cast<CompleterItem*>(index.internalPointer());
        switch (role) {
        case Qt::DisplayRole:
        case Qt::EditRole:
            if (!item)
                return QVariant();

            if (index.column() == 0) {
                return item->text();
            } else {
                return "";
            }
        case Qt::DecorationRole:
            if (!item)
                return QIcon();
            if (index.column() == 0) {
                return item->icon();
            } else
                return QIcon();
        default:
            return QVariant();
        }
    }
    return QVariant();
}

QList<CompleterItem*> CompleterModel::findItems(const QString& text, Qt::MatchFlags flags,
                                                int column) const
{
    QModelIndexList indexes
        = match(index(0, column, QModelIndex()), Qt::DisplayRole, text, -1, flags);
    QList<CompleterItem*> items;
    const int numIndexes = indexes.size();
    items.reserve(numIndexes);
    for (int i = 0; i < numIndexes; ++i)
        items.append(itemFromIndex(indexes.at(i)));
    return items;
}

void CompleterModel::clear()
{
    beginResetModel();
    m_root.clear();
    endResetModel();
}

CompleterItem* CompleterModel::itemFromIndex(const QModelIndex index) const
{
    if ((index.row() < 0) || (index.column() < 0) || (index.model() != this))
        return nullptr;
    CompleterItem* parent = static_cast<CompleterItem*>(index.internalPointer());
    if (parent == nullptr)
        return nullptr;
    CompleterItem* item = parent->child(index.row());
    return item;
}

QModelIndex CompleterModel::indexFromItem(CompleterItem* item) const
{
    if (item && item->parent()) {
        return createIndex(item->row(), 0, item);
    }
    return QModelIndex();
}

CompleterItem::~CompleterItem() { }

void CompleterItem::setIcon(const QIcon& newIcon) { m_icon = newIcon; }

void CompleterItem::setText(const QString& newText) { m_text = newText; }

void CompleterItem::appendRow(CompleterItem* child)
{
    child->m_parent = this;
    child->m_model = this->m_model;
    m_children.append(QSharedPointer<CompleterItem>(child));
    if (m_model) {
        QModelIndex start = m_model->indexFromItem(child);
        emit m_model->dataChanged(start, start);
    }
}

void CompleterItem::appendRows(const QList<CompleterItem*>& children)
{
    foreach (CompleterItem* item, children) {
        appendRow(item);
    }
}