Implement support for IRCv3 client-to-client tags.

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

diff --git a/src/modules/m_ircv3_echomessage.cpp b/src/modules/m_ircv3_echomessage.cpp
index f6eae5a00741c92038d8ca61c3811ad8831b026a..3ec534e916725dd7359009e8abe992d49368754a 100644 (file)
--- a/src/modules/m_ircv3_echomessage.cpp
+++ b/src/modules/m_ircv3_echomessage.cpp
@@ -20,14 +20,21 @@
 
 #include "inspircd.h"
 #include "modules/cap.h"
+#include "modules/ctctags.h"
 
-class ModuleIRCv3EchoMessage : public Module
+class ModuleIRCv3EchoMessage
+       : public Module
+       , public CTCTags::EventListener
 {
+ private:
        Cap::Capability cap;
+       ClientProtocol::EventProvider tagmsgprov;
 
  public:
        ModuleIRCv3EchoMessage()
-               : cap(this, "echo-message")
+               : CTCTags::EventListener(this)
+               , cap(this, "echo-message")
+               , tagmsgprov(this, "TAGMSG")
        {
        }
 
@@ -40,29 +47,59 @@ class ModuleIRCv3EchoMessage : public Module
                LocalUser* const localuser = static_cast<LocalUser*>(user);
 
                const std::string& text = details.echo_original ? details.original_text : details.text;
+               const ClientProtocol::TagMap& tags = details.echo_original ? details.tags_in : details.tags_out;
                if (target.type == MessageTarget::TYPE_USER)
                {
                        User* destuser = target.Get<User>();
                        ClientProtocol::Messages::Privmsg privmsg(ClientProtocol::Messages::Privmsg::nocopy, user, destuser, text, details.type);
-                       privmsg.AddTags(details.tags_in);
+                       privmsg.AddTags(tags);
                        localuser->Send(ServerInstance->GetRFCEvents().privmsg, privmsg);
                }
                else if (target.type == MessageTarget::TYPE_CHANNEL)
                {
                        Channel* chan = target.Get<Channel>();
                        ClientProtocol::Messages::Privmsg privmsg(ClientProtocol::Messages::Privmsg::nocopy, user, chan, text, details.type, target.status);
-                       privmsg.AddTags(details.tags_in);
+                       privmsg.AddTags(tags);
                        localuser->Send(ServerInstance->GetRFCEvents().privmsg, privmsg);
                }
                else
                {
                        const std::string* servername = target.Get<std::string>();
                        ClientProtocol::Messages::Privmsg privmsg(ClientProtocol::Messages::Privmsg::nocopy, user, *servername, text, details.type);
-                       privmsg.AddTags(details.tags_in);
+                       privmsg.AddTags(tags);
                        localuser->Send(ServerInstance->GetRFCEvents().privmsg, privmsg);
                }
        }
 
+       void OnUserPostTagMessage(User* user, const MessageTarget& target, const CTCTags::TagMessageDetails& details) CXX11_OVERRIDE
+       {
+               if (!cap.get(user) || !details.echo)
+                       return;
+
+               // Caps are only set on local users
+               LocalUser* const localuser = static_cast<LocalUser*>(user);
+
+               const ClientProtocol::TagMap& tags = details.echo_original ? details.tags_in : details.tags_out;
+               if (target.type == MessageTarget::TYPE_USER)
+               {
+                       User* destuser = target.Get<User>();
+                       CTCTags::TagMessage message(user, destuser, tags);
+                       localuser->Send(tagmsgprov, message);
+               }
+               else if (target.type == MessageTarget::TYPE_CHANNEL)
+               {
+                       Channel* chan = target.Get<Channel>();
+                       CTCTags::TagMessage message(user, chan, tags);
+                       localuser->Send(tagmsgprov, message);
+               }
+               else
+               {
+                       const std::string* servername = target.Get<std::string>();
+                       CTCTags::TagMessage message(user, servername->c_str(), tags);
+                       localuser->Send(tagmsgprov, message);
+               }
+       }
+
        void OnUserMessageBlocked(User* user, const MessageTarget& target, const MessageDetails& details) CXX11_OVERRIDE
        {
                // Prevent spammers from knowing that their spam was blocked.
@@ -70,6 +107,13 @@ class ModuleIRCv3EchoMessage : public Module
                        OnUserPostMessage(user, target, details);
        }
 
+       void OnUserTagMessageBlocked(User* user, const MessageTarget& target, const CTCTags::TagMessageDetails& details) CXX11_OVERRIDE
+       {
+               // Prevent spammers from knowing that their spam was blocked.
+               if (details.echo_original)
+                       OnUserPostTagMessage(user, target, details);
+       }
+
        Version GetVersion() CXX11_OVERRIDE
        {
                return Version("Provides the echo-message IRCv3 extension", VF_VENDOR);