iw4x-client/src/Components/Modules/Download.hpp

namespace Components
{
	class Download : public Component
	{
	public:
		Download();
		~Download();

#ifdef DEBUG
		const char* GetName() { return "Download"; };
#endif

		static void InitiateClientDownload(std::string mod);

	private:
		class ClientDownload
		{
		public:
			ClientDownload() : Valid(false), Running(false) {}
			~ClientDownload() { this->Clear(); }

			bool Running;
			bool Valid;
			mg_mgr Mgr;
			Network::Address Target;
			std::string Mod;
			std::mutex Mutex;
			std::thread Thread;
			std::string Progress;

			class File
			{
			public:
				std::string Name;
				std::string Hash;
				size_t Size;
			};

			std::vector<File> Files;

			void Clear()
			{
				this->Running = false;
				this->Mod.clear();
				this->Files.clear();

				if (this->Valid)
				{
					this->Valid = false;
					mg_mgr_free(&(this->Mgr));
				}

				this->Mutex.lock();
				this->Progress.clear();
				this->Mutex.unlock();
			}
		};

		class FileDownload
		{
		public:
			ClientDownload* download;
			ClientDownload::File file;

			bool downloading;
			unsigned int index;
			std::string buffer;
			size_t receivedBytes;
		};

		static mg_mgr Mgr;
		static ClientDownload CLDownload;

		static void EventHandler(mg_connection *nc, int ev, void *ev_data);
		static void ListHandler(mg_connection *nc, int ev, void *ev_data);
		static void FileHandler(mg_connection *nc, int ev, void *ev_data);
		static void InfoHandler(mg_connection *nc, int ev, void *ev_data);
		static void DownloadHandler(mg_connection *nc, int ev, void *ev_data);

		static bool IsClient(mg_connection *nc);
		static Game::client_t* GetClient(mg_connection *nc);
		static void Forbid(mg_connection *nc);

		static void ModDownloader(ClientDownload* download);
		static bool ParseModList(ClientDownload* download, std::string list);
		static bool DownloadFile(ClientDownload* download, unsigned int index);
	};
}