Header update: 2007 -> 2008

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

diff --git a/src/modules/m_blockamsg.cpp b/src/modules/m_blockamsg.cpp
index f1a388890f8a3a2c154b9d48b7a9f158f7a36462..70fb99b9a25acd68924c14ab45cec9c276b08131 100644 (file)
--- a/src/modules/m_blockamsg.cpp
+++ b/src/modules/m_blockamsg.cpp
@@ -2,7 +2,7 @@
  *       | Inspire Internet Relay Chat Daemon |
  *       +------------------------------------+
  *
- *  InspIRCd: (C) 2002-2007 InspIRCd Development Team
+ *  InspIRCd: (C) 2002-2008 InspIRCd Development Team
  * See: http://www.inspircd.org/wiki/index.php/Credits
  *
  * This program is free but copyrighted software; see
@@ -11,19 +11,17 @@
  * ---------------------------------------------------
  */
 
-#include <string>
-#include <time.h>
-#include "inspircd_config.h"
-#include "users.h"
-#include "channels.h"
-#include "modules.h"
-
-#include "hashcomp.h"
 #include "inspircd.h"
 
 /* $ModDesc: Attempt to block /amsg, at least some of the irritating mIRC scripts. */
 
 enum BlockAction { IBLOCK_KILL, IBLOCK_KILLOPERS, IBLOCK_NOTICE, IBLOCK_NOTICEOPERS, IBLOCK_SILENT };
+/*     IBLOCK_NOTICE           - Send a notice to the user informing them of what happened.
+ *     IBLOCK_NOTICEOPERS      - Send a notice to the user informing them and send an oper notice.
+ *     IBLOCK_SILENT           - Generate no output, silently drop messages.
+ *     IBLOCK_KILL                     - Kill the user with the reason "Global message (/amsg or /ame) detected".
+ *     IBLOCK_KILLOPERS        - As above, but send an oper notice as well. This is the default.
+ */
 
 /** Holds a blocked message's details
  */
@@ -34,27 +32,25 @@ public:
        irc::string target;
        time_t sent;
 
-       BlockedMessage(const std::string &msg, const irc::string &tgt, time_t when) : message(msg), target(tgt), sent(when)
+       BlockedMessage(const std::string &msg, const irc::string &tgt, time_t when)
+       : message(msg), target(tgt), sent(when)
        {
        }
 };
 
 class ModuleBlockAmsg : public Module
 {
-       
        int ForgetDelay;
        BlockAction action;
-public:
+       
+ public:
        ModuleBlockAmsg(InspIRCd* Me) : Module(Me)
        {
-               
                this->OnRehash(NULL,"");
+               Implementation eventlist[] = { I_OnRehash, I_OnPreCommand, I_OnCleanup };
+               ServerInstance->Modules->Attach(eventlist, this, 3);
        }
 
-       void Implements(char* List)
-       {
-               List[I_OnRehash] = List[I_OnPreCommand] = List[I_OnCleanup] = 1;
-       }
        
        virtual ~ModuleBlockAmsg()
        {
@@ -65,15 +61,16 @@ public:
                return Version(1,1,0,0,VF_VENDOR,API_VERSION);
        }
        
-       virtual void OnRehash(userrec* user, const std::string &parameter)
+       virtual void OnRehash(User* user, const std::string &parameter)
        {
-               ConfigReader* Conf = new ConfigReader(ServerInstance);
+               ConfigReader Conf(ServerInstance);
                
-               ForgetDelay = Conf->ReadInteger("blockamsg", "delay", 0, false);
-               if(Conf->GetError() == CONF_VALUE_NOT_FOUND)
+               ForgetDelay = Conf.ReadInteger("blockamsg", "delay", 0, false);
+               
+               if(Conf.GetError() == CONF_VALUE_NOT_FOUND)
                        ForgetDelay = -1;
                        
-               std::string act = Conf->ReadValue("blockamsg", "action", 0);
+               std::string act = Conf.ReadValue("blockamsg", "action", 0);
                
                if(act == "notice")
                        action = IBLOCK_NOTICE;
@@ -85,11 +82,9 @@ public:
                        action = IBLOCK_KILL;
                else
                        action = IBLOCK_KILLOPERS;
-
-               DELETE(Conf);
        }
 
-       virtual int OnPreCommand(const std::string &command, const char** parameters, int pcnt, userrec *user, bool validated, const std::string &original_line)
+       virtual int OnPreCommand(const std::string &command, const char** parameters, int pcnt, User *user, bool validated, const std::string &original_line)
        {
                // Don't do anything with unregistered users, or remote ones.
                if(!user || (user->registered != REG_ALL) || !IS_LOCAL(user))
@@ -107,14 +102,25 @@ public:
                        
                        int targets = 1;
                        int userchans = 0;
-                       
-                       // Decrement if the first target wasn't a channel.
+               
                        if(*parameters[0] != '#')
+                       {
+                               // Decrement if the first target wasn't a channel.
                                targets--;
+                       }
                        
                        for(const char* c = parameters[0]; *c; c++)
                                if((*c == ',') && *(c+1) && (*(c+1) == '#'))
                                        targets++;
+                               
+                       /* targets should now contain the number of channel targets the msg/notice was pointed at.
+                        * If the msg/notice was a PM there should be no channel targets and 'targets' should = 0.
+                        * We don't want to block PMs so...
+                        */
+                       if(targets == 0)
+                       {
+                               return 0;
+                       }
                                        
                        userchans = user->chans.size();
 
@@ -134,7 +140,7 @@ public:
                                        ServerInstance->WriteOpers("*** %s had an /amsg or /ame denied", user->nick);
 
                                if(action == IBLOCK_KILL || action == IBLOCK_KILLOPERS)
-                                       userrec::QuitUser(ServerInstance, user, "Global message (/amsg or /ame) detected");
+                                       User::QuitUser(ServerInstance, user, "Global message (/amsg or /ame) detected");
                                else if(action == IBLOCK_NOTICE || action == IBLOCK_NOTICEOPERS)
                                        user->WriteServ( "NOTICE %s :Global message (/amsg or /ame) detected", user->nick);
                                                                        
@@ -161,12 +167,12 @@ public:
        {
                if(target_type == TYPE_USER)
                {
-                       userrec* user = (userrec*)item;
+                       User* user = (User*)item;
                        BlockedMessage* m;
                        user->GetExt("amsgblock", m);
                        if(m)
                        {
-                               DELETE(m);
+                               delete m;
                                user->Shrink("amsgblock");
                        }
                }
@@ -174,25 +180,4 @@ public:
 };
 
 
-class ModuleBlockAmsgFactory : public ModuleFactory
-{
-       public:
-               ModuleBlockAmsgFactory()
-               {
-               }
-       
-               ~ModuleBlockAmsgFactory()
-               {
-               }
-       
-               virtual Module * CreateModule(InspIRCd* Me)
-               {
-                       return new ModuleBlockAmsg(Me);
-               }
-};
-
-
-extern "C" DllExport void * init_module( void )
-{
-       return new ModuleBlockAmsgFactory;
-}
+MODULE_INIT(ModuleBlockAmsg)