Jump to content

Module:Parameter names example/sandbox

From Wikipedia, the free encyclopedia
This is an old revision of this page, as edited by Mr. Stradivarius (talk | contribs) at 12:47, 29 August 2014 (fix bug where we tried to use non-existent args). The present address (URL) is a permanent link to this revision, which may differ significantly from the current revision.
-- This module implements {{parameter names example}}.

local p = {}

local function makeParam(s)
	local lb = '{'
	local rb = '}'
	return lb:rep(3) .. s .. rb:rep(3)
end

local function italicize(s)
	return "''" .. s .. "''"
end

function p._main(origArgs, frame)
	frame = frame or mw.getCurrentFrame()

	-- Separate the args and the options.
	local args, options = {}, {}
	for k, v in pairs(origArgs) do
		if type(k) == 'number' or not k:find('^_') then
			args[k] = v
		else
			options[k:gsub('^_', '')] = v
		end
	end

	-- Find how we want to format the arguments to the template.
	local formatFunc
	if options.display == 'italics' or options.display == 'italic' then
		formatFunc = italicize
	else
		formatFunc = makeParam
	end

	-- Build the table of template arguments, and record whether any named
	-- arguments are being used.
	local targs = {}
	local hasNamedArgs = false
	for k, v in pairs(args) do
		if type(k) == 'number' then
			targs[v] = formatFunc(v)
		else
			hasNamedArgs = true
			targs[k] = v
		end
	end

	-- Find the template name.
	local templateName
	if options.template then
		templateName = options.template
	else
		-- Find the title object for the template.
		local templateTitle
		local currentTitle = mw.title.getCurrentTitle()
		if currentTitle.prefixedText:find('/sandbox$') then
			templateTitle = currentTitle
		else
			templateTitle = currentTitle.basePageTitle
		end

		-- Format the template name according to the namespace we are in.
		if templateTitle.namespace == 10 then -- NS_TEMPLATE
			templateName = templateTitle.text
		elseif templateTitle.namespace == 0 then -- NS_MAIN
			templateName = ':' .. templateTitle.text
		else
			templateName = templateTitle.prefixedText
		end
	end

	local ret = ''

	-- Build the template invocation.
	if options.code then
		-- Order the arguments. If we have named arguments this is alphabetical
		-- order, otherwise it's the order that they were passed into the
		-- template.
		local keys = {}
		local displayArgs = hasNamedArgs and targs or args
		for k in pairs(displayArgs) do
			keys[#keys + 1] = k
		end
		table.sort(keys, function (a, b)
			local typeA = type(a)
			local typeB = type(b)
			if typeA ~= typeB then
				return typeA == 'number' -- numbers go first
			else
				return a < b
			end
		end)

		-- Assemble the invocation.
		local code = { templateName }
		for _, k in ipairs(keys) do
			code[#code + 1] = ' ' .. tostring(k) .. ' = ' .. displayArgs[k]
		end
		code = '{{' .. table.concat(code, '\n|') .. '\n}}'
		code = '<pre style="white-space: pre-wrap;">\n' .. code .. '\n</pre>\n\n'
		ret = ret .. frame:preprocess(code)
	end

	-- Call the template with the arguments.
	local success, result = pcall(
		frame.expandTemplate,
		frame,
		{title = templateName, args = targs}
	)
	if success then
		ret = ret .. result
	end

	return ret
end

function p.main(frame)
	local args = require('Module:Arguments').getArgs(frame, {
		wrappers = 'Template:Parameter names example'
	})
	return p._main(args, frame)
end

return p