trashfilterproxymodel.cpp

/*
 Copyright (c) 2011 Christian Mollekopf <chrigi_1@fastmail.fm>

 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 "trashfilterproxymodel.h"

#include <akonadi/entity.h>
#include <akonadi/entitydeletedattribute.h>
#include <akonadi/item.h>
#include <akonadi/entitytreemodel.h>
#include <akonadi/resourcesettings.h>

using namespace Akonadi;

class TrashFilterProxyModel::TrashFilterProxyModelPrivate
{
public:
    TrashFilterProxyModelPrivate()
        : mTrashIsShown(false)
    {
    };
    bool showEntity(const Entity &) const;
    bool mTrashIsShown;
};

bool TrashFilterProxyModel::TrashFilterProxyModelPrivate::showEntity(const Akonadi::Entity &entity) const
{
    if (entity.hasAttribute<EntityDeletedAttribute>()) {
        return true;
    }
    if (entity.id() == Collection::root().id()) {
        return false;
    }
    return showEntity(entity.parentCollection());
}

TrashFilterProxyModel::TrashFilterProxyModel(QObject *parent)
    : KRecursiveFilterProxyModel(parent)
    , d_ptr(new TrashFilterProxyModelPrivate())
{

}

TrashFilterProxyModel::~TrashFilterProxyModel()
{
    delete d_ptr;
}

void TrashFilterProxyModel::showTrash(bool enable)
{
    Q_D(TrashFilterProxyModel);
    d->mTrashIsShown = enable;
    invalidateFilter();
}

bool TrashFilterProxyModel::trashIsShown() const
{
    Q_D(const TrashFilterProxyModel);
    return d->mTrashIsShown;
}

bool TrashFilterProxyModel::acceptRow(int sourceRow, const QModelIndex &sourceParent) const
{
    Q_D(const TrashFilterProxyModel);
    const QModelIndex &index = sourceModel()->index(sourceRow, 0, sourceParent);
    const Item &item = index.data(EntityTreeModel::ItemRole).value<Item>();
    //kDebug() << item.id();
    if (item.isValid()) {
        if (item.hasAttribute<EntityDeletedAttribute>()/* d->showEntity(item)*/) {
            return d->mTrashIsShown;
        }
    }
    const Collection &collection = index.data(EntityTreeModel::CollectionRole).value<Collection>();
    if (collection.isValid()) {
        if (collection.hasAttribute<EntityDeletedAttribute>()/*d->showEntity(collection) */) {
            return d->mTrashIsShown;
        }
    }
    return !d->mTrashIsShown;
}