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 313d8eefac2b5e9619deee5eb18f69da83f5e0ae..11a4e9493daa73794deff82f6d067556de7a5013 100644 (file)
--- a/src/modules/m_noinvite.cpp
+++ b/src/modules/m_noinvite.cpp
@@ -2,14 +2,14 @@
  *   | 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.
  *
  * ---------------------------------------------------
  */
@@ -24,65 +24,72 @@ using namespace std;
 
 /* $ModDesc: Provides support for unreal-style channel mode +V */
 
+class NoInvite : public ModeHandler
+{
+ public:
+       NoInvite() : ModeHandler('V', 0, 0, false, MODETYPE_CHANNEL, false) { }
+
+       ModeAction OnModeChange(userrec* source, userrec* dest, chanrec* channel, std::string &parameter, bool adding)
+       {
+               if (adding)
+               {
+                       if (!channel->IsModeSet('V'))
+                       {
+                               channel->SetMode('V',true);
+                               return MODEACTION_ALLOW;
+                       }
+               }
+               else
+               {
+                       if (channel->IsModeSet('V'))
+                       {
+                               channel->SetMode('V',false);
+                               return MODEACTION_ALLOW;
+                       }
+               }
+
+               return MODEACTION_DENY;
+       }
+};
+
 class ModuleNoInvite : public Module
 {
        Server *Srv;
+       NoInvite *ni;
        
        public:
  
-               ModuleNoInvite()
+               ModuleNoInvite(Server* Me) : Module::Module(Me)
                {
-                       Srv = new Server;
-                       Srv->AddExtendedMode('V',MT_CHANNEL,false,0,0);
+                       Srv = Me;
+                       ni = new NoInvite();
+                       Srv->AddMode(ni, 'V');
                }
 
-               virtual void On005Numeric(std::string &output)
+               void Implements(char* List)
                {
-                       std::stringstream line(output);
-                       std::string temp1, temp2;
+                       List[I_On005Numeric] = List[I_OnUserPreInvite] = 1;
+               }
 
-                       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 void On005Numeric(std::string &output)
+               {
+                       InsertMode(output,"V",4);
                }
 
 
                virtual int OnUserPreInvite(userrec* user,userrec* dest,chanrec* channel)
                {
-                       if (channel->IsCustomModeSet('V'))
+                       if (channel->IsModeSet('V'))
                        {
-                               WriteServ(user->fd,"492 %s %s :Can't invite %s to channel (+V set)",user->nick, channel->name, dest->nick);
+                               user->WriteServ("492 %s %s :Can't invite %s to channel (+V set)",user->nick, channel->name, dest->nick);
                                return 1;
                        }
                        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 == 'V') && (type == MT_CHANNEL))
-                       {
-                               return 1;
-                       }
-                       else
-                       {
-                               return 0;
-                       }
-               }
 
                virtual ~ModuleNoInvite()
                {
-                       delete Srv;
+                       DELETE(ni);
                }
        
                virtual Version GetVersion()
@@ -103,9 +110,9 @@ class ModuleNoInviteFactory : public ModuleFactory
                {
                }
        
-               virtual Module * CreateModule()
+               virtual Module * CreateModule(Server* Me)
                {
-                       return new ModuleNoInvite;
+                       return new ModuleNoInvite(Me);
                }
        
 };