Differential D5464 Diff 13470 src/plugins/nextcloud/nextcloudjob.cpp

Changeset View

Standalone View

View Options

src/plugins/nextcloud/nextcloudjob.cpp

  • This file was added.
1​/*
2​ Copyright 2017 Lim Yuen Hoe <yuenhoe86@gmail.com>
3​
4​ This library is free software; you can redistribute it and/or
5​ modify it under the terms of the GNU Lesser General Public
6​ License as published by the Free Software Foundation; either
7​ version 2.1 of the License, or (at your option) any later version.
8​
9​ This library is distributed in the hope that it will be useful,
10​ but WITHOUT ANY WARRANTY; without even the implied warranty of
11​ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12​ Lesser General Public License for more details.
13​
14​ You should have received a copy of the GNU Lesser General Public
15​ License along with this library. If not, see <http://www.gnu.org/licenses/>.
16​*/
17​
18​#include "nextcloudjob.h"
19​#include <QDebug>
20​#include <QFileInfo>
21​#include <KStringHandler>
22​#include <KLocalizedString>
23​#include <KAccounts/getcredentialsjob.h>
24​#include <KAccounts/core.h>
25​#include <KIO/Job>
26​#include <KIO/DavJob>
27​
28​void NextcloudJob::start()
29​{
30​ // get all the info we need
31​ const QString folder = data().value(QStringLiteral("folder")).toString();
32​ const Accounts::AccountId id = data().value(QStringLiteral("accountId")).toInt();
33​ Accounts::Account* acc = Accounts::Account::fromId(KAccounts::accountsManager(), id);
34​ auto job = new GetCredentialsJob(id, this);
35​ bool b = job->exec();
36​ if (!b) {
37​ qWarning() << "Couldn't fetch credentials";
38​ setError(job->error());
39​ setErrorText(job->errorText());
40​ emitResult();
41​ return;
42​ }
43​ Q_FOREACH(const Accounts::Service &service, acc->services()) {
44​ if (service.name() == QStringLiteral("nextcloud-upload")) {
45​ acc->selectService(service);
46​ }
47​ }
48​ m_davUrl = QUrl(acc->valueAsString(QStringLiteral("server")) +
49​ QStringLiteral("remote.php/webdav/") + folder + QStringLiteral("/"));
50​ m_davUrl.setUserName(job->credentialsData().value(QStringLiteral("UserName")).toString());
51​ m_davUrl.setPassword(job->credentialsData().value(QStringLiteral("Secret")).toString());
52​
53​ // first we check that the folder exists
54​ KIO::DavJob* davjob = KIO::davPropFind(m_davUrl, QDomDocument(), QStringLiteral("0"), KIO::HideProgressInfo);
55​ connect(davjob, &KJob::finished, this, &NextcloudJob::checkTargetFolder);
56​}
57​
58​void NextcloudJob::checkTargetFolder(KJob* j)
59​{
60​ QString responseString = qobject_cast<KIO::DavJob*>(j)->response().toString();
61​ // TODO: prob a better way to do this
62​ if (responseString.contains(QStringLiteral("<d:collection xmlns:d=\"DAV:\"/>"))) {
63​ const QJsonArray urls = data().value(QStringLiteral("urls")).toArray();
64​
65​ foreach(const QJsonValue& url, urls) {
66​ // before uploading, we try to avoid overwrite by checking for existing file on nextcloud
67​ QUrl local = QUrl(url.toString());
68​ QUrl uploadTarget = createTargetUrl(local.fileName(), m_davUrl);
69​ KIO::DavJob* davjob = KIO::davPropFind(uploadTarget, QDomDocument(), QStringLiteral("0"), KIO::HideProgressInfo);
70​ connect(davjob, &KJob::finished, this, [=](KJob* job) { NextcloudJob::checkTargetFile(1, local, job); });
71​ m_pendingJobs++;
72​ }
73​
74​ } else {
75​ qWarning() << "invalid folder";
76​ setError(KIO::Error::ERR_CANNOT_ENTER_DIRECTORY);
77​ setErrorText(i18n("Invalid folder!"));
78​ emitResult();
79​ }
80​}
81​
82​void NextcloudJob::checkTargetFile(const int suffix, const QUrl& local, KJob* j)
davidedmundsonUnsubmitted
Done

Replace this first argument with KIO::suggestName(); you just need to know the name of the last one you tried; which you can extract from the URL you're using

Then you can kill: createTargetUrl

davidedmundson: Replace this first argument with KIO::suggestName(); you just need to know the name of the last…
83​{
84​ if (j->error()) {
85​ setError(j->error());
86​ setErrorText(j->errorText());
87​ emitResult();
88​ return;
89​ }
90​
91​ KIO::DavJob* job = qobject_cast<KIO::DavJob*>(j);
92​ QString responseString = job->response().toString();
93​
94​ // TODO: better way to do this
95​ if (responseString.contains(QStringLiteral("DAV\\Exception\\NotFound</s:exception>"))) {
96​ // okay file doesn't exist on nextcloud, we'll fetch the file, then upload it
97​ KIO::StoredTransferJob* next = KIO::storedGet(local);
98​ QUrl targetUrl = job->url();
99​ connect(next, &KJob::finished, this, [=](KJob* jj) { NextcloudJob::fileFetched(targetUrl, jj); });
100​ } else {
101​ // file already exists! we try successive suffixes until we find a free name
102​ QUrl uploadTarget = createTargetUrl(local.fileName(), m_davUrl, suffix);
103​ qDebug() << "Trying: " << uploadTarget.toString();
104​ KIO::DavJob* davjob = KIO::davPropFind(uploadTarget, QDomDocument(), QStringLiteral("0"), KIO::HideProgressInfo);
105​ connect(davjob, &KJob::finished, this, [=](KJob* jj) { NextcloudJob::checkTargetFile(suffix + 1, local, jj); });
106​ }
107​}
108​
109​void NextcloudJob::fileFetched(const QUrl& uploadUrl, KJob* j)
110​{
111​ if (j->error()) {
112​ setError(j->error());
113​ setErrorText(j->errorText());
114​ emitResult();
115​ return;
116​ }
117​
118​ KIO::StoredTransferJob* job = qobject_cast<KIO::StoredTransferJob*>(j);
119​
120​ // we fetched our file and we have a place to upload to. Time to upload!
121​ KIO::StoredTransferJob *tJob = KIO::storedPut(job->data(), uploadUrl, KIO::HideProgressInfo);
122​ connect(tJob, &KJob::result, this, &NextcloudJob::fileUploaded);
123​}
124​
125​void NextcloudJob::fileUploaded(KJob* j)
126​{
127​ if (j->error()) {
128​ setError(j->error());
129​ setErrorText(j->errorText());
130​ emitResult();
131​ return;
132​ }
133​
134​ //KIO::StoredTransferJob *sjob = qobject_cast<KIO::StoredTransferJob *>(j);
135​ m_pendingJobs--;
136​ if (m_pendingJobs == 0) {
137​ setOutput( {{ QStringLiteral("url"), QString() }});
138​ emitResult();
139​ }
140​}
141​
142​QUrl NextcloudJob::createTargetUrl(QString filename, QUrl baseUrl, const int suffix)
143​{
144​ // disallow giant filenames
145​ // TODO: this doesn't currently deal well with say image clipboard data. might see if we can somehow add the right extension.
146​ filename = KStringHandler::csqueeze(filename, 100);
147​ if (suffix == 0) {
148​ baseUrl.setPath(baseUrl.path() + filename);
149​ } else {
150​ QFileInfo info = QFileInfo(filename);
151​ if (info.completeSuffix().isEmpty()) {
152​ baseUrl.setPath(baseUrl.path() + filename + QString::number(suffix));
153​ } else {
154​ baseUrl.setPath(baseUrl.path() + info.baseName() + QString::number(suffix) + QStringLiteral(".") + info.completeSuffix());
155​ }
156​ }
157​ return baseUrl;
158​}