max1220 · February 7, 2018 22:32
diff --git a/xml_generate.lua b/xml_generate.lua
 -- Simple Lua XML generation library. 
 -- For example usage see test() function below.
 -- Does not atempt to escape characters anywhere, so be carefull!
 -- released under MIT License. 

 local insert = table.insert
 local concat = table.concat

 local function elem_t(name, parms, elems, parms_list)
 	local elem_t = {
 		name = name,
 		parms = parms,
 		elems = elems,
 		parms_list = parms_list
 	}
 	local elem_mt = {
 		__tostring = function(t)
 			local args = t._args or {}
 			local ret = {"<", t.name}
 			if t.parms then
 				local parms_str = {}
 				if t.parm_list then
 					for _,parm in ipairs(t.parms) do
 						insert(parms_str, parm.k.."=".."\""..parm.v.."\"")
 					end
 				else
 					for k,v in pairs(t.parms) do
 						insert(parms_str, k.."=".."\""..v.."\"")
 					end
 				end			
 				if #parms_str >= 1 then
 					insert(ret, " ")
 					insert(ret, table.concat(parms_str, " "))
 				end
 			end
 			insert(ret, ">")
 			if t.elems then
 				for _,elem in pairs(t.elems) do
 					if type(elem) == "function" then
 						insert(ret, tostring(elem(args) or ""))
 					else
 						elem._args = args
 						insert(ret, tostring(elem or ""))
 						elem._args = nil
 					end
 				end
 				insert(ret,"</"..t.name..">")
 			end				
 			return concat(ret)
 		end,
 		newindex = function(t,k,v) end -- TODO: insert elements/text or modify parameters based on type of k
 	}
 	setmetatable(elem_t, elem_mt)
 	return elem_t
 end

 local function generate(tree, args)
 	local ret = {}
 	local args = args or {}
 	for _, elem in pairs(tree) do
 		if type(elem) == "function" then
 			insert(ret, tostring(elem(args) or ""))
 		else
 			elem._args = args
 			insert(ret, tostring(elem or ""))
 			elem._args = nil
 		end
 	end
 	return concat(ret)
 end

 local function text(str)
 	local text_t = { text = str}
 	local text_mt = {
 		__tostring = function(t)
 			return tostring(t.text)
 		end
 	}
 	setmetatable(text_t, text_mt)
 	return text_t
 end

 local function variable(name, default)
 	return function(args)
 		if args and type(args) == "table" then
 			return args[name] or default
 		else
 			return default
 		end
 	end
 end

 local function test(t)
 	local text = t.text
 	local elem = t.element
 	local var = t.variable

 	local tree = {
 		text("<!DOCTYPE html>"),
 		elem("html", {}, {
 			elem("body", {style="background-color: grey; color: #333; font-family: sans-serif; padding: 0 0; margin: 0 0;"}, {
 				elem("div", {style="background-color: white; max-width: 800px; margin: 0 auto;"}, {
 					elem("h1", {}, {
 						text("Hello World!")
 					}),
 					elem("p", {}, {
 						text("Current date: "),
 						var("date", "No date aviable")
 					}),
 					function(args)
 						if args.secret then
 							return elem("p", {}, {text("I'm secret!")})
 						end
 					end
 				})
 			})
 		})
 	}
 	
 	table.insert(tree, function()
 		print("Running custom funtion!")
 		return "<!-- Comment from custom function! -->"
 	end)
 	
 	print("Stylesheet of body: (Should be: background-color: grey; color: #333; font-family: sans-serif; padding: 0 0; margin: 0 0;)")
 	print(tree[2].elems[1].parms.style)
 	print()
 	print("Body as string:")
 	print(tree[2].elems[1])
 	print()
 	print("Generating full tree as string without date")
 	print(generate(tree))
 	print("Generating full tree as string with date")
 	print(generate(tree, {date = os.date()}))
 	print("Generating full tree as string with date and secret")
 	print(generate(tree, {date = os.date(), secret = true}))
 	
 end

 return {
 	generate = generate,
 	element = elem_t,
 	variable = variable,
 	text = text,
 	_test = test,
 }
	-- Simple Lua XML generation library.
	-- For example usage see test() function below.
	-- Does not atempt to escape characters anywhere, so be carefull!
	-- released under MIT License.

	local insert = table.insert
	local concat = table.concat

	local function elem_t(name, parms, elems, parms_list)
	local elem_t = {
	name = name,
	parms = parms,
	elems = elems,
	parms_list = parms_list
	}
	local elem_mt = {
	__tostring = function(t)
	local args = t._args or {}
	local ret = {"<", t.name}
	if t.parms then
	local parms_str = {}
	if t.parm_list then
	for _,parm in ipairs(t.parms) do
	insert(parms_str, parm.k.."=".."\""..parm.v.."\"")
	end
	else
	for k,v in pairs(t.parms) do
	insert(parms_str, k.."=".."\""..v.."\"")
	end
	end
	if #parms_str >= 1 then
	insert(ret, " ")
	insert(ret, table.concat(parms_str, " "))
	end
	end
	insert(ret, ">")
	if t.elems then
	for _,elem in pairs(t.elems) do
	if type(elem) == "function" then
	insert(ret, tostring(elem(args) or ""))
	else
	elem._args = args
	insert(ret, tostring(elem or ""))
	elem._args = nil
	end
	end
	insert(ret,"</"..t.name..">")
	end
	return concat(ret)
	end,
	newindex = function(t,k,v) end -- TODO: insert elements/text or modify parameters based on type of k
	}
	setmetatable(elem_t, elem_mt)
	return elem_t
	end

	local function generate(tree, args)
	local ret = {}
	local args = args or {}
	for _, elem in pairs(tree) do
	if type(elem) == "function" then
	insert(ret, tostring(elem(args) or ""))
	else
	elem._args = args
	insert(ret, tostring(elem or ""))
	elem._args = nil
	end
	end
	return concat(ret)
	end

	local function text(str)
	local text_t = { text = str}
	local text_mt = {
	__tostring = function(t)
	return tostring(t.text)
	end
	}
	setmetatable(text_t, text_mt)
	return text_t
	end

	local function variable(name, default)
	return function(args)
	if args and type(args) == "table" then
	return args[name] or default
	else
	return default
	end
	end
	end

	local function test(t)
	local text = t.text
	local elem = t.element
	local var = t.variable

	local tree = {
	text("<!DOCTYPE html>"),
	elem("html", {}, {
	elem("body", {style="background-color: grey; color: #333; font-family: sans-serif; padding: 0 0; margin: 0 0;"}, {
	elem("div", {style="background-color: white; max-width: 800px; margin: 0 auto;"}, {
	elem("h1", {}, {
	text("Hello World!")
	}),
	elem("p", {}, {
	text("Current date: "),
	var("date", "No date aviable")
	}),
	function(args)
	if args.secret then
	return elem("p", {}, {text("I'm secret!")})
	end
	end
	})
	})
	})
	}

	table.insert(tree, function()
	print("Running custom funtion!")
	return "<!-- Comment from custom function! -->"
	end)

	print("Stylesheet of body: (Should be: background-color: grey; color: #333; font-family: sans-serif; padding: 0 0; margin: 0 0;)")
	print(tree[2].elems[1].parms.style)
	print()
	print("Body as string:")
	print(tree[2].elems[1])
	print()
	print("Generating full tree as string without date")
	print(generate(tree))
	print("Generating full tree as string with date")
	print(generate(tree, {date = os.date()}))
	print("Generating full tree as string with date and secret")
	print(generate(tree, {date = os.date(), secret = true}))

	end

	return {
	generate = generate,
	element = elem_t,
	variable = variable,
	text = text,
	_test = test,
	}