response.cpp

/*
 * SPDX-FileCopyrightText: (C) 2013-2022 Daniel Nicoletti <dantti12@gmail.com>
 * SPDX-License-Identifier: BSD-3-Clause
 */
#include "common.h"
#include "context_p.h"
#include "engine.h"
#include "enginerequest.h"
#include "response_p.h"
 
#include <QCryptographicHash>
#include <QEventLoop>
#include <QJsonArray>
#include <QJsonObject>
#include <QtCore/QJsonDocument>
 
using namespace Cutelyst;
 
Response::Response(const Headers &defaultHeaders, EngineRequest *engineRequest)
    : d_ptr(new ResponsePrivate(defaultHeaders, engineRequest))
{
    open(QIODevice::WriteOnly);
}
 
qint64 Response::readData(char *data, qint64 maxlen)
{
    Q_UNUSED(data)
    Q_UNUSED(maxlen)
    return -1;
}
 
qint64 Response::writeData(const char *data, qint64 len)
{
    Q_D(Response);
 
    if (len <= 0) {
        return len;
    }
 
    // Finalize headers if someone manually writes output
    if (!(d->engineRequest->status & EngineRequest::FinalizedHeaders)) {
        if (d->headers.header("Transfer-Encoding"_qba).compare("chunked") == 0) {
            d->engineRequest->status |= EngineRequest::IOWrite | EngineRequest::Chunked;
        } else {
            // When chunked encoding is not set the client can only know
            // that data is finished if we close the connection
            d->headers.setHeader("Connection"_qba, "Close"_qba);
            d->engineRequest->status |= EngineRequest::IOWrite;
        }
        delete d->bodyIODevice;
        d->bodyIODevice = nullptr;
        d->bodyData     = QByteArray();
 
        d->engineRequest->finalizeHeaders();
    }
 
    return d->engineRequest->write(data, len);
}
 
Response::~Response()
{
    delete d_ptr->bodyIODevice;
    delete d_ptr;
}
 
quint16 Response::status() const noexcept
{
    Q_D(const Response);
    return d->status;
}
 
void Response::setStatus(quint16 status) noexcept
{
    Q_D(Response);
    d->status = status;
}
 
bool Response::hasBody() const noexcept
{
    Q_D(const Response);
    return !d->bodyData.isEmpty() || d->bodyIODevice ||
           d->engineRequest->status & EngineRequest::IOWrite;
}
 
QByteArray &Response::body()
{
    Q_D(Response);
    if (d->bodyIODevice) {
        delete d->bodyIODevice;
        d->bodyIODevice = nullptr;
    }
 
    return d->bodyData;
}
 
QIODevice *Response::bodyDevice() const noexcept
{
    Q_D(const Response);
    return d->bodyIODevice;
}
 
void Response::setBody(QIODevice *body)
{
    Q_D(Response);
    Q_ASSERT(body && body->isOpen() && body->isReadable());
 
    if (!(d->engineRequest->status & EngineRequest::IOWrite)) {
        d->bodyData = QByteArray();
        if (d->bodyIODevice) {
            delete d->bodyIODevice;
        }
        d->bodyIODevice = body;
    }
}
 
void Response::setBody(const QByteArray &body)
{
    Q_D(Response);
    d->setBodyData(body);
}
 
void Response::setJsonBody(const QByteArray &json)
{
    Q_D(Response);
    d->setBodyData(json);
    d->headers.setContentType("application/json"_qba);
}
 
void Response::setJsonObjectBody(const QJsonObject &obj)
{
    setJsonBody(QJsonDocument(obj).toJson(QJsonDocument::Compact));
}
 
void Response::setJsonArrayBody(const QJsonArray &array)
{
    setJsonBody(QJsonDocument(array).toJson(QJsonDocument::Compact));
}
 
QByteArray Response::contentEncoding() const noexcept
{
    Q_D(const Response);
    return d->headers.contentEncoding();
}
 
void Cutelyst::Response::setContentEncoding(const QByteArray &encoding)
{
    Q_D(Response);
    Q_ASSERT_X(!(d->engineRequest->status & EngineRequest::FinalizedHeaders),
               "setContentEncoding",
               "setting a header value after finalize_headers and the response callback has been "
               "called. Not what you want.");
 
    d->headers.setContentEncoding(encoding);
}
 
qint64 Response::contentLength() const
{
    Q_D(const Response);
    return d->headers.contentLength();
}
 
void Response::setContentLength(qint64 length)
{
    Q_D(Response);
    Q_ASSERT_X(!(d->engineRequest->status & EngineRequest::FinalizedHeaders),
               "setContentLength",
               "setting a header value after finalize_headers and the response callback has been "
               "called. Not what you want.");
 
    d->headers.setContentLength(length);
}
 
QByteArray Response::contentType() const
{
    Q_D(const Response);
    return d->headers.contentType();
}
 
QByteArray Response::contentTypeCharset() const
{
    Q_D(const Response);
    return d->headers.contentTypeCharset();
}
 
QVariant Response::cookie(const QByteArray &name) const
{
    Q_D(const Response);
    return QVariant::fromValue(d->cookies.value(name));
}
 
QList<QNetworkCookie> Response::cookies() const
{
    Q_D(const Response);
    return d->cookies.values();
}
 
void Response::setCookie(const QNetworkCookie &cookie)
{
    Q_D(Response);
    d->cookies.insert(cookie.name(), cookie);
}
 
void Response::setCookies(const QList<QNetworkCookie> &cookies)
{
    Q_D(Response);
    for (const QNetworkCookie &cookie : cookies) {
        d->cookies.insert(cookie.name(), cookie);
    }
}
 
int Response::removeCookies(const QByteArray &name)
{
    Q_D(Response);
    return d->cookies.remove(name);
}
 
void Response::redirect(const QUrl &url, quint16 status)
{
    Q_D(Response);
    d->location = url;
    d->status   = status;
 
    if (url.isValid()) {
        const auto location = url.toEncoded(QUrl::FullyEncoded);
        qCDebug(CUTELYST_RESPONSE) << "Redirecting to" << location << status;
 
        d->headers.setHeader("Location"_qba, location);
        d->headers.setContentType("text/html; charset=utf-8"_qba);
 
        const QByteArray buf = R"V0G0N(<!DOCTYPE html>
<html xmlns="http://www.w3.org/1999/xhtml">
  <head>
    <title>Moved</title>
  </head>
  <body>
     <p>This item has moved <a href=")V0G0N" +
                               location + R"V0G0N(">here</a>.</p>
  </body>
</html>
)V0G0N";
        setBody(buf);
    } else {
        d->headers.removeHeader("Location"_qba);
        qCDebug(CUTELYST_ENGINE) << "Invalid redirect removing header" << url << status;
    }
}
 
void Response::redirect(const QString &url, quint16 status)
{
    redirect(QUrl(url), status);
}
 
void Response::redirectSafe(const QUrl &url, const QUrl &fallback)
{
    Q_D(const Response);
    if (url.matches(d->engineRequest->context->req()->uri(),
                    QUrl::RemovePath | QUrl::RemoveQuery)) {
        redirect(url);
    } else {
        redirect(fallback);
    }
}
 
QUrl Response::location() const noexcept
{
    Q_D(const Response);
    return d->location;
}
 
QByteArray Response::header(const QByteArray &field) const noexcept
{
    Q_D(const Response);
    return d->headers.header(field);
}
 
void Response::setHeader(const QByteArray &key, const QByteArray &value)
{
    Q_D(Response);
    Q_ASSERT_X(!(d->engineRequest->status & EngineRequest::FinalizedHeaders),
               "setHeader",
               "setting a header value after finalize_headers and the response callback has been "
               "called. Not what you want.");
 
    d->headers.setHeader(key, value);
}
 
Headers &Response::headers() noexcept
{
    Q_D(Response);
    return d->headers;
}
 
bool Response::isFinalizedHeaders() const noexcept
{
    Q_D(const Response);
    return d->engineRequest->status & EngineRequest::FinalizedHeaders;
}
 
bool Response::isSequential() const noexcept
{
    return true;
}
 
qint64 Response::size() const noexcept
{
    Q_D(const Response);
    if (d->engineRequest->status & EngineRequest::IOWrite) {
        return -1;
    } else if (d->bodyIODevice) {
        return d->bodyIODevice->size();
    } else {
        return d->bodyData.size();
    }
}
 
bool Response::webSocketHandshake(const QByteArray &key,
                                  const QByteArray &origin,
                                  const QByteArray &protocol)
{
    Q_D(Response);
    return d->engineRequest->webSocketHandshake(key, origin, protocol);
}
 
bool Response::webSocketTextMessage(const QString &message)
{
    Q_D(Response);
    return d->engineRequest->webSocketSendTextMessage(message);
}
 
bool Response::webSocketBinaryMessage(const QByteArray &message)
{
    Q_D(Response);
    return d->engineRequest->webSocketSendBinaryMessage(message);
}
 
bool Response::webSocketPing(const QByteArray &payload)
{
    Q_D(Response);
    return d->engineRequest->webSocketSendPing(payload);
}
 
bool Response::webSocketClose(quint16 code, const QString &reason)
{
    Q_D(Response);
    return d->engineRequest->webSocketClose(code, reason);
}
 
void ResponsePrivate::setBodyData(const QByteArray &body)
{
    if (!(engineRequest->status & EngineRequest::IOWrite)) {
        if (bodyIODevice) {
            delete bodyIODevice;
            bodyIODevice = nullptr;
        }
        bodyData = body;
        headers.setContentLength(body.size());
    }
}
 
#include "moc_response.cpp"