Difference between revisions of "Module:Protection banner"
Jump to navigation
Jump to search
(use pipes as separators instead of hyphens for the protection category keys and validate reasons when we create the protection object to make sure they don't contain pipes) |
m (231 revisions imported: citations) |
||
(49 intermediate revisions by 6 users not shown) | |||
Line 4: | Line 4: | ||
-- Initialise necessary modules. | -- Initialise necessary modules. | ||
require('Module:No globals') | require('Module:No globals') | ||
− | local | + | local makeFileLink = require('Module:File link')._main |
− | |||
local effectiveProtectionLevel = require('Module:Effective protection level')._main | local effectiveProtectionLevel = require('Module:Effective protection level')._main | ||
+ | local effectiveProtectionExpiry = require('Module:Effective protection expiry')._main | ||
local yesno = require('Module:Yesno') | local yesno = require('Module:Yesno') | ||
-- Lazily initialise modules and objects we don't always need. | -- Lazily initialise modules and objects we don't always need. | ||
local getArgs, makeMessageBox, lang | local getArgs, makeMessageBox, lang | ||
+ | |||
+ | -- Set constants. | ||
+ | local CONFIG_MODULE = 'Module:Protection banner/config' | ||
-------------------------------------------------------------------------------- | -------------------------------------------------------------------------------- | ||
Line 16: | Line 19: | ||
-------------------------------------------------------------------------------- | -------------------------------------------------------------------------------- | ||
− | local function makeCategoryLink(cat) | + | local function makeCategoryLink(cat, sort) |
if cat then | if cat then | ||
return string.format( | return string.format( | ||
− | '[[%s:%s]]', | + | '[[%s:%s|%s]]', |
mw.site.namespaces[14].name, | mw.site.namespaces[14].name, | ||
− | cat | + | cat, |
+ | sort | ||
) | ) | ||
− | |||
− | |||
end | end | ||
end | end | ||
Line 30: | Line 32: | ||
-- Validation function for the expiry and the protection date | -- Validation function for the expiry and the protection date | ||
local function validateDate(dateString, dateType) | local function validateDate(dateString, dateType) | ||
− | lang = | + | if not lang then |
+ | lang = mw.language.getContentLanguage() | ||
+ | end | ||
local success, result = pcall(lang.formatDate, lang, 'U', dateString) | local success, result = pcall(lang.formatDate, lang, 'U', dateString) | ||
if success then | if success then | ||
Line 39: | Line 43: | ||
end | end | ||
error(string.format( | error(string.format( | ||
− | 'invalid %s | + | 'invalid %s: %s', |
dateType, | dateType, | ||
tostring(dateString) | tostring(dateString) | ||
− | ), | + | ), 4) |
end | end | ||
Line 53: | Line 57: | ||
end | end | ||
− | local function | + | -- Given a directed graph formatted as node -> table of direct successors, |
− | -- | + | -- get a table of all nodes reachable from a given node (though always |
− | + | -- including the given node). | |
− | + | local function getReachableNodes(graph, start) | |
+ | local toWalk, retval = {[start] = true}, {} | ||
+ | while true do | ||
+ | -- Can't use pairs() since we're adding and removing things as we're iterating | ||
+ | local k = next(toWalk) -- This always gets the "first" key | ||
+ | if k == nil then | ||
+ | return retval | ||
+ | end | ||
+ | toWalk[k] = nil | ||
+ | retval[k] = true | ||
+ | for _,v in ipairs(graph[k]) do | ||
+ | if not retval[v] then | ||
+ | toWalk[v] = true | ||
+ | end | ||
+ | end | ||
+ | end | ||
end | end | ||
Line 63: | Line 82: | ||
-------------------------------------------------------------------------------- | -------------------------------------------------------------------------------- | ||
− | local Protection = | + | local Protection = {} |
+ | Protection.__index = Protection | ||
Protection.supportedActions = { | Protection.supportedActions = { | ||
edit = true, | edit = true, | ||
move = true, | move = true, | ||
− | autoreview = true | + | autoreview = true, |
+ | upload = true | ||
} | } | ||
Line 80: | Line 101: | ||
} | } | ||
− | function Protection | + | function Protection.new(args, cfg, title) |
− | + | local obj = {} | |
− | + | obj._cfg = cfg | |
+ | obj.title = title or mw.title.getCurrentTitle() | ||
-- Set action | -- Set action | ||
if not args.action then | if not args.action then | ||
− | + | obj.action = 'edit' | |
− | elseif | + | elseif Protection.supportedActions[args.action] then |
− | + | obj.action = args.action | |
else | else | ||
error(string.format( | error(string.format( | ||
− | 'invalid action | + | 'invalid action: %s', |
tostring(args.action) | tostring(args.action) | ||
− | ), | + | ), 3) |
end | end | ||
-- Set level | -- Set level | ||
− | + | obj.level = args.demolevel or effectiveProtectionLevel(obj.action, obj.title) | |
− | if | + | if not obj.level or (obj.action == 'move' and obj.level == 'autoconfirmed') then |
− | |||
− | |||
− | |||
− | |||
-- Users need to be autoconfirmed to move pages anyway, so treat | -- Users need to be autoconfirmed to move pages anyway, so treat | ||
-- semi-move-protected pages as unprotected. | -- semi-move-protected pages as unprotected. | ||
− | + | obj.level = '*' | |
end | end | ||
-- Set expiry | -- Set expiry | ||
− | + | local effectiveExpiry = effectiveProtectionExpiry(obj.action, obj.title) | |
− | + | if effectiveExpiry == 'infinity' then | |
− | + | obj.expiry = 'indef' | |
− | + | elseif effectiveExpiry ~= 'unknown' then | |
− | + | obj.expiry = validateDate(effectiveExpiry, 'expiry date') | |
− | |||
− | |||
− | |||
end | end | ||
-- Set reason | -- Set reason | ||
if args[1] then | if args[1] then | ||
− | + | obj.reason = mw.ustring.lower(args[1]) | |
− | if | + | if obj.reason:find('|') then |
− | error('reasons cannot contain the pipe character ("|")', | + | error('reasons cannot contain the pipe character ("|")', 3) |
end | end | ||
end | end | ||
-- Set protection date | -- Set protection date | ||
− | + | if args.date then | |
+ | obj.protectionDate = validateDate(args.date, 'protection date') | ||
+ | end | ||
-- Set banner config | -- Set banner config | ||
do | do | ||
− | + | obj.bannerConfig = {} | |
local configTables = {} | local configTables = {} | ||
− | if cfg.banners[ | + | if cfg.banners[obj.action] then |
− | configTables[#configTables + 1] = cfg.banners[ | + | configTables[#configTables + 1] = cfg.banners[obj.action][obj.reason] |
end | end | ||
− | if cfg.defaultBanners[ | + | if cfg.defaultBanners[obj.action] then |
− | configTables[#configTables + 1] = cfg.defaultBanners[ | + | configTables[#configTables + 1] = cfg.defaultBanners[obj.action][obj.level] |
− | configTables[#configTables + 1] = cfg.defaultBanners[ | + | configTables[#configTables + 1] = cfg.defaultBanners[obj.action].default |
end | end | ||
configTables[#configTables + 1] = cfg.masterBanner | configTables[#configTables + 1] = cfg.masterBanner | ||
− | for i, field in ipairs( | + | for i, field in ipairs(Protection.bannerConfigFields) do |
for j, t in ipairs(configTables) do | for j, t in ipairs(configTables) do | ||
if t[field] then | if t[field] then | ||
− | + | obj.bannerConfig[field] = t[field] | |
break | break | ||
end | end | ||
Line 151: | Line 168: | ||
end | end | ||
end | end | ||
+ | return setmetatable(obj, Protection) | ||
end | end | ||
function Protection:isProtected() | function Protection:isProtected() | ||
return self.level ~= '*' | return self.level ~= '*' | ||
+ | end | ||
+ | |||
+ | function Protection:isTemporary() | ||
+ | return type(self.expiry) == 'number' | ||
end | end | ||
Line 175: | Line 197: | ||
-- Get the namespace key fragment. | -- Get the namespace key fragment. | ||
− | local | + | local namespaceFragment = cfg.categoryNamespaceKeys[title.namespace] |
− | + | if not namespaceFragment and title.namespace % 2 == 1 then | |
− | + | namespaceFragment = 'talk' | |
− | |||
− | |||
− | |||
end | end | ||
Line 205: | Line 224: | ||
-- instead. | -- instead. | ||
--]] | --]] | ||
− | + | table.insert(order, table.remove(order, self.reason and cfg.reasonsWithNamespacePriority[self.reason] and 2 or 3)) | |
− | |||
− | |||
− | |||
− | |||
− | |||
--[[ | --[[ | ||
Line 287: | Line 301: | ||
local attempt = cats[key] | local attempt = cats[key] | ||
if attempt then | if attempt then | ||
− | return makeCategoryLink(attempt) | + | return makeCategoryLink(attempt, title.text) |
end | end | ||
end | end | ||
− | + | return '' | |
− | |||
− | |||
− | |||
− | return | ||
− | |||
− | |||
− | |||
end | end | ||
Line 318: | Line 325: | ||
local msg = self._cfg.msg | local msg = self._cfg.msg | ||
local ret = { self:makeProtectionCategory() } | local ret = { self:makeProtectionCategory() } | ||
− | |||
− | |||
− | |||
if self:isIncorrect() then | if self:isIncorrect() then | ||
− | ret[#ret + 1] = makeCategoryLink(msg['tracking-category-incorrect']) | + | ret[#ret + 1] = makeCategoryLink( |
+ | msg['tracking-category-incorrect'], | ||
+ | self.title.text | ||
+ | ) | ||
end | end | ||
if self:isTemplateProtectedNonTemplate() then | if self:isTemplateProtectedNonTemplate() then | ||
− | ret[#ret + 1] = makeCategoryLink(msg['tracking-category-template']) | + | ret[#ret + 1] = makeCategoryLink( |
+ | msg['tracking-category-template'], | ||
+ | self.title.text | ||
+ | ) | ||
end | end | ||
return table.concat(ret) | return table.concat(ret) | ||
Line 334: | Line 344: | ||
-------------------------------------------------------------------------------- | -------------------------------------------------------------------------------- | ||
− | local Blurb = | + | local Blurb = {} |
+ | Blurb.__index = Blurb | ||
Blurb.bannerTextFields = { | Blurb.bannerTextFields = { | ||
Line 344: | Line 355: | ||
} | } | ||
− | function Blurb | + | function Blurb.new(protectionObj, args, cfg) |
− | + | return setmetatable({ | |
− | + | _cfg = cfg, | |
− | + | _protectionObj = protectionObj, | |
+ | _args = args | ||
+ | }, Blurb) | ||
end | end | ||
− | -- | + | -- Private methods -- |
− | function Blurb | + | function Blurb:_formatDate(num) |
-- Formats a Unix timestamp into dd Month, YYYY format. | -- Formats a Unix timestamp into dd Month, YYYY format. | ||
lang = lang or mw.language.getContentLanguage() | lang = lang or mw.language.getContentLanguage() | ||
Line 358: | Line 371: | ||
lang.formatDate, | lang.formatDate, | ||
lang, | lang, | ||
− | 'j F Y', | + | self._cfg.msg['expiry-date-format'] or 'j F Y', |
'@' .. tostring(num) | '@' .. tostring(num) | ||
) | ) | ||
Line 365: | Line 378: | ||
end | end | ||
end | end | ||
− | |||
− | |||
function Blurb:_getExpandedMessage(msgKey) | function Blurb:_getExpandedMessage(msgKey) | ||
Line 377: | Line 388: | ||
parameterFuncs.CURRENTVERSION = self._makeCurrentVersionParameter | parameterFuncs.CURRENTVERSION = self._makeCurrentVersionParameter | ||
− | |||
− | |||
parameterFuncs.EDITREQUEST = self._makeEditRequestParameter | parameterFuncs.EDITREQUEST = self._makeEditRequestParameter | ||
parameterFuncs.EXPIRY = self._makeExpiryParameter | parameterFuncs.EXPIRY = self._makeExpiryParameter | ||
Line 384: | Line 393: | ||
parameterFuncs.IMAGELINK = self._makeImageLinkParameter | parameterFuncs.IMAGELINK = self._makeImageLinkParameter | ||
parameterFuncs.INTROBLURB = self._makeIntroBlurbParameter | parameterFuncs.INTROBLURB = self._makeIntroBlurbParameter | ||
− | parameterFuncs. | + | parameterFuncs.INTROFRAGMENT = self._makeIntroFragmentParameter |
parameterFuncs.PAGETYPE = self._makePagetypeParameter | parameterFuncs.PAGETYPE = self._makePagetypeParameter | ||
parameterFuncs.PROTECTIONBLURB = self._makeProtectionBlurbParameter | parameterFuncs.PROTECTIONBLURB = self._makeProtectionBlurbParameter | ||
Line 390: | Line 399: | ||
parameterFuncs.PROTECTIONLEVEL = self._makeProtectionLevelParameter | parameterFuncs.PROTECTIONLEVEL = self._makeProtectionLevelParameter | ||
parameterFuncs.PROTECTIONLOG = self._makeProtectionLogParameter | parameterFuncs.PROTECTIONLOG = self._makeProtectionLogParameter | ||
− | |||
parameterFuncs.TALKPAGE = self._makeTalkPageParameter | parameterFuncs.TALKPAGE = self._makeTalkPageParameter | ||
parameterFuncs.TOOLTIPBLURB = self._makeTooltipBlurbParameter | parameterFuncs.TOOLTIPBLURB = self._makeTooltipBlurbParameter | ||
+ | parameterFuncs.TOOLTIPFRAGMENT = self._makeTooltipFragmentParameter | ||
parameterFuncs.VANDAL = self._makeVandalTemplateParameter | parameterFuncs.VANDAL = self._makeVandalTemplateParameter | ||
Line 437: | Line 446: | ||
local action = self._protectionObj.action | local action = self._protectionObj.action | ||
local level = self._protectionObj.level | local level = self._protectionObj.level | ||
− | |||
− | |||
− | |||
− | |||
− | |||
− | |||
− | |||
− | |||
− | |||
-- Get the edit request type. | -- Get the edit request type. | ||
Line 452: | Line 452: | ||
if level == 'autoconfirmed' then | if level == 'autoconfirmed' then | ||
requestType = 'semi' | requestType = 'semi' | ||
+ | elseif level == 'extendedconfirmed' then | ||
+ | requestType = 'extended' | ||
elseif level == 'templateeditor' then | elseif level == 'templateeditor' then | ||
requestType = 'template' | requestType = 'template' | ||
Line 458: | Line 460: | ||
requestType = requestType or 'full' | requestType = requestType or 'full' | ||
− | return mEditRequest. | + | -- Get the display value. |
+ | local display = self:_getExpandedMessage('edit-request-display') | ||
+ | |||
+ | return mEditRequest._link{type = requestType, display = display} | ||
end | end | ||
Line 464: | Line 469: | ||
local expiry = self._protectionObj.expiry | local expiry = self._protectionObj.expiry | ||
if type(expiry) == 'number' then | if type(expiry) == 'number' then | ||
− | return | + | return self:_formatDate(expiry) |
else | else | ||
return expiry | return expiry | ||
Line 471: | Line 476: | ||
function Blurb:_makeExplanationBlurbParameter() | function Blurb:_makeExplanationBlurbParameter() | ||
+ | -- Cover special cases first. | ||
+ | if self._protectionObj.title.namespace == 8 then | ||
+ | -- MediaWiki namespace | ||
+ | return self:_getExpandedMessage('explanation-blurb-nounprotect') | ||
+ | end | ||
+ | |||
+ | -- Get explanation blurb table keys | ||
local action = self._protectionObj.action | local action = self._protectionObj.action | ||
local level = self._protectionObj.level | local level = self._protectionObj.level | ||
− | local | + | local talkKey = self._protectionObj.title.isTalkPage and 'talk' or 'subject' |
− | |||
− | -- | + | -- Find the message in the explanation blurb table and substitute any |
− | local | + | -- parameters. |
− | if | + | local explanations = self._cfg.explanationBlurbs |
− | + | local msg | |
− | + | if explanations[action][level] and explanations[action][level][talkKey] then | |
− | elseif action | + | msg = explanations[action][level][talkKey] |
− | + | elseif explanations[action][level] and explanations[action][level].default then | |
− | elseif action | + | msg = explanations[action][level].default |
− | + | elseif explanations[action].default and explanations[action].default[talkKey] then | |
− | + | msg = explanations[action].default[talkKey] | |
− | + | elseif explanations[action].default and explanations[action].default.default then | |
− | + | msg = explanations[action].default.default | |
− | |||
else | else | ||
− | + | error(string.format( | |
+ | 'could not find explanation blurb for action "%s", level "%s" and talk key "%s"', | ||
+ | action, | ||
+ | level, | ||
+ | talkKey | ||
+ | ), 8) | ||
end | end | ||
− | return self: | + | return self:_substituteParameters(msg) |
end | end | ||
Line 511: | Line 526: | ||
function Blurb:_makeIntroBlurbParameter() | function Blurb:_makeIntroBlurbParameter() | ||
− | if | + | if self._protectionObj:isTemporary() then |
return self:_getExpandedMessage('intro-blurb-expiry') | return self:_getExpandedMessage('intro-blurb-expiry') | ||
else | else | ||
return self:_getExpandedMessage('intro-blurb-noexpiry') | return self:_getExpandedMessage('intro-blurb-noexpiry') | ||
+ | end | ||
+ | end | ||
+ | |||
+ | function Blurb:_makeIntroFragmentParameter() | ||
+ | if self._protectionObj:isTemporary() then | ||
+ | return self:_getExpandedMessage('intro-fragment-expiry') | ||
+ | else | ||
+ | return self:_getExpandedMessage('intro-fragment-noexpiry') | ||
end | end | ||
end | end | ||
Line 522: | Line 545: | ||
return pagetypes[self._protectionObj.title.namespace] | return pagetypes[self._protectionObj.title.namespace] | ||
or pagetypes.default | or pagetypes.default | ||
− | or error('no default pagetype defined') | + | or error('no default pagetype defined', 8) |
end | end | ||
Line 537: | Line 560: | ||
msg = protectionBlurbs.edit.default | msg = protectionBlurbs.edit.default | ||
else | else | ||
− | error('no protection blurb defined for protectionBlurbs.edit.default') | + | error('no protection blurb defined for protectionBlurbs.edit.default', 8) |
end | end | ||
return self:_substituteParameters(msg) | return self:_substituteParameters(msg) | ||
Line 545: | Line 568: | ||
local protectionDate = self._protectionObj.protectionDate | local protectionDate = self._protectionObj.protectionDate | ||
if type(protectionDate) == 'number' then | if type(protectionDate) == 'number' then | ||
− | return | + | return self:_formatDate(protectionDate) |
else | else | ||
return protectionDate | return protectionDate | ||
Line 563: | Line 586: | ||
msg = protectionLevels.edit.default | msg = protectionLevels.edit.default | ||
else | else | ||
− | error('no protection level defined for protectionLevels.edit.default') | + | error('no protection level defined for protectionLevels.edit.default', 8) |
end | end | ||
return self:_substituteParameters(msg) | return self:_substituteParameters(msg) | ||
Line 598: | Line 621: | ||
function Blurb:_makeTooltipBlurbParameter() | function Blurb:_makeTooltipBlurbParameter() | ||
− | if | + | if self._protectionObj:isTemporary() then |
return self:_getExpandedMessage('tooltip-blurb-expiry') | return self:_getExpandedMessage('tooltip-blurb-expiry') | ||
else | else | ||
return self:_getExpandedMessage('tooltip-blurb-noexpiry') | return self:_getExpandedMessage('tooltip-blurb-noexpiry') | ||
+ | end | ||
+ | end | ||
+ | |||
+ | function Blurb:_makeTooltipFragmentParameter() | ||
+ | if self._protectionObj:isTemporary() then | ||
+ | return self:_getExpandedMessage('tooltip-fragment-expiry') | ||
+ | else | ||
+ | return self:_getExpandedMessage('tooltip-fragment-noexpiry') | ||
end | end | ||
end | end | ||
Line 634: | Line 665: | ||
tostring(key), | tostring(key), | ||
type(msg) | type(msg) | ||
− | )) | + | ), 4) |
end | end | ||
return self:_substituteParameters(msg) | return self:_substituteParameters(msg) | ||
Line 644: | Line 675: | ||
-------------------------------------------------------------------------------- | -------------------------------------------------------------------------------- | ||
− | local BannerTemplate = | + | local BannerTemplate = {} |
+ | BannerTemplate.__index = BannerTemplate | ||
− | function BannerTemplate | + | function BannerTemplate.new(protectionObj, cfg) |
− | + | local obj = {} | |
+ | obj._cfg = cfg | ||
-- Set the image filename. | -- Set the image filename. | ||
local imageFilename = protectionObj.bannerConfig.image | local imageFilename = protectionObj.bannerConfig.image | ||
if imageFilename then | if imageFilename then | ||
− | + | obj._imageFilename = imageFilename | |
else | else | ||
-- If an image filename isn't specified explicitly in the banner config, | -- If an image filename isn't specified explicitly in the banner config, | ||
Line 658: | Line 691: | ||
local action = protectionObj.action | local action = protectionObj.action | ||
local level = protectionObj.level | local level = protectionObj.level | ||
− | |||
local namespace = protectionObj.title.namespace | local namespace = protectionObj.title.namespace | ||
+ | local reason = protectionObj.reason | ||
-- Deal with special cases first. | -- Deal with special cases first. | ||
− | if (namespace == 10 or namespace == 828) | + | if ( |
+ | namespace == 10 | ||
+ | or namespace == 828 | ||
+ | or reason and obj._cfg.indefImageReasons[reason] | ||
+ | ) | ||
and action == 'edit' | and action == 'edit' | ||
and level == 'sysop' | and level == 'sysop' | ||
− | and not | + | and not protectionObj:isTemporary() |
then | then | ||
-- Fully protected modules and templates get the special red "indef" | -- Fully protected modules and templates get the special red "indef" | ||
-- padlock. | -- padlock. | ||
− | + | obj._imageFilename = obj._cfg.msg['image-filename-indef'] | |
else | else | ||
-- Deal with regular protection types. | -- Deal with regular protection types. | ||
− | local images = | + | local images = obj._cfg.images |
if images[action] then | if images[action] then | ||
if images[action][level] then | if images[action][level] then | ||
− | + | obj._imageFilename = images[action][level] | |
elseif images[action].default then | elseif images[action].default then | ||
− | + | obj._imageFilename = images[action].default | |
end | end | ||
end | end | ||
end | end | ||
end | end | ||
− | + | return setmetatable(obj, BannerTemplate) | |
− | |||
− | |||
− | |||
− | |||
− | |||
− | |||
− | |||
end | end | ||
Line 696: | Line 726: | ||
or self._cfg.msg['image-filename-default'] | or self._cfg.msg['image-filename-default'] | ||
or 'Transparent.gif' | or 'Transparent.gif' | ||
− | return | + | return makeFileLink{ |
− | + | file = filename, | |
− | + | size = (self.imageWidth or 20) .. 'px', | |
− | + | alt = self._imageAlt, | |
− | + | link = self._imageLink, | |
− | + | caption = self.imageCaption | |
+ | } | ||
end | end | ||
Line 708: | Line 739: | ||
-------------------------------------------------------------------------------- | -------------------------------------------------------------------------------- | ||
− | local Banner = BannerTemplate | + | local Banner = setmetatable({}, BannerTemplate) |
+ | Banner.__index = Banner | ||
− | function Banner | + | function Banner.new(protectionObj, blurbObj, cfg) |
− | BannerTemplate. | + | local obj = BannerTemplate.new(protectionObj, cfg) -- This doesn't need the blurb. |
− | + | obj.imageWidth = 40 | |
− | + | obj.imageCaption = blurbObj:makeBannerText('alt') -- Large banners use the alt text for the tooltip. | |
− | + | obj._reasonText = blurbObj:makeBannerText('text') | |
− | + | obj._explanationText = blurbObj:makeBannerText('explanation') | |
− | + | obj._page = protectionObj.title.prefixedText -- Only makes a difference in testing. | |
+ | return setmetatable(obj, Banner) | ||
end | end | ||
Line 722: | Line 755: | ||
-- Renders the banner. | -- Renders the banner. | ||
makeMessageBox = makeMessageBox or require('Module:Message box').main | makeMessageBox = makeMessageBox or require('Module:Message box').main | ||
− | local reasonText = self._reasonText or error('no reason text set') | + | local reasonText = self._reasonText or error('no reason text set', 2) |
local explanationText = self._explanationText | local explanationText = self._explanationText | ||
local mbargs = { | local mbargs = { | ||
Line 741: | Line 774: | ||
-------------------------------------------------------------------------------- | -------------------------------------------------------------------------------- | ||
− | local Padlock = BannerTemplate | + | local Padlock = setmetatable({}, BannerTemplate) |
+ | Padlock.__index = Padlock | ||
− | function Padlock | + | function Padlock.new(protectionObj, blurbObj, cfg) |
− | BannerTemplate. | + | local obj = BannerTemplate.new(protectionObj, cfg) -- This doesn't need the blurb. |
− | + | obj.imageWidth = 20 | |
− | + | obj.imageCaption = blurbObj:makeBannerText('tooltip') | |
− | + | obj._imageAlt = blurbObj:makeBannerText('alt') | |
− | + | obj._imageLink = blurbObj:makeBannerText('link') | |
− | + | obj._indicatorName = cfg.padlockIndicatorNames[protectionObj.action] | |
− | or cfg. | + | or cfg.padlockIndicatorNames.default |
− | or ' | + | or 'pp-default' |
+ | return setmetatable(obj, Padlock) | ||
end | end | ||
function Padlock:__tostring() | function Padlock:__tostring() | ||
− | local | + | local frame = mw.getCurrentFrame() |
− | + | -- The nowiki tag helps prevent whitespace at the top of articles. | |
− | + | return frame:extensionTag{name = 'nowiki'} .. frame:extensionTag{ | |
− | + | name = 'indicator', | |
− | + | args = {name = self._indicatorName}, | |
− | + | content = self:renderImage() | |
− | + | } | |
end | end | ||
Line 783: | Line 818: | ||
function p._main(args, cfg, title) | function p._main(args, cfg, title) | ||
args = args or {} | args = args or {} | ||
− | + | cfg = cfg or require(CONFIG_MODULE) | |
− | |||
− | |||
− | + | local protectionObj = Protection.new(args, cfg, title) | |
− | local | ||
− | |||
− | |||
− | |||
− | |||
− | |||
− | |||
− | |||
− | |||
− | |||
− | |||
− | |||
− | |||
− | |||
− | |||
− | |||
− | |||
− | |||
− | |||
local ret = {} | local ret = {} | ||
− | -- Render the banner | + | -- If a page's edit protection is equally or more restrictive than its |
− | + | -- protection from some other action, then don't bother displaying anything | |
− | + | -- for the other action (except categories). | |
− | + | if protectionObj.action == 'edit' or | |
− | + | args.demolevel or | |
− | + | not getReachableNodes( | |
+ | cfg.hierarchy, | ||
+ | protectionObj.level | ||
+ | )[effectiveProtectionLevel('edit', protectionObj.title)] | ||
+ | then | ||
+ | -- Initialise the blurb object | ||
+ | local blurbObj = Blurb.new(protectionObj, args, cfg) | ||
+ | |||
+ | -- Render the banner | ||
+ | if protectionObj:isProtected() then | ||
+ | ret[#ret + 1] = tostring( | ||
+ | (yesno(args.small) and Padlock or Banner) | ||
+ | .new(protectionObj, blurbObj, cfg) | ||
+ | ) | ||
+ | end | ||
end | end | ||
− | + | ||
-- Render the categories | -- Render the categories | ||
if yesno(args.category) ~= false then | if yesno(args.category) ~= false then | ||
Line 826: | Line 854: | ||
end | end | ||
− | function p.main(frame) | + | function p.main(frame, cfg) |
− | if | + | cfg = cfg or require(CONFIG_MODULE) |
− | + | ||
+ | -- Find default args, if any. | ||
+ | local parent = frame.getParent and frame:getParent() | ||
+ | local defaultArgs = parent and cfg.wrappers[parent:getTitle():gsub('/sandbox$', '')] | ||
+ | |||
+ | -- Find user args, and use the parent frame if we are being called from a | ||
+ | -- wrapper template. | ||
+ | getArgs = getArgs or require('Module:Arguments').getArgs | ||
+ | local userArgs = getArgs(frame, { | ||
+ | parentOnly = defaultArgs, | ||
+ | frameOnly = not defaultArgs | ||
+ | }) | ||
+ | |||
+ | -- Build the args table. User-specified args overwrite default args. | ||
+ | local args = {} | ||
+ | for k, v in pairs(defaultArgs or {}) do | ||
+ | args[k] = v | ||
+ | end | ||
+ | for k, v in pairs(userArgs) do | ||
+ | args[k] = v | ||
end | end | ||
− | + | return p._main(args, cfg) | |
− | return p._main(args) | ||
end | end | ||
return p | return p |
Latest revision as of 18:47, 2 November 2019
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 makeFileLink = require('Module:File link')._main
local effectiveProtectionLevel = require('Module:Effective protection level')._main
local effectiveProtectionExpiry = require('Module:Effective protection expiry')._main
local yesno = require('Module:Yesno')
-- Lazily initialise modules and objects we don't always need.
local getArgs, makeMessageBox, lang
-- Set constants.
local CONFIG_MODULE = 'Module:Protection banner/config'
--------------------------------------------------------------------------------
-- Helper functions
--------------------------------------------------------------------------------
local function makeCategoryLink(cat, sort)
if cat then
return string.format(
'[[%s:%s|%s]]',
mw.site.namespaces[14].name,
cat,
sort
)
end
end
-- Validation function for the expiry and the protection date
local function validateDate(dateString, dateType)
if not lang then
lang = mw.language.getContentLanguage()
end
local success, result = pcall(lang.formatDate, lang, 'U', dateString)
if success then
result = tonumber(result)
if result then
return result
end
end
error(string.format(
'invalid %s: %s',
dateType,
tostring(dateString)
), 4)
end
local function makeFullUrl(page, query, display)
return string.format(
'[%s %s]',
tostring(mw.uri.fullUrl(page, query)),
display
)
end
-- Given a directed graph formatted as node -> table of direct successors,
-- get a table of all nodes reachable from a given node (though always
-- including the given node).
local function getReachableNodes(graph, start)
local toWalk, retval = {[start] = true}, {}
while true do
-- Can't use pairs() since we're adding and removing things as we're iterating
local k = next(toWalk) -- This always gets the "first" key
if k == nil then
return retval
end
toWalk[k] = nil
retval[k] = true
for _,v in ipairs(graph[k]) do
if not retval[v] then
toWalk[v] = true
end
end
end
end
--------------------------------------------------------------------------------
-- Protection class
--------------------------------------------------------------------------------
local Protection = {}
Protection.__index = Protection
Protection.supportedActions = {
edit = true,
move = true,
autoreview = true,
upload = true
}
Protection.bannerConfigFields = {
'text',
'explanation',
'tooltip',
'alt',
'link',
'image'
}
function Protection.new(args, cfg, title)
local obj = {}
obj._cfg = cfg
obj.title = title or mw.title.getCurrentTitle()
-- Set action
if not args.action then
obj.action = 'edit'
elseif Protection.supportedActions[args.action] then
obj.action = args.action
else
error(string.format(
'invalid action: %s',
tostring(args.action)
), 3)
end
-- Set level
obj.level = args.demolevel or effectiveProtectionLevel(obj.action, obj.title)
if not obj.level or (obj.action == 'move' and obj.level == 'autoconfirmed') then
-- Users need to be autoconfirmed to move pages anyway, so treat
-- semi-move-protected pages as unprotected.
obj.level = '*'
end
-- Set expiry
local effectiveExpiry = effectiveProtectionExpiry(obj.action, obj.title)
if effectiveExpiry == 'infinity' then
obj.expiry = 'indef'
elseif effectiveExpiry ~= 'unknown' then
obj.expiry = validateDate(effectiveExpiry, 'expiry date')
end
-- Set reason
if args[1] then
obj.reason = mw.ustring.lower(args[1])
if obj.reason:find('|') then
error('reasons cannot contain the pipe character ("|")', 3)
end
end
-- Set protection date
if args.date then
obj.protectionDate = validateDate(args.date, 'protection date')
end
-- Set banner config
do
obj.bannerConfig = {}
local configTables = {}
if cfg.banners[obj.action] then
configTables[#configTables + 1] = cfg.banners[obj.action][obj.reason]
end
if cfg.defaultBanners[obj.action] then
configTables[#configTables + 1] = cfg.defaultBanners[obj.action][obj.level]
configTables[#configTables + 1] = cfg.defaultBanners[obj.action].default
end
configTables[#configTables + 1] = cfg.masterBanner
for i, field in ipairs(Protection.bannerConfigFields) do
for j, t in ipairs(configTables) do
if t[field] then
obj.bannerConfig[field] = t[field]
break
end
end
end
end
return setmetatable(obj, Protection)
end
function Protection:isProtected()
return self.level ~= '*'
end
function Protection:isTemporary()
return type(self.expiry) == 'number'
end
function Protection:makeProtectionCategory()
local cfg = self._cfg
local title = self.title
-- Exit if the page is not protected.
if not self:isProtected() then
return ''
end
-- Get the expiry key fragment.
local expiryFragment
if self.expiry == 'indef' then
expiryFragment = self.expiry
elseif type(self.expiry) == 'number' then
expiryFragment = 'temp'
end
-- Get the namespace key fragment.
local namespaceFragment = cfg.categoryNamespaceKeys[title.namespace]
if not namespaceFragment and title.namespace % 2 == 1 then
namespaceFragment = 'talk'
end
-- Define the order that key fragments are tested in. This is done with an
-- array of tables containing the value to be tested, along with its
-- position in the cfg.protectionCategories table.
local order = {
{val = expiryFragment, keypos = 1},
{val = namespaceFragment, keypos = 2},
{val = self.reason, keypos = 3},
{val = self.level, keypos = 4},
{val = self.action, keypos = 5}
}
--[[
-- The old protection templates used an ad-hoc protection category system,
-- with some templates prioritising namespaces in their categories, and
-- others prioritising the protection reason. To emulate this in this module
-- we use the config table cfg.reasonsWithNamespacePriority to set the
-- reasons for which namespaces have priority over protection reason.
-- If we are dealing with one of those reasons, move the namespace table to
-- the end of the order table, i.e. give it highest priority. If not, the
-- reason should have highest priority, so move that to the end of the table
-- instead.
--]]
table.insert(order, table.remove(order, self.reason and cfg.reasonsWithNamespacePriority[self.reason] and 2 or 3))
--[[
-- Define the attempt order. Inactive subtables (subtables with nil "value"
-- fields) are moved to the end, where they will later be given the key
-- "all". This is to cut down on the number of table lookups in
-- cfg.protectionCategories, which grows exponentially with the number of
-- non-nil keys. We keep track of the number of active subtables with the
-- noActive parameter.
--]]
local noActive, attemptOrder
do
local active, inactive = {}, {}
for i, t in ipairs(order) 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 key fragments we are
-- given, that match will be found first. If not, we keep trying different
-- key fragment combinations until we match using the key
-- "all-all-all-all-all".
--
-- To generate the keys, we index the key subtables using a binary matrix
-- with indexes i and j. j is only calculated up to the number of active
-- subtables. For example, if there were three active subtables, the matrix
-- would look like this, with 0 corresponding to the key fragment "all", and
-- 1 corresponding to other key fragments.
--
-- 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 subtables are set
-- to the string "all".
--
-- A key for cfg.protectionCategories is constructed for each value of i.
-- The position of the value in the key is determined by the keypos field in
-- each subtable.
--]]
local cats = cfg.protectionCategories
for i = 1, 2^noActive do
local key = {}
for j, t in ipairs(attemptOrder) do
if j > noActive then
key[t.keypos] = 'all'
else
local quotient = i / 2 ^ (j - 1)
quotient = math.ceil(quotient)
if quotient % 2 == 1 then
key[t.keypos] = t.val
else
key[t.keypos] = 'all'
end
end
end
key = table.concat(key, '|')
local attempt = cats[key]
if attempt then
return makeCategoryLink(attempt, title.text)
end
end
return ''
end
function Protection:isIncorrect()
local expiry = self.expiry
return not self:isProtected()
or type(expiry) == 'number' and expiry < os.time()
end
function Protection:isTemplateProtectedNonTemplate()
local action, namespace = self.action, self.title.namespace
return self.level == 'templateeditor'
and (
(action ~= 'edit' and action ~= 'move')
or (namespace ~= 10 and namespace ~= 828)
)
end
function Protection:makeCategoryLinks()
local msg = self._cfg.msg
local ret = { self:makeProtectionCategory() }
if self:isIncorrect() then
ret[#ret + 1] = makeCategoryLink(
msg['tracking-category-incorrect'],
self.title.text
)
end
if self:isTemplateProtectedNonTemplate() then
ret[#ret + 1] = makeCategoryLink(
msg['tracking-category-template'],
self.title.text
)
end
return table.concat(ret)
end
--------------------------------------------------------------------------------
-- Blurb class
--------------------------------------------------------------------------------
local Blurb = {}
Blurb.__index = Blurb
Blurb.bannerTextFields = {
text = true,
explanation = true,
tooltip = true,
alt = true,
link = true
}
function Blurb.new(protectionObj, args, cfg)
return setmetatable({
_cfg = cfg,
_protectionObj = protectionObj,
_args = args
}, Blurb)
end
-- Private methods --
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,
self._cfg.msg['expiry-date-format'] or 'j F Y',
'@' .. tostring(num)
)
if success then
return date
end
end
function Blurb:_getExpandedMessage(msgKey)
return self:_substituteParameters(self._cfg.msg[msgKey])
end
function Blurb:_substituteParameters(msg)
if not self._params then
local parameterFuncs = {}
parameterFuncs.CURRENTVERSION = self._makeCurrentVersionParameter
parameterFuncs.EDITREQUEST = self._makeEditRequestParameter
parameterFuncs.EXPIRY = self._makeExpiryParameter
parameterFuncs.EXPLANATIONBLURB = self._makeExplanationBlurbParameter
parameterFuncs.IMAGELINK = self._makeImageLinkParameter
parameterFuncs.INTROBLURB = self._makeIntroBlurbParameter
parameterFuncs.INTROFRAGMENT = self._makeIntroFragmentParameter
parameterFuncs.PAGETYPE = self._makePagetypeParameter
parameterFuncs.PROTECTIONBLURB = self._makeProtectionBlurbParameter
parameterFuncs.PROTECTIONDATE = self._makeProtectionDateParameter
parameterFuncs.PROTECTIONLEVEL = self._makeProtectionLevelParameter
parameterFuncs.PROTECTIONLOG = self._makeProtectionLogParameter
parameterFuncs.TALKPAGE = self._makeTalkPageParameter
parameterFuncs.TOOLTIPBLURB = self._makeTooltipBlurbParameter
parameterFuncs.TOOLTIPFRAGMENT = self._makeTooltipFragmentParameter
parameterFuncs.VANDAL = self._makeVandalTemplateParameter
self._params = setmetatable({}, {
__index = function (t, k)
local param
if parameterFuncs[k] then
param = parameterFuncs[k](self)
end
param = param or ''
t[k] = param
return param
end
})
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 pagename = self._protectionObj.title.prefixedText
if self._protectionObj.action == 'move' then
-- We need the move log link.
return makeFullUrl(
'Special:Log',
{type = 'move', page = pagename},
self:_getExpandedMessage('current-version-move-display')
)
else
-- We need the history link.
return makeFullUrl(
pagename,
{action = 'history'},
self:_getExpandedMessage('current-version-edit-display')
)
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 edit request type.
local requestType
if action == 'edit' then
if level == 'autoconfirmed' then
requestType = 'semi'
elseif level == 'extendedconfirmed' then
requestType = 'extended'
elseif level == 'templateeditor' then
requestType = 'template'
end
end
requestType = requestType or 'full'
-- Get the display value.
local display = self:_getExpandedMessage('edit-request-display')
return mEditRequest._link{type = requestType, display = display}
end
function Blurb:_makeExpiryParameter()
local expiry = self._protectionObj.expiry
if type(expiry) == 'number' then
return self:_formatDate(expiry)
else
return expiry
end
end
function Blurb:_makeExplanationBlurbParameter()
-- Cover special cases first.
if self._protectionObj.title.namespace == 8 then
-- MediaWiki namespace
return self:_getExpandedMessage('explanation-blurb-nounprotect')
end
-- Get explanation blurb table keys
local action = self._protectionObj.action
local level = self._protectionObj.level
local talkKey = self._protectionObj.title.isTalkPage and 'talk' or 'subject'
-- Find the message in the explanation blurb table and substitute any
-- parameters.
local explanations = self._cfg.explanationBlurbs
local msg
if explanations[action][level] and explanations[action][level][talkKey] then
msg = explanations[action][level][talkKey]
elseif explanations[action][level] and explanations[action][level].default then
msg = explanations[action][level].default
elseif explanations[action].default and explanations[action].default[talkKey] then
msg = explanations[action].default[talkKey]
elseif explanations[action].default and explanations[action].default.default then
msg = explanations[action].default.default
else
error(string.format(
'could not find explanation blurb for action "%s", level "%s" and talk key "%s"',
action,
level,
talkKey
), 8)
end
return self:_substituteParameters(msg)
end
function Blurb:_makeImageLinkParameter()
local imageLinks = self._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()
if self._protectionObj:isTemporary() then
return self:_getExpandedMessage('intro-blurb-expiry')
else
return self:_getExpandedMessage('intro-blurb-noexpiry')
end
end
function Blurb:_makeIntroFragmentParameter()
if self._protectionObj:isTemporary() then
return self:_getExpandedMessage('intro-fragment-expiry')
else
return self:_getExpandedMessage('intro-fragment-noexpiry')
end
end
function Blurb:_makePagetypeParameter()
local pagetypes = self._cfg.pagetypes
return pagetypes[self._protectionObj.title.namespace]
or pagetypes.default
or error('no default pagetype defined', 8)
end
function Blurb:_makeProtectionBlurbParameter()
local protectionBlurbs = self._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 protectionBlurbs.edit.default', 8)
end
return self:_substituteParameters(msg)
end
function Blurb:_makeProtectionDateParameter()
local protectionDate = self._protectionObj.protectionDate
if type(protectionDate) == 'number' then
return self:_formatDate(protectionDate)
else
return protectionDate
end
end
function Blurb:_makeProtectionLevelParameter()
local protectionLevels = self._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 protectionLevels.edit.default', 8)
end
return self:_substituteParameters(msg)
end
function Blurb:_makeProtectionLogParameter()
local pagename = self._protectionObj.title.prefixedText
if self._protectionObj.action == 'autoreview' then
-- We need the pending changes log.
return makeFullUrl(
'Special:Log',
{type = 'stable', page = pagename},
self:_getExpandedMessage('pc-log-display')
)
else
-- We need the protection log.
return makeFullUrl(
'Special:Log',
{type = 'protect', page = pagename},
self:_getExpandedMessage('protection-log-display')
)
end
end
function Blurb:_makeTalkPageParameter()
return string.format(
'[[%s:%s#%s|%s]]',
mw.site.namespaces[self._protectionObj.title.namespace].talk.name,
self._protectionObj.title.text,
self._args.section or 'top',
self:_getExpandedMessage('talk-page-link-display')
)
end
function Blurb:_makeTooltipBlurbParameter()
if self._protectionObj:isTemporary() then
return self:_getExpandedMessage('tooltip-blurb-expiry')
else
return self:_getExpandedMessage('tooltip-blurb-noexpiry')
end
end
function Blurb:_makeTooltipFragmentParameter()
if self._protectionObj:isTemporary() then
return self:_getExpandedMessage('tooltip-fragment-expiry')
else
return self:_getExpandedMessage('tooltip-fragment-noexpiry')
end
end
function Blurb:_makeVandalTemplateParameter()
return require('Module:Vandal-m')._main{
self._args.user or self._protectionObj.title.baseText
}
end
-- Public methods --
function Blurb:makeBannerText(key)
-- Validate input.
if not key or not Blurb.bannerTextFields[key] then
error(string.format(
'"%s" is not a valid banner config field',
tostring(key)
), 2)
end
-- Generate the text.
local msg = self._protectionObj.bannerConfig[key]
if type(msg) == 'string' then
return self:_substituteParameters(msg)
elseif type(msg) == 'function' then
msg = msg(self._protectionObj, self._args)
if type(msg) ~= 'string' then
error(string.format(
'bad output from banner config function with key "%s"'
.. ' (expected string, got %s)',
tostring(key),
type(msg)
), 4)
end
return self:_substituteParameters(msg)
end
end
--------------------------------------------------------------------------------
-- BannerTemplate class
--------------------------------------------------------------------------------
local BannerTemplate = {}
BannerTemplate.__index = BannerTemplate
function BannerTemplate.new(protectionObj, cfg)
local obj = {}
obj._cfg = cfg
-- Set the image filename.
local imageFilename = protectionObj.bannerConfig.image
if imageFilename then
obj._imageFilename = imageFilename
else
-- If an image filename isn't specified explicitly in the banner config,
-- generate it from the protection status and the namespace.
local action = protectionObj.action
local level = protectionObj.level
local namespace = protectionObj.title.namespace
local reason = protectionObj.reason
-- Deal with special cases first.
if (
namespace == 10
or namespace == 828
or reason and obj._cfg.indefImageReasons[reason]
)
and action == 'edit'
and level == 'sysop'
and not protectionObj:isTemporary()
then
-- Fully protected modules and templates get the special red "indef"
-- padlock.
obj._imageFilename = obj._cfg.msg['image-filename-indef']
else
-- Deal with regular protection types.
local images = obj._cfg.images
if images[action] then
if images[action][level] then
obj._imageFilename = images[action][level]
elseif images[action].default then
obj._imageFilename = images[action].default
end
end
end
end
return setmetatable(obj, BannerTemplate)
end
function BannerTemplate:renderImage()
local filename = self._imageFilename
or self._cfg.msg['image-filename-default']
or 'Transparent.gif'
return makeFileLink{
file = filename,
size = (self.imageWidth or 20) .. 'px',
alt = self._imageAlt,
link = self._imageLink,
caption = self.imageCaption
}
end
--------------------------------------------------------------------------------
-- Banner class
--------------------------------------------------------------------------------
local Banner = setmetatable({}, BannerTemplate)
Banner.__index = Banner
function Banner.new(protectionObj, blurbObj, cfg)
local obj = BannerTemplate.new(protectionObj, cfg) -- This doesn't need the blurb.
obj.imageWidth = 40
obj.imageCaption = blurbObj:makeBannerText('alt') -- Large banners use the alt text for the tooltip.
obj._reasonText = blurbObj:makeBannerText('text')
obj._explanationText = blurbObj:makeBannerText('explanation')
obj._page = protectionObj.title.prefixedText -- Only makes a difference in testing.
return setmetatable(obj, Banner)
end
function Banner:__tostring()
-- Renders the banner.
makeMessageBox = makeMessageBox or require('Module:Message box').main
local reasonText = self._reasonText or error('no reason text set', 2)
local explanationText = self._explanationText
local mbargs = {
page = self._page,
type = 'protection',
image = self:renderImage(),
text = string.format(
"'''%s'''%s",
reasonText,
explanationText and '<br />' .. explanationText or ''
)
}
return makeMessageBox('mbox', mbargs)
end
--------------------------------------------------------------------------------
-- Padlock class
--------------------------------------------------------------------------------
local Padlock = setmetatable({}, BannerTemplate)
Padlock.__index = Padlock
function Padlock.new(protectionObj, blurbObj, cfg)
local obj = BannerTemplate.new(protectionObj, cfg) -- This doesn't need the blurb.
obj.imageWidth = 20
obj.imageCaption = blurbObj:makeBannerText('tooltip')
obj._imageAlt = blurbObj:makeBannerText('alt')
obj._imageLink = blurbObj:makeBannerText('link')
obj._indicatorName = cfg.padlockIndicatorNames[protectionObj.action]
or cfg.padlockIndicatorNames.default
or 'pp-default'
return setmetatable(obj, Padlock)
end
function Padlock:__tostring()
local frame = mw.getCurrentFrame()
-- The nowiki tag helps prevent whitespace at the top of articles.
return frame:extensionTag{name = 'nowiki'} .. frame:extensionTag{
name = 'indicator',
args = {name = self._indicatorName},
content = self:renderImage()
}
end
--------------------------------------------------------------------------------
-- Exports
--------------------------------------------------------------------------------
local p = {}
function p._exportClasses()
-- This is used for testing purposes.
return {
Protection = Protection,
Blurb = Blurb,
BannerTemplate = BannerTemplate,
Banner = Banner,
Padlock = Padlock,
}
end
function p._main(args, cfg, title)
args = args or {}
cfg = cfg or require(CONFIG_MODULE)
local protectionObj = Protection.new(args, cfg, title)
local ret = {}
-- If a page's edit protection is equally or more restrictive than its
-- protection from some other action, then don't bother displaying anything
-- for the other action (except categories).
if protectionObj.action == 'edit' or
args.demolevel or
not getReachableNodes(
cfg.hierarchy,
protectionObj.level
)[effectiveProtectionLevel('edit', protectionObj.title)]
then
-- Initialise the blurb object
local blurbObj = Blurb.new(protectionObj, args, cfg)
-- Render the banner
if protectionObj:isProtected() then
ret[#ret + 1] = tostring(
(yesno(args.small) and Padlock or Banner)
.new(protectionObj, blurbObj, cfg)
)
end
end
-- Render the categories
if yesno(args.category) ~= false then
ret[#ret + 1] = protectionObj:makeCategoryLinks()
end
return table.concat(ret)
end
function p.main(frame, cfg)
cfg = cfg or require(CONFIG_MODULE)
-- Find default args, if any.
local parent = frame.getParent and frame:getParent()
local defaultArgs = parent and cfg.wrappers[parent:getTitle():gsub('/sandbox$', '')]
-- Find user args, and use the parent frame if we are being called from a
-- wrapper template.
getArgs = getArgs or require('Module:Arguments').getArgs
local userArgs = getArgs(frame, {
parentOnly = defaultArgs,
frameOnly = not defaultArgs
})
-- Build the args table. User-specified args overwrite default args.
local args = {}
for k, v in pairs(defaultArgs or {}) do
args[k] = v
end
for k, v in pairs(userArgs) do
args[k] = v
end
return p._main(args, cfg)
end
return p