src/modules/m_kicknorejoin.cpp

   1 #include <time.h>
   2 #include <map>
   3 #include <vector>
   4 #include <sstream>
   5 #include "users.h"
   6 #include "channels.h"
   7 #include "modules.h"
   8 #include "helperfuncs.h"
   9
  10 /* $ModDesc: Provides channel mode +J (delay rejoin after kick) */
  11
  12 inline int strtoint(const std::string &str)
  13 {
  14         std::istringstream ss(str);
  15         int result;
  16         ss >> result;
  17         return result;
  18 }
  19
  20 typedef std::map<userrec*, time_t> delaylist;
  21
  22 class ModuleKickNoRejoin : public Module
  23 {
  24         Server *Srv;
  25
  26 public:
  27
  28         ModuleKickNoRejoin(Server* Me)
  29                 : Module::Module(Me)
  30         {
  31                 Srv = Me;
  32
  33                 Srv->AddExtendedMode('J', MT_CHANNEL, false, 1, 0);
  34         }
  35
  36         virtual int OnExtendedMode(userrec* user, void* target, char modechar, int type, bool mode_on, string_list &params)
  37         {
  38                 if ((modechar == 'J') && (type == MT_CHANNEL))
  39                 {
  40                         if(!mode_on)
  41                         {
  42                                 // Taking the mode off, we need to clean up.
  43                                 chanrec* c = (chanrec*)target;
  44
  45                                 delaylist* dl = (delaylist*)c->GetExt("norejoinusers");
  46
  47                                 if(dl)
  48                                 {
  49                                         delete dl;
  50                                         c->Shrink("norejoinusers");
  51                                 }
  52                         }
  53                         return 1;
  54                 }
  55                 return 0;
  56         }
  57
  58         virtual int OnUserPreJoin(userrec* user, chanrec* chan, const char* cname)
  59         {
  60                 if(chan)
  61                 {
  62                         delaylist* dl = (delaylist*)chan->GetExt("norejoinusers");
  63                         log(DEBUG, "m_kicknorejoin.so: tried to grab delay list");
  64
  65                         if(dl)
  66                         {
  67                                 log(DEBUG, "m_kicknorejoin.so: got delay list, iterating over it");
  68                                 std::vector<userrec*> itemstoremove;
  69
  70                                 for(delaylist::iterator iter = dl->begin(); iter != dl->end(); iter++)
  71                                 {
  72                                         log(DEBUG, "m_kicknorejoin.so:\t[%s] => %d", iter->first->nick, iter->second);
  73                                         if(iter->second > time(NULL))
  74                                         {
  75                                                 log(DEBUG, "m_kicknorejoin.so: still inside time slot");
  76                                                 if(iter->first == user)
  77                                                 {
  78                                                         log(DEBUG, "m_kicknorejoin.so: and we have the right user");
  79                                                         WriteServ(user->fd, "495 %s %s :You cannot rejoin this channel yet after being kicked (+J)", user->nick, chan->name);
  80                                                         return 1;
  81                                                 }
  82                                         }
  83                                         else
  84                                         {
  85                                                 // Expired record, remove.
  86                                                 log(DEBUG, "m_kicknorejoin.so: record expired");
  87                                                 itemstoremove.push_back(iter->first);
  88                                         }
  89                                 }
  90
  91                                 for(unsigned int i = 0; i < itemstoremove.size(); i++)
  92                                         dl->erase(itemstoremove[i]);
  93
  94                                 if(!dl->size())
  95                                 {
  96                                         // Now it's empty..
  97                                         delete dl;
  98                                         chan->Shrink("norejoinusers");
  99                                 }
 100                         }
 101                 }
 102                 return 0;
 103         }
 104
 105         virtual void OnUserKick(userrec* source, userrec* user, chanrec* chan, std::string reason)
 106         {
 107                 if(chan->IsCustomModeSet('J'))
 108                 {
 109                         delaylist* dl = (delaylist*)chan->GetExt("norejoinusers");
 110
 111                         if(!dl)
 112                         {
 113                                 dl = new delaylist;
 114                                 chan->Extend("norejoinusers", (char*)dl);
 115                         }
 116
 117                         log(DEBUG, "m_kicknorejoin.so: setting record for %s, %d second delay", user->nick, strtoint(chan->GetModeParameter('J')));
 118                         (*dl)[user] = time(NULL) + strtoint(chan->GetModeParameter('J'));
 119                 }
 120         }
 121
 122         virtual void OnChannelDelete(chanrec* chan)
 123         {
 124                 delaylist* dl = (delaylist*)chan->GetExt("norejoinusers");
 125
 126                 if(dl)
 127                 {
 128                         delete dl;
 129                         chan->Shrink("norejoinusers");
 130                 }
 131         }
 132
 133         virtual void OnCleanup(int target_type, void* item)
 134         {
 135                 if(target_type == TYPE_CHANNEL)
 136                         OnChannelDelete((chanrec*)item);
 137         }
 138
 139         virtual void Implements(char* List)
 140         {
 141                 List[I_OnCleanup] = List[I_On005Numeric] = List[I_OnExtendedMode] = List[I_OnChannelDelete] = List[I_OnUserPreJoin] = List[I_OnUserKick] = 1;
 142         }
 143
 144         virtual void On005Numeric(std::string &output)
 145         {
 146                 InsertMode(output, "J", 3);
 147         }
 148
 149         virtual ~ModuleKickNoRejoin()
 150         {
 151         }
 152
 153         virtual Version GetVersion()
 154         {
 155                 return Version(1, 0, 0, 0, VF_STATIC | VF_VENDOR);
 156         }
 157 };
 158
 159
 160 class ModuleKickNoRejoinFactory : public ModuleFactory
 161 {
 162  public:
 163         ModuleKickNoRejoinFactory()
 164         {
 165         }
 166
 167         ~ModuleKickNoRejoinFactory()
 168         {
 169         }
 170
 171         virtual Module * CreateModule(Server* Me)
 172         {
 173                 return new ModuleKickNoRejoin(Me);
 174         }
 175
 176 };
 177
 178
 179 extern "C" void * init_module( void )
 180 {
 181         return new ModuleKickNoRejoinFactory;
 182 }
 183