1 /* +------------------------------------+
2 * | Inspire Internet Relay Chat Daemon |
3 * +------------------------------------+
5 * InspIRCd: (C) 2002-2009 InspIRCd Development Team
6 * See: http://wiki.inspircd.org/Credits
8 * This program is free but copyrighted software; see
9 * the file COPYING for details.
11 * ---------------------------------------------------
17 #include "m_sqlutils.h"
19 /* $ModDesc: Provides some utilities to SQL client modules, such as mapping queries to users and channels */
20 /* $ModDep: m_sqlutils.h */
22 typedef std::map<unsigned long, User*> IdUserMap;
23 typedef std::map<unsigned long, Channel*> IdChanMap;
24 typedef std::list<unsigned long> AssocIdList;
26 class ModuleSQLutils : public Module
31 SimpleExtItem<AssocIdList> idExt;
34 ModuleSQLutils() : idExt("sqlutils_list", this)
36 ServerInstance->Modules->PublishInterface("SQLutils", this);
37 Implementation eventlist[] = { I_OnChannelDelete, I_OnUnloadModule, I_OnUserDisconnect };
38 ServerInstance->Modules->Attach(eventlist, this, 3);
43 ServerInstance->Modules->UnpublishInterface("SQLutils", this);
47 void OnRequest(Request& request)
49 if(strcmp(SQLUTILAU, request.id) == 0)
51 AssociateUser* req = (AssociateUser*)&request;
53 iduser.insert(std::make_pair(req->id, req->user));
55 AttachList(req->user, req->id);
57 else if(strcmp(SQLUTILAC, request.id) == 0)
59 AssociateChan* req = (AssociateChan*)&request;
61 idchan.insert(std::make_pair(req->id, req->chan));
63 AttachList(req->chan, req->id);
65 else if(strcmp(SQLUTILUA, request.id) == 0)
67 UnAssociate* req = (UnAssociate*)&request;
69 /* Unassociate a given query ID with all users and channels
70 * it is associated with.
73 DoUnAssociate(iduser, req->id);
74 DoUnAssociate(idchan, req->id);
76 else if(strcmp(SQLUTILGU, request.id) == 0)
78 GetAssocUser* req = (GetAssocUser*)&request;
80 IdUserMap::iterator iter = iduser.find(req->id);
82 if(iter != iduser.end())
84 req->user = iter->second;
87 else if(strcmp(SQLUTILGC, request.id) == 0)
89 GetAssocChan* req = (GetAssocChan*)&request;
91 IdChanMap::iterator iter = idchan.find(req->id);
93 if(iter != idchan.end())
95 req->chan = iter->second;
100 void OnUserDisconnect(User* user)
102 /* A user is disconnecting, first we need to check if they have a list of queries associated with them.
103 * Then, if they do, we need to erase each of them from our IdUserMap (iduser) so when the module that
104 * associated them asks to look them up then it gets a NULL result and knows to discard the query.
106 AssocIdList* il = idExt.get(user);
110 for(AssocIdList::iterator listiter = il->begin(); listiter != il->end(); listiter++)
112 IdUserMap::iterator iter;
114 iter = iduser.find(*listiter);
116 if(iter != iduser.end())
118 if(iter->second != user)
120 ServerInstance->Logs->Log("m_sqlutils",DEBUG, "BUG: ID associated with user %s doesn't have the same User* associated with it in the map (erasing anyway)", user->nick.c_str());
127 ServerInstance->Logs->Log("m_sqlutils",DEBUG, "BUG: user %s was extended with sqlutils_queryids but there was nothing matching in the map", user->nick.c_str());
135 void AttachList(Extensible* obj, unsigned long id)
137 AssocIdList* il = idExt.get(obj);
141 /* Doesn't already exist, create a new list and attach it. */
142 il = new AssocIdList;
146 /* Now either way we have a valid list in il, attached. */
150 void RemoveFromList(Extensible* obj, unsigned long id)
152 AssocIdList* il = idExt.get(obj);
156 /* Only do anything if the list exists... (which it ought to) */
161 /* If we just emptied it.. */
167 template <class T> void DoUnAssociate(T &map, unsigned long id)
169 /* For each occurence of 'id' (well, only one..it's not a multimap) in 'map'
170 * remove it from the map, take an Extensible* value from the map and remove
171 * 'id' from the list of query IDs attached to it.
173 typename T::iterator iter = map.find(id);
175 if(iter != map.end())
177 /* Found a value indexed by 'id', call RemoveFromList()
178 * on it with 'id' to remove 'id' from the list attached
181 RemoveFromList(iter->second, id);
185 void OnChannelDelete(Channel* chan)
187 /* A channel is being destroyed, first we need to check if it has a list of queries associated with it.
188 * Then, if it does, we need to erase each of them from our IdChanMap (idchan) so when the module that
189 * associated them asks to look them up then it gets a NULL result and knows to discard the query.
191 AssocIdList* il = idExt.get(chan);
195 for(AssocIdList::iterator listiter = il->begin(); listiter != il->end(); listiter++)
197 IdChanMap::iterator iter;
199 iter = idchan.find(*listiter);
201 if(iter != idchan.end())
203 if(iter->second != chan)
205 ServerInstance->Logs->Log("m_sqlutils",DEBUG, "BUG: ID associated with channel %s doesn't have the same Channel* associated with it in the map (erasing anyway)", chan->name.c_str());
211 ServerInstance->Logs->Log("m_sqlutils",DEBUG, "BUG: channel %s was extended with sqlutils_queryids but there was nothing matching in the map", chan->name.c_str());
221 return Version("Provides some utilities to SQL client modules, such as mapping queries to users and channels", VF_VENDOR | VF_SERVICEPROVIDER, API_VERSION);
226 MODULE_INIT(ModuleSQLutils)