New 'Implements' system

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

diff --git a/src/modules/m_restrictmsg.cpp b/src/modules/m_restrictmsg.cpp
index 68c842ae86e8460b5b48919429ddf120cd97b289..5043b3f8dd3dabfa51ccf7940598925fe82ccdcd 100644 (file)
--- a/src/modules/m_restrictmsg.cpp
+++ b/src/modules/m_restrictmsg.cpp
@@ -14,12 +14,15 @@
  * ---------------------------------------------------
  */
 
+using namespace std;
+
 #include <stdio.h>
 #include <string>
 #include <vector>
 #include "users.h"
 #include "channels.h"
 #include "modules.h"
+#include "helperfuncs.h"
 
 /* $ModDesc: Forbids users from messaging each other. Users may still message opers and opers may message other opers. */
 
@@ -29,9 +32,10 @@ class ModuleRestrictMsg : public Module
        Server *Srv;
  public:
  
-       ModuleRestrictMsg()
+       ModuleRestrictMsg(Server* Me)
+               : Module::Module(Me)
        {
-               Srv = new Server;
+               Srv = Me;
        }
 
        virtual int OnUserPreMessage(userrec* user,void* dest,int target_type, std::string &text)
@@ -62,7 +66,6 @@ class ModuleRestrictMsg : public Module
 
        virtual ~ModuleRestrictMsg()
        {
-               delete Srv;
        }
        
        virtual Version GetVersion()
@@ -83,9 +86,9 @@ class ModuleRestrictMsgFactory : public ModuleFactory
        {
        }
        
-       virtual Module * CreateModule()
+       virtual Module * CreateModule(Server* Me)
        {
-               return new ModuleRestrictMsg;
+               return new ModuleRestrictMsg(Me);
        }
        
 };