src/modules/m_kicknorejoin.cpp

   1 /*
   2  * InspIRCd -- Internet Relay Chat Daemon
   3  *
   4  *   Copyright (C) 2009 Daniel De Graaf <danieldg@inspircd.org>
   5  *   Copyright (C) 2008 Pippijn van Steenhoven <pip88nl@gmail.com>
   6  *   Copyright (C) 2007 Dennis Friis <peavey@inspircd.org>
   7  *   Copyright (C) 2006-2007 Robin Burchell <robin+git@viroteck.net>
   8  *   Copyright (C) 2006 John Brooks <john.brooks@dereferenced.net>
   9  *   Copyright (C) 2006 Craig Edwards <craigedwards@brainbox.cc>
  10  *   Copyright (C) 2006 Oliver Lupton <oliverlupton@gmail.com>
  11  *
  12  * This file is part of InspIRCd.  InspIRCd is free software: you can
  13  * redistribute it and/or modify it under the terms of the GNU General Public
  14  * License as published by the Free Software Foundation, version 2.
  15  *
  16  * This program is distributed in the hope that it will be useful, but WITHOUT
  17  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
  18  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
  19  * details.
  20  *
  21  * You should have received a copy of the GNU General Public License
  22  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  23  */
  24
  25
  26 #include "inspircd.h"
  27
  28 /* $ModDesc: Provides channel mode +J (delay rejoin after kick) */
  29
  30 typedef std::map<User*, time_t> delaylist;
  31
  32 /** Handles channel mode +J
  33  */
  34 class KickRejoin : public ParamChannelModeHandler
  35 {
  36  public:
  37         SimpleExtItem<delaylist> ext;
  38         KickRejoin(Module* Creator) : ParamChannelModeHandler(Creator, "kicknorejoin", 'J'), ext("norejoinusers", Creator) { }
  39
  40         bool ParamValidate(std::string& parameter)
  41         {
  42                 int v = atoi(parameter.c_str());
  43                 if (v <= 0)
  44                         return false;
  45                 parameter = ConvToStr(v);
  46                 return true;
  47         }
  48
  49         ModeAction OnModeChange(User* source, User* dest, Channel* channel, std::string &parameter, bool adding)
  50         {
  51                 ModeAction rv = ParamChannelModeHandler::OnModeChange(source, dest, channel, parameter, adding);
  52                 if (rv == MODEACTION_ALLOW && !adding)
  53                         ext.unset(channel);
  54                 return rv;
  55         }
  56 };
  57
  58 class ModuleKickNoRejoin : public Module
  59 {
  60         KickRejoin kr;
  61
  62 public:
  63
  64         ModuleKickNoRejoin()
  65                 : kr(this)
  66         {
  67                 if (!ServerInstance->Modes->AddMode(&kr))
  68                         throw ModuleException("Could not add new modes!");
  69                 ServerInstance->Extensions.Register(&kr.ext);
  70                 Implementation eventlist[] = { I_OnUserPreJoin, I_OnUserKick };
  71                 ServerInstance->Modules->Attach(eventlist, this, 2);
  72         }
  73
  74         ModResult OnUserPreJoin(User* user, Channel* chan, const char* cname, std::string &privs, const std::string &keygiven)
  75         {
  76                 if (chan)
  77                 {
  78                         delaylist* dl = kr.ext.get(chan);
  79                         if (dl)
  80                         {
  81                                 std::vector<User*> itemstoremove;
  82
  83                                 for (delaylist::iterator iter = dl->begin(); iter != dl->end(); iter++)
  84                                 {
  85                                         if (iter->second > ServerInstance->Time())
  86                                         {
  87                                                 if (iter->first == user)
  88                                                 {
  89                                                         user->WriteNumeric(ERR_DELAYREJOIN, "%s %s :You must wait %s seconds after being kicked to rejoin (+J)",
  90                                                                 user->nick.c_str(), chan->name.c_str(), chan->GetModeParameter(&kr).c_str());
  91                                                         return MOD_RES_DENY;
  92                                                 }
  93                                         }
  94                                         else
  95                                         {
  96                                                 // Expired record, remove.
  97                                                 itemstoremove.push_back(iter->first);
  98                                         }
  99                                 }
 100
 101                                 for (unsigned int i = 0; i < itemstoremove.size(); i++)
 102                                         dl->erase(itemstoremove[i]);
 103
 104                                 if (!dl->size())
 105                                         kr.ext.unset(chan);
 106                         }
 107                 }
 108                 return MOD_RES_PASSTHRU;
 109         }
 110
 111         void OnUserKick(User* source, Membership* memb, const std::string &reason, CUList& excepts)
 112         {
 113                 if (memb->chan->IsModeSet(&kr) && (source != memb->user))
 114                 {
 115                         delaylist* dl = kr.ext.get(memb->chan);
 116                         if (!dl)
 117                         {
 118                                 dl = new delaylist;
 119                                 kr.ext.set(memb->chan, dl);
 120                         }
 121                         (*dl)[memb->user] = ServerInstance->Time() + atoi(memb->chan->GetModeParameter(&kr).c_str());
 122                 }
 123         }
 124
 125         ~ModuleKickNoRejoin()
 126         {
 127         }
 128
 129         Version GetVersion()
 130         {
 131                 return Version("Channel mode to delay rejoin after kick", VF_VENDOR);
 132         }
 133 };
 134
 135
 136 MODULE_INIT(ModuleKickNoRejoin)