Module:Protection banner

From Amanita Research
Revision as of 14:47, 10 March 2014 by wikipedia>Mr. Stradivarius (tidy this up and comment it so it will be easier for others to understand)
Jump to navigation Jump to search

Documentation for this module may be created at Module:Protection banner/doc

-- This module implements {{pp-meta}} and its daughter templates such as
-- {{pp-dispute}}, {{pp-vandalism}} and {{pp-sock}}.

local categories = {
	-- The key strings follow this format:
	-- type, level, ns, reason, expiry
	['edit-autoconfirmed-user-all-all'] = 'Wikipedia semi-protected user and user talk pages',
	['edit-autoconfirmed-project-all-all'] = 'Semi-protected project pages',
	['edit-autoconfirmed-file-all-all'] = 'Semi-protected images',
	['edit-autoconfirmed-template-all-all'] = 'Wikipedia semi-protected templates',
	['edit-autoconfirmed-portal-all-all'] = 'Semi-protected portals',
	['edit-autoconfirmed-talk-all-all'] = 'Semi-protected talk pages',
	['edit-autoconfirmed-all-vandalism-all'] = 'Wikipedia pages semi-protected against vandalism',
	['edit-sysop-user-all-all'] = 'Wikipedia protected user and user talk pages',
	['edit-sysop-file-all-all'] = 'Protected images',
	['edit-sysop-project-all-all'] = 'Protected project pages',
	['edit-sysop-template-all-all'] = 'Wikipedia protected templates',
	['edit-sysop-talk-all-all'] = 'Protected talk pages',
	['edit-sysop-all-vandalism-all'] = 'Wikipedia pages protected against vandalism',
	['edit-autoconfirmed-all-dispute-all'] = 'Wikipedia pages semi-protected due to dispute',
	['edit-sysop-all-dispute-all'] = 'Wikipedia pages protected due to dispute',
	['move-sysop-all-dispute-all'] = 'Wikipedia pages move-protected due to dispute',
	['move-sysop-user-all-all'] = 'Wikipedia move-protected user and user talk pages',
	['move-sysop-project-all-all'] = 'Wikipedia move-protected project pages',
	['move-sysop-portal-all-all'] = 'Wikipedia move-protected portals',
	['move-sysop-all-vandalism-all'] = 'Wikipedia pages move-protected due to vandalism',
	['edit-autoconfirmed-template-all-all'] = 'Wikipedia semi-protected templates',
	['move-sysop-template-all-all'] = 'Wikipedia move-protected templates',
	['edit-all-template-all-all'] = 'Wikipedia protected templates',
	['move-sysop-portal-all-all'] = 'Wikipedia move-protected portals',
	['edit-autoconfirmed-all-sock-all'] = 'Wikipedia pages semi-protected from banned users',
	['edit-sysop-all-sock-all'] = 'Wikipedia pages protected from banned users',
	['edit-autoconfirmed-all-blp-temp'] = 'Wikipedia temporarily semi-protected biographies of living people',
	['edit-sysop-all-blp-temp'] = 'Wikipedia temporarily protected biographies of living people',
	['edit-autoconfirmed-all-blp-all'] = 'Wikipedia indefinitely semi-protected biographies of living people',
	['edit-sysop-all-blp-all'] = 'Wikipedia indefinitely protected biographies of living people',
	['edit-autoconfirmed-all-all-indef'] = 'Wikipedia indefinitely semi-protected pages',
	['edit-autoconfirmed-category-all-all'] = 'Wikipedia semi-protected categories',
	['edit-sysop-category-all-all'] = 'Wikipedia protected categories',
	['move-sysop-talk-all-all'] = 'Wikipedia move-protected talk pages',
	['move-sysop-all-all-indef'] = 'Wikipedia indefinitely move-protected pages',
	['edit-autoconfirmed-all-all-all'] = 'Wikipedia semi-protected pages',
	['move-sysop-all-all-all'] = 'Wikipedia move-protected pages',
	['pc-autoconfirmed-all-all-all'] = 'Wikipedia pending changes protected pages (level 1)',
	['pc-reviewer-all-all-all'] = 'Wikipedia pending changes protected pages (level 2)',
	['all-all-all-office-all'] = 'Wikipedia Office-protected pages',
	['all-all-all-all-all'] = 'Wikipedia protected pages',
}

local nskeys = {
	[2] = 'user',
	[3] = 'user',
	[4] = 'project',
	[6] = 'file',
	[10] = 'template',
	[12] = 'project',
	[14] = 'category',
	[100] = 'portal',
}

local reasonBehaviors = {
	vandalism = 'namespaceFirst',
	dispute = 'reasonFirst',
	blp = 'reasonFirst',
	sock = 'reasonFirst',
	office = 'reasonFirst',
}

--[[
-- Not currently used
local error_categories = {
	incorrect = 'Wikipedia pages with incorrect protection templates',
	no_expiry = 'Wikipedia protected pages without expiry',
	create = 'Wikipedia pages tagged as create-protected',
	template = 'Wikipedia template-protected pages other than templates and modules'
}
--]]

--------------------------------------------------------------------------------
-- Main functions
--------------------------------------------------------------------------------

-- Define often-used functions as local variables.
local tconcat = table.concat

local p = {}

function p.matchNamespace(ns)
	-- Matches a namespace number to a string that can be passed to the
	-- namespacep parameter of p.getCategoryName.
	if not ns or type(ns) ~= 'number' then
		return 'all'
	end
	local nskey = nskeys[ns]
	if not nskey and ns % 2 == 1 then
		nskey = 'talk'
	elseif not nskey then
		nskey = 'all'
	end
	return nskey
end

function p.getCategoryName(cats, protType, protLevel, namespace, reason, expiry)
	-- Gets a category name from the category table, given a combination of
	-- the protection type, the protection level, the namespace number, the
	-- reason for protection, and the expiry date.
	
	-- Preprocess parameters
	cats = cats or categories
	protType = protType or 'all'
	protLevel = protLevel or 'all'
	namespace = p.matchNamespace(namespace)
	reason = reason or 'all'
	if not expiry then
		expiry = 'all'
	elseif expiry ~= 'indef' then
		expiry = 'temp'
	end
	
	--[[
	-- Define the key table and the order to test it in.
	-- The idiosyncracies of the old template system have been distilled down to
	-- two basic behaviors - templates that prioritise matching the reason over
	-- matching the namespace, and templates that prioritise matching the
	-- namespace over matching the reason. In each case the function tries to
	-- match specific keys, and if there is no match, it proceeds to match
	-- increasingly generic keys, until eventually it will match the key
	-- "all-all-all-all-all".
	--]]
	local protectionProperties = {protType, protLevel, namespace, reason, expiry}
	local reasonBehavior = reasonBehaviors[reason]
	local attemptOrder 
	if reasonBehavior == 'namespaceFirst' then
		attemptOrder = {
			-- type, level, ns, reason, expiry
			-- true corresponds to the original protectionProperties key,
			-- and false corresponds to the key "all".
			{true, true, true, true, true},     -- 1
			{true, true, true, true, false},    -- 2
			{true, true, true, false, true},    -- 3
			{true, true, true, false, false},   -- 4
			{true, false, true, true, true},    -- 5
			{true, false, true, true, false},   -- 6
			{true, false, true, false, true},   -- 7
			{true, false, true, false, false},  -- 8
			{false, false, true, false, false}, -- 9
			{true, true, false, true, true},    -- 10
			{true, true, false, true, false},   -- 11
			{true, true, false, false, true},   -- 12
			{true, false, false, true, true},   -- 13
			{true, false, false, true, false},  -- 14
			{true, false, false, false, true},  -- 15
			{true, false, false, false, false}, -- 16
			{false, false, false, false, false} -- 17
		}
	elseif reasonBehavior == 'reasonFirst' or reason == 'all' then
		attemptOrder = {
			-- type, level, ns, reason, expiry
			{true, true, true, true, true},     -- 1
			{true, true, true, true, false},    -- 2
			{true, true, false, true, true},    -- 3
			{true, true, false, true, false},   -- 4
			{true, false, true, true, true},    -- 5
			{true, false, true, true, false},   -- 6
			{true, false, false, true, true},   -- 7
			{true, false, false, true, false},  -- 8
			{false, false, false, true, false}, -- 9
			{true, true, true, false, true},    -- 10
			{true, true, true, false, false},   -- 11
			{true, true, false, false, true},   -- 12
			{true, false, true, false, true},   -- 13
			{true, false, true, false, false},  -- 14
			{true, false, false, false, true},  -- 15
			{true, false, false, false, false}, -- 16
			{false, false, false, false, false} -- 17
		}
	else
		error(reason .. ' is not a valid reason')
	end
	
	-- Try successively generic matches until we run out of key combinations
	for i, t in ipairs(attemptOrder) do
		local key = {}
		for j, useProperty in ipairs(t) do
			if useProperty then
				key[j] = protectionProperties[j]
			else
				key[j] = 'all'
			end
		end
		key = tconcat(key, '-')
		mw.log(key) -- for debugging
		local attempt = cats[key]
		if attempt then
			return attempt
		end
	end
	error('No category match found; please define the category for key "all-all-all-all-all"')
end			

return p