mirror-chatterino2/src/common/NetworkRequest.hpp

#pragma once

#include "Application.hpp"
#include "common/NetworkCommon.hpp"
#include "common/NetworkData.hpp"
#include "common/NetworkRequester.hpp"
#include "common/NetworkResult.hpp"
#include "common/NetworkTimer.hpp"
#include "common/NetworkWorker.hpp"

namespace chatterino {

class NetworkRequest
{
    // Stores all data about the request that needs to be passed around to each part of the request
    std::shared_ptr<NetworkData> data;

    // Timer that tracks the timeout
    // By default, there's no explicit timeout for the request
    // to enable the timer, the "setTimeout" function needs to be called before execute is called
    std::unique_ptr<NetworkTimer> timer;

    // The NetworkRequest destructor will assert if executed_ hasn't been set to true before dying
    bool executed_ = false;

public:
    NetworkRequest() = delete;
    NetworkRequest(const NetworkRequest &other) = delete;
    NetworkRequest &operator=(const NetworkRequest &other) = delete;

    NetworkRequest(NetworkRequest &&other) = default;
    NetworkRequest &operator=(NetworkRequest &&other) = default;

    explicit NetworkRequest(const std::string &url,
                            NetworkRequestType requestType = NetworkRequestType::Get);
    NetworkRequest(QUrl url, NetworkRequestType requestType = NetworkRequestType::Get);

    ~NetworkRequest();

    void setRequestType(NetworkRequestType newRequestType);

    void onReplyCreated(NetworkReplyCreatedCallback cb);
    void onError(NetworkErrorCallback cb);
    void onSuccess(NetworkSuccessCallback cb);

    void setPayload(const QByteArray &payload);
    void setUseQuickLoadCache(bool value);
    void setCaller(const QObject *caller);
    void setRawHeader(const char *headerName, const char *value);
    void setRawHeader(const char *headerName, const QByteArray &value);
    void setRawHeader(const char *headerName, const QString &value);
    void setTimeout(int ms);
    void makeAuthorizedV5(const QString &clientID, const QString &oauthToken = QString());

    void execute();

private:
    // Returns true if the file was successfully loaded from cache
    // Returns false if the cache file either didn't exist, or it contained "invalid" data
    // "invalid" is specified by the onSuccess callback
    bool tryLoadCachedFile();

    void doRequest();

public:
    // Helper creator functions
    static NetworkRequest twitchRequest(QUrl url);
};

}  // namespace chatterino