Module:HtmlBuilder

-- Module for building complex HTML (e.g. infoboxes, navboxes) using a fluent interface.
local HtmlBuilder = {}
local metatable = {}
metatable.__index = function(t, key)
 local ret = rawget(t, key)
 if ret then
 return ret
 end
 
 ret = metatable[key]
 if type(ret) == 'function' then
 return function(...) 
 return ret(t, ...) 
 end 
 else
 return ret
 end
end
metatable.__tostring = function(t)
 local ret = {}
 t._build(ret)
 return table.concat(ret)
end
metatable._build = function(t, ret)
 if t.tagName then 
 table.insert(ret, '<' .. t.tagName)
 for i, attr in ipairs(t.attributes) do
 table.insert(ret, ' ' .. attr.name .. '="' .. attr.val .. '"') 
 end
 if #t.styles > 0 then
 table.insert(ret, ' style="')
 for i, prop in ipairs(t.styles) do
 if type(prop) == 'string' then -- added with cssText()
 table.insert(ret, prop .. ';')
 else -- added with css()
 table.insert(ret, prop.name .. ':' .. prop.val .. ';')
 end
 end
 table.insert(ret, '"')
 end
 if t.selfClosing then
 table.insert(ret, ' /')
 end
 table.insert(ret, '>') 
 end
 for i, node in ipairs(t.nodes) do
 if node then
 if type(node) == 'table' then
 node._build(ret)
 else
 table.insert(ret, tostring(node))
 end
 end
 end
 if t.tagName and not t.unclosed and not t.selfClosing then
 table.insert(ret, '</' .. t.tagName .. '>')
 end
end
metatable.node = function(t, builder)
 if builder then
 table.insert(t.nodes, builder)
 end
 return t
end
metatable.wikitext = function(t, ...)
 local vals = {...}
 for i = 1, #vals do
 if vals[i] then
 table.insert(t.nodes, vals[i])
 end
 end
 return t
end
metatable.newline = function(t)
 table.insert(t.nodes, '\n')
 return t
end
metatable.tag = function(t, tagName, args)
 args = args or {}
 args.parent = t
 local builder = HtmlBuilder.create(tagName, args)
 table.insert(t.nodes, builder)
 return builder
end
local function getAttr(t, name)
 for i, attr in ipairs(t.attributes) do
 if attr.name == name then
 return attr
 end
 end
end
metatable.attr = function(t, name, val)
 if type(val) == 'string' or type(val) == 'number' then
 -- if caller sets the style attribute explicitly, then replace all styles previously added with css() and cssText()
 if name == 'style' then
 t.styles = {val}
 return t
 end
 
 local attr = getAttr(t, name)
 if attr then
 attr.val = val
 else
 table.insert(t.attributes, {name = name, val = val})
 end
 end
 
 return t
end
metatable.addClass = function(t, class)
 if class then
 local attr = getAttr(t, 'class')
 if attr then
 attr.val = attr.val .. ' ' .. class
 else
 t.attr('class', class)
 end
 end
 
 return t
end
metatable.css = function(t, name, val)
 if type(val) == 'string' or type(val) == 'number' then
 for i, prop in ipairs(t.styles) do
 if prop.name == name then
 prop.val = val
 return t
 end
 end
 
 table.insert(t.styles, {name = name, val = val})
 end
 
 return t
end
metatable.cssText = function(t, css)
 if css then
 table.insert(t.styles, css)
 end
 return t
end
metatable.done = function(t)
 return t.parent or t
end
metatable.allDone = function(t)
 while t.parent do
 t = t.parent
 end
 return t
end
function HtmlBuilder.create(tagName, args)
 args = args or {}
 local builder = {}
 setmetatable(builder, metatable)
 builder.nodes = {}
 builder.attributes = {}
 builder.styles = {}
 builder.tagName = tagName
 builder.parent = args.parent
 builder.unclosed = args.unclosed or false
 builder.selfClosing = args.selfClosing or false
 return builder
end
return HtmlBuilder