From d81eea635c2151c2c9c62a53c08e427c38ad0acd Mon Sep 17 00:00:00 2001
From: Christian Mollekopf <chrigi_1@fastmail.fm>
Date: Fri, 29 Jun 2018 17:27:31 +0200
Subject: Partial fix for multipart/mixed in alternative part

We only render the first part right now, which is not correct.
---
 .../src/domain/mime/mimetreeparser/messagepart.cpp | 27 ++++++++++++---------
 .../src/domain/mime/mimetreeparser/messagepart.h   |  1 -
 .../mime/mimetreeparser/multipartalternative.cpp   | 28 +---------------------
 .../mimetreeparser/tests/mimetreeparsertest.cpp    | 20 ++++++++++++++++
 4 files changed, 37 insertions(+), 39 deletions(-)

(limited to 'framework/src/domain/mime/mimetreeparser')

diff --git a/framework/src/domain/mime/mimetreeparser/messagepart.cpp b/framework/src/domain/mime/mimetreeparser/messagepart.cpp
index 1a1b2003..9c17aafc 100644
--- a/framework/src/domain/mime/mimetreeparser/messagepart.cpp
+++ b/framework/src/domain/mime/mimetreeparser/messagepart.cpp
@@ -587,30 +587,35 @@ AlternativeMessagePart::AlternativeMessagePart(ObjectTreeParser *otp, KMime::Con
         // when displaying plain text.
         if (!dataHtml) {
             dataHtml  = findTypeInDirectChilds(mNode, "multipart/mixed");
+            if (dataHtml) {
+                const auto parts = dataHtml->contents();
+                for (int i = 0; i < parts.size(); i++) {
+                    const auto p = parts.at(i);
+                    if (i == 0 ) {
+                        // FIXME multipart/mixed should display all types serially, not just one
+                        dataHtml = p;
+                    } else if (KMime::isAttachment(p)) {
+                        appendSubPart(MimeMessagePart::Ptr(new MimeMessagePart(otp, p, true)));
+                    }
+                }
+            }
         }
     }
 
     if (dataIcal) {
-        mChildNodes[Util::MultipartIcal] = dataIcal;
+        mChildParts[Util::MultipartIcal] = MimeMessagePart::Ptr(new MimeMessagePart(mOtp, dataIcal, true));
     }
 
     if (dataText) {
-        mChildNodes[Util::MultipartPlain] = dataText;
+        mChildParts[Util::MultipartPlain] = MimeMessagePart::Ptr(new MimeMessagePart(mOtp, dataText, true));
     }
 
     if (dataHtml) {
-        mChildNodes[Util::MultipartHtml] = dataHtml;
+        mChildParts[Util::MultipartHtml] = MimeMessagePart::Ptr(new MimeMessagePart(mOtp, dataHtml, true));
     }
 
-    if (mChildNodes.isEmpty()) {
+    if (mChildParts.isEmpty()) {
         qCWarning(MIMETREEPARSER_LOG) << "no valid nodes";
-        return;
-    }
-
-    QMapIterator<Util::HtmlMode, KMime::Content *> i(mChildNodes);
-    while (i.hasNext()) {
-        i.next();
-        mChildParts[i.key()] = MimeMessagePart::Ptr(new MimeMessagePart(mOtp, i.value(), true));
     }
 }
 
diff --git a/framework/src/domain/mime/mimetreeparser/messagepart.h b/framework/src/domain/mime/mimetreeparser/messagepart.h
index c576699e..9add81d2 100644
--- a/framework/src/domain/mime/mimetreeparser/messagepart.h
+++ b/framework/src/domain/mime/mimetreeparser/messagepart.h
@@ -245,7 +245,6 @@ public:
 
     QList<Util::HtmlMode> availableModes();
 private:
-    QMap<Util::HtmlMode, KMime::Content *> mChildNodes;
     QMap<Util::HtmlMode, MimeMessagePart::Ptr> mChildParts;
 
     friend class DefaultRendererPrivate;
diff --git a/framework/src/domain/mime/mimetreeparser/multipartalternative.cpp b/framework/src/domain/mime/mimetreeparser/multipartalternative.cpp
index d0657edb..b1500136 100644
--- a/framework/src/domain/mime/mimetreeparser/multipartalternative.cpp
+++ b/framework/src/domain/mime/mimetreeparser/multipartalternative.cpp
@@ -47,35 +47,9 @@ MessagePart::Ptr MultiPartAlternativeBodyPartFormatter::process(Interface::BodyP
         return MessagePart::Ptr();
     }
 
-    //Hardcoded after removing the source
-    auto preferredMode = MimeTreeParser::Util::Html;
     AlternativeMessagePart::Ptr mp(new AlternativeMessagePart(part.objectTreeParser(), node));
-    if (mp->mChildNodes.isEmpty()) {
+    if (mp->mChildParts.isEmpty()) {
         return MimeMessagePart::Ptr(new MimeMessagePart(part.objectTreeParser(), node->contents().at(0)));
     }
-
-    KMime::Content *dataIcal = mp->mChildNodes.contains(Util::MultipartIcal) ? mp->mChildNodes[Util::MultipartIcal] : nullptr;
-    KMime::Content *dataHtml = mp->mChildNodes.contains(Util::MultipartHtml) ? mp->mChildNodes[Util::MultipartHtml] : nullptr;
-    KMime::Content *dataPlain = mp->mChildNodes.contains(Util::MultipartPlain) ? mp->mChildNodes[Util::MultipartPlain] : nullptr;
-
-    // Make sure that in default ical is prefered over html and plain text
-    if (dataIcal && ((preferredMode != Util::MultipartHtml && preferredMode != Util::MultipartPlain))) {
-        if (dataHtml) {
-            part.nodeHelper()->setNodeProcessed(dataHtml, false);
-        }
-        if (dataPlain) {
-            part.nodeHelper()->setNodeProcessed(dataPlain, false);
-        }
-        preferredMode = Util::MultipartIcal;
-    } else if ((dataHtml && (preferredMode == Util::MultipartHtml || preferredMode == Util::Html)) ||
-               (dataHtml && dataPlain && dataPlain->body().isEmpty())) {
-        if (dataPlain) {
-            part.nodeHelper()->setNodeProcessed(dataPlain, false);
-        }
-        preferredMode = Util::MultipartHtml;
-    } else if (!(preferredMode == Util::MultipartHtml) && dataPlain) {
-        part.nodeHelper()->setNodeProcessed(dataHtml, false);
-        preferredMode = Util::MultipartPlain;
-    }
     return mp;
 }
diff --git a/framework/src/domain/mime/mimetreeparser/tests/mimetreeparsertest.cpp b/framework/src/domain/mime/mimetreeparser/tests/mimetreeparsertest.cpp
index 6b4280a9..7fa651bc 100644
--- a/framework/src/domain/mime/mimetreeparser/tests/mimetreeparsertest.cpp
+++ b/framework/src/domain/mime/mimetreeparser/tests/mimetreeparsertest.cpp
@@ -418,6 +418,26 @@ private slots:
         auto attachments = otp.collectAttachmentParts();
         QCOMPARE(attachments.size(), 1);
     }
+
+    void testAppleHtmlWithAttachmentsMixed()
+    {
+        MimeTreeParser::ObjectTreeParser otp;
+        otp.parseObjectTree(readMailFromFile("applehtmlwithattachmentsmixed.mbox"));
+        otp.decryptParts();
+        otp.print();
+        auto partList = otp.collectContentParts();
+        QCOMPARE(partList.size(), 1);
+        auto part = partList[0].dynamicCast<MimeTreeParser::MessagePart>();
+        QCOMPARE(part->encryptions().size(), 0);
+        QCOMPARE(part->signatures().size(), 0);
+        qWarning() << otp.plainTextContent();
+        qWarning() << otp.htmlContent();
+        QCOMPARE(otp.plainTextContent(), QString::fromUtf8("Hello\n\n\n\nRegards\n\nFsdfsdf"));
+        QCOMPARE(otp.htmlContent(), QString::fromUtf8("<html><head><meta http-equiv=\"Content-Type\" content=\"text/html; charset=us-ascii\"></head><body style=\"word-wrap: break-word; -webkit-nbsp-mode: space; line-break: after-white-space;\" class=\"\"><strike class=\"\">Hello</strike><div class=\"\"><br class=\"\"></div><div class=\"\"></div></body></html><html><head><meta http-equiv=\"Content-Type\" content=\"text/html; charset=us-ascii\"></head><body style=\"word-wrap: break-word; -webkit-nbsp-mode: space; line-break: after-white-space;\" class=\"\"><div class=\"\"></div><div class=\"\"><br class=\"\"></div><div class=\"\"><b class=\"\">Regards</b></div><div class=\"\"><b class=\"\"><br class=\"\"></b></div><div class=\"\">Fsdfsdf</div></body></html>"));
+
+        auto attachments = otp.collectAttachmentParts();
+        QCOMPARE(attachments.size(), 1);
+    }
 };
 
 QTEST_GUILESS_MAIN(InterfaceTest)
-- 
cgit v1.2.3