#include "inspircd.h"
+#include "modules/exemption.h"
-typedef insp::flat_map<irc::string, irc::string> censor_t;
-
-/** Handles usermode +G
- */
-class CensorUser : public SimpleUserModeHandler
-{
- public:
- CensorUser(Module* Creator) : SimpleUserModeHandler(Creator, "u_censor", 'G') { }
-};
-
-/** Handles channel mode +G
- */
-class CensorChannel : public SimpleChannelModeHandler
-{
- public:
- CensorChannel(Module* Creator) : SimpleChannelModeHandler(Creator, "censor", 'G') { }
-};
+typedef insp::flat_map<std::string, std::string, irc::insensitive_swo> censor_t;
class ModuleCensor : public Module
{
+ CheckExemption::EventProvider exemptionprov;
censor_t censors;
- CensorUser cu;
- CensorChannel cc;
+ SimpleUserModeHandler cu;
+ SimpleChannelModeHandler cc;
public:
- ModuleCensor() : cu(this), cc(this) { }
+ ModuleCensor()
+ : exemptionprov(this)
+ , cu(this, "u_censor", 'G')
+ , cc(this, "censor", 'G')
+ {
+ }
// format of a config entry is <badword text="shit" replace="poo">
- ModResult OnUserPreMessage(User* user, void* dest, int target_type, std::string& text, char status, CUList& exempt_list, MessageType msgtype) CXX11_OVERRIDE
+ ModResult OnUserPreMessage(User* user, const MessageTarget& target, MessageDetails& details) CXX11_OVERRIDE
{
if (!IS_LOCAL(user))
return MOD_RES_PASSTHRU;
- bool active = false;
+ int numeric = 0;
+ const char* targetname = NULL;
- if (target_type == TYPE_USER)
- active = ((User*)dest)->IsModeSet(cu);
- else if (target_type == TYPE_CHANNEL)
+ switch (target.type)
{
- Channel* c = (Channel*)dest;
- active = c->IsModeSet(cc);
- ModResult res = ServerInstance->OnCheckExemption(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;
+
+ ModResult result = CheckExemption::Call(exemptionprov, user, targchan, "censor");
+ if (result == MOD_RES_ALLOW)
+ return MOD_RES_PASSTHRU;
- if (res == MOD_RES_ALLOW)
+ numeric = ERR_CANNOTSENDTOCHAN;
+ targetname = targchan->name.c_str();
+ break;
+ }
+
+ default:
return MOD_RES_PASSTHRU;
}
- if (!active)
- return MOD_RES_PASSTHRU;
-
- irc::string text2 = text.c_str();
for (censor_t::iterator index = censors.begin(); index != censors.end(); index++)
{
- if (text2.find(index->first) != irc::string::npos)
+ size_t censorpos;
+ while ((censorpos = irc::find(details.text, index->first)) != std::string::npos)
{
if (index->second.empty())
{
- user->WriteNumeric(ERR_WORDFILTERED, "%s %s :Your message contained a censored word, and was blocked", ((Channel*)dest)->name.c_str(), index->first.c_str());
+ user->WriteNumeric(numeric, targetname, "Your message contained a censored word (" + index->first + "), and was blocked");
return MOD_RES_DENY;
}
- SearchAndReplace(text2, index->first, index->second);
+ details.text.replace(censorpos, index->first.size(), index->second);
}
}
- text = text2.c_str();
return MOD_RES_PASSTHRU;
}
for (ConfigIter i = badwords.first; i != badwords.second; ++i)
{
ConfigTag* tag = i->second;
- std::string str = tag->getString("text");
- irc::string pattern(str.c_str());
- str = tag->getString("replace");
- censors[pattern] = irc::string(str.c_str());
+ const std::string text = tag->getString("text");
+ if (text.empty())
+ continue;
+
+ const std::string replace = tag->getString("replace");
+ censors[text] = replace;
}
}