Review and optimize

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

diff --git a/src/modules/m_noctcp.cpp b/src/modules/m_noctcp.cpp
index 4d58151f09aab1e6620d70013caf9a812fed265b..2dd24955916b77867da7e0abb1681b20cc173d0c 100644 (file)
--- a/src/modules/m_noctcp.cpp
+++ b/src/modules/m_noctcp.cpp
@@ -14,10 +14,13 @@
  * ---------------------------------------------------
  */
 
+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 +c */
 
@@ -27,11 +30,31 @@ class ModuleNoCTCP : public Module
        
  public:
  
-       ModuleNoCTCP()
+       ModuleNoCTCP(Server* Me)
+               : Module::Module(Me)
        {
-               Srv = new Server;
+               Srv = Me;
                Srv->AddExtendedMode('C',MT_CHANNEL,false,0,0);
        }
+
+        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 + "C";
+                        }
+                        temp2 = temp2 + temp1 + " ";
+                }
+               if (temp2.length())
+                       output = temp2.substr(0,temp2.length()-1);
+        }
        
        virtual int OnUserPreMessage(userrec* user,void* dest,int target_type, std::string &text)
        {
@@ -78,6 +101,7 @@ class ModuleNoCTCP : public Module
                // check if this is our mode character...
                if ((modechar == 'C') && (type == MT_CHANNEL))
                {
+                       log(DEBUG,"Allowing C change");
                        return 1;
                }
                else
@@ -88,12 +112,11 @@ class ModuleNoCTCP : public Module
 
        virtual ~ModuleNoCTCP()
        {
-               delete Srv;
        }
        
        virtual Version GetVersion()
        {
-               return Version(1,0,0,0);
+               return Version(1,0,0,0,VF_STATIC|VF_VENDOR);
        }
 };
 
@@ -109,9 +132,9 @@ class ModuleNoCTCPFactory : public ModuleFactory
        {
        }
        
-       virtual Module * CreateModule()
+       virtual Module * CreateModule(Server* Me)
        {
-               return new ModuleNoCTCP;
+               return new ModuleNoCTCP(Me);
        }
        
 };