//////////////////////////////////////////////////////////////////////////
//
// ZPatcher - Patcher system - Part of the ZUpdater suite
// Felipe "Zoc" Silveira - (c) 2016-2018
//
//////////////////////////////////////////////////////////////////////////
//
// DownloadFileWriter.h
// Class to download a file from an URL and save to disk
//
//////////////////////////////////////////////////////////////////////////


#ifndef _DOWNLOADFILEWRITER_H_
#define _DOWNLOADFILEWRITER_H_

#include <string>
#include "curl/curl.h"

#ifdef WIN32
	#include <windows.h>
#endif


// Zoc (2016-04-17): It is mandatory to call curl_global_init() before using DownloadFileWriter!
class DownloadFileWriter
{
public:
	typedef					int(*TransferInfoFunc)(void* p, curl_off_t dltotal, curl_off_t dlnow, curl_off_t ultotal, curl_off_t ulnow);

protected:
	DownloadFileWriter*		m_pSelf;
	CURL*					m_CurlHandle;
	double					m_LastUpdateTime;		// Optionally used to limit the transfer info functin update frequency.
	char					m_CurlErrorMessage[CURL_ERROR_SIZE];

	TransferInfoFunc		m_pTransferInfoFunc;

	// Zoc (2016-04-26): File to write
	std::string				m_DestFileName;
	FILE*					m_hDestFile;


public:
	DownloadFileWriter();

	int						PrepareFileToWrite(const std::string& DestFile);
	int						PrepareCurlHandle();
	void					SetTransferInfoFunction(TransferInfoFunc func);
	void					SetupTransfer(const std::string& URL);
	CURLcode				StartDownload();
	void					FinishDownload();

	// Information getters for the transfer function pointer
	CURL*					GetCurlHandle();
	std::string				GetDestFileName();
	double					GetLastUpdateTime();
	void					SetLastUpdateTime(const double& lastUpdate);

	static size_t			Callback_WriteFile(void* Buffer, size_t ElementSize, size_t NumElements, void* DownloadFileWriterPointer);
	static int				TransferInfo(void *p, curl_off_t dltotal, curl_off_t dlnow, curl_off_t ultotal, curl_off_t ulnow);
	static void				PrintProgressBar(const int Percentage, curl_off_t CurrentDownload);

};

#endif // _DOWNLOADFILEWRITER_H_