qt-migration-legatus/qt6-after/qt6-migrated-legatus/Lts/Common/httq/include/httq/HttpRequest.h

#pragma once

#include "httq_export.h"

#include <QByteArray>
#include <QMap>
#include <QObject>
#include <QString>
#include <QUrl>
#include <QUrlQuery>
#include <http_parser.h>

class QTcpSocket;
class QIODevice;

namespace httq
{
class AbstractServer;
class Logger;
class DataStream;
class HttpRequest;

struct HttpRequestData
{
    QByteArray mData;
    QUrl mUrl;
    QString mCurrentHeaderField;
    QMap<QString, QString> mHeaders;
    bool mDone{false};
    QByteArray mBodyPartial;
    QUrlQuery mQuery;
    HttpRequest* mHttpRquest;
};

class HTTQ_EXPORT HttpRequest : public QObject
{
    Q_OBJECT

  private:
    explicit HttpRequest(QTcpSocket* cli, AbstractServer* parent = nullptr);

  public:
    ~HttpRequest();

    qint64 availableBytes() const;

    http_method method() const
    {
        return (enum http_method)mParser.method;
    }

    QString methodString() const
    {
        return QString::fromUtf8(http_method_str(method()));
    }

    const QUrl& url() const
    {
        return mData.mUrl;
    }

    const QUrlQuery& query() const
    {
        return mData.mQuery;
    }

    qint64 contentLength() const
    {
        return mContentLength;
    }

    const QMap<QString, QString>& requestHeaders() const
    {
        return mData.mHeaders;
    }

    Logger* logger()
    {
        return mLogger;
    }

    QString toString() const;

    void addHeader(const QString& key, const QString& value);
    void write(int status, const QByteArray& data, const QString& contentType);
    void write(int status, const QJsonValue& jsonValue);
    DataStream* createDataStreamFromClient(QIODevice* to, qint64 fileSize, qint64 bufferSize = 1024 * 1024);
    DataStream* createDataStreamToClient(int status, const QString& contentType, QIODevice* from, qint64 fileSize,
                                         qint64 bufferSize = 1024 * 1024);

    friend class DataStream;

  private:
    friend class AbstractServer;

    void writeStatusHeader(int status);
    void writeHeaders();

    static int on_message_begin(http_parser* parser);
    static int on_url(http_parser* parser, const char* at, size_t length);
    static int on_status(http_parser* parser, const char* at, size_t length);
    static int on_header_field(http_parser* parser, const char* at, size_t length);
    static int on_header_value(http_parser* parser, const char* at, size_t length);
    static int on_headers_complete(http_parser* parser);
    static int on_body(http_parser* parser, const char* at, size_t length);
    static int on_message_complete(http_parser* parser);
    static int on_chunk_header(http_parser* parser);
    static int on_chunk_complete(http_parser* parser);

    static HttpRequestData* data(http_parser* parser)
    {
        return static_cast<HttpRequestData*>(parser->data);
    }

    static Logger* dataLogger(http_parser* parser)
    {
        return static_cast<HttpRequestData*>(parser->data)->mHttpRquest->logger();
    }

    QTcpSocket* mCli;
    http_parser mParser;
    http_parser_settings mParserSettings;
    HttpRequestData mData;
    bool mDone{false};
    qint64 mContentLength{0};
    qint64 alreadyRead{0};
    QMap<QString, QString> mHeaders;
    Logger* mLogger;

  private slots:
    void slotReadyRead();

  signals:
    void signalUpgrade(QTcpSocket* sock);
    void signalReady();
};
} // namespace httq