Difference between revisions of "Module:Namespace detect/data"

From MINR.ORG WIKI
(avoid using local variables to save table lookups per Jackmcbarn's suggestion, and because this will be cached with mw.loadData so performance of this function is not such a worry)
(Undid revision 601018314 by Mr. Stradivarius (talk) looks like that broke something)
Line 52: Line 52:
 
-- }
 
-- }
 
--]]
 
--]]
 +
local ustringLower = mw.ustring.lower
 +
local tinsert = table.insert
 +
local clone = mw.clone
 +
local subjectNamespaces = mw.site.subjectNamespaces
 
local mappings = {}
 
local mappings = {}
local mainNsName = mw.site.subjectNamespaces[0].name
+
mappings[ustringLower(subjectNamespaces[0].name)] = clone(argKeys.main)
mainNsName = mw.ustring.lower(mainNsName)
+
mappings['talk'] = clone(argKeys.talk)
mappings[mainNsName] = mw.clone(argKeys.main)
+
for nsid, ns in pairs(subjectNamespaces) do
mappings['talk'] = mw.clone(argKeys.talk)
 
for nsid, ns in pairs(mw.site.subjectNamespaces) do
 
 
if nsid ~= 0 then -- Exclude main namespace.
 
if nsid ~= 0 then -- Exclude main namespace.
local nsname = ns.name
+
local nsname = ustringLower(ns.name)
local canonicalName = ns.canonicalName
+
local canonicalName = ustringLower(ns.canonicalName)
mappings[nsname] = {mw.ustring.lower(nsname)}
+
mappings[nsname] = {nsname}
 
if canonicalName ~= nsname then
 
if canonicalName ~= nsname then
table.insert(mappings[nsname], mw.ustring.lower(canonicalName))
+
tinsert(mappings[nsname], canonicalName)
 
end
 
end
 
for _, alias in ipairs(ns.aliases) do
 
for _, alias in ipairs(ns.aliases) do
table.insert(mappings[nsname], mw.ustring.lower(alias))
+
tinsert(mappings[nsname], ustringLower(alias))
 
end
 
end
 
end
 
end

Revision as of 12:37, 24 March 2014

Documentation for this module may be created at Module:Namespace detect/data/doc

--------------------------------------------------------------------------------
--                          Namespace detect data                             --
-- This module holds data for [[Module:Namespace detect]] to be loaded per    --
-- page, rather than per #invoke, for performance reasons.                    --
--------------------------------------------------------------------------------

local cfg = require('Module:Namespace detect/config')

local function addKey(t, key, defaultKey)
	if key ~= defaultKey then
		t[#t + 1] = key
	end
end

-- Get a table of parameters to query for each default parameter name.
-- This allows wikis to customise parameter names in the cfg table while
-- ensuring that default parameter names will always work. The cfg table
-- values can be added as a string, or as an array of strings.

local argKeys = {
	main = {'main'},
	talk = {'talk'},
	other = {'other'},
	subjectns = {'subjectns'},
	demospace = {'demospace'},
	page = {'page'}
}

for defaultKey, t in pairs(argKeys) do
	local cfgValue = cfg[defaultKey]
	local cfgValueType = type(cfgValue)
	if cfgValueType == 'string' then
		addKey(t, cfgValue, defaultKey)
	elseif cfgValueType == 'table' then
		for i, key in ipairs(cfgValue) do
			addKey(t, key, defaultKey)
		end
	end
	cfg[defaultKey] = nil -- Free the cfg value as we don't need it any more.
end

local function getParamMappings()
	--[[
	-- Returns a table of how parameter names map to namespace names. The keys
	-- are the actual namespace names, in lower case, and the values are the
	-- possible parameter names for that namespace, also in lower case. The
	-- table entries are structured like this:
	-- {
	--   [''] = {'main'},
	--   ['wikipedia'] = {'wikipedia', 'project', 'wp'},
	--   ...
	-- }
	--]]
	local ustringLower = mw.ustring.lower
	local tinsert = table.insert
	local clone = mw.clone
	local subjectNamespaces = mw.site.subjectNamespaces
	local mappings = {}
	mappings[ustringLower(subjectNamespaces[0].name)] = clone(argKeys.main)
	mappings['talk'] = clone(argKeys.talk)
	for nsid, ns in pairs(subjectNamespaces) do
		if nsid ~= 0 then -- Exclude main namespace.
			local nsname = ustringLower(ns.name)
			local canonicalName = ustringLower(ns.canonicalName)
			mappings[nsname] = {nsname}
			if canonicalName ~= nsname then
				tinsert(mappings[nsname], canonicalName)
			end
			for _, alias in ipairs(ns.aliases) do
				tinsert(mappings[nsname], ustringLower(alias))
			end
		end
	end
	return mappings
end

return {
	argKeys = argKeys,
	cfg = cfg,
	mappings = getParamMappings()
}