1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
|
/*
Copyright (c) 2016 Christian Mollekopf <mollekopf@kolabsys.com>
This library is free software; you can redistribute it and/or modify it
under the terms of the GNU Library General Public License as published by
the Free Software Foundation; either version 2 of the License, or (at your
option) any later version.
This library 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 Library General Public
License for more details.
You should have received a copy of the GNU Library General Public License
along with this library; see the file COPYING.LIB. If not, write to the
Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
02110-1301, USA.
*/
#include "messageparser.h"
#include "stringhtmlwriter.h"
#include "objecttreesource.h"
#include "mimetreeparser/interface.h"
#include <QRegExp>
#include <QFile>
#include <QImage>
#include <QDebug>
#include <QTime>
#include <QUrl>
#include <MimeTreeParser/ObjectTreeParser>
#include <MimeTreeParser/MessagePart>
class MessagePartPrivate
{
public:
QSharedPointer<MimeTreeParser::MessagePart> mPartTree;
QString mHtml;
QMap<QByteArray, QUrl> mEmbeddedPartMap;
std::shared_ptr<MimeTreeParser::NodeHelper> mNodeHelper;
std::shared_ptr<Parser> mParser;
};
MessageParser::MessageParser(QObject *parent)
: QObject(parent)
, d(std::unique_ptr<MessagePartPrivate>(new MessagePartPrivate))
{
}
MessageParser::~MessageParser()
{
}
QString MessageParser::html() const
{
return d->mHtml;
}
QVariant MessageParser::message() const
{
return QVariant();
}
void MessageParser::setMessage(const QVariant &message)
{
QTime time;
time.start();
d->mParser = std::shared_ptr<Parser>(new Parser(message.toByteArray()));
const auto mailData = KMime::CRLFtoLF(message.toByteArray());
KMime::Message::Ptr msg(new KMime::Message);
msg->setContent(mailData);
msg->parse();
qWarning() << "parsed: " << time.elapsed();
// render the mail
StringHtmlWriter htmlWriter;
//temporary files only have the lifetime of the nodehelper, so we keep it around until the mail changes.
d->mNodeHelper = std::make_shared<MimeTreeParser::NodeHelper>();
ObjectTreeSource source(&htmlWriter);
MimeTreeParser::ObjectTreeParser otp(&source, d->mNodeHelper.get());
otp.parseObjectTree(msg.data());
d->mPartTree = otp.parsedPart().dynamicCast<MimeTreeParser::MessagePart>();
d->mEmbeddedPartMap = htmlWriter.embeddedParts();
d->mHtml = htmlWriter.html();
emit htmlChanged();
}
QAbstractItemModel *MessageParser::partTree() const
{
return new PartModel(d->mPartTree, d->mParser);
}
QAbstractItemModel *MessageParser::newTree() const
{
return new NewModel(d->mParser);
}
|