src/modules/m_abbreviation.cpp

   1 /*       +------------------------------------+
   2  *       | Inspire Internet Relay Chat Daemon |
   3  *       +------------------------------------+
   4  *
   5  *  InspIRCd: (C) 2002-2009 InspIRCd Development Team
   6  * See: http://wiki.inspircd.org/Credits
   7  *
   8  * This program is free but copyrighted software; see
   9  *            the file COPYING for details.
  10  *
  11  * ---------------------------------------------------
  12  */
  13
  14 #include "inspircd.h"
  15
  16 /* $ModDesc: Provides the ability to abbreviate commands a-la BBC BASIC keywords. */
  17
  18 class ModuleAbbreviation : public Module
  19 {
  20
  21  public:
  22
  23         ModuleAbbreviation(InspIRCd* Me)
  24                 : Module(Me)
  25         {
  26                 Me->Modules->Attach(I_OnPreCommand, this);
  27                 /* Must do this first */
  28                 Me->Modules->SetPriority(this, I_OnPreCommand, PRIORITY_FIRST);
  29         }
  30
  31         virtual Version GetVersion()
  32         {
  33                 return Version("Provides the ability to abbreviate commands a-la BBC BASIC keywords.",VF_VENDOR,API_VERSION);
  34         }
  35
  36         virtual ModResult OnPreCommand(std::string &command, std::vector<std::string> &parameters, User *user, bool validated, const std::string &original_line)
  37         {
  38                 /* Command is already validated, has a length of 0, or last character is not a . */
  39                 if (validated || command.empty() || *command.rbegin() != '.')
  40                         return MOD_RES_PASSTHRU;
  41
  42                 /* Whack the . off the end */
  43                 command.erase(command.end() - 1);
  44
  45                 /* Look for any command that starts with the same characters, if it does, replace the command string with it */
  46                 size_t clen = command.length();
  47                 std::string foundcommand, matchlist;
  48                 bool foundmatch = false;
  49                 for (Commandtable::iterator n = ServerInstance->Parser->cmdlist.begin(); n != ServerInstance->Parser->cmdlist.end(); ++n)
  50                 {
  51                         if (n->first.length() < clen)
  52                                 continue;
  53
  54                         if (command == n->first.substr(0, clen))
  55                         {
  56                                 if (matchlist.length() > 450)
  57                                 {
  58                                         user->WriteNumeric(420, "%s :Ambiguous abbreviation and too many possible matches.", user->nick.c_str());
  59                                         return MOD_RES_DENY;
  60                                 }
  61
  62                                 if (!foundmatch)
  63                                 {
  64                                         /* Found the command */
  65                                         foundcommand = n->first;
  66                                         foundmatch = true;
  67                                 }
  68                                 else
  69                                         matchlist.append(" ").append(n->first);
  70                         }
  71                 }
  72
  73                 /* Ambiguous command, list the matches */
  74                 if (!matchlist.empty())
  75                 {
  76                         user->WriteNumeric(420, "%s :Ambiguous abbreviation, posssible matches: %s%s", user->nick.c_str(), foundcommand.c_str(), matchlist.c_str());
  77                         return MOD_RES_DENY;
  78                 }
  79
  80                 if (foundcommand.empty())
  81                 {
  82                         /* No match, we have to put the . back again so that the invalid command numeric looks correct. */
  83                         command += '.';
  84                 }
  85                 else
  86                 {
  87                         command = foundcommand;
  88                 }
  89
  90                 return MOD_RES_PASSTHRU;
  91         }
  92 };
  93
  94 MODULE_INIT(ModuleAbbreviation)