Module:Sandbox/ChaoticShadow/InfoboxBuilder

Documentation for this module may be created at Module:Sandbox/ChaoticShadow/InfoboxBuilder/doc

local InfoboxBuilder = {}
InfoboxBuilder.__index = InfoboxBuilder
InfoboxBuilder.__tostring = InfoboxBuilder.tostring

local tagmap = {
	th = 'th',
	td = 'td',
	argth = 'th',
	argtd = 'td'
}

local function tableSize(tbl)
	local size = 0
	for i, v in ipairs(tbl) do size = size + 1 end
	return size
end

function InfoboxBuilder.new()
	local obj = setmetatable({
		name = '',
		headerColors = {
			text = '',
			bg   = ''
		},
		params = {},
		paramnames = {},
		args = {},
		infobox = mw.html.create('table'):addClass('infobox')
	}, InfoboxBuilder)

	return obj
end

function InfoboxBuilder:setName(arg)
	assert(arg ~= nil)
	self.name = arg
	
	return self
end

function InfoboxBuilder:setWidth(arg)
	assert(arg ~= nil)
	self.infobox:css('width', arg)
	
	return self
end

function InfoboxBuilder:setHeaderTextColor(arg)
	assert(arg ~= nil)
	self.headerColors.text = arg
	
	return self
end

function InfoboxBuilder:setHeaderBackgroundColor(arg)
	assert(arg ~= nil)
	self.headerColors.bg = arg
	
	return self
end

function InfoboxBuilder:setHeaderColors(arg)
	assert(arg ~= nil)
	assert(arg.text ~= nil)
	assert(arg.bg ~= nil)
	self.headerColors.text = arg.text
	self.headerColors.bg   = arg.bg
	
	return self
end

function InfoboxBuilder:setParams(...)
	for i, v in ipairs(...) do
		if v.name then
			if type(v.func) == 'function' or type(v.func) == 'table' or type(v.func) == 'string' then
				self.params[v.name] = {
					['type'] = type(v.func),
					func = v.func,
					default = v.default
				}
			table.insert(self.paramnames, v.name)
			end
		end
	end
	
	return self
end

function InfoboxBuilder:setArgs(args)
	for k,v in pairs(args) do
		self.args[k] = v
	end
	
	return self
end

function InfoboxBuilder:getContent(v)
	local content = '?'
	local argparams = self.params[v.content]
	local arg = self.args[v.content] or self.params[v.content].default or ''
	
	if argparams['type'] == 'function' then
		content = self.params[v.content].func(arg)
	elseif argparams['type'] == 'table' then
		content = self.params[v.content].func[arg]
	elseif argparams['type'] == 'string' then
		content = arg
	end
	
	return content
end

function InfoboxBuilder:addHeader(arg)
	local _cell = self.infobox:tag('tr'):tag('th')
	_cell:css({
		['text-align']       = 'center',
		['background-color'] = self.headerColors.bg,
		['color']            = self.headerColors.text
	})
	
	if arg.attr then
		_cell:attr(arg.attr)
	end
	if arg.colspan then
		_cell:attr('colspan', arg.colspan)
	end
	if arg.rowspan then
		_cell:attr('rowspan', arg.rowspan)
	end
	if arg.css then
		_cell:css(arg.css)
	end
	
	_cell:wikitext(arg.content)
	
	return self
end

function InfoboxBuilder:addImage(...)
	local argt = ...
	
	local _cell = self.infobox:tag('tr'):tag('td')
	local content = '?'
	
	if #argt < 2 then
		content = self:getContent(argt[0])
	else
		local t = {}
		for i, v in ipairs(argt) do
			table.insert(t, i, self:getContent(v))
		end
		content = mw.getCurrentFrame()
					:callParserFunction({
						name = '#tag',
						args = {
							'tabber',
							table.concat(t, "|-|")
						}
					})
	end
	
	_cell:attr('colspan', 30)
	_cell:wikitext(content)
	
	return self
end

function InfoboxBuilder:addRow(...)
	local _row = self.infobox:tag('tr')
	
	for i, v in ipairs(...) do
		mw.log(v.tag)
		mw.log(tagmap[v.tag])
		local _cell = _row:tag(tagmap[v.tag] or 'td')
		
		if v.attr then
			_cell:attr(v.attr)
		end
		if v.colspan then
			_cell:attr('colspan', v.colspan)
		end
		if v.rowspan then
			_cell:attr('rowspan', v.rowspan)
		end
		if v.css then
			_cell:css(v.css)
		end
		
		if v.tag == 'th' or v.tag == 'td' then
			_cell:wikitext(v.content)
		elseif v.tag == 'argth' or v.tag == 'argtd' then
			_cell:wikitext(self:getContent(v))
		end
	end
	
	return self
end

function InfoboxBuilder:setupCols()
	local spacer = self.infobox:tag('tr')
	for i=1,30,1 do
		spacer:tag('td')
			:attr('colspan', 1)
			:css('width', 'calc(100% / 30)')
	end
end

function InfoboxBuilder:tostring()
	return tostring(self.infobox)
end

return InfoboxBuilder