mobile/mail/emailsfilterproxymodel.cpp

   1 /*
   2     Copyright (c) 2010 Tobias Koenig <tokoe@kde.org>
   3
   4     This library is free software; you can redistribute it and/or modify it
   5     under the terms of the GNU Library General Public License as published by
   6     the Free Software Foundation; either version 2 of the License, or (at your
   7     option) any later version.
   8
   9     This library is distributed in the hope that it will be useful, but WITHOUT
  10     ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11     FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Library General Public
  12     License for more details.
  13
  14     You should have received a copy of the GNU Library General Public License
  15     along with this library; see the file COPYING.LIB.  If not, write to the
  16     Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
  17     02110-1301, USA.
  18 */
  19
  20 #include "emailsfilterproxymodel.h"
  21
  22 #include <akonadi/entitytreemodel.h>
  23 #include <KMime/Message>
  24
  25 static bool emailMatchesFilter( const KMime::Message::Ptr &message, const QString &filterString );
  26
  27 using namespace Akonadi;
  28
  29 class EmailsFilterProxyModel::Private
  30 {
  31   public:
  32     QString mFilter;
  33 };
  34
  35 EmailsFilterProxyModel::EmailsFilterProxyModel( QObject *parent )
  36   : QSortFilterProxyModel( parent ), d( new Private )
  37 {
  38   setSortLocaleAware( true );
  39   setDynamicSortFilter( true );
  40 }
  41
  42 EmailsFilterProxyModel::~EmailsFilterProxyModel()
  43 {
  44   delete d;
  45 }
  46
  47 void EmailsFilterProxyModel::setFilterString( const QString &filter )
  48 {
  49   d->mFilter = filter;
  50   invalidateFilter();
  51 }
  52
  53 bool EmailsFilterProxyModel::filterAcceptsRow( int row, const QModelIndex &parent ) const
  54 {
  55   if ( d->mFilter.isEmpty() )
  56     return true;
  57
  58   const QModelIndex index = sourceModel()->index( row, 0, parent );
  59
  60   const Akonadi::Item item = index.data( Akonadi::EntityTreeModel::ItemRole ).value<Akonadi::Item>();
  61
  62   if ( item.hasPayload<KMime::Message::Ptr>() ) {
  63     const KMime::Message::Ptr message = item.payload<KMime::Message::Ptr>();
  64     return emailMatchesFilter( message, d->mFilter );
  65   }
  66
  67   return true;
  68 }
  69
  70 static bool emailMatchesFilter( const KMime::Message::Ptr &message, const QString &filterString )
  71 {
  72   if ( message->subject()->asUnicodeString().contains( filterString, Qt::CaseInsensitive ) )
  73     return true;
  74
  75   foreach ( const KMime::Types::Mailbox &mailbox, message->from()->mailboxes() ) {
  76     if ( mailbox.hasName() ) {
  77       if ( mailbox.name().contains( filterString, Qt::CaseInsensitive ) )
  78         return true;
  79     } else {
  80       if ( mailbox.addrSpec().asPrettyString().contains( filterString, Qt::CaseInsensitive ) )
  81         return true;
  82     }
  83   }
  84
  85   return false;
  86 }
  87
  88 #include "emailsfilterproxymodel.moc"