#ifndef _MODULES_H
#define _MODULES_H

#include "main.h"
#include <dlfcn.h>
#include <vector>
using std::vector;

#define VERSION 0.034
#define MODULEDEFS(CLASS) extern "C" { CModule* Load(void* p, CUser* pUser, const CString& sModName); void Unload(CModule* pMod); double GetVersion(); } double GetVersion() { return VERSION; } CModule* Load(void* p, CUser* pUser, const CString& sModName) { return new CLASS(p, pUser, sModName); } void Unload(CModule* pMod) { if (pMod) { delete pMod; } }
#define MODCONSTRUCTOR(CLASS) CLASS(void *pDLL, CUser* pUser, const CString& sModName) : CModule(pDLL, pUser, sModName)

// Forward Declarations
class CZNC;
class CUser;
class CNick;
class CChan;
class Csock;
class CModule;
template<class T> class TSocketManager;
// !Forward Declarations

class CTimer : public CCron {
public:
	CTimer(CModule* pModule, unsigned int uInterval, unsigned int uCycles, const CString& sLabel, const CString& sDescription);

	virtual ~CTimer();

	// Setters
	void SetModule(CModule* p);
	void SetDescription(const CString& s);
	// !Setters

	// Getters
	CModule* GetModule() const;
	const CString& GetDescription() const;
	// !Getters
private:
protected:
	CModule*	m_pModule;
	CString		m_sDescription;
};

class CModInfo {
public:

	CModInfo(const CString& sName, const CString& sPath, bool bSystem) {
		m_bSystem = bSystem;
		m_sName = sName;
		m_sPath = sPath;
	}
	virtual ~CModInfo() {}

	bool operator < (const CModInfo& Info) const {
		return (GetName() < Info.GetName());
	}

	// Getters
	const CString& GetName() const { return m_sName; }
	const CString& GetPath() const { return m_sPath; }
	bool IsSystem() const { return m_bSystem; }
	// !Getters
private:
protected:
	bool	m_bSystem;
	CString	m_sName;
	CString	m_sPath;
};

class CModule {
public:
	CModule(void* pDLL, CUser* pUser, const CString& sModName);
	virtual ~CModule();

	virtual CString GetDescription();

	virtual bool OnLoad(const CString& sArgs);
	virtual bool OnBoot();
	virtual void OnUserAttached();
	virtual void OnUserDetached();
	virtual void OnIRCDisconnected();
	virtual void OnIRCConnected();

	virtual bool OnDCCUserSend(const CNick& RemoteNick, unsigned long uLongIP, unsigned short uPort, const CString& sFile, unsigned long uFileSize);

	virtual void OnOp(const CNick& OpNick, const CNick& Nick, const CChan& Channel, bool bNoChange);
	virtual void OnDeop(const CNick& OpNick, const CNick& Nick, const CChan& Channel, bool bNoChange);
	virtual void OnVoice(const CNick& OpNick, const CNick& Nick, const CChan& Channel, bool bNoChange);
	virtual void OnDevoice(const CNick& OpNick, const CNick& Nick, const CChan& Channel, bool bNoChange);
	virtual void OnRawMode(const CNick& OpNick, const CChan& Channel, const CString& sModes, const CString& sArgs);

	virtual bool OnUserRaw(CString& sLine);
	virtual bool OnRaw(CString& sLine);

	virtual bool OnStatusCommand(const CString& sCommand);
	virtual void OnModCommand(const CString& sCommand);
	virtual void OnModNotice(const CString& sMessage);
	virtual void OnModCTCP(const CString& sMessage);

	virtual void OnQuit(const CNick& Nick, const CString& sMessage, const vector<CChan*>& vChans);
	virtual void OnNick(const CNick& Nick, const CString& sNewNick, const vector<CChan*>& vChans);
	virtual void OnKick(const CNick& Nick, const CString& sOpNick, const CChan& Channel, const CString& sMessage);
	virtual void OnJoin(const CNick& Nick, const CChan& Channel);
	virtual void OnPart(const CNick& Nick, const CChan& Channel);

	virtual bool OnUserCTCPReply(const CNick& Nick, CString& sMessage);
	virtual bool OnCTCPReply(const CNick& Nick, CString& sMessage);
	virtual bool OnUserCTCP(const CString& sTarget, CString& sMessage);
	virtual bool OnPrivCTCP(const CNick& Nick, CString& sMessage);
	virtual bool OnChanCTCP(const CNick& Nick, const CChan& Channel, CString& sMessage);
	virtual bool OnUserMsg(const CString& sTarget, CString& sMessage);
	virtual bool OnPrivMsg(const CNick& Nick, CString& sMessage);
	virtual bool OnChanMsg(const CNick& Nick, const CChan& Channel, CString& sMessage);
	virtual bool OnUserNotice(const CString& sTarget, CString& sMessage);
	virtual bool OnPrivNotice(const CNick& Nick, CString& sMessage);
	virtual bool OnChanNotice(const CNick& Nick, const CChan& Channel, CString& sMessage);

	void * GetDLL();
	static double GetVersion() { return VERSION; }

	virtual bool PutIRC(const CString& sLine);
	virtual bool PutUser(const CString& sLine);
	virtual bool PutStatus(const CString& sLine);
	virtual bool PutModule(const CString& sLine, const CString& sIdent = "znc", const CString& sHost = "znc.com");
	virtual bool PutModNotice(const CString& sLine, const CString& sIdent = "znc", const CString& sHost = "znc.com");

	const CString& GetModName();
	CString GetModNick();

	// Timer stuff
	bool AddTimer(CTimer* pTimer);
	bool RemTimer(const CString& sLabel);
	bool UnlinkTimer(CTimer* pTimer);
	CTimer* FindTimer(const CString& sLabel);
	virtual void ListTimers();
	// !Timer stuff

protected:
	vector<CTimer*>			m_vTimers;
	void*					m_pDLL;
	TSocketManager<Csock>*	m_pManager;
	CUser*					m_pUser;
	CString					m_sModName;
};

class CModules : public vector<CModule*> {
public:
	CModules();
	virtual ~CModules();

	void UnloadAll();

	virtual bool OnLoad(const CString& sArgs);
	virtual bool OnBoot();
	virtual void OnUserAttached();
	virtual void OnUserDetached();
	virtual void OnIRCDisconnected();
	virtual void OnIRCConnected();

	virtual bool OnDCCUserSend(const CNick& RemoteNick, unsigned long uLongIP, unsigned short uPort, const CString& sFile, unsigned long uFileSize);

	virtual void OnOp(const CNick& OpNick, const CNick& Nick, const CChan& Channel, bool bNoChange);
	virtual void OnDeop(const CNick& OpNick, const CNick& Nick, const CChan& Channel, bool bNoChange);
	virtual void OnVoice(const CNick& OpNick, const CNick& Nick, const CChan& Channel, bool bNoChange);
	virtual void OnDevoice(const CNick& OpNick, const CNick& Nick, const CChan& Channel, bool bNoChange);
	virtual void OnRawMode(const CNick& OpNick, const CChan& Channel, const CString& sModes, const CString& sArgs);

	virtual bool OnUserRaw(CString& sLine);
	virtual bool OnRaw(CString& sLine);

	virtual bool OnStatusCommand(const CString& sCommand);
	virtual void OnModCommand(const CString& sCommand);
	virtual void OnModNotice(const CString& sMessage);
	virtual void OnModCTCP(const CString& sMessage);

	virtual void OnQuit(const CNick& Nick, const CString& sMessage, const vector<CChan*>& vChans);
	virtual void OnNick(const CNick& Nick, const CString& sNewNick, const vector<CChan*>& vChans);
	virtual void OnKick(const CNick& Nick, const CString& sOpNick, const CChan& Channel, const CString& sMessage);
	virtual void OnJoin(const CNick& Nick, const CChan& Channel);
	virtual void OnPart(const CNick& Nick, const CChan& Channel);

	virtual bool OnUserCTCPReply(const CNick& Nick, CString& sMessage);
	virtual bool OnCTCPReply(const CNick& Nick, CString& sMessage);
	virtual bool OnUserCTCP(const CString& sTarget, CString& sMessage);
	virtual bool OnPrivCTCP(const CNick& Nick, CString& sMessage);
	virtual bool OnChanCTCP(const CNick& Nick, const CChan& Channel, CString& sMessage);
	virtual bool OnUserMsg(const CString& sTarget, CString& sMessage);
	virtual bool OnPrivMsg(const CNick& Nick, CString& sMessage);
	virtual bool OnChanMsg(const CNick& Nick, const CChan& Channel, CString& sMessage);
	virtual bool OnUserNotice(const CString& sTarget, CString& sMessage);
	virtual bool OnPrivNotice(const CNick& Nick, CString& sMessage);
	virtual bool OnChanNotice(const CNick& Nick, const CChan& Channel, CString& sMessage);

	CModule* FindModule(const CString& sModule);
	bool LoadModule(const CString& sModule, const CString& sArgs, CUser* pUser, CString& sRetMsg);
	bool UnloadModule(const CString& sModule, CString& sRetMsg);
	bool ReloadModule(const CString& sModule, const CString& sArgs, CUser* pUser, CString& sRetMsg);

	static void GetAvailableMods(set<CModInfo>& ssMods, CZNC* pZNC);
};

#endif // !_MODULES_H