Difference between revisions of "Module:Protection banner"

From Amanita Research
Jump to navigation Jump to search
(convert categories back to a flat table scheme)
(add namespace parser)
Line 43: Line 43:
 
['pc-autoconfirmed-all-all-all'] = 'Wikipedia pending changes protected pages (level 1)',
 
['pc-autoconfirmed-all-all-all'] = 'Wikipedia pending changes protected pages (level 1)',
 
['pc-reviewer-all-all-all'] = 'Wikipedia pending changes protected pages (level 2)',
 
['pc-reviewer-all-all-all'] = 'Wikipedia pending changes protected pages (level 2)',
 +
}
 +
 +
local nskeys = {
 +
[2] = 'user',
 +
[3] = 'user',
 +
[4] = 'project',
 +
[6] = 'file',
 +
[10] = 'template',
 +
[12] = 'project',
 +
[14] = 'category',
 +
[100] = 'portal',
 
}
 
}
  
Line 95: Line 106:
 
end
 
end
  
function p.matchCategory(cats, protType, protLevel, namespace, expiry, reason)
+
function p.parseNamespace(ns)
local function isValidCat(t)
+
-- The ns variable is the namespace number.
--[[
+
if not ns or type(ns) ~= 'number' then
-- Does a first parse over the categories, removing any that explicitly
+
return 'all'
-- name a wrong parameter. This accounts for the majority of all of the
 
-- categories.
 
--
 
-- We check the parameters most likely to be wrong first. This saves on
 
-- table lookups, which shoud make the function faster.
 
--]]
 
local testType, testLevel, testNs, testExpiry, testReason
 
testReason = t.reason
 
if testReason and testReason ~= reason then
 
return false
 
end
 
testNs = t.namespace
 
if testNs and not testNs[namespace] then
 
return false
 
end
 
testExpiry = t.expiry
 
if testExpiry and testExpiry ~= expiry then
 
return false
 
end
 
testLevel = t.level
 
if testLevel and testLevel ~= protLevel then
 
return false
 
end
 
testType = t.type
 
if testType and testType ~= protType then
 
return false
 
end
 
return true
 
 
end
 
end
 
+
local nskey = nskeys[ns]
-- Remove invalid cats.
+
if not nskey and ns % 2 == 1 then
local sortCats = {}
+
nskey = 'talk'
for i, t in ipairs(cats) do
+
else
if isValidCat(t) then
+
nskey = 'all'
sortCats[#sortCats + 1] = t
 
end
 
 
end
 
end
 +
return nskey
 +
end
  
-- Sort the remaining cats.
+
function p.matchCategory(cats, protType, protLevel, namespace, expiry, reason)
-- The categories which are left either explicitly name the right parameter,
+
local nskey = p.parseNamespace(namespace)
-- or implicitly allow all values for that parameter.
 
table.sort(sortCats, function (t1, t2)
 
local type1 = t1.type
 
local type2 = t2.type
 
local level1 = t1.level
 
local level2 = t2.level
 
local ns1 = t1.namespace
 
local ns2 = t2.namespace
 
local exp1 = t1.expiry
 
local exp2 = t2.expiry
 
local reason1 = t1.reason
 
local reason2 = t2.reason
 
 
 
local score1 = 0
 
local score2 = 0
 
end)
 
 
end
 
end
  
 
return p
 
return p

Revision as of 23:48, 3 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 = {
	['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-user-all-all'] = 'Wikipedia move-protected project pages',
	['move-sysop-portal-all-all'] = 'Wikipedia move-protected portals',
	['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-template-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-indef'] = 'Wikipedia indefinitely semi-protected biographies of living people',
	['edit-sysop-all-blp-indef'] = 'Wikipedia indefinitely protected biographies of living people',
	['edit-autoconfirmed-all-all-indef'] = 'Wikipedia indefinitely semi-protected pages',
	['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',
	['all-all-all-all-all'] = 'Wikipedia protected pages',
	['edit-autoconfirmed-category-all-all'] = 'Wikipedia semi-protected categories',
	['edit-sysop-all-all-all'] = 'Wikipedia protected categories',
	['move-sysop-all-all-all'] = 'Wikipedia move-protected pages',
	['all-all-all-office-all'] = 'Wikipedia Office-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)',
}

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

local error_categories = {
	incorrect = 'Wikipedia pages with incorrect protection templates',
	no_expiry = 'Wikipedia protected pages without expiry',
	create = 'Wikipedia pages tagged as create-protected'
}

local template_types = {
	vandalism = {
		namespaces = false, -- defaults to all namespaces
		editLevels = false, -- all edit protection levels
		moveLevels = false, -- all move protection levels
		createLevels = false, -- all create protection levels
		header = false,
		reason = 'due to [[Wikipedia:Vandalism|vandalism]]',
		icon_reason = 'due to vandalism',
	}
}

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

local getArgs = require('Module:Arguments').getArgs
local yesno = require('Module:Yesno')

local p = {}

function p.main(frame)
	-- Get data
	local args = getArgs(frame)
	local title = p.getTitle(args.page)
	local protType, protLevel = p.getProtectionStatus(title)
	local namespace = p.getNamespace(title)
	local expiry = p.getExpiry(args.expiry)
	local reason = p.getReason(args.reason)

	-- Sort categories and build a category link with the best match
	categories = p.sortCategories(categories, protType, protLevel, namespace, expiry, reason)
	local category = categories[1].name
	category = string.format('[[Category:%s]]', category)

	-- Find whether we are small or not and output a padlock or a banner as appropriate
	local isSmall = yesno(args.small, true)
	if isSmall then
		return p.exportPadlock()
	else
		return p.exportBanner()
	end
end

function p.parseNamespace(ns)
	-- The ns variable is the namespace number.
	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'
	else
		nskey = 'all'
	end
	return nskey
end

function p.matchCategory(cats, protType, protLevel, namespace, expiry, reason)
	local nskey = p.parseNamespace(namespace)
end			

return p