Difference between revisions of "Module:Protection banner"

Revision as of 08:17, 11 March 2014

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 behaviors = {
	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 floor = math.floor

local p = {}

function p.matchNamespace(ns)
	-- Matches a namespace number to a string that can be passed to the
	-- namespace parameter of p.getCategoryName.
	if not ns or type(ns) ~= 'number' then
		return 'all'
	end
	local nskey = nskeys[ns]
	if not nskey then
		if ns % 2 == 1 then
			nskey = 'talk'
		else
			nskey = 'all'
		end
	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

	local properties = {expiry, namespace, protLevel, protType, reason}
	local behavior = behaviors[reason]
	if behavior == 'namespaceFirst' then
		tinsert(properties, table.remove(properties, 2)) -- move namespace to the end
	elseif behavior ~= 'reasonFirst' and reason ~= 'all' then
		error(reason .. ' is not a valid reason')
	end

	local activeProperties = {}
	for i, property in ipairs(properties) do
		if property ~= 'all' then
			activeProperties[#activeProperties + 1] = property
		end
	end
	local noActiveProperties = #activeProperties
	
	-- Try successively generic matches until we run out of key combinations
	for i = 1, 2^noActiveProperties do
		local key = {}
		for pos = 1, 5 do
			if pos > noActiveProperties then
				key[pos] = 'all'
			else
				local quotient = i / 2 ^ (pos - 1)
				quotient = floor(quotient)
				if quotient % 2 == 1 then
					key[pos] = protectionProperties[j]
				else -- quotient % 2 == 0
					key[pos] = 'all'
				end
			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

@@ Line 59: / Line 59: @@
 }
-local reasonBehaviors = {
+local behaviors = {
 	vandalism = 'namespaceFirst',
 	dispute = 'reasonFirst',
@@ Line 83: / Line 83: @@
 -- Define often-used functions as local variables.
 local tconcat = table.concat
+local floor = math.floor
 local p = {}
@@ Line 119: / Line 120: @@
 		expiry = 'temp'
 	end
-	--[[
+	local properties = {expiry, namespace, protLevel, protType, reason}
-	-- Define the key table and the order to test it in.
+	local behavior = behaviors[reason]
-	-- The idiosyncracies of the old template system have been distilled down to
+	if behavior == 'namespaceFirst' then
-	-- two basic behaviors - templates that prioritise matching the reason over
+		tinsert(properties, table.remove(properties, 2)) -- move namespace to the end
-	-- matching the namespace, and templates that prioritise matching the
+	elseif behavior ~= 'reasonFirst' and reason ~= 'all' then
-	-- 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
+	local activeProperties = {}
+	for i, property in ipairs(properties) do
+		if property ~= 'all' then
+			activeProperties[#activeProperties + 1] = property
+		end
+	end
+	local noActiveProperties = #activeProperties
 	-- Try successively generic matches until we run out of key combinations
-	for i, t in ipairs(attemptOrder) do
+	for i = 1, 2^noActiveProperties do
 		local key = {}
-		for j, useProperty in ipairs(t) do
+		for pos = 1, 5 do
-			if useProperty then
+			if pos > noActiveProperties then
-				key[j] = protectionProperties[j]
+				key[pos] = 'all'
 			else
-				key[j] = 'all'
+				local quotient = i / 2 ^ (pos - 1)
+				quotient = floor(quotient)
+				if quotient % 2 == 1 then
+					key[pos] = protectionProperties[j]
+				else -- quotient % 2 == 0
+					key[pos] = 'all'
+				end
 			end
 		end
@@ Line 199: / Line 161: @@
 	end
 	error('No category match found; please define the category for key "all-all-all-all-all"')
 end
 return p