-/* +------------------------------------+
- * | Inspire Internet Relay Chat Daemon |
- * +------------------------------------+
- *
- * 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 "inspircd.h"
-#include <sstream>
-#include <list>
-#include "users.h"
-#include "channels.h"
-#include "modules.h"
-#include "configreader.h"
-#include "m_sqlutils.h"
-
-/* $ModDesc: Provides some utilities to SQL client modules, such as mapping queries to users and channels */
-/* $ModDep: m_sqlutils.h */
-
-typedef std::map<unsigned long, userrec*> IdUserMap;
-typedef std::map<unsigned long, chanrec*> IdChanMap;
-typedef std::list<unsigned long> AssocIdList;
-
-class ModuleSQLutils : public Module
-{
-private:
- IdUserMap iduser;
- IdChanMap idchan;
-
-public:
- ModuleSQLutils(InspIRCd* Me)
- : Module::Module(Me)
- {
- ServerInstance->PublishInterface("SQLutils", this);
- }
-
- virtual ~ModuleSQLutils()
- {
- ServerInstance->UnpublishInterface("SQLutils", this);
- }
-
- void Implements(char* List)
- {
- List[I_OnChannelDelete] = List[I_OnUnloadModule] = List[I_OnRequest] = List[I_OnUserDisconnect] = 1;
- }
-
- virtual char* OnRequest(Request* request)
- {
- if(strcmp(SQLUTILAU, request->GetId()) == 0)
- {
- AssociateUser* req = (AssociateUser*)request;
-
- iduser.insert(std::make_pair(req->id, req->user));
-
- AttachList(req->user, req->id);
- }
- else if(strcmp(SQLUTILAC, request->GetId()) == 0)
- {
- AssociateChan* req = (AssociateChan*)request;
-
- idchan.insert(std::make_pair(req->id, req->chan));
-
- AttachList(req->chan, req->id);
- }
- else if(strcmp(SQLUTILUA, request->GetId()) == 0)
- {
- UnAssociate* req = (UnAssociate*)request;
-
- /* Unassociate a given query ID with all users and channels
- * it is associated with.
- */
-
- DoUnAssociate(iduser, req->id);
- DoUnAssociate(idchan, req->id);
- }
- else if(strcmp(SQLUTILGU, request->GetId()) == 0)
- {
- GetAssocUser* req = (GetAssocUser*)request;
-
- IdUserMap::iterator iter = iduser.find(req->id);
-
- if(iter != iduser.end())
- {
- req->user = iter->second;
- }
- }
- else if(strcmp(SQLUTILGC, request->GetId()) == 0)
- {
- GetAssocChan* req = (GetAssocChan*)request;
-
- IdChanMap::iterator iter = idchan.find(req->id);
-
- if(iter != idchan.end())
- {
- req->chan = iter->second;
- }
- }
-
- return SQLUTILSUCCESS;
- }
-
- virtual void OnUserDisconnect(userrec* user)
- {
- /* A user is disconnecting, first we need to check if they have a list of queries associated with them.
- * Then, if they do, we need to erase each of them from our IdUserMap (iduser) so when the module that
- * associated them asks to look them up then it gets a NULL result and knows to discard the query.
- */
- AssocIdList* il;
-
- if(user->GetExt("sqlutils_queryids", il))
- {
- for(AssocIdList::iterator listiter = il->begin(); listiter != il->end(); listiter++)
- {
- IdUserMap::iterator iter;
-
- iter = iduser.find(*listiter);
-
- if(iter != iduser.end())
- {
- if(iter->second != user)
- {
- ServerInstance->Log(DEBUG, "BUG: ID associated with user %s doesn't have the same userrec* associated with it in the map (erasing anyway)", user->nick);
- }
-
- iduser.erase(iter);
- }
- else
- {
- ServerInstance->Log(DEBUG, "BUG: user %s was extended with sqlutils_queryids but there was nothing matching in the map", user->nick);
- }
- }
-
- user->Shrink("sqlutils_queryids");
- delete il;
- }
- }
-
- void AttachList(Extensible* obj, unsigned long id)
- {
- AssocIdList* il;
-
- if(!obj->GetExt("sqlutils_queryids", il))
- {
- /* Doesn't already exist, create a new list and attach it. */
- il = new AssocIdList;
- obj->Extend("sqlutils_queryids", il);
- }
-
- /* Now either way we have a valid list in il, attached. */
- il->push_back(id);
- }
-
- void RemoveFromList(Extensible* obj, unsigned long id)
- {
- AssocIdList* il;
-
- if(obj->GetExt("sqlutils_queryids", il))
- {
- /* Only do anything if the list exists... (which it ought to) */
- il->remove(id);
-
- if(il->empty())
- {
- /* If we just emptied it.. */
- delete il;
- obj->Shrink("sqlutils_queryids");
- }
- }
- }
-
- template <class T> void DoUnAssociate(T &map, unsigned long id)
- {
- /* For each occurence of 'id' (well, only one..it's not a multimap) in 'map'
- * remove it from the map, take an Extensible* value from the map and remove
- * 'id' from the list of query IDs attached to it.
- */
- typename T::iterator iter = map.find(id);
-
- if(iter != map.end())
- {
- /* Found a value indexed by 'id', call RemoveFromList()
- * on it with 'id' to remove 'id' from the list attached
- * to the value.
- */
- RemoveFromList(iter->second, id);
- }
- }
-
- virtual void OnChannelDelete(chanrec* chan)
- {
- /* A channel is being destroyed, first we need to check if it has a list of queries associated with it.
- * Then, if it does, we need to erase each of them from our IdChanMap (idchan) so when the module that
- * associated them asks to look them up then it gets a NULL result and knows to discard the query.
- */
- AssocIdList* il;
-
- if(chan->GetExt("sqlutils_queryids", il))
- {
- for(AssocIdList::iterator listiter = il->begin(); listiter != il->end(); listiter++)
- {
- IdChanMap::iterator iter;
-
- iter = idchan.find(*listiter);
-
- if(iter != idchan.end())
- {
- if(iter->second != chan)
- {
- ServerInstance->Log(DEBUG, "BUG: ID associated with channel %s doesn't have the same chanrec* associated with it in the map (erasing anyway)", chan->name);
- }
- idchan.erase(iter);
- }
- else
- {
- ServerInstance->Log(DEBUG, "BUG: channel %s was extended with sqlutils_queryids but there was nothing matching in the map", chan->name);
- }
- }
-
- chan->Shrink("sqlutils_queryids");
- delete il;
- }
- }
-
- virtual Version GetVersion()
- {
- return Version(1, 1, 0, 0, VF_VENDOR|VF_SERVICEPROVIDER, API_VERSION);
- }
-
-};
-
-MODULE_INIT(ModuleSQLutils);
-
+/* +------------------------------------+\r * | Inspire Internet Relay Chat Daemon |\r * +------------------------------------+\r *\r * InspIRCd: (C) 2002-2007 InspIRCd Development Team\r * See: http://www.inspircd.org/wiki/index.php/Credits\r *\r * This program is free but copyrighted software; see\r * the file COPYING for details.\r *\r * ---------------------------------------------------\r */\r\r#include "inspircd.h"\r#include <sstream>\r#include <list>\r#include "users.h"\r#include "channels.h"\r#include "modules.h"\r#include "configreader.h"\r#include "m_sqlutils.h"\r\r/* $ModDesc: Provides some utilities to SQL client modules, such as mapping queries to users and channels */\r/* $ModDep: m_sqlutils.h */\r\rtypedef std::map<unsigned long, userrec*> IdUserMap;\rtypedef std::map<unsigned long, chanrec*> IdChanMap;\rtypedef std::list<unsigned long> AssocIdList;\r\rclass ModuleSQLutils : public Module\r{\rprivate:\r IdUserMap iduser;\r IdChanMap idchan;\r\rpublic:\r ModuleSQLutils(InspIRCd* Me)\r : Module::Module(Me)\r {\r ServerInstance->PublishInterface("SQLutils", this);\r }\r\r virtual ~ModuleSQLutils()\r {\r ServerInstance->UnpublishInterface("SQLutils", this);\r } \r\r void Implements(char* List)\r {\r List[I_OnChannelDelete] = List[I_OnUnloadModule] = List[I_OnRequest] = List[I_OnUserDisconnect] = 1;\r }\r\r virtual char* OnRequest(Request* request)\r {\r if(strcmp(SQLUTILAU, request->GetId()) == 0)\r {\r AssociateUser* req = (AssociateUser*)request;\r \r iduser.insert(std::make_pair(req->id, req->user));\r \r AttachList(req->user, req->id);\r }\r else if(strcmp(SQLUTILAC, request->GetId()) == 0)\r {\r AssociateChan* req = (AssociateChan*)request;\r \r idchan.insert(std::make_pair(req->id, req->chan)); \r \r AttachList(req->chan, req->id);\r }\r else if(strcmp(SQLUTILUA, request->GetId()) == 0)\r {\r UnAssociate* req = (UnAssociate*)request;\r \r /* Unassociate a given query ID with all users and channels\r * it is associated with.\r */\r \r DoUnAssociate(iduser, req->id);\r DoUnAssociate(idchan, req->id);\r }\r else if(strcmp(SQLUTILGU, request->GetId()) == 0)\r {\r GetAssocUser* req = (GetAssocUser*)request;\r \r IdUserMap::iterator iter = iduser.find(req->id);\r \r if(iter != iduser.end())\r {\r req->user = iter->second;\r }\r }\r else if(strcmp(SQLUTILGC, request->GetId()) == 0)\r {\r GetAssocChan* req = (GetAssocChan*)request; \r \r IdChanMap::iterator iter = idchan.find(req->id);\r \r if(iter != idchan.end())\r {\r req->chan = iter->second;\r }\r }\r \r return SQLUTILSUCCESS;\r }\r \r virtual void OnUserDisconnect(userrec* user)\r {\r /* A user is disconnecting, first we need to check if they have a list of queries associated with them.\r * Then, if they do, we need to erase each of them from our IdUserMap (iduser) so when the module that\r * associated them asks to look them up then it gets a NULL result and knows to discard the query.\r */\r AssocIdList* il;\r \r if(user->GetExt("sqlutils_queryids", il))\r {\r for(AssocIdList::iterator listiter = il->begin(); listiter != il->end(); listiter++)\r {\r IdUserMap::iterator iter;\r \r iter = iduser.find(*listiter);\r \r if(iter != iduser.end())\r {\r if(iter->second != user)\r {\r ServerInstance->Log(DEBUG, "BUG: ID associated with user %s doesn't have the same userrec* associated with it in the map (erasing anyway)", user->nick);\r }\r\r iduser.erase(iter);\r }\r else\r {\r ServerInstance->Log(DEBUG, "BUG: user %s was extended with sqlutils_queryids but there was nothing matching in the map", user->nick);\r }\r }\r \r user->Shrink("sqlutils_queryids");\r delete il;\r }\r }\r \r void AttachList(Extensible* obj, unsigned long id)\r {\r AssocIdList* il;\r \r if(!obj->GetExt("sqlutils_queryids", il))\r {\r /* Doesn't already exist, create a new list and attach it. */\r il = new AssocIdList;\r obj->Extend("sqlutils_queryids", il);\r }\r \r /* Now either way we have a valid list in il, attached. */\r il->push_back(id);\r }\r \r void RemoveFromList(Extensible* obj, unsigned long id)\r {\r AssocIdList* il;\r \r if(obj->GetExt("sqlutils_queryids", il))\r {\r /* Only do anything if the list exists... (which it ought to) */\r il->remove(id);\r \r if(il->empty())\r {\r /* If we just emptied it.. */\r delete il;\r obj->Shrink("sqlutils_queryids");\r }\r }\r }\r \r template <class T> void DoUnAssociate(T &map, unsigned long id)\r {\r /* For each occurence of 'id' (well, only one..it's not a multimap) in 'map'\r * remove it from the map, take an Extensible* value from the map and remove\r * 'id' from the list of query IDs attached to it.\r */\r typename T::iterator iter = map.find(id);\r \r if(iter != map.end())\r {\r /* Found a value indexed by 'id', call RemoveFromList()\r * on it with 'id' to remove 'id' from the list attached\r * to the value.\r */\r RemoveFromList(iter->second, id);\r }\r }\r \r virtual void OnChannelDelete(chanrec* chan)\r {\r /* A channel is being destroyed, first we need to check if it has a list of queries associated with it.\r * Then, if it does, we need to erase each of them from our IdChanMap (idchan) so when the module that\r * associated them asks to look them up then it gets a NULL result and knows to discard the query.\r */\r AssocIdList* il;\r \r if(chan->GetExt("sqlutils_queryids", il))\r {\r for(AssocIdList::iterator listiter = il->begin(); listiter != il->end(); listiter++)\r {\r IdChanMap::iterator iter;\r \r iter = idchan.find(*listiter);\r \r if(iter != idchan.end())\r {\r if(iter->second != chan)\r {\r ServerInstance->Log(DEBUG, "BUG: ID associated with channel %s doesn't have the same chanrec* associated with it in the map (erasing anyway)", chan->name);\r }\r idchan.erase(iter); \r }\r else\r {\r ServerInstance->Log(DEBUG, "BUG: channel %s was extended with sqlutils_queryids but there was nothing matching in the map", chan->name);\r }\r }\r \r chan->Shrink("sqlutils_queryids");\r delete il;\r }\r }\r \r virtual Version GetVersion()\r {\r return Version(1, 1, 0, 0, VF_VENDOR|VF_SERVICEPROVIDER, API_VERSION);\r }\r \r};\r\rMODULE_INIT(ModuleSQLutils);\r\r
\ No newline at end of file