2 * InspIRCd -- Internet Relay Chat Daemon
4 * Copyright (C) 2019 Peter Powell <petpow@saberuk.com>
5 * Copyright (C) 2016 Attila Molnar <attilamolnar@hush.com>
7 * This file is part of InspIRCd. InspIRCd is free software: you can
8 * redistribute it and/or modify it under the terms of the GNU General Public
9 * License as published by the Free Software Foundation, version 2.
11 * This program is distributed in the hope that it will be useful, but WITHOUT
12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
13 * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
16 * You should have received a copy of the GNU General Public License
17 * along with this program. If not, see <http://www.gnu.org/licenses/>.
22 #include "modules/cap.h"
23 #include "modules/ctctags.h"
29 const std::string::size_type baselen;
34 , strid(InspIRCd::Format("%s~%lu~", ServerInstance->Config->GetSID().c_str(), ServerInstance->startup_time))
35 , baselen(strid.length())
39 const std::string& GetNext()
42 strid.append(ConvToStr(counter++));
47 class MsgIdTag : public ClientProtocol::MessageTagProvider
50 Cap::Reference ctctagcap;
53 MsgIdGenerator generator;
56 : ClientProtocol::MessageTagProvider(mod)
57 , ctctagcap(mod, "message-tags")
61 void OnPopulateTags(ClientProtocol::Message& msg) CXX11_OVERRIDE
63 const ClientProtocol::TagMap& tags = msg.GetTags();
64 if (tags.find("msgid") == tags.end())
65 msg.AddTag("msgid", this, generator.GetNext());
68 ModResult OnProcessTag(User* user, const std::string& tagname, std::string& tagvalue) CXX11_OVERRIDE
70 if (!irc::equals(tagname, "msgid"))
71 return MOD_RES_PASSTHRU;
73 // We should only allow this tag if it is added by a remote server.
74 return IS_LOCAL(user) ? MOD_RES_DENY : MOD_RES_ALLOW;
77 bool ShouldSendTag(LocalUser* user, const ClientProtocol::MessageTagData& tagdata) CXX11_OVERRIDE
79 return ctctagcap.get(user);
85 , public CTCTags::EventListener
90 ModResult CopyMessageId(const ClientProtocol::TagMap& tags_in, ClientProtocol::TagMap& tags_out)
92 ClientProtocol::TagMap::const_iterator iter = tags_in.find("msgid");
93 if (iter != tags_in.end())
95 // If the remote server has sent a message identifier we should use that as
96 // identifiers need to be the same on all sides of the network.
97 tags_out.insert(*iter);
99 return MOD_RES_PASSTHRU;
104 : CTCTags::EventListener(this)
109 ModResult OnUserPreMessage(User* user, const MessageTarget& target, MessageDetails& details) CXX11_OVERRIDE
111 return CopyMessageId(details.tags_in, details.tags_out);
114 ModResult OnUserPreTagMessage(User* user, const MessageTarget& target, CTCTags::TagMessageDetails& details) CXX11_OVERRIDE
116 return CopyMessageId(details.tags_in, details.tags_out);
119 Version GetVersion() CXX11_OVERRIDE
121 return Version("Provides the msgid IRCv3 tag", VF_VENDOR);
125 MODULE_INIT(ModuleMsgId)