there were two.. yes, you're right Special

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

diff --git a/src/modules/m_randquote.cpp b/src/modules/m_randquote.cpp
index 98160bda01725b11971a5283f944561d58fb3d2f..2ad7831ce90f0f5aba9e3a1b6d32c9b362b92f39 100644 (file)
--- a/src/modules/m_randquote.cpp
+++ b/src/modules/m_randquote.cpp
@@ -2,62 +2,87 @@
  *       | 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-2007 InspIRCd Development Team
+ * See: http://www.inspircd.org/wiki/index.php/Credits
+ *
  * This program is free but copyrighted software; see
  *            the file COPYING for details.
  *
  * ---------------------------------------------------
  */
 
-#include <stdio.h>
-#include <stdlib.h>
-#include <fstream>
+#include "inspircd.h"
 #include "users.h"
 #include "channels.h"
 #include "modules.h"
 
-Server *Srv = NULL;
-FileReader *quotes = NULL;
+static FileReader *quotes = NULL;
 
-std::string q_file = "";
-std::string prefix = "";
-std::string suffix = "";
+std::string q_file;
+std::string prefix;
+std::string suffix;
 
 /* $ModDesc: Provides random Quotes on Connect. */
 
-void handle_randquote(char** parameters, int pcntl, userrec *user)
+/** Handle /RANDQUOTE
+ */
+class cmd_randquote : public command_t
 {
-       std::string str;
-       int fsize;
-       char buf[MAXBUF];
-       if (quotes)
+ public:
+       cmd_randquote (InspIRCd* Instance) : command_t(Instance,"RANDQUOTE", 0, 0)
        {
-               fsize = quotes->FileSize();
-               str = quotes->GetLine(rand() % fsize);
-               sprintf(buf,"NOTICE %s :%s%s%s",user->nick,prefix.c_str(),str.c_str(),suffix.c_str());
-               Srv->SendServ(user->fd, buf);
+               this->source = "m_randquote.so";
        }
-       return;
-}
 
+       CmdResult Handle (const char** parameters, int pcntl, userrec *user)
+       {
+               std::string str;
+               int fsize;
 
+               if (q_file.empty() || quotes->Exists())
+               {
+                       fsize = quotes->FileSize();
+                       str = quotes->GetLine(rand() % fsize);
+                       user->WriteServ("NOTICE %s :%s%s%s",user->nick,prefix.c_str(),str.c_str(),suffix.c_str());
+               }
+               else
+               {
+                       user->WriteServ("NOTICE %s :Your administrator specified an invalid quotes file, please bug them about this.", user->nick);
+                       return CMD_FAILURE;
+               }
 
+               return CMD_LOCALONLY;
+       }
+};
 
-class ModuleRandQuote : public Module
+/** Thrown by m_randquote
+ */
+class RandquoteException : public ModuleException
 {
  private:
+       const std::string err;
+ public:
+       RandquoteException(const std::string &message) : err(message) { }
+
+       ~RandquoteException() throw () { }
+
+       virtual const char* GetReason()
+       {
+               return err.c_str();
+       }
+};
 
+class ModuleRandQuote : public Module
+{
+ private:
+       cmd_randquote* mycommand;
        ConfigReader *conf;
  public:
-       ModuleRandQuote()
+       ModuleRandQuote(InspIRCd* Me)
+               : Module(Me)
        {
-               Srv = new Server;
-               conf = new ConfigReader;
+               
+               conf = new ConfigReader(ServerInstance);
                // Sort the Randomizer thingie..
                srand(time(NULL));
 
@@ -65,63 +90,49 @@ class ModuleRandQuote : public Module
                prefix = conf->ReadValue("randquote","prefix",0);
                suffix = conf->ReadValue("randquote","suffix",0);
 
-               if (q_file == "") {
-                       log(DEFAULT,"m_randquote: Quotefile not specified - Please check your config.");
-                       return;
-                }
+               mycommand = NULL;
 
-               quotes = new FileReader(q_file);
+               if (q_file.empty())
+               {
+                       RandquoteException e("m_randquote: Quotefile not specified - Please check your config.");
+                       throw(e);
+               }
+
+               quotes = new FileReader(ServerInstance, q_file);
                if(!quotes->Exists())
                {
-                       log(DEFAULT,"m_randquote: QuoteFile not Found!! Please check your config - module will not function.");
-                       return;
+                       RandquoteException e("m_randquote: QuoteFile not Found!! Please check your config - module will not function.");
+                       throw(e);
+               }
+               else
+               {
+                       /* Hidden Command -- Mode clients assume /quote sends raw data to an IRCd >:D */
+                       mycommand = new cmd_randquote(ServerInstance);
+                       ServerInstance->AddCommand(mycommand);
                }
-               /* Hidden Command -- Mode clients assume /quote sends raw data to an IRCd >:D */
-               else Srv->AddCommand("QUOTE",handle_randquote,0,0,"m_randquote.so");
+       }
+
+       void Implements(char* List)
+       {
+               List[I_OnUserConnect] = 1;
        }
        
        virtual ~ModuleRandQuote()
        {
-               delete Srv;
-               delete conf;
-               delete quotes;
+               DELETE(conf);
+               DELETE(quotes);
        }
        
        virtual Version GetVersion()
        {
-               return Version(1,0,0,1,VF_VENDOR);
+               return Version(1,1,0,1,VF_VENDOR,API_VERSION);
        }
        
        virtual void OnUserConnect(userrec* user)
        {
-               // Make a fake pointer to be passed to handle_randquote()
-               // Dont try this at home kiddies :D
-               char *rar = "RAR";
-               handle_randquote(&rar, 0, user);
-       }
-};
-
-
-class ModuleRandQuoteFactory : public ModuleFactory
-{
- public:
-       ModuleRandQuoteFactory()
-       {
-       }
-       
-       ~ModuleRandQuoteFactory()
-       {
+               if (mycommand)
+                       mycommand->Handle(NULL, 0, user);
        }
-       
-       virtual Module * CreateModule()
-       {
-               return new ModuleRandQuote;
-       }
-       
 };
 
-
-extern "C" void * init_module( void )
-{
-       return new ModuleRandQuoteFactory;
-}
+MODULE_INIT(ModuleRandQuote)