查看“模块:Toolbar”的源代码

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