1 files changed, 39 insertions, 71 deletions

diff --git a/src/modules/m_kicknorejoin.cpp b/src/modules/m_kicknorejoin.cpp
index a914f3869..fdb7b8f24 100644
--- a/src/modules/m_kicknorejoin.cpp
+++ b/src/modules/m_kicknorejoin.cpp
@@ -25,49 +25,44 @@
 
 #include "inspircd.h"
 
-/* $ModDesc: Provides channel mode +J (delay rejoin after kick) */
-
 typedef std::map<std::string, time_t> delaylist;
 
+struct KickRejoinData
+{
+	delaylist kicked;
+	unsigned int delay;
+
+	KickRejoinData(unsigned int Delay) : delay(Delay) { }
+};
+
 /** Handles channel mode +J
  */
-class KickRejoin : public ModeHandler
+class KickRejoin : public ParamMode<KickRejoin, SimpleExtItem<KickRejoinData> >
 {
+	static const unsigned int max = 60;
  public:
-	unsigned int max;
-	SimpleExtItem<delaylist> ext;
 	KickRejoin(Module* Creator)
-		: ModeHandler(Creator, "kicknorejoin", 'J', PARAM_SETONLY, MODETYPE_CHANNEL)
-		, ext("norejoinusers", Creator)
+		: ParamMode<KickRejoin, SimpleExtItem<KickRejoinData> >(Creator, "kicknorejoin", 'J')
 	{
 	}
 
-	ModeAction OnModeChange(User* source, User* dest, Channel* channel, std::string& parameter, bool adding)
+	ModeAction OnSet(User* source, Channel* channel, std::string& parameter)
 	{
-		if (adding)
-		{
-			int v = ConvToInt(parameter);
-			if (v <= 0)
-				return MODEACTION_DENY;
-			if (parameter == channel->GetModeParameter(this))
-				return MODEACTION_DENY;
-
-			if ((IS_LOCAL(source) && ((unsigned int)v > max)))
-				v = max;
+		int v = ConvToInt(parameter);
+		if (v <= 0)
+			return MODEACTION_DENY;
 
-			parameter = ConvToStr(v);
-			channel->SetModeParam(this, parameter);
-		}
-		else
-		{
-			if (!channel->IsModeSet(this))
-				return MODEACTION_DENY;
+		if ((IS_LOCAL(source) && ((unsigned int)v > max)))
+			v = max;
 
-			ext.unset(channel);
-			channel->SetModeParam(this, "");
-		}
+		ext.set(channel, new KickRejoinData(v));
 		return MODEACTION_ALLOW;
 	}
+
+	void SerializeParam(Channel* chan, const KickRejoinData* krd, std::string& out)
+	{
+		out.append(ConvToStr(krd->delay));
+	}
 };
 
 class ModuleKickNoRejoin : public Module
@@ -75,44 +70,27 @@ class ModuleKickNoRejoin : public Module
 	KickRejoin kr;
 
 public:
-
 	ModuleKickNoRejoin()
 		: kr(this)
 	{
 	}
 
-	void init()
-	{
-		ServerInstance->Modules->AddService(kr);
-		ServerInstance->Modules->AddService(kr.ext);
-		Implementation eventlist[] = { I_OnUserPreJoin, I_OnUserKick, I_OnRehash };
-		ServerInstance->Modules->Attach(eventlist, this, sizeof(eventlist)/sizeof(Implementation));
-		OnRehash(NULL);
-	}
-
-	void OnRehash(User* user)
-	{
-		kr.max = ServerInstance->Duration(ServerInstance->Config->ConfValue("kicknorejoin")->getString("maxtime"));
-		if (!kr.max)
-			kr.max = 30*60;
-	}
-
-	ModResult OnUserPreJoin(User* user, Channel* chan, const char* cname, std::string &privs, const std::string &keygiven)
+	ModResult OnUserPreJoin(LocalUser* user, Channel* chan, const std::string& cname, std::string& privs, const std::string& keygiven) CXX11_OVERRIDE
 	{
 		if (chan)
 		{
-			delaylist* dl = kr.ext.get(chan);
-			if (dl)
+			KickRejoinData* data = kr.ext.get(chan);
+			if (data)
 			{
-				for (delaylist::iterator iter = dl->begin(); iter != dl->end(); )
+				delaylist& kicked = data->kicked;
+				for (delaylist::iterator iter = kicked.begin(); iter != kicked.end(); )
 				{
 					if (iter->second > ServerInstance->Time())
 					{
 						if (iter->first == user->uuid)
 						{
-							std::string modeparam = chan->GetModeParameter(&kr);
-							user->WriteNumeric(ERR_DELAYREJOIN, "%s %s :You must wait %s seconds after being kicked to rejoin (+J)",
-								user->nick.c_str(), chan->name.c_str(), modeparam.c_str());
+							user->WriteNumeric(ERR_DELAYREJOIN, "%s :You must wait %u seconds after being kicked to rejoin (+J)",
+								chan->name.c_str(), data->delay);
 							return MOD_RES_DENY;
 						}
 						++iter;
@@ -120,40 +98,30 @@ public:
 					else
 					{
 						// Expired record, remove.
-						dl->erase(iter++);
+						kicked.erase(iter++);
 					}
 				}
-
-				if (dl->empty())
-					kr.ext.unset(chan);
 			}
 		}
 		return MOD_RES_PASSTHRU;
 	}
 
-	void OnUserKick(User* source, Membership* memb, const std::string &reason, CUList& excepts)
+	void OnUserKick(User* source, Membership* memb, const std::string &reason, CUList& excepts) CXX11_OVERRIDE
 	{
-		if (memb->chan->IsModeSet(&kr) && (IS_LOCAL(memb->user)) && (source != memb->user))
+		if ((!IS_LOCAL(memb->user)) || (source == memb->user))
+			return;
+
+		KickRejoinData* data = kr.ext.get(memb->chan);
+		if (data)
 		{
-			delaylist* dl = kr.ext.get(memb->chan);
-			if (!dl)
-			{
-				dl = new delaylist;
-				kr.ext.set(memb->chan, dl);
-			}
-			(*dl)[memb->user->uuid] = ServerInstance->Time() + ConvToInt(memb->chan->GetModeParameter(&kr));
+			data->kicked[memb->user->uuid] = ServerInstance->Time() + data->delay;
 		}
 	}
 
-	~ModuleKickNoRejoin()
-	{
-	}
-
-	Version GetVersion()
+	Version GetVersion() CXX11_OVERRIDE
 	{
 		return Version("Channel mode to delay rejoin after kick", VF_VENDOR);
 	}
 };
 
-
 MODULE_INIT(ModuleKickNoRejoin)