src/Imap/Model/OneMessageModel.cpp

   1 /* Copyright (C) 2006 - 2014 Jan Kundrát <jkt@flaska.net>
   2
   3    This file is part of the Trojita Qt IMAP e-mail client,
   4    http://trojita.flaska.net/
   5
   6    This program is free software; you can redistribute it and/or
   7    modify it under the terms of the GNU General Public License as
   8    published by the Free Software Foundation; either version 2 of
   9    the License or (at your option) version 3 or any later version
  10    accepted by the membership of KDE e.V. (or its successor approved
  11    by the membership of KDE e.V.), which shall act as a proxy
  12    defined in Section 14 of version 3 of the license.
  13
  14    This program 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
  17    GNU General Public License for more details.
  18
  19    You should have received a copy of the GNU General Public License
  20    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  21 */
  22
  23 #include "OneMessageModel.h"
  24 #include "kdeui-itemviews/kdescendantsproxymodel.h"
  25 #include "FindInterestingPart.h"
  26 #include "ItemRoles.h"
  27 #include "Model.h"
  28 #include "SubtreeModel.h"
  29 #include "Imap/Network/MsgPartNetAccessManager.h"
  30
  31 namespace Imap
  32 {
  33 namespace Mailbox
  34 {
  35
  36 OneMessageModel::OneMessageModel(Model *model): QObject(model), m_subtree(0)
  37 {
  38     m_subtree = new SubtreeModelOfModel(this);
  39     m_subtree->setSourceModel(model);
  40     connect(m_subtree, &QAbstractItemModel::modelReset, this, &OneMessageModel::envelopeChanged);
  41     connect(m_subtree, &QAbstractItemModel::dataChanged, this, &OneMessageModel::envelopeChanged);
  42     connect(this, &OneMessageModel::envelopeChanged, this, &OneMessageModel::flagsChanged);
  43     connect(model, &QAbstractItemModel::dataChanged, this, &OneMessageModel::handleModelDataChanged);
  44     m_flatteningModel = new KDescendantsProxyModel(this);
  45     m_flatteningModel->setSourceModel(m_subtree);
  46     QHash<int, QByteArray> roleNames;
  47     roleNames[RoleIsFetched] = "isFetched";
  48     roleNames[RolePartMimeType] = "mimeType";
  49     roleNames[RolePartCharset] = "charset";
  50     roleNames[RolePartContentFormat] = "contentFormat";
  51     roleNames[RolePartEncoding] = "encoding";
  52     roleNames[RolePartBodyFldId] = "bodyFldId";
  53     roleNames[RolePartBodyDisposition] = "bodyDisposition";
  54     roleNames[RolePartFileName] = "fileName";
  55     roleNames[RolePartOctets] = "size";
  56     roleNames[RolePartId] = "partId";
  57     roleNames[RolePartPathToPart] = "pathToPart";
  58     m_flatteningModel->proxySetRoleNames(roleNames);
  59 }
  60
  61 void OneMessageModel::setMessage(const QString &mailbox, const uint uid)
  62 {
  63     m_mainPartUrl = QUrl(QStringLiteral("about:blank"));
  64     emit mainPartUrlChanged();
  65     QAbstractItemModel *abstractModel = qobject_cast<QAbstractItemModel*>(QObject::parent());
  66     Q_ASSERT(abstractModel);
  67     Model *model = qobject_cast<Model*>(abstractModel);
  68     Q_ASSERT(model);
  69     setMessage(model->messageIndexByUid(mailbox, uid));
  70 }
  71
  72 void OneMessageModel::setMessage(const QModelIndex &message)
  73 {
  74     Q_ASSERT(!message.isValid() || message.model() == QObject::parent());
  75     m_message = message;
  76     m_subtree->setRootItem(message);
  77
  78     // Now try to locate the interesting part of the current message
  79     QModelIndex idx;
  80     QString partMessage;
  81     FindInterestingPart::MainPartReturnCode res = FindInterestingPart::findMainPartOfMessage(m_message, idx, partMessage, 0);
  82     switch (res) {
  83     case Imap::Mailbox::FindInterestingPart::MAINPART_FOUND:
  84     case Imap::Mailbox::FindInterestingPart::MAINPART_PART_LOADING:
  85         Q_ASSERT(idx.isValid());
  86         m_mainPartUrl = QLatin1String("trojita-imap://msg") + idx.data(RolePartPathToPart).toString();
  87         break;
  88     case Imap::Mailbox::FindInterestingPart::MAINPART_MESSAGE_NOT_LOADED:
  89         Q_ASSERT(false);
  90         m_mainPartUrl = QLatin1String("data:text/plain;charset=utf-8;base64,") + QString::fromLocal8Bit(QByteArray("").toBase64());
  91         break;
  92     case Imap::Mailbox::FindInterestingPart::MAINPART_PART_CANNOT_DETERMINE:
  93         // FIXME: show a decent message here
  94         m_mainPartUrl = QLatin1String("data:text/plain;charset=utf-8;base64,") + QString::fromLocal8Bit(QByteArray(partMessage.toUtf8()).toBase64());
  95         break;
  96     }
  97     emit mainPartUrlChanged();
  98 }
  99
 100 QDateTime OneMessageModel::date() const
 101 {
 102     return m_message.data(RoleMessageDate).toDateTime();
 103 }
 104
 105 QDateTime OneMessageModel::receivedDate() const
 106 {
 107     return m_message.data(RoleMessageInternalDate).toDateTime();
 108 }
 109
 110 QString OneMessageModel::subject() const
 111 {
 112     return m_message.data(RoleMessageSubject).toString();
 113 }
 114
 115 QVariantList OneMessageModel::from() const
 116 {
 117     return m_message.data(RoleMessageFrom).toList();
 118 }
 119
 120 QVariantList OneMessageModel::to() const
 121 {
 122     return m_message.data(RoleMessageTo).toList();
 123 }
 124
 125 QVariantList OneMessageModel::cc() const
 126 {
 127     return m_message.data(RoleMessageCc).toList();
 128 }
 129
 130 QVariantList OneMessageModel::bcc() const
 131 {
 132     return m_message.data(RoleMessageBcc).toList();
 133 }
 134
 135 QVariantList OneMessageModel::sender() const
 136 {
 137     return m_message.data(RoleMessageSender).toList();
 138 }
 139
 140 QVariantList OneMessageModel::replyTo() const
 141 {
 142     return m_message.data(RoleMessageReplyTo).toList();
 143 }
 144
 145 QByteArray OneMessageModel::inReplyTo() const
 146 {
 147     return m_message.data(RoleMessageInReplyTo).toByteArray();
 148 }
 149
 150 QByteArray OneMessageModel::messageId() const
 151 {
 152     return m_message.data(RoleMessageMessageId).toByteArray();
 153 }
 154
 155 bool OneMessageModel::isMarkedDeleted() const
 156 {
 157     return m_message.data(RoleMessageIsMarkedDeleted).toBool();
 158 }
 159
 160 bool OneMessageModel::isMarkedRead() const
 161 {
 162     return m_message.data(RoleMessageIsMarkedRead).toBool();
 163 }
 164
 165 bool OneMessageModel::isMarkedForwarded() const
 166 {
 167     return m_message.data(RoleMessageIsMarkedForwarded).toBool();
 168 }
 169
 170 bool OneMessageModel::isMarkedReplied() const
 171 {
 172     return m_message.data(RoleMessageIsMarkedReplied).toBool();
 173 }
 174
 175 bool OneMessageModel::isMarkedRecent() const
 176 {
 177     return m_message.data(RoleMessageIsMarkedRecent).toBool();
 178 }
 179
 180 bool OneMessageModel::isMarkedFlagged() const
 181 {
 182     return m_message.data(RoleMessageIsMarkedFlagged).toBool();
 183 }
 184
 185 bool OneMessageModel::isMarkedJunk() const
 186 {
 187     return m_message.data(RoleMessageIsMarkedJunk).toBool() &&
 188         !m_message.data(RoleMessageIsMarkedNotJunk).toBool();
 189 }
 190
 191 bool OneMessageModel::isMarkedNotJunk() const
 192 {
 193     return (m_message.data(RoleMessageIsMarkedNotJunk).toBool() &&
 194         !m_message.data(RoleMessageIsMarkedJunk).toBool());
 195 }
 196
 197
 198 QUrl OneMessageModel::mainPartUrl() const
 199 {
 200     return m_mainPartUrl;
 201 }
 202
 203 QObject *OneMessageModel::attachmentsModel() const
 204 {
 205     return m_flatteningModel;
 206 }
 207
 208 QModelIndex OneMessageModel::mainPartModelIndex()
 209 {
 210     return Imap::Network::MsgPartNetAccessManager::pathToPart(
 211                 m_message, m_mainPartUrl.path().toUtf8());
 212 }
 213
 214 void OneMessageModel::setMarkedDeleted(const bool marked)
 215 {
 216     if (!m_message.isValid())
 217         return;
 218
 219     qobject_cast<Model*>(parent())->markMessagesDeleted(QList<QModelIndex>() << m_message, marked ? FLAG_ADD : FLAG_REMOVE);
 220 }
 221
 222 void OneMessageModel::setMarkedRead(const bool marked)
 223 {
 224     if (!m_message.isValid())
 225         return;
 226
 227     qobject_cast<Model*>(parent())->markMessagesRead(QList<QModelIndex>() << m_message, marked ? FLAG_ADD : FLAG_REMOVE);
 228 }
 229
 230 void OneMessageModel::handleModelDataChanged(const QModelIndex &topLeft, const QModelIndex &bottomRight)
 231 {
 232     Q_ASSERT(topLeft.row() == bottomRight.row());
 233     Q_ASSERT(topLeft.parent() == bottomRight.parent());
 234     Q_ASSERT(topLeft.model() == bottomRight.model());
 235
 236     if (m_message.isValid() && topLeft == m_message)
 237         emit flagsChanged();
 238 }
 239
 240 bool OneMessageModel::hasValidIndex() const
 241 {
 242     return m_message.isValid();
 243 }
 244
 245 }
 246 }