Mercurial > lxmppd-hg-xmllex / file comparison
comparison: util/discohelper.lua
util/discohelper.lua
- changeset 390
- 160d69c6eb89
- parent 387
- 700e95c00c5b
- child 519
- cccd610a0ef9
equal
deleted
inserted
replaced
| 383:32b586d9e915 | 390:160d69c6eb89 |
|---|---|
| 1 | |
| 2 local t_insert = table.insert; | |
| 3 local jid_split = require "util.jid".split; | |
| 4 local ipairs = ipairs; | |
| 5 local st = require "util.stanza"; | |
| 6 | |
| 7 module "discohelper"; | |
| 8 | |
| 9 local function addDiscoItemsHandler(self, jid, func) | |
| 10 if self.item_handlers[jid] then | |
| 11 t_insert(self.item_handlers[jid], func); | |
| 12 else | |
| 13 self.item_handlers[jid] = {func}; | |
| 14 end | |
| 15 end | |
| 16 | |
| 17 local function addDiscoInfoHandler(self, jid, func) | |
| 18 if self.info_handlers[jid] then | |
| 19 t_insert(self.info_handlers[jid], func); | |
| 20 else | |
| 21 self.info_handlers[jid] = {func}; | |
| 22 end | |
| 23 end | |
| 24 | |
| 25 local function handle(self, stanza) | |
| 26 if stanza.name == "iq" and stanza.tags[1].name == "query" then | |
| 27 local query = stanza.tags[1]; | |
| 28 local to = stanza.attr.to; | |
| 29 local from = stanza.attr.from | |
| 30 local node = query.attr.node or ""; | |
| 31 local to_node, to_host = jid_split(to); | |
| 32 | |
| 33 local reply = st.reply(stanza):query(query.attr.xmlns); | |
| 34 local handlers; | |
| 35 if query.attr.xmlns == "http://jabber.org/protocol/disco#info" then -- select handler set | |
| 36 handlers = self.info_handlers; | |
| 37 elseif query.attr.xmlns == "http://jabber.org/protocol/disco#items" then | |
| 38 handlers = self.item_handlers; | |
| 39 end | |
| 40 local handler = handlers[to]; -- get the handler | |
| 41 if not handler then -- if not found then use default handler | |
| 42 if to_node then | |
| 43 handler = handlers["*defaultnode"]; | |
| 44 else | |
| 45 handler = handlers["*defaulthost"]; | |
| 46 end | |
| 47 end | |
| 48 local found; -- to keep track of any handlers found | |
| 49 if handler then | |
| 50 for _, h in ipairs(handler) do | |
| 51 if h(reply, to, from, node) then found = true; end | |
| 52 end | |
| 53 end | |
| 54 if to_node then -- handlers which get called always | |
| 55 handler = handlers["*node"]; | |
| 56 else | |
| 57 handler = handlers["*host"]; | |
| 58 end | |
| 59 if handler then -- call always called handler | |
| 60 for _, h in ipairs(handler) do | |
| 61 if h(reply, to, from, node) then found = true; end | |
| 62 end | |
| 63 end | |
| 64 if found then return reply; end -- return the reply if there was one | |
| 65 return st.error_reply(stanza, "cancel", "service-unavailable"); | |
| 66 end | |
| 67 end | |
| 68 | |
| 69 function new() | |
| 70 return { | |
| 71 item_handlers = {}; | |
| 72 info_handlers = {}; | |
| 73 addDiscoItemsHandler = addDiscoItemsHandler; | |
| 74 addDiscoInfoHandler = addDiscoInfoHandler; | |
| 75 handle = handle; | |
| 76 }; | |
| 77 end | |
| 78 | |
| 79 return _M; |
