kde-playground/kdepim/examples/coisceim/tripcomponentfactory.cpp
2015-04-14 21:49:29 +00:00

89 lines
3.1 KiB
C++

/*
This file is part of Akonadi.
Copyright (c) 2011 Stephen Kelly <steveire@gmail.com>
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.
This program 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 General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301,
USA.
*/
#include "tripcomponentfactory.h"
#include <Akonadi/EntityTreeModel>
#include <Akonadi/ChangeRecorder>
#include <Akonadi/ItemFetchScope>
#include <Akonadi/EntityMimeTypeFilterModel>
#include <KMime/Message>
#include <KCalCore/Todo>
#include "note.h"
#include "todocheckableproxymodel.h"
#include "mixedtreemodel.h"
Akonadi::ChangeRecorder* TripComponentFactory::createMailChangeRecorder(QObject *parent)
{
Akonadi::ChangeRecorder* chRec = new Akonadi::ChangeRecorder(parent);
chRec->setMimeTypeMonitored(KMime::Message::mimeType());
chRec->itemFetchScope().fetchFullPayload(true);
return chRec;
}
Akonadi::ChangeRecorder* TripComponentFactory::createNotesChangeRecorder(QObject *parent)
{
Akonadi::ChangeRecorder* chRec = new Akonadi::ChangeRecorder(parent);
chRec->setMimeTypeMonitored(Akonotes::Note::mimeType());
chRec->itemFetchScope().fetchFullPayload(true);
return chRec;
}
Akonadi::ChangeRecorder* TripComponentFactory::createTodoChangeRecorder(QObject *parent)
{
Akonadi::ChangeRecorder* chRec = new Akonadi::ChangeRecorder(parent);
chRec->setMimeTypeMonitored(KCalCore::Todo::todoMimeType());
chRec->itemFetchScope().fetchFullPayload(true);
return chRec;
}
QAbstractItemModel* TripComponentFactory::createMailModel(Akonadi::ChangeRecorder *changeRecorder)
{
MixedTreeModel *model = new MixedTreeModel(changeRecorder);
model->setCollectionFetchStrategy(MixedTreeModel::FetchNoCollections);
return model;
}
QAbstractItemModel* TripComponentFactory::createNotesModel(Akonadi::ChangeRecorder *changeRecorder)
{
MixedTreeModel *model = new MixedTreeModel(changeRecorder);
model->setCollectionFetchStrategy(MixedTreeModel::FetchNoCollections);
return model;
}
QAbstractItemModel* TripComponentFactory::createTodoModel(Akonadi::ChangeRecorder *changeRecorder)
{
MixedTreeModel *model = new MixedTreeModel(changeRecorder);
model->setCollectionFetchStrategy(MixedTreeModel::FetchNoCollections);
Akonadi::EntityMimeTypeFilterModel *filterModel = new Akonadi::EntityMimeTypeFilterModel(changeRecorder);
filterModel->addMimeTypeInclusionFilter(KCalCore::Todo::todoMimeType());
filterModel->setHeaderGroup(MixedTreeModel::ItemListHeaders);
filterModel->setSourceModel(model);
TodoCheckableProxyModel *checkableProxy = new TodoCheckableProxyModel(model);
checkableProxy->setSourceModel(filterModel);
return checkableProxy;
}