davjob.cpp

/*
    Copyright (c) 2014 Gregory Oestreicher <greg@kamago.net>
 
    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 "davjob.h"
 
#include "davmanager.h"
#include "libkdav2_debug.h"
 
#include <QTextStream>
 
using namespace KDAV2;
 
class DavJobPrivate {
public:
    QByteArray data;
    QDomDocument doc;
    QUrl url;
 
    QString location;
    QString etag;
    QString contentType;
    QNetworkReply::NetworkError responseCode = QNetworkReply::NoError;
    int httpStatusCode = 0;
};
 
DavJob::DavJob(QNetworkReply *reply, QUrl url, QObject *parent)
    : KJob(parent),
    d(new DavJobPrivate)
{
    d->url = url;
    connectToReply(reply);
}
 
DavJob::~DavJob()
{
}
 
void DavJob::connectToReply(QNetworkReply *reply)
{
    QObject::connect(reply, &QNetworkReply::readyRead, this, [=] () {
        d->data.append(reply->readAll());
    });
    QObject::connect(reply, static_cast<void(QNetworkReply::*)(QNetworkReply::NetworkError)>(&QNetworkReply::error), this, [=] (QNetworkReply::NetworkError error) {
        qCWarning(KDAV2_LOG) << "Network error:" << error << "Message:" << reply->errorString() << "HTTP Status code:" << reply->attribute(QNetworkRequest::HttpStatusCodeAttribute).toInt() << "\nAvailable data:" << reply->readAll();
    });
    QObject::connect(reply, &QNetworkReply::metaDataChanged, this, [=] () {
        qCDebug(KDAV2_LOG) << "Metadata changed: " << reply->rawHeaderPairs();
        d->location = reply->rawHeader("Location");
        d->etag = reply->rawHeader("ETag");
        //"text/x-vcard; charset=utf-8" -> "text/x-vcard"
        d->contentType = reply->rawHeader("Content-Type").split(';').first();
    });
    QObject::connect(reply, &QNetworkReply::finished, this, [=] () {
        //This is a workaround for QNetworkAccessManager::setRedirectPolicy(QNetworkRequest::UserVerifiedRedirectPolicy),
        //which does not seem to work with multiple redirects.
        const auto possibleRedirectUrl = reply->attribute(QNetworkRequest::RedirectionTargetAttribute).toUrl();
        if(!possibleRedirectUrl.isEmpty()) {
            qCDebug(KDAV2_LOG) << "Redirecting to " << possibleRedirectUrl;
            auto request = reply->request();
            request.setUrl(possibleRedirectUrl);
            reply->disconnect(this);
 
            //Set in QWebdav
            const auto requestData = reply->property("requestData").toByteArray();
            d->data.clear();
 
            auto redirectReply = [&] {
                if (reply->property("isPut").toBool()) {
                    return DavManager::networkAccessManager()->put(request, requestData);
                }
                return DavManager::networkAccessManager()->sendCustomRequest(request, request.attribute(QNetworkRequest::CustomVerbAttribute).toByteArray(), requestData);
            }();
            redirectReply->setProperty("requestData", requestData);
            connectToReply(redirectReply);
            return;
        }
 
        //Could have changed due to redirects
        d->url = reply->url();
 
        d->doc.setContent(d->data, true);
 
        if (KDAV2_LOG().isDebugEnabled()) {
            QTextStream stream(stdout, QIODevice::WriteOnly);
            d->doc.save(stream, 2);
        }
 
        d->responseCode = reply->error();
        d->httpStatusCode = reply->attribute(QNetworkRequest::HttpStatusCodeAttribute).toInt();
        if (d->responseCode) {
            setError(KJob::UserDefinedError);
            setErrorText(reply->errorString());
        } else if (d->httpStatusCode >= 400) {
            qWarning() << "No error set even though we clearly have an http error?" << d->responseCode << d->httpStatusCode;
            Q_ASSERT(false);
            setError(KJob::UserDefinedError);
            setErrorText(reply->errorString());
        }
        emitResult();
    });
 
}
 
void DavJob::start()
{
}
 
QDomDocument DavJob::response() const
{
    return d->doc;
}
 
QByteArray DavJob::data() const
{
    return d->data;
}
 
QUrl DavJob::url() const
{
    return d->url;
}
 
QString DavJob::getLocationHeader() const
{
    return d->location;
}
 
QString DavJob::getETagHeader() const
{
    return d->etag;
}
 
QString DavJob::getContentTypeHeader() const
{
    return d->contentType;
}
 
QNetworkReply::NetworkError DavJob::responseCode() const
{
    return d->responseCode;
}
 
int DavJob::httpStatusCode() const
{
    return d->httpStatusCode;
}