src/modules/m_ircv3.cpp

   1 /*
   2  * InspIRCd -- Internet Relay Chat Daemon
   3  *
   4  *   Copyright (C) 2012 Attila Molnar <attilamolnar@hush.com>
   5  *
   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.
   9  *
  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
  13  * details.
  14  *
  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/>.
  17  */
  18
  19 #include "inspircd.h"
  20 #include "modules/account.h"
  21 #include "modules/cap.h"
  22
  23 class WriteNeighboursWithExt : public User::ForEachNeighborHandler
  24 {
  25         const LocalIntExt& ext;
  26         const std::string& msg;
  27
  28         void Execute(LocalUser* user) CXX11_OVERRIDE
  29         {
  30                 if (ext.get(user))
  31                         user->Write(msg);
  32         }
  33
  34  public:
  35         WriteNeighboursWithExt(User* user, const std::string& message, const LocalIntExt& extension)
  36                 : ext(extension)
  37                 , msg(message)
  38         {
  39                 user->ForEachNeighbor(*this, false);
  40         }
  41 };
  42
  43 class ModuleIRCv3 : public Module, public AccountEventListener
  44 {
  45         GenericCap cap_accountnotify;
  46         GenericCap cap_awaynotify;
  47         GenericCap cap_extendedjoin;
  48
  49         CUList last_excepts;
  50
  51  public:
  52         ModuleIRCv3()
  53                 : AccountEventListener(this)
  54                 , cap_accountnotify(this, "account-notify"),
  55                                         cap_awaynotify(this, "away-notify"),
  56                                         cap_extendedjoin(this, "extended-join")
  57         {
  58         }
  59
  60         void ReadConfig(ConfigStatus& status) CXX11_OVERRIDE
  61         {
  62                 ConfigTag* conf = ServerInstance->Config->ConfValue("ircv3");
  63                 cap_accountnotify.SetActive(conf->getBool("accountnotify", true));
  64                 cap_awaynotify.SetActive(conf->getBool("awaynotify", true));
  65                 cap_extendedjoin.SetActive(conf->getBool("extendedjoin", true));
  66         }
  67
  68         void OnAccountChange(User* user, const std::string& newaccount) CXX11_OVERRIDE
  69         {
  70                 // :nick!user@host ACCOUNT account
  71                 // or
  72                 // :nick!user@host ACCOUNT *
  73                 std::string line = ":" + user->GetFullHost() + " ACCOUNT ";
  74                 if (newaccount.empty())
  75                         line += "*";
  76                 else
  77                         line += newaccount;
  78
  79                 WriteNeighboursWithExt(user, line, cap_accountnotify.ext);
  80         }
  81
  82         void OnUserJoin(Membership* memb, bool sync, bool created, CUList& excepts) CXX11_OVERRIDE
  83         {
  84                 // Remember who is not going to see the JOIN because of other modules
  85                 if ((cap_awaynotify.IsActive()) && (memb->user->IsAway()))
  86                         last_excepts = excepts;
  87
  88                 if (!cap_extendedjoin.IsActive())
  89                         return;
  90
  91                 /*
  92                  * Send extended joins to clients who have the extended-join capability.
  93                  * An extended join looks like this:
  94                  *
  95                  * :nick!user@host JOIN #chan account :realname
  96                  *
  97                  * account is the joining user's account if he's logged in, otherwise it's an asterisk (*).
  98                  */
  99
 100                 std::string line;
 101                 std::string mode;
 102
 103                 const Channel::MemberMap& userlist = memb->chan->GetUsers();
 104                 for (Channel::MemberMap::const_iterator it = userlist.begin(); it != userlist.end(); ++it)
 105                 {
 106                         // Send the extended join line if the current member is local, has the extended-join cap and isn't excepted
 107                         User* member = IS_LOCAL(it->first);
 108                         if ((member) && (cap_extendedjoin.ext.get(member)) && (excepts.find(member) == excepts.end()))
 109                         {
 110                                 // Construct the lines we're going to send if we haven't constructed them already
 111                                 if (line.empty())
 112                                 {
 113                                         bool has_account = false;
 114                                         line = ":" + memb->user->GetFullHost() + " JOIN " + memb->chan->name + " ";
 115                                         const AccountExtItem* accountext = GetAccountExtItem();
 116                                         if (accountext)
 117                                         {
 118                                                 std::string* accountname;
 119                                                 accountname = accountext->get(memb->user);
 120                                                 if (accountname)
 121                                                 {
 122                                                         line += *accountname;
 123                                                         has_account = true;
 124                                                 }
 125                                         }
 126
 127                                         if (!has_account)
 128                                                 line += "*";
 129
 130                                         line += " :" + memb->user->fullname;
 131
 132                                         // If the joining user received privileges from another module then we must send them as well,
 133                                         // since silencing the normal join means the MODE will be silenced as well
 134                                         if (!memb->modes.empty())
 135                                         {
 136                                                 const std::string& modefrom = ServerInstance->Config->CycleHostsFromUser ? memb->user->GetFullHost() : ServerInstance->Config->ServerName;
 137                                                 mode = ":" + modefrom + " MODE " + memb->chan->name + " +" + memb->modes;
 138
 139                                                 for (unsigned int i = 0; i < memb->modes.length(); i++)
 140                                                         mode += " " + memb->user->nick;
 141                                         }
 142                                 }
 143
 144                                 // Write the JOIN and the MODE, if any
 145                                 member->Write(line);
 146                                 if ((!mode.empty()) && (member != memb->user))
 147                                         member->Write(mode);
 148
 149                                 // Prevent the core from sending the JOIN and MODE to this user
 150                                 excepts.insert(it->first);
 151                         }
 152                 }
 153         }
 154
 155         ModResult OnSetAway(User* user, const std::string &awaymsg) CXX11_OVERRIDE
 156         {
 157                 if (cap_awaynotify.IsActive())
 158                 {
 159                         // Going away: n!u@h AWAY :reason
 160                         // Back from away: n!u@h AWAY
 161                         std::string line = ":" + user->GetFullHost() + " AWAY";
 162                         if (!awaymsg.empty())
 163                                 line += " :" + awaymsg;
 164
 165                         WriteNeighboursWithExt(user, line, cap_awaynotify.ext);
 166                 }
 167                 return MOD_RES_PASSTHRU;
 168         }
 169
 170         void OnPostJoin(Membership *memb) CXX11_OVERRIDE
 171         {
 172                 if ((!cap_awaynotify.IsActive()) || (!memb->user->IsAway()))
 173                         return;
 174
 175                 std::string line = ":" + memb->user->GetFullHost() + " AWAY :" + memb->user->awaymsg;
 176
 177                 const Channel::MemberMap& userlist = memb->chan->GetUsers();
 178                 for (Channel::MemberMap::const_iterator it = userlist.begin(); it != userlist.end(); ++it)
 179                 {
 180                         // Send the away notify line if the current member is local, has the away-notify cap and isn't excepted
 181                         User* member = IS_LOCAL(it->first);
 182                         if ((member) && (cap_awaynotify.ext.get(member)) && (last_excepts.find(member) == last_excepts.end()))
 183                         {
 184                                 member->Write(line);
 185                         }
 186                 }
 187
 188                 last_excepts.clear();
 189         }
 190
 191         void Prioritize()
 192         {
 193                 ServerInstance->Modules->SetPriority(this, I_OnUserJoin, PRIORITY_LAST);
 194         }
 195
 196         Version GetVersion() CXX11_OVERRIDE
 197         {
 198                 return Version("Provides support for extended-join, away-notify and account-notify CAP capabilities", VF_VENDOR);
 199         }
 200 };
 201
 202 MODULE_INIT(ModuleIRCv3)