Change to Duration for second param

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

diff --git a/src/modules/m_lockserv.cpp b/src/modules/m_lockserv.cpp
index 0a19d56df911ccc91d5c0cd49f7e790771aa7a25..100b7864b70803a5521fac51ae2c1e69a0b5295b 100644 (file)
--- a/src/modules/m_lockserv.cpp
+++ b/src/modules/m_lockserv.cpp
@@ -25,19 +25,18 @@ class CommandLockserv : public Command
 {
        bool& locked;
 public:
-       CommandLockserv (InspIRCd* Instance, Module* Creator, bool& lock)
-               : Command(Instance, Creator, "LOCKSERV", "o", 0), locked(lock)
+       CommandLockserv(Module* Creator, bool& lock) : Command(Creator, "LOCKSERV", 0), locked(lock)
        {
-               syntax.clear();
+               flags_needed = 'o'; syntax.clear();
        }
 
        CmdResult Handle (const std::vector<std::string> &parameters, User *user)
        {
                locked = true;
-               user->WriteNumeric(988, "%s %s :Closed for new connections", user->nick.c_str(), user->server);
+               user->WriteNumeric(988, "%s %s :Closed for new connections", user->nick.c_str(), user->server.c_str());
                ServerInstance->SNO->WriteGlobalSno('a', "Oper %s used LOCKSERV to temporarily close for new connections", user->nick.c_str());
                /* Dont send to the network */
-               return CMD_LOCALONLY;
+               return CMD_SUCCESS;
        }
 };
 
@@ -47,19 +46,18 @@ private:
        bool& locked;
 
 public:
-       CommandUnlockserv (InspIRCd* Instance, Module* Creator, bool &lock)
-               : Command(Instance, Creator, "UNLOCKSERV", "o", 0), locked(lock)
+       CommandUnlockserv(Module* Creator, bool &lock) : Command(Creator, "UNLOCKSERV", 0), locked(lock)
        {
-               syntax.clear();
+               flags_needed = 'o'; syntax.clear();
        }
 
        CmdResult Handle (const std::vector<std::string> &parameters, User *user)
        {
                locked = false;
-               user->WriteNumeric(989, "%s %s :Open for new connections", user->nick.c_str(), user->server);
+               user->WriteNumeric(989, "%s %s :Open for new connections", user->nick.c_str(), user->server.c_str());
                ServerInstance->SNO->WriteGlobalSno('a', "Oper %s used UNLOCKSERV to allow for new connections", user->nick.c_str());
                /* Dont send to the network */
-               return CMD_LOCALONLY;
+               return CMD_SUCCESS;
        }
 };
 
@@ -76,7 +74,7 @@ private:
        }
 
 public:
-       ModuleLockserv(InspIRCd* Me) : Module(Me), lockcommand(Me, this, locked), unlockcommand(Me, this, locked)
+       ModuleLockserv() : lockcommand(this, locked), unlockcommand(this, locked)
        {
                ResetLocked();
                ServerInstance->AddCommand(&lockcommand);
@@ -95,24 +93,24 @@ public:
                ResetLocked();
        }
 
-       virtual int OnUserRegister(User* user)
+       virtual ModResult OnUserRegister(User* user)
        {
                if (locked)
                {
                        ServerInstance->Users->QuitUser(user, "Server is temporarily closed. Please try again later.");
-                       return 1;
+                       return MOD_RES_DENY;
                }
-               return 0;
+               return MOD_RES_PASSTHRU;
        }
 
-       virtual bool OnCheckReady(User* user)
+       virtual ModResult OnCheckReady(User* user)
        {
-               return !locked;
+               return locked ? MOD_RES_DENY : MOD_RES_PASSTHRU;
        }
 
        virtual Version GetVersion()
        {
-               return Version("$Id$", VF_VENDOR, API_VERSION);
+               return Version("Allows locking of the server to stop all incoming connections till unlocked again", VF_VENDOR);
        }
 };