Fix

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

diff --git a/src/snomasks.cpp b/src/snomasks.cpp
index dea7cc446aff57d61534a2275ee0609f14830291..dbbc11aaf876c945553e443e5bab9799ac442bf3 100644 (file)
--- a/src/snomasks.cpp
+++ b/src/snomasks.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
@@ -24,14 +24,41 @@ 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)
+bool SnomaskManager::SetLocalOnly(char letter, bool local)
+{
+       SnoList::iterator n = SnoMasks.find(letter);
+       if (n != SnoMasks.end())
+       {
+               n->second->LocalOnly = local;
+               return n->second->LocalOnly;
+       }
+
+       ServerInstance->Logs->Log("SNOMASK", DEFAULT, "Snomask %d not found, unmatched modules?", letter);
+       return true;
+}
+
+bool SnomaskManager::EnableSnomask(char letter, const std::string &type, bool local)
 {
        if (SnoMasks.find(letter) == SnoMasks.end())
        {
-               SnoMasks[letter] = type;
+               Snomask *s = new Snomask(ServerInstance, letter, type, local);
+               SnoMasks[letter] = s;
                return true;
        }
        return false;
@@ -42,6 +69,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;
        }
@@ -54,16 +82,7 @@ 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::list<User*>::iterator i = ServerInstance->all_opers.begin(); i != ServerInstance->all_opers.end(); i++)
-               {
-                       User* a = *i;
-                       if (IS_LOCAL(a) && a->IsModeSet('s') && a->IsModeSet('n') && 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);
        }
 }
 
@@ -86,17 +105,67 @@ bool SnomaskManager::IsEnabled(char letter)
 
 void SnomaskManager::SetupDefaults()
 {
-       this->EnableSnomask('c',"CONNECT");                     /* Local connect notices */
-       this->EnableSnomask('C',"REMOTECONNECT");       /* Remote connect notices */
-       this->EnableSnomask('q',"QUIT");                        /* Local quit notices */
-       this->EnableSnomask('Q',"REMOTEQUIT");          /* Remote quit notices */
-       this->EnableSnomask('k',"KILL");                        /* Kill notices */
-       this->EnableSnomask('K',"REMOTEKILL");          /* Remote kill notices */
+       this->EnableSnomask('c',"CONNECT", true);                       /* Local connect notices */
+       this->EnableSnomask('C',"REMOTECONNECT");               /* Remote connect notices */
+       this->EnableSnomask('q',"QUIT", true);                  /* Local quit notices */
+       this->EnableSnomask('Q',"REMOTEQUIT");                  /* Remote quit notices */
+       this->EnableSnomask('k',"KILL", true);                  /* 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;
+
+       /* Only opers can receive snotices, so we iterate the oper list */
+       for (std::list<User*>::iterator i = ServerInstance->Users->all_opers.begin(); i != ServerInstance->Users->all_opers.end(); i++)
+       {
+               User* a = *i;
+               if (IS_LOCAL(a) && a->IsModeSet('s') && a->IsModeSet('n') && a->IsNoticeMaskSet(MySnomask) && !a->quitting)
+               {
+
+                       a->WriteServ("NOTICE %s :*** %s: %s", a->nick, this->Description.c_str(), this->LastMessage.c_str());
+                       if (Count > 1)
+                       {
+                               a->WriteServ("NOTICE %s :*** %s: (last message repeated %u times)", a->nick, this->Description.c_str(), Count);
+                       }
+               }
+       }
+
+       if (!LocalOnly)
+       {
+               // XXX this is a bit ugly.
+               std::string sno;
+               sno += MySnomask;
+
+               ServerInstance->PI->SendSNONotice(sno, this->Description + ": " + this->LastMessage);
+               if (Count > 1)
+                       ServerInstance->PI->SendSNONotice(sno, this->Description + ": (last message repeated " + ConvToStr(Count) + " times)");
+       }
+
+       LastMessage = "";
+       Count = 1;
+}