WebSocket: use one WebSocket message per IRC message.

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

diff --git a/src/modules/m_censor.cpp b/src/modules/m_censor.cpp
index cb2bec85a9c70077241471d461cabf2959a3bc66..a9c55386c60c7d54a188cc3eeed560cba4a1cac2 100644 (file)
--- a/src/modules/m_censor.cpp
+++ b/src/modules/m_censor.cpp
@@ -46,23 +46,41 @@ class ModuleCensor : public Module
                if (!IS_LOCAL(user))
                        return MOD_RES_PASSTHRU;
 
-               bool active = false;
+               int numeric = 0;
+               const char* targetname = NULL;
 
-               if (target.type == MessageTarget::TYPE_USER)
-                       active = target.Get<User>()->IsModeSet(cu);
-               else if (target.type == MessageTarget::TYPE_CHANNEL)
+               switch (target.type)
                {
-                       Channel* c = target.Get<Channel>();
-                       active = c->IsModeSet(cc);
-                       ModResult res = CheckExemption::Call(exemptionprov, user, c, "censor");
+                       case MessageTarget::TYPE_USER:
+                       {
+                               User* targuser = target.Get<User>();
+                               if (!targuser->IsModeSet(cu))
+                                       return MOD_RES_PASSTHRU;
+
+                               numeric = ERR_CANTSENDTOUSER;
+                               targetname = targuser->nick.c_str();
+                               break;
+                       }
+
+                       case MessageTarget::TYPE_CHANNEL:
+                       {
+                               Channel* targchan = target.Get<Channel>();
+                               if (!targchan->IsModeSet(cc))
+                                       return MOD_RES_PASSTHRU;
 
-                       if (res == MOD_RES_ALLOW)
+                               ModResult result = CheckExemption::Call(exemptionprov, user, targchan, "censor");
+                               if (result == MOD_RES_ALLOW)
+                                       return MOD_RES_PASSTHRU;
+
+                               numeric = ERR_CANNOTSENDTOCHAN;
+                               targetname = targchan->name.c_str();
+                               break;
+                       }
+
+                       default:
                                return MOD_RES_PASSTHRU;
                }
 
-               if (!active)
-                       return MOD_RES_PASSTHRU;
-
                for (censor_t::iterator index = censors.begin(); index != censors.end(); index++)
                {
                        size_t censorpos;
@@ -70,8 +88,7 @@ class ModuleCensor : public Module
                        {
                                if (index->second.empty())
                                {
-                                       const std::string targname = target.type == MessageTarget::TYPE_CHANNEL ? target.Get<Channel>()->name : target.Get<User>()->nick;
-                                       user->WriteNumeric(ERR_WORDFILTERED, targname, index->first.c_str(), "Your message contained a censored word, and was blocked");
+                                       user->WriteNumeric(numeric, targetname, "Your message contained a censored word (" + index->first + "), and was blocked");
                                        return MOD_RES_DENY;
                                }