render mail to html

1 files changed, 46 insertions, 0 deletions

diff --git a/framework/mail/maillistmodel.cpp b/framework/mail/maillistmodel.cpp
index e6a9c218..c37e2be4 100644
--- a/framework/mail/maillistmodel.cpp
+++ b/framework/mail/maillistmodel.cpp
@@ -1,6 +1,12 @@
 #include "maillistmodel.h"
 
+#include "filehtmlwriter.h"
+#include "objecttreesource.h"
+
 #include <QFile>
+#include <QImage>
+#include <KF5/MessageViewer/ObjectTreeParser>
+#include <KF5/MessageViewer/CSSHelper>
 
 MailListModel::MailListModel(QObject *parent)
     : QIdentityProxyModel()
@@ -25,6 +31,7 @@ QHash< int, QByteArray > MailListModel::roleNames() const
     roles[Important] = "important";
     roles[Id] = "id";
     roles[MimeMessage] = "mimeMessage";
+    roles[RenderedMessage] = "renderedMessage";
     roles[DomainObject] = "domainObject";
 
     return roles;
@@ -61,6 +68,45 @@ QVariant MailListModel::data(const QModelIndex &idx, int role) const
             }
             return "Failed to read mail.";
         }
+        case RenderedMessage: {
+            auto filename = srcIdx.sibling(srcIdx.row(), 6).data(Qt::DisplayRole).toString();
+            QFile file(filename);
+            if (file.open(QFile::ReadOnly)) {
+                const auto mailData = KMime::CRLFtoLF(file.readAll());
+                KMime::Message::Ptr msg(new KMime::Message);
+                msg->setContent(mailData);
+                msg->parse();
+
+                // render the mail
+                const QString fname("/tmp/test.html");
+                FileHtmlWriter htmlWriter(fname);
+                QImage paintDevice;
+                MessageViewer::CSSHelper cssHelper(&paintDevice);
+                MessageViewer::NodeHelper nodeHelper;
+                ObjectTreeSource source(&htmlWriter, &cssHelper);
+                MessageViewer::ObjectTreeParser otp(&source, &nodeHelper);
+
+                htmlWriter.begin(QString());
+                htmlWriter.queue(cssHelper.htmlHead(false));
+
+                otp.parseObjectTree(msg.data());
+
+                htmlWriter.queue(QStringLiteral("</body></html>"));
+                htmlWriter.flush();
+                htmlWriter.end();
+
+                QFile file(fname);
+                if (file.open(QFile::ReadOnly)) {
+                    const auto content = file.readAll();
+                    return content;
+                } else {
+                    qWarning() << "Failed to open the file";
+                }
+            } else {
+                qWarning() << "Failed to open the file";
+            }
+            return "Failed to read mail.";
+        }
     }
     return QIdentityProxyModel::data(idx, role);
 }