Jump to content

Module:Namespace detect/sandbox

From Wikipedia, the free encyclopedia
This is an old revision of this page, as edited by Mr. Stradivarius (talk | contribs) at 03:28, 21 March 2014 (try and save a couple of table lookups - more changes to come in a bit). The present address (URL) is a permanent link to this revision, which may differ significantly from the current revision.
--[[
----------------------------------------------------------------------------------------------------
--                                                                                                --
--                                           NAMESPACE DETECT                                     --
--                                                                                                --
--      This module implements the {{namespace detect}} template in Lua, with a few               --
--      improvements: all namespaces and all namespace aliases are supported, and namespace       --
--      names are detected automatically for the local wiki. The module can also use the          --
--      corresponding subject namespace value if it is used on a talk page. Parameter names       --
--      can be configured for different wikis by altering the values in the "cfg" table           --
--      in Module:Namespace detect/config.                                                        --
--                                                                                                --
----------------------------------------------------------------------------------------------------
--]]

local data = mw.loadData('Module:Namespace detect/data')
local cfg = data.cfg
local mappings = data.mappings

local yesno = require('Module:Yesno')
local ustringLower = mw.ustring.lower

local p = {}

function p.getPageObject(page)
	-- Get the page object, passing the function through pcall in case we are over the expensive function count limit.
	if page then
		local noError, pageObject = pcall(mw.title.new, page)
		if not noError then
			return nil
		else
			return pageObject
		end
	else
		return mw.title.getCurrentTitle()
	end
end

-- Provided for backward compatibility with other modules
function p.getParamMappings()
	return mappings
end

local function getNamespace(args)
	-- Gets the namespace name from the page object.
	local page = args[cfg.page]
	local demospace = args[cfg.demospace]
	local subjectns = args[cfg.subjectns]
	local ret
	if demospace then
		-- Handle "demospace = main" properly.
		if ustringLower(demospace) == cfg.main then
			ret = mw.site.namespaces[0].name
		else
			ret = demospace
		end
	else
		local pageObject = p.getPageObject(page)
		if pageObject then
			if pageObject.isTalkPage then
				-- If cfg.subjectns is set, get the subject namespace, otherwise use cfg.talk.
				if yesno(subjectns) then
					ret = mw.site.namespaces[pageObject.namespace].subject.name
				else
					ret = cfg.talk
				end
			else
				ret = pageObject.nsText
			end
		else
			return nil -- return nil if the page object doesn't exist.
		end
	end
	ret = ret:gsub('_', ' ')
	return ustringLower(ret)
end

function p._main(args)
	-- Get the namespace to compare the parameters to, and the parameter mapping table.
	local namespace = getNamespace(args)
	-- Check for any matches in the namespace arguments. The order we check them doesn't matter,
	-- as there can only be one match.
	for ns, params in pairs(mappings) do
		if ns == namespace then
			-- Check all aliases for matches. The default local namespace is checked first, as
			-- {{namespace detect}} checked these before alias names.
			for _, param in ipairs(params) do
				if args[param] ~= nil then
					return args[param]
				end
			end
		end
	end
	-- If there were no matches, return parameters for other namespaces. This happens if there
	-- was no text specified for the namespace that was detected or if the demospace parameter
	-- is not a valid namespace. Note that the parameter for the detected namespace must be
	-- completely absent for this to happen, not merely blank.
	if args[cfg.other] ~= nil then
		return args[cfg.other]
	end
end

function p.main(frame)
	-- If called via #invoke, use the args passed into the invoking template, or the args
	-- passed to #invoke if any exist. Otherwise 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
	-- Trim whitespace and remove blank arguments for demospace and page parameters.
	local args = {}
	for k, v in pairs(origArgs) do
		if type(v) == 'string' then
			v = mw.text.trim(v) -- Trim whitespace.
		end
		if k == cfg.demospace or k == cfg.page then
			if v ~= '' then
				args[k] = v
			end
		else
			args[k] = v
		end
	end
	return p._main(args)
end

function p.table(frame)
	--[[ Create a wikitable of all subject namespace parameters, for documentation purposes. The talk 
	  parameter is optional, in case it needs to be excluded in the documentation.
	]]
	local useTalk = type(frame) == 'table' and type(frame.args) == 'table' and frame.args.talk == 'yes' -- Whether to use the talk parameter.
	-- Start the wikitable.
	local ret = '{| class="wikitable"'
		.. '\n|-'
		.. '\n! ' .. cfg.wikitableNamespaceHeader
		.. '\n! ' .. cfg.wikitableAliasesHeader
	
	-- Generate the row for the main namespace, as we want this to be first in the list.
	ret = ret .. '\n|-'
		.. '\n| <code>' .. cfg.main .. '</code>'
		.. '\n|'
	if useTalk then
		ret = ret .. '\n|-'
			.. '\n| <code>' .. cfg.talk .. '</code>'
			.. '\n|'
	end
	-- Generate the other wikitable rows.
	for ns, params in pairs(mappings) do
		if ns ~= mw.site.namespaces[0].name then -- Ignore the main namespace.
			for i, param in ipairs(params) do
				if i == 1 then
					ret = ret .. '\n|-'
						.. '\n| <code>' .. param .. '</code>'
						.. '\n| '
				elseif i == 2 then
					ret = ret .. '<code>' .. param .. '</code>'
				else
					ret = ret .. ', <code>' .. param .. '</code>'
				end
			end
		end
	end
	-- End the wikitable.
	ret = ret .. '\n|-'
		.. '\n|}'
	return ret
end

return p