Module:Protection banner

    From Nonbinary Wiki
    Revision as of 01:51, 4 March 2014 by wikipedia>Mr. Stradivarius (expand category matching code - still buggy)

    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'
    	elseif not nskey then
    		nskey = 'all'
    	end
    	return nskey
    end
    
    function p.matchCategory(cats, protType, protLevel, namespace, reason, expiry)
    	-- Preprocess parameters
    	cats = cats or categories
    	protType = protType or 'all'
    	protLevel = protLevel or 'all'
    	namespace = p.parseNamespace(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
    	local keyTable = {protType, protLevel, namespace, reason, expiry}
    	local tryOrder = {5, 4, 3, 2, 1}
    	
    	-- Try successively generic matches until we run out of key combinations
    	local function attemptMatch()
    		local key = table.concat(keyTable, '-')
    		mw.log(key) -- for debugging
    		return cats[key]
    	end
    	local attempt = attemptMatch()
    	if attempt then
    		return attempt
    	end
    	for i, keyTableKey in ipairs(tryOrder) do
    		keyTable[keyTableKey] = 'all'
    		attempt = attemptMatch()
    		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