X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=src%2Fmodules%2Fm_kicknorejoin.cpp;h=34df89ae3f398103317ee312f87ab55b2fad926e;hb=8f915e5ddbab4e36bb08f9f9d726e953db1f601f;hp=5a8233a8f76cad8bab3355c93119ce1f1fbf49eb;hpb=c4d6ce8c5e1802e5b834d5845e31c3adf7700585;p=user%2Fhenk%2Fcode%2Finspircd.git

diff --git a/src/modules/m_kicknorejoin.cpp b/src/modules/m_kicknorejoin.cpp
index 5a8233a8f..34df89ae3 100644
--- a/src/modules/m_kicknorejoin.cpp
+++ b/src/modules/m_kicknorejoin.cpp
@@ -2,7 +2,7 @@
  *       | Inspire Internet Relay Chat Daemon |
  *       +------------------------------------+
  *
- *  InspIRCd: (C) 2002-2009 InspIRCd Development Team
+ *  InspIRCd: (C) 2002-2010 InspIRCd Development Team
  * See: http://wiki.inspircd.org/Credits
  *
  * This program is free but copyrighted software; see
@@ -15,86 +15,31 @@
 
 /* $ModDesc: Provides channel mode +J (delay rejoin after kick) */
 
-static inline int strtoint(const std::string &str)
-{
-	std::istringstream ss(str);
-	int result;
-	ss >> result;
-	return result;
-}
-
 typedef std::map<User*, time_t> delaylist;
 
 /** Handles channel mode +J
  */
-class KickRejoin : public ModeHandler
+class KickRejoin : public ParamChannelModeHandler
 {
  public:
 	SimpleExtItem<delaylist> ext;
-	KickRejoin(Module* Creator) : ModeHandler(Creator, "kicknorejoin", 'J', PARAM_SETONLY, MODETYPE_CHANNEL),
-		ext("norejoinusers", Creator) { }
+	KickRejoin(Module* Creator) : ParamChannelModeHandler(Creator, "kicknorejoin", 'J'), ext("norejoinusers", Creator) { }
 
-	ModePair ModeSet(User* source, User* dest, Channel* channel, const std::string &parameter)
+	bool ParamValidate(std::string& parameter)
 	{
-		if (channel->IsModeSet('J'))
-			return std::make_pair(true, channel->GetModeParameter('J'));
-		else
-			return std::make_pair(false, parameter);
+		int v = atoi(parameter.c_str());
+		if (v <= 0)
+			return false;
+		parameter = ConvToStr(v);
+		return true;
 	}
 
 	ModeAction OnModeChange(User* source, User* dest, Channel* channel, std::string &parameter, bool adding)
 	{
-		if (!adding)
-		{
+		ModeAction rv = ParamChannelModeHandler::OnModeChange(source, dest, channel, parameter, adding);
+		if (rv == MODEACTION_ALLOW && !adding)
 			ext.unset(channel);
-
-			if (!channel->IsModeSet('J'))
-			{
-				return MODEACTION_DENY;
-			}
-			else
-			{
-				channel->SetModeParam('J', "");
-				return MODEACTION_ALLOW;
-			}
-		}
-		else if (atoi(parameter.c_str()) > 0)
-		{
-			if (!channel->IsModeSet('J'))
-			{
-				parameter = ConvToStr(atoi(parameter.c_str()));
-				channel->SetModeParam('J', parameter);
-				return MODEACTION_ALLOW;
-			}
-			else
-			{
-				std::string cur_param = channel->GetModeParameter('J');
-				if (cur_param == parameter)
-				{
-					// mode params match, don't change mode
-					return MODEACTION_DENY;
-				}
-				else
-				{
-					// new mode param, replace old with new
-					parameter = ConvToStr(atoi(parameter.c_str()));
-					if (parameter != "0")
-					{
-						channel->SetModeParam('J', parameter);
-						return MODEACTION_ALLOW;
-					}
-					else
-					{
-						/* Fix to jamie's fix, dont allow +J 0 on the new value! */
-						return MODEACTION_DENY;
-					}
-				}
-			}
-		}
-		else
-		{
-			return MODEACTION_DENY;
-		}
+		return rv;
 	}
 };
 
@@ -109,7 +54,7 @@ public:
 	{
 		if (!ServerInstance->Modes->AddMode(&kr))
 			throw ModuleException("Could not add new modes!");
-		Extensible::Register(&kr.ext);
+		ServerInstance->Extensions.Register(&kr.ext);
 		Implementation eventlist[] = { I_OnUserPreJoin, I_OnUserKick };
 		ServerInstance->Modules->Attach(eventlist, this, 2);
 	}
@@ -129,7 +74,8 @@ public:
 					{
 						if (iter->first == user)
 						{
-							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(), chan->GetModeParameter('J').c_str());
+							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(), chan->GetModeParameter(&kr).c_str());
 							return MOD_RES_DENY;
 						}
 					}
@@ -152,15 +98,15 @@ public:
 
 	void OnUserKick(User* source, Membership* memb, const std::string &reason, CUList& excepts)
 	{
-		if (memb->chan->IsModeSet('J') && (source != memb->user))
+		if (memb->chan->IsModeSet(&kr) && (source != memb->user))
 		{
 			delaylist* dl = kr.ext.get(memb->chan);
-			if (dl)
+			if (!dl)
 			{
 				dl = new delaylist;
 				kr.ext.set(memb->chan, dl);
 			}
-			(*dl)[memb->user] = ServerInstance->Time() + strtoint(memb->chan->GetModeParameter('J'));
+			(*dl)[memb->user] = ServerInstance->Time() + atoi(memb->chan->GetModeParameter(&kr).c_str());
 		}
 	}
 
@@ -170,7 +116,7 @@ public:
 
 	Version GetVersion()
 	{
-		return Version("Channel mode J, kick-no-rejoin", VF_COMMON | VF_VENDOR);
+		return Version("Channel mode to delay rejoin after kick", VF_VENDOR);
 	}
 };