Software /
code /
prosody
File
util/discohelper.lua @ 1111:a1cf1d623695
net.http, net.httpserver: Remove urlcodes table... it really isn't needed (thanks Jan Harkes)
author | Matthew Wild <mwild1@gmail.com> |
---|---|
date | Mon, 04 May 2009 19:53:31 +0100 |
parent | 896:2c0b9e3c11c3 |
child | 1509:a2ea99238466 |
line wrap: on
line source
-- Prosody IM v0.4 -- Copyright (C) 2008-2009 Matthew Wild -- Copyright (C) 2008-2009 Waqas Hussain -- -- This project is MIT/X11 licensed. Please see the -- COPYING file in the source package for more information. -- local t_insert = table.insert; local jid_split = require "util.jid".split; local ipairs = ipairs; local st = require "util.stanza"; module "discohelper"; local function addDiscoItemsHandler(self, jid, func) if self.item_handlers[jid] then t_insert(self.item_handlers[jid], func); else self.item_handlers[jid] = {func}; end end local function addDiscoInfoHandler(self, jid, func) if self.info_handlers[jid] then t_insert(self.info_handlers[jid], func); else self.info_handlers[jid] = {func}; end end local function handle(self, stanza) if stanza.name == "iq" and stanza.tags[1].name == "query" then local query = stanza.tags[1]; local to = stanza.attr.to; local from = stanza.attr.from local node = query.attr.node or ""; local to_node, to_host = jid_split(to); local reply = st.reply(stanza):query(query.attr.xmlns); local handlers; if query.attr.xmlns == "http://jabber.org/protocol/disco#info" then -- select handler set handlers = self.info_handlers; elseif query.attr.xmlns == "http://jabber.org/protocol/disco#items" then handlers = self.item_handlers; end local handler; local found; -- to keep track of any handlers found if to_node then -- handlers which get called always handler = handlers["*node"]; else handler = handlers["*host"]; end if handler then -- call always called handler for _, h in ipairs(handler) do if h(reply, to, from, node) then found = true; end end end handler = handlers[to]; -- get the handler if not handler then -- if not found then use default handler if to_node then handler = handlers["*defaultnode"]; else handler = handlers["*defaulthost"]; end end if handler then for _, h in ipairs(handler) do if h(reply, to, from, node) then found = true; end end end if found then return reply; end -- return the reply if there was one return st.error_reply(stanza, "cancel", "service-unavailable"); end end function new() return { item_handlers = {}; info_handlers = {}; addDiscoItemsHandler = addDiscoItemsHandler; addDiscoInfoHandler = addDiscoInfoHandler; handle = handle; }; end return _M;