Software / code / prosody
Diff
plugins/mod_xmlrpc.lua @ 875:e73b6ef579c7
Added mod_xmlrpc
| author | Waqas Hussain <waqas20@gmail.com> |
|---|---|
| date | Wed, 04 Mar 2009 22:12:52 +0500 |
| child | 877:0bababc930dd |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/plugins/mod_xmlrpc.lua Wed Mar 04 22:12:52 2009 +0500 @@ -0,0 +1,115 @@ +-- Prosody IM v0.3 +-- 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. +-- + + +module.host = "*" -- Global module + +local httpserver = require "net.httpserver"; +local st = require "util.stanza"; +local pcall = pcall; +local unpack = unpack; +local tostring = tostring; + +local translate_request = require "util.xmlrpc".translate_request; +local create_response = require "util.xmlrpc".create_response; +local create_error_response = require "util.xmlrpc".create_error_response; + +local entity_map = setmetatable({ + ["amp"] = "&"; + ["gt"] = ">"; + ["lt"] = "<"; + ["apos"] = "'"; + ["quot"] = "\""; +}, {__index = function(_, s) + if s:sub(1,1) == "#" then + if s:sub(2,2) == "x" then + return string.char(tonumber(s:sub(3), 16)); + else + return string.char(tonumber(s:sub(2))); + end + end + end +}); +local function xml_unescape(str) + return (str:gsub("&(.-);", entity_map)); +end +local function parse_xml(xml) + local stanza = st.stanza("root"); + local regexp = "<([^>]*)>([^<]*)"; + for elem, text in xml:gmatch(regexp) do + --print("[<"..elem..">|"..text.."]"); + if elem:sub(1,1) == "!" or elem:sub(1,1) == "?" then -- neglect comments and processing-instructions + elseif elem:sub(1,1) == "/" then -- end tag + elem = elem:sub(2); + stanza:up(); -- TODO check for start-end tag name match + elseif elem:sub(-1,-1) == "/" then -- empty tag + elem = elem:sub(1,-2); + stanza:tag(elem):up(); + else -- start tag + stanza:tag(elem); + end + if #text ~= 0 then -- text + stanza:text(xml_unescape(text)); + end + end + return stanza.tags[1]; +end + +local function get_method(method) + return function(...) + return {method = method; args = {...}}; + end +end + +local function handle_xmlrpc_request(method, args) + method = get_method(method); + if not method then return create_error_response(404, "method not found"); end + args = args or {}; + local success, result = pcall(method, unpack(args)); + if success then + success, result = pcall(create_response, result or "nil"); + if success then + return result; + end + return create_error_response(500, "Error in creating response: "..result); + end + return create_error_response(0, result or "nil"); +end + +--[[local function handle_xmpp_request(origin, stanza) + local query = stanza.tags[1]; + if query.name == "query" then + if #query.tags == 1 then + local success, method, args = pcall(translate_request, query.tags[1]); + if success then + local result = handle_xmlrpc_request(method, args); + origin.send(st.reply(stanza):tag('query', {xmlns='jabber:iq:rpc'}):add_child(result)); + else + origin.send(st.error_reply(stanza, "modify", "bad-request", method)); + end + else + origin.send(st.error_reply(stanza, "modify", "bad-request", "No content in XML-RPC request")); + end + else + origin.send(st.error_reply(stanza, "cancel", "service-unavailable")); + end +end +module:add_iq_handler({"c2s", "s2sin"}, "jabber:iq:rpc", handle_xmpp_request);]] + +local function handle_http_request(method, body, request) + local stanza = body and parse_xml(body); + if (not stanza) or request.method ~= "POST" then + return "<html><body>You really don't look like an XML-RPC client to me... what do you want?</body></html>"; + end + local success, method, args = pcall(translate_request, stanza); + if success then + return tostring(handle_xmlrpc_request(method, args)); + end + return "<html><body>You really don't look like an XML-RPC client to me... what do you want?</body></html>"; +end +httpserver.new{ port = 9000, base = "xmlrpc", handler = handle_http_request }
