mirror of
https://bitbucket.org/smil3y/kde-playground.git
synced 2025-02-23 18:32:51 +00:00
146 lines
4 KiB
C++
146 lines
4 KiB
C++
/*
|
||
* This file is part of the Akonadi Mail example.
|
||
*
|
||
* Copyright 2009 Stephen Kelly <steveire@gmail.com>
|
||
*
|
||
* This library is free software; you can redistribute it and/or
|
||
* modify it under the terms of the GNU Lesser General Public
|
||
* License as published by the Free Software Foundation; either
|
||
* version 2.1 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
|
||
* Lesser General Public License for more details.
|
||
*
|
||
* You should have received a copy of the GNU Lesser General Public
|
||
* License along with this library; if not, write to the Free Software
|
||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
|
||
* 02110-1301 USA
|
||
*/
|
||
|
||
#include "mailmodel.h"
|
||
|
||
#include <kmime/kmime_message.h>
|
||
|
||
#include <boost/shared_ptr.hpp>
|
||
|
||
#include <KDebug>
|
||
|
||
typedef boost::shared_ptr<KMime::Message> MessagePtr;
|
||
|
||
using namespace Akonadi;
|
||
|
||
class MailModelPrivate
|
||
{
|
||
public:
|
||
MailModelPrivate(MailModel *model)
|
||
: q_ptr(model)
|
||
{
|
||
m_collectionHeaders << QLatin1String( "Folder" ) << QLatin1String( "Count" );
|
||
m_itemHeaders << QLatin1String( "Subject" ) << QLatin1String( "From" ) << QLatin1String( "Date" );
|
||
}
|
||
Q_DECLARE_PUBLIC(MailModel)
|
||
MailModel *q_ptr;
|
||
|
||
QStringList m_itemHeaders;
|
||
QStringList m_collectionHeaders;
|
||
|
||
};
|
||
|
||
MailModel::MailModel( ChangeRecorder *monitor, QObject *parent)
|
||
: EntityTreeModel(monitor, parent), d_ptr(new MailModelPrivate(this))
|
||
{
|
||
|
||
}
|
||
|
||
MailModel::~MailModel()
|
||
{
|
||
delete d_ptr;
|
||
}
|
||
|
||
QVariant MailModel::entityData(const Item &item, int column, int role) const
|
||
{
|
||
if (!item.hasPayload<MessagePtr>())
|
||
{
|
||
kWarning() << "Not a message" << item.id() << item.remoteId() << item.mimeType();
|
||
return QVariant();
|
||
}
|
||
const MessagePtr mail = item.payload<MessagePtr>();
|
||
if (role == Qt::DisplayRole)
|
||
{
|
||
switch (column)
|
||
{
|
||
case 0:
|
||
return mail->subject()->asUnicodeString();
|
||
case 1:
|
||
return mail->from()->asUnicodeString();
|
||
case 2:
|
||
return mail->date()->asUnicodeString();
|
||
}
|
||
} else if (role == Qt::ToolTipRole)
|
||
{
|
||
QString d;
|
||
d.append(QString::fromLatin1("Subject: %1\n").arg(mail->subject()->asUnicodeString()));
|
||
d.append(QString::fromLatin1("From: %1\n").arg(mail->from()->asUnicodeString()));
|
||
d.append(QString::fromLatin1("Date: %1\n").arg(mail->date()->asUnicodeString()));
|
||
return d;
|
||
}
|
||
return EntityTreeModel::entityData(item, column, role);
|
||
}
|
||
|
||
QVariant MailModel::entityData(const Collection &collection, int column, int role) const
|
||
{
|
||
if (role == Qt::DisplayRole)
|
||
{
|
||
switch (column)
|
||
{
|
||
case 0:
|
||
return EntityTreeModel::entityData(collection, column, role);
|
||
case 1:
|
||
{
|
||
const QModelIndex index = EntityTreeModel::modelIndexForCollection( this, collection );
|
||
Q_ASSERT( index.isValid() );
|
||
return rowCount(index );
|
||
}
|
||
default:
|
||
// Return a QString to pass modeltest.
|
||
return QString();
|
||
// return QVariant();
|
||
}
|
||
}
|
||
return EntityTreeModel::entityData(collection, column, role);
|
||
}
|
||
|
||
int MailModel::columnCount(const QModelIndex &index) const
|
||
{
|
||
Q_UNUSED(index);
|
||
return 3;
|
||
}
|
||
|
||
QVariant MailModel::entityHeaderData( int section, Qt::Orientation orientation, int role, HeaderGroup headerGroup ) const
|
||
{
|
||
Q_D(const MailModel);
|
||
|
||
if (orientation == Qt::Horizontal)
|
||
{
|
||
if (headerGroup == EntityTreeModel::CollectionTreeHeaders)
|
||
{
|
||
if (role == Qt::DisplayRole)
|
||
{
|
||
if (section >= d->m_collectionHeaders.size() )
|
||
return QVariant();
|
||
return d->m_collectionHeaders.at(section);
|
||
}
|
||
} else if (headerGroup == EntityTreeModel::ItemListHeaders)
|
||
{
|
||
if (role == Qt::DisplayRole)
|
||
{
|
||
if (section >= d->m_itemHeaders.size() )
|
||
return QVariant();
|
||
return d->m_itemHeaders.at(section);
|
||
}
|
||
}
|
||
}
|
||
return EntityTreeModel::entityHeaderData(section, orientation, role, headerGroup);
|
||
}
|