Hold reference to the associated ConfigTag inside ConnectClass

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

diff --git a/src/modules/m_conn_umodes.cpp b/src/modules/m_conn_umodes.cpp
index 8c962ae5715a8d46e382077e8f5369a26b50c2d2..0fcd73fe92493c2fd17b60d13f09c3d82e114d1f 100644 (file)
--- a/src/modules/m_conn_umodes.cpp
+++ b/src/modules/m_conn_umodes.cpp
@@ -2,136 +2,81 @@
  *       | Inspire Internet Relay Chat Daemon |
  *       +------------------------------------+
  *
- *  InspIRCd is copyright (C) 2002-2006 ChatSpike-Dev.
- *                       E-mail:
- *                <brain@chatspike.net>
- *               <Craig@chatspike.net>
- *     
- * Written by Craig Edwards, Craig McLure, and others.
+ *  InspIRCd: (C) 2002-2009 InspIRCd Development Team
+ * See: http://wiki.inspircd.org/Credits
+ *
  * This program is free but copyrighted software; see
  *            the file COPYING for details.
  *
  * ---------------------------------------------------
  */
 
-using namespace std;
-
-#include <stdio.h>
-#include <vector>
-#include "users.h"
-#include "channels.h"
 #include "inspircd.h"
-#include "modules.h"
-#include "wildcard.h"
 
 /* $ModDesc: Sets (and unsets) modes on users when they connect */
 
 class ModuleModesOnConnect : public Module
 {
- private:
-
-       ConfigReader *Conf;
-
  public:
-       ModuleModesOnConnect(InspIRCd* Me)
-               : Module::Module(Me)
-       {
-               
-               Conf = new ConfigReader(ServerInstance);
+       ModuleModesOnConnect()  {
+               Implementation eventlist[] = { I_OnUserConnect, I_OnRehash };
+               ServerInstance->Modules->Attach(eventlist, this, 2);
+               // for things like +x on connect, important, otherwise we have to resort to config order (bleh) -- w00t
+               ServerInstance->Modules->SetPriority(this, PRIORITY_FIRST);
        }
 
-       void Implements(char* List)
-       {
-               List[I_OnPostConnect] = List[I_OnRehash] = 1;
-       }
 
-       virtual void OnRehash(const std::string &parameter)
-       {
-               DELETE(Conf);
-               Conf = new ConfigReader(ServerInstance);
-       }
-       
        virtual ~ModuleModesOnConnect()
        {
-               DELETE(Conf);
        }
-       
+
        virtual Version GetVersion()
        {
-               return Version(1,0,0,1,VF_VENDOR);
+               return Version("Sets (and unsets) modes on users when they connect", VF_VENDOR,API_VERSION);
        }
-       
-       virtual void OnPostConnect(userrec* user)
+
+       virtual void OnUserConnect(User* user)
        {
                if (!IS_LOCAL(user))
                        return;
 
-               ServerInstance->Log(DEBUG,"Post connect for mode setting");
-               for (int j = 0; j < Conf->Enumerate("connect"); j++)
+               // Backup and zero out the disabled usermodes, so that we can override them here.
+               char save[64];
+               memcpy(save, ServerInstance->Config->DisabledUModes,
+                               sizeof(ServerInstance->Config->DisabledUModes));
+               memset(ServerInstance->Config->DisabledUModes, 0, 64);
+
+               ConfigTag* tag = user->MyClass->config;
+               std::string ThisModes = tag->getString("modes");
+               if (!ThisModes.empty())
                {
-                       std::string hostn = Conf->ReadValue("connect","allow",j);
-                       if ((match(user->GetIPString(),hostn.c_str(),true)) || (match(user->host,hostn.c_str())))
+                       std::string buf;
+                       std::stringstream ss(ThisModes);
+
+                       std::vector<std::string> tokens;
+
+                       // split ThisUserModes into modes and mode params
+                       while (ss >> buf)
+                               tokens.push_back(buf);
+
+                       std::vector<std::string> modes;
+                       modes.push_back(user->nick);
+                       modes.push_back(tokens[0]);
+
+                       if (tokens.size() > 1)
                        {
-                               ServerInstance->Log(DEBUG,"Found matching connect block '%s'",hostn.c_str());
-                               std::string ThisModes = Conf->ReadValue("connect","modes",j);
-                               if (ThisModes != "")
+                               // process mode params
+                               for (unsigned int k = 1; k < tokens.size(); k++)
                                {
-                                       std::string buf;
-                                       stringstream ss(ThisModes);
-
-                                       vector<string> tokens;
-
-                                       // split ThisOperModes into modes and mode params
-                                       while (ss >> buf)
-                                               tokens.push_back(buf);
-
-                                       int size = tokens.size() + 1;
-                                       const char* modes[size];
-                                       modes[0] = user->nick;
-                                       modes[1] = tokens[0].c_str();
-
-                                       if (tokens.size() > 1)
-                                       {
-                                               // process mode params
-                                               int i = 2;
-                                               for (unsigned int k = 1; k < tokens.size(); k++)
-                                               {
-                                                       modes[i] = tokens[k].c_str();
-                                                       i++;
-                                               }
-                                       }
-
-                                       ServerInstance->SendMode(modes, size, user);
+                                       modes.push_back(tokens[k]);
                                }
-                               break;
                        }
-               }
-       }
-};
 
-// stuff down here is the module-factory stuff. For basic modules you can ignore this.
+                       ServerInstance->Parser->CallHandler("MODE", modes, user);
+               }
 
-class ModuleModesOnConnectFactory : public ModuleFactory
-{
- public:
-       ModuleModesOnConnectFactory()
-       {
+               memcpy(ServerInstance->Config->DisabledUModes, save, 64);
        }
-       
-       ~ModuleModesOnConnectFactory()
-       {
-       }
-       
-       virtual Module * CreateModule(InspIRCd* Me)
-       {
-               return new ModuleModesOnConnect(Me);
-       }
-       
 };
 
-
-extern "C" void * init_module( void )
-{
-       return new ModuleModesOnConnectFactory;
-}
-
+MODULE_INIT(ModuleModesOnConnect)