metaforce/hecl-gui/DownloadManager.hpp

#pragma once

#include <QObject>
#include <QtNetwork>
#include <QNetworkAccessManager>
#include <QProgressBar>
#include <QLabel>

class QuaZip;

class DownloadManager : public QObject
{
    Q_OBJECT
    QNetworkAccessManager m_netManager;
    QNetworkReply* m_indexInProgress = nullptr;
    QNetworkReply* m_binaryInProgress = nullptr;
    QString m_outPath;
    bool m_hasError = false;
    QProgressBar* m_progBar = nullptr;
    QLabel* m_errorLabel = nullptr;
    std::function<void(const QStringList& index)> m_indexCompletionHandler;
    std::function<void(QuaZip& file)> m_completionHandler;
    std::function<void()> m_failedHandler;

    void resetError()
    {
        m_hasError = false;
        if (m_errorLabel)
            m_errorLabel->setText(QString());
    }

    void setError(QNetworkReply::NetworkError error, const QString& errStr)
    {
        if (m_hasError && error == QNetworkReply::OperationCanceledError)
            return;
        m_hasError = true;
        if (m_errorLabel)
            m_errorLabel->setText(errStr);
    }

    void _validateCert(QNetworkReply* reply);

public:
    explicit DownloadManager(QObject* parent = Q_NULLPTR)
    : QObject(parent), m_netManager(this) {}
    void connectWidgets(QProgressBar* progBar, QLabel* errorLabel,
                        std::function<void(const QStringList& index)>&& indexCompletionHandler,
                        std::function<void(QuaZip& file)>&& completionHandler,
                        std::function<void()>&& failedHandler)
    {
        m_progBar = progBar;
        m_errorLabel = errorLabel;
        m_indexCompletionHandler = std::move(indexCompletionHandler);
        m_completionHandler = std::move(completionHandler);
        m_failedHandler = std::move(failedHandler);
    }
    void fetchIndex();
    void fetchBinary(const QString& str, const QString& outPath);
    bool hasError() const { return m_hasError; }

public slots:
    void indexFinished();
    void indexError(QNetworkReply::NetworkError error);
    void indexValidateCert();

    void binaryFinished();
    void binaryError(QNetworkReply::NetworkError error);
    void binaryValidateCert();
    void binaryDownloadProgress(qint64 bytesReceived, qint64 bytesTotal);

};