Module:Auto compact TOC/sandbox

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

Implements {{auto compact TOC}}

The above documentation is transcluded from Module:Auto compact TOC/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.

local p = {}
local horizontal = require('Module:List').horizontal

local function make_TOC_item_linked(anchor, text)
	text = text or anchor
--	if (frame.args.auto == "")
	return ("[[#%s|%s]]"):format(anchor, text)
end
local function make_TOC_item_unlinked(anchor, text)
	text = text or anchor
	return ("%s"):format(anchor, text)
end
local Array_mt = { __index = table }
local function Array()
	return setmetatable({}, Array_mt)
end

function p.make_TOC(frame)
	local content = mw.title.getCurrentTitle():getContent()
	
	if not content then
		error "The current page has no content"
	end
	
	local letters = Array()
	-- Find uppermost headers containing a single ASCII letter.
	if frame.args.auto:upper() == "LINK" then -- If we wanted them linked, do that
		for letter in content:gmatch "%f[^\n]==%s*(%a)%s*==%f[^=]" do
		letter = letter:upper()
		letters:insert(make_TOC_item_linked(letter))
		end
	else -- f we do not want them linked, don't
		for letter in content:gmatch "%f[^\n]==%s*(%a)%s*==%f[^=]" do
		letter = letter:upper()
		letters:insert(make_TOC_item_unlinked(letter))
		end
	end
	
	
	return   horizontal(letters) .. horizontal(rest) 
end

return p