]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/modules/m_httpd_stats.cpp
Add call to protocol interface to get useful info on the server map. Return a std...
[user/henk/code/inspircd.git] / src / modules / m_httpd_stats.cpp
index a421ed36628e4185aba066408668f3b0fb7e9e5a..aa64dc3eabc63783a30a82f9cf38f91632bacae7 100644 (file)
@@ -2,44 +2,77 @@
  *       | Inspire Internet Relay Chat Daemon |
  *       +------------------------------------+
  *
- *  InspIRCd is copyright (C) 2002-2006 ChatSpike-Dev.
- *                    E-mail:
- *             <brain@chatspike.net>
- *               <Craig@chatspike.net>
- *     
- * Written by Craig Edwards, Craig McLure, and others.
+ *  InspIRCd: (C) 2002-2008 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.
  *
  * ---------------------------------------------------
  */
 
-using namespace std;
-
-#include <stdio.h>
-#include "users.h"
-#include "channels.h"
-#include "modules.h"
-#include "inspsocket.h"
-#include "helperfuncs.h"
+#include "inspircd.h"
 #include "httpd.h"
+#include "protocol.h"
 
 /* $ModDesc: Provides statistics over HTTP via m_httpd.so */
+/* $ModDep: httpd.h */
+
+typedef std::map<irc::string,int> StatsHash;
+typedef StatsHash::iterator StatsIter;
+
+typedef std::vector<std::pair<int,irc::string> > SortedList;
+typedef SortedList::iterator SortedIter;
+
+static StatsHash* sh = new StatsHash();
+static SortedList* so = new SortedList();
 
 class ModuleHttpStats : public Module
 {
-       Server* Srv;
+       
+       std::string stylesheet;
+       bool changed;
 
  public:
 
        void ReadConfig()
        {
+               ConfigReader c(ServerInstance);
+               this->stylesheet = c.ReadValue("httpstats", "stylesheet", 0);
        }
 
-       ModuleHttpStats(Server* Me) : Module::Module(Me)
+       ModuleHttpStats(InspIRCd* Me) : Module(Me)
        {
-               Srv = Me;
                ReadConfig();
+               this->changed = true;
+               Implementation eventlist[] = { I_OnEvent, I_OnRequest, I_OnChannelDelete, I_OnUserJoin, I_OnUserPart, I_OnUserQuit };
+               ServerInstance->Modules->Attach(eventlist, this, 6);
+       }
+
+       void InsertOrder(irc::string channel, int count)
+       {
+               /* This function figures out where in the sorted list to put an item from the hash */
+               SortedIter a;
+               for (a = so->begin(); a != so->end(); a++)
+               {
+                       /* Found an item equal to or less than, we insert our item before it */
+                       if (a->first <= count)
+                       {
+                               so->insert(a,std::pair<int,irc::string>(count,channel));
+                               return;
+                       }
+               }
+               /* There are no items in the list yet, insert something at the beginning */
+               so->insert(so->begin(), std::pair<int,irc::string>(count,channel));
+       }
+
+       void SortList()
+       {
+               /* Sorts the hash into the sorted list using an insertion sort */
+               so->clear();
+               for (StatsIter a = sh->begin(); a != sh->end(); a++)
+                       InsertOrder(a->first, a->second);
+               this->changed = false;
        }
 
        void OnEvent(Event* event)
@@ -48,59 +81,160 @@ class ModuleHttpStats : public Module
 
                if (event->GetEventID() == "httpd_url")
                {
-                       log(DEBUG,"HTTP URL!");
-
-                       data << "<html><h1>Chickens</h1></html>";
-
+                       ServerInstance->Logs->Log("m_http_stats", DEBUG,"Handling httpd event");
                        HTTPRequest* http = (HTTPRequest*)event->GetData();
-                       HTTPDocument response(http->sock, &data, 200, "X-Powered-By: m_http_stats.so\r\n");
-                       Request req((char*)&response, (Module*)this, event->GetSource());
-                       req.Send();
 
-                       log(DEBUG,"Sent");
+                       if ((http->GetURI() == "/stats") || (http->GetURI() == "/stats/"))
+                       {
+                               data << "<inspircdstats>";
+
+                               data << "<server><name>" << ServerInstance->Config->ServerName << "</name><gecos>" << ServerInstance->Config->ServerDesc << "</gecos></server>";
+
+                               data << "<general>";
+                               data << "<usercount>" << ServerInstance->Users->clientlist->size() << "</usercount>";
+                               data << "<channelcount>" << ServerInstance->chanlist->size() << "</channelcount>";
+                               data << "<opercount>" << ServerInstance->Users->all_opers.size() << "</opercount>";
+                               data << "<socketcount>" << (ServerInstance->SE->GetMaxFds() - ServerInstance->SE->GetRemainingFds()) << "</socketcount><socketmax>" << ServerInstance->SE->GetMaxFds() <<
+                                       "</socketmax><socketengine>" << ServerInstance->SE->GetName() << "</socketengine>";
+
+                               time_t current_time = 0;
+                               current_time = ServerInstance->Time();
+                               time_t server_uptime = current_time - ServerInstance->startup_time;
+                               struct tm* stime;
+                               stime = gmtime(&server_uptime);
+                               data << "<uptime><days>" << stime->tm_yday << "</days><hours>" << stime->tm_hour << "</hours><mins>" << stime->tm_min << "</mins><secs>" << stime->tm_sec << "</secs></uptime>";
+
+
+                               data << "</general>";
+                               data << "<modulelist>";
+                               std::vector<std::string> module_names = ServerInstance->Modules->GetAllModuleNames(0);
+                               for (std::vector<std::string>::iterator i = module_names.begin(); i != module_names.end(); ++i)
+                               {
+                                       Module* m = ServerInstance->Modules->Find(i->c_str());
+                                       Version v = m->GetVersion();
+                                       data << "<module><name>" << *i << "</name><version>" << v.Major << "." <<  v.Minor << "." << v.Revision << "." << v.Build << "</version></module>";
+                               }
+                               data << "</modulelist>";
+
+                               data << "<channellist>";
+                               /* If the list has changed since last time it was displayed, re-sort it
+                                * this time only (not every time, as this would be moronic)
+                                */
+                               if (this->changed)
+                                       this->SortList();
+
+                               for (SortedIter a = so->begin(); a != so->end(); a++)
+                               {
+                                       Channel* c = ServerInstance->FindChan(a->second.c_str());
+                                       if (c && !c->IsModeSet('s') && !c->IsModeSet('p'))
+                                       {
+                                               data << "<channel>";
+                                               data << "<usercount>" << c->GetUsers()->size() << "</usercount><channelname>" << c->name << "</channelname>";
+                                               data << "<channelops>" << c->GetOppedUsers()->size() << "</channelops>";
+                                               data << "<channelhalfops>" << c->GetHalfoppedUsers()->size() << "</channelhalfops>";
+                                               data << "<channelvoices>" << c->GetVoicedUsers()->size() << "</channelvoices>";
+                                               data << "<channeltopic>" << c->topic << "</channeltopic>";
+                                               data << "<channelmodes>" << c->ChanModes(false) << "</channelmodes>";
+                                               data << "</channel>";
+                                       }
+                               }
+
+                               data << "</channellist>";
+
+                               data << "<serverlist>";
+                               
+                               ProtoServerList sl;
+                               ServerInstance->PI->GetServerList(sl);
+
+                               for (ProtoServerList::iterator b = sl.begin(); b != sl.end(); ++b)
+                               {
+                                       data << "<server>";
+                                       data << "<servername>" << b->servername << "</servername>";
+                                       data << "<parentname>" << b->parentname << "</parentname>";
+                                       data << "<usercount>" << b->usercount << "</usercount>";
+                                       data << "<opercount>" << b->opercount << "</opercount>";
+                                       data << "<lagmillisecs>" << b->latencyms << "</lagmillisecs>";
+                                       data << "</server>";
+                               }
+                               data << "</serverlist>";
+
+                               data << "</inspircdstats>";
+
+                               /* Send the document back to m_httpd */
+                               HTTPDocument response(http->sock, &data, 200);
+                               response.headers.SetHeader("X-Powered-By", "m_httpd_stats.so");
+                               response.headers.SetHeader("Content-Type", "text/xml");
+                               Request req((char*)&response, (Module*)this, event->GetSource());
+                               req.Send();
+                       }
                }
        }
 
-       char* OnRequest(Request* request)
+       void OnChannelDelete(Channel* chan)
        {
-               return NULL;
+               StatsIter a = sh->find(chan->name);
+               if (a != sh->end())
+               {
+                       sh->erase(a);
+               }
+               this->changed = true;
        }
 
-       void Implements(char* List)
+       void OnUserJoin(User* user, Channel* channel, bool sync, bool &silent)
        {
-               List[I_OnEvent] = List[I_OnRequest] = 1;
+               StatsIter a = sh->find(channel->name);
+               if (a != sh->end())
+               {
+                       a->second++;
+               }
+               else
+               {
+                       irc::string name = channel->name;
+                       sh->insert(std::pair<irc::string,int>(name,1));
+               }
+               this->changed = true;
        }
 
-       virtual ~ModuleHttpStats()
+       void OnUserPart(User* user, Channel* channel, const std::string &partmessage, bool &silent)
        {
+               StatsIter a = sh->find(channel->name);
+               if (a != sh->end())
+               {
+                       a->second--;
+               }
+               this->changed = true;
        }
 
-       virtual Version GetVersion()
+       void OnUserQuit(User* user, const std::string &message, const std::string &oper_message)
        {
-               return Version(1,0,0,0,VF_STATIC|VF_VENDOR);
+               for (UCListIter v = user->chans.begin(); v != user->chans.end(); v++)
+               {
+                       Channel* c = v->first;
+                       StatsIter a = sh->find(c->name);
+                       if (a != sh->end())
+                       {
+                               a->second--;
+                       }
+               }
+               this->changed = true;
        }
-};
-
 
-class ModuleHttpStatsFactory : public ModuleFactory
-{
- public:
-       ModuleHttpStatsFactory()
+       const char* OnRequest(Request* request)
        {
+               return NULL;
        }
-       
-       ~ModuleHttpStatsFactory()
+
+
+       virtual ~ModuleHttpStats()
        {
+               delete sh;
+               delete so;
        }
-       
-       virtual Module * CreateModule(Server* Me)
+
+       virtual Version GetVersion()
        {
-               return new ModuleHttpStats(Me);
+               return Version(1, 2, 0, 0, VF_VENDOR, API_VERSION);
        }
 };
 
-
-extern "C" void * init_module( void )
-{
-       return new ModuleHttpStatsFactory;
-}
+MODULE_INIT(ModuleHttpStats)