]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/modules/m_httpd_stats.cpp
XHTML 1.1 spec validation and charset
[user/henk/code/inspircd.git] / src / modules / m_httpd_stats.cpp
index 526f9fda25a9623eea48baeef78d9538159b1244..83bcc97aaf7dfb0f41deae7712b37bbf75272150 100644 (file)
@@ -19,17 +19,37 @@ using namespace std;
 #include <stdio.h>
 #include "users.h"
 #include "channels.h"
+#include "configreader.h"
 #include "modules.h"
 #include "inspsocket.h"
 #include "helperfuncs.h"
 #include "httpd.h"
+#include "inspircd.h"
 
 /* $ModDesc: Provides statistics over HTTP via m_httpd.so */
 
+extern user_hash clientlist;
+extern chan_hash chanlist;
+extern std::vector<userrec*> all_opers;
+extern InspIRCd* ServerInstance;
+extern ServerConfig* Config;
+
+extern int MODCOUNT;
+
+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:
 
@@ -43,6 +63,33 @@ class ModuleHttpStats : public Module
        {
                Srv = Me;
                ReadConfig();
+               this->changed = false;
+       }
+
+       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)
@@ -51,23 +98,135 @@ class ModuleHttpStats : public Module
 
                if (event->GetEventID() == "httpd_url")
                {
-                       log(DEBUG,"HTTP URL!");
+                       HTTPRequest* http = (HTTPRequest*)event->GetData();
 
-                       data << "<HTML><HEAD>";
-                       data << "<TITLE>InspIRCd server statisitics for " << Srv->GetServerName() << " (" << Srv->GetServerDescription() << ")</TITLE>";
-                       data << "</HEAD><BODY>";
-                       data << "<H1>InspIRCd server statisitics for " << Srv->GetServerName() << " (" << Srv->GetServerDescription() << ")</H1>";
-                       
-                       data << "</BODY>";
-                       data << "</HTML>";
+                       if ((http->GetURI() == "/stats") || (http->GetURI() == "/stats/"))
+                       {
+                               log(DEBUG,"HTTP URL!");
 
-                       HTTPRequest* http = (HTTPRequest*)event->GetData();
-                       HTTPDocument response(http->sock, &data, 200, "X-Powered-By: m_http_stats.so\r\nContent-Type: text/html\r\n");
-                       Request req((char*)&response, (Module*)this, event->GetSource());
-                       req.Send();
+                               data << "<!DOCTYPE html PUBLIC \
+                                       \"-//W3C//DTD XHTML 1.1//EN\" \
+                                       \"http://www.w3.org/TR/xhtml11/DTD/xhtml11.dtd\">\n\
+                                       <html xmlns=\"http://www.w3.org/1999/xhtml\" xml:lang=\"en\">";
+
+                               data << "<head>";
+                               data << "<title>InspIRCd server statisitics for " << Srv->GetServerName() << " (" << Srv->GetServerDescription() << ")</title>";
+                               data << "</head><body>";
+                               data << "<h1>InspIRCd server statisitics for " << Srv->GetServerName() << " (" << Srv->GetServerDescription() << ")</h1>";
+
+                               data << "<div id='TOTALS'>";
+                               data << "<h2>Totals</h2>";
+                               data << "<table>";
+                               data << "<tr><td>Users</td><td>" << clientlist.size() << "</td></tr>";
+                               data << "<tr><td>Channels</td><td>" << chanlist.size() << "</td></tr>";
+                               data << "<tr><td>Opers</td><td>" << all_opers.size() << "</td></tr>";
+                               data << "<tr><td>Sockets</td><td>" << (ServerInstance->SE->GetMaxFds() - ServerInstance->SE->GetRemainingFds()) << " (Max: " << ServerInstance->SE->GetMaxFds() << " via socket engine '" << ServerInstance->SE->GetName() << "')</td></tr>";
+                               data << "</table>";
+                               data << "</div>";
+
+                               data << "<div id='MODULES'>";
+                               data << "<h2>Modules</h2>";
+                               data << "<table>";
+                               for (int i = 0; i <= MODCOUNT; i++)
+                               {
+                                       if (Config->module_names[i] != "")
+                                               data << "<tr><td>" << Config->module_names[i] << "</td></tr>";
+                               }
+                               data << "</table>";
+                               data << "</div>";
+
+                               data << "<div id='CHANNELS'>";
+                               data << "<h2>Channels</h2>";
+                               data << "<table>";
+                               data << "<tr><th>Users</th><th>Name</th><th>@</th><th>%</th><th>+</th></tr>";
+
+                               /* 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();
+
+                               int n = 0;
+                               for (SortedIter a = so->begin(); ((a != so->end()) && (n < 25)); a++, n++)
+                               {
+                                       chanrec* c = Srv->FindChannel(a->second.c_str());
+                                       if (c)
+                                       {
+                                               data << "<tr><td>" << a->first << "</td><td>" << a->second << "</td>";
+                                               data << "<td>" << c->GetOppedUsers()->size() << "</td>";
+                                               data << "<td>" << c->GetHalfoppedUsers()->size() << "</td>";
+                                               data << "<td>" << c->GetVoicedUsers()->size() << "</td>";
+                                               data << "</tr>";
+                                       }
+                               }
+
+                               data << "</table>";
+                               data << "</div>";
+
+                               
+                               data << "</body>";
+                               data << "</html>";
 
-                       log(DEBUG,"Sent");
+                               /* Send the document back to m_httpd */
+                               HTTPDocument response(http->sock, &data, 200, "X-Powered-By: m_http_stats.so\r\nContent-Type: text/html; charset=iso-8859-1\r\n");
+                               Request req((char*)&response, (Module*)this, event->GetSource());
+                               req.Send();
+
+                               log(DEBUG,"Sent");
+                       }
+               }
+       }
+
+       void OnChannelDelete(chanrec* chan)
+       {
+               StatsIter a = sh->find(chan->name);
+               if (a != sh->end())
+               {
+                       sh->erase(a);
+               }
+               this->changed = true;
+       }
+
+       void OnUserJoin(userrec* user, chanrec* channel)
+       {
+               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;
+       }
+
+       void OnUserPart(userrec* user, chanrec* channel, const std::string &partmessage)
+       {
+               StatsIter a = sh->find(channel->name);
+               if (a != sh->end())
+               {
+                       a->second--;
+               }
+               this->changed = true;
+       }
+
+       void OnUserQuit(userrec* user, const std::string &message)
+       {
+               for (std::vector<ucrec*>::const_iterator v = user->chans.begin(); v != user->chans.end(); v++)
+               {
+                       if (((ucrec*)(*v))->channel)
+                       {
+                               chanrec* c = ((ucrec*)(*v))->channel;
+                               StatsIter a = sh->find(c->name);
+                               if (a != sh->end())
+                               {
+                                       a->second--;
+                               }
+                       }
                }
+               this->changed = true;
        }
 
        char* OnRequest(Request* request)
@@ -77,11 +236,12 @@ class ModuleHttpStats : public Module
 
        void Implements(char* List)
        {
-               List[I_OnEvent] = List[I_OnRequest] = 1;
+               List[I_OnEvent] = List[I_OnRequest] = List[I_OnChannelDelete] = List[I_OnUserJoin] = List[I_OnUserPart] = List[I_OnUserQuit] = 1;
        }
 
        virtual ~ModuleHttpStats()
        {
+               delete sh;
        }
 
        virtual Version GetVersion()