Implement feature in bug #395 reported by stealth, and tidy up a bit

[user/henk/code/inspircd.git] / src / modules / m_nonicks.cpp

diff --git a/src/modules/m_nonicks.cpp b/src/modules/m_nonicks.cpp
index d465251525d7f5d02efca9b59509e8e2b92fc05d..bb1843a958e20b6eae05c41f65b387ee7e760e8a 100644 (file)
--- a/src/modules/m_nonicks.cpp
+++ b/src/modules/m_nonicks.cpp
@@ -1,92 +1,102 @@
-#include <stdio.h>
-#include <string>
+/*       +------------------------------------+
+ *       | Inspire Internet Relay Chat Daemon |
+ *       +------------------------------------+
+ *
+ *  InspIRCd: (C) 2002-2007 InspIRCd Development Team
+ * See: http://www.inspircd.org/wiki/index.php/Credits
+ *
+ * This program is free but copyrighted software; see
+ *            the file COPYING for details.
+ *
+ * ---------------------------------------------------
+ */
+
+#include "inspircd.h"
 #include "users.h"
 #include "channels.h"
 #include "modules.h"
+#include "hashcomp.h"
+#include "configreader.h"
 
-/* $ModDesc: Provides support for unreal-style GLOBOPS and umode +g */
-
-Server *Srv;
+/* $ModDesc: Provides support for channel mode +N which prevents nick changes on channel */
 
-class ModuleNoNickChange : public Module
+class NoNicks : public ModeHandler
 {
  public:
-       ModuleNoNickChange()
-       {
-               Srv = new Server;
-               
-               Srv->AddExtendedMode('N',MT_CHANNEL,false,0,0);
-       }
-       
-       virtual ~ModuleNoNickChange()
-       {
-               delete Srv;
-       }
-       
-       virtual Version GetVersion()
-       {
-               return Version(1,0,0,1);
-       }
-       
-       virtual int OnUserPreNick(userrec* user, std::string newnick)
+       NoNicks(InspIRCd* Instance) : ModeHandler(Instance, 'N', 0, 0, false, MODETYPE_CHANNEL, false) { }
+
+       ModeAction OnModeChange(userrec* source, userrec* dest, chanrec* channel, std::string &parameter, bool adding)
        {
-               if (!strcasecmp(user->server,Srv->GetServerName().c_str()))
+               if (adding)
                {
-                       for (int i =0; i != MAXCHANS; i++)
+                       if (!channel->IsModeSet('N'))
                        {
-                               if (user->chans[i].channel != NULL)
-                               {
-                                       chanrec* curr = user->chans[i].channel;
-                                       if (curr->IsCustomModeSet('N'))
-                                       {
-                                               // don't allow the nickchange, theyre on at least one channel with +N set
-                                               WriteServ(user->fd,"447 %s :Can't change nickname while on %s (+N is set)",user->nick,curr->name);
-                                               return 1;
-                                       }
-                               }
+                               channel->SetMode('N',true);
+                               return MODEACTION_ALLOW;
                        }
                }
-               return 0;
-       }
-       
-       virtual int OnExtendedMode(userrec* user, void* target, char modechar, int type, bool mode_on, string_list &params)
-       {
-               // check if this is our mode character...
-               if ((modechar == 'N') && (type == MT_CHANNEL))
-               {
-                       return 1;
-               }
                else
                {
-                       return 0;
+                       if (channel->IsModeSet('N'))
+                       {
+                               channel->SetMode('N',false);
+                               return MODEACTION_ALLOW;
+                       }
                }
-       }
 
+               return MODEACTION_DENY;
+       }
 };
 
-// stuff down here is the module-factory stuff. For basic modules you can ignore this.
-
-class ModuleNoNickChangeFactory : public ModuleFactory
+class ModuleNoNickChange : public Module
 {
+       NoNicks* nn;
  public:
-       ModuleNoNickChangeFactory()
+       ModuleNoNickChange(InspIRCd* Me)
+               : Module(Me)
        {
+               
+               nn = new NoNicks(ServerInstance);
+               ServerInstance->AddMode(nn, 'N');
        }
        
-       ~ModuleNoNickChangeFactory()
+       virtual ~ModuleNoNickChange()
        {
+               ServerInstance->Modes->DelMode(nn);
+               DELETE(nn);
        }
        
-       virtual Module * CreateModule()
+       virtual Version GetVersion()
        {
-               return new ModuleNoNickChange;
+               return Version(1,1,0,1,VF_COMMON|VF_VENDOR,API_VERSION);
        }
-       
-};
 
+       void Implements(char* List)
+       {
+               List[I_OnUserPreNick] = 1;
+       }
 
-extern "C" void * init_module( void )
-{
-       return new ModuleNoNickChangeFactory;
-}
+       virtual int OnUserPreNick(userrec* user, const std::string &newnick)
+       {
+               if (IS_LOCAL(user))
+               {
+                       for (UCListIter i = user->chans.begin(); i != user->chans.end(); i++)
+                       {
+                               chanrec* curr = i->first;
+
+                               if (curr->IsModeSet('N'))
+                               {
+                                       if (CHANOPS_EXEMPT(ServerInstance, 'N') && curr->GetStatus(user) == STATUS_OP)
+                                               continue;
+
+                                       user->WriteServ("447 %s :Can't change nickname while on %s (+N is set)", user->nick, curr->name);
+                                       return 1;
+                               }
+                       }
+               }
+
+               return 0;
+       }
+};
 
+MODULE_INIT(ModuleNoNickChange)