diff --git a/framework/src/entitymodel.cpp b/framework/src/entitymodel.cpp index 4c79f45d..9b61cb67 100644 --- a/framework/src/entitymodel.cpp +++ b/framework/src/entitymodel.cpp @@ -1,239 +1,243 @@ /* Copyright (c) 2018 Christian Mollekopf This library is free software; you can redistribute it and/or modify it under the terms of the GNU Library General Public License as published by the Free Software Foundation; either version 2 of the License, or (at your option) any later version. This library is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Library General Public License for more details. You should have received a copy of the GNU Library General Public License along with this library; see the file COPYING.LIB. If not, write to the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. */ #include "entitymodel.h" #include #include using namespace Sink; using namespace Sink::ApplicationDomain; enum Roles { IdRole = Qt::UserRole + 1, ObjectRole, LastRole }; EntityModel::EntityModel(QObject *parent) : QSortFilterProxyModel(parent) { setDynamicSortFilter(true); - sort(0, Qt::AscendingOrder); setFilterCaseSensitivity(Qt::CaseInsensitive); + setSortCaseSensitivity(Qt::CaseInsensitive); } EntityModel::~EntityModel() { } QHash EntityModel::roleNames() const { return mRoleNames; } QVariant EntityModel::data(const QModelIndex &idx, int role) const { auto srcIdx = mapToSource(idx); auto entity = srcIdx.data(Sink::Store::DomainObjectBaseRole).value(); const auto roleName = mRoleNames.value(role); if (roleName == "identifier") { return entity->identifier(); } else if (roleName == "object") { return QVariant::fromValue(entity); } else { return entity->getProperty(roleName); } } +bool EntityModel::lessThan(const QModelIndex &sourceLeft, const QModelIndex &sourceRight) const +{ + auto left = sourceLeft.data(Sink::Store::DomainObjectBaseRole).value(); + auto right = sourceRight.data(Sink::Store::DomainObjectBaseRole).value(); + const auto leftProperty = left->getProperty(mSortRole.toUtf8()).toString(); + const auto rightProperty = right->getProperty(mSortRole.toUtf8()).toString(); + return leftProperty < rightProperty; +} + void EntityModel::runQuery(const Query &query) { if (mType == "calendar") { mModel = Store::loadModel(query); } else if (mType == "addressbook") { mModel = Store::loadModel(query); } else { qWarning() << "Type not supported " << mType; Q_ASSERT(false); } setSourceModel(mModel.data()); } void EntityModel::updateQuery() { if (mType.isEmpty() || mRoles.isEmpty()) { return; } Query query; if (!mAccountId.isEmpty()) { query.resourceFilter(mAccountId.toUtf8()); } query.setFlags(Sink::Query::LiveQuery | Sink::Query::UpdateStatus); for (const auto &property: mRoles.keys()) { query.requestedProperties << property; } runQuery(query); } void EntityModel::setAccountId(const QString &accountId) { //Get all folders of an account mAccountId = accountId; updateQuery(); } QString EntityModel::accountId() const { return {}; } void EntityModel::setType(const QString &type) { mType = type; updateQuery(); } QString EntityModel::type() const { return {}; } void EntityModel::setRoles(const QStringList &roles) { mRoleNames.clear(); mRoleNames.insert(IdRole, "identifier"); mRoleNames.insert(ObjectRole, "object"); int role = LastRole; for (int i = 0; i < roles.size(); i++) { mRoleNames.insert(role++, roles.at(i).toLatin1()); } mRoles.clear(); for (const auto &r : mRoleNames.keys()) { mRoles.insert(mRoleNames.value(r), r); } - if (!mSortRole.isEmpty()) { - QSortFilterProxyModel::setSortRole(mRoles.value(mSortRole.toUtf8())); - } updateQuery(); } QStringList EntityModel::roles() const { // return mRoleNames.values(); return {}; } void EntityModel::setFilter(const QVariantMap &) { //TODO } QVariantMap EntityModel::filter() const { return {}; } void EntityModel::setSortRole(const QString &sortRole) { mSortRole = sortRole; - if (!mRoles.isEmpty()) { - QSortFilterProxyModel::setSortRole(mRoles.value(sortRole.toUtf8())); - } + sort(0, Qt::AscendingOrder); } QString EntityModel::sortRole() const { return mSortRole; } QVariantMap EntityModel::data(int row) const { QVariantMap map; for (const auto &r : mRoleNames.keys()) { map.insert(mRoleNames.value(r), data(index(row, 0), r)); } return map; } bool EntityModel::setData(const QModelIndex &index, const QVariant &value, int role) { if (!mRoleNames.contains(role)) { return false; } const auto entity = EntityModel::data(index, ObjectRole).value(); //FIXME hardcoding calendar is not a great idea here. Sink::ApplicationDomain::Calendar modifiedEntity{*entity}; const auto propertyName = mRoleNames.value(role); modifiedEntity.setProperty(propertyName, value.toBool()); //Ignore if we didn't modify anything. if (!modifiedEntity.changedProperties().isEmpty()) { Sink::Store::modify(modifiedEntity).exec(); } return true; } CheckableEntityModel::CheckableEntityModel(QObject *parent) : EntityModel(parent) { } CheckableEntityModel::~CheckableEntityModel() { } QHash CheckableEntityModel::roleNames() const { auto roleNames = EntityModel::roleNames(); roleNames.insert(Qt::CheckStateRole, "checked"); return roleNames; } QVariant CheckableEntityModel::data(const QModelIndex &index, int role) const { if (role == Qt::CheckStateRole) { return mCheckedEntities.contains(EntityModel::data(index, IdRole).toByteArray()); } return EntityModel::data(index, role); } bool CheckableEntityModel::setData(const QModelIndex &index, const QVariant &value, int role) { if (role == Qt::CheckStateRole) { auto identifier = EntityModel::data(index, IdRole).toByteArray(); if (value.toBool()) { mCheckedEntities.insert(identifier); } else { mCheckedEntities.remove(identifier); } emit checkedEntitiesChanged(); return true; } return EntityModel::setData(index, value, role); } QSet CheckableEntityModel::checkedEntities() const { return mCheckedEntities; } diff --git a/framework/src/entitymodel.h b/framework/src/entitymodel.h index f3e0c35b..380f5337 100644 --- a/framework/src/entitymodel.h +++ b/framework/src/entitymodel.h @@ -1,106 +1,109 @@ /* Copyright (c) 2018 Christian Mollekopf This library is free software; you can redistribute it and/or modify it under the terms of the GNU Library General Public License as published by the Free Software Foundation; either version 2 of the License, or (at your option) any later version. This library is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Library General Public License for more details. You should have received a copy of the GNU Library General Public License along with this library; see the file COPYING.LIB. If not, write to the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. */ #pragma once #include "kube_export.h" #include #include #include #include namespace Sink { class Query; } class KUBE_EXPORT EntityModel : public QSortFilterProxyModel { Q_OBJECT Q_PROPERTY (QString accountId READ accountId WRITE setAccountId) Q_PROPERTY (QString type READ type WRITE setType) Q_PROPERTY (QStringList roles READ roles WRITE setRoles) Q_PROPERTY (QString sortRole READ sortRole WRITE setSortRole) Q_PROPERTY (QVariantMap filter READ filter WRITE setFilter) public: enum Status { NoStatus, InProgressStatus, ErrorStatus, SuccessStatus, }; Q_ENUMS(Status) EntityModel(QObject *parent = Q_NULLPTR); virtual ~EntityModel(); virtual QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override; bool setData(const QModelIndex &index, const QVariant &value, int role = Qt::EditRole) override; virtual QHash roleNames() const override; void setAccountId(const QString &); QString accountId() const; void setType(const QString &); QString type() const; void setRoles(const QStringList &); QStringList roles() const; void setFilter(const QVariantMap &); QVariantMap filter() const; void setSortRole(const QString &); QString sortRole() const; Q_INVOKABLE QVariantMap data(int row) const; +protected: + bool lessThan(const QModelIndex &source_left, const QModelIndex &source_right) const override; + private: void runQuery(const Sink::Query &query); void updateQuery(); QSharedPointer mModel; QHash mRoleNames; QHash mRoles; QString mAccountId; QString mType; QString mSortRole; }; class KUBE_EXPORT CheckableEntityModel : public EntityModel { Q_OBJECT Q_PROPERTY (QSet checkedEntities READ checkedEntities NOTIFY checkedEntitiesChanged) public: CheckableEntityModel(QObject *parent = Q_NULLPTR); virtual ~CheckableEntityModel(); QHash roleNames() const override; QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override; bool setData(const QModelIndex &index, const QVariant &value, int role = Qt::EditRole) override; QSet checkedEntities() const; signals: void checkedEntitiesChanged(); private: QSet mCheckedEntities; };