Diff

util/datamapper.lua @ 11435:a1fa6202fa13

util.datamapper: Library for extracting data from stanzas Based on the XML support in the OpenAPI specification.
author Kim Alvefur <zash@zash.se>
date Sun, 07 Mar 2021 00:57:36 +0100
child 11436:5df9ffc25bb4
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/util/datamapper.lua	Sun Mar 07 00:57:36 2021 +0100
@@ -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}