Differential D28448 Diff 79782 messagecomposer/src/job/protectedheaders.cpp

Changeset View

Standalone View

messagecomposer/src/job/protectedheaders.cpp

This file was added.

1		/*
2		Copyright (C) 2020 Sandro Knauß <sknauss@kde.org>
3
4		This library is free software; you can redistribute it and/or modify it
5		under the terms of the GNU Library General Public License as published by
6		the Free Software Foundation; either version 2 of the License, or (at your
7		option) any later version.
8
9		This library is distributed in the hope that it will be useful, but WITHOUT
10		ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11		FITNESS FOR A PARTICULAR PURPOSE. See the GNU Library General Public
12		License for more details.
13
14		You should have received a copy of the GNU Library General Public License
15		along with this library; see the file COPYING.LIB. If not, write to the
16		Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
17		02110-1301, USA.
18		*/
19
20		#include "job/protectedheaders.h"
21
22		#include "contentjobbase_p.h"
23		#include "job/singlepartjob.h"
24		#include "utils/util_p.h"
25
26		#include "messagecomposer_debug.h"
27
28		#include <kmime/kmime_message.h>
29		#include <kmime/kmime_content.h>
30
31		using namespace MessageComposer;
32
33		class MessageComposer::ProtectedHeadersJobPrivate : public ContentJobBasePrivate
34		{
35		public:
36		ProtectedHeadersJobPrivate(ProtectedHeadersJob *qq)
37		: ContentJobBasePrivate(qq)
38		{
39		}
40
41		KMime::Content *content = nullptr;
42		KMime::Message *skeletonMessage = nullptr;
43
44		bool obvoscate = false;
45
46		Q_DECLARE_PUBLIC(ProtectedHeadersJob)
47		};
48
49		ProtectedHeadersJob::ProtectedHeadersJob(QObject *parent)
50		: ContentJobBase(*new ProtectedHeadersJobPrivate(this), parent)
51		{
52		}
53
54		ProtectedHeadersJob::~ProtectedHeadersJob()
55		{
56		}
57
58		void ProtectedHeadersJob::setContent(KMime::Content *content)
59		{
60		Q_D(ProtectedHeadersJob);
61
62		d->content = content;
63		if (content)
64		{
65		d->content->assemble();
66		}
67		}
68
69		void ProtectedHeadersJob::setSkeletonMessage(KMime::Message *skeletonMessage)
70		{
	mlaurentUnsubmitted Done add missing space before ( mlaurent: add missing space before (
71		Q_D(ProtectedHeadersJob);
72
73		d->skeletonMessage = skeletonMessage;
74		}
75
76		void ProtectedHeadersJob::setObvoscate(bool obvoscate)
77		{
78		Q_D(ProtectedHeadersJob);
79
80		d->obvoscate = obvoscate;
81		}
82
83		void ProtectedHeadersJob::doStart() {
84		Q_D(ProtectedHeadersJob);
85		Q_ASSERT(d->resultContent == nullptr); // Not processed before.
86		Q_ASSERT(d->skeletonMessage); // We need a skeletonMessage to proceed
87
88		auto subject = d->skeletonMessage->header<KMime::Headers::Subject>();
89		if (d->obvoscate && subject) {
90		// Create protected header lagacy mimepart with replaced headers
91		SinglepartJob *cjob = new SinglepartJob;
92		cjob->contentType()->setMimeType("text/plain");
93		cjob->contentType()->setCharset(subject->rfc2047Charset());
94		cjob->contentType()->setParameter(QStringLiteral("protected-headers"), QStringLiteral("v1"));
95		cjob->contentDisposition()->setDisposition(KMime::Headers::contentDisposition::CDinline);
96		cjob->setData(subject->type() + QByteArray(": ") + subject->asUnicodeString().toUtf8());
97
98		QObject::connect(cjob, &SinglepartJob::finished, this, [d, cjob](KJob *job) {
99		KMime::Content *mixedPart = new KMime::Content();
100		const QByteArray boundary = KMime::multiPartBoundary();
101		mixedPart->contentType()->setMimeType("multipart/mixed");
102		mixedPart->contentType()->setBoundary(boundary);
103		mixedPart->addContent(cjob->content());
104
105		// if setContent hasn't been called, we assume that a subjob was added
106		// and we want to use that
107		if (!d->content \|\| !d->content->hasContent()) {
108		Q_ASSERT(d->subjobContents.size() == 1);
109		d->content = d->subjobContents.first();
110		}
111
112		mixedPart->addContent(d->content);
113		d->content = mixedPart;
114		});
115		appendSubjob(cjob);
116		}
117
118		ContentJobBase::doStart();
119		}
120
121		void ProtectedHeadersJob::process()
122		{
123		Q_D(ProtectedHeadersJob);
124
125		// if setContent hasn't been called, we assume that a subjob was added
126		// and we want to use that
127		if (!d->content \|\| !d->content->hasContent()) {
128		Q_ASSERT(d->subjobContents.size() == 1);
129		d->content = d->subjobContents.first();
130		}
131
132		auto subject = d->skeletonMessage->header<KMime::Headers::Subject>();
133		const auto headers = d->skeletonMessage->headers();
134		for (const auto &header: headers) {
	mlaurentUnsubmitted Done const auto &header ? mlaurent: const auto &header ?
	mlaurentUnsubmitted Done const auto headers = d->skeletonMessage->headers(); for (const auto &header : headers) { } fix missing space after "for" and '(' too mlaurent: const auto headers = d->skeletonMessage->headers(); for (const auto &header : headers) { } fix…
135		const QByteArray headerType(header->type());
	mlaurentUnsubmitted Done const mlaurent: const
136		if (headerType.startsWith("X-KMail-")) {
137		continue;
138		}
139		if (headerType == "MIME-Version") {
140		continue;
141		}
142		if (headerType == "Bcc") {
143		continue;
144		}
145		if (headerType.startsWith("Content-")) {
146		continue;
147		}
148		if (headerType == "Subject") {
149		KMime::Headers::Subject *copySubject = new KMime::Headers::Subject();
150		copySubject->from7BitString(subject->as7BitString(false));
151		d->content->appendHeader(copySubject);
152		} else {
153		d->content->appendHeader(header);
154		}
155		}
156
157		if (d->obvoscate && subject) {
158		subject->clear();
159		subject->from7BitString("...");
160		}
161		auto contentType = d->content->header<KMime::Headers::ContentType>();
162		contentType->setParameter(QStringLiteral("protected-headers"), QStringLiteral("v1"));
163
164		d->resultContent = d->content;
165
166		emitResult();
167		}