mirror of
https://github.com/beefytech/Beef.git
synced 2025-06-08 11:38:21 +02:00
123 lines
2.2 KiB
C++
123 lines
2.2 KiB
C++
#pragma once
|
|
|
|
#include "BeefySysLib/Common.h"
|
|
#include "BeefySysLib/FileStream.h"
|
|
#include "BeefySysLib/util/Deque.h"
|
|
#include "BeefySysLib/util/ThreadPool.h"
|
|
#include "BeefySysLib/util/Dictionary.h"
|
|
|
|
NS_BF_BEGIN
|
|
|
|
#ifdef BF_PLATFORM_WINDOWS
|
|
#define BF_CURL
|
|
#endif
|
|
|
|
class DebugManager;
|
|
class NetManager;
|
|
class NetResult;
|
|
|
|
#ifdef BF_CURL
|
|
typedef void CURL;
|
|
typedef void CURLM;
|
|
#endif
|
|
|
|
class NetRequest : public ThreadPool::Job
|
|
{
|
|
public:
|
|
NetManager* mNetManager;
|
|
|
|
String mURL;
|
|
String mOutPath;
|
|
String mOutTempPath;
|
|
FileStream mOutFile;
|
|
#ifdef BF_CURL
|
|
CURL* mCURL;
|
|
CURLM* mCURLMulti;
|
|
#else
|
|
#endif
|
|
volatile bool mCancelling;
|
|
bool mFailed;
|
|
String mError;
|
|
uint32 mLastUpdateTick;
|
|
bool mShowTracking;
|
|
NetResult* mResult;
|
|
NetResult* mCancelOnSuccess;
|
|
|
|
NetRequest()
|
|
{
|
|
mLastUpdateTick = 0;
|
|
#ifdef BF_CURL
|
|
mCURL = NULL;
|
|
mCURLMulti = NULL;
|
|
#else
|
|
#endif
|
|
mCancelling = false;
|
|
mFailed = false;
|
|
mShowTracking = false;
|
|
mResult = NULL;
|
|
mCancelOnSuccess = NULL;
|
|
}
|
|
~NetRequest();
|
|
|
|
void DoTransfer();
|
|
|
|
void Cleanup();
|
|
void Fail(const StringImpl& error);
|
|
bool Cancel() override;
|
|
void Perform() override;
|
|
void ShowTracking();
|
|
};
|
|
|
|
class NetResult
|
|
{
|
|
public:
|
|
String mURL;
|
|
String mOutPath;
|
|
bool mFailed;
|
|
NetRequest* mCurRequest;
|
|
bool mRemoved;
|
|
SyncEvent* mDoneEvent;
|
|
|
|
NetResult()
|
|
{
|
|
mFailed = false;
|
|
mCurRequest = NULL;
|
|
mRemoved = false;
|
|
mDoneEvent = NULL;
|
|
}
|
|
|
|
~NetResult()
|
|
{
|
|
delete mDoneEvent;
|
|
}
|
|
};
|
|
|
|
class NetManager
|
|
{
|
|
public:
|
|
Dictionary<String, NetResult*> mCachedResults;
|
|
|
|
ThreadPool mThreadPool;
|
|
DebugManager* mDebugManager;
|
|
Deque<NetRequest*> mRequests;
|
|
Array<NetResult*> mOldResults;
|
|
SyncEvent mRequestDoneEvent;
|
|
NetResult* mWaitingResult;
|
|
NetRequest* mWaitingRequest;
|
|
|
|
public:
|
|
NetManager();
|
|
~NetManager();
|
|
|
|
NetRequest* CreateGetRequest(const StringImpl& url, const StringImpl& destPath, bool useCache);
|
|
NetResult* QueueGet(const StringImpl& url, const StringImpl& destPath, bool useCache);
|
|
bool Get(const StringImpl& url, const StringImpl& destPath);
|
|
|
|
void CancelAll();
|
|
void Clear();
|
|
void CancelCurrent();
|
|
void Cancel(NetResult* netResult);
|
|
void SetCancelOnSuccess(NetResult* dependentResult, NetResult* cancelOnSucess);
|
|
};
|
|
|
|
NS_BF_END
|