Converted more stuff to 'Implements' system

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

diff --git a/src/modules/m_conn_lusers.cpp b/src/modules/m_conn_lusers.cpp
index c0106566a2f640d98dcfb41782ae34bedf2fe8c1..b4bb0ce6d32a5348765e24a5ab722eff3e66f781 100644 (file)
--- a/src/modules/m_conn_lusers.cpp
+++ b/src/modules/m_conn_lusers.cpp
@@ -32,24 +32,43 @@ class ModuleConnLUSERS : public Module
         
         Server *Srv;
  public:
-       ModuleConnLUSERS()
+       ModuleConnLUSERS(Server* Me)
+               : Module::Module(Me)
        {
-               Srv = new Server;
+               Srv = Me;
        }
        
        virtual ~ModuleConnLUSERS()
        {
-               delete Srv;
        }
        
        virtual Version GetVersion()
        {
                return Version(1,0,0,1,VF_VENDOR);
        }
+
+       void Implements(char* List)
+       {
+               List[I_OnUserConnect] = 1;
+       }
        
        virtual void OnUserConnect(userrec* user)
        {
-               Srv->CallCommandHandler("LUSERS", NULL, 0, user);
+               // if we're using a protocol module, we cant just call
+               // the command handler because the protocol module
+               // has hooked it. We must call OnPreCommand in the
+               // protocol module. Yes, at some point there will
+               // be a way to get the current protocol module's name
+               // from the core and probably a pointer to its class.
+               Module* Proto = Srv->FindModule("m_spanningtree.so");
+               if (Proto)
+               {
+                       Proto->OnPreCommand("LUSERS", NULL, 0, user, true);
+               }
+               else
+               {
+                       Srv->CallCommandHandler("LUSERS", NULL, 0, user);
+               }
        }
 };
 
@@ -70,9 +89,9 @@ class ModuleConnLUSERSFactory : public ModuleFactory
        {
        }
        
-       virtual Module * CreateModule()
+       virtual Module * CreateModule(Server* Me)
        {
-               return new ModuleConnLUSERS;
+               return new ModuleConnLUSERS(Me);
        }
        
 };