-/* +------------------------------------+\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
+/* +------------------------------------+
+ * | 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);
+
projects / user / henk / code / inspircd.git / blobdiff