Module:TOC: Difference between revisions

local p = {}

function p.main( 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('template-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
	local ul = root:tag('ul')

	for i, num in ipairs(rowNums) do
		local header = args['header' .. num]
		local section = args['section' .. num]
		local link = args['link' .. num]

		if header then
			root:tag('div')
				:addClass('template-toctitle')
				:css('font-weight', 'bold')
				:wikitext(header)
			if not i == 1 then local ul = root:tag('ul')
		end

		if section then
			if link then
				section = '[[' .. link .. '|' .. section .. ']]'
			end
			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
	end
	return tostring(root)
end
return p