Changeset

11453:f0037234b2e9

util.datamapper: Enumerated elements E.g. error conditions or chat states.
author Kim Alvefur <zash@zash.se>
date Fri, 12 Mar 2021 01:33:15 +0100
parents 11452:c799bac7ca59
children 11454:1d9c1893cc5e
files spec/util_datamapper_spec.lua teal-src/util/datamapper.tl util/datamapper.lua
diffstat 3 files changed, 49 insertions(+), 3 deletions(-) [+]
line wrap: on
line diff
--- a/spec/util_datamapper_spec.lua	Wed Mar 10 00:19:45 2021 +0100
+++ b/spec/util_datamapper_spec.lua	Fri Mar 12 01:33:15 2021 +0100
@@ -29,6 +29,13 @@
 				};
 				state = {
 					type = "string";
+					enum = {
+						"active",
+						"inactive",
+						"gone",
+						"composing",
+						"paused",
+					};
 					xml = {x_name_is_value = true; namespace = "http://jabber.org/protocol/chatstates"};
 				};
 				fallback = {
@@ -46,6 +53,7 @@
 				<message xmlns="jabber:client" xml:lang="en" to="a@test" from="b@test" type="chat" id="1">
 				<body>Hello</body>
 				<delay xmlns='urn:xmpp:delay' from='test' stamp='2021-03-07T15:59:08+00:00'>Becasue</delay>
+				<UNRELATED xmlns='http://jabber.org/protocol/chatstates'/>
 				<active xmlns='http://jabber.org/protocol/chatstates'/>
 				<fallback xmlns='urn:xmpp:fallback:0'/>
 				<origin-id xmlns='urn:xmpp:sid:0' id='qgkmMdPB'/>
@@ -77,7 +85,7 @@
 			local u = map.unparse(s, d);
 			assert.equal("message", u.name);
 			assert.same(x.attr, u.attr);
-			assert.equal(#x.tags, #u.tags)
+			assert.equal(#x.tags-1, #u.tags)
 			assert.equal(x:get_child_text("body"), u:get_child_text("body"));
 			assert.equal(x:get_child_text("delay", "urn:xmpp:delay"), u:get_child_text("delay", "urn:xmpp:delay"));
 			assert.same(x:get_child("delay", "urn:xmpp:delay").attr, u:get_child("delay", "urn:xmpp:delay").attr);
--- a/teal-src/util/datamapper.tl	Wed Mar 10 00:19:45 2021 +0100
+++ b/teal-src/util/datamapper.tl	Fri Mar 12 01:33:15 2021 +0100
@@ -42,6 +42,7 @@
 			local is_text = false
 			local name_is_value = false;
 			local single_attribute : string
+			local enums : { any }
 
 			local proptype : js.schema_t.type_e
 			if propschema is js.schema_t then
@@ -72,10 +73,28 @@
 					-- XXX Custom extension
 					single_attribute = propschema.xml.x_single_attribute
 				end
+				if propschema["const"] then
+					enums = { propschema["const"] }
+				elseif propschema["enum"] then
+					enums = propschema["enum"]
+				end
 			end
 
 			if name_is_value then
-				local c = s:get_child(nil, namespace);
+				local c : st.stanza_t
+				if proptype == "boolean" then
+					c = s:get_child(name, namespace);
+				elseif enums and proptype == "string" then
+					-- XXX O(n²) ?
+					-- Probably better to flip the table and loop over :childtags(nil, ns), should be 2xO(n)
+					-- BUT works first, optimize later
+					for i = 1, #enums do
+						c = s:get_child(enums[i] as string, namespace);
+						if c then break end
+					end
+				else
+					c = s:get_child(nil, namespace);
+				end
 				if c and proptype == "string" then
 					out[prop] = c.name;
 				elseif proptype == "boolean" and c then
--- a/util/datamapper.lua	Wed Mar 10 00:19:45 2021 +0100
+++ b/util/datamapper.lua	Fri Mar 12 01:33:15 2021 +0100
@@ -20,6 +20,7 @@
 			local is_text = false
 			local name_is_value = false;
 			local single_attribute
+			local enums
 
 			local proptype
 			if type(propschema) == "table" then
@@ -50,10 +51,28 @@
 
 					single_attribute = propschema.xml.x_single_attribute
 				end
+				if propschema["const"] then
+					enums = {propschema["const"]}
+				elseif propschema["enum"] then
+					enums = propschema["enum"]
+				end
 			end
 
 			if name_is_value then
-				local c = s:get_child(nil, namespace);
+				local c
+				if proptype == "boolean" then
+					c = s:get_child(name, namespace);
+				elseif enums and proptype == "string" then
+
+					for i = 1, #enums do
+						c = s:get_child(enums[i], namespace);
+						if c then
+							break
+						end
+					end
+				else
+					c = s:get_child(nil, namespace);
+				end
 				if c and proptype == "string" then
 					out[prop] = c.name;
 				elseif proptype == "boolean" and c then