src/modules/m_botmode.cpp

   1 /*
   2  * InspIRCd -- Internet Relay Chat Daemon
   3  *
   4  *   Copyright (C) 2004, 2008 Craig Edwards <craigedwards@brainbox.cc>
   5  *   Copyright (C) 2007 Dennis Friis <peavey@inspircd.org>
   6  *   Copyright (C) 2007 Robin Burchell <robin+git@viroteck.net>
   7  *
   8  * This file is part of InspIRCd.  InspIRCd is free software: you can
   9  * redistribute it and/or modify it under the terms of the GNU General Public
  10  * License as published by the Free Software Foundation, version 2.
  11  *
  12  * This program is distributed in the hope that it will be useful, but WITHOUT
  13  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
  14  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
  15  * details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19  */
  20
  21
  22 #include "inspircd.h"
  23 #include "modules/cap.h"
  24 #include "modules/whois.h"
  25
  26 enum
  27 {
  28         // From UnrealIRCd.
  29         RPL_WHOISBOT = 335
  30 };
  31
  32 class BotTag : public ClientProtocol::MessageTagProvider
  33 {
  34  private:
  35         SimpleUserModeHandler& botmode;
  36         Cap::Reference ctctagcap;
  37
  38  public:
  39         BotTag(Module* mod, SimpleUserModeHandler& bm)
  40                 : ClientProtocol::MessageTagProvider(mod)
  41                 , botmode(bm)
  42                 , ctctagcap(mod, "message-tags")
  43         {
  44         }
  45
  46         void OnPopulateTags(ClientProtocol::Message& msg) CXX11_OVERRIDE
  47         {
  48                 User* const user = msg.GetSourceUser();
  49                 if (user && user->IsModeSet(botmode))
  50                         msg.AddTag("inspircd.org/bot", this, "");
  51         }
  52
  53         bool ShouldSendTag(LocalUser* user, const ClientProtocol::MessageTagData& tagdata) CXX11_OVERRIDE
  54         {
  55                 return ctctagcap.get(user);
  56         }
  57 };
  58
  59 class ModuleBotMode : public Module, public Whois::EventListener
  60 {
  61  private:
  62         SimpleUserModeHandler bm;
  63         BotTag tag;
  64
  65  public:
  66         ModuleBotMode()
  67                 : Whois::EventListener(this)
  68                 , bm(this, "bot", 'B')
  69                 , tag(this, bm)
  70         {
  71         }
  72
  73         Version GetVersion() CXX11_OVERRIDE
  74         {
  75                 return Version("Provides user mode +B to mark the user as a bot",VF_VENDOR);
  76         }
  77
  78         void OnWhois(Whois::Context& whois) CXX11_OVERRIDE
  79         {
  80                 if (whois.GetTarget()->IsModeSet(bm))
  81                 {
  82                         whois.SendLine(RPL_WHOISBOT, "is a bot on " + ServerInstance->Config->Network);
  83                 }
  84         }
  85 };
  86
  87 MODULE_INIT(ModuleBotMode)