Module:Aligned table/sandbox

Module documentation[view] [edit] [history] [purge]

Implements {{Aligned table}}

The above documentation is transcluded from Module:Aligned table/doc. (edit | history)
Editors can experiment in this module's sandbox (edit | diff) and testcases (create) pages.
Add categories to the /doc subpage. Subpages of this module.

-- 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
	-- create the root table
	local root = mw.html.create('table')
	-- add table style for fullwidth	
	if fullwidth ~= '' then
		root
			:css('width', '100%')
			:css('border-collapse', 'collapse')
			:css('border-spacing', '0px')
			:css('border', 'none')
	end
	-- add table classes
	if class ~= '' then
		root:addClass(class)
	end
	-- add table style
	if style ~= '' then
		root:cssText(style)
	end
	-- build arrays with the column styles and classes
	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
	-- compute the maximum cell index
	local cellcount = 0
	for k, v in pairs( args ) do
		if type( k ) == 'number' then
			cellcount = math.max(cellcount, k)
		end
	end
	-- compute the number of rows
	local rows = math.ceil(cellcount / cols)
	
	-- build the table content
	for j=1,rows do
		-- start a new row
		local row = root:tag('tr')
		row:css('vertical-align', 'top')
		-- loop over the cells in each row
		for i=1,cols do
			local cell = row:tag('td')
			if args['class' .. tostring(j) .. '.' .. tostring(i)] then
				cell:addClass(args['class' .. tostring(j) .. '.' .. tostring(i)])
			elseif args['rowclass' .. tostring(j)] then
				cell:addClass(args['rowclass' .. tostring(j)] .. '"')
			elseif colclass[i] ~= '' then
				cell:addClass(colclass[i])
			end
			if args['style' .. tostring(j) .. '.' .. tostring(i)] then
				cell:cssText(args['style' .. tostring(j) .. '.' .. tostring(i)])
			elseif args['rowstyle' .. tostring(j)] then
				cell:cssText(args['rowstyle' .. tostring(j)])
			elseif colstyle[i] ~= '' then
				cell:cssText(colstyle[i])
			end
			cell:wikitext(args[rows*(j - 1) + i] or '')
        end
    end
	-- return the root table    
    return tostring(root)
 end

return p