libkdepim/maillistdrag.cpp

   1 /*
   2     This file is part of libkdepim.
   3
   4     Copyright (c) 2003 Don Sanders <sanders@kde.org>
   5     Copyright (c) 2005 George Staikos <staikos@kde.org>
   6     Copyright (c) 2005 Rafal Rzepecki <divide@users.sourceforge.net>
   7     Copyright (c) 2008 Thomas McGuire <mcguire@kde.org>
   8
   9     This library is free software; you can redistribute it and/or
  10     modify it under the terms of the GNU Library General Public
  11     License as published by the Free Software Foundation; either
  12     version 2 of the License, or (at your option) any later version.
  13
  14     This library is distributed in the hope that it will be useful,
  15     but WITHOUT ANY WARRANTY; without even the implied warranty of
  16     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  17     Library General Public License for more details.
  18
  19     You should have received a copy of the GNU Library General Public License
  20     along with this library; see the file COPYING.LIB.  If not, write to
  21     the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
  22     Boston, MA 02110-1301, USA.
  23 */
  24
  25 #include "maillistdrag.h"
  26
  27 #include <KDateTime>
  28 #include <KLocale>
  29 #include <KProgressDialog>
  30 #include <KUrl>
  31
  32 #include <QBuffer>
  33 #include <QDataStream>
  34 #include <QEventLoop>
  35 #include <QProgressBar>
  36 #include <QTextStream>
  37 #include <QDropEvent>
  38
  39 using namespace KPIM;
  40
  41
  42 // Have to define before use
  43 QDataStream& operator<< ( QDataStream &s, const MailSummary &d )
  44 {
  45   s << d.serialNumber();
  46   s << d.messageId();
  47   s << d.subject();
  48   s << d.from();
  49   s << d.to();
  50   KDateTime tempTime;
  51   tempTime.setTime_t( d.date() );
  52   s << tempTime.dateTime();
  53   return s;
  54 }
  55
  56 QDataStream& operator>> ( QDataStream &s, MailSummary &d )
  57 {
  58   quint32 serialNumber;
  59   QString messageId, subject, from, to;
  60   time_t date;
  61   s >> serialNumber;
  62   s >> messageId;
  63   s >> subject;
  64   s >> from;
  65   s >> to;
  66   QDateTime tempTime;
  67   s >> tempTime;
  68   date = KDateTime( tempTime ).toTime_t();
  69   d.set( serialNumber, messageId, subject, from, to, date );
  70   return s;
  71 }
  72
  73 QDataStream& operator<< ( QDataStream &s, const MailList &mailList )
  74 {
  75   MailList::const_iterator it;
  76   for (it = mailList.begin(); it != mailList.end(); ++it) {
  77     MailSummary mailDrag = *it;
  78     s << mailDrag;
  79   }
  80   return s;
  81 }
  82
  83 QDataStream& operator>> ( QDataStream &s, MailList &mailList )
  84 {
  85   mailList.clear();
  86   MailSummary mailDrag;
  87   while (!s.atEnd()) {
  88     s >> mailDrag;
  89     mailList.append( mailDrag );
  90   }
  91   return s;
  92 }
  93
  94 MailSummary::MailSummary( quint32 serialNumber, const QString &messageId,
  95                           const QString &subject, const QString &from, const QString &to,
  96                           time_t date )
  97     : mSerialNumber( serialNumber ), mMessageId( messageId ),
  98       mSubject( subject ), mFrom( from ), mTo( to ), mDate( date )
  99 {}
 100
 101 quint32 MailSummary::serialNumber() const
 102 {
 103     return mSerialNumber;
 104 }
 105
 106 QString MailSummary::messageId()  const
 107 {
 108     return mMessageId;
 109 }
 110
 111 QString MailSummary::subject() const
 112 {
 113     return mSubject;
 114 }
 115
 116 QString MailSummary::from() const
 117 {
 118     return mFrom;
 119 }
 120
 121 QString MailSummary::to() const
 122 {
 123     return mTo;
 124 }
 125
 126 time_t MailSummary::date() const
 127 {
 128     return mDate;
 129 }
 130
 131 void MailSummary::set( quint32 serialNumber, const QString &messageId,
 132                        const QString &subject, const QString &from, const QString &to, time_t date )
 133 {
 134     mSerialNumber = serialNumber;
 135     mMessageId = messageId;
 136     mSubject = subject;
 137     mFrom = from;
 138     mTo = to;
 139     mDate = date;
 140 }
 141
 142 #ifdef Q_CC_MSVC
 143 MailSummary::operator KUrl() const { return KUrl(); }
 144 #endif
 145
 146 QString MailList::mimeDataType()
 147 {
 148   return QLatin1String( "x-kmail-drag/message-list" );
 149 }
 150
 151 bool MailList::canDecode( const QMimeData *md )
 152 {
 153   return md->hasFormat( mimeDataType() );
 154 }
 155
 156 void MailList::populateMimeData( QMimeData *md )
 157 {
 158   /* We have three different possible mime types: x-kmail-drag/message-list, message/rfc822, and URL
 159      Add them in this order */
 160
 161   /* Popuplate the MimeData with the custom streaming x-kmail-drag/message-list mime type */
 162   if ( count() ) {
 163     QByteArray array;
 164     QBuffer buffer( &array, 0 );
 165     buffer.open( QIODevice::WriteOnly);
 166     QDataStream stream( &buffer );
 167     stream << (*this);
 168     buffer.close();
 169     md->setData( MailList::mimeDataType(), array );
 170   }
 171 }
 172
 173 MailList MailList::fromMimeData( const QMimeData *md )
 174 {
 175   if ( canDecode(md) ) {
 176     return decode( md->data( mimeDataType() ) );
 177   } else {
 178     return MailList();
 179   }
 180 }
 181
 182 MailList MailList::decode( const QByteArray& payload )
 183 {
 184   MailList mailList;
 185   // A read-only data stream
 186   QDataStream stream( payload );
 187   if ( payload.size() ) {
 188     stream >> mailList;
 189   }
 190   return mailList;
 191 }
 192
 193 QByteArray MailList::serialsFromMimeData( const QMimeData *md )
 194 {
 195   MailList mailList = fromMimeData( md );
 196   if ( mailList.count() ) {
 197     MailList::iterator it;
 198     QByteArray a;
 199     QBuffer buffer( &a );
 200     buffer.open( QIODevice::WriteOnly );
 201     QDataStream stream( &buffer );
 202     for (it = mailList.begin(); it != mailList.end(); ++it) {
 203       MailSummary mailDrag = *it;
 204       stream << mailDrag.serialNumber();
 205     }
 206     buffer.close();
 207     return a;
 208   } else {
 209     return QByteArray();
 210   }
 211 }
 212
 213 MailListMimeData::MailListMimeData( MailTextSource *src )
 214   : mMailTextSource( src )
 215 {
 216 }
 217
 218 MailListMimeData::~MailListMimeData()
 219 {
 220   delete mMailTextSource;
 221   mMailTextSource = 0;
 222 }
 223
 224 bool MailListMimeData::hasFormat ( const QString & mimeType ) const
 225 {
 226   if ( mimeType == QLatin1String( "message/rfc822" ) && mMailTextSource )
 227     return true;
 228   else
 229     return QMimeData::hasFormat( mimeType );
 230 }
 231
 232 QStringList MailListMimeData::formats () const
 233 {
 234   QStringList theFormats = QMimeData::formats();
 235   if ( mMailTextSource )
 236     theFormats.prepend( QLatin1String( "message/rfc822" ) );
 237   return theFormats;
 238 }
 239
 240 QVariant MailListMimeData::retrieveData( const QString & mimeType,
 241                                          QVariant::Type type ) const
 242 {
 243   if ( ( mimeType == QLatin1String( "message/rfc822" ) ) && mMailTextSource ) {
 244
 245     if ( mMails.isEmpty() ) {
 246       MailList list = MailList::fromMimeData( this );
 247       KProgressDialog *dlg = new KProgressDialog( 0, QString(),
 248                                    i18n("Retrieving and storing messages...") );
 249       dlg->setWindowModality( Qt::WindowModal );
 250       dlg->setAllowCancel( true );
 251       dlg->progressBar()->setMaximum( list.size() );
 252       int i = 0;
 253       dlg->progressBar()->setValue( i );
 254       dlg->show();
 255
 256       for ( MailList::ConstIterator it = list.constBegin(); it != list.constEnd(); ++it ) {
 257
 258         // Get the serial number from the mail summary and use the mail text source
 259         // to get the actual text of the mail.
 260         MailSummary mailSummary = *it;
 261         mMails.append( mMailTextSource->text( mailSummary.serialNumber() ) );
 262         if ( dlg->wasCancelled() ) {
 263           break;
 264         }
 265         dlg->progressBar()->setValue(++i);
 266 #ifdef __GNUC__
 267 #warning Port me!
 268 #endif
 269         //kapp->eventLoop()->processEvents(QEventLoop::ExcludeSocketNotifiers);
 270       }
 271       delete dlg;
 272     }
 273     return mMails;
 274   }
 275   else
 276     return QMimeData::retrieveData( mimeType, type );
 277 }