Rewrite SnomaskManager::SendMessage() and Flush(), split code into functions

1 files changed, 61 insertions, 63 deletions

diff --git a/src/snomasks.cpp b/src/snomasks.cpp
index 0a0d3cab2..fa8e5ab60 100644
--- a/src/snomasks.cpp
+++ b/src/snomasks.cpp
@@ -76,45 +76,35 @@ SnomaskManager::SnomaskManager()
 	EnableSnomask('t',"STATS");			/* Local or remote stats request */
 }
 
-/*************************************************************************************/
+bool SnomaskManager::IsSnomaskUsable(char ch) const
+{
+	return ((isalpha(ch)) && (!masks[tolower(ch) - 'a'].Description.empty()));
+}
+
+Snomask::Snomask()
+	: Count(0)
+{
+}
 
-void Snomask::SendMessage(const std::string &message, char mysnomask)
+void Snomask::SendMessage(const std::string& message, char letter)
 {
-	if (ServerInstance->Config->NoSnoticeStack || message != LastMessage || mysnomask != LastLetter)
+	if ((!ServerInstance->Config->NoSnoticeStack) && (message == LastMessage) && (letter == LastLetter))
 	{
-		this->Flush();
-		LastMessage = message;
-		LastLetter = mysnomask;
-
-		std::string desc = Description;
-		if (desc.empty())
-			desc = std::string("SNO-") + (char)tolower(mysnomask);
-		if (isupper(mysnomask))
-			desc = "REMOTE" + desc;
-		ModResult MOD_RESULT;
-		ServerInstance->Logs->Log("snomask", LOG_DEFAULT, "%s: %s", desc.c_str(), message.c_str());
-
-		FIRST_MOD_RESULT(OnSendSnotice, MOD_RESULT, (mysnomask, desc, message));
-
-		LastBlocked = (MOD_RESULT == MOD_RES_DENY);
-
-		if (!LastBlocked)
-		{
-			/* 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->WriteNotice("*** " + desc + ": " + message);
-				}
-
-				i++;
-			}
-		}
+		Count++;
+		return;
 	}
+
+	this->Flush();
+
+	std::string desc = GetDescription(letter);
+	ModResult MOD_RESULT;
+	FIRST_MOD_RESULT(OnSendSnotice, MOD_RESULT, (letter, desc, message));
+	if (MOD_RESULT == MOD_RES_DENY)
+		return;
+
+	Snomask::Send(letter, desc, message);
+	LastMessage = message;
+	LastLetter = letter;
 	Count++;
 }
 
@@ -122,36 +112,44 @@ void Snomask::Flush()
 {
 	if (Count > 1)
 	{
-		std::string desc = Description;
-		if (desc.empty())
-			desc = std::string("SNO-") + (char)tolower(LastLetter);
-		if (isupper(LastLetter))
-			desc = "REMOTE" + desc;
-		std::string mesg = "(last message repeated "+ConvToStr(Count)+" times)";
-
-		ServerInstance->Logs->Log("snomask", LOG_DEFAULT, "%s: %s", desc.c_str(), mesg.c_str());
-
-		FOREACH_MOD(I_OnSendSnotice, OnSendSnotice(LastLetter, desc, mesg));
-
-		if (!LastBlocked)
-		{
-			/* 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(LastLetter) && !a->quitting)
-				{
-					a->WriteNotice("*** " + desc + ": " + mesg);
-				}
-
-				i++;
-			}
-		}
+		std::string desc = GetDescription(LastLetter);
+		std::string msg = "(last message repeated " + ConvToStr(Count) + " times)";
 
+		FOREACH_MOD(I_OnSendSnotice, OnSendSnotice(LastLetter, desc, msg));
+		Snomask::Send(LastLetter, desc, msg);
 	}
+
 	LastMessage.clear();
-	LastBlocked = false;
 	Count = 0;
 }
+
+void Snomask::Send(char letter, const std::string& desc, const std::string& msg)
+{
+	std::string log = desc;
+	log.append(": ").append(msg);
+	ServerInstance->Logs->Log("snomask", LOG_DEFAULT, log);
+
+	std::string finalmsg = "*** ";
+	finalmsg.append(log);
+	/* Only opers can receive snotices, so we iterate the oper list */
+	const std::list<User*>& opers = ServerInstance->Users->all_opers;
+	for (std::list<User*>::const_iterator i = opers.begin(); i != opers.end(); ++i)
+	{
+		User* user = *i;
+		// IsNoticeMaskSet() returns false for opers who aren't +s, no need to check for it seperately
+		if (IS_LOCAL(user) && user->IsNoticeMaskSet(letter))
+			user->WriteNotice(finalmsg);
+	}
+}
+
+std::string Snomask::GetDescription(char letter) const
+{
+	std::string ret;
+	if (isupper(letter))
+		ret = "REMOTE";
+	if (!Description.empty())
+		ret += Description;
+	else
+		ret += std::string("SNO-") + (char)tolower(letter);
+	return ret;
+}