Module:TOC

From ProleWiki, the proletarian encyclopedia
Revision as of 17:09, 23 June 2022 by Forte (talk | contribs) (Improved code for numbering sections)
local p = {}
function p.toc( frame )
	local args = frame:getParent().args
	local root = mw.html.create()
	root = root:tag('div')
	root:addClass('template-toc')
	if args.title then
		root:tag('div')
		:addClass('toctitle')
		:tag('big')
		:css('font-weight', 'bold')
		:wikitext(args.title)
	end
	local rowNums = {}
	for k,v in pairs(args) do
		local num = k:match('^header(%d+)$') or k:match('^section(%d+)$') or k:match('^link(%d+)$')
		if num then table.insert(rowNums, tonumber(num))
		end
	end
	table.sort(rowNums)
	for i = #rowNums, 1, -1 do
		if rowNums[i] == rowNums[i - 1] then
			table.remove(rowNums, i)
		end
	end
	local section_number = 1
	for i, num in ipairs(rowNums) do
		local header = args['header' .. num]
		if header then
			root:tag('div')
			:addClass('toctitle')
			:css('font-weight', 'bold')
			:wikitext(header)
		end
		local section = args['section' .. num]
		local ul = root:tag('ul')
		if section then
			ul:tag('li')
				:addClass('toclevel-1')
				:tag('span')
					:addClass('tocnumber')
					:wikitext(section_number)
					:done()
				:tag('span')
					:addClass('toctext')
					:wikitext(section)
			section_number = section_number + 1
		end
	end
	return tostring(root)
end
return p