Add Module* creator to Command and ModeHandler

[user/henk/code/inspircd.git] / src / modules / m_sethost.cpp

diff --git a/src/modules/m_sethost.cpp b/src/modules/m_sethost.cpp
index 0894e8f24c3588c2aaab5e8090a6127524281335..8442167cfb9c4ba6c785c56fca1d3f3f25a9a5ce 100644 (file)
--- a/src/modules/m_sethost.cpp
+++ b/src/modules/m_sethost.cpp
@@ -2,104 +2,102 @@
  *       | Inspire Internet Relay Chat Daemon |
  *       +------------------------------------+
  *
- *  Inspire is copyright (C) 2002-2004 ChatSpike-Dev.
- *                       E-mail:
- *                <brain@chatspike.net>
- *               <Craig@chatspike.net>
- *     
- * Written by Craig Edwards, Craig McLure, and others.
+ *  InspIRCd: (C) 2002-2009 InspIRCd Development Team
+ * See: http://wiki.inspircd.org/Credits
+ *
  * This program is free but copyrighted software; see
  *            the file COPYING for details.
  *
  * ---------------------------------------------------
  */
 
-using namespace std;
-
-#include <stdio.h>
-#include <string>
-#include "users.h"
-#include "channels.h"
-#include "modules.h"
-#include "helperfuncs.h"
+#include "inspircd.h"
 
 /* $ModDesc: Provides support for the SETHOST command */
 
-Server *Srv;
-
-class cmd_sethost : public command_t
+/** Handle /SETHOST
+ */
+class CommandSethost : public Command
 {
+ private:
+       char* hostmap;
  public:
-       cmd_sethost() : command_t("SETHOST",'o',1)
+       CommandSethost (InspIRCd* Instance, Module* Creator, char* hmap) : Command(Instance,Creator,"SETHOST","o",1), hostmap(hmap)
        {
-               this->source = "m_sethost.so";
+               syntax = "<new-hostname>";
+               TRANSLATE2(TR_TEXT, TR_END);
        }
 
-       void Handle (char **parameters, int pcnt, userrec *user)
+       CmdResult Handle (const std::vector<std::string>& parameters, User *user)
        {
-               for (unsigned int x = 0; x < strlen(parameters[0]); x++)
+               size_t len = 0;
+               for (std::string::const_iterator x = parameters[0].begin(); x != parameters[0].end(); x++, len++)
                {
-                       if (((tolower(parameters[0][x]) < 'a') || (tolower(parameters[0][x]) > 'z')) && (parameters[0][x] != '.'))
+                       if (!hostmap[(const unsigned char)*x])
                        {
-                               if (((parameters[0][x] < '0') || (parameters[0][x]> '9')) && (parameters[0][x] != '-'))
-                               {
-                                       Srv->SendTo(NULL,user,"NOTICE "+std::string(user->nick)+" :*** Invalid characters in hostname");
-                                       return;
-                               }
+                               user->WriteServ("NOTICE "+std::string(user->nick)+" :*** SETHOST: Invalid characters in hostname");
+                               return CMD_FAILURE;
                        }
                }
-               Srv->ChangeHost(user,parameters[0]);
-               Srv->SendOpers(std::string(user->nick)+" used SETHOST to change their displayed host to "+std::string(parameters[0]));
+               if (len == 0)
+               {
+                       user->WriteServ("NOTICE %s :*** SETHOST: Host must be specified", user->nick.c_str());
+                       return CMD_FAILURE;
+               }
+               if (len > 64)
+               {
+                       user->WriteServ("NOTICE %s :*** SETHOST: Host too long",user->nick.c_str());
+                       return CMD_FAILURE;
+               }
+
+               if (user->ChangeDisplayedHost(parameters[0].c_str()))
+               {
+                       ServerInstance->SNO->WriteGlobalSno('a', std::string(user->nick)+" used SETHOST to change their displayed host to "+user->dhost);
+                       return CMD_LOCALONLY;
+               }
+
+               return CMD_FAILURE;
        }
 };
 
 
 class ModuleSetHost : public Module
 {
-       cmd_sethost*    mycommand;
+       CommandSethost cmd;
+       char hostmap[256];
  public:
-       ModuleSetHost(Server* Me)
-               : Module::Module(Me)
+       ModuleSetHost(InspIRCd* Me)
+               : Module(Me), cmd(Me, this, hostmap)
        {
-               Srv = Me;
-               mycommand = new cmd_sethost();
-               Srv->AddCommand(mycommand);
+               OnRehash(NULL);
+               ServerInstance->AddCommand(&cmd);
+               Implementation eventlist[] = { I_OnRehash };
+               ServerInstance->Modules->Attach(eventlist, this, 1);
        }
-       
-       virtual ~ModuleSetHost()
-       {
-       }
-       
-       virtual Version GetVersion()
-       {
-               return Version(1,0,0,1,VF_VENDOR);
-       }
-       
-};
 
-// stuff down here is the module-factory stuff. For basic modules you can ignore this.
 
-class ModuleSetHostFactory : public ModuleFactory
-{
- public:
-       ModuleSetHostFactory()
+       void OnRehash(User* user)
        {
+               ConfigReader Conf(ServerInstance);
+               std::string hmap = Conf.ReadValue("hostname", "charmap", 0);
+
+               if (hmap.empty())
+                       hmap = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz.-_/0123456789";
+
+               memset(hostmap, 0, sizeof(hostmap));
+               for (std::string::iterator n = hmap.begin(); n != hmap.end(); n++)
+                       hostmap[(unsigned char)*n] = 1;
        }
-       
-       ~ModuleSetHostFactory()
+
+       virtual ~ModuleSetHost()
        {
        }
-       
-       virtual Module * CreateModule(Server* Me)
+
+       virtual Version GetVersion()
        {
-               return new ModuleSetHost(Me);
+               return Version("$Id$", VF_VENDOR, API_VERSION);
        }
-       
-};
 
+};
 
-extern "C" void * init_module( void )
-{
-       return new ModuleSetHostFactory;
-}
-
+MODULE_INIT(ModuleSetHost)