pmrwebserviceresponse.cpp

/*******************************************************************************

Copyright (C) The University of Auckland

OpenCOR 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 3 of the License, or
(at your option) any later version.

OpenCOR 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, see <https://gnu.org/licenses>.

*******************************************************************************/

//==============================================================================
// PMR Web service response
//==============================================================================

#include "pmrwebservice.h"
#include "pmrwebserviceresponse.h"

//==============================================================================

#include <QNetworkReply>

//==============================================================================

#include <array>

//==============================================================================

#include "zlib.h"

//==============================================================================

namespace OpenCOR {
namespace PMRSupport {

//==============================================================================

PmrWebServiceResponse::PmrWebServiceResponse(QNetworkReply *pNetworkReply) :
    mNetworkReply(pNetworkReply)
{
    // Process the response once we have got it

    connect(mNetworkReply, &QNetworkReply::finished,
            this, &PmrWebServiceResponse::processResponse);
}

//==============================================================================

void PmrWebServiceResponse::processResponse()
{
    // Retrieve the contents of the response and uncompress it if needed

    QByteArray jsonResponse = QByteArray();

    if (mNetworkReply->rawHeader("Content-Encoding") != "gzip") {
        // The response is not compressed, so just retrieve its raw contents

        jsonResponse = mNetworkReply->readAll();
    } else {
        // Retrieve the raw contents of the response and uncompress it

        QByteArray compressedJsonResponse = mNetworkReply->readAll();
        z_stream stream;

        memset(&stream, 0, sizeof(z_stream));

        if (inflateInit2_(&stream, MAX_WBITS+16, ZLIB_VERSION, sizeof(z_stream)) == Z_OK) {
            enum {
                BufferSize = 32768
            };

            std::array<Bytef, BufferSize> buffer = {};

            stream.next_in = reinterpret_cast<Bytef *>(compressedJsonResponse.data());
            stream.avail_in = uint(compressedJsonResponse.size());

            do {
                stream.next_out = buffer.data();
                stream.avail_out = BufferSize;

                inflate(&stream, Z_NO_FLUSH);

                if (stream.msg == nullptr) {
                    jsonResponse += QByteArray::fromRawData(reinterpret_cast<char *>(buffer.data()), int(BufferSize-stream.avail_out));
                } else {
                    jsonResponse = QByteArray();
                }
            } while (stream.avail_out == 0);

            inflateEnd(&stream);
        }
    }

    // Check for network errors

    static const QStringList ResponseMimeTypes = QStringList() << "application/json"
                                                               << RequestMimeType
                                                               << CollectionMimeType;

    int httpStatusCode = mNetworkReply->attribute(QNetworkRequest::HttpStatusCodeAttribute).toInt();

    if (mNetworkReply->error() != QNetworkReply::NoError) {
        if (httpStatusCode == 403) {
            emit forbidden(mNetworkReply->url().toString());
        } else {
            emit error(mNetworkReply->errorString());
        }
    } else if (httpStatusCode == 302) {
        emit found(mNetworkReply->header(QNetworkRequest::LocationHeader).toString());
    } else if (!ResponseMimeTypes.contains(mNetworkReply->header(QNetworkRequest::ContentTypeHeader).toString())) {
        emit error(tr("PMR response has unexpected content type"));
    } else {
        // Parse our response

        QJsonParseError jsonParseError = QJsonParseError();
        QJsonDocument jsonDocument = QJsonDocument::fromJson(jsonResponse, &jsonParseError);

        // Check for parse errors

        if (jsonParseError.error != QJsonParseError::NoError) {
            emit error(jsonParseError.errorString());
        } else {
            emit response(jsonDocument);
        }
    }

    emit busy(false);
    emit finished();

    mNetworkReply->deleteLater();
}

//==============================================================================

} // namespace PMRSupport
} // namespace OpenCOR

//==============================================================================
// End of file
//==============================================================================