File indexing completed on 2024-06-23 05:20:44

 /* Copyright (C) 2006 - 2017 Jan Kundrát <jkt@kde.org>
 
    This file is part of the Trojita Qt IMAP e-mail client,
    http://trojita.flaska.net/
 
    This program is free software; you can redistribute it and/or
    modify it under the terms of the GNU General Public License as
    published by the Free Software Foundation; either version 2 of
    the License or (at your option) version 3 or any later version
    accepted by the membership of KDE e.V. (or its successor approved
    by the membership of KDE e.V.), which shall act as a proxy
    defined in Section 14 of version 3 of the license.
 
    This program 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 General Public License for more details.
 
    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
 #ifndef IMAP_MESSAGECOMPOSER_H
 #define IMAP_MESSAGECOMPOSER_H
 
 #include <QAbstractListModel>
 #include <QPointer>
 
 #include "Composer/AbstractComposer.h"
 #include "Composer/ContentDisposition.h"
 #include "Imap/Model/CatenateData.h"
 #include "Imap/Parser/Message.h"
 
 namespace Imap {
 namespace Mailbox {
 class Model;
 }
 }
 
 namespace Composer {
 
 class AttachmentItem;
 
 /** @short Model storing individual parts of a composed message */
 class MessageComposer : public QAbstractListModel, public AbstractComposer
 {
     Q_OBJECT
 public:
 
     explicit MessageComposer(Imap::Mailbox::Model *model);
     ~MessageComposer();
 
     int rowCount(const QModelIndex &parent = QModelIndex()) const override;
     QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override;
     Qt::DropActions supportedDropActions() const override;
     Qt::ItemFlags flags(const QModelIndex &index) const override;
     bool dropMimeData(const QMimeData *data, Qt::DropAction action, int row, int column, const QModelIndex &parent) override;
     QStringList mimeTypes() const override;
     QMimeData *mimeData(const QModelIndexList &indexes) const override;
 
     void setInReplyTo(const QList<QByteArray> &inReplyTo);
     void setReferences(const QList<QByteArray> &references);
     void setTimestamp(const QDateTime &timestamp);
     void setSubject(const QString &subject);
     void setOrganization(const QString &organization);
     void setText(const QString &text);
     void setReplyingToMessage(const QModelIndex &index);
     void prepareForwarding(const QModelIndex &index, const ForwardMode mode);
 
     bool isReadyForSerialization() const override;
     bool asRawMessage(QIODevice *target, QString *errorMessage) const override;
     bool asCatenateData(QList<Imap::Mailbox::CatenatePair> &target, QString *errorMessage) const override;
     void setPreloadEnabled(const bool preload) override;
     void setRecipients(const QList<QPair<Composer::RecipientKind, Imap::Message::MailAddress> > &recipients) override;
     void setFrom(const Imap::Message::MailAddress &from) override;
 
     QDateTime timestamp() const override;
     QList<QByteArray> inReplyTo() const;
     QList<QByteArray> references() const;
     QByteArray rawFromAddress() const override;
     QList<QByteArray> rawRecipientAddresses() const override;
     QModelIndex replyingToMessage() const override;
     QModelIndex forwardingMessage() const override;
 
     bool addFileAttachment(const QString &path);
     void removeAttachment(const QModelIndex &index);
     void setAttachmentContentDisposition(const QModelIndex &index, const ContentDisposition disposition);
     void setAttachmentName(const QModelIndex &index, const QString &newName);
 
     void setReportTrojitaVersions(const bool reportVersion);
 
 private:
     static QByteArray encodeHeaderField(const QString &text);
 
     void ensureRandomStrings() const;
     void writeCommonMessageBeginning(QIODevice *target) const;
     bool writeAttachmentHeader(QIODevice *target, QString *errorMessage, const AttachmentItem *attachment) const;
     bool writeAttachmentBody(QIODevice *target, QString *errorMessage, const AttachmentItem *attachment) const;
 
     void writeHeaderWithMsgIds(QIODevice *target, const QByteArray &headerName, const QList<QByteArray> &messageIds) const;
 
     bool validateDropImapMessage(QDataStream &stream, QString &mailbox, uint &uidValidity, QList<uint> &uids) const;
     bool validateDropImapPart(QDataStream &stream, QString &mailbox, uint &uidValidity, uint &uid, QByteArray &trojitaPath) const;
     bool dropImapMessage(QDataStream &stream);
     bool dropImapPart(QDataStream &stream);
     bool dropAttachmentList(QDataStream &stream);
 
     Imap::Message::MailAddress m_from;
     QList<QPair<Composer::RecipientKind, Imap::Message::MailAddress> > m_recipients;
     QList<QByteArray> m_inReplyTo;
     QList<QByteArray> m_references;
     QDateTime m_timestamp;
     QString m_subject;
     QString m_organization;
     QString m_text;
     QPersistentModelIndex m_replyingTo;
     QPersistentModelIndex m_forwarding;
     mutable QByteArray m_messageId;
     mutable QByteArray m_mimeBoundary;
 
     QList<AttachmentItem *> m_attachments;
     QPointer<Imap::Mailbox::Model> m_model;
     bool m_shouldPreload;
     bool m_reportTrojitaVersions;
 };
 
 }
 
 #endif // IMAP_MESSAGECOMPOSER_H