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 b407aece4a1af3a4c29519f1424f204b8af0f7bd..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")
        {
        }
 
@@ -64,6 +71,35 @@ class ModuleIRCv3EchoMessage : public Module
                }
        }
 
+       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.
@@ -71,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);