Remove unneeded headers from spanningtree. This was done to the rest of the source...

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

diff --git a/src/modules/m_blockcaps.cpp b/src/modules/m_blockcaps.cpp
index 662bdc865ab93cf7245ac27b50bb3a8419cff274..21e08470f7518694756d0e9a99b128de4ffc5927 100644 (file)
--- a/src/modules/m_blockcaps.cpp
+++ b/src/modules/m_blockcaps.cpp
@@ -11,11 +11,7 @@
  * ---------------------------------------------------
  */
 
-#include "users.h"
-#include "channels.h"
-#include "modules.h"
 #include "inspircd.h"
-#include "mode.h"
 
 /* $ModDesc: Provides support for channel mode +P to block all-CAPS channel messages and notices */
 
@@ -53,11 +49,12 @@ class BlockCaps : public ModeHandler
 class ModuleBlockCAPS : public Module
 {
        BlockCaps* bc;
-       float percent;
+       int percent;
        unsigned int minlen;
+       char capsmap[256];
 public:
        
-       ModuleBlockCAPS(InspIRCd* Me) : Module::Module(Me)
+       ModuleBlockCAPS(InspIRCd* Me) : Module(Me)
        {
                OnRehash(NULL,"");
                bc = new BlockCaps(ServerInstance);
@@ -77,7 +74,6 @@ public:
 
        virtual int OnUserPreMessage(userrec* user,void* dest,int target_type, std::string &text, char status, CUList &exempt_list)
        {
-               ServerInstance->Log(DEBUG, "*** " + ConvToStr( ( 20 * 100 / 26)  ));
                if (target_type == TYPE_CHANNEL)
                {
                        if ((!IS_LOCAL(user)) || (text.length() < minlen))
@@ -87,15 +83,12 @@ public:
 
                        if (c->IsModeSet('P'))
                        {
-                               float caps = 0;
+                               int caps = 0;
                                for (std::string::iterator i = text.begin(); i != text.end(); i++)
+                                       caps += capsmap[(unsigned char)*i];
+                               if ( ((caps*100)/(int)text.length()) >= percent )
                                {
-                                       if ( (*i >= 'A') && (*i <= 'Z'))
-                                               caps++;
-                               }
-                               if ( ((caps / text.length()) * 100) >= percent )
-                               {
-                                       user->WriteServ( "404 %s %s :Can't send all-CAPS to channel (+P set)", user->nick, c->name);
+                                       user->WriteServ( "404 %s %s :Your line cannot be more than %d%% capital letters if it is %d or more letters long", user->nick, c->name, percent, minlen);
                                        return 1;
                                }
                        }
@@ -113,6 +106,12 @@ public:
                ConfigReader Conf(ServerInstance);
                percent = Conf.ReadInteger("blockcaps", "percent", "100", 0, true);
                minlen = Conf.ReadInteger("blockcaps", "minlen", "0", 0, true);
+               std::string hmap = Conf.ReadValue("blockcaps", "capsmap", 0);
+               if (hmap.empty())
+                       hmap = "ABCDEFGHIJKLMNOPQRSTUVWXYZ";
+               memset(&capsmap, 0, 255);
+               for (std::string::iterator n = hmap.begin(); n != hmap.end(); n++)
+                       capsmap[(unsigned char)*n] = 1;
                if (percent < 0 || percent > 100)
                {
                        ServerInstance->Log(DEFAULT, "<blockcaps:percent> out of range, setting to default of 100.");
@@ -137,27 +136,4 @@ public:
        }
 };
 
-
-class ModuleBlockCAPSFactory : public ModuleFactory
-{
- public:
-       ModuleBlockCAPSFactory()
-       {
-       }
-       
-       ~ModuleBlockCAPSFactory()
-       {
-       }
-       
-       virtual Module * CreateModule(InspIRCd* Me)
-       {
-               return new ModuleBlockCAPS(Me);
-       }
-       
-};
-
-
-extern "C" void * init_module( void )
-{
-       return new ModuleBlockCAPSFactory;
-}
+MODULE_INIT(ModuleBlockCAPS)