Difference between revisions of "Module:Protection banner"

From MINR.ORG WIKI
(allow passing the config object into the exportToLua and exportToWiki functions)
(move the category methods to the Protection class, and get rid of all the category objects)
Line 19: Line 19:
  
 
function Protection:initialize(args, configObj, titleObj)
 
function Protection:initialize(args, configObj, titleObj)
 +
self._configObj = configObj
 +
self._titleObj = titleObj
 +
 
-- Set action
 
-- Set action
 
do
 
do
Line 123: Line 126:
 
function Protection:isProtected()
 
function Protection:isProtected()
 
return self.level ~= '*'
 
return self.level ~= '*'
 +
end
 +
 +
function Protection._makeCategoryLink(cat)
 +
-- Static method for rendering category wikitext.
 +
if cat then
 +
return string.format(
 +
'[[%s:%s]]',
 +
mw.site.namespaces[14].name,
 +
cat
 +
)
 +
else
 +
return ''
 +
end
 +
end
 +
 +
function Protection:makeProtectionCategory()
 +
local configObj = self._configObj
 +
local titleObj = self._titleObj
 +
 +
-- Exit if the page is not protected.
 +
if not self:isProtected() then
 +
return ''
 +
end
 +
 +
-- Get the expiry.
 +
local expiry = self.expiry
 +
if type(expiry) == 'number' then
 +
expiry = 'temp'
 +
elseif expiry ~= 'indef' then
 +
expiry = nil
 +
end
 +
 +
-- Get the namespace category key.
 +
local nskey
 +
do
 +
local namespace = titleObj.namespace
 +
local categoryNamespaces = configObj.cfg.categoryNamespaceKeys
 +
nskey = categoryNamespaces[namespace]
 +
if not nskey and namespace % 2 == 1 then
 +
nskey = 'talk'
 +
end
 +
end
 +
 +
-- Get the other inputs.
 +
local reason = self.reason
 +
local action = self.action
 +
local level = self.level
 +
 +
--[[
 +
-- Define the properties table. Each property is a table containing the
 +
-- canonical order that the property is tested in, the position the
 +
-- property has in the category key strings, and the property value itself.
 +
--]]
 +
local properties = {
 +
expiry    = {order = 1, val = expiry},
 +
namespace = {order = 2, val = nskey},
 +
reason    = {order = 3, val = reason},
 +
level    = {order = 4, val = level},
 +
action    = {order = 5, val = action}
 +
}
 +
 +
--[[
 +
-- Apply the category order configuration, if any. The configuration value
 +
-- will be a property string, e.g. 'reason', 'namespace', etc. The property
 +
-- corresponding to that string is tested last (i.e. it is the most
 +
-- important, because it keeps its specified value the longest) and the
 +
-- other properties are tested in the canonical order. If no configuration
 +
-- value is specified then the canonical order is used.
 +
--]]
 +
local configOrder = {}
 +
do
 +
local reasonsWithNamespacePriority = configObj.cfg.reasonsWithNamespacePriority
 +
local namespaceFirst = reason and reasonsWithNamespacePriority[reason] or false
 +
for propertiesKey, t in pairs(properties) do
 +
configOrder[t.order] = t
 +
end
 +
if namespaceFirst then
 +
-- Swap namespace and reason around.
 +
local namespaceTable = table.remove(configOrder, 2)
 +
table.insert(configOrder, 3, namespaceTable)
 +
end
 +
end
 +
 +
--[[
 +
-- Define the attempt order. Properties with no value defined are moved
 +
-- to the end, where they will later be given the value "all". This is
 +
-- to cut down on the number of table lookups in the cats table, which
 +
-- grows exponentially with the number of properties with valid values.
 +
-- We keep track of the number of active properties with the noActive
 +
-- parameter.
 +
--]]
 +
local noActive, attemptOrder
 +
do
 +
local active, inactive = {}, {}
 +
for i, t in ipairs(configOrder) do
 +
if t.val then
 +
active[#active + 1] = t
 +
else
 +
inactive[#inactive + 1] = t
 +
end
 +
end
 +
noActive = #active
 +
attemptOrder = active
 +
for i, t in ipairs(inactive) do
 +
attemptOrder[#attemptOrder + 1] = t
 +
end
 +
end
 +
 +
--[[
 +
-- Check increasingly generic key combinations until we find a match.
 +
-- If a specific category exists for the combination of properties
 +
-- we are given, that match will be found first. If not, we keep
 +
-- trying different key combinations until we match using the key
 +
-- "all-all-all-all-all".
 +
--
 +
-- To generate the keys, we index the property subtables using a
 +
-- binary matrix with indexes i and j. j is only calculated up to
 +
-- the number of active properties. For example, if there were three
 +
-- active properties, the matrix would look like this, with 0
 +
-- corresponding to the string "all", and 1 corresponding to the
 +
-- val field in the property table:
 +
--
 +
--  j 1  2  3
 +
-- i 
 +
-- 1  1  1  1
 +
-- 2  0  1  1
 +
-- 3  1  0  1
 +
-- 4  0  0  1
 +
-- 5  1  1  0
 +
-- 6  0  1  0
 +
-- 7  1  0  0
 +
-- 8  0  0  0
 +
--
 +
-- Values of j higher than the number of active properties are set
 +
-- to the string "all".
 +
--
 +
-- A key for the category table is constructed for each value of i.
 +
-- The correct position of the value in the key is determined by the
 +
-- pos field in the property table.
 +
--]]
 +
local cats = configObj.cfg.protectionCategories
 +
local cat
 +
for i = 1, 2^noActive do
 +
local key = {}
 +
for j, t in ipairs(attemptOrder) do
 +
if j > noActive then
 +
key[t.order] = 'all'
 +
else
 +
local quotient = i / 2 ^ (j - 1)
 +
quotient = math.ceil(quotient)
 +
if quotient % 2 == 1 then
 +
key[t.order] = t.val
 +
else
 +
key[t.order] = 'all'
 +
end
 +
end
 +
end
 +
key = table.concat(key, '-')
 +
local attempt = cats[key]
 +
if attempt then
 +
cat = attempt
 +
break
 +
end
 +
end
 +
 +
return self._makeCategoryLink(cat)
 +
end
 +
 +
function Protection:makeExpiryCategory()
 +
local reasonsWithoutExpiryCheck = self._configObj.cfg.reasonsWithoutExpiryCheck
 +
local expiryCheckActions = self._configObj.cfg.expiryCheckActions
 +
 +
local cat
 +
if not self.expiry
 +
and expiryCheckActions[self.action]
 +
and self.reason -- the old {{pp-protected}} didn't check for expiry
 +
and not reasonsWithoutExpiryCheck[self.reason]
 +
then
 +
cat = self._configObj.msg['tracking-category-expiry']
 +
end
 +
return self._makeCategoryLink(cat)
 +
end
 +
 +
function Protection:makeErrorCategory()
 +
local configObj = self._configObj
 +
local cat
 +
if not self:isProtected()
 +
or type(self.expiry) == 'number' and self.expiry < os.time()
 +
then
 +
cat = configObj.msg['tracking-category-incorrect']
 +
end
 +
return self._makeCategoryLink(cat)
 +
end
 +
 +
function Protection:makeTemplateCategory()
 +
local configObj = self._configObj
 +
local titleObj = self._titleObj
 +
 +
local cat
 +
if self.level == 'templateeditor'
 +
and (
 +
(self.action ~= 'edit' and self.action ~= 'move')
 +
or (titleObj.namespace ~= 10 and titleObj.namespace ~= 828)
 +
)
 +
then
 +
cat = configObj.msg['tracking-category-template']
 +
end
 +
return self._makeCategoryLink(cat)
 
end
 
end
  
Line 660: Line 871:
 
:wikitext(self:renderImage())
 
:wikitext(self:renderImage())
 
return tostring(root)
 
return tostring(root)
end
 
 
--------------------------------------------------------------------------------
 
-- Category class
 
--------------------------------------------------------------------------------
 
 
local Category = class('Category')
 
 
function Category:initialize(configObj, protectionObj)
 
self._configObj = configObj
 
self._protectionObj = protectionObj
 
end
 
 
function Category:setName(name)
 
self._name = name
 
end
 
 
function Category:render()
 
if self._name then
 
return string.format(
 
'[[%s:%s]]',
 
mw.site.namespaces[14].name,
 
self._name
 
)
 
end
 
end
 
 
--------------------------------------------------------------------------------
 
-- ProtectionCategory class
 
--------------------------------------------------------------------------------
 
 
local ProtectionCategory = Category:subclass('ProtectionCategory')
 
 
function ProtectionCategory:initialize(configObj, protectionObj, titleObj)
 
Category.initialize(self, configObj, protectionObj)
 
self._titleObj = titleObj
 
end
 
 
function ProtectionCategory:render()
 
local configObj = self._configObj
 
local protectionObj = self._protectionObj
 
local titleObj = self._titleObj
 
 
-- Get the level and exit if the page is not protected.
 
if not protectionObj:isProtected() then
 
return ''
 
end
 
 
-- Get the expiry.
 
local expiry = protectionObj.expiry
 
if type(expiry) == 'number' then
 
expiry = 'temp'
 
elseif expiry ~= 'indef' then
 
expiry = nil
 
end
 
 
-- Get the namespace category key.
 
local nskey
 
do
 
local namespace = titleObj.namespace
 
local categoryNamespaces = configObj.cfg.categoryNamespaceKeys
 
nskey = categoryNamespaces[namespace]
 
if not nskey and namespace % 2 == 1 then
 
nskey = 'talk'
 
end
 
end
 
 
-- Get the other inputs.
 
local reason = protectionObj.reason
 
local action = protectionObj.action
 
local level = protectionObj.level
 
 
--[[
 
-- Define the properties table. Each property is a table containing the
 
-- canonical order that the property is tested in, the position the
 
-- property has in the category key strings, and the property value itself.
 
--]]
 
local properties = {
 
expiry    = {order = 1, val = expiry},
 
namespace = {order = 2, val = nskey},
 
reason    = {order = 3, val = reason},
 
level    = {order = 4, val = level},
 
action    = {order = 5, val = action}
 
}
 
 
--[[
 
-- Apply the category order configuration, if any. The configuration value
 
-- will be a property string, e.g. 'reason', 'namespace', etc. The property
 
-- corresponding to that string is tested last (i.e. it is the most
 
-- important, because it keeps its specified value the longest) and the
 
-- other properties are tested in the canonical order. If no configuration
 
-- value is specified then the canonical order is used.
 
--]]
 
local configOrder = {}
 
do
 
local reasonsWithNamespacePriority = configObj.cfg.reasonsWithNamespacePriority
 
local namespaceFirst = reason and reasonsWithNamespacePriority[reason] or false
 
for propertiesKey, t in pairs(properties) do
 
configOrder[t.order] = t
 
end
 
if namespaceFirst then
 
-- Swap namespace and reason around.
 
local namespaceTable = table.remove(configOrder, 2)
 
table.insert(configOrder, 3, namespaceTable)
 
end
 
end
 
 
--[[
 
-- Define the attempt order. Properties with no value defined are moved
 
-- to the end, where they will later be given the value "all". This is
 
-- to cut down on the number of table lookups in the cats table, which
 
-- grows exponentially with the number of properties with valid values.
 
-- We keep track of the number of active properties with the noActive
 
-- parameter.
 
--]]
 
local noActive, attemptOrder
 
do
 
local active, inactive = {}, {}
 
for i, t in ipairs(configOrder) do
 
if t.val then
 
active[#active + 1] = t
 
else
 
inactive[#inactive + 1] = t
 
end
 
end
 
noActive = #active
 
attemptOrder = active
 
for i, t in ipairs(inactive) do
 
attemptOrder[#attemptOrder + 1] = t
 
end
 
end
 
 
--[[
 
-- Check increasingly generic key combinations until we find a match.
 
-- If a specific category exists for the combination of properties
 
-- we are given, that match will be found first. If not, we keep
 
-- trying different key combinations until we match using the key
 
-- "all-all-all-all-all".
 
--
 
-- To generate the keys, we index the property subtables using a
 
-- binary matrix with indexes i and j. j is only calculated up to
 
-- the number of active properties. For example, if there were three
 
-- active properties, the matrix would look like this, with 0
 
-- corresponding to the string "all", and 1 corresponding to the
 
-- val field in the property table:
 
--
 
--  j 1  2  3
 
-- i 
 
-- 1  1  1  1
 
-- 2  0  1  1
 
-- 3  1  0  1
 
-- 4  0  0  1
 
-- 5  1  1  0
 
-- 6  0  1  0
 
-- 7  1  0  0
 
-- 8  0  0  0
 
--
 
-- Values of j higher than the number of active properties are set
 
-- to the string "all".
 
--
 
-- A key for the category table is constructed for each value of i.
 
-- The correct position of the value in the key is determined by the
 
-- pos field in the property table.
 
--]]
 
local cats = configObj.cfg.protectionCategories
 
local cat
 
for i = 1, 2^noActive do
 
local key = {}
 
for j, t in ipairs(attemptOrder) do
 
if j > noActive then
 
key[t.order] = 'all'
 
else
 
local quotient = i / 2 ^ (j - 1)
 
quotient = math.ceil(quotient)
 
if quotient % 2 == 1 then
 
key[t.order] = t.val
 
else
 
key[t.order] = 'all'
 
end
 
end
 
end
 
key = table.concat(key, '-')
 
local attempt = cats[key]
 
if attempt then
 
cat = attempt
 
break
 
end
 
end
 
if cat then
 
self:setName(cat)
 
end
 
return Category.render(self)
 
end
 
 
--------------------------------------------------------------------------------
 
-- ExpiryCategory class
 
--------------------------------------------------------------------------------
 
 
local ExpiryCategory = Category:subclass('ExpiryCategory')
 
 
function ExpiryCategory:render()
 
local reasonsWithoutExpiryCheck = self._configObj.cfg.reasonsWithoutExpiryCheck
 
local expiryCheckActions = self._configObj.cfg.expiryCheckActions
 
local expiry = self._protectionObj.expiry
 
local action = self._protectionObj.action
 
local reason = self._protectionObj.reason
 
 
if not expiry
 
and expiryCheckActions[action]
 
and reason -- the old {{pp-protected}} didn't check for expiry
 
and not reasonsWithoutExpiryCheck[reason]
 
then
 
self:setName(self._configObj.msg['tracking-category-expiry'])
 
end
 
return Category.render(self)
 
end
 
 
--------------------------------------------------------------------------------
 
-- ErrorCategory class
 
--------------------------------------------------------------------------------
 
 
local ErrorCategory = Category:subclass('ErrorCategory')
 
 
function ErrorCategory:render()
 
local configObj = self._configObj
 
local protectionObj = self._protectionObj
 
local expiry = protectionObj.expiry
 
local action = protectionObj.action
 
local level = protectionObj.level
 
 
if not protectionObj:isProtected()
 
or type(expiry) == 'number' and expiry < os.time()
 
then
 
self:setName(configObj.msg['tracking-category-incorrect'])
 
end
 
return Category.render(self)
 
end
 
 
--------------------------------------------------------------------------------
 
-- TemplateCategory class
 
--------------------------------------------------------------------------------
 
 
local TemplateCategory = Category:subclass('TemplateCategory')
 
 
function TemplateCategory:initialize(configObj, protectionObj, titleObj)
 
Category.initialize(self, configObj, protectionObj)
 
self._titleObj = titleObj
 
end
 
 
function TemplateCategory:render()
 
local configObj = self._configObj
 
local titleObj = self._titleObj
 
local action = self._protectionObj.action
 
local level = self._protectionObj.level
 
local namespace = titleObj.namespace
 
 
if level == 'templateeditor'
 
and (
 
(action ~= 'edit' and action ~= 'move')
 
or (namespace ~= 10 and namespace ~= 828)
 
)
 
then
 
self:setName(configObj.msg['tracking-category-template'])
 
end
 
return Category.render(self)
 
 
end
 
end
  
Line 986: Line 932:
 
-- Render the categories
 
-- Render the categories
 
if yesno(args.category) ~= false then
 
if yesno(args.category) ~= false then
local objects = {
+
ret[#ret + 1] = protectionObj:makeProtectionCategory()
ProtectionCategory:new(configObj, protectionObj, titleObj),
+
ret[#ret + 1] = protectionObj:makeExpiryCategory()
ExpiryCategory:new(configObj, protectionObj),
+
ret[#ret + 1] = protectionObj:makeErrorCategory()
ErrorCategory:new(configObj, protectionObj),
+
ret[#ret + 1] = protectionObj:makeTemplateCategory()
TemplateCategory:new(configObj, protectionObj, titleObj)
 
}
 
for _, obj in ipairs(objects) do
 
ret[#ret + 1] = obj:render()
 
end
 
 
end
 
end
 
 
Line 1,008: Line 949:
 
Banner = Banner,
 
Banner = Banner,
 
Padlock = Padlock,
 
Padlock = Padlock,
Category = Category,
 
ProtectionCategory = ProtectionCategory,
 
ErrorCategory = ErrorCategory,
 
ExpiryCategory = ExpiryCategory,
 
 
}
 
}
 
end
 
end
  
 
return ProtectionBanner
 
return ProtectionBanner

Revision as of 14:54, 28 June 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}}.

-- Initialise necessary modules.
require('Module:No globals')
local class = require('Module:Middleclass').class
local newFileLink = require('Module:File link').new
local effectiveProtectionLevel = require('Module:Effective protection level')._main
local yesno = require('Module:Yesno')

-- Lazily initialise modules and objects we don't always need.
local mArguments, mMessageBox, lang

--------------------------------------------------------------------------------
-- Protection class
--------------------------------------------------------------------------------

local Protection = class('Protection')

function Protection:initialize(args, configObj, titleObj)
	self._configObj = configObj
	self._titleObj = titleObj

	-- Set action
	do
		local actions = {
			create = true,
			edit = true,
			move = true,
			autoreview = true
		}
		if args.action and actions[args.action] then
			self.action = args.action
		else
			self.action = 'edit'
		end
	end

	-- Set level
	do
		local level = effectiveProtectionLevel(self.action, titleObj)
		if level == 'accountcreator' then
			-- Lump titleblacklisted pages in with template-protected pages,
			-- since templateeditors can do both.
			level = 'templateeditor'
		end
		if self.action == 'move' and level == 'autoconfirmed' then
			-- Users need to be autoconfirmed to move pages anyway, so treat
			-- semi-move-protected pages as unprotected.
			level = '*'
		end
		self.level = level or '*'
	end

	-- Validation function for the expiry and the protection date
	local function validateDate(date, dateType)
		lang = lang or mw.language.getContentLanguage()
		local success, expiry = pcall(lang.formatDate, lang, 'U', args.expiry)
		expiry = tonumber(expiry)
		if success and expiry then
			return expiry
		else
			return string.format(
				'<strong class="error">Error: invalid %s ("%s")</strong>',
				dateType,
				tostring(args.expiry)
			)
		end
	end

	-- Set expiry
	if args.expiry then
		local indefStrings = configObj.cfg.indefStrings
		if indefStrings[args.expiry] then
			self.expiry = 'indef'
		elseif type(args.expiry) == 'number' then
			self.expiry = args.expiry
		else
			self.expiry = validateDate(args.expiry, 'expiry date')
		end
	end

	-- Set reason
	do
		local reason = args.reason or args[1]
		if reason then
			self.reason = reason:lower()
		end
	end

	-- Set protection date
	self.protectionDate = validateDate(args.date, 'protection date')
	
	-- Set banner config
	do
		self.bannerConfig = {}
		local cfg = configObj.cfg
		local fields = {
			'text',
			'explanation',
			'tooltip',
			'alt',
			'link',
			'image'
		}
		local configTables = {}
		if cfg.banners[self.action] then
			configTables[#configTables + 1] = cfg.banners[self.action][self.reason]
		end
		if cfg.defaultBanners[self.action] then
			configTables[#configTables + 1] = cfg.defaultBanners[self.action][self.level]
			configTables[#configTables + 1] = cfg.defaultBanners[self.action].default
		end
		configTables[#configTables + 1] = cfg.masterBanner
		for i, field in ipairs(fields) do
			for j, t in ipairs(configTables) do
				if t[field] then
					self.bannerConfig[field] = t[field]
					break
				end
			end
		end
	end
end

function Protection:isProtected()
	return self.level ~= '*'
end

function Protection._makeCategoryLink(cat)
	-- Static method for rendering category wikitext.
	if cat then
		return string.format(
			'[[%s:%s]]',
			mw.site.namespaces[14].name,
			cat
		)
	else
		return ''
	end
end

function Protection:makeProtectionCategory()
	local configObj = self._configObj
	local titleObj = self._titleObj
	
	-- Exit if the page is not protected.
	if not self:isProtected() then
		return ''
	end
	
	-- Get the expiry.
	local expiry = self.expiry
	if type(expiry) == 'number' then
		expiry = 'temp'
	elseif expiry ~= 'indef' then
		expiry = nil
	end

	-- Get the namespace category key.
	local nskey
	do
		local namespace = titleObj.namespace
		local categoryNamespaces = configObj.cfg.categoryNamespaceKeys
		nskey = categoryNamespaces[namespace]
		if not nskey and namespace % 2 == 1 then
				nskey = 'talk'
		end
	end

	-- Get the other inputs.
	local reason = self.reason
	local action = self.action
	local level = self.level
 
	--[[
	-- Define the properties table. Each property is a table containing the
	-- canonical order that the property is tested in, the position the
	-- property has in the category key strings, and the property value itself.
	--]]
	local properties = {
		expiry    = {order = 1, val = expiry},
		namespace = {order = 2, val = nskey},
		reason    = {order = 3, val = reason},
		level     = {order = 4, val = level},
		action    = {order = 5, val = action}
	}
 
	--[[
	-- Apply the category order configuration, if any. The configuration value
	-- will be a property string, e.g. 'reason', 'namespace', etc. The property
	-- corresponding to that string is tested last (i.e. it is the most
	-- important, because it keeps its specified value the longest) and the
	-- other properties are tested in the canonical order. If no configuration
	-- value is specified then the canonical order is used.
	--]]
	local configOrder = {}
	do
		local reasonsWithNamespacePriority = configObj.cfg.reasonsWithNamespacePriority
		local namespaceFirst = reason and reasonsWithNamespacePriority[reason] or false
		for propertiesKey, t in pairs(properties) do
			configOrder[t.order] = t
		end
		if namespaceFirst then
			-- Swap namespace and reason around.
			local namespaceTable = table.remove(configOrder, 2)
			table.insert(configOrder, 3, namespaceTable)
		end
	end
 
	--[[
	-- Define the attempt order. Properties with no value defined are moved
	-- to the end, where they will later be given the value "all". This is
	-- to cut down on the number of table lookups in the cats table, which
	-- grows exponentially with the number of properties with valid values.
	-- We keep track of the number of active properties with the noActive
	-- parameter.
	--]]
	local noActive, attemptOrder
	do
		local active, inactive = {}, {}
		for i, t in ipairs(configOrder) do
			if t.val then
				active[#active + 1] = t
			else
				inactive[#inactive + 1] = t
			end
		end
		noActive = #active
		attemptOrder = active
		for i, t in ipairs(inactive) do
			attemptOrder[#attemptOrder + 1] = t
		end
	end
 
	--[[
	-- Check increasingly generic key combinations until we find a match.
	-- If a specific category exists for the combination of properties
	-- we are given, that match will be found first. If not, we keep
	-- trying different key combinations until we match using the key
	-- "all-all-all-all-all".
	--
	-- To generate the keys, we index the property subtables using a
	-- binary matrix with indexes i and j. j is only calculated up to
	-- the number of active properties. For example, if there were three
	-- active properties, the matrix would look like this, with 0
	-- corresponding to the string "all", and 1 corresponding to the
	-- val field in the property table:
	-- 
	--   j 1  2  3
	-- i  
	-- 1   1  1  1
	-- 2   0  1  1
	-- 3   1  0  1
	-- 4   0  0  1
	-- 5   1  1  0
	-- 6   0  1  0
	-- 7   1  0  0
	-- 8   0  0  0
	-- 
	-- Values of j higher than the number of active properties are set
	-- to the string "all".
	--
	-- A key for the category table is constructed for each value of i.
	-- The correct position of the value in the key is determined by the
	-- pos field in the property table.
	--]]
	local cats = configObj.cfg.protectionCategories
	local cat
	for i = 1, 2^noActive do
		local key = {}
		for j, t in ipairs(attemptOrder) do
			if j > noActive then
				key[t.order] = 'all'
			else
				local quotient = i / 2 ^ (j - 1)
				quotient = math.ceil(quotient)
				if quotient % 2 == 1 then
					key[t.order] = t.val
				else
					key[t.order] = 'all'
				end
			end
		end
		key = table.concat(key, '-')
		local attempt = cats[key]
		if attempt then
			cat = attempt
			break
		end
	end

	return self._makeCategoryLink(cat)
end

function Protection:makeExpiryCategory()
	local reasonsWithoutExpiryCheck = self._configObj.cfg.reasonsWithoutExpiryCheck
	local expiryCheckActions = self._configObj.cfg.expiryCheckActions
	
	local cat
	if not self.expiry
		and expiryCheckActions[self.action]
		and self.reason -- the old {{pp-protected}} didn't check for expiry
		and not reasonsWithoutExpiryCheck[self.reason]
	then
		cat = self._configObj.msg['tracking-category-expiry']
	end
	return self._makeCategoryLink(cat)
end

function Protection:makeErrorCategory()
	local configObj = self._configObj
	local cat
	if not self:isProtected()
		or type(self.expiry) == 'number' and self.expiry < os.time()
	then
		cat = configObj.msg['tracking-category-incorrect']
	end
	return self._makeCategoryLink(cat)
end

function Protection:makeTemplateCategory()
	local configObj = self._configObj
	local titleObj = self._titleObj
	
	local cat
	if self.level == 'templateeditor'
		and (
			(self.action ~= 'edit' and self.action ~= 'move')
			or (titleObj.namespace ~= 10 and titleObj.namespace ~= 828)
		)
	then
		cat = configObj.msg['tracking-category-template']
	end
	return self._makeCategoryLink(cat)
end

--------------------------------------------------------------------------------
-- Blurb class
--------------------------------------------------------------------------------

local Blurb = class('Blurb')

function Blurb:initialize(configObj, protectionObj, titleObj)
	self._configObj = configObj
	self._protectionObj = protectionObj
	self._bannerConfig = protectionObj.bannerConfig
	self._titleObj = titleObj
end

-- Static methods --

function Blurb.makeFullUrl(page, query, display)
	local url = mw.uri.fullUrl(page, query)
	url = tostring(url)
	return string.format('[%s %s]', url, display)
end

function Blurb.formatDate(num)
	-- Formats a Unix timestamp into dd Month, YYYY format.
	lang = lang or mw.language.getContentLanguage()
	local success, date = pcall(
		lang.formatDate,
		lang,
		'j F Y',
		'@' .. tostring(num)
	)
	if success then
		return date
	end
end

-- Private methods --

function Blurb:_getExpandedMessage(msg)
	local msg = self._configObj.msg[msg]
	return self:_substituteParameters(msg)
end

function Blurb:_substituteParameters(msg)
	if not self._params then
		local params, parameterFuncs = {}, {}
		setmetatable(params, {
			__index = function (t, k)
				local param
				if parameterFuncs[k] then
					param = parameterFuncs[k](self)
				end
				param = param or ''
				params[k] = param
				return param
			end
		})

		parameterFuncs.CURRENTVERSION     = self._makeCurrentVersionParameter
		parameterFuncs.DELETIONDISCUSSION = self._makeDeletionDiscussionParameter
		parameterFuncs.DISPUTEBLURB       = self._makeDisputeBlurbParameter
		parameterFuncs.DISPUTESECTION     = self._makeDisputeSectionParameter
		parameterFuncs.EDITREQUEST        = self._makeEditRequestParameter
		parameterFuncs.EXPIRY             = self._makeExpiryParameter
		parameterFuncs.EXPLANATIONBLURB   = self._makeExplanationBlurbParameter
		parameterFuncs.IMAGELINK          = self._makeImageLinkParameter
		parameterFuncs.INTROBLURB         = self._makeIntroBlurbParameter
		parameterFuncs.OFFICEBLURB        = self._makeOfficeBlurbParameter
		parameterFuncs.PAGETYPE           = self._makePagetypeParameter
		parameterFuncs.PROTECTIONBLURB    = self._makeProtectionBlurbParameter
		parameterFuncs.PROTECTIONDATE     = self._makeProtectionDateParameter
		parameterFuncs.PROTECTIONLEVEL    = self._makeProtectionLevelParameter
		parameterFuncs.PROTECTIONLOG      = self._makeProtectionLogParameter
		parameterFuncs.RESETBLURB         = self._makeResetBlurbParameter
		parameterFuncs.TALKPAGE           = self._makeTalkPageParameter
		parameterFuncs.TOOLTIPBLURB       = self._makeTooltipBlurbParameter
		parameterFuncs.VANDAL             = self._makeVandalTemplateParameter
		
		self._params = params
	end
	
	msg = msg:gsub('${(%u+)}', self._params)
	return msg
end

function Blurb:_makeCurrentVersionParameter()
	-- A link to the page history or the move log, depending on the kind of
	-- protection.
	local action = self._protectionObj.action
	local pagename = self._titleObj.prefixedText
	if action == 'move' then
		-- We need the move log link.
		return self.makeFullUrl(
			'Special:Log',
			{type = 'move', page = pagename},
			self:_getExpandedMessage('current-version-move-display')
		)
	else
		-- We need the history link.
		return self.makeFullUrl(
			pagename,
			{action = 'history'},
			self:_getExpandedMessage('current-version-edit-display')
		)
	end
end

function Blurb:_makeDeletionDiscussionLinkParameter()
	local deletionDiscussionPage = self._deletionDiscussionPage
	if deletionDiscussionPage then
		local display = self:_getExpandedMessage('deletion-discussion-link-display')
		return string.format('[[%s|%s]]', deletionDiscussionPage, display)
	end
end

function Blurb:_makeDisputeBlurbParameter()
	local expiry = self._protectionObj.expiry
	if type(expiry) == 'number' then
		return self:_getExpandedMessage('dispute-blurb-expiry')
	else
		return self:_getExpandedMessage('dispute-blurb-noexpiry')
	end
end

function Blurb:_makeDisputeSectionParameter()
	-- "disputes", with or without a section link
	local section = self._section
	local disputes = self:_getExpandedMessage('dispute-section-link-display')
	if section then
		return string.format(
			'[[%s:%s#%s|%s]]',
			mw.site.namespaces[self._titleObj.namespace].talk.name,
			self._titleObj.text,
			section,
			disputes
		)
	else
		return disputes
	end
end

function Blurb:_makeEditRequestParameter()
	local mEditRequest = require('Module:Submit an edit request')
	local action = self._protectionObj.action
	local level = self._protectionObj.level
	
	-- Get the display message key.
	local key
	if action == 'edit' and level == 'autoconfirmed' then
		key = 'edit-request-semi-display'
	else
		key = 'edit-request-full-display'
	end
	local display = self:_getExpandedMessage(key)
	
	-- Get the edit request type.
	local requestType
	if action == 'edit' then
		if level == 'autoconfirmed' then
			requestType = 'semi'
		elseif level == 'templateeditor' then
			requestType = 'template'
		end
	end
	requestType = requestType or 'full'
	
	return mEditRequest.exportLinkToLua{type = requestType, display = display}
end

function Blurb:_makeExpiryParameter()
	local expiry = self._protectionObj.expiry
	if expiry == 'indef' then
		return nil
	elseif type(expiry) == 'number' then
		return Blurb.formatDate(expiry)
	elseif expiry then
		-- Expiry is an error string.
		return expiry
	end
end

function Blurb:_makeExplanationBlurbParameter()
	local action = self._protectionObj.action
	local level = self._protectionObj.level
	local namespace = self._titleObj.namespace
	local isTalk = self._titleObj.isTalkPage

	-- @TODO: add semi-protection and pending changes blurbs
	local key
	if namespace == 8 then
		-- MediaWiki namespace
		key = 'explanation-blurb-full-nounprotect'
	elseif action == 'edit' and level == 'sysop' and not isTalk then
		key = 'explanation-blurb-full-subject'
	elseif action == 'move' then
		if isTalk then
			key = 'explanation-blurb-move-talk'
		else
			key = 'explanation-blurb-move-subject'
		end
	elseif action == 'create' then
		local xfd = self._deletionDiscussion
		if xfd then
			key = 'explanation-blurb-create-xfd'
		else
			key = 'explanation-blurb-create-noxfd'
		end
	else
		key = 'explanation-blurb-default'
	end
	return self:_getExpandedMessage(key)
end

function Blurb:_makeImageLinkParameter()
	local imageLinks = self._configObj.cfg.imageLinks
	local action = self._protectionObj.action
	local level = self._protectionObj.level
	local msg
	if imageLinks[action][level] then
		msg = imageLinks[action][level]
	elseif imageLinks[action].default then
		msg = imageLinks[action].default
	else
		msg = imageLinks.edit.default
	end
	return self:_substituteParameters(msg)
end

function Blurb:_makeIntroBlurbParameter()
	local expiry = self._protectionObj.expiry
	if type(expiry) == 'number' then
		return self:_getExpandedMessage('intro-blurb-expiry')
	else
		return self:_getExpandedMessage('intro-blurb-noexpiry')
	end
end

function Blurb:_makeOfficeBlurbParameter()
	local protectionDate = self._protectionObj.protectionDate
	if protectionDate then
		return self:_getExpandedMessage('office-blurb-protectiondate')
	else
		return self:_getExpandedMessage('office-blurb-noprotectiondate')
	end
end

function Blurb:_makePagetypeParameter()
	local pagetypes = self._configObj.cfg.pagetypes
	local namespace = self._titleObj.namespace
	return pagetypes[namespace] or pagetypes.default or error('no default pagetype defined')
end

function Blurb:_makeProtectionBlurbParameter()
	local protectionBlurbs = self._configObj.cfg.protectionBlurbs
	local action = self._protectionObj.action
	local level = self._protectionObj.level
	local msg
	if protectionBlurbs[action][level] then
		msg = protectionBlurbs[action][level]
	elseif protectionBlurbs[action].default then
		msg = protectionBlurbs[action].default
	elseif protectionBlurbs.edit.default then
		msg = protectionBlurbs.edit.default
	else
		error('no protection blurb defined for cfg.protectionBlurbs.edit.default')
	end
	return self:_substituteParameters(msg)
end

function Blurb:_makeProtectionDateParameter()
	local protectionDate = self._protectionObj.protectionDate
	if type(protectionDate) == 'number' then
		return Blurb.formatDate(protectionDate)
	else
		return protectionDate
	end
end

function Blurb:_makeProtectionLevelParameter()
	local protectionLevels = self._configObj.cfg.protectionLevels
	local action = self._protectionObj.action
	local level = self._protectionObj.level
	local msg
	if protectionLevels[action][level] then
		msg = protectionLevels[action][level]
	elseif protectionLevels[action].default then
		msg = protectionLevels[action].default
	elseif protectionLevels.edit.default then
		msg = protectionLevels.edit.default
	else
		error('no protection level defined for cfg.protectionLevels.edit.default')
	end
	return self:_substituteParameters(msg)
end

function Blurb:_makeProtectionLogParameter()
	local action = self._protectionObj.action
	local pagename = self._titleObj.prefixedText
	if action == 'autoreview' then
		-- We need the pending changes log.
		return self.makeFullUrl(
			'Special:Log',
			{type = 'stable', page = pagename},
			self:_getExpandedMessage('pc-log-display')
		)
	else
		-- We need the protection log.
		return self.makeFullUrl(
			'Special:Log',
			{type = 'protect', page = pagename},
			self:_getExpandedMessage('protection-log-display')
		)
	end
end

function Blurb:_makeResetBlurbParameter()
	local protectionDate = self._protectionObj.protectionDate
	if protectionDate then
		return self:_getExpandedMessage('reset-blurb-protectiondate')
	else
		return self:_getExpandedMessage('reset-blurb-noprotectiondate')
	end
end

function Blurb:_makeTalkPageParameter()
	local section = self._section
	local display = self:_getExpandedMessage('talk-page-link-display')
	return string.format(
		'[[%s:%s#%s|%s]]',
		mw.site.namespaces[self._titleObj.namespace].talk.name,
		self._titleObj.text,
		section or 'top',
		display
	)
end

function Blurb:_makeTooltipBlurbParameter()
	local expiry = self._protectionObj.expiry
	if type(expiry) == 'number' then
		return self:_getExpandedMessage('tooltip-blurb-expiry')
	else
		return self:_getExpandedMessage('tooltip-blurb-noexpiry')
	end
end

function Blurb:_makeVandalTemplateParameter()
	local mVandalM = require('Module:Vandal-m')
	local username = self._username
	username = username or self._titleObj.baseText
	return mVandalM._main{username}
end

-- Public methods --

function Blurb:setDeletionDiscussionPage(page)
	self._deletionDiscussionPage = page
end

function Blurb:setUsername(username)
	self._username = username
end

function Blurb:setSection(section)
	self._section = section
end

function Blurb:makeReasonText()
	local msg = self._bannerConfig.text
	if msg then
		return self:_substituteParameters(msg)
	end
end

function Blurb:makeExplanationText()
	local msg = self._bannerConfig.explanation
	return self:_substituteParameters(msg)
end

function Blurb:makeTooltipText()
	local msg = self._bannerConfig.tooltip
	return self:_substituteParameters(msg)
end

function Blurb:makeAltText()
	local msg = self._bannerConfig.alt
	return self:_substituteParameters(msg)
end

function Blurb:makeLinkText()
	local msg = self._bannerConfig.link
	return self:_substituteParameters(msg)
end

--------------------------------------------------------------------------------
-- BannerTemplate class
--------------------------------------------------------------------------------

local BannerTemplate = class('BannerTemplate')

function BannerTemplate:initialize(configObj)
	self._configObj = configObj
end

function BannerTemplate:setImageFilename(filename, protectionObj, titleObj)
	if filename then
		self._imageFilename = filename
		return nil
	end

	local action = protectionObj.action
	local level = protectionObj.level
	local expiry = protectionObj.expiry
	local namespace = titleObj.namespace
	
	-- Deal with special cases first.
	if (namespace == 10 or namespace == 828) -- Maybe we don't need the namespace check?
		and action == 'edit'
		and level == 'sysop'
		and not expiry
	then
		-- Fully protected modules and templates get the special red "indef"
		-- padlock.
		self._imageFilename = self._configObj.msg['image-filename-indef']
		return nil
	end

	-- Deal with regular protection types.
	local images = self._configObj.cfg.images
	if images[action] then
		if images[action][level] then
			self._imageFilename = images[action][level]
			return nil
		elseif images[action].default then
			self._imageFilename = images[action].default
			return nil
		end
	end

	return nil
end

function BannerTemplate:setImageWidth(width)
	self._imageWidth = width
end

function BannerTemplate:setImageTooltip(tooltip)
	self._imageCaption = tooltip
end

function BannerTemplate:renderImage()
	local filename = self._imageFilename
		or self._configObj.msg['image-filename-default']
		or 'Transparent.gif'
	return newFileLink(filename)
		:width(self._imageWidth or 20)
		:alt(self._imageAlt)
		:link(self._imageLink)
		:caption(self._imageCaption)
		:render()
end

function BannerTemplate:render()
	-- Dummy method, to be implemented by the subclasses.
	return ''
end

--------------------------------------------------------------------------------
-- Banner class
--------------------------------------------------------------------------------

local Banner = BannerTemplate:subclass('Banner')

function Banner:initialize(configObj)
	BannerTemplate.initialize(self, configObj)
	self:setImageWidth(40)
end

function Banner:setReasonText(s)
	self._reasonText = s
end

function Banner:setExplanationText(s)
	self._explanationText = s
end

function Banner:render(page)
	-- Renders the banner.
	-- The page parameter specifies the page to generate the banner for, for
	-- testing purposes.
	mMessageBox = mMessageBox or require('Module:Message box')
	local reasonText = self._reasonText or error('no reason text set')
	local explanationText = self._explanationText
	local mbargs = {
		page = page,
		type = 'protection',
		image = self:renderImage(),
		text = string.format(
			"'''%s'''%s",
			reasonText,
			explanationText and '<br />' .. explanationText or ''
		)
	}
	return mMessageBox.main('mbox', mbargs)
end

--------------------------------------------------------------------------------
-- Padlock class
--------------------------------------------------------------------------------

local Padlock = BannerTemplate:subclass('Padlock')

function Padlock:initialize(configObj)
	BannerTemplate.initialize(self, configObj)
	self:setImageWidth(20)
end

function Padlock:setImageAlt(alt)
	self._imageAlt = alt
end

function Padlock:setImageLink(link)
	self._imageLink = link
end

function Padlock:setRight(px)
	self._right = px
end

function Padlock:render()
	local root = mw.html.create('div')
	root
		:addClass('metadata topicon nopopups')
		:attr('id', 'protected-icon')
		:css{display = 'none', right = self._right or '55px'}
		:wikitext(self:renderImage())
	return tostring(root)
end

--------------------------------------------------------------------------------
-- ProtectionBanner class
--------------------------------------------------------------------------------

local ProtectionBanner = {}

function ProtectionBanner.exportToWiki(frame, configObj, titleObj)
	mArguments = mArguments or require('Module:Arguments')
	local args = mArguments.getArgs(frame)
	return ProtectionBanner.exportToLua(args, configObj, titleObj)
end

function ProtectionBanner.exportToLua(args, configObj, titleObj)
	configObj = configObj or mw.loadData('Module:Protection banner/config')
	titleObj = titleObj or mw.title.getCurrentTitle()

	-- Initialise protection and blurb objects
	local protectionObj = Protection:new(args, configObj, titleObj)
	local blurbObj = Blurb:new(configObj, protectionObj, titleObj)
	blurbObj:setDeletionDiscussionPage(args.xfd)
	blurbObj:setUsername(args.user)
	blurbObj:setSection(args.section)

	local ret = {}

	-- Render the banner
	if protectionObj:isProtected() then
		-- Get the banner object
		local isPadlock = yesno(args.small)
		local bannerObj
		if isPadlock then
			bannerObj = Padlock:new(configObj)
		else
			bannerObj = Banner:new(configObj)
		end

		-- Set the image fields
		local bannerConfig = protectionObj.bannerConfig
		bannerObj:setImageFilename(bannerConfig.image, protectionObj, titleObj)
		if isPadlock then
			bannerObj:setImageTooltip(blurbObj:makeTooltipText())
			bannerObj:setImageAlt(blurbObj:makeAltText())
			bannerObj:setImageLink(blurbObj:makeLinkText())
		else
			-- Large banners use the alt text for the tooltip.
			bannerObj:setImageTooltip(blurbObj:makeAltText())
		end

		-- Set the text fields
		if not isPadlock then
			bannerObj:setReasonText(blurbObj:makeReasonText())
			bannerObj:setExplanationText(blurbObj:makeExplanationText())
		end

		ret[#ret + 1] = bannerObj:render()
	end
	
	-- Render the categories
	if yesno(args.category) ~= false then
		ret[#ret + 1] = protectionObj:makeProtectionCategory()
		ret[#ret + 1] = protectionObj:makeExpiryCategory()
		ret[#ret + 1] = protectionObj:makeErrorCategory()
		ret[#ret + 1] = protectionObj:makeTemplateCategory()
	end
	
	return table.concat(ret)	
end

function ProtectionBanner._exportClasses()
	-- This is used to export the classes for testing purposes.
	return {
		Protection = Protection,
		Blurb = Blurb,
		BannerTemplate = BannerTemplate,
		Banner = Banner,
		Padlock = Padlock,
	}
end

return ProtectionBanner