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