src/modules/extra/m_testclient.cpp

   1 #include <string>
   2
   3 #include "users.h"
   4 #include "channels.h"
   5 #include "modules.h"
   6 #include "helperfuncs.h"
   7 #include "inspircd.h"
   8 #include "configreader.h"
   9 #include "m_sqlv2.h"
  10
  11 class ModuleTestClient : public Module
  12 {
  13 private:
  14         Server* Srv;
  15
  16 public:
  17         ModuleTestClient(Server* Me)
  18         : Module::Module(Me), Srv(Me)
  19         {
  20         }
  21
  22         void Implements(char* List)
  23         {
  24                 List[I_OnRequest] = List[I_OnBackgroundTimer] = 1;
  25         }
  26
  27         virtual Version GetVersion()
  28         {
  29                 return Version(1, 0, 0, 0, VF_VENDOR);
  30         }
  31
  32         virtual void OnBackgroundTimer(time_t foo)
  33         {
  34                 Module* target = Srv->FindFeature("SQL");
  35
  36                 if(target)
  37                 {
  38                         SQLrequest foo = SQLreq(this, target, "foo", "SELECT foo, bar FROM ?", "rawr");
  39
  40                         if(foo.Send())
  41                         {
  42                                 log(DEBUG, "Sent query, got given ID %lu", foo.id);
  43                         }
  44                         else
  45                         {
  46                                 log(DEBUG, "SQLrequest failed: %s", foo.error.Str());
  47                         }
  48                 }
  49         }
  50
  51         virtual char* OnRequest(Request* request)
  52         {
  53                 if(strcmp(SQLRESID, request->GetData()) == 0)
  54                 {
  55                         log(DEBUG, "Got SQL result (%s)", request->GetData());
  56
  57                         SQLresult* res = (SQLresult*)request;
  58
  59                         log(DEBUG, "Got result with %d rows and %d columns", res->Rows(), res->Cols());
  60
  61                         for (int r = 0; r < res->Rows(); r++)
  62                         {
  63                                 log(DEBUG, "Row %d:", r);
  64
  65                                 for(int i = 0; i < res->Cols(); i++)
  66                                 {
  67                                         log(DEBUG, "\t[%s]: %s", res->ColName(i).c_str(), res->GetValue(r, i).d.c_str());
  68                                 }
  69                         }
  70
  71                         return SQLSUCCESS;
  72                 }
  73
  74                 log(DEBUG, "Got unsupported API version string: %s", request->GetData());
  75
  76                 return NULL;
  77         }
  78
  79         virtual ~ModuleTestClient()
  80         {
  81         }
  82 };
  83
  84 class ModuleTestClientFactory : public ModuleFactory
  85 {
  86  public:
  87         ModuleTestClientFactory()
  88         {
  89         }
  90
  91         ~ModuleTestClientFactory()
  92         {
  93         }
  94
  95         virtual Module * CreateModule(Server* Me)
  96         {
  97                 return new ModuleTestClient(Me);
  98         }
  99 };
 100
 101
 102 extern "C" void * init_module( void )
 103 {
 104         return new ModuleTestClientFactory;
 105 }