Diff

mod_rest/mod_rest.lua @ 3810:91ff86fc3b20

mod_rest: Factor out payload parsing To make it easier to add new format parsers in a single place
author Kim Alvefur <zash@zash.se>
date Wed, 01 Jan 2020 15:43:47 +0100
parent 3807:b0449faca52b
child 3811:eb25110696cd
line wrap: on
line diff
--- a/mod_rest/mod_rest.lua	Wed Jan 01 12:08:25 2020 +0100
+++ b/mod_rest/mod_rest.lua	Wed Jan 01 15:43:47 2020 +0100
@@ -22,6 +22,16 @@
 	return request.headers.authorization == secret;
 end
 
+local function parse(mimetype, data)
+	mimetype = mimetype:match("^[^; ]*");
+	if mimetype == "application/xmpp+xml" then
+		return xml.parse(data);
+	elseif mimetype == "text/plain" then
+		return st.message({ type = "chat" }, data);
+	end
+	return nil, "unknown-payload-type";
+end
+
 local function handle_post(event)
 	local request, response = event.request, event.response;
 	if not request.headers.authorization then
@@ -30,10 +40,7 @@
 	elseif not check_credentials(request) then
 		return 401;
 	end
-	if request.headers.content_type ~= "application/xmpp+xml" then
-		return errors.new({ code = 415, text = "'application/xmpp+xml' expected"  });
-	end
-	local payload, err = xml.parse(request.body);
+	local payload, err = parse(request.headers.content_type, request.body);
 	if not payload then
 		-- parse fail
 		return errors.new({ code = 400, text = err });
@@ -166,34 +173,25 @@
 					-- Delivered, no reply
 					return;
 				end
-				local reply, reply_text;
+				local reply;
 
-				if response.headers["content-type"] == "application/xmpp+xml" then
-					local parsed, err = xml.parse(body);
-					if not parsed then
-						module:log("warn", "REST callback responded with invalid XML: %s, %q", err, body);
-					elseif parsed.name ~= stanza.name then
-						module:log("warn", "REST callback responded with the wrong stanza type, got %s but expected %s", parsed.name, stanza.name);
-					else
-						parsed.attr = {
-							from = stanza.attr.to,
-							to = stanza.attr.from,
-							id = parsed.attr.id or id.medium();
-							type = parsed.attr.type,
-							["xml:lang"] = parsed.attr["xml:lang"],
-						};
-						if parsed.name == "iq" or parsed.attr.type == "error" then
-							parsed.attr.id = stanza.attr.id;
-						end
-						reply = parsed;
+				local parsed, err = parse(response.headers["content-type"], body);
+				if not parsed then
+					module:log("warn", "Failed parsing data from REST callback: %s, %q", err, body);
+				elseif parsed.name ~= stanza.name then
+					module:log("warn", "REST callback responded with the wrong stanza type, got %s but expected %s", parsed.name, stanza.name);
+				else
+					parsed.attr = {
+						from = stanza.attr.to,
+						to = stanza.attr.from,
+						id = parsed.attr.id or id.medium();
+						type = parsed.attr.type,
+						["xml:lang"] = parsed.attr["xml:lang"],
+					};
+					if parsed.name == "iq" or parsed.attr.type == "error" then
+						parsed.attr.id = stanza.attr.id;
 					end
-				elseif response.headers["content-type"] == "text/plain" then
-					reply = st.reply(stanza);
-					if body ~= "" then
-						reply_text = body;
-					end
-				elseif body ~= "" then -- ignore empty body
-					module:log("debug", "Callback returned response of unhandled type %q", response.headers["content-type"]);
+					reply = parsed;
 				end
 
 				if not reply then
@@ -203,18 +201,15 @@
 						if stanza.name ~= "iq" then
 							reply.attr.id = id.medium();
 						end
-						if reply_text and reply.name == "message" then
-							reply:body(reply_text, { ["xml:lang"] = response.headers["content-language"] });
-						end
 						-- TODO presence/status=body ?
 					elseif code2err[code] then
 						reply = st.error_reply(stanza, errors.new(code, nil, code2err));
 					elseif code_hundreds == 400 then
-						reply = st.error_reply(stanza, "modify", "bad-request", reply_text);
+						reply = st.error_reply(stanza, "modify", "bad-request", body);
 					elseif code_hundreds == 500 then
-						reply = st.error_reply(stanza, "cancel", "internal-server-error", reply_text);
+						reply = st.error_reply(stanza, "cancel", "internal-server-error", body);
 					else
-						reply = st.error_reply(stanza, "cancel", "undefined-condition", reply_text);
+						reply = st.error_reply(stanza, "cancel", "undefined-condition", body);
 					end
 				end