util/xmlrpc.lua

   1 -- Prosody IM v0.3
   2 -- Copyright (C) 2008-2009 Matthew Wild
   3 -- Copyright (C) 2008-2009 Waqas Hussain
   4 --
   5 -- This project is MIT/X11 licensed. Please see the
   6 -- COPYING file in the source package for more information.
   7 --
   8
   9
  10 local pairs = pairs;
  11 local type = type;
  12 local error = error;
  13 local t_concat = table.concat;
  14 local t_insert = table.insert;
  15 local tostring = tostring;
  16 local tonumber = tonumber;
  17 local st = require "util.stanza";
  18
  19 module "xmlrpc"
  20
  21 local _lua_to_xmlrpc;
  22 local map = {
  23         table=function(stanza, object)
  24                 stanza:tag("struct");
  25                 for name, value in pairs(object) do
  26                         stanza:tag("member");
  27                                 stanza:tag("name"):text(tostring(name)):up();
  28                                 stanza:tag("value");
  29                                         _lua_to_xmlrpc(stanza, value);
  30                                 stanza:up();
  31                         stanza:up();
  32                 end
  33                 stanza:up();
  34         end;
  35         boolean=function(stanza, object)
  36                 stanza:tag("boolean"):text(object and "1" or "0"):up();
  37         end;
  38         string=function(stanza, object)
  39                 stanza:tag("string"):text(object):up();
  40         end;
  41         number=function(stanza, object)
  42                 stanza:tag("int"):text(tostring(object)):up();
  43         end;
  44         ["nil"]=function(stanza, object) -- nil extension
  45                 stanza:tag("nil"):up();
  46         end;
  47 };
  48 _lua_to_xmlrpc = function(stanza, object)
  49         local h = map[type(object)];
  50         if h then
  51                 h(stanza, object);
  52         else
  53                 error("Type not supported by XML-RPC: " .. type(object));
  54         end
  55 end
  56 function create_response(object)
  57         local stanza = st.stanza("methodResponse"):tag("params"):tag("param"):tag("value");
  58         _lua_to_xmlrpc(stanza, object);
  59         stanza:up():up():up();
  60         return stanza;
  61 end
  62 function create_error_response(faultCode, faultString)
  63         local stanza = st.stanza("methodResponse"):tag("fault"):tag("value");
  64         _lua_to_xmlrpc(stanza, {faultCode=faultCode, faultString=faultString});
  65         stanza:up():up();
  66         return stanza;
  67 end
  68
  69
  70 local _xmlrpc_to_lua;
  71 local int_parse = function(stanza)
  72         if #stanza.tags ~= 0 or #stanza == 0 then error("<"..stanza.name.."> must have a single text child"); end
  73         local n = tonumber(t_concat(stanza));
  74         if n then return n; end
  75         error("Failed to parse content of <"..stanza.name..">");
  76 end
  77 local rmap = {
  78         methodCall=function(stanza)
  79                 if #stanza.tags ~= 2 then error("<methodCall> must have exactly two subtags"); end -- FIXME <params> is optional
  80                 if stanza.tags[1].name ~= "methodName" then error("First <methodCall> child tag must be <methodName>") end
  81                 if stanza.tags[2].name ~= "params" then error("Second <methodCall> child tag must be <params>") end
  82                 return _xmlrpc_to_lua(stanza.tags[1]), _xmlrpc_to_lua(stanza.tags[2]);
  83         end;
  84         methodName=function(stanza)
  85                 if #stanza.tags ~= 0 then error("<methodName> must not have any subtags"); end
  86                 if #stanza == 0 then error("<methodName> must have text content"); end
  87                 return t_concat(stanza);
  88         end;
  89         params=function(stanza)
  90                 local t = {};
  91                 for _, child in pairs(stanza.tags) do
  92                         if child.name ~= "param" then error("<params> can only have <param> children"); end;
  93                         t_insert(t, _xmlrpc_to_lua(child));
  94                 end
  95                 return t;
  96         end;
  97         param=function(stanza)
  98                 if not(#stanza.tags == 1 and stanza.tags[1].name == "value") then error("<param> must have exactly one <value> child"); end
  99                 return _xmlrpc_to_lua(stanza.tags[1]);
 100         end;
 101         value=function(stanza)
 102                 if #stanza.tags == 0 then return t_concat(stanza); end
 103                 if #stanza.tags ~= 1 then error("<value> must have a single child"); end
 104                 return _xmlrpc_to_lua(stanza.tags[1]);
 105         end;
 106         int=int_parse;
 107         i4=int_parse;
 108         double=int_parse;
 109         boolean=function(stanza)
 110                 if #stanza.tags ~= 0 or #stanza == 0 then error("<boolean> must have a single text child"); end
 111                 local b = t_concat(stanza);
 112                 if b ~= "1" and b ~= "0" then error("Failed to parse content of <boolean>"); end
 113                 return b == "1" and true or false;
 114         end;
 115         string=function(stanza)
 116                 if #stanza.tags ~= 0 then error("<string> must have a single text child"); end
 117                 return t_concat(stanza);
 118         end;
 119         array=function(stanza)
 120                 if #stanza.tags ~= 1 then error("<array> must have a single <data> child"); end
 121                 return _xmlrpc_to_lua(stanza.tags[1]);
 122         end;
 123         data=function(stanza)
 124                 local t = {};
 125                 for _,child in pairs(stanza.tags) do
 126                         if child.name ~= "value" then error("<data> can only have <value> children"); end
 127                         t_insert(t, _xmlrpc_to_lua(child));
 128                 end
 129                 return t;
 130         end;
 131         struct=function(stanza)
 132                 local t = {};
 133                 for _,child in pairs(stanza.tags) do
 134                         if child.name ~= "member" then error("<struct> can only have <member> children"); end
 135                         local name, value = _xmlrpc_to_lua(child);
 136                         t[name] = value;
 137                 end
 138                 return t;
 139         end;
 140         member=function(stanza)
 141                 if #stanza.tags ~= 2 then error("<member> must have exactly two subtags"); end -- FIXME <params> is optional
 142                 if stanza.tags[1].name ~= "name" then error("First <member> child tag must be <name>") end
 143                 if stanza.tags[2].name ~= "value" then error("Second <member> child tag must be <value>") end
 144                 return _xmlrpc_to_lua(stanza.tags[1]), _xmlrpc_to_lua(stanza.tags[2]);
 145         end;
 146         name=function(stanza)
 147                 if #stanza.tags ~= 0 then error("<name> must have a single text child"); end
 148                 local n = t_concat(stanza)
 149                 if tostring(tonumber(n)) == n then n = tonumber(n); end
 150                 return n;
 151         end;
 152         ["nil"]=function(stanza) -- nil extension
 153                 return nil;
 154         end;
 155 }
 156 _xmlrpc_to_lua = function(stanza)
 157         local h = rmap[stanza.name];
 158         if h then
 159                 return h(stanza);
 160         else
 161                 error("Unknown element: "..stanza.name);
 162         end
 163 end
 164 function translate_request(stanza)
 165         if stanza.name ~= "methodCall" then error("XML-RPC requests must have <methodCall> as root element"); end
 166         return _xmlrpc_to_lua(stanza);
 167 end
 168
 169 return _M;