src/modules/m_randquote.cpp

   1 /*       +------------------------------------+
   2  *       | Inspire Internet Relay Chat Daemon |
   3  *       +------------------------------------+
   4  *
   5  *  InspIRCd is copyright (C) 2002-2006 ChatSpike-Dev.
   6  *                       E-mail:
   7  *                <brain@chatspike.net>
   8  *                <Craig@chatspike.net>
   9  *
  10  * Written by Craig Edwards, Craig McLure, and others.
  11  * This program is free but copyrighted software; see
  12  *            the file COPYING for details.
  13  *
  14  * ---------------------------------------------------
  15  */
  16
  17 using namespace std;
  18
  19 #include "users.h"
  20 #include "channels.h"
  21 #include "modules.h"
  22
  23 #include "inspircd.h"
  24
  25
  26 static FileReader *quotes = NULL;
  27
  28 std::string q_file = "";
  29 std::string prefix = "";
  30 std::string suffix = "";
  31
  32 /* $ModDesc: Provides random Quotes on Connect. */
  33
  34 class cmd_randquote : public command_t
  35 {
  36  public:
  37  cmd_randquote (InspIRCd* Instance) : command_t(Instance,"RANDQUOTE", 0, 0)
  38         {
  39                 this->source = "m_randquote.so";
  40         }
  41
  42         CmdResult Handle (const char** parameters, int pcntl, userrec *user)
  43         {
  44                 std::string str;
  45                 int fsize;
  46
  47                 if (q_file == "" || quotes->Exists())
  48                 {
  49                         fsize = quotes->FileSize();
  50                         str = quotes->GetLine(rand() % fsize);
  51                         user->WriteServ("NOTICE %s :%s%s%s",user->nick,prefix.c_str(),str.c_str(),suffix.c_str());
  52                 }
  53                 else
  54                 {
  55                         user->WriteServ("NOTICE %s :Your administrator specified an invalid quotes file, please bug them about this.", user->nick);
  56                         return CMD_FAILURE;
  57                 }
  58                 return CMD_SUCCESS;
  59         }
  60 };
  61
  62 class RandquoteException : public ModuleException
  63 {
  64  private:
  65         std::string err;
  66  public:
  67         RandquoteException(std::string message) : err(message) { }
  68
  69         virtual const char* GetReason()
  70         {
  71                 return (char*)err.c_str();
  72         }
  73 };
  74
  75 class ModuleRandQuote : public Module
  76 {
  77  private:
  78         cmd_randquote* mycommand;
  79         ConfigReader *conf;
  80  public:
  81         ModuleRandQuote(InspIRCd* Me)
  82                 : Module::Module(Me)
  83         {
  84
  85                 conf = new ConfigReader(ServerInstance);
  86                 // Sort the Randomizer thingie..
  87                 srand(time(NULL));
  88
  89                 q_file = conf->ReadValue("randquote","file",0);
  90                 prefix = conf->ReadValue("randquote","prefix",0);
  91                 suffix = conf->ReadValue("randquote","suffix",0);
  92
  93                 mycommand = NULL;
  94
  95                 if (q_file == "")
  96                 {
  97                         RandquoteException e("m_randquote: Quotefile not specified - Please check your config.");
  98                         throw(e);
  99                 }
 100
 101                 quotes = new FileReader(ServerInstance, q_file);
 102                 if(!quotes->Exists())
 103                 {
 104                         RandquoteException e("m_randquote: QuoteFile not Found!! Please check your config - module will not function.");
 105                         throw(e);
 106                 }
 107                 else
 108                 {
 109                         /* Hidden Command -- Mode clients assume /quote sends raw data to an IRCd >:D */
 110                         mycommand = new cmd_randquote(ServerInstance);
 111                         ServerInstance->AddCommand(mycommand);
 112                 }
 113         }
 114
 115         void Implements(char* List)
 116         {
 117                 List[I_OnUserConnect] = 1;
 118         }
 119
 120         virtual ~ModuleRandQuote()
 121         {
 122                 DELETE(conf);
 123                 DELETE(quotes);
 124         }
 125
 126         virtual Version GetVersion()
 127         {
 128                 return Version(1,0,0,1,VF_VENDOR);
 129         }
 130
 131         virtual void OnUserConnect(userrec* user)
 132         {
 133                 if (mycommand)
 134                         mycommand->Handle(NULL, 0, user);
 135         }
 136 };
 137
 138
 139 class ModuleRandQuoteFactory : public ModuleFactory
 140 {
 141  public:
 142         ModuleRandQuoteFactory()
 143         {
 144         }
 145
 146         ~ModuleRandQuoteFactory()
 147         {
 148         }
 149
 150         virtual Module * CreateModule(InspIRCd* Me)
 151         {
 152                 return new ModuleRandQuote(Me);
 153         }
 154
 155 };
 156
 157
 158 extern "C" void * init_module( void )
 159 {
 160         return new ModuleRandQuoteFactory;
 161 }