util.dataforms: Add a simple function for identifying form types
authorKim Alvefur <zash@zash.se>
Sat, 02 Jun 2018 19:57:46 +0200
changeset 8867 cf2f66b233d1
parent 8866 64fa8d80c09f
child 8868 2a8bbfcb6868
util.dataforms: Add a simple function for identifying form types This is meant to allow identifying forms without parsing them completely.
spec/util_dataforms_spec.lua
util/dataforms.lua
--- a/spec/util_dataforms_spec.lua	Sat Jun 02 19:49:15 2018 +0200
+++ b/spec/util_dataforms_spec.lua	Sat Jun 02 19:57:46 2018 +0200
@@ -301,5 +301,14 @@
 		assert.equal("text-single-value", f:get_child_text("value"));
 	end);
 
+	describe("get_type()", function ()
+		it("identifes dataforms", function ()
+			assert.equal(nil, dataforms.get_type(nil));
+			assert.equal(nil, dataforms.get_type(""));
+			assert.equal(nil, dataforms.get_type({}));
+			assert.equal(nil, dataforms.get_type(st.stanza("no-a-form")));
+			assert.equal("xmpp:prosody.im/spec/util.dataforms#1", dataforms.get_type(xform));
+		end);
+	end);
 end);
 
--- a/util/dataforms.lua	Sat Jun 02 19:49:15 2018 +0200
+++ b/util/dataforms.lua	Sat Jun 02 19:57:46 2018 +0200
@@ -249,8 +249,24 @@
 		return field_tag:get_child_text("value");
 	end
 
+
+local function get_form_type(form)
+	if not st.is_stanza(form) then
+		return nil, "not a stanza object";
+	elseif form.attr.xmlns ~= "jabber:x:data" or form.name ~= "x" then
+		return nil, "not a dataform element";
+	end
+	for field in form:childtags("field") do
+		if field.attr.var == "FORM_TYPE" then
+			return field:get_child_text("value");
+		end
+	end
+	return "";
+end
+
 return {
 	new = new;
+	get_type = get_form_type;
 };