* | Inspire Internet Relay Chat Daemon |
* +------------------------------------+
*
- * InspIRCd: (C) 2002-2009 InspIRCd Development Team
+ * InspIRCd: (C) 2002-2010 InspIRCd Development Team
* See: http://wiki.inspircd.org/Credits
*
* This program is free but copyrighted software; see
#include <map>
#include <string>
-class CapData : public classbase
+class CapEvent : public Event
{
public:
irc::string type;
std::vector<std::string> ack;
User* user;
Module* creator;
+ CapEvent(Module* sender, const std::string& t) : Event(sender, t) {}
};
-void GenericCapHandler(Event* ev, const std::string &extname, const std::string &cap)
+class GenericCap
{
- if (ev->GetEventID() == "cap_req")
- {
- CapData *data = (CapData *) ev->GetData();
-
- std::vector<std::string>::iterator it;
- if ((it = std::find(data->wanted.begin(), data->wanted.end(), cap)) != data->wanted.end())
- {
- // we can handle this, so ACK it, and remove it from the wanted list
- data->ack.push_back(*it);
- data->wanted.erase(it);
- data->user->Extend(extname);
- }
- }
-
- if (ev->GetEventID() == "cap_ls")
+ public:
+ LocalIntExt ext;
+ const std::string cap;
+ GenericCap(Module* parent, const std::string &Cap) : ext("cap_" + Cap, parent), cap(Cap)
{
- CapData *data = (CapData *) ev->GetData();
- data->wanted.push_back(cap);
+ ServerInstance->Extensions.Register(&ext);
}
- if (ev->GetEventID() == "cap_list")
+ void HandleEvent(Event& ev)
{
- CapData *data = (CapData *) ev->GetData();
+ CapEvent *data = static_cast<CapEvent*>(&ev);
+ if (ev.id == "cap_req")
+ {
+ std::vector<std::string>::iterator it;
+ if ((it = std::find(data->wanted.begin(), data->wanted.end(), cap)) != data->wanted.end())
+ {
+ // we can handle this, so ACK it, and remove it from the wanted list
+ data->ack.push_back(*it);
+ data->wanted.erase(it);
+ ext.set(data->user, 1);
+ }
+ }
- if (data->user->GetExt(extname))
+ if (ev.id == "cap_ls")
+ {
data->wanted.push_back(cap);
- }
+ }
- if (ev->GetEventID() == "cap_clear")
- {
- CapData *data = (CapData *) ev->GetData();
- data->ack.push_back("-" + cap);
- data->user->Shrink(extname);
+ if (ev.id == "cap_list")
+ {
+ if (ext.get(data->user))
+ data->wanted.push_back(cap);
+ }
+
+ if (ev.id == "cap_clear")
+ {
+ data->ack.push_back("-" + cap);
+ ext.set(data->user, 0);
+ }
}
-}
+};
#endif