#ifndef _IRCSOCK_H
#define _IRCSOCK_H

#include "main.h"
#include "User.h"
#include "Chan.h"
#include "Buffer.h"

// Forward Declarations
class CZNC;
class CUserSock;
// !Forward Declarations

class CIRCSock : public Csock {
public:
	CIRCSock(CZNC* pZNC, CUser* pUser);
	virtual ~CIRCSock();

	typedef enum {
		ListArg		= 0,	// These values must line up with their position in the CHANMODE argument to raw 005
		HasArg		= 1,
		ArgWhenSet	= 2,
		NoArg		= 3
	} EChanModeArgs;

	// Message Handlers
	bool OnCTCPReply(const CString& sNickMask, CString& sMessage);
	bool OnPrivCTCP(const CString& sNickMask, CString& sMessage);
	bool OnChanCTCP(const CString& sNickMask, const CString& sChan, CString& sMessage);
	bool OnPrivMsg(const CString& sNickMask, CString& sMessage);
	bool OnChanMsg(const CString& sNickMask, const CString& sChan, CString& sMessage);
	bool OnPrivNotice(const CString& sNickMask, CString& sMessage);
	bool OnChanNotice(const CString& sNickMask, const CString& sChan, CString& sMessage);
	// !Message Handlers

	virtual void ReadLine(const CString& sData);
	virtual void Connected();
	virtual void Disconnected();
	virtual void ConnectionRefused();
	virtual void SockError(int iErrno);
	virtual void Timeout();

	void KeepNick();
	bool IsUserAttached() { return (m_pUserSock != NULL); }
	void UserConnected(CUserSock* pUserSock);
	void UserDisconnected();
	void PutServ(const CString& sLine);
	void PutUser(const CString& sLine);
	void PutStatus(const CString& sLine);
	void CIRCSock::ParseISupport(const CString& sLine);

	// Setters
	void SetPass(const CString& s) { m_sPass = s; }
	// !Setters

	// Getters

	EChanModeArgs GetModeType(unsigned char uMode) const;
	unsigned char GetPermFromMode(unsigned char uMode) const;
	const map<unsigned char, EChanModeArgs>& GetChanModes() const { return m_mueChanModes; }
	bool IsPermChar(const char c) const { return (c != '\0' && GetPerms().find(c) != CString::npos); }
	bool IsPermMode(const char c) const { return (c != '\0' && GetPermModes().find(c) != CString::npos); }
	const CString& GetPerms() const { return m_sPerms; }
	const CString& GetPermModes() const { return m_sPermModes; }
	CString GetNickMask() const { return m_Nick.GetNickMask(); }
	const CString& GetNick() const { return m_Nick.GetNick(); }
	const CString& GetPass() const { return m_sPass; }
	// !Getters
private:
	void SetNick(const CString& sNick);
protected:
	bool						m_bISpoofReleased;
	bool						m_bAuthed;
	bool						m_bKeepNick;
	CString						m_sPerms;
	CString						m_sPermModes;
	map<unsigned char, EChanModeArgs>	m_mueChanModes;
	CZNC*						m_pZNC;
	CUser*						m_pUser;
	CNick						m_Nick;
	CString						m_sPass;
	CBuffer						m_RawBuffer;
	CBuffer						m_MotdBuffer;
	CUserSock*					m_pUserSock;
	map<CString, CChan*>		m_msChans;
	unsigned int				m_uQueryBufferCount;
	CBuffer						m_QueryBuffer;
};

#endif // !_IRCSOCK_H