Software /
code /
prosody
Diff
util/datamapper.lua @ 11439:9abcdfdcdb01
util.datamapper: Add support for mapping of elements where only one attribute matters
E.g. <feature var='foo'/> in XEP-0030 and some other simple
specifications.
author | Kim Alvefur <zash@zash.se> |
---|---|
date | Sun, 07 Mar 2021 12:48:49 +0100 |
parent | 11438:b7807583de34 |
child | 11451:ee4f2296e7df |
line wrap: on
line diff
--- a/util/datamapper.lua Sun Mar 07 01:41:39 2021 +0100 +++ b/util/datamapper.lua Sun Mar 07 12:48:49 2021 +0100 @@ -19,6 +19,7 @@ local is_attribute = false local is_text = false local name_is_value = false; + local single_attribute local proptype if type(propschema) == "table" then @@ -43,6 +44,8 @@ is_text = true elseif propschema.xml.x_name_is_value then name_is_value = true + elseif propschema.xml.x_single_attribute then + single_attribute = propschema.xml.x_single_attribute end end @@ -77,6 +80,16 @@ out[prop] = tonumber(s:get_text()) end + elseif single_attribute then + local c = s:get_child(name, namespace) + local a = c and c.attr[single_attribute] + if proptype == "string" then + out[prop] = a + elseif proptype == "integer" or proptype == "number" then + out[prop] = tonumber(a) + elseif proptype == "boolean" then + out[prop] = toboolean(a) + end else if proptype == "string" then @@ -135,6 +148,7 @@ local is_attribute = false local is_text = false local name_is_value = false; + local single_attribute if type(propschema) == "table" and propschema.xml then @@ -155,6 +169,8 @@ is_text = true elseif propschema.xml.x_name_is_value then name_is_value = true + elseif propschema.xml.x_single_attribute then + single_attribute = propschema.xml.x_single_attribute end end @@ -179,6 +195,24 @@ if type(v) == "string" then out:text(v) end + elseif single_attribute then + local propattr = {} + + if namespace ~= current_ns then + propattr.xmlns = namespace + end + + if proptype == "string" and type(v) == "string" then + propattr[single_attribute] = v + elseif proptype == "number" and type(v) == "number" then + propattr[single_attribute] = string.format("%g", v) + elseif proptype == "integer" and type(v) == "number" then + propattr[single_attribute] = string.format("%d", v) + elseif proptype == "boolean" and type(v) == "boolean" then + propattr[single_attribute] = v and "1" or "0" + end + out:tag(name, propattr):up(); + else local propattr if namespace ~= current_ns then