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
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
|
/*
SPDX-FileCopyrightText: 2013 Daniel Vrátil <dvratil@redhat.com>
SPDX-License-Identifier: LGPL-2.0-or-later
*/
#include "changeitemsflagstask.h"
#include "imapresource_debug.h"
#include <KIMAP/SelectJob>
#include <KIMAP/Session>
#include <KIMAP/StoreJob>
ChangeItemsFlagsTask::ChangeItemsFlagsTask(const ResourceStateInterface::Ptr &resource, QObject *parent)
: ResourceTask(ResourceTask::DeferIfNoSession, resource, parent)
{
}
ChangeItemsFlagsTask::~ChangeItemsFlagsTask() = default;
void ChangeItemsFlagsTask::doStart(KIMAP::Session *session)
{
const QString mailBox = mailBoxForCollection(items().at(0).parentCollection());
qCDebug(IMAPRESOURCE_LOG) << mailBox;
if (session->selectedMailBox() != mailBox) {
auto select = new KIMAP::SelectJob(session);
select->setMailBox(mailBox);
connect(select, &KJob::result, this, &ChangeItemsFlagsTask::onSelectDone);
select->start();
} else {
if (!addedFlags().isEmpty()) {
triggerAppendFlagsJob(session);
} else if (!removedFlags().isEmpty()) {
triggerRemoveFlagsJob(session);
} else {
qCDebug(IMAPRESOURCE_LOG) << "nothing to do";
changeProcessed();
}
}
}
void ChangeItemsFlagsTask::onSelectDone(KJob *job)
{
if (job->error()) {
qCWarning(IMAPRESOURCE_LOG) << "Select failed: " << job->errorString();
cancelTask(job->errorString());
} else {
auto select = static_cast<KIMAP::SelectJob *>(job);
qCDebug(IMAPRESOURCE_LOG) << addedFlags();
if (!addedFlags().isEmpty()) {
triggerAppendFlagsJob(select->session());
} else if (!removedFlags().isEmpty()) {
triggerRemoveFlagsJob(select->session());
} else {
qCDebug(IMAPRESOURCE_LOG) << "nothing to do";
changeProcessed();
}
}
}
KIMAP::StoreJob *ChangeItemsFlagsTask::prepareJob(KIMAP::Session *session)
{
KIMAP::ImapSet set;
const Akonadi::Item::List &allItems = items();
// Split the request to multiple smaller requests of 2000 UIDs each - various IMAP
// servers have various limits on maximum size of a request
// 2000 is a random number that sounds like a good compromise between performance
// and functionality (i.e. 2000 UIDs should be supported by any server out there)
for (int i = 0, count = qMin(2000, allItems.count() - m_processedItems); i < count; ++i) {
set.add(allItems[m_processedItems].remoteId().toLong());
++m_processedItems;
}
auto store = new KIMAP::StoreJob(session);
store->setUidBased(true);
store->setSequenceSet(set);
return store;
}
void ChangeItemsFlagsTask::triggerAppendFlagsJob(KIMAP::Session *session)
{
const auto supportedFlags = fromAkonadiToSupportedImapFlags(addedFlags().values(), items().at(0).parentCollection());
if (supportedFlags.isEmpty()) {
if (!removedFlags().isEmpty()) {
m_processedItems = 0;
triggerRemoveFlagsJob(session);
} else {
changeProcessed();
}
} else {
KIMAP::StoreJob *store = prepareJob(session);
store->setFlags(supportedFlags);
store->setMode(KIMAP::StoreJob::AppendFlags);
connect(store, &KIMAP::StoreJob::result, this, &ChangeItemsFlagsTask::onAppendFlagsDone);
store->start();
}
}
void ChangeItemsFlagsTask::triggerRemoveFlagsJob(KIMAP::Session *session)
{
const auto supportedFlags = fromAkonadiToSupportedImapFlags(removedFlags().values(), items().at(0).parentCollection());
if (supportedFlags.isEmpty()) {
changeProcessed();
} else {
KIMAP::StoreJob *store = prepareJob(session);
store->setFlags(supportedFlags);
store->setMode(KIMAP::StoreJob::RemoveFlags);
connect(store, &KIMAP::StoreJob::result, this, &ChangeItemsFlagsTask::onRemoveFlagsDone);
store->start();
}
}
void ChangeItemsFlagsTask::onAppendFlagsDone(KJob *job)
{
if (job->error()) {
qCWarning(IMAPRESOURCE_LOG) << "Flag append failed: " << job->errorString();
cancelTask(job->errorString());
} else {
KIMAP::Session *session = qobject_cast<KIMAP::Job *>(job)->session();
if (m_processedItems < items().count()) {
triggerAppendFlagsJob(session);
} else if (removedFlags().isEmpty()) {
changeProcessed();
} else {
qCDebug(IMAPRESOURCE_LOG) << removedFlags();
m_processedItems = 0;
triggerRemoveFlagsJob(session);
}
}
}
void ChangeItemsFlagsTask::onRemoveFlagsDone(KJob *job)
{
if (job->error()) {
qCWarning(IMAPRESOURCE_LOG) << "Flag remove failed: " << job->errorString();
cancelTask(job->errorString());
} else {
if (m_processedItems < items().count()) {
triggerRemoveFlagsJob(qobject_cast<KIMAP::Job *>(job)->session());
} else {
changeProcessed();
}
}
}
#include "moc_changeitemsflagstask.cpp"
|
