| 1 | /* |
| 2 | SPDX-FileCopyrightText: 2010 Tobias Koenig <tokoe@kde.org> |
| 3 | |
| 4 | SPDX-License-Identifier: LGPL-2.0-or-later |
| 5 | */ |
| 6 | |
| 7 | #include "davitemcreatejob.h" |
| 8 | #include "davjobbase_p.h" |
| 9 | |
| 10 | #include "daverror.h" |
| 11 | #include "davitemfetchjob.h" |
| 12 | |
| 13 | #include <KIO/DavJob> |
| 14 | #include <KIO/StoredTransferJob> |
| 15 | |
| 16 | using namespace KDAV; |
| 17 | |
| 18 | namespace KDAV |
| 19 | { |
| 20 | class DavItemCreateJobPrivate : public DavJobBasePrivate |
| 21 | { |
| 22 | public: |
| 23 | void davJobFinished(KJob *job); |
| 24 | void itemRefreshed(KJob *job); |
| 25 | |
| 26 | DavItem mItem; |
| 27 | int mRedirectCount = 0; |
| 28 | |
| 29 | Q_DECLARE_PUBLIC(DavItemCreateJob) |
| 30 | }; |
| 31 | } |
| 32 | |
| 33 | DavItemCreateJob::DavItemCreateJob(const DavItem &item, QObject *parent) |
| 34 | : DavJobBase(new DavItemCreateJobPrivate, parent) |
| 35 | { |
| 36 | Q_D(DavItemCreateJob); |
| 37 | d->mItem = item; |
| 38 | } |
| 39 | |
| 40 | void DavItemCreateJob::start() |
| 41 | { |
| 42 | Q_D(DavItemCreateJob); |
| 43 | QString = QStringLiteral("Content-Type: " ); |
| 44 | headers += d->mItem.contentType(); |
| 45 | headers += QLatin1String("\r\n" ); |
| 46 | headers += QLatin1String("If-None-Match: *" ); |
| 47 | |
| 48 | KIO::StoredTransferJob *job = KIO::storedPut(arr: d->mItem.data(), url: itemUrl(), permissions: -1, flags: KIO::HideProgressInfo | KIO::DefaultFlags); |
| 49 | job->addMetaData(QStringLiteral("PropagateHttpHeader" ), QStringLiteral("true" )); |
| 50 | job->addMetaData(QStringLiteral("customHTTPHeader" ), value: headers); |
| 51 | job->addMetaData(QStringLiteral("cookies" ), QStringLiteral("none" )); |
| 52 | job->addMetaData(QStringLiteral("no-auth-prompt" ), QStringLiteral("true" )); |
| 53 | job->setRedirectionHandlingEnabled(false); |
| 54 | |
| 55 | connect(sender: job, signal: &KIO::StoredTransferJob::result, context: this, slot: [d](KJob *job) { |
| 56 | d->davJobFinished(job); |
| 57 | }); |
| 58 | } |
| 59 | |
| 60 | DavItem DavItemCreateJob::item() const |
| 61 | { |
| 62 | Q_D(const DavItemCreateJob); |
| 63 | return d->mItem; |
| 64 | } |
| 65 | |
| 66 | QUrl DavItemCreateJob::itemUrl() const |
| 67 | { |
| 68 | Q_D(const DavItemCreateJob); |
| 69 | return d->mItem.url().url(); |
| 70 | } |
| 71 | |
| 72 | void DavItemCreateJobPrivate::davJobFinished(KJob *job) |
| 73 | { |
| 74 | Q_Q(DavItemCreateJob); |
| 75 | KIO::StoredTransferJob *storedJob = qobject_cast<KIO::StoredTransferJob *>(object: job); |
| 76 | const QString responseCodeStr = storedJob->queryMetaData(QStringLiteral("responsecode" )); |
| 77 | const int responseCode = responseCodeStr.isEmpty() ? 0 : responseCodeStr.toInt(); |
| 78 | if (storedJob->error()) { |
| 79 | setLatestResponseCode(responseCode); |
| 80 | setError(ERR_ITEMCREATE); |
| 81 | setJobErrorText(storedJob->errorText()); |
| 82 | setJobError(storedJob->error()); |
| 83 | setErrorTextFromDavError(); |
| 84 | |
| 85 | emitResult(); |
| 86 | return; |
| 87 | } |
| 88 | |
| 89 | // The 'Location:' HTTP header is used to indicate the new URL |
| 90 | const QStringList = storedJob->queryMetaData(QStringLiteral("HTTP-Headers" )).split(sep: QLatin1Char('\n')); |
| 91 | QString location; |
| 92 | for (const QString & : allHeaders) { |
| 93 | if (header.startsWith(s: QLatin1String("location:" ), cs: Qt::CaseInsensitive)) { |
| 94 | location = header.section(asep: QLatin1Char(' '), astart: 1); |
| 95 | } |
| 96 | } |
| 97 | |
| 98 | QUrl url; |
| 99 | if (location.isEmpty()) { |
| 100 | url = storedJob->url(); |
| 101 | } else if (location.startsWith(c: QLatin1Char('/'))) { |
| 102 | url = storedJob->url(); |
| 103 | url.setPath(path: location, mode: QUrl::TolerantMode); |
| 104 | } else { |
| 105 | url = QUrl::fromUserInput(userInput: location); |
| 106 | } |
| 107 | |
| 108 | if (responseCode == 301 || responseCode == 302 || responseCode == 307 || responseCode == 308) { |
| 109 | if (mRedirectCount > 4) { |
| 110 | setLatestResponseCode(responseCode); |
| 111 | setError(DavItemCreateJob::UserDefinedError + responseCode); |
| 112 | emitResult(); |
| 113 | } else { |
| 114 | QUrl _itemUrl(url); |
| 115 | _itemUrl.setUserInfo(userInfo: q->itemUrl().userInfo()); |
| 116 | mItem.setUrl(DavUrl(_itemUrl, mItem.url().protocol())); |
| 117 | |
| 118 | ++mRedirectCount; |
| 119 | q->start(); |
| 120 | } |
| 121 | |
| 122 | return; |
| 123 | } |
| 124 | |
| 125 | url.setUserInfo(userInfo: q->itemUrl().userInfo()); |
| 126 | mItem.setUrl(DavUrl(url, mItem.url().protocol())); |
| 127 | |
| 128 | DavItemFetchJob *fetchJob = new DavItemFetchJob(mItem); |
| 129 | QObject::connect(sender: fetchJob, signal: &DavItemFetchJob::result, context: q, slot: [this](KJob *job) { |
| 130 | itemRefreshed(job); |
| 131 | }); |
| 132 | fetchJob->start(); |
| 133 | } |
| 134 | |
| 135 | void DavItemCreateJobPrivate::itemRefreshed(KJob *job) |
| 136 | { |
| 137 | if (!job->error()) { |
| 138 | DavItemFetchJob *fetchJob = qobject_cast<DavItemFetchJob *>(object: job); |
| 139 | mItem.setEtag(fetchJob->item().etag()); |
| 140 | } |
| 141 | emitResult(); |
| 142 | } |
| 143 | |
| 144 | #include "moc_davitemcreatejob.cpp" |
| 145 | |