]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/modules/m_kicknorejoin.cpp
Untested! New m_watch that should be hundreds of times faster (im not joking either)
[user/henk/code/inspircd.git] / src / modules / m_kicknorejoin.cpp
index 1c5333defeaf06ff896701d08ff94524e724a5c2..2c485fe34bc37ab4276893e60990b7034615ff93 100644 (file)
@@ -5,11 +5,12 @@
 #include "users.h"
 #include "channels.h"
 #include "modules.h"
-#include "helperfuncs.h"
 #include "inspircd.h"
 
 /* $ModDesc: Provides channel mode +J (delay rejoin after kick) */
 
+
+
 inline int strtoint(const std::string &str)
 {
        std::istringstream ss(str);
@@ -20,19 +21,35 @@ inline int strtoint(const std::string &str)
 
 typedef std::map<userrec*, time_t> delaylist;
 
+/** Handles channel mode +J
+ */
 class KickRejoin : public ModeHandler
 {
  public:
-       KickRejoin() : ModeHandler('J', 1, 0, false, MODETYPE_CHANNEL, false) { }
+       KickRejoin(InspIRCd* Instance) : ModeHandler(Instance, 'J', 1, 0, false, MODETYPE_CHANNEL, false) { }
+
+        ModePair ModeSet(userrec* source, userrec* dest, chanrec* channel, const std::string &parameter)
+        {
+                if (channel->IsModeSet('J'))
+                        return std::make_pair(true, channel->GetModeParameter('J'));
+                else
+                        return std::make_pair(false, parameter);
+        } 
 
+        bool CheckTimeStamp(time_t theirs, time_t ours, const std::string &their_param, const std::string &our_param, chanrec* channel)
+       {
+               /* When TS is equal, the alphabetically later one wins */
+               return (their_param < our_param);
+       }
+       
        ModeAction OnModeChange(userrec* source, userrec* dest, chanrec* channel, std::string &parameter, bool adding)
        {
                if (!adding)
                {
                        // Taking the mode off, we need to clean up.
-                       delaylist* dl = (delaylist*)channel->GetExt("norejoinusers");
+                       delaylist* dl;
 
-                       if (dl)
+                       if (channel->GetExt("norejoinusers", dl))
                        {
                                DELETE(dl);
                                channel->Shrink("norejoinusers");
@@ -40,9 +57,47 @@ class KickRejoin : public ModeHandler
                }
                if ((!adding) || (atoi(parameter.c_str()) > 0))
                {
-                       channel->SetModeParam('J', parameter.c_str(), adding);
-                       channel->SetMode('J', adding);
-                       return MODEACTION_ALLOW;
+                       ServerInstance->Log(DEBUG,"Got parameter: '%s'",parameter.c_str());
+
+                       if (!channel->IsModeSet('J'))
+                       {
+                               parameter = ConvToStr(atoi(parameter.c_str()));
+                               channel->SetModeParam('J', parameter.c_str(), adding);
+                               channel->SetMode('J', adding);
+                               return MODEACTION_ALLOW;
+                       }
+                       else
+                       {
+                               if (!adding)
+                               {
+                                       channel->SetModeParam('J', parameter.c_str(), adding);
+                                       return MODEACTION_ALLOW;
+                               }
+
+                               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()));
+                                       cur_param = ConvToStr(atoi(cur_param.c_str()));
+                                       if (parameter != "0")
+                                       {
+                                               channel->SetModeParam('J', cur_param.c_str(), false);
+                                               channel->SetModeParam('J', parameter.c_str(), adding);
+                                               return MODEACTION_ALLOW;
+                                       }
+                                       else
+                                       {
+                                               /* Fix to jamie's fix, dont allow +J 0 on the new value! */
+                                               return MODEACTION_DENY;
+                                       }
+                               }
+                       }
                }
                else
                {
@@ -53,48 +108,46 @@ class KickRejoin : public ModeHandler
 
 class ModuleKickNoRejoin : public Module
 {
-       Server *Srv;
+       
        KickRejoin* kr;
        
 public:
  
-       ModuleKickNoRejoin(Server* Me)
+       ModuleKickNoRejoin(InspIRCd* Me)
                : Module::Module(Me)
        {
-               Srv = Me;
-               kr = new KickRejoin();
-               Srv->AddMode(kr, 'J');
+               
+               kr = new KickRejoin(ServerInstance);
+               ServerInstance->AddMode(kr, 'J');
        }
 
-       virtual int OnUserPreJoin(userrec* user, chanrec* chan, const char* cname)
+       virtual int OnUserPreJoin(userrec* user, chanrec* chan, const char* cname, std::string &privs)
        {
                if (chan)
                {
-                       delaylist* dl = (delaylist*)chan->GetExt("norejoinusers");
-                       log(DEBUG, "m_kicknorejoin.so: tried to grab delay list");
-                       
-                       if (dl)
+                       delaylist* dl;
+                       if (chan->GetExt("norejoinusers", dl))
                        {
-                               log(DEBUG, "m_kicknorejoin.so: got delay list, iterating over it");
+                               ServerInstance->Log(DEBUG, "m_kicknorejoin.so: got delay list, iterating over it");
                                std::vector<userrec*> itemstoremove;
                        
                                for (delaylist::iterator iter = dl->begin(); iter != dl->end(); iter++)
                                {
-                                       log(DEBUG, "m_kicknorejoin.so:\t[%s] => %d", iter->first->nick, iter->second);
+                                       ServerInstance->Log(DEBUG, "m_kicknorejoin.so:\t[%s] => %d", iter->first->nick, iter->second);
                                        if (iter->second > time(NULL))
                                        {
-                                               log(DEBUG, "m_kicknorejoin.so: still inside time slot");
+                                               ServerInstance->Log(DEBUG, "m_kicknorejoin.so: still inside time slot");
                                                if (iter->first == user)                                        
                                                {
-                                                       log(DEBUG, "m_kicknorejoin.so: and we have the right user");
-                                                       WriteServ(user->fd, "495 %s %s :You cannot rejoin this channel yet after being kicked (+J)", user->nick, chan->name);
+                                                       ServerInstance->Log(DEBUG, "m_kicknorejoin.so: and we have the right user");
+                                                       user->WriteServ( "495 %s %s :You cannot rejoin this channel yet after being kicked (+J)", user->nick, chan->name);
                                                        return 1;
                                                }
                                        }
                                        else
                                        {
                                                // Expired record, remove.
-                                               log(DEBUG, "m_kicknorejoin.so: record expired");
+                                               ServerInstance->Log(DEBUG, "m_kicknorejoin.so: record expired");
                                                itemstoremove.push_back(iter->first);
                                        }
                                }
@@ -105,38 +158,37 @@ public:
                                if (!dl->size())
                                {
                                        // Now it's empty..
-                                               DELETE(dl);
-                                               chan->Shrink("norejoinusers");
-                                       }
+                                       DELETE(dl);
+                                       chan->Shrink("norejoinusers");
                                }
                        }
-                       return 0;
                }
+               return 0;
+       }
                
-               virtual void OnUserKick(userrec* source, userrec* user, chanrec* chan, const std::string &reason)
+       virtual void OnUserKick(userrec* source, userrec* user, chanrec* chan, const std::string &reason)
+       {
+               if (chan->IsModeSet('J') && (source != user))
                {
-                       if (chan->IsModeSet('J') && (source != user))
+                       delaylist* dl;
+                       if (!chan->GetExt("norejoinusers", dl))
                        {
-                               delaylist* dl = (delaylist*)chan->GetExt("norejoinusers");
-                               
-                               if (!dl)
-                               {
-                                       dl = new delaylist;
-                                       chan->Extend("norejoinusers", (char*)dl);
-                               }
-                               
-                               log(DEBUG, "m_kicknorejoin.so: setting record for %s, %d second delay", user->nick, strtoint(chan->GetModeParameter('J')));
-                               (*dl)[user] = time(NULL) + strtoint(chan->GetModeParameter('J'));
+                               dl = new delaylist;
+                               chan->Extend("norejoinusers", dl);
                        }
+                       
+                       ServerInstance->Log(DEBUG, "m_kicknorejoin.so: setting record for %s, %d second delay", user->nick, strtoint(chan->GetModeParameter('J')));
+                       (*dl)[user] = time(NULL) + strtoint(chan->GetModeParameter('J'));
                }
-               
-               virtual void OnChannelDelete(chanrec* chan)
-               {
-                       delaylist* dl = (delaylist*)chan->GetExt("norejoinusers");
+       }
+       
+       virtual void OnChannelDelete(chanrec* chan)
+       {
+               delaylist* dl;
                        
-                       if (dl)
-                       {
-                               DELETE(dl);
+               if (chan->GetExt("norejoinusers", dl))
+               {
+                       DELETE(dl);
                        chan->Shrink("norejoinusers");
                }
        }
@@ -149,22 +201,18 @@ public:
 
        virtual void Implements(char* List)
        {
-               List[I_OnCleanup] = List[I_On005Numeric] = List[I_OnChannelDelete] = List[I_OnUserPreJoin] = List[I_OnUserKick] = 1;
-       }
-
-       virtual void On005Numeric(std::string &output)
-       {
-               InsertMode(output, "J", 3);
+               List[I_OnCleanup] = List[I_OnChannelDelete] = List[I_OnUserPreJoin] = List[I_OnUserKick] = 1;
        }
 
        virtual ~ModuleKickNoRejoin()
        {
+               ServerInstance->Modes->DelMode(kr);
                DELETE(kr);
        }
        
        virtual Version GetVersion()
        {
-               return Version(1, 0, 0, 0, VF_STATIC | VF_VENDOR);
+               return Version(1, 1, 0, 0, VF_COMMON | VF_VENDOR, API_VERSION);
        }
 };
 
@@ -180,7 +228,7 @@ class ModuleKickNoRejoinFactory : public ModuleFactory
        {
        }
        
-       virtual Module * CreateModule(Server* Me)
+       virtual Module * CreateModule(InspIRCd* Me)
        {
                return new ModuleKickNoRejoin(Me);
        }