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

From blackwiki
Jump to navigation Jump to search
blackwiki>Mr. Stradivarius
(try and save a couple of table lookups - more changes to come in a bit)
m (29 revisions imported)
 
(9 intermediate revisions by one other user not shown)
Line 1: Line 1:
 
--[[
 
--[[
----------------------------------------------------------------------------------------------------
+
--------------------------------------------------------------------------------
--                                                                                               --
+
--                                                                           --
--                                           NAMESPACE DETECT                                     --
+
--                           NAMESPACE DETECT                               --
--                                                                                               --
+
--                                                                           --
--     This module implements the {{namespace detect}} template in Lua, with a few              --
+
-- This module implements the {{namespace detect}} template in Lua, with a   --
--     improvements: all namespaces and all namespace aliases are supported, and namespace      --
+
-- few improvements: all namespaces and all namespace aliases are supported, --
--     names are detected automatically for the local wiki. The module can also use the         --
+
-- and namespace names are detected automatically for the local wiki. The     --
--      corresponding subject namespace value if it is used on a talk page. Parameter names       --
+
-- module can also use the corresponding subject namespace value if it is     --
--      can be configured for different wikis by altering the values in the "cfg" table           --
+
-- used on a talk page. Parameter names can be configured for different wikis --
--     in Module:Namespace detect/config.                                                       --
+
-- by altering the values in the "cfg" table in                              --
--                                                                                               --
+
-- Module:Namespace detect/config.                                           --
----------------------------------------------------------------------------------------------------
+
--                                                                           --
 +
--------------------------------------------------------------------------------
 
--]]
 
--]]
  
 
local data = mw.loadData('Module:Namespace detect/data')
 
local data = mw.loadData('Module:Namespace detect/data')
 +
local argKeys = data.argKeys
 
local cfg = data.cfg
 
local cfg = data.cfg
 
local mappings = data.mappings
 
local mappings = data.mappings
  
 
local yesno = require('Module:Yesno')
 
local yesno = require('Module:Yesno')
 +
local mArguments -- Lazily initialise Module:Arguments
 +
local mTableTools -- Lazily initilalise Module:TableTools
 
local ustringLower = mw.ustring.lower
 
local ustringLower = mw.ustring.lower
  
 
local p = {}
 
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 we are over the expensive function count limit.
+
-- 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 noError, pageObject = pcall(mw.title.new, page)
+
local success, pageObject = pcall(mw.title.new, page)
if not noError then
+
if success then
 +
return pageObject
 +
else
 
return nil
 
return nil
else
 
return pageObject
 
 
end
 
end
 
else
 
else
Line 43: Line 71:
  
 
local function getNamespace(args)
 
local function getNamespace(args)
-- Gets the namespace name from the page object.
+
-- This function gets the namespace name from the page object.
local page = args[cfg.page]
+
local page = fetchValue(args, argKeys.demopage)
local demospace = args[cfg.demospace]
+
if page == '' then
local subjectns = args[cfg.subjectns]
+
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 ustringLower(demospace) == cfg.main then
+
if equalsArrayValue(argKeys.main, ustringLower(demospace)) then
 
ret = mw.site.namespaces[0].name
 
ret = mw.site.namespaces[0].name
 
else
 
else
Line 59: Line 93:
 
if pageObject then
 
if pageObject then
 
if pageObject.isTalkPage then
 
if pageObject.isTalkPage then
-- If cfg.subjectns is set, get the subject namespace, otherwise use cfg.talk.
+
-- 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 = cfg.talk
+
ret = 'talk'
 
end
 
end
 
else
 
else
Line 77: Line 112:
  
 
function p._main(args)
 
function p._main(args)
-- Get the namespace to compare the parameters to, and the parameter mapping table.
+
-- 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
-- Check for any matches in the namespace arguments. The order we check them doesn't matter,
+
local params = mappings[namespace] or {}
-- as there can only be one match.
+
local ret = fetchValue(args, params)
for ns, params in pairs(mappings) do
+
--[[
if ns == namespace then
+
-- If there were no matches, return parameters for other namespaces.
-- Check all aliases for matches. The default local namespace is checked first, as
+
-- This happens if there was no text specified for the namespace that
-- {{namespace detect}} checked these before alias names.
+
-- was detected or if the demospace parameter is not a valid
for _, param in ipairs(params) do
+
-- namespace. Note that the parameter for the detected namespace must be
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.
 
-- completely absent for this to happen, not merely blank.
if args[cfg.other] ~= nil then
+
--]]
return args[cfg.other]
+
if ret == nil then
 +
ret = fetchValue(args, argKeys.other)
 
end
 
end
 +
return ret
 
end
 
end
  
 
function p.main(frame)
 
function p.main(frame)
-- If called via #invoke, use the args passed into the invoking template, or the args
+
mArguments = require('Module:Arguments')
-- passed to #invoke if any exist. Otherwise assume args are being passed directly in.
+
local args = mArguments.getArgs(frame, {removeBlanks = false})
local origArgs
+
local ret = p._main(args)
if frame == mw.getCurrentFrame() then
+
return ret or ''
origArgs = frame:getParent().args
+
end
for k, v in pairs(frame.args) do
+
 
origArgs = frame.args
+
function p.table(frame)
break
+
--[[
 +
-- 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
else
 
origArgs = frame
 
 
end
 
end
-- Trim whitespace and remove blank arguments for demospace and page parameters.
+
local nsHeader = checkValue(cfg.wikitableNamespaceHeader, 'Namespace')
local args = {}
+
local aliasesHeader = checkValue(cfg.wikitableAliasesHeader, 'Aliases')
for k, v in pairs(origArgs) do
+
 
if type(v) == 'string' then
+
-- Put the namespaces in order.
v = mw.text.trim(v) -- Trim whitespace.
+
local mappingsOrdered = {}
end
+
for nsname, params in pairs(mappings) do
if k == cfg.demospace or k == cfg.page then
+
if useTalk or nsname ~= 'talk' then
if v ~= '' then
+
local nsid = namespaces[nsname].id
args[k] = v
+
-- Add 1, as the array must start with 1; nsid 0 would be lost otherwise.
end
+
nsid = nsid + 1
else
+
mappingsOrdered[nsid] = params
args[k] = v
 
 
end
 
end
 
end
 
end
return p._main(args)
+
mappingsOrdered = mTableTools.compressSparseArray(mappingsOrdered)
end
 
  
function p.table(frame)
+
-- Build the table.
--[[ 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"'
 
local ret = '{| class="wikitable"'
 
.. '\n|-'
 
.. '\n|-'
.. '\n! ' .. cfg.wikitableNamespaceHeader
+
.. '\n! ' .. nsHeader
.. '\n! ' .. cfg.wikitableAliasesHeader
+
.. '\n! ' .. aliasesHeader
+
for i, params in ipairs(mappingsOrdered) do
-- Generate the row for the main namespace, as we want this to be first in the list.
+
for j, param in ipairs(params) do
ret = ret .. '\n|-'
+
if j == 1 then
.. '\n| <code>' .. cfg.main .. '</code>'
+
ret = ret .. '\n|-'
.. '\n|'
+
.. '\n| <code>' .. param .. '</code>'
if useTalk then
+
.. '\n| '
ret = ret .. '\n|-'
+
elseif j == 2 then
.. '\n| <code>' .. cfg.talk .. '</code>'
+
ret = ret .. '<code>' .. param .. '</code>'
.. '\n|'
+
else
end
+
ret = ret .. ', <code>' .. param .. '</code>'
-- 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
 
end
 
end
-- End the wikitable.
 
 
ret = ret .. '\n|-'
 
ret = ret .. '\n|-'
 
.. '\n|}'
 
.. '\n|}'

Latest revision as of 08:05, 27 September 2020

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

--[[
--------------------------------------------------------------------------------
--                                                                            --
--                            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 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| <code>' .. param .. '</code>'
					.. '\n| '
			elseif j == 2 then
				ret = ret .. '<code>' .. param .. '</code>'
			else
				ret = ret .. ', <code>' .. param .. '</code>'
			end
		end
	end
	ret = ret .. '\n|-'
		.. '\n|}'
	return ret
end

return p