mirror of
https://github.com/bjc/prosody.git
synced 2025-04-03 21:27:38 +03:00
98 lines
2.8 KiB
Lua
98 lines
2.8 KiB
Lua
-- luacheck: ignore 213/i
|
|
local stanza_mt = require "util.stanza".stanza_mt;
|
|
local setmetatable = setmetatable;
|
|
local pairs = pairs;
|
|
local ipairs = ipairs;
|
|
local error = error;
|
|
local envload = require "util.envload".envload;
|
|
local debug = debug;
|
|
local t_remove = table.remove;
|
|
local parse_xml = require "util.xml".parse;
|
|
|
|
local _ENV = nil;
|
|
-- luacheck: std none
|
|
|
|
local function trim_xml(stanza)
|
|
for i=#stanza,1,-1 do
|
|
local child = stanza[i];
|
|
if child.name then
|
|
trim_xml(child);
|
|
else
|
|
child = child:gsub("^%s*", ""):gsub("%s*$", "");
|
|
stanza[i] = child;
|
|
if child == "" then t_remove(stanza, i); end
|
|
end
|
|
end
|
|
end
|
|
|
|
local function create_string_string(str)
|
|
str = ("%q"):format(str);
|
|
str = str:gsub("{([^}]*)}", function(s)
|
|
return '"..(data["'..s..'"]or"").."';
|
|
end);
|
|
return str;
|
|
end
|
|
local function create_attr_string(attr, xmlns)
|
|
local str = '{';
|
|
for name,value in pairs(attr) do
|
|
if name ~= "xmlns" or value ~= xmlns then
|
|
str = str..("[%q]=%s;"):format(name, create_string_string(value));
|
|
end
|
|
end
|
|
return str..'}';
|
|
end
|
|
local function create_clone_string(stanza, lookup, xmlns)
|
|
if not lookup[stanza] then
|
|
local s = ('setmetatable({name=%q,attr=%s,tags={'):format(stanza.name, create_attr_string(stanza.attr, xmlns));
|
|
-- add tags
|
|
for i,tag in ipairs(stanza.tags) do
|
|
s = s..create_clone_string(tag, lookup, stanza.attr.xmlns)..";";
|
|
end
|
|
s = s..'};';
|
|
-- add children
|
|
for i,child in ipairs(stanza) do
|
|
if child.name then
|
|
s = s..create_clone_string(child, lookup, stanza.attr.xmlns)..";";
|
|
else
|
|
s = s..create_string_string(child)..";"
|
|
end
|
|
end
|
|
s = s..'}, stanza_mt)';
|
|
s = s:gsub('%.%.""', ""):gsub('([=;])""%.%.', "%1"):gsub(';"";', ";"); -- strip empty strings
|
|
local n = #lookup + 1;
|
|
lookup[n] = s;
|
|
lookup[stanza] = "_"..n;
|
|
end
|
|
return lookup[stanza];
|
|
end
|
|
local function create_cloner(stanza, chunkname)
|
|
local lookup = {};
|
|
local name = create_clone_string(stanza, lookup, "");
|
|
local src = "local setmetatable,stanza_mt=...;return function(data)";
|
|
for i=1,#lookup do
|
|
src = src.."local _"..i.."="..lookup[i]..";";
|
|
end
|
|
src = src.."return "..name..";end";
|
|
local f,err = envload(src, chunkname);
|
|
if not f then error(err); end
|
|
return f(setmetatable, stanza_mt);
|
|
end
|
|
|
|
local template_mt = { __tostring = function(t) return t.name end };
|
|
local function create_template(templates, text)
|
|
local stanza, err = parse_xml(text);
|
|
if not stanza then error(err); end
|
|
trim_xml(stanza);
|
|
|
|
local info = debug.getinfo(3, "Sl");
|
|
info = info and ("template(%s:%d)"):format(info.short_src:match("[^\\/]*$"), info.currentline) or "template(unknown)";
|
|
|
|
local template = setmetatable({ apply = create_cloner(stanza, info), name = info, text = text }, template_mt);
|
|
templates[text] = template;
|
|
return template;
|
|
end
|
|
|
|
local templates = setmetatable({}, { __mode = 'k', __index = create_template });
|
|
return function(text)
|
|
return templates[text];
|
|
end;
|