Difference between revisions of "Module:Namespace detect"
Shawndouglas (talk | contribs)  (Created as needed.)  | 
				Shawndouglas (talk | contribs)  m (Updated code)  | 
				||
| (6 intermediate revisions by the same user not shown) | |||
| Line 1: | Line 1: | ||
--  | --[[  | ||
--   | --------------------------------------------------------------------------------  | ||
--   | --                                                                            --  | ||
--   | --                            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 argKeys = data.argKeys  | |||
local cfg = data.cfg  | |||
local mappings = data.mappings  | |||
local   | local yesno = require('Module:Yesno')  | ||
local mArguments -- Lazily initialise Module:Arguments  | |||
local mTableTools -- Lazily initilalise Module:TableTools  | |||
local ustringLower = mw.ustring.lower  | |||
local p = {}  | |||
--   | local function fetchValue(t1, t2)  | ||
	-- Fetches a value from the table t1 for the first key in array t2 where  | |||
	-- a non-nil value of t1 exists.  | |||
	for i, key in ipairs(t2) do  | |||
		local value = t1[key]  | |||
		if value ~= nil then  | |||
			return value  | |||
		end  | |||
	end  | |||
	return nil  | |||
end  | |||
local function equalsArrayValue(t, value)  | |||
	-- Returns true if value equals a value in the array t. Otherwise  | |||
	-- returns false.  | |||
	for i, arrayValue in ipairs(t) do  | |||
--   | 		if value == arrayValue then  | ||
--   | 			return true  | ||
		end  | |||
	end  | |||
	return false  | |||
end  | |||
function p.getPageObject(page)  | function p.getPageObject(page)  | ||
	-- Get the page object, passing the function through pcall in case   | 	-- Get the page object, passing the function through pcall in case of  | ||
	-- errors, e.g. being over the expensive function count limit.  | |||
	if page then  | 	if page then  | ||
		local   | 		local success, pageObject = pcall(mw.title.new, page)  | ||
		if   | 		if success then  | ||
			return pageObject  | |||
		else  | |||
			return nil  | 			return nil  | ||
		end  | 		end  | ||
	else  | 	else  | ||
| Line 67: | Line 65: | ||
end  | end  | ||
-- Provided for backward compatibility with other modules  | |||
function p.getParamMappings()  | function p.getParamMappings()  | ||
	return mappings  | 	return mappings  | ||
end  | end  | ||
local function getNamespace(args)  | local function getNamespace(args)  | ||
	--   | 	-- This function gets the namespace name from the page object.  | ||
	local page = args  | 	local page = fetchValue(args, argKeys.demopage)  | ||
	local demospace = args  | 	if page == '' then  | ||
	local subjectns = args  | 		page = nil  | ||
	end  | |||
	local demospace = fetchValue(args, argKeys.demospace)  | |||
	if demospace == '' then  | |||
		demospace = nil  | |||
	end  | |||
	local subjectns = fetchValue(args, argKeys.subjectns)  | |||
	local ret  | 	local ret  | ||
	if demospace then  | 	if demospace then  | ||
		-- Handle "demospace = main" properly.  | 		-- Handle "demospace = main" properly.  | ||
		if   | 		if equalsArrayValue(argKeys.main, ustringLower(demospace)) then  | ||
			ret = mw.site.namespaces[0].name  | 			ret = mw.site.namespaces[0].name  | ||
		else  | 		else  | ||
| Line 113: | Line 93: | ||
		if pageObject then  | 		if pageObject then  | ||
			if pageObject.isTalkPage then  | 			if pageObject.isTalkPage then  | ||
				--   | 				-- Get the subject namespace if the option is set,  | ||
				-- otherwise use "talk".  | |||
				if yesno(subjectns) then  | 				if yesno(subjectns) then  | ||
					ret = mw.site.namespaces[pageObject.namespace].subject.name  | 					ret = mw.site.namespaces[pageObject.namespace].subject.name  | ||
				else  | 				else  | ||
					ret =   | 					ret = 'talk'  | ||
				end  | 				end  | ||
			else  | 			else  | ||
| Line 126: | Line 107: | ||
		end  | 		end  | ||
	end  | 	end  | ||
	ret =   | 	ret = ret:gsub('_', ' ')  | ||
	return   | 	return ustringLower(ret)  | ||
end  | end  | ||
function p._main(args)  | function p._main(args)  | ||
	--   | 	-- Check the parameters stored in the mappings table for any matches.  | ||
	local namespace = getNamespace(args)  | 	local namespace = getNamespace(args) or 'other' -- "other" avoids nil table keys  | ||
	local params = mappings[namespace] or {}  | |||
	local ret = fetchValue(args, params)  | |||
	--[[  | |||
	-- 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  | |||
	-- If there were no matches, return parameters for other namespaces. This happens if there  | |||
	--   | |||
	-- completely absent for this to happen, not merely blank.  | 	-- completely absent for this to happen, not merely blank.  | ||
	if   | 	--]]  | ||
	if ret == nil then  | |||
		ret = fetchValue(args, argKeys.other)  | |||
	end  | 	end  | ||
	return ret  | |||
end  | end  | ||
function p.main(frame)  | function p.main(frame)  | ||
	--   | 	mArguments = require('Module:Arguments')  | ||
	--   | 	local args = mArguments.getArgs(frame, {removeBlanks = false})  | ||
	local ret = p._main(args)  | |||
	return ret or ''  | |||
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.  | |||
	--]]  | |||
	-- Load modules and initialise variables.  | |||
	mTableTools = require('Module:TableTools')  | |||
	local namespaces = mw.site.namespaces  | |||
	local cfg = data.cfg  | |||
	local useTalk = type(frame) == 'table'   | |||
		and type(frame.args) == 'table'   | |||
		and yesno(frame.args.talk) -- Whether to use the talk parameter.  | |||
	-- Get the header names.  | |||
	local function checkValue(value, default)  | |||
		if type(value) == 'string' then  | |||
			return value  | |||
		else  | |||
			return default  | |||
		end  | 		end  | ||
	end  | 	end  | ||
	--   | 	local nsHeader = checkValue(cfg.wikitableNamespaceHeader, 'Namespace')  | ||
	local   | 	local aliasesHeader = checkValue(cfg.wikitableAliasesHeader, 'Aliases')  | ||
	for   | |||
		if   | 	-- Put the namespaces in order.  | ||
	local mappingsOrdered = {}  | |||
	for nsname, params in pairs(mappings) do  | |||
		if useTalk or nsname ~= 'talk' then  | |||
			local nsid = namespaces[nsname].id  | |||
			-- Add 1, as the array must start with 1; nsid 0 would be lost otherwise.  | |||
			nsid = nsid + 1   | |||
			mappingsOrdered[nsid] = params  | |||
		end  | 		end  | ||
	end  | 	end  | ||
	mappingsOrdered = mTableTools.compressSparseArray(mappingsOrdered)  | |||
	-- Build the table.  | |||
	--  | |||
	local ret = '{| class="wikitable"'  | 	local ret = '{| class="wikitable"'  | ||
		.. '\n|-'  | 		.. '\n|-'  | ||
		.. '\n! ' ..   | 		.. '\n! ' .. nsHeader  | ||
		.. '\n! ' ..   | 		.. '\n! ' .. aliasesHeader  | ||
	for i, params in ipairs(mappingsOrdered) do  | |||
		for j, param in ipairs(params) do  | |||
			if j == 1 then  | |||
				ret = ret .. '\n|-'  | |||
					.. '\n| <code>' .. param .. '</code>'  | |||
					.. '\n| '  | |||
			elseif j == 2 then  | |||
				ret = ret .. '<code>' .. param .. '</code>'  | |||
			else  | |||
				ret = ret .. ', <code>' .. param .. '</code>'  | |||
			end  | 			end  | ||
		end  | 		end  | ||
	end  | 	end  | ||
	ret = ret .. '\n|-'  | 	ret = ret .. '\n|-'  | ||
		.. '\n|}'  | 		.. '\n|}'  | ||
| Line 230: | Line 198: | ||
return p  | return p  | ||
Latest revision as of 16:31, 22 November 2014
--[[
-- -- -- NAMESPACE DETECT -- -- -- -- This module implements the 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 argKeys = data.argKeys local cfg = data.cfg local mappings = data.mappings
local yesno = require('Module:Yesno') local mArguments -- Lazily initialise Module:Arguments local mTableTools -- Lazily initilalise Module:TableTools local ustringLower = mw.ustring.lower
local p = {}
local function fetchValue(t1, t2) -- Fetches a value from the table t1 for the first key in array t2 where -- a non-nil value of t1 exists. for i, key in ipairs(t2) do local value = t1[key] if value ~= nil then return value end end return nil end
local function equalsArrayValue(t, value) -- Returns true if value equals a value in the array t. Otherwise -- returns false. for i, arrayValue in ipairs(t) do if value == arrayValue then return true end end return false end
function p.getPageObject(page) -- Get the page object, passing the function through pcall in case of -- errors, e.g. being over the expensive function count limit. if page then local success, pageObject = pcall(mw.title.new, page) if success then return pageObject else return nil 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) -- This function gets the namespace name from the page object. local page = fetchValue(args, argKeys.demopage) if page == then page = nil end local demospace = fetchValue(args, argKeys.demospace) if demospace == then demospace = nil end local subjectns = fetchValue(args, argKeys.subjectns) local ret if demospace then -- Handle "demospace = main" properly. if equalsArrayValue(argKeys.main, ustringLower(demospace)) then ret = mw.site.namespaces[0].name else ret = demospace end else local pageObject = p.getPageObject(page) if pageObject then if pageObject.isTalkPage then -- Get the subject namespace if the option is set, -- otherwise use "talk". if yesno(subjectns) then ret = mw.site.namespaces[pageObject.namespace].subject.name else ret = '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) -- Check the parameters stored in the mappings table for any matches. local namespace = getNamespace(args) or 'other' -- "other" avoids nil table keys local params = mappings[namespace] or {} local ret = fetchValue(args, params) --[[ -- 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 ret == nil then ret = fetchValue(args, argKeys.other) end return ret end
function p.main(frame) mArguments = require('Module:Arguments') local args = mArguments.getArgs(frame, {removeBlanks = false}) local ret = p._main(args) return ret or 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. --]]
-- Load modules and initialise variables. mTableTools = require('Module:TableTools') local namespaces = mw.site.namespaces local cfg = data.cfg local useTalk = type(frame) == 'table' and type(frame.args) == 'table' and yesno(frame.args.talk) -- Whether to use the talk parameter.
-- Get the header names. local function checkValue(value, default) if type(value) == 'string' then return value else return default end end local nsHeader = checkValue(cfg.wikitableNamespaceHeader, 'Namespace') local aliasesHeader = checkValue(cfg.wikitableAliasesHeader, 'Aliases')
-- Put the namespaces in order. local mappingsOrdered = {} for nsname, params in pairs(mappings) do if useTalk or nsname ~= 'talk' then local nsid = namespaces[nsname].id -- Add 1, as the array must start with 1; nsid 0 would be lost otherwise. nsid = nsid + 1 mappingsOrdered[nsid] = params end end mappingsOrdered = mTableTools.compressSparseArray(mappingsOrdered)
	-- Build the table.
	local ret = '{| class="wikitable"'
		.. '\n|-'
		.. '\n! ' .. nsHeader
		.. '\n! ' .. aliasesHeader
	for i, params in ipairs(mappingsOrdered) do
		for j, param in ipairs(params) do
			if j == 1 then
				ret = ret .. '\n|-'
					.. '\n| ' .. param .. ''
					.. '\n| '
			elseif j == 2 then
				ret = ret .. '' .. param .. ''
			else
				ret = ret .. ', ' .. param .. ''
			end
		end
	end
	ret = ret .. '\n|-'
		.. '\n|}'
	return ret
end
return p







