Module:Toolbar

From Zoophilia Wiki
Revision as of 01:00, 29 October 2013 by meta>Mr. Stradivarius (allow access to the p._main function from other modules, return the blank string if no arguments were specified, and switch indentation to tabs)
Jump to navigationJump to search

Documentation for this module may be created at Module:Toolbar/doc

local p = {}

local function getArgNums(args)
	-- Get the keys of the numerical arguments that are present.
	local nums = {}
	local tinsert = table.insert
	for k, v in pairs(args) do
		if type(k) == 'number' then
			tinsert(nums, k)
		end
	end
	table.sort(nums)
	return nums
end

local function makeToolbarItems(args)
	local nums = getArgNums(args)
	local sep = (args.separator or 'pipe') .. '-separator'
	sep = mw.message.new(sep):plain()
	local ret = {}
	local tinsert = table.insert
	for i, v in ipairs(nums) do
		tinsert(ret, args[v])
	end
	if #ret > 0 then
		return table.concat(ret, sep)
	end
end

function p._main(args)
	local toolbarItems = makeToolbarItems(args)
	if not toolbarItems then return '' end -- Return the blank string if no arguments were specified, rather than returning empty brackets
	return mw.ustring.format(
		'<span class="plainlinks%s"%s>(%s)</span>',
		type(args.class) == 'string' and ' ' .. args.class or '',
		type(args.style) == 'string' and mw.ustring.format(' style="%s"', args.style) or '',
		toolbarItems
	)
end

function p.main(frame)
	-- If called via #invoke, use the args passed into the invoking template.
	-- Otherwise, for testing purposes, assume args are being passed directly in.
	local origArgs
	if frame == mw.getCurrentFrame() then
		origArgs = frame:getParent().args
		for k, v in pairs(frame.args) do
			origArgs = frame.args
			break
		end
	else
		origArgs = frame
	end
	-- Strip whitespace and remove nil values
	local args = {}
	for k, v in pairs(origArgs) do
		if type(v) == 'string' then
			v = mw.text.trim(v)
		end
		if v ~= '' then
			args[k] = v
		end
	end
	return p._main(args)
end

return p