File indexing completed on 2024-11-17 04:44:01

 /*
     Copyright (c) 2010 Grégory Oestreicher <greg@kamago.net>
     Based on DavItemsListJob which is copyright (c) 2010 Tobias Koenig <tokoe@kde.org>
 
     This program is free software; you can redistribute it and/or modify
     it under the terms of the GNU General Public License as published by
     the Free Software Foundation; either version 2 of the License, or
     (at your option) any later version.
 
     This program 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 General Public License for more details.
 
     You should have received a copy of the GNU General Public License
     along with this program; if not, write to the Free Software
     Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 */
 
 #include "davitemsfetchjob.h"
 
 #include "davmanager.h"
 #include "davmultigetprotocol.h"
 #include "utils.h"
 #include "daverror.h"
 #include "davjob.h"
 
 using namespace KDAV2;
 
 DavItemsFetchJob::DavItemsFetchJob(const DavUrl &collectionUrl, const QStringList &urls, QObject *parent)
     : DavJobBase(parent), mCollectionUrl(collectionUrl), mUrls(urls)
 {
 }
 
 void DavItemsFetchJob::start()
 {
     const DavMultigetProtocol *protocol =
         dynamic_cast<const DavMultigetProtocol *>(DavManager::self()->davProtocol(mCollectionUrl.protocol()));
     if (!protocol) {
         setError(ERR_NO_MULTIGET);
         setErrorTextFromDavError();
         emitResult();
         return;
     }
 
     if (mUrls.isEmpty()) {
         setError(ERR_PROBLEM_WITH_REQUEST);
         setErrorText("DavItemsFetchJob without urls.");
         emitResult();
         return;
     }
 
     const QDomDocument report = protocol->itemsReportQuery(mUrls)->buildQuery();
     DavJob *job = DavManager::self()->createReportJob(mCollectionUrl.url(), report, QStringLiteral("0"));
     connect(job, &DavJob::result, this, &DavItemsFetchJob::davJobFinished);
 }
 
 DavItem::List DavItemsFetchJob::items() const
 {
     DavItem::List values;
     values.reserve(mItems.size());
     Q_FOREACH (const auto &value, mItems) {
         values << value;
     }
     return values;
 }
 
 DavItem DavItemsFetchJob::item(const QString &url) const
 {
     return mItems.value(url);
 }
 
 void DavItemsFetchJob::davJobFinished(KJob *job)
 {
     auto davJob = static_cast<DavJob *>(job);
     if (davJob->error()) {
         setErrorFromJob(davJob);
         emitResult();
         return;
     }
 
     const DavMultigetProtocol *protocol =
         static_cast<const DavMultigetProtocol *>(DavManager::self()->davProtocol(mCollectionUrl.protocol()));
 
     const QDomDocument document = davJob->response();
     const QDomElement documentElement = document.documentElement();
 
     QDomElement responseElement = Utils::firstChildElementNS(documentElement, QStringLiteral("DAV:"), QStringLiteral("response"));
 
     while (!responseElement.isNull()) {
         QDomElement propstatElement = Utils::firstChildElementNS(responseElement, QStringLiteral("DAV:"), QStringLiteral("propstat"));
 
         if (propstatElement.isNull()) {
             responseElement = Utils::nextSiblingElementNS(responseElement, QStringLiteral("DAV:"), QStringLiteral("response"));
             continue;
         }
 
         // Check for errors
         const QDomElement statusElement = Utils::firstChildElementNS(propstatElement, QStringLiteral("DAV:"), QStringLiteral("status"));
         if (!statusElement.text().contains(QLatin1String("200"))) {
             responseElement = Utils::nextSiblingElementNS(responseElement, QStringLiteral("DAV:"), QStringLiteral("response"));
             continue;
         }
 
         const QDomElement propElement = Utils::firstChildElementNS(propstatElement, QStringLiteral("DAV:"), QStringLiteral("prop"));
 
         DavItem item;
 
         // extract path
         const QDomElement hrefElement = Utils::firstChildElementNS(responseElement, QStringLiteral("DAV:"), QStringLiteral("href"));
         const QString href = hrefElement.text();
 
         QUrl url = davJob->url();
         if (href.startsWith(QLatin1Char('/'))) {
             // href is only a path, use request url to complete
             url.setPath(href, QUrl::TolerantMode);
         } else {
             // href is a complete url
             url = QUrl::fromUserInput(href);
         }
 
         auto _url = url;
         _url.setUserInfo(mCollectionUrl.url().userInfo());
         item.setUrl(DavUrl(_url, mCollectionUrl.protocol()));
 
         // extract etag
         const QDomElement getetagElement = Utils::firstChildElementNS(propElement, QStringLiteral("DAV:"), QStringLiteral("getetag"));
         item.setEtag(getetagElement.text());
 
         // extract content
         const QDomElement dataElement = Utils::firstChildElementNS(propElement,
                                         protocol->responseNamespace(),
                                         protocol->dataTagName());
 
         if (dataElement.isNull()) {
             responseElement = Utils::nextSiblingElementNS(responseElement, QStringLiteral("DAV:"), QStringLiteral("response"));
             continue;
         }
 
         const QByteArray data = dataElement.firstChild().toText().data().toUtf8();
         if (data.isEmpty()) {
             responseElement = Utils::nextSiblingElementNS(responseElement, QStringLiteral("DAV:"), QStringLiteral("response"));
             continue;
         }
 
         item.setData(data);
 
         mItems.insert(item.url().toDisplayString(), item);
         responseElement = Utils::nextSiblingElementNS(responseElement, QStringLiteral("DAV:"), QStringLiteral("response"));
     }
 
     emitResult();
 }