#include "MListModel.h"

/*!
 *  \brief      Retrieve the role names set for this list model.
 *  \return     Role names used for this model.
 */
QHash<int, QByteArray> View::MListModel::roleNames() const
{
    return _roleNames;
}

/*!
 *  \brief      Set the role names for the data in this list model.
 *  \param[in]  roleNames   New role names for this list model.
 *  \note       This will clear the list model of any data.
 */
void View::MListModel::setRoleNames(const QHash<int, QByteArray>& vRoleNames)
{
    if (_roleNames != vRoleNames)
    {
        clear();
        _roleNames = vRoleNames;
    }
}

/*!
 *  \brief      Get the number of rows in this list model.
 *  \return     Number of rows in this list model.
 */
int View::MListModel::rowCount(const QModelIndex &) const
{
    return _data.size();
}

/*!
 *  \brief      Retrieve the data stored at the given index for the specified role.
 *  \param[in]  index       Index of the data in the list model.
 *  \param[in]  role        Role of the data to fetch.
 *  \return     Data at index for role or Invalid if index or role are not valid.
 */
QVariant View::MListModel::data(const QModelIndex &vIndex, int vRole) const
{
    // check for invalid or out of bounds index
    if ( ! vIndex.isValid() || vIndex.row() >= rowCount()) {
        return QVariant();
    }

    // ensure the role is in the data and return it, otherwise return invalid
    return (_data[vIndex.row()].find(vRole) != _data[vIndex.row()].end())
               ? _data[vIndex.row()][vRole]
               : QVariant();
}

/*!
 *  \brief      Retrieve the data stored at the given index for the specified role and expose to QML
 *  \param[in]  vRow       Index of the data in the list model.
 */
QVariantMap View::MListModel::get(int vRow) const
{
    if (vRow < 0 || vRow >= rowCount()) {
        return {};
    }

    QHash<int, QByteArray> roles = roleNames();
    QVariantMap map;
    for (auto it = roles.begin(); it != roles.end(); ++it) {
        map[it.value()] = data(index(vRow, 0), it.key());
    }
    return map;
}

/*!
 *  \brief      Clear any data contained in this list model.
 */
void View::MListModel::clear() {
    beginRemoveRows(QModelIndex(), 0, rowCount());
    _data.clear();
    endRemoveRows();
}

/*!
 *  \brief      Sets the value of the data in this list model at the given index and for the given role.
 *  \param[in]  index   Location of data to update.
 *  \param[in]  value   Value to set.
 *  \param[in]  role    The role of the data at the given index to set.
 *  \return     Return true if data was set properly, otherwise false.
 */
bool View::MListModel::setData(const QModelIndex &vIndex, const QVariant& vValue, int vRole) {
    if (! vIndex.isValid() || vIndex.row() >= _data.count()) {
        return false;
    }

    if (_data[vIndex.row()][vRole] != vValue)
    {
        _data[vIndex.row()][vRole] = vValue;
        // explicitly emit a dataChanged signal to notify anybody bound to this property (vRole)
        emit dataChanged(vIndex, vIndex, QVector<int>(1, vRole));
    }

    return true;
}

/*!
 *  \brief      Assignment operator
 *  \return     A reference to this object.
 */
View::MListModel& View::MListModel::operator = (const QList<QHash<int, QVariant>> &src)
{
    clear();

    if (_data != src)
    {
        beginInsertRows(QModelIndex(), 0, src.count() - 1);
        _data = src;
        endInsertRows();
    }
    return *this;
}