From 00584c4d610a381e4181ab438102420aa601f310 Mon Sep 17 00:00:00 2001 From: Kim Alvefur Date: Sun, 17 Nov 2024 12:35:51 +0100 Subject: util.stanza: Handle namespace prefixes for attributes in :find() More correct handling of namespaces here. This works with both prefixes from the parser and hacky .attr["foo:bar"] --- spec/util_stanza_spec.lua | 4 ++++ spec/util_xtemplate_spec.lua | 5 ++--- util/stanza.lua | 4 ++++ 3 files changed, 10 insertions(+), 3 deletions(-) diff --git a/spec/util_stanza_spec.lua b/spec/util_stanza_spec.lua index e11c70dd..cabb27bf 100644 --- a/spec/util_stanza_spec.lua +++ b/spec/util_stanza_spec.lua @@ -577,5 +577,9 @@ describe("util.stanza", function() assert.equal("text", s:find("{urn:example:not:same}child/nested#"), "finds nested text") assert.is_nil(s:find("child"), "respects namespaces") end); + it("handles namespaced attributes", function() + local s = st.stanza("root", { ["urn:example:namespace\1attr"] = "value" }, { e = "urn:example:namespace" }); + assert.equal("value", s:find("@e:attr"), "finds prefixed attr") + end) end); end); diff --git a/spec/util_xtemplate_spec.lua b/spec/util_xtemplate_spec.lua index 421be43f..3b0ecaa9 100644 --- a/spec/util_xtemplate_spec.lua +++ b/spec/util_xtemplate_spec.lua @@ -10,7 +10,7 @@ describe("util.xtemplate", function () end) it("supports conditionals", function () local atom_tmpl = "{@pubsub:title|and{*{@pubsub:title}*\n\n}}{summary|or{{author/name|and{{author/name} posted }}{title}}}"; - local atom_data = st.stanza("entry", { xmlns = "http://www.w3.org/2005/Atom" }); + local atom_data = st.stanza("entry", { xmlns = "http://www.w3.org/2005/Atom" }, {["pubsub"] = "http://jabber.org/protocol/pubsub"}); assert.same("", xtemplate.render(atom_tmpl, atom_data)); atom_data:text_tag("title", "an Entry") @@ -22,8 +22,7 @@ describe("util.xtemplate", function () atom_data:text_tag("summary", "Juliet just posted a new entry"); assert.same("Juliet just posted a new entry", xtemplate.render(atom_tmpl, atom_data)); - atom_data.attr["xmlns:pubsub"] = "http://jabber.org/protocol/pubsub"; - atom_data.attr["pubsub:title"] = "Juliets musings"; + atom_data.attr["http://jabber.org/protocol/pubsub\1title"] = "Juliets musings"; assert.same("*Juliets musings*\n\nJuliet just posted a new entry", xtemplate.render(atom_tmpl, atom_data)); end) it("can strip surrounding whitespace", function () diff --git a/util/stanza.lua b/util/stanza.lua index 77110470..325c494e 100644 --- a/util/stanza.lua +++ b/util/stanza.lua @@ -277,6 +277,10 @@ function stanza_mt:find(path) local xmlns, name, text; local char = s_sub(path, pos, pos); if char == "@" then + local prefix, attr = s_match(path, "^([^:]+):(.*)", pos+1); + if prefix and self.namespaces and self.namespaces[prefix] then + return self.attr[self.namespaces[prefix] .. "\1" .. attr]; + end return self.attr[s_sub(path, pos + 1)]; elseif char == "{" then xmlns, pos = s_match(path, "^([^}]+)}()", pos + 1); -- cgit v1.2.3