Mercurial > prosody-modules

comparison mod_pubsub_post/mod_pubsub_post.lua @ 3014:72dbc9b66de8

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
mod_pubsub_post: Change to support arbitrary XML payloads
author Kim Alvefur <zash@zash.se>
date Sat, 19 May 2018 23:56:39 +0200
parents 8bfba31a1f8b
children 338b7c808ecc
comparison
equal deleted inserted replaced
3013:8bfba31a1f8b 3014:72dbc9b66de8
1 module:depends("http"); 1 module:depends("http");
2 2
3 local st = require "util.stanza"; 3 local st = require "util.stanza";
4 local json = require "util.json"; 4 local json = require "util.json";
5 local xml = require "util.xml";
5 local uuid_generate = require "util.uuid".generate; 6 local uuid_generate = require "util.uuid".generate;
6 local timestamp_generate = require "util.datetime".datetime; 7 local timestamp_generate = require "util.datetime".datetime;
7 8
8 local pubsub_service = module:depends("pubsub").service; 9 local pubsub_service = module:depends("pubsub").service;
9 10
11 local error_mapping = {
12 ["forbidden"] = 403;
13 ["item-not-found"] = 404;
14 ["internal-server-error"] = 500;
15 ["conflict"] = 409;
16 };
17
18 local function publish_payload(node, item_id, payload)
19 local post_item = st.stanza("item", { xmlns = "http://jabber.org/protocol/pubsub", id = item_id, })
20 :add_child(payload);
21 local ok, err = pubsub_service:publish(node, true, item_id, post_item);
22 module:log("debug", ":publish(%q, true, %q, %s) -> %q", node, item_id, payload:top_tag(), err or "");
23 if not ok then
24 return error_mapping[err] or 500;
25 end
26 return 202;
27 end
28
29 local function handle_xml(node, payload)
30 local xmlpayload, err = xml.parse(payload);
31 if not xmlpayload then
32 module:log("debug", "XML parse error: %s\n%q", err, payload);
33 return { status_code = 400, body = tostring(err) };
34 end
35 return publish_payload(node, "current", xmlpayload);
36 end
37
10 function handle_POST(event, path) 38 function handle_POST(event, path)
11 local data = event.request.body; 39 local request = event.request;
12 local item_id = "default"; 40 module:log("debug", "Handling POST: \n%s\n", tostring(request.body));
13 41
14 local post_item = st.stanza("item", { id = item_id, xmlns = "http://jabber.org/protocol/pubsub" }) 42 local content_type = request.headers.content_type or "application/octet-stream";
15 :tag("entry", { xmlns = "http://www.w3.org/2005/Atom" }) 43
16 :tag("id"):text(uuid_generate()):up() 44 if content_type == "application/xml" or content_type:sub(-4) == "+xml" then
17 :tag("title"):text(data):up() 45 return handle_xml(path, request.body);
18 :tag("author") 46 end
19 :tag("name"):text(event.request.conn:ip()):up() 47
20 :up() 48 module:log("debug", "Unsupported content-type: %q", content_type);
21 :tag("published"):text(timestamp_generate()):up(); 49 return 415;
22
23 local ok, err = pubsub_service:publish(path, true, item_id, post_item);
24 module:log("debug", "Handled POST: \n%s\n", tostring(event.request.body));
25 return ok and "Posted" or ("Error: "..err);
26 end 50 end
27 51
28 module:provides("http", { 52 module:provides("http", {
29 route = { 53 route = {
30 ["POST /*"] = handle_POST; 54 ["POST /*"] = handle_POST;