util.datamapper: Library for extracting data from stanzas

Based on the XML support in the OpenAPI specification.

1 files changed, 96 insertions, 0 deletions

diff --git a/util/datamapper.lua b/util/datamapper.lua
new file mode 100644
index 00000000..05f4da8c
--- /dev/null
+++ b/util/datamapper.lua
@@ -0,0 +1,96 @@
+local st = require("util.stanza");
+
+local function toboolean(s)
+	if s == "true" or s == "1" then
+		return true
+	elseif s == "false" or s == "0" then
+		return false
+	end
+end
+
+local function parse_object(schema, s)
+	local out = {}
+	if schema.properties then
+		for prop, propschema in pairs(schema.properties) do
+
+			local name = prop
+			local namespace = s.attr.xmlns;
+			local prefix = nil
+			local is_attribute = false
+			local is_text = false
+
+			local proptype
+			if type(propschema) == "table" then
+				proptype = propschema.type
+			elseif type(propschema) == "string" then
+				proptype = propschema
+			end
+
+			if type(propschema) == "table" and propschema.xml then
+				if propschema.xml.name then
+					name = propschema.xml.name
+				end
+				if propschema.xml.namespace then
+					namespace = propschema.xml.namespace
+				end
+				if propschema.xml.prefix then
+					prefix = propschema.xml.prefix
+				end
+				if propschema.xml.attribute then
+					is_attribute = true
+				elseif propschema.xml.text then
+					is_text = true
+				end
+			end
+
+			if is_attribute then
+				local attr = name
+				if prefix then
+					attr = prefix .. ":" .. name
+				elseif namespace ~= s.attr.xmlns then
+					attr = namespace .. "\1" .. name
+				end
+				if proptype == "string" then
+					out[prop] = s.attr[attr]
+				elseif proptype == "integer" or proptype == "number" then
+
+					out[prop] = tonumber(s.attr[attr])
+				elseif proptype == "boolean" then
+					out[prop] = toboolean(s.attr[attr])
+
+				end
+
+			elseif is_text then
+				if proptype == "string" then
+					out[prop] = s:get_text()
+				elseif proptype == "integer" or proptype == "number" then
+					out[prop] = tonumber(s:get_text())
+				end
+
+			else
+
+				if proptype == "string" then
+					out[prop] = s:get_child_text(name, namespace)
+				elseif proptype == "integer" or proptype == "number" then
+					out[prop] = tonumber(s:get_child_text(name, namespace))
+				elseif proptype == "object" and type(propschema) == "table" then
+					local c = s:get_child(name, namespace)
+					if c then
+						out[prop] = parse_object(propschema, c);
+					end
+
+				end
+			end
+		end
+	end
+
+	return out
+end
+
+local function parse(schema, s)
+	if schema.type == "object" then
+		return parse_object(schema, s)
+	end
+end
+
+return {parse = parse}