DownloadQueue.hpp

#pragma once
 
#if !defined(NETWORK) && !defined(NETWORK_MOCK)
#define NETWORK_MOCK
#endif
 
#ifndef NETWORK_MOCK
#include <curl/curl.h>
#include <curl/easy.h>
#endif
 
#include <functional>
#include <string>
#include <list>
 
namespace Chesto {
 
struct DownloadOperation;
 
enum class DownloadStatus
{
    QUEUED,
    DOWNLOADING,
    COMPLETE,
    FAILED,
};
 
struct DownloadOperation
{
    std::string url;
    std::string buffer;
    DownloadStatus status;
#ifndef NETWORK_MOCK
    CURL *eh;
#endif
 
    std::function<void(DownloadOperation*)> cb;
    void *cbdata;
};
 
class DownloadQueue
{
public:
    DownloadQueue();
    ~DownloadQueue();
 
    void downloadAdd(DownloadOperation *download);
 
    void downloadCancel(DownloadOperation *download);
 
    int process();
 
    // static instance
    static void init();
    static void quit();
    static DownloadQueue* downloadQueue;
 
private:
    void transferStart(DownloadOperation *download);
 
    void transferFinish(DownloadOperation *download);
 
    void startTransfersFromQueue();
 
#ifndef NETWORK_MOCK
    // curl multi handle
    CURLM *cm;
 
    void setPlatformCurlFlags(CURL* c);
#endif
 
    std::list<DownloadOperation*> queue;
 
    int transfers = 0;
};
 
} // namespace Chesto