siadrive/SiaDrive.Api/UploadManager.h

#pragma once
#include "AutoThread.h"
#include "SQLiteCpp/Database.h"
#include <deque>
#include <EventSystem.h>

NS_BEGIN(Sia)
NS_BEGIN(Api)

class AFX_EXT_CLASS CUploadManager :
	public CAutoThread
{
public:
	enum class _UploadStatus : unsigned
	{
		NotFound,
		Copying,
		Queued,
		Modified,
		Uploading,
		Complete,
		Error
	};

	enum class _UploadError
	{
		Success,
		SourceFileNotFound
	};

private:
	typedef struct
	{
		std::uint64_t Id;
		String SiaPath;
		String FilePath;
		String TempPath;
		_UploadStatus Status;
	} UploadData;

public:
	CUploadManager(const CSiaCurl& siaCurl, CSiaDriveConfig* siaDriveConfig);

public:
	virtual ~CUploadManager();

private:
	SQLite::Database _uploadDatabase;
	std::mutex _uploadMutex;
	CSiaDriveConfig* _siaDriveConfig;
	CAutoThread _fileThread;
	std::mutex _fileQueueMutex;
	std::deque<std::function<void()>> _fileQueue;

private:
	void FileUploadAction(const String& siaPath, const String& filePath, const String& tempSourcePath, const String& siaDriveFilePath);

protected:
	virtual void AutoThreadCallback(const CSiaCurl& siaCurl, CSiaDriveConfig* siaDriveConfig) override;
	void FileThreadCallback(const CSiaCurl& siaCurl, CSiaDriveConfig* siaDriveConfig);

public:
	static String UploadStatusToString(const _UploadStatus& uploadStatus);

public:
	_UploadStatus GetUploadStatus(const String& siaPath);
	_UploadError AddOrUpdate(const String& siaPath, String filePath);
	_UploadError Remove(const String& siaPath);
	void PurgeCompleteStatus();
	void PurgeErrorStatus();
};

typedef Sia::Api::CUploadManager::_UploadStatus UploadStatus;
typedef Sia::Api::CUploadManager::_UploadError UploadError;

// Event Notifications
class CreatingTemporarySiaDriveFile :
	public CEvent
{
public:
	CreatingTemporarySiaDriveFile(const String& siaPath, const String& filePath, const String& tempSourcePath) :
		_siaPath(siaPath),
		_filePath(filePath),
		_tempSourcePath(tempSourcePath)
	{

	}

public:
	virtual ~CreatingTemporarySiaDriveFile()
	{
	}

private:
	const String _siaPath;
	const String _filePath;
	const String _tempSourcePath;

public:
	virtual String GetSingleLineMessage() const override
	{
		return L"CreatingTemporarySiaDriveFile|SP|" + _siaPath + L"|FP|" + _filePath + L"|TSP|" + _tempSourcePath;
	}
};

class CreatingTemporarySiaDriveFileFailed :
	public CEvent
{
public:
	CreatingTemporarySiaDriveFileFailed(const String& siaPath, const String& filePath, const String& tempSourcePath) :
		_siaPath(siaPath),
		_filePath(filePath),
		_tempSourcePath(tempSourcePath)
	{

	}

public:
	virtual ~CreatingTemporarySiaDriveFileFailed()
	{
	}

private:
	const String _siaPath;
	const String _filePath;
	const String _tempSourcePath;

public:
	virtual String GetSingleLineMessage() const override
	{
		return L"CreatingTemporarySiaDriveFileFailed|SP|" + _siaPath + L"|FP|" + _filePath + L"|TSP|" + _tempSourcePath;
	}
};

class DeleteSiaDriveFileFailed :
	public CEvent
{
public:
	DeleteSiaDriveFileFailed(const String& siaPath, const String& filePath, const String& siaDriveFilePath) :
		_siaPath(siaPath),
		_filePath(filePath),
		_siaDriveFilePath(siaDriveFilePath)
	{

	}

public:
	virtual ~DeleteSiaDriveFileFailed()
	{
	}

private:
	const String _siaPath;
	const String _filePath;
	const String _siaDriveFilePath;

public:
	virtual String GetSingleLineMessage() const override
	{
		return L"DeleteSiaDriveFileFailed|SP|" + _siaPath + L"|FP|" + _filePath + L"|SDP|" + _siaDriveFilePath;
	}
};

class DeleteTemporarySiaDriveFileFailed :
	public CEvent
{
public:
	DeleteTemporarySiaDriveFileFailed(const String& siaPath, const String& filePath, const String& tempSourcePath) :
		_siaPath(siaPath),
		_filePath(filePath),
		_tempSourcePath(tempSourcePath)
	{

	}

public:
	virtual ~DeleteTemporarySiaDriveFileFailed()
	{
	}

private:
	const String _siaPath;
	const String _filePath;
	const String _tempSourcePath;

public:
	virtual String GetSingleLineMessage() const override
	{
		return L"DeleteTemporarySiaDriveFileFailed|SP|" + _siaPath + L"|FP|" + _filePath + L"|TSP|" + _tempSourcePath;
	}
};

class RenamingTemporarySiaDriveFile :
	public CEvent
{
public:
	RenamingTemporarySiaDriveFile(const String& siaPath, const String& filePath, const String& tempSourcePath, const String& siaDriveFilePath) :
		_siaPath(siaPath),
		_filePath(filePath),
		_tempSourcePath(tempSourcePath),
		_siaDriveFilePath(siaDriveFilePath)
	{

	}

public:
	virtual ~RenamingTemporarySiaDriveFile()
	{
	}

private:
	const String _siaPath;
	const String _filePath;
	const String _tempSourcePath;
	const String _siaDriveFilePath;

public:
	virtual String GetSingleLineMessage() const override
	{
		return L"RenamingTemporarySiaDriveFile|SP|" + _siaPath + L"|FP|" + _filePath + L"|TSP|" + _tempSourcePath + L"|SDP|" + _siaDriveFilePath;
	}
};

class RenamingTemporarySiaDriveFileFailed :
	public CEvent
{
public:
	RenamingTemporarySiaDriveFileFailed(const String& siaPath, const String& filePath, const String& tempSourcePath, const String& siaDriveFilePath) :
		_siaPath(siaPath),
		_filePath(filePath),
		_tempSourcePath(tempSourcePath),
		_siaDriveFilePath(siaDriveFilePath)
	{

	}

public:
	virtual ~RenamingTemporarySiaDriveFileFailed()
	{
	}

private:
	const String _siaPath;
	const String _filePath;
	const String _tempSourcePath;
	const String _siaDriveFilePath;

public:
	virtual String GetSingleLineMessage() const override
	{
		return L"RenamingTemporarySiaDriveFileFailed|SP|" + _siaPath + L"|FP|" + _filePath + L"|TSP|" + _tempSourcePath + L"|SDP|" + _siaDriveFilePath;
	}
};

class ExistingUploadFound :
	public CEvent
{
public:
	ExistingUploadFound(const String& siaPath, const String& filePath, const UploadStatus& uploadStatus) :
		_siaPath(siaPath),
		_filePath(filePath),
		_uploadStatus(uploadStatus)
	{

	}

public:
	virtual ~ExistingUploadFound()
	{
	}

private:
	const String _siaPath;
	const String _filePath;
	const UploadStatus _uploadStatus;

public:
	virtual String GetSingleLineMessage() const override
	{
		return L"ExistingUploadFound|SP|" + _siaPath + L"|FP|" + _filePath + L"|ST|" + CUploadManager::UploadStatusToString(_uploadStatus);
	}
};

class NewUploadAdded :
	public CEvent
{
public:
	NewUploadAdded(const String& siaPath, const String& filePath, const String& siaDriveFilePath) :
		_siaPath(siaPath),
		_filePath(filePath),
		_siaDriveFilePath(siaDriveFilePath)
	{

	}

public:
	virtual ~NewUploadAdded()
	{
	}

private:
	const String _siaPath;
	const String _filePath;
	const String _siaDriveFilePath;

public:
	virtual String GetSingleLineMessage() const override
	{
		return L"NewUploadAdded|SP|" + _siaPath + L"|FP|" + _filePath + L"|SDP|" + _siaDriveFilePath;
	}
};

class SourceFileNotFound :
	public CEvent
{
public:
	SourceFileNotFound(const String& siaPath, const String& filePath) :
		_siaPath(siaPath),
		_filePath(filePath)
	{

	}

public:
	virtual ~SourceFileNotFound()
	{
	}

private:
	const String _siaPath;
	const String _filePath;

public:
	virtual String GetSingleLineMessage() const override
	{
		return L"SourceFileNotFound|SP|" + _siaPath + L"|FP|" + _filePath;
	}
};

NS_END(2)