2 * InspIRCd -- Internet Relay Chat Daemon
4 * Copyright (C) 2019-2020 Sadie Powell <sadie@witchery.services>
6 * This file is part of InspIRCd. InspIRCd is free software: you can
7 * redistribute it and/or modify it under the terms of the GNU General Public
8 * License as published by the Free Software Foundation, version 2.
10 * This program is distributed in the hope that it will be useful, but WITHOUT
11 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
12 * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
15 * You should have received a copy of the GNU General Public License
16 * along with this program. If not, see <http://www.gnu.org/licenses/>.
21 #include "modules/ctctags.h"
23 class MsgIdTag : public ClientProtocol::MessageTagProvider
26 CTCTags::CapReference ctctagcap;
30 : ClientProtocol::MessageTagProvider(mod)
35 ModResult OnProcessTag(User* user, const std::string& tagname, std::string& tagvalue) CXX11_OVERRIDE
37 if (!irc::equals(tagname, "msgid"))
38 return MOD_RES_PASSTHRU;
40 // We should only allow this tag if it is added by a remote server.
41 return IS_LOCAL(user) ? MOD_RES_DENY : MOD_RES_ALLOW;
44 bool ShouldSendTag(LocalUser* user, const ClientProtocol::MessageTagData& tagdata) CXX11_OVERRIDE
46 return ctctagcap.get(user);
54 const std::string::size_type baselen;
59 , strid(InspIRCd::Format("%s~%lu~", ServerInstance->Config->GetSID().c_str(), ServerInstance->startup_time))
60 , baselen(strid.length())
64 const std::string& GetNext()
67 strid.append(ConvToStr(counter++));
74 , public CTCTags::EventListener
78 MsgIdGenerator generator;
80 ModResult CopyMessageId(const ClientProtocol::TagMap& tags_in, ClientProtocol::TagMap& tags_out)
82 ClientProtocol::TagMap::const_iterator iter = tags_in.find("msgid");
83 if (iter != tags_in.end())
85 // If the remote server has sent a message identifier we should use that as
86 // identifiers need to be the same on all sides of the network.
87 tags_out.insert(*iter);
88 return MOD_RES_PASSTHRU;
91 // Otherwise, we can just create a new message identifier.
92 tags_out.insert(std::make_pair("msgid", ClientProtocol::MessageTagData(&tag, generator.GetNext())));
93 return MOD_RES_PASSTHRU;
98 : CTCTags::EventListener(this)
103 ModResult OnUserPreMessage(User* user, const MessageTarget& target, MessageDetails& details) CXX11_OVERRIDE
105 return CopyMessageId(details.tags_in, details.tags_out);
108 ModResult OnUserPreTagMessage(User* user, const MessageTarget& target, CTCTags::TagMessageDetails& details) CXX11_OVERRIDE
110 return CopyMessageId(details.tags_in, details.tags_out);
113 Version GetVersion() CXX11_OVERRIDE
115 return Version("Provides support for the IRCv3 Message IDs specification.", VF_VENDOR);
119 MODULE_INIT(ModuleMsgId)