Module:Aligned table

Documentation for this module may be created at Module:Aligned table/doc

-- This module implements {{aligned table}}

local p = {}

function p.table(frame)
	local args = (frame.args[3] ~= nil) and frame.args or frame:getParent().args
	local entries = {}
	local colclass = {}
	local colstyle = {}
	local cols = tonumber(args['cols']) or 2
	local class = args['class'] or ''
	local style = args['style'] or ''
	local leftright = args['leftright'] or ''
	local fullwidth = args['fullwidth'] or ''
	
	if leftright ~= '' then
		colstyle[1] = 'text-align:left;'
		colstyle[2] = 'text-align:right;'
	end

	local root = mw.html.create('table')
	
	if fullwidth ~= '' then
		root
			:css('width', '100%')
			:css('border-collapse', 'collapse')
			:css('border-spacing', '0px')
			:css('border', 'none')
	end
	if class ~= '' then
		root:addClass(class)
	end
	if style ~= '' then
		root:cssText(style)
	end
	for i = 1,cols do
		colclass[ i ] = colclass[ i ] or ''
		colstyle[ i ] = colstyle[ i ] or ''
		if args['align' .. tostring(i)] then
			colstyle[ i ] = 'text-align:' .. args['align' .. tostring(i)] .. ';' .. colstyle[ i ]
		end
		if args['nowrap' .. tostring(i)] and args['nowrap' .. tostring(i)] ~= '' then
			colstyle[ i ] = 'white-space:nowrap;' .. colstyle[ i ]
		end
		if args['style' .. tostring(i)] then
			colstyle[ i ] = colstyle[ i ] .. args['style' .. tostring(i)]
		end
		if args['class' .. tostring(i)] then
			colclass[ i ] =  args['class' .. tostring(i)]
		end
	end
	
	for k, v in pairs( args ) do
		if type( k ) == 'number' then
			i = math.fmod(k-1,cols) + 1
			local j = (k - i) / cols + 1
			local tdstyle = ''
			if args['class' .. tostring(j) .. '.' .. tostring(i)] then
				tdstyle = tdstyle .. ' class="' .. args['class' .. tostring(j) .. '.' .. tostring(i)] .. '"'
			elseif args['rowclass' .. tostring(j)] then
				tdstyle = tdstyle .. ' class="' .. args['rowclass' .. tostring(j)] .. '"'
			elseif colclass[i] ~= '' then
				tdstyle = tdstyle .. ' class="' .. colclass[i] .. '"'
			end
			if args['style' .. tostring(j) .. '.' .. tostring(i)] then
				tdstyle = tdstyle .. ' style="' .. args['style' .. tostring(j) .. '.' .. tostring(i)] .. '"'
			elseif args['rowstyle' .. tostring(j)] then
				tdstyle = tdstyle .. ' style="' .. args['rowstyle' .. tostring(j)] .. '"'
			elseif colstyle[i] ~= '' then
				tdstyle = tdstyle .. ' style="' .. colstyle[i] .. '"'
			end
			entries[ k ] = '<td' .. tdstyle .. '>' .. v .. '</td>'
			if i == 1 then
				entries[ k ] = '<tr style="vertical-align:top">' .. entries[ k ]
			end
			if i == cols then
				entries[ k ] = entries[k] .. '</tr>'
			end
        end
    end
    root:wikitext(table.concat( entries, '\n' ))

    return tostring(root)
 end

return p