Module:Large category TOC/sandbox

This is the module sandbox page for Module:Large category TOC (diff).
local p = {}

local azupper = mw.text.split('ABCDEFGHIJKLMNOPQRSTUVWXYZ','')
local azlower = mw.text.split('abcdefghijklmnopqrstuvwxyz','')
local aejot = mw.text.split('aejot','')

local function main(toc_type)
	-- It should be much faster to only process these once, and just re use them as variables
	local pageurl = mw.title.getCurrentTitle():fullUrl()
	local toc = mw.message.new('Toc'):plain()
	
	-- Highest level div
	local toc_frame = mw.html.create('div')
		:addClass('large-cat-toc')
		:addClass('plainlinks')
		:addClass('hlist')
		:addClass('toc')
		-- :attr('id','toc')
		:css({
			['box-sizing'] = 'border-box',
			display = 'block',
			clear = 'both',
			width = '100%'
		})

	-- Contains "Content: Top 0-9 A - Z"
	local header = toc_frame:tag('div')
		:attr('id', 'toctitle')
		:attr('class', 'toctitle')
	
	-- Contains all the rest
	local body_wrapper
	local body = toc_frame:tag('div')
		:css('text-align', 'center')
		
	local jumpy_collapse = mw.html.create('div')
		:cssText('margin: 0 4em')
	
	if toc_type == 'collapsible' then
		toc_frame:addClass('mw-collapsible mw-collapsed')
			:cssText('padding: 4px; text-align: center; border: 1px solid #a2a9b1; font-size: 95%')
		header:cssText('font-weight: bold; line-height: 1.6em')
		body:addClass('mw-collapsible-content')
			:css({
				background = 'white'
			})
	elseif toc_type == 'scrollable' then
		body:css({ ['overflow-x'] = 'scroll',
					['overflow-y']= 'hidden',
					['white-space'] = 'nowrap' })
	end
	
	local header_content = {'<span style="font-weight: bold">',toc,':</span>',
						' [',pageurl,' Top]',
						' [',pageurl,'?from=0 0–9]' }
	
	for _, v in ipairs(azupper) do
		table.insert(header_content,string.format(' [%s?from=%s %s]',pageurl,v,v))
	end
	
	if toc_type == 'collapsible' then
		jumpy_collapse:wikitext(table.concat(header_content))
		header:node(jumpy_collapse)
	else
		header:wikitext(table.concat(header_content))
	end
	
	local body_content = {}
	
	if toc_type == 'collapsible' then
		table.insert(body_content,'<b>#</b> ')
		body_wrapper = body:tag('code')
						:css('background','White')
	else
		table.insert(body_content,'['..pageurl..'?from=* <b>*</b>] <b>#</b> ')
		body_wrapper = body:tag('span')
	end
	
	for i=0,9 do
		table.insert(body_content,string.format(' [%s?from=%s %s]',pageurl,i,i))
	end
	
	local function atoz(letter)
		local azlist = {}
		local letterlist
		if toc_type == 'aejot' then
			letterlist = aejot
		else
			letterlist = azlower
		end
				
		if toc_type == 'aejot' or toc_type == 'scrollable' then
			table.insert(azlist,' • <b>'..letter..'</b> ')
		else
			table.insert(azlist,'<br /><b>'..letter..'</b> ')
		end
			
		for _, v in ipairs(letterlist) do
			table.insert(azlist, string.format(
				' [%s?from=%s%s %s%s] ', pageurl, letter, v, letter, v
			))
		end
		return table.concat(azlist)
	end
	
	for _, v in ipairs(azupper) do
		table.insert(body_content, atoz(v))
	end
	
	body_wrapper:wikitext(table.concat(body_content))
	
	return '__NOTOC__\n' .. mw.getCurrentFrame():extensionTag{
		name = 'templatestyles', args = { src = 'Module:Large category TOC/styles.css' }
	} .. tostring(toc_frame)
end

function p.scrollable(frame)
	return main('scrollable')
end

function p.collapsible(frame)
	return main('collapsible')
end

function p.aejot(frame)
	return main('aejot')
end

return p
Usage
