/*
FatRat download manager
http://fatrat.dolezel.info

Copyright (C) 2006-2010 Lubos Dolezel <lubos a dolezel.info>

This program is free software; you can redistribute it and/or
modify it under the terms of the GNU General Public License
version 2 as published by the Free Software Foundation.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

In addition, as a special exemption, Luboš Doležel gives permission
to link the code of FatRat with the OpenSSL project's
"OpenSSL" library (or with modified versions of it that use the; same
license as the "OpenSSL" library), and distribute the linked
executables. You must obey the GNU General Public License in all
respects for all of the code used other than "OpenSSL".
*/


#ifndef CURL_POLLER
#define CURL_POLLER
#include <QThread>
#include <QMutex>
#include <QMap>
#include <QHash>
#include <QQueue>
#include <curl/curl.h>
#include "engines/CurlUser.h"
#include "poller/Poller.h"

class CurlPollingMaster;

class CurlPoller : public QThread
{
public:
	CurlPoller();
	~CurlPoller();
	
	void addTransfer(CurlUser* obj);
	// will handle the underlying CURL* too
	void removeTransfer(CurlUser* obj, bool nodeep = false);
	//void removeSafely(CURL* curl);
	void addTransfer(CurlPollingMaster* obj);
	void removeTransfer(CurlPollingMaster* obj);
	
	void run();
	void checkErrors(timeval tvNow);
	
	static CurlPoller* instance() { return m_instance; }
protected:
	void epollEnable(int socket, int events);
	void pollingCycle(bool oneshot);
	static int socket_callback(CURL* easy, curl_socket_t s, int action, CurlPoller* This, void* socketp);
	static int timer_callback(CURLM* multi, long newtimeout, long* timeout);
	static void setTransferTimeout(int timeout);
	static int getTransferTimeout() { return m_nTransferTimeout; }
protected:
	static CurlPoller* m_instance;
	static int m_nTransferTimeout;

	bool m_bAbort;
	CURLM* m_curlm;
	Poller* m_poller;
	int m_curlTimeout;
	long m_timeout;
	
	typedef QMap<int, QPair<int,CurlStat*> > sockets_hash;
	
	QMap<CURL*, CurlUser*> m_users;
	QMap<int, CurlPollingMaster*> m_masters;
	sockets_hash m_sockets;
	QMutex m_usersLock;
	QQueue<CurlUser*> m_queueToDelete;
	
	QList<int> m_socketsToRemove;
	sockets_hash m_socketsToAdd;

	friend class HttpFtpSettings;
	friend class CurlDownload;
	friend class CurlPollingMaster;
	friend class CurlUser;
};

#endif
