From 23acbe42a6c244c3ce132f55ae06d01a959ccc5b Mon Sep 17 00:00:00 2001
From: Toon Schoenmakers <nighteyes1993@gmail.com>
Date: Sun, 25 Sep 2011 01:58:23 +0200
Subject: [PATCH] Implemented a OnInvite hook

This hook will be called when the user get's invited into a channel
because we don't have a CChan of this channel yet this won't be send
through the hook. Instead a CString with the name of the channel is send.

Signed-off-by: Toon Schoenmakers <nighteyes1993@gmail.com>
---
 IRCSock.cpp | 2 ++
 Modules.cpp | 2 ++
 Modules.h   | 5 +++++
 3 files changed, 9 insertions(+)

diff --git a/IRCSock.cpp b/IRCSock.cpp
index ff40d730..7867f6d6 100644
--- a/IRCSock.cpp
+++ b/IRCSock.cpp
@@ -711,6 +711,8 @@ void CIRCSock::ReadLine(const CString& sData) {
 			}
 			// Don't forward any CAP stuff to the client
 			return;
+		} else if (sCmd.Equals("INVITE")) {
+			NETWORKMODULECALL(OnInvite(sLine.Token(3).TrimPrefix_n(":")), m_pNetwork->GetUser(), m_pNetwork, NULL, NOTHING);
 		}
 	}
 
diff --git a/Modules.cpp b/Modules.cpp
index 7d1a9fec..33600418 100644
--- a/Modules.cpp
+++ b/Modules.cpp
@@ -513,6 +513,7 @@ void CModule::OnNick(const CNick& Nick, const CString& sNewNick, const vector<CC
 void CModule::OnKick(const CNick& Nick, const CString& sKickedNick, CChan& Channel, const CString& sMessage) {}
 void CModule::OnJoin(const CNick& Nick, CChan& Channel) {}
 void CModule::OnPart(const CNick& Nick, CChan& Channel, const CString& sMessage) {}
+void CModule::OnInvite(const CString& sChan) {}
 
 CModule::EModRet CModule::OnChanBufferStarting(CChan& Chan, CClient& Client) { return CONTINUE; }
 CModule::EModRet CModule::OnChanBufferEnding(CChan& Chan, CClient& Client) { return CONTINUE; }
@@ -685,6 +686,7 @@ bool CModules::OnNick(const CNick& Nick, const CString& sNewNick, const vector<C
 bool CModules::OnKick(const CNick& Nick, const CString& sKickedNick, CChan& Channel, const CString& sMessage) { MODUNLOADCHK(OnKick(Nick, sKickedNick, Channel, sMessage)); return false; }
 bool CModules::OnJoin(const CNick& Nick, CChan& Channel) { MODUNLOADCHK(OnJoin(Nick, Channel)); return false; }
 bool CModules::OnPart(const CNick& Nick, CChan& Channel, const CString& sMessage) { MODUNLOADCHK(OnPart(Nick, Channel, sMessage)); return false; }
+bool CModules::OnInvite(const CString& sChan) { MODUNLOADCHK(OnInvite(sChan)); return false; }
 bool CModules::OnChanBufferStarting(CChan& Chan, CClient& Client) { MODHALTCHK(OnChanBufferStarting(Chan, Client)); }
 bool CModules::OnChanBufferEnding(CChan& Chan, CClient& Client) { MODHALTCHK(OnChanBufferEnding(Chan, Client)); }
 bool CModules::OnChanBufferPlayLine(CChan& Chan, CClient& Client, CString& sLine) { MODHALTCHK(OnChanBufferPlayLine(Chan, Client, sLine)); }
diff --git a/Modules.h b/Modules.h
index 0e3e142e..ede88ff2 100644
--- a/Modules.h
+++ b/Modules.h
@@ -552,6 +552,10 @@ public:
 	 *  @param sMessage The part message.
 	 */
 	virtual void OnPart(const CNick& Nick, CChan& Channel, const CString& sMessage);
+	/** Called when user is invited into a channel
+	 *  @param sChan The channel the user got invited into
+	 */
+	virtual void OnInvite(const CString& sChan);
 
 	/** Called before a channel buffer is played back to a client.
 	 *  @param Chan The channel which will be played back.
@@ -1046,6 +1050,7 @@ public:
 	bool OnKick(const CNick& Nick, const CString& sOpNick, CChan& Channel, const CString& sMessage);
 	bool OnJoin(const CNick& Nick, CChan& Channel);
 	bool OnPart(const CNick& Nick, CChan& Channel, const CString& sMessage);
+	bool OnInvite(const CString& sChan);
 
 	bool OnChanBufferStarting(CChan& Chan, CClient& Client);
 	bool OnChanBufferEnding(CChan& Chan, CClient& Client);