card_picture_loader_worker.h

Go to the documentation of this file.

#ifndef PICTURE_LOADER_WORKER_H
#define PICTURE_LOADER_WORKER_H
 
#include "card_picture_loader_local.h"
#include "card_picture_loader_worker_work.h"
#include "card_picture_to_load.h"
 
#include <QDateTime>
#include <QHash>
#include <QLoggingCategory>
#include <QMutex>
#include <QNetworkAccessManager>
#include <QNetworkDiskCache>
#include <QObject>
#include <QQueue>
#include <QTimer>
#include <libcockatrice/card/card_info.h>
#include <libcockatrice/card/database/card_database.h>
 
#define REDIRECT_HEADER_NAME "redirects"
#define REDIRECT_ORIGINAL_URL "original"
#define REDIRECT_URL "redirect"
#define REDIRECT_TIMESTAMP "timestamp"
#define REDIRECT_CACHE_FILENAME "cache.ini"
 
inline Q_LOGGING_CATEGORY(CardPictureLoaderWorkerLog, "card_picture_loader.worker")
 
    class CardPictureLoaderWorkerWork;

class CardPictureLoaderWorker : public QObject
{
    Q_OBJECT
 
public:
    explicit CardPictureLoaderWorker();
 
    ~CardPictureLoaderWorker() override;

    void enqueueImageLoad(const ExactCard &card);

    void queueRequest(const QUrl &url, CardPictureLoaderWorkerWork *worker);

    void onHostRateLimited(const QString &host);

    void clearNetworkCache();
 
public slots:
    QNetworkReply *makeRequest(const QUrl &url, CardPictureLoaderWorkerWork *workThread);

    void processQueuedRequests();

    bool processSingleRequest();

    void handleImageLoaded(const ExactCard &card, const QImage &image);

    void cacheRedirect(const QUrl &originalUrl, const QUrl &redirectUrl);

    void removedCachedUrl(const QUrl &url);
 
private:
    QThread *pictureLoaderThread;                      
    QNetworkAccessManager *networkManager;             
    QNetworkDiskCache *cache;                          
    QHash<QUrl, QPair<QUrl, QDateTime>> redirectCache; 
    QString cacheFilePath;                             
    static constexpr int CacheTTLInDays = 30;          
    bool picDownload;                                  
    QQueue<QPair<QUrl, CardPictureLoaderWorkerWork *>> requestLoadQueue; 
 
    int requestQuota;                       
    QTimer requestTimer;                    
    QHash<QString, int> hostRequestQuota;   
    QHash<QString, int> hostQuotaRemaining; 
    QHash<QString, QDateTime> hostLast429;  
 
    CardPictureLoaderLocal *localLoader; 
    QSet<QString> currentlyLoading;      

    [[nodiscard]] QUrl getCachedRedirect(const QUrl &originalUrl) const;

    void loadRedirectCache();

    void saveRedirectCache() const;

    void cleanStaleEntries();
 
private slots:
    void resetRequestQuota();

    void handleImageLoadEnqueued(const ExactCard &card);
 
signals:
    void imageLoadEnqueued(const ExactCard &card);

    void imageLoaded(const ExactCard &card, const QImage &image);

    void imageRequestQueued(const QUrl &url, const ExactCard &card, const QString &setName);

    void imageRequestSucceeded(const QUrl &url);
};
 
#endif // PICTURE_LOADER_WORKER_H