In the grand tradition of huge fucking commits:

[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 f7ee81fd5ed7b10632e7abca56014de6f5489bb7..167a12bec07722eab197892a9a9438cccda94772 100644 (file)
--- a/src/modules/m_blockcaps.cpp
+++ b/src/modules/m_blockcaps.cpp
@@ -2,121 +2,138 @@
  *       | Inspire Internet Relay Chat Daemon |
  *       +------------------------------------+
  *
- *  InspIRCd is copyright (C) 2002-2006 ChatSpike-Dev.
- *                       E-mail:
- *                <brain@chatspike.net>
- *               <Craig@chatspike.net>
- *                <omster@gmail.com>
- *     
- * Written by Craig Edwards, Craig McLure, and others.
+ *  InspIRCd: (C) 2002-2007 InspIRCd Development Team
+ * See: http://www.inspircd.org/wiki/index.php/Credits
+ *
  * This program is free but copyrighted software; see
  *            the file COPYING for details.
  *
  * ---------------------------------------------------
  */
 
-#include "users.h"
-#include "channels.h"
-#include "modules.h"
-#include "helperfuncs.h"
+#include "inspircd.h"
 
 /* $ModDesc: Provides support for channel mode +P to block all-CAPS channel messages and notices */
 
+
+/** Handles the +P channel mode
+ */
+class BlockCaps : public ModeHandler
+{
+ public:
+       BlockCaps(InspIRCd* Instance) : ModeHandler(Instance, 'P', 0, 0, false, MODETYPE_CHANNEL, false) { }
+
+       ModeAction OnModeChange(User* source, User* dest, Channel* channel, std::string &parameter, bool adding)
+       {
+               if (adding)
+               {
+                       if (!channel->IsModeSet('P'))
+                       {
+                               channel->SetMode('P',true);
+                               return MODEACTION_ALLOW;
+                       }
+               }
+               else
+               {
+                       if (channel->IsModeSet('P'))
+                       {
+                               channel->SetMode('P',false);
+                               return MODEACTION_ALLOW;
+                       }
+               }
+
+               return MODEACTION_DENY;
+       }
+};
+
 class ModuleBlockCAPS : public Module
 {
-       Server *Srv;
+       BlockCaps* bc;
+       int percent;
+       unsigned int minlen;
+       char capsmap[256];
 public:
        
-       ModuleBlockCAPS(Server* Me) : Module::Module(Me)
+       ModuleBlockCAPS(InspIRCd* Me) : Module(Me)
        {
-               Srv = Me;
-               Srv->AddExtendedMode('P', MT_CHANNEL, false, 0, 0);
+               OnRehash(NULL,"");
+               bc = new BlockCaps(ServerInstance);
+               if (!ServerInstance->AddMode(bc, 'P'))
+                       throw ModuleException("Could not add new modes!");
        }
 
        void Implements(char* List)
        {
-               List[I_On005Numeric] = List[I_OnUserPreMessage] = List[I_OnUserPreNotice] = List[I_OnExtendedMode] = 1;
+               List[I_OnUserPreMessage] = List[I_OnUserPreNotice] = List[I_OnRehash] = 1;
        }
 
-       virtual void On005Numeric(std::string &output)
+       virtual void OnRehash(User* user, const std::string &param)
        {
-               InsertMode(output, "P", 4);
+               ReadConf();
        }
 
-       virtual int OnUserPreMessage(userrec* user,void* dest,int target_type, std::string &text, char status)
+       virtual int OnUserPreMessage(User* user,void* dest,int target_type, std::string &text, char status, CUList &exempt_list)
        {
                if (target_type == TYPE_CHANNEL)
                {
-                       chanrec* c = (chanrec*)dest;
+                       if ((!IS_LOCAL(user)) || (text.length() < minlen))
+                               return 0;
+
+                       Channel* c = (Channel*)dest;
 
-                       if (c->IsCustomModeSet('P'))
+                       if (c->IsModeSet('P'))
                        {
-                               char* i = (char*)text.c_str();
-                               for (; *i; i++)
+                               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 != ' ') && (*i != '\t')) && ((*i < 'A') || (*i > 'Z')))
-                                       {
-                                               return 0;
-                                       }
+                                       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;
                                }
-                               
-                               WriteServ(user->fd, "404 %s %s :Can't send all-CAPS to channel (+P set)", user->nick, c->name);
-                               return 1;
                        }
                }
                return 0;
        }
-       
-       virtual int OnUserPreNotice(userrec* user,void* dest,int target_type, std::string &text, char status)
+
+       virtual int OnUserPreNotice(User* user,void* dest,int target_type, std::string &text, char status, CUList &exempt_list)
        {
-               return OnUserPreMessage(user,dest,target_type,text,status);
+               return OnUserPreMessage(user,dest,target_type,text,status,exempt_list);
        }
-       
-       virtual int OnExtendedMode(userrec* user, void* target, char modechar, int type, bool mode_on, string_list &params)
+
+       void ReadConf()
        {
-               // check if this is our mode character...
-               if ((modechar == 'P') && (type == MT_CHANNEL))
-               {
-                       log(DEBUG,"Allowing P change");
-                       return 1;
+               ConfigReader Conf(ServerInstance);
+               percent = Conf.ReadInteger("blockcaps", "percent", "100", 0, true);
+               minlen = Conf.ReadInteger("blockcaps", "minlen", "1", 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 < 1 || percent > 100)
+               {
+                       ServerInstance->Log(DEFAULT, "<blockcaps:percent> out of range, setting to default of 100.");
+                       percent = 100;
                }
-               else
+               if (minlen < 1 || minlen > MAXBUF-1)
                {
-                       return 0;
+                       ServerInstance->Log(DEFAULT, "<blockcaps:minlen> out of range, setting to default of 1.");
+                       minlen = 1;
                }
        }
 
        virtual ~ModuleBlockCAPS()
        {
+               ServerInstance->Modes->DelMode(bc);
+               DELETE(bc);
        }
-       
-       virtual Version GetVersion()
-       {
-               return Version(1,0,0,0,VF_STATIC|VF_VENDOR);
-       }
-};
 
-
-class ModuleBlockCAPSFactory : public ModuleFactory
-{
- public:
-       ModuleBlockCAPSFactory()
-       {
-       }
-       
-       ~ModuleBlockCAPSFactory()
-       {
-       }
-       
-       virtual Module * CreateModule(Server* Me)
+       virtual Version GetVersion()
        {
-               return new ModuleBlockCAPS(Me);
+               return Version(1,1,0,0,VF_COMMON|VF_VENDOR,API_VERSION);
        }
-       
 };
 
-
-extern "C" void * init_module( void )
-{
-       return new ModuleBlockCAPSFactory;
-}
+MODULE_INIT(ModuleBlockCAPS)