akonadiconsole/akonadibrowsermodel.cpp

   1 /*
   2     Copyright (c) 2009 Stephen Kelly <steveire@gmail.com>
   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
  21 #include "akonadibrowsermodel.h"
  22
  23 #include <kmime/kmime_message.h>
  24
  25 #include <kabc/addressee.h>
  26 #include <kabc/contactgroup.h>
  27
  28 #include <kcalcore/incidence.h>
  29 #include <kcalcore/event.h>
  30
  31 #include <boost/shared_ptr.hpp>
  32
  33 typedef boost::shared_ptr<KMime::Message> MessagePtr;
  34 typedef QSharedPointer<KCalCore::Incidence> IncidencePtr;
  35
  36 class AkonadiBrowserModel::State
  37 {
  38 public:
  39   virtual ~State() {}
  40   QStringList m_collectionHeaders;
  41   QStringList m_itemHeaders;
  42
  43   virtual QVariant entityData( const Item &item, int column, int role ) const = 0;
  44
  45 };
  46
  47 class GenericState : public AkonadiBrowserModel::State
  48 {
  49 public:
  50   GenericState()
  51   {
  52     m_collectionHeaders << "Collection";
  53     m_itemHeaders << "Id" << "Remote Id" << "MimeType";
  54   }
  55   virtual ~GenericState() {}
  56
  57   QVariant entityData( const Item &item, int column, int role ) const
  58   {
  59     if (Qt::DisplayRole != role)
  60       return QVariant();
  61
  62     switch (column)
  63     {
  64     case 0:
  65       return item.id();
  66     case 1:
  67       return item.remoteId();
  68     case 2:
  69       return item.mimeType();
  70     }
  71     return QVariant();
  72   }
  73
  74
  75 };
  76
  77 class MailState : public AkonadiBrowserModel::State
  78 {
  79 public:
  80   MailState()
  81   {
  82     m_collectionHeaders << "Collection";
  83     m_itemHeaders << "Subject" << "Sender" << "Date";
  84   }
  85   virtual ~MailState() {}
  86
  87   QVariant entityData( const Item &item, int column, int role ) const
  88   {
  89     if (Qt::DisplayRole != role)
  90       return QVariant();
  91
  92     if (!item.hasPayload<MessagePtr>())
  93     {
  94       return QVariant();
  95     }
  96     const MessagePtr mail = item.payload<MessagePtr>();
  97
  98     switch (column)
  99     {
 100     case 0:
 101       return mail->subject()->asUnicodeString();
 102     case 1:
 103       return mail->from()->asUnicodeString();
 104     case 2:
 105       return mail->date()->asUnicodeString();
 106     }
 107
 108     return QVariant();
 109   }
 110
 111 };
 112
 113 class ContactsState : public AkonadiBrowserModel::State
 114 {
 115 public:
 116   ContactsState()
 117   {
 118     m_collectionHeaders << "Collection";
 119     m_itemHeaders << "Given Name" << "Family Name" << "Email";
 120   }
 121   virtual ~ContactsState() {}
 122
 123   QVariant entityData( const Item &item, int column, int role ) const
 124   {
 125     if (Qt::DisplayRole != role)
 126       return QVariant();
 127
 128     if ( !item.hasPayload<KABC::Addressee>() && !item.hasPayload<KABC::ContactGroup>() )
 129     {
 130       return QVariant();
 131     }
 132
 133     if ( item.hasPayload<KABC::Addressee>() )
 134     {
 135       const KABC::Addressee addr = item.payload<KABC::Addressee>();
 136
 137       switch (column)
 138       {
 139       case 0:
 140         return addr.givenName();
 141       case 1:
 142         return addr.familyName();
 143       case 2:
 144         return addr.preferredEmail();
 145       }
 146       return QVariant();
 147     }
 148     if ( item.hasPayload<KABC::ContactGroup>() ) {
 149
 150       switch (column)
 151       {
 152       case 0:
 153         const KABC::ContactGroup group = item.payload<KABC::ContactGroup>();
 154         return group.name();
 155       }
 156       return QVariant();
 157     }
 158     return QVariant();
 159   }
 160 };
 161
 162 class CalendarState : public AkonadiBrowserModel::State
 163 {
 164 public:
 165   CalendarState()
 166   {
 167     m_collectionHeaders << "Collection";
 168     m_itemHeaders << "Summary" << "DateTime start" << "DateTime End" << "Type";
 169   }
 170   virtual ~CalendarState() {}
 171
 172   QVariant entityData( const Item &item, int column, int role ) const
 173   {
 174     if (Qt::DisplayRole != role)
 175       return QVariant();
 176
 177     if ( !item.hasPayload<IncidencePtr>() )
 178     {
 179       return QVariant();
 180     }
 181     const IncidencePtr incidence = item.payload<IncidencePtr>();
 182     switch (column)
 183     {
 184     case 0:
 185       return incidence->summary();
 186       break;
 187     case 1:
 188       return incidence->dtStart().toString();
 189       break;
 190     case 2:
 191       return incidence->dateTime( KCalCore::Incidence::RoleEnd ).toString();
 192       break;
 193     case 3:
 194       return incidence->typeStr();
 195       break;
 196     default:
 197       break;
 198     }
 199     return QVariant();
 200   }
 201 };
 202
 203 AkonadiBrowserModel::AkonadiBrowserModel( ChangeRecorder* monitor, QObject* parent )
 204     : EntityTreeModel( monitor, parent ),
 205       m_itemDisplayMode( GenericMode )
 206 {
 207
 208   m_genericState = new GenericState();
 209   m_mailState = new MailState();
 210   m_contactsState = new ContactsState();
 211   m_calendarState = new CalendarState();
 212
 213   m_currentState = m_genericState;
 214 }
 215
 216 QVariant AkonadiBrowserModel::entityData( const Item &item, int column, int role ) const
 217 {
 218   QVariant var = m_currentState->entityData( item, column, role );
 219   if ( !var.isValid() )
 220   {
 221     if ( column < 1 )
 222       return EntityTreeModel::entityData( item, column, role );
 223     return QString();
 224   }
 225
 226   return var;
 227 }
 228
 229 QVariant AkonadiBrowserModel::entityData(const Akonadi::Collection& collection, int column, int role) const
 230 {
 231   return Akonadi::EntityTreeModel::entityData( collection, column, role );
 232 }
 233
 234 int AkonadiBrowserModel::entityColumnCount( HeaderGroup headerGroup ) const
 235 {
 236   if ( ItemListHeaders == headerGroup )
 237   {
 238     return m_currentState->m_itemHeaders.size();
 239   }
 240
 241   if ( CollectionTreeHeaders == headerGroup )
 242   {
 243     return m_currentState->m_collectionHeaders.size();
 244   }
 245   // Practically, this should never happen.
 246   return EntityTreeModel::entityColumnCount( headerGroup );
 247 }
 248
 249
 250 QVariant AkonadiBrowserModel::entityHeaderData( int section, Qt::Orientation orientation, int role, HeaderGroup headerGroup ) const
 251 {
 252   if ( section < 0 )
 253     return QVariant();
 254
 255   if ( orientation == Qt::Vertical )
 256      return EntityTreeModel::entityHeaderData( section, orientation, role, headerGroup );
 257
 258   if ( headerGroup == EntityTreeModel::CollectionTreeHeaders )
 259   {
 260     if ( role == Qt::DisplayRole )
 261     {
 262       if ( section >= m_currentState->m_collectionHeaders.size() )
 263         return QVariant();
 264       return m_currentState->m_collectionHeaders.at( section );
 265     }
 266   } else if ( headerGroup == EntityTreeModel::ItemListHeaders )
 267   {
 268     if ( role == Qt::DisplayRole )
 269     {
 270       if ( section >= m_currentState->m_itemHeaders.size() )
 271         return QVariant();
 272       return m_currentState->m_itemHeaders.at( section );
 273     }
 274   }
 275   return EntityTreeModel::entityHeaderData( section, orientation, role, headerGroup );
 276 }
 277
 278 AkonadiBrowserModel::ItemDisplayMode AkonadiBrowserModel::itemDisplayMode() const
 279 {
 280   return m_itemDisplayMode;
 281 }
 282
 283 void AkonadiBrowserModel::setItemDisplayMode( AkonadiBrowserModel::ItemDisplayMode itemDisplayMode )
 284 {
 285   beginResetModel();
 286   m_itemDisplayMode = itemDisplayMode;
 287   switch (itemDisplayMode)
 288   {
 289   case MailMode:
 290     m_currentState = m_mailState;
 291     break;
 292   case ContactsMode:
 293     m_currentState = m_contactsState;
 294     break;
 295   case CalendarMode:
 296     m_currentState = m_calendarState;
 297     break;
 298   case GenericMode:
 299   default:
 300     m_currentState = m_genericState;
 301     break;
 302   }
 303   endResetModel();
 304 }