Use the error name constants here now rather than magic numbers. magic numbers blow...

[user/henk/code/inspircd.git] / src / snomasks.cpp

diff --git a/src/snomasks.cpp b/src/snomasks.cpp
index e465b2ec80d063658edfc9ae0efef2fa048bed31..5f04931f991cb3335e7d4a7dbe3f678ec2ad45ce 100644 (file)
--- a/src/snomasks.cpp
+++ b/src/snomasks.cpp
@@ -2,26 +2,19 @@
  *       | Inspire Internet Relay Chat Daemon |
  *       +------------------------------------+
  *
- *  InspIRCd is copyright (C) 2002-2006 ChatSpike-Dev.
- *                       E-mail:
- *                <brain@chatspike.net>
- *                <Craig@chatspike.net>
+ *  InspIRCd: (C) 2002-2008 InspIRCd Development Team
+ * See: http://www.inspircd.org/wiki/index.php/Credits
  *
- * Written by Craig Edwards, Craig McLure, and others.
  * This program is free but copyrighted software; see
  *            the file COPYING for details.
  *
  * ---------------------------------------------------
  */
 
-#include <string>
-#include <vector>
+/* $Core */
+
+#include "inspircd.h"
 #include <stdarg.h>
-#include "configreader.h"
-#include "users.h"
-#include "modules.h"
-#include "commands.h"
-#include "xline.h"
 #include "snomasks.h"
 
 SnomaskManager::SnomaskManager(InspIRCd* Instance) : ServerInstance(Instance)
@@ -31,14 +24,28 @@ SnomaskManager::SnomaskManager(InspIRCd* Instance) : ServerInstance(Instance)
 }
 
 SnomaskManager::~SnomaskManager()
+{      
+       for (std::map<char, Snomask *>::iterator i = SnoMasks.begin(); i != SnoMasks.end(); i++)
+       {
+               delete i->second;
+       }
+       SnoMasks.clear();
+}
+
+void SnomaskManager::FlushSnotices()
 {
+       for (std::map<char, Snomask *>::iterator i = SnoMasks.begin(); i != SnoMasks.end(); i++)
+       {
+               i->second->Flush();
+       }
 }
 
 bool SnomaskManager::EnableSnomask(char letter, const std::string &type)
 {
        if (SnoMasks.find(letter) == SnoMasks.end())
        {
-               SnoMasks[letter] = type;
+               Snomask *s = new Snomask(ServerInstance, letter, type);
+               SnoMasks[letter] = s;
                return true;
        }
        return false;
@@ -49,6 +56,7 @@ bool SnomaskManager::DisableSnomask(char letter)
        SnoList::iterator n = SnoMasks.find(letter);
        if (n != SnoMasks.end())
        {
+               delete n->second; // destroy the snomask
                SnoMasks.erase(n);
                return true;
        }
@@ -61,22 +69,13 @@ void SnomaskManager::WriteToSnoMask(char letter, const std::string &text)
        SnoList::iterator n = SnoMasks.find(letter);
        if (n != SnoMasks.end())
        {
-               /* Only opers can receive snotices, so we iterate the oper list */
-               for (std::vector<userrec*>::iterator i = ServerInstance->all_opers.begin(); i != ServerInstance->all_opers.end(); i++)
-               {
-                       userrec* a = *i;
-                       if (IS_LOCAL(a) && a->modes[UM_SERVERNOTICE] && a->modes[UM_SNOMASK] && a->IsNoticeMaskSet(n->first))
-                       {
-                               /* send server notices to all with +ns */
-                               a->WriteServ("NOTICE %s :*** %s: %s",a->nick, n->second.c_str(), text.c_str());
-                       }
-               }
+               n->second->SendMessage(text);
        }
 }
 
 void SnomaskManager::WriteToSnoMask(char letter, const char* text, ...)
 {
-        char textbuffer[MAXBUF];
+       char textbuffer[MAXBUF];
        va_list argsPtr;
 
        va_start(argsPtr, text);
@@ -93,15 +92,74 @@ bool SnomaskManager::IsEnabled(char letter)
 
 void SnomaskManager::SetupDefaults()
 {
-       this->EnableSnomask('c',"CONNECT");             /* Local connect notices */
+       this->EnableSnomask('c',"CONNECT");                     /* Local connect notices */
        this->EnableSnomask('C',"REMOTECONNECT");       /* Remote connect notices */
-       this->EnableSnomask('q',"QUIT");                /* Local quit notices */
+       this->EnableSnomask('q',"QUIT");                        /* Local quit notices */
        this->EnableSnomask('Q',"REMOTEQUIT");          /* Remote quit notices */
-       this->EnableSnomask('k',"KILL");                /* Kill notices */
-       this->EnableSnomask('n',"NICK");                /* Nickchange notices */
-       this->EnableSnomask('l',"LINK");                /* Link notices */
-       this->EnableSnomask('o',"OPER");                /* Oper up/down notices */
-       this->EnableSnomask('d',"DEBUG");               /* Debug notices */
-       this->EnableSnomask('x',"XLINE");               /* Xline notice (g/z/q/k/e) */
+       this->EnableSnomask('k',"KILL");                        /* Kill notices */
+       this->EnableSnomask('K',"REMOTEKILL");          /* Remote kill notices */
+       this->EnableSnomask('l',"LINK");                        /* Link notices */
+       this->EnableSnomask('o',"OPER");                        /* Oper up/down notices */
+       this->EnableSnomask('A',"ANNOUNCEMENT");        /* formerly WriteOpers() - generic notices to all opers */
+       this->EnableSnomask('d',"DEBUG");                       /* Debug notices */
+       this->EnableSnomask('x',"XLINE");                       /* Xline notice (g/z/q/k/e) */
+       this->EnableSnomask('t',"STATS");                       /* Local or remote stats request */
+       this->EnableSnomask('f',"FLOOD");                       /* Flooding notices */
 }
 
+/*************************************************************************************/
+
+void Snomask::SendMessage(const std::string &message)
+{
+       if (message != LastMessage)
+       {
+               this->Flush();
+               LastMessage = message;
+       }
+       else
+       {
+               Count++;
+       }
+}
+
+void Snomask::Flush()
+{
+       if (this->LastMessage.empty())
+               return;
+
+       ServerInstance->Logs->Log("snomask", DEFAULT, "%s: %s", this->Description.c_str(), this->LastMessage.c_str());
+       if (Count > 1)
+               ServerInstance->Logs->Log("snomask", DEFAULT, "%s: (last message repeated %u times)", this->Description.c_str(), Count);
+
+
+       int MOD_RESULT = 0;
+       char mysnomask = MySnomask;
+       std::string desc = this->Description;
+
+       FOREACH_RESULT(I_OnSendSnotice, OnSendSnotice(mysnomask, desc, this->LastMessage));
+
+       if (MOD_RESULT != 1) // 1 blocks the message
+       {
+               /* Only opers can receive snotices, so we iterate the oper list */
+               std::list<User*>::iterator i = ServerInstance->Users->all_opers.begin();
+
+               while (i != ServerInstance->Users->all_opers.end())
+               {
+                       User* a = *i;
+                       if (IS_LOCAL(a) && a->IsModeSet('s') && a->IsNoticeMaskSet(mysnomask) && !a->quitting)
+                       {
+
+                               a->WriteServ("NOTICE %s :*** %s: %s", a->nick.c_str(), desc.c_str(), this->LastMessage.c_str());
+                               if (Count > 1)
+                               {
+                                       a->WriteServ("NOTICE %s :*** %s: (last message repeated %u times)", a->nick.c_str(), this->Description.c_str(), Count);
+                               }
+                       }
+
+                       i++;
+               }
+       }
+
+       LastMessage = "";
+       Count = 1;
+}