prosody/util/xtemplate.lua
Kim Alvefur 2ef0e122fd util.xtemplate: Use same argument order in filters even without 'args'
This removes the different argument order used between '{x|foo}' and
'{x|foo(y)}' because the differing order was awkward and confusing.

This util does not seem to be widely used so should not be problematic
to change this part. The only known use is in mod_pubsub, which does not
use the filter function feature.
2024-10-16 16:15:05 +02:00

91 lines
2.8 KiB
Lua

local s_gsub = string.gsub;
local s_match = string.match;
local s_sub = string.sub;
local t_concat = table.concat;
local st = require("prosody.util.stanza");
local function render(template, root, escape, filters)
escape = escape or st.xml_escape;
return (s_gsub(template, "(%s*)(%b{})(%s*)", function(pre_blank, block, post_blank)
local inner = s_sub(block, 2, -2);
if inner:sub(1, 1) == "-" then
pre_blank = "";
inner = inner:sub(2);
end
if inner:sub(-1, -1) == "-" then
post_blank = "";
inner = inner:sub(1, -2);
end
local path, pipe, pos = s_match(inner, "^([^|]+)(|?)()");
if not (type(path) == "string") then return end
local value
if path == "." then
value = root;
elseif path == "#" then
value = root:get_text();
else
value = root:find(path);
end
local is_escaped = false;
while pipe == "|" do
local func, args, tmpl, p = s_match(inner, "^(%w+)(%b())(%b{})()", pos);
if not func then func, args, p = s_match(inner, "^(%w+)(%b())()", pos); end
if not func then func, tmpl, p = s_match(inner, "^(%w+)(%b{})()", pos); end
if not func then func, p = s_match(inner, "^(%w+)()", pos); end
if not func then break end
if tmpl then tmpl = s_sub(tmpl, 2, -2); end
if args then args = s_sub(args, 2, -2); end
if func == "each" and tmpl then
if not st.is_stanza(value) then return pre_blank .. post_blank end
if not args then value, args = root, path; end
local ns, name = s_match(args, "^(%b{})(.*)$");
if ns then
ns = s_sub(ns, 2, -2);
else
name, ns = args, nil;
end
if ns == "" then ns = nil; end
if name == "" then name = nil; end
local out, i = {}, 1;
for c in (value):childtags(name, ns) do out[i], i = render(tmpl, c, escape, filters), i + 1; end
value = t_concat(out);
is_escaped = true;
elseif func == "and" and tmpl then
local condition = value;
if args then condition = root:find(args); end
if condition then
value = render(tmpl, root, escape, filters);
is_escaped = true;
end
elseif func == "or" and tmpl then
local condition = value;
if args then condition = root:find(args); end
if not condition then
value = render(tmpl, root, escape, filters);
is_escaped = true;
end
elseif filters and filters[func] then
local f = filters[func];
value, is_escaped = f(value, args, tmpl);
else
error("No such filter function: " .. func);
end
pipe, pos = s_match(inner, "^(|?)()", p);
end
if type(value) == "string" then
if not is_escaped then value = escape(value); end
return pre_blank .. value .. post_blank
elseif st.is_stanza(value) then
value = value:get_text();
if value then return pre_blank .. escape(value) .. post_blank end
end
return pre_blank .. post_blank
end))
end
return { render = render }