WHEEEEE!!!!!

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

diff --git a/src/modules/m_noinvite.cpp b/src/modules/m_noinvite.cpp
index 30f313a92d27ca5070bddaab95a7dedd211a1291..11a4e9493daa73794deff82f6d067556de7a5013 100644 (file)
--- a/src/modules/m_noinvite.cpp
+++ b/src/modules/m_noinvite.cpp
@@ -1,108 +1,119 @@
-/*       +------------------------------------+
- *       | Inspire Internet Relay Chat Daemon |
- *       +------------------------------------+
+/*   +------------------------------------+
+ *   | Inspire Internet Relay Chat Daemon |
+ *   +------------------------------------+
  *
- *  Inspire is copyright (C) 2002-2004 ChatSpike-Dev.
- *                       E-mail:
- *                <brain@chatspike.net>
- *               <Craig@chatspike.net>
- *     
+ *  InspIRCd is copyright (C) 2002-2006 ChatSpike-Dev.
+ *                   E-mail:
+ *            <brain@chatspike.net>
+ *            <Craig@chatspike.net>
+ * 
  * Written by Craig Edwards, Craig McLure, and others.
  * This program is free but copyrighted software; see
- *            the file COPYING for details.
+ * the file COPYING for details.
  *
  * ---------------------------------------------------
  */
 
+using namespace std;
+
 #include <stdio.h>
 #include "users.h"
 #include "channels.h"
 #include "modules.h"
+#include "helperfuncs.h"
 
 /* $ModDesc: Provides support for unreal-style channel mode +V */
 
-class ModuleNoInvite : public Module
+class NoInvite : public ModeHandler
 {
-       Server *Srv;
-       
  public:
- 
-       ModuleNoInvite()
-       {
-               Srv = new Server;
-               Srv->AddExtendedMode('V',MT_CHANNEL,false,0,0);
-       }
+       NoInvite() : ModeHandler('V', 0, 0, false, MODETYPE_CHANNEL, false) { }
 
-        virtual void On005Numeric(std::string &output)
-        {
-                std::stringstream line(output);
-                std::string temp1, temp2;
-                while (!line.eof())
-                {
-                        line >> temp1;
-                        if (temp1.substr(0,10) == "CHANMODES=")
-                        {
-                                // append the chanmode to the end
-                                temp1 = temp1.substr(10,temp1.length());
-                                temp1 = "CHANMODES=" + temp1 + "V";
-                        }
-                        temp2 = temp2 + temp1 + " ";
-                }
-               if (temp2.length())
-                       output = temp2.substr(0,temp2.length()-1);
-        }
-
-
-       virtual int OnUserPreInvite(userrec* user,userrec* dest,chanrec* channel)
+       ModeAction OnModeChange(userrec* source, userrec* dest, chanrec* channel, std::string &parameter, bool adding)
        {
-               if (channel->IsCustomModeSet('V'))
+               if (adding)
                {
-                       WriteServ(user->fd,"492 %s %s :Can't invite %s to channel (+V set)",user->nick, channel->name, dest->nick);
-                       return 1;
+                       if (!channel->IsModeSet('V'))
+                       {
+                               channel->SetMode('V',true);
+                               return MODEACTION_ALLOW;
+                       }
                }
-               return 0;
+               else
+               {
+                       if (channel->IsModeSet('V'))
+                       {
+                               channel->SetMode('V',false);
+                               return MODEACTION_ALLOW;
+                       }
+               }
+
+               return MODEACTION_DENY;
        }
+};
+
+class ModuleNoInvite : public Module
+{
+       Server *Srv;
+       NoInvite *ni;
        
-       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 == 'V') && (type == MT_CHANNEL))
-               {
-                       return 1;
+       public:
+ 
+               ModuleNoInvite(Server* Me) : Module::Module(Me)
+               {
+                       Srv = Me;
+                       ni = new NoInvite();
+                       Srv->AddMode(ni, 'V');
                }
-               else
+
+               void Implements(char* List)
+               {
+                       List[I_On005Numeric] = List[I_OnUserPreInvite] = 1;
+               }
+
+               virtual void On005Numeric(std::string &output)
+               {
+                       InsertMode(output,"V",4);
+               }
+
+
+               virtual int OnUserPreInvite(userrec* user,userrec* dest,chanrec* channel)
                {
+                       if (channel->IsModeSet('V'))
+                       {
+                               user->WriteServ("492 %s %s :Can't invite %s to channel (+V set)",user->nick, channel->name, dest->nick);
+                               return 1;
+                       }
                        return 0;
                }
-       }
 
-       virtual ~ModuleNoInvite()
-       {
-               delete Srv;
-       }
+               virtual ~ModuleNoInvite()
+               {
+                       DELETE(ni);
+               }
        
-       virtual Version GetVersion()
-       {
-               return Version(1,0,0,0,VF_STATIC|VF_VENDOR);
-       }
+               virtual Version GetVersion()
+               {
+                       return Version(1,0,0,0,VF_STATIC|VF_VENDOR);
+               }
 };
 
 
 class ModuleNoInviteFactory : public ModuleFactory
 {
- public:
-       ModuleNoInviteFactory()
-       {
-       }
+       public:
+               ModuleNoInviteFactory()
+               {
+               }
        
-       ~ModuleNoInviteFactory()
-       {
-       }
+               ~ModuleNoInviteFactory()
+               {
+               }
        
-       virtual Module * CreateModule()
-       {
-               return new ModuleNoInvite;
-       }
+               virtual Module * CreateModule(Server* Me)
+               {
+                       return new ModuleNoInvite(Me);
+               }
        
 };