Module:Protection banner: Difference between revisions

    From Nonbinary Wiki
    (change this to something that should work if I can get the order figured out)
    (key components are in the correct order now - property order still a long way off, though)
    Line 84: Line 84:
    local tconcat = table.concat
    local tconcat = table.concat
    local tinsert = table.insert
    local tinsert = table.insert
    local tremove = table.remove
    local floor = math.floor
    local floor = math.floor
    local function toTableEnd(t, pos)
    -- Sends the item at position pos to the end of array t,
    -- and shifts all the other array items down accordingly.
    tinsert(t, tremove(t, pos))
    end


    local p = {}
    local p = {}
    Line 92: Line 99:
    -- namespace parameter of p.getCategoryName.
    -- namespace parameter of p.getCategoryName.
    if not ns or type(ns) ~= 'number' then
    if not ns or type(ns) ~= 'number' then
    return 'all'
    return nil
    end
    end
    local nskey = nskeys[ns]
    local nskey = nskeys[ns]
    if not nskey then
    if not nskey and ns % 2 == 1 then
    if ns % 2 == 1 then
    nskey = 'talk'
    nskey = 'talk'
    else
    nskey = 'all'
    end
    end
    end
    return nskey
    return nskey
    Line 109: Line 112:
    -- the protection type, the protection level, the namespace number, the
    -- the protection type, the protection level, the namespace number, the
    -- reason for protection, and the expiry date.
    -- reason for protection, and the expiry date.
    -- Preprocess parameters
    cats = cats or categories
    cats = cats or categories
    local properties = {}
     
    properties.protType = protType or 'all'
    local properties = {
    properties.protLevel = protLevel or 'all'
    {pos = 5, val = expiry},
    properties.namespace = p.matchNamespace(namespace)
    {pos = 3, val = namespace},
    properties.reason = reason or 'all'
    {pos = 2, val = protLevel},
    if not expiry then
    {pos = 1, val = protType},
    properties.expiry = 'all'
    {pos = 4, val = reason}
    elseif expiry ~= 'indef' then
    }
    properties.expiry = 'temp'
     
    local behavior = 'reasonFirst'
    if reason then
    behavior = behaviors[reason]
    if behavior == 'namespaceFirst' then
    toTableEnd(properties, 2) -- move namespace to the end
    elseif behavior ~= 'reasonFirst' then
    error(reason .. ' is not a valid reason')
    end
    end
    end


    local order = {'expiry', 'namespace', 'protLevel', 'protType', 'reason'}  
    local active, inactive = {}, {}
    local behavior = behaviors[reason]
    for i, t in ipairs(properties) do
    if behavior == 'namespaceFirst' then
    if t.val then
    tinsert(order, table.remove(order, 2)) -- move namespace to the end
    active[#active + 1] = t
    elseif behavior ~= 'reasonFirst' and reason ~= 'all' then
    else
    error(reason .. ' is not a valid reason')
    inactive[#inactive + 1] = t
    end
    end
    end
    local noActive = #active


    local activePropertyKeys = {}
    local attemptOrder = active
    for i, propertyKey in ipairs(order) do
    for i, t in ipairs(inactive) do
    if properties[propertyKey] ~= 'all' then
    attemptOrder[#attemptOrder + 1] = t
    activePropertyKeys[#activePropertyKeys + 1] = propertyKey
    end
    end
    end
    local noActiveProperties = #activePropertyKeys
     
    for i = 1, 2^noActive do
    -- Try successively generic matches until we run out of key combinations
    local key = {}
    for i = 1, 2^noActiveProperties do
    for j, t in ipairs(attemptOrder) do
    local categoryKey = {}
    local pos = t.pos
    for pos = 1, 5 do
    local val = t.val
    if pos > noActiveProperties then
    if j > noActive then
    categoryKey[pos] = 'all'
    key[pos] = 'all'
    else
    else
    local quotient = i / 2 ^ (pos - 1)
    local quotient = i / 2 ^ (j - 1)
    quotient = floor(quotient)
    quotient = floor(quotient)
    if quotient % 2 == 1 then
    if quotient % 2 == 1 then
    categoryKey[pos] = properties[activePropertyKeys[pos]]
    key[pos] = val
    else -- quotient % 2 == 0
    else
    categoryKey[pos] = 'all'
    key[pos] = 'all'
    end
    end
    end
    end
    end
    end
    categoryKey = tconcat(categoryKey, '-')
    key = tconcat(key, '-')
    mw.log(categoryKey) -- for debugging
    mw.log(key) -- for debugging
    local attempt = cats[categoryKey]
    local attempt = cats[key]
    if attempt then
    if attempt then
    return attempt
    return attempt

    Revision as of 12:52, 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 tinsert = table.insert
    local tremove = table.remove
    local floor = math.floor
    
    local function toTableEnd(t, pos)
    	-- Sends the item at position pos to the end of array t,
    	-- and shifts all the other array items down accordingly.
    	tinsert(t, tremove(t, pos))
    end
    
    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 nil
    	end
    	local nskey = nskeys[ns]
    	if not nskey and ns % 2 == 1 then
    			nskey = 'talk'
    	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.
    	cats = cats or categories
    
    	local properties = {
    		{pos = 5, val = expiry},
    		{pos = 3, val = namespace},
    		{pos = 2, val = protLevel},
    		{pos = 1, val = protType},
    		{pos = 4, val = reason}
    	}
    
    	local behavior = 'reasonFirst'
    	if reason then
    		behavior = behaviors[reason]
    		if behavior == 'namespaceFirst' then
    			toTableEnd(properties, 2) -- move namespace to the end
    		elseif behavior ~= 'reasonFirst' then
    			error(reason .. ' is not a valid reason')
    		end
    	end
    
    	local active, inactive = {}, {}
    	for i, t in ipairs(properties) do
    		if t.val then
    			active[#active + 1] = t
    		else
    			inactive[#inactive + 1] = t
    		end
    	end
    	local noActive = #active
    
    	local attemptOrder = active
    	for i, t in ipairs(inactive) do
    		attemptOrder[#attemptOrder + 1] = t
    	end
    
    	for i = 1, 2^noActive do
    		local key = {}
    		for j, t in ipairs(attemptOrder) do
    			local pos = t.pos
    			local val = t.val
    			if j > noActive then
    				key[pos] = 'all'
    			else
    				local quotient = i / 2 ^ (j - 1)
    				quotient = floor(quotient)
    				if quotient % 2 == 1 then
    					key[pos] = val
    				else
    					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