src/modules/m_showfile.cpp

   1 /*
   2  * InspIRCd -- Internet Relay Chat Daemon
   3  *
   4  *   Copyright (C) 2013 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
  20 #include "inspircd.h"
  21
  22 enum
  23 {
  24         // From UnrealIRCd.
  25         RPL_RULES = 232,
  26         RPL_RULESTART = 308,
  27         RPL_RULESEND = 309,
  28         ERR_NORULES = 434
  29 };
  30
  31 class CommandShowFile : public Command
  32 {
  33         enum Method
  34         {
  35                 SF_MSG,
  36                 SF_NOTICE,
  37                 SF_NUMERIC
  38         };
  39
  40         std::string introtext;
  41         std::string endtext;
  42         unsigned int intronumeric;
  43         unsigned int textnumeric;
  44         unsigned int endnumeric;
  45         file_cache contents;
  46         Method method;
  47
  48  public:
  49         CommandShowFile(Module* parent, const std::string& cmdname)
  50                 : Command(parent, cmdname)
  51         {
  52         }
  53
  54         CmdResult Handle(User* user, const Params& parameters) CXX11_OVERRIDE
  55         {
  56                 if (method == SF_NUMERIC)
  57                 {
  58                         if (!introtext.empty())
  59                                 user->WriteRemoteNumeric(intronumeric, introtext);
  60
  61                         for (file_cache::const_iterator i = contents.begin(); i != contents.end(); ++i)
  62                                 user->WriteRemoteNumeric(textnumeric, InspIRCd::Format("- %s", i->c_str()));
  63
  64                         user->WriteRemoteNumeric(endnumeric, endtext.c_str());
  65                 }
  66                 else if (IS_LOCAL(user))
  67                 {
  68                         LocalUser* const localuser = IS_LOCAL(user);
  69                         for (file_cache::const_iterator i = contents.begin(); i != contents.end(); ++i)
  70                         {
  71                                 const std::string& line = *i;
  72                                 ClientProtocol::Messages::Privmsg msg(ClientProtocol::Messages::Privmsg::nocopy, ServerInstance->FakeClient, localuser, line, ((method == SF_MSG) ? MSG_PRIVMSG : MSG_NOTICE));
  73                                 localuser->Send(ServerInstance->GetRFCEvents().privmsg, msg);
  74                         }
  75                 }
  76                 return CMD_SUCCESS;
  77         }
  78
  79         void UpdateSettings(ConfigTag* tag, const std::vector<std::string>& filecontents)
  80         {
  81                 introtext = tag->getString("introtext", "Showing " + name);
  82                 endtext = tag->getString("endtext", "End of " + name);
  83                 intronumeric = tag->getUInt("intronumeric", RPL_RULESTART, 0, 999);
  84                 textnumeric = tag->getUInt("numeric", RPL_RULES, 0, 999);
  85                 endnumeric = tag->getUInt("endnumeric", RPL_RULESEND, 0, 999);
  86                 std::string smethod = tag->getString("method");
  87
  88                 method = SF_NUMERIC;
  89                 if (smethod == "msg")
  90                         method = SF_MSG;
  91                 else if (smethod == "notice")
  92                         method = SF_NOTICE;
  93
  94                 contents = filecontents;
  95                 InspIRCd::ProcessColors(contents);
  96         }
  97 };
  98
  99 class ModuleShowFile : public Module
 100 {
 101         std::vector<CommandShowFile*> cmds;
 102
 103         void ReadTag(ConfigTag* tag, std::vector<CommandShowFile*>& newcmds)
 104         {
 105                 std::string cmdname = tag->getString("name");
 106                 if (cmdname.empty())
 107                         throw ModuleException("Empty value for 'name'");
 108
 109                 std::transform(cmdname.begin(), cmdname.end(), cmdname.begin(), ::toupper);
 110
 111                 const std::string file = tag->getString("file", cmdname);
 112                 if (file.empty())
 113                         throw ModuleException("Empty value for 'file'");
 114                 FileReader reader(file);
 115
 116                 CommandShowFile* sfcmd;
 117                 Command* handler = ServerInstance->Parser.GetHandler(cmdname);
 118                 if (handler)
 119                 {
 120                         // Command exists, check if it is ours
 121                         if (handler->creator != this)
 122                                 throw ModuleException("Command " + cmdname + " already exists");
 123
 124                         // This is our command, make sure we don't have the same entry twice
 125                         sfcmd = static_cast<CommandShowFile*>(handler);
 126                         if (stdalgo::isin(newcmds, sfcmd))
 127                                 throw ModuleException("Command " + cmdname + " is already used in a <showfile> tag");
 128                 }
 129                 else
 130                 {
 131                         // Command doesn't exist, create it
 132                         sfcmd = new CommandShowFile(this, cmdname);
 133                         ServerInstance->Modules->AddService(*sfcmd);
 134                 }
 135
 136                 sfcmd->UpdateSettings(tag, reader.GetVector());
 137                 newcmds.push_back(sfcmd);
 138         }
 139
 140  public:
 141         void ReadConfig(ConfigStatus& status) CXX11_OVERRIDE
 142         {
 143                 std::vector<CommandShowFile*> newcmds;
 144
 145                 ConfigTagList tags = ServerInstance->Config->ConfTags("showfile");
 146                 for (ConfigIter i = tags.first; i != tags.second; ++i)
 147                 {
 148                         ConfigTag* tag = i->second;
 149                         try
 150                         {
 151                                 ReadTag(tag, newcmds);
 152                         }
 153                         catch (CoreException& ex)
 154                         {
 155                                 ServerInstance->Logs->Log(MODNAME, LOG_DEFAULT, "Error: " + ex.GetReason() + " at " + tag->getTagLocation());
 156                         }
 157                 }
 158
 159                 // Remove all commands that were removed from the config
 160                 std::vector<CommandShowFile*> removed(cmds.size());
 161                 std::sort(newcmds.begin(), newcmds.end());
 162                 std::set_difference(cmds.begin(), cmds.end(), newcmds.begin(), newcmds.end(), removed.begin());
 163
 164                 stdalgo::delete_all(removed);
 165                 cmds.swap(newcmds);
 166         }
 167
 168         ~ModuleShowFile()
 169         {
 170                 stdalgo::delete_all(cmds);
 171         }
 172
 173         Version GetVersion() CXX11_OVERRIDE
 174         {
 175                 return Version("Provides support for showing text files to users", VF_VENDOR);
 176         }
 177 };
 178
 179 MODULE_INIT(ModuleShowFile)