src/modules/m_namesx.cpp

   1 /*
   2  * InspIRCd -- Internet Relay Chat Daemon
   3  *
   4  *   Copyright (C) 2019 linuxdaemon <linuxdaemon.irc@gmail.com>
   5  *   Copyright (C) 2013, 2018-2020 Sadie Powell <sadie@witchery.services>
   6  *   Copyright (C) 2012-2016 Attila Molnar <attilamolnar@hush.com>
   7  *   Copyright (C) 2012, 2019 Robby <robby@chatbelgie.be>
   8  *   Copyright (C) 2009-2010 Daniel De Graaf <danieldg@inspircd.org>
   9  *   Copyright (C) 2007 Dennis Friis <peavey@inspircd.org>
  10  *   Copyright (C) 2006, 2008 Craig Edwards <brain@inspircd.org>
  11  *
  12  * This file is part of InspIRCd.  InspIRCd is free software: you can
  13  * redistribute it and/or modify it under the terms of the GNU General Public
  14  * License as published by the Free Software Foundation, version 2.
  15  *
  16  * This program is distributed in the hope that it will be useful, but WITHOUT
  17  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
  18  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
  19  * details.
  20  *
  21  * You should have received a copy of the GNU General Public License
  22  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  23  */
  24
  25
  26 #include "inspircd.h"
  27 #include "modules/cap.h"
  28 #include "modules/names.h"
  29 #include "modules/who.h"
  30
  31 class ModuleNamesX
  32         : public Module
  33         , public Names::EventListener
  34         , public Who::EventListener
  35 {
  36  private:
  37         Cap::Capability cap;
  38
  39  public:
  40         ModuleNamesX()
  41                 : Names::EventListener(this)
  42                 , Who::EventListener(this)
  43                 , cap(this, "multi-prefix")
  44         {
  45         }
  46
  47         Version GetVersion() CXX11_OVERRIDE
  48         {
  49                 return Version("Provides the IRCv3 multi-prefix client capability.", VF_VENDOR);
  50         }
  51
  52         void On005Numeric(std::map<std::string, std::string>& tokens) CXX11_OVERRIDE
  53         {
  54                 // The legacy PROTOCTL system is a wrapper around the cap.
  55                 dynamic_reference_nocheck<Cap::Manager> capmanager(this, "capmanager");
  56                 if (capmanager)
  57                         tokens["NAMESX"];
  58         }
  59
  60         ModResult OnPreCommand(std::string& command, CommandBase::Params& parameters, LocalUser* user, bool validated) CXX11_OVERRIDE
  61         {
  62                 /* We don't actually create a proper command handler class for PROTOCTL,
  63                  * because other modules might want to have PROTOCTL hooks too.
  64                  * Therefore, we just hook its as an unvalidated command therefore we
  65                  * can capture it even if it doesnt exist! :-)
  66                  */
  67                 if (command == "PROTOCTL")
  68                 {
  69                         if (!parameters.empty() && irc::equals(parameters[0], "NAMESX"))
  70                         {
  71                                 cap.set(user, true);
  72                                 return MOD_RES_DENY;
  73                         }
  74                 }
  75                 return MOD_RES_PASSTHRU;
  76         }
  77
  78         ModResult OnNamesListItem(LocalUser* issuer, Membership* memb, std::string& prefixes, std::string& nick) CXX11_OVERRIDE
  79         {
  80                 if (cap.get(issuer))
  81                         prefixes = memb->GetAllPrefixChars();
  82
  83                 return MOD_RES_PASSTHRU;
  84         }
  85
  86         ModResult OnWhoLine(const Who::Request& request, LocalUser* source, User* user, Membership* memb, Numeric::Numeric& numeric) CXX11_OVERRIDE
  87         {
  88                 if ((!memb) || (!cap.get(source)))
  89                         return MOD_RES_PASSTHRU;
  90
  91                 // Don't do anything if the user has only one prefix
  92                 std::string prefixes = memb->GetAllPrefixChars();
  93                 if (prefixes.length() <= 1)
  94                         return MOD_RES_PASSTHRU;
  95
  96                 size_t flag_index;
  97                 if (!request.GetFieldIndex('f', flag_index))
  98                         return MOD_RES_PASSTHRU;
  99
 100                 // #chan ident localhost insp22.test nick H@ :0 Attila
 101                 if (numeric.GetParams().size() <= flag_index)
 102                         return MOD_RES_PASSTHRU;
 103
 104                 numeric.GetParams()[flag_index].append(prefixes, 1, std::string::npos);
 105                 return MOD_RES_PASSTHRU;
 106         }
 107 };
 108
 109 MODULE_INIT(ModuleNamesX)