src/modules/m_kicknorejoin.cpp

   1 /*       +------------------------------------+
   2  *       | Inspire Internet Relay Chat Daemon |
   3  *       +------------------------------------+
   4  *
   5  *  InspIRCd: (C) 2002-2009 InspIRCd Development Team
   6  * See: http://wiki.inspircd.org/Credits
   7  *
   8  * This program is free but copyrighted software; see
   9  *            the file COPYING for details.
  10  *
  11  * ---------------------------------------------------
  12  */
  13
  14 #include "inspircd.h"
  15
  16 /* $ModDesc: Provides channel mode +J (delay rejoin after kick) */
  17
  18 inline int strtoint(const std::string &str)
  19 {
  20         std::istringstream ss(str);
  21         int result;
  22         ss >> result;
  23         return result;
  24 }
  25
  26 typedef std::map<User*, time_t> delaylist;
  27
  28 /** Handles channel mode +J
  29  */
  30 class KickRejoin : public ModeHandler
  31 {
  32  public:
  33         KickRejoin(InspIRCd* Instance, Module* Creator) : ModeHandler(Instance, Creator, 'J', 1, 0, false, MODETYPE_CHANNEL, false) { }
  34
  35         ModePair ModeSet(User* source, User* dest, Channel* channel, const std::string &parameter)
  36         {
  37                 if (channel->IsModeSet('J'))
  38                         return std::make_pair(true, channel->GetModeParameter('J'));
  39                 else
  40                         return std::make_pair(false, parameter);
  41         }
  42
  43         ModeAction OnModeChange(User* source, User* dest, Channel* channel, std::string &parameter, bool adding, bool)
  44         {
  45                 if (!adding)
  46                 {
  47                         // Taking the mode off, we need to clean up.
  48                         delaylist* dl;
  49
  50                         if (channel->GetExt("norejoinusers", dl))
  51                         {
  52                                 delete dl;
  53                                 channel->Shrink("norejoinusers");
  54                         }
  55
  56                         if (!channel->IsModeSet('J'))
  57                         {
  58                                 return MODEACTION_DENY;
  59                         }
  60                         else
  61                         {
  62                                 channel->SetModeParam('J', "");
  63                                 return MODEACTION_ALLOW;
  64                         }
  65                 }
  66                 else if (atoi(parameter.c_str()) > 0)
  67                 {
  68                         if (!channel->IsModeSet('J'))
  69                         {
  70                                 parameter = ConvToStr(atoi(parameter.c_str()));
  71                                 channel->SetModeParam('J', parameter);
  72                                 return MODEACTION_ALLOW;
  73                         }
  74                         else
  75                         {
  76                                 std::string cur_param = channel->GetModeParameter('J');
  77                                 if (cur_param == parameter)
  78                                 {
  79                                         // mode params match, don't change mode
  80                                         return MODEACTION_DENY;
  81                                 }
  82                                 else
  83                                 {
  84                                         // new mode param, replace old with new
  85                                         parameter = ConvToStr(atoi(parameter.c_str()));
  86                                         if (parameter != "0")
  87                                         {
  88                                                 channel->SetModeParam('J', parameter);
  89                                                 return MODEACTION_ALLOW;
  90                                         }
  91                                         else
  92                                         {
  93                                                 /* Fix to jamie's fix, dont allow +J 0 on the new value! */
  94                                                 return MODEACTION_DENY;
  95                                         }
  96                                 }
  97                         }
  98                 }
  99                 else
 100                 {
 101                         return MODEACTION_DENY;
 102                 }
 103         }
 104 };
 105
 106 class ModuleKickNoRejoin : public Module
 107 {
 108
 109         KickRejoin kr;
 110
 111 public:
 112
 113         ModuleKickNoRejoin(InspIRCd* Me)
 114                 : Module(Me), kr(Me, this)
 115         {
 116                 if (!ServerInstance->Modes->AddMode(&kr))
 117                         throw ModuleException("Could not add new modes!");
 118                 Implementation eventlist[] = { I_OnCleanup, I_OnChannelDelete, I_OnUserPreJoin, I_OnUserKick };
 119                 ServerInstance->Modules->Attach(eventlist, this, 4);
 120         }
 121
 122         virtual int OnUserPreJoin(User* user, Channel* chan, const char* cname, std::string &privs, const std::string &keygiven)
 123         {
 124                 if (chan)
 125                 {
 126                         delaylist* dl;
 127                         if (chan->GetExt("norejoinusers", dl))
 128                         {
 129                                 std::vector<User*> itemstoremove;
 130
 131                                 for (delaylist::iterator iter = dl->begin(); iter != dl->end(); iter++)
 132                                 {
 133                                         if (iter->second > ServerInstance->Time())
 134                                         {
 135                                                 if (iter->first == user)
 136                                                 {
 137                                                         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());
 138                                                         return 1;
 139                                                 }
 140                                         }
 141                                         else
 142                                         {
 143                                                 // Expired record, remove.
 144                                                 itemstoremove.push_back(iter->first);
 145                                         }
 146                                 }
 147
 148                                 for (unsigned int i = 0; i < itemstoremove.size(); i++)
 149                                         dl->erase(itemstoremove[i]);
 150
 151                                 if (!dl->size())
 152                                 {
 153                                         // Now it's empty..
 154                                         delete dl;
 155                                         chan->Shrink("norejoinusers");
 156                                 }
 157                         }
 158                 }
 159                 return 0;
 160         }
 161
 162         virtual void OnUserKick(User* source, User* user, Channel* chan, const std::string &reason, bool &silent)
 163         {
 164                 if (chan->IsModeSet('J') && (source != user))
 165                 {
 166                         delaylist* dl;
 167                         if (!chan->GetExt("norejoinusers", dl))
 168                         {
 169                                 dl = new delaylist;
 170                                 chan->Extend("norejoinusers", dl);
 171                         }
 172                         (*dl)[user] = ServerInstance->Time() + strtoint(chan->GetModeParameter('J'));
 173                 }
 174         }
 175
 176         virtual void OnChannelDelete(Channel* chan)
 177         {
 178                 delaylist* dl;
 179
 180                 if (chan->GetExt("norejoinusers", dl))
 181                 {
 182                         delete dl;
 183                         chan->Shrink("norejoinusers");
 184                 }
 185         }
 186
 187         virtual void OnCleanup(int target_type, void* item)
 188         {
 189                 if(target_type == TYPE_CHANNEL)
 190                         OnChannelDelete((Channel*)item);
 191         }
 192
 193
 194         virtual ~ModuleKickNoRejoin()
 195         {
 196                 ServerInstance->Modes->DelMode(&kr);
 197         }
 198
 199         virtual Version GetVersion()
 200         {
 201                 return Version("$Id$", VF_COMMON | VF_VENDOR, API_VERSION);
 202         }
 203 };
 204
 205
 206 MODULE_INIT(ModuleKickNoRejoin)