Difference between revisions of "Module:Message box"

From Game Detectives Wiki
Jump to: navigation, search
(remove the presentButBlank function, fix bug with blank date values, and trim whitespace from all arguments)
(Oops)
 
(46 intermediate revisions by 15 users not shown)
Line 1: Line 1:
-- This is a meta-module for producing message box templates, including {{mbox}}, {{ambox}}, {{imbox}}, {{tmbox}}, {{ombox}}, {{cmbox}} and {{fmbox}}.
+
-- This is a meta-module for producing message box templates, including
 +
-- {{mbox}}, {{ambox}}, {{imbox}}, {{tmbox}}, {{ombox}}, {{cmbox}} and {{fmbox}}.
  
-- Require necessary modules.
+
-- Load necessary modules.
local htmlBuilder = require('Module:HtmlBuilder')
+
require('Module:No globals')
local categoryHandler = require('Module:Category handler').main
+
local getArgs
 +
local categoryHandler = require('Module:Category handler')._main
 
local yesno = require('Module:Yesno')
 
local yesno = require('Module:Yesno')
  
Line 9: Line 11:
 
local lang = mw.language.getContentLanguage()
 
local lang = mw.language.getContentLanguage()
  
-- Set aliases for often-used functions to reduce table lookups.
+
--------------------------------------------------------------------------------
local format = mw.ustring.format
+
-- Helper functions
local tinsert = table.insert
+
--------------------------------------------------------------------------------
local tconcat = table.concat
 
local trim = mw.text.trim
 
  
local box = {}
+
local function getTitleObject(...)
 
+
-- Get the title object, passing the function through pcall
local function getTitleObject(page)
+
-- in case we are over the expensive function count limit.
    if type(page) == 'string' then
+
local success, title = pcall(mw.title.new, ...)
        -- Get the title object, passing the function through pcall  
+
if success then
        -- in case we are over the expensive function count limit.
+
return title
        local success, title = pcall(mw.title.new, page)
+
end
        if success then
 
            return title
 
        end
 
    end
 
 
end
 
end
  
 
local function union(t1, t2)
 
local function union(t1, t2)
    -- Returns the union of two arrays.
+
-- Returns the union of two arrays.
    local vals = {}
+
local vals = {}
    for i, v in ipairs(t1) do
+
for i, v in ipairs(t1) do
        vals[v] = true
+
vals[v] = true
    end
+
end
    for i, v in ipairs(t2) do
+
for i, v in ipairs(t2) do
        vals[v] = true
+
vals[v] = true
    end
+
end
    local ret = {}
+
local ret = {}
    for k in pairs(vals) do
+
for k in pairs(vals) do
        tinsert(ret, k)
+
table.insert(ret, k)
    end
+
end
    table.sort(ret)
+
table.sort(ret)
    return ret
+
return ret
 
end
 
end
  
 
local function getArgNums(args, prefix)
 
local function getArgNums(args, prefix)
    local nums = {}
+
local nums = {}
    for k, v in pairs(args) do
+
for k, v in pairs(args) do
        local num = mw.ustring.match(tostring(k), '^' .. prefix .. '([1-9]%d*)$')
+
local num = mw.ustring.match(tostring(k), '^' .. prefix .. '([1-9]%d*)$')
        if num then
+
if num then
            tinsert(nums, tonumber(num))
+
table.insert(nums, tonumber(num))
        end
+
end
    end
+
end
    table.sort(nums)
+
table.sort(nums)
    return nums
+
return nums
 
end
 
end
  
function box.getNamespaceId(ns)
+
--------------------------------------------------------------------------------
    if not ns then return end
+
-- Box class definition
    if type(ns) == 'string' then
+
--------------------------------------------------------------------------------
        ns = lang:ucfirst(mw.ustring.lower(ns))
+
 
        if ns == 'Main' then
+
local MessageBox = {}
            ns = 0
+
MessageBox.__index = MessageBox
        end
+
 
    end
+
function MessageBox.new(boxType, args, cfg)
    local nsTable = mw.site.namespaces[ns]
+
args = args or {}
    if nsTable then
+
local obj = {}
        return nsTable.id
+
 
    end
+
-- Set the title object and the namespace.
 +
obj.title = getTitleObject(args.page) or mw.title.getCurrentTitle()
 +
 
 +
-- Set the config for our box type.
 +
obj.cfg = cfg[boxType]
 +
if not obj.cfg then
 +
local ns = obj.title.namespace
 +
-- boxType is "mbox" or invalid input
 +
if ns == 0 then
 +
obj.cfg = cfg.ambox -- main namespace
 +
elseif ns == 6 then
 +
obj.cfg = cfg.imbox -- file namespace
 +
elseif ns == 14 then
 +
obj.cfg = cfg.cmbox -- category namespace
 +
else
 +
local nsTable = mw.site.namespaces[ns]
 +
if nsTable and nsTable.isTalk then
 +
obj.cfg = cfg.tmbox -- any talk namespace
 +
else
 +
obj.cfg = cfg.ombox -- other namespaces or invalid input
 +
end
 +
end
 +
end
 +
 
 +
-- Set the arguments, and remove all blank arguments except for the ones
 +
-- listed in cfg.allowBlankParams.
 +
do
 +
local newArgs = {}
 +
for k, v in pairs(args) do
 +
if v ~= '' then
 +
newArgs[k] = v
 +
end
 +
end
 +
for i, param in ipairs(obj.cfg.allowBlankParams or {}) do
 +
newArgs[param] = args[param]
 +
end
 +
obj.args = newArgs
 +
end
 +
 
 +
-- Define internal data structure.
 +
obj.categories = {}
 +
obj.classes = {}
 +
 
 +
return setmetatable(obj, MessageBox)
 
end
 
end
  
function box.getMboxType(nsid)
+
function MessageBox:addCat(ns, cat, sort)
    -- Gets the mbox type from a namespace number.
+
if not cat then
    if nsid == 0 then
+
return nil
        return 'ambox' -- main namespace
+
end
    elseif nsid == 6 then
+
if sort then
        return 'imbox' -- file namespace
+
cat = string.format('[[Category:%s|%s]]', cat, sort)
    elseif nsid == 14 then
+
else
        return 'cmbox' -- category namespace
+
cat = string.format('[[Category:%s]]', cat)
    else
+
end
        local nsTable = mw.site.namespaces[nsid]
+
self.categories[ns] = self.categories[ns] or {}
        if nsTable and nsTable.isTalk then
+
table.insert(self.categories[ns], cat)
            return 'tmbox' -- any talk namespace
 
        else
 
            return 'ombox' -- other namespaces or invalid input
 
        end
 
    end
 
 
end
 
end
  
function box:addCat(ns, cat, sort)
+
function MessageBox:addClass(class)
    if type(cat) ~= 'string' then return end
+
if not class then
    local nsVals = {'main', 'template', 'all'}
+
return nil
    local tname
+
end
    for i, val in ipairs(nsVals) do
+
table.insert(self.classes, class)
        if ns == val then
 
            tname = ns .. 'Cats'
 
        end
 
    end
 
    if not tname then
 
        for i, val in ipairs(nsVals) do
 
            nsVals[i] = format('"%s"', val)
 
        end
 
        error('invalid ns parameter passed to box:addCat; valid values are ' .. mw.text.listToText(nsVals, nil, ' or '))
 
    end
 
    self[tname] = self[tname] or {}
 
    if type(sort) == 'string' then
 
        tinsert(self[tname], format('[[Category:%s|%s]]', cat, sort))
 
    else
 
        tinsert(self[tname], format('[[Category:%s]]', cat))
 
    end
 
 
end
 
end
  
function box:addClass(class)
+
function MessageBox:setParameters()
    if type(class) ~= 'string' then return end
+
local args = self.args
    self.classes = self.classes or {}
+
local cfg = self.cfg
    tinsert(self.classes, class)
+
 
end
+
-- Get type data.
 +
self.type = args.type
 +
local typeData = cfg.types[self.type]
 +
self.invalidTypeError = cfg.showInvalidTypeError
 +
and self.type
 +
and not typeData
 +
typeData = typeData or cfg.types[cfg.default]
 +
self.typeClass = typeData.class
 +
self.typeImage = typeData.image
 +
 
 +
-- Find if the box has been wrongly substituted.
 +
self.isSubstituted = cfg.substCheck and args.subst == 'SUBST'
 +
 
 +
-- Find whether we are using a small message box.
 +
self.isSmall = cfg.allowSmall and (
 +
cfg.smallParam and args.small == cfg.smallParam
 +
or not cfg.smallParam and yesno(args.small)
 +
)
 +
 
 +
-- Add attributes, classes and styles.
 +
self.id = args.id
 +
self:addClass(
 +
cfg.usePlainlinksParam and yesno(args.plainlinks or true) and 'plainlinks'
 +
)
 +
for _, class in ipairs(cfg.classes or {}) do
 +
self:addClass(class)
 +
end
 +
if self.isSmall then
 +
self:addClass(cfg.smallClass or 'mbox-small')
 +
end
 +
self:addClass(self.typeClass)
 +
self:addClass(args.class)
 +
self.style = args.style
 +
self.attrs = args.attrs
  
function box:setTitle(args)
+
-- Set text style.
    -- Get the title object and the namespace.
+
self.textstyle = args.textstyle
    self.pageTitle = getTitleObject(args.page ~= '' and args.page)
 
    self.title = self.pageTitle or mw.title.getCurrentTitle()
 
    self.demospace = args.demospace ~= '' and args.demospace or nil
 
    self.nsid = box.getNamespaceId(self.demospace) or self.title.namespace
 
end
 
  
function box:getConfig(boxType)
+
-- Find if we are on the template page or not. This functionality is only
    -- Get the box config data from the data page.
+
-- used if useCollapsibleTextFields is set, or if both cfg.templateCategory
    if boxType == 'mbox' then
+
-- and cfg.templateCategoryRequireName are set.
        boxType = box.getMboxType(self.nsid)
+
self.useCollapsibleTextFields = cfg.useCollapsibleTextFields
    end
+
if self.useCollapsibleTextFields
    local cfgTables = mw.loadData('Module:Message box/configuration')
+
or cfg.templateCategory
    local cfg = cfgTables[boxType]
+
and cfg.templateCategoryRequireName
    if not cfg then
+
then
        local boxTypes = {}
+
self.name = args.name
        for k, v in pairs(dataTables) do
+
if self.name then
            tinsert(boxTypes, format('"%s"', k))
+
local templateName = mw.ustring.match(
        end
+
self.name,
        tinsert(boxTypes, '"mbox"')
+
'^[tT][eE][mM][pP][lL][aA][tT][eE][%s_]*:[%s_]*(.*)$'
        error(format('invalid message box type "%s"; valid types are %s', tostring(boxType), mw.text.listToText(boxTypes)), 2)
+
) or self.name
    end
+
templateName = 'Template:' .. templateName
    return cfg
+
self.templateTitle = getTitleObject(templateName)
end
+
end
   
+
self.isTemplatePage = self.templateTitle
function box:removeBlankArgs(cfg, args)
+
and mw.title.equals(self.title, self.templateTitle)
    -- Only allow blank arguments for the parameter names listed in cfg.allowBlankParams.
+
end
    local newArgs = {}
 
    for k, v in pairs(args) do
 
        if v ~= '' then
 
            newArgs[k] = v
 
        end
 
    end
 
    for i, param in ipairs(cfg.allowBlankParams or {}) do
 
        newArgs[param] = args[param]
 
    end
 
    return newArgs
 
end
 
  
function box:setBoxParameters(cfg, args)
+
-- Process data for collapsible text fields. At the moment these are only
    -- Get type data.
+
-- used in {{ambox}}.
    self.type = args.type
+
if self.useCollapsibleTextFields then
    local typeData = cfg.types[self.type]
+
-- Get the self.issue value.
    self.invalidTypeError = cfg.showInvalidTypeError and self.type and not typeData and true or false
+
if self.isSmall and args.smalltext then
    typeData = typeData or cfg.types[cfg.default]
+
self.issue = args.smalltext
    self.typeClass = typeData.class
+
else
    self.typeImage = typeData.image
+
local sect
 +
if args.sect == '' then
 +
sect = 'This ' .. (cfg.sectionDefault or 'page')
 +
elseif type(args.sect) == 'string' then
 +
sect = 'This ' .. args.sect
 +
end
 +
local issue = args.issue
 +
issue = type(issue) == 'string' and issue ~= '' and issue or nil
 +
local text = args.text
 +
text = type(text) == 'string' and text or nil
 +
local issues = {}
 +
table.insert(issues, sect)
 +
table.insert(issues, issue)
 +
table.insert(issues, text)
 +
self.issue = table.concat(issues, ' ')
 +
end
  
    -- Find if the box has been wrongly substituted.
+
-- Get the self.talk value.
    if cfg.substCheck and args.subst == 'SUBST' then
+
local talk = args.talk
        self.isSubstituted = true
+
-- Show talk links on the template page or template subpages if the talk
    end
+
-- parameter is blank.
 +
if talk == ''
 +
and self.templateTitle
 +
and (
 +
mw.title.equals(self.templateTitle, self.title)
 +
or self.title:isSubpageOf(self.templateTitle)
 +
)
 +
then
 +
talk = '#'
 +
elseif talk == '' then
 +
talk = nil
 +
end
 +
if talk then
 +
-- If the talk value is a talk page, make a link to that page. Else
 +
-- assume that it's a section heading, and make a link to the talk
 +
-- page of the current page with that section heading.
 +
local talkTitle = getTitleObject(talk)
 +
local talkArgIsTalkPage = true
 +
if not talkTitle or not talkTitle.isTalkPage then
 +
talkArgIsTalkPage = false
 +
talkTitle = getTitleObject(
 +
self.title.text,
 +
mw.site.namespaces[self.title.namespace].talk.id
 +
)
 +
end
 +
if talkTitle and talkTitle.exists then
 +
local talkText = 'Relevant discussion may be found on'
 +
if talkArgIsTalkPage then
 +
talkText = string.format(
 +
'%s [[%s|%s]].',
 +
talkText,
 +
talk,
 +
talkTitle.prefixedText
 +
)
 +
else
 +
talkText = string.format(
 +
'%s the [[%s#%s|talk page]].',
 +
talkText,
 +
talkTitle.prefixedText,
 +
talk
 +
)
 +
end
 +
self.talk = talkText
 +
end
 +
end
  
    -- Find whether we are using a small message box.
+
-- Get other values.
    if cfg.allowSmall and (
+
self.fix = args.fix ~= '' and args.fix or nil
            cfg.smallParam and args.small == cfg.smallParam
+
local date
            or not cfg.smallParam and yesno(args.small)
+
if args.date and args.date ~= '' then
    )
+
date = args.date
    then
+
elseif args.date == '' and self.isTemplatePage then
        self.isSmall = true
+
date = lang:formatDate('F Y')
    else
+
end
        self.isSmall = false
+
if date then
    end
+
self.date = string.format(" <small>''(%s)''</small>", date)
           
+
end
 +
self.info = args.info
 +
end
  
    -- Add attributes, classes and styles.
+
-- Set the non-collapsible text field. At the moment this is used by all box
    self.id = args.id
+
-- types other than ambox, and also by ambox when small=yes.
    self:addClass(cfg.usePlainlinksParam and yesno(args.plainlinks or true) and 'plainlinks')
+
if self.isSmall then
    for _, class in ipairs(cfg.classes or {}) do
+
self.text = args.smalltext or args.text
        self:addClass(class)
+
else
    end
+
self.text = args.text
    if self.isSmall then
+
end
        self:addClass(cfg.smallClass or 'mbox-small')
 
    end
 
    self:addClass(self.typeClass)
 
    self:addClass(args.class)
 
    self.style = args.style
 
  
    -- Set text style.
+
-- Set the below row.
    self.textstyle = args.textstyle
+
self.below = cfg.below and args.below
  
    -- Process data for collapsible text fields. At the moment these are only used in {{ambox}}.
+
-- General image settings.
    self.useCollapsibleTextFields = cfg.useCollapsibleTextFields
+
self.imageCellDiv = true
    if self.useCollapsibleTextFields then
+
self.imageEmptyCell = cfg.imageEmptyCell
        self.name = args.name
+
if cfg.imageEmptyCellStyle then
        local nameTitle = getTitleObject(self.name)
+
self.imageEmptyCellStyle = 'border:none;padding:0px;width:1px'
        self.isTemplatePage = nameTitle and self.title.prefixedText == ('Template:' .. nameTitle.text) and true or false
+
end
  
        -- Get the self.issue value.
+
-- Left image settings.
        if self.isSmall and args.smalltext then
+
local imageLeft = self.isSmall and args.smallimage or args.image
            self.issue = args.smalltext
+
if cfg.imageCheckBlank and imageLeft ~= 'blank' and imageLeft ~= 'none'
        else
+
or not cfg.imageCheckBlank and imageLeft ~= 'none'
            local sect
+
then
            if args.sect == '' then
+
self.imageLeft = imageLeft
                sect = 'This ' .. (cfg.sectionDefault or 'page')
+
if not imageLeft then
            elseif type(args.sect) == 'string' then
+
local imageSize = self.isSmall
                sect = 'This ' .. sect
+
and (cfg.imageSmallSize or '30x30px')
            end
+
or '40x40px'
            local issue = args.issue
+
self.imageLeft = string.format('[[File:%s|%s|link=|alt=]]', self.typeImage
            issue = type(issue) == 'string' and issue or nil
+
or 'Imbox notice.png', imageSize)
            local text = args.text
+
end
            text = type(text) == 'string' and text or nil
+
end
            local issues = {}
 
            tinsert(issues, sect)
 
            tinsert(issues, issue)
 
            tinsert(issues, text)
 
            self.issue = tconcat(issues, ' ')
 
        end
 
  
        -- Get the self.talk value.
+
-- Right image settings.
        local talk = args.talk
+
local imageRight = self.isSmall and args.smallimageright or args.imageright
        if talk == '' and self.isTemplatePage then
+
if not (cfg.imageRightNone and imageRight == 'none') then
            talk = '#'
+
self.imageRight = imageRight
        end
+
end
        if talk then
+
end
            -- See if the talk link exists and is for a talk or a content namespace.
 
            local talkTitle = getTitleObject(talk)
 
            if not talkTitle or not talkTitle.isTalkPage then
 
                -- If we couldn't process the talk page link, get the talk page of the current page.
 
                local success
 
                success, talkTitle = pcall(self.title.talkPageTitle, self.title)
 
                if not success then
 
                    talkTitle = nil
 
                end
 
            end
 
            if talkTitle and talkTitle.exists then
 
                local talkText = 'Relevant discussion may be found on'
 
                if talkTitle.isTalkPage then
 
                    talkText = format('%s [[%s|%s]].', talkText, talk, talkTitle.prefixedText)
 
                else
 
                    talkText = format('%s the [[%s#%s|talk page]].', talkText, talkTitle.prefixedText, talk)
 
                end
 
                self.talk = talkText
 
            end
 
        end
 
  
        -- Get other values.
+
function MessageBox:setMainspaceCategories()
        self.fix = args.fix
+
local args = self.args
        if args.date and args.date ~= '' then
+
local cfg = self.cfg
            self.date = format(" <small>''(%s)''</small>", args.date)
 
        elseif args.date == '' and self.isTemplatePage then
 
            self.date = lang:formatDate('F Y')
 
        end
 
        self.info = args.info
 
    end
 
  
    -- Set the non-collapsible text field. At the moment this is used by all box types other than ambox,
+
if not cfg.allowMainspaceCategories then
    -- and also by ambox when small=yes.
+
return nil
    if self.isSmall then
+
end
        self.text = args.smalltext or args.text
 
    else
 
        self.text = args.text
 
    end
 
  
    -- Set the below row.
+
local nums = {}
    self.below = cfg.below and args.below
+
for _, prefix in ipairs{'cat', 'category', 'all'} do
 +
args[prefix .. '1'] = args[prefix]
 +
nums = union(nums, getArgNums(args, prefix))
 +
end
  
    -- General image settings.
+
-- The following is roughly equivalent to the old {{Ambox/category}}.
    self.imageCellDiv = not self.isSmall and cfg.imageCellDiv and true or false
+
local date = args.date
    self.imageEmptyCell = cfg.imageEmptyCell
+
date = type(date) == 'string' and date
    if cfg.imageEmptyCellStyle then
+
local preposition = 'from'
        self.imageEmptyCellStyle = 'border:none;padding:0px;width:1px'
+
for _, num in ipairs(nums) do
    end
+
local mainCat = args['cat' .. tostring(num)]
 +
or args['category' .. tostring(num)]
 +
local allCat = args['all' .. tostring(num)]
 +
mainCat = type(mainCat) == 'string' and mainCat
 +
allCat = type(allCat) == 'string' and allCat
 +
if mainCat and date and date ~= '' then
 +
local catTitle = string.format('%s %s %s', mainCat, preposition, date)
 +
self:addCat(0, catTitle)
 +
catTitle = getTitleObject('Category:' .. catTitle)
 +
if not catTitle or not catTitle.exists then
 +
self:addCat(0, 'Articles with invalid date parameter in template')
 +
end
 +
elseif mainCat and (not date or date == '') then
 +
self:addCat(0, mainCat)
 +
end
 +
if allCat then
 +
self:addCat(0, allCat)
 +
end
 +
end
 +
end
  
    -- Left image settings.
+
function MessageBox:setTemplateCategories()
    local imageLeft = self.isSmall and args.smallimage or args.image
+
local args = self.args
    if cfg.imageCheckBlank and imageLeft ~= 'blank' and imageLeft ~= 'none'
+
local cfg = self.cfg
        or not cfg.imageCheckBlank and imageLeft ~= 'none'
 
    then
 
        self.imageLeft = imageLeft
 
        if not imageLeft then
 
            local imageSize = self.isSmall and (cfg.imageSmallSize or '30x30px') or '40x40px'
 
            self.imageLeft = format('[[File:%s|%s|link=|alt=]]', self.typeImage or 'Imbox notice.png', imageSize)
 
        end
 
    end
 
  
    -- Right image settings.
+
-- Add template categories.
    local imageRight = self.isSmall and args.smallimageright or args.imageright
+
if cfg.templateCategory then
    if not (cfg.imageRightNone and imageRight == 'none') then
+
if cfg.templateCategoryRequireName then
        self.imageRight = imageRight
+
if self.isTemplatePage then
    end
+
self:addCat(10, cfg.templateCategory)
 +
end
 +
elseif not self.title.isSubpage then
 +
self:addCat(10, cfg.templateCategory)
 +
end
 +
end
  
    -- Add mainspace categories. At the moment these are only used in {{ambox}}.
+
-- Add template error categories.
    if cfg.allowMainspaceCategories then
+
if cfg.templateErrorCategory then
        if args.cat then
+
local templateErrorCategory = cfg.templateErrorCategory
            args.cat1 = args.cat
+
local templateCat, templateSort
        end
+
if not self.name and not self.title.isSubpage then
        self.catNums = getArgNums(args, 'cat')
+
templateCat = templateErrorCategory
        if args.category then
+
elseif self.isTemplatePage then
            args.category1 = args.category
+
local paramsToCheck = cfg.templateErrorParamsToCheck or {}
        end
+
local count = 0
        self.categoryNums = getArgNums(args, 'category')
+
for i, param in ipairs(paramsToCheck) do
        self.categoryParamNums = union(self.catNums, self.categoryNums)
+
if not args[param] then
        -- The following is roughly equivalent to the old {{Ambox/category}}.
+
count = count + 1
        local date = args.date
+
end
        date = type(date) == 'string' and date
+
end
        local preposition = 'from'
+
if count > 0 then
        for _, num in ipairs(self.categoryParamNums) do
+
templateCat = templateErrorCategory
            local mainCat = args['cat' .. tostring(num)] or args['category' .. tostring(num)]
+
templateSort = tostring(count)
            local allCat = args['all' .. tostring(num)]
+
end
            mainCat = type(mainCat) == 'string' and mainCat
+
if self.categoryNums and #self.categoryNums > 0 then
            allCat = type(allCat) == 'string' and allCat
+
templateCat = templateErrorCategory
            if mainCat and date then
+
templateSort = 'C'
                local catTitle = format('%s %s %s', mainCat, preposition, date)
+
end
                self:addCat('main', catTitle)
+
end
                catTitle = getTitleObject('Category:' .. catTitle)
+
self:addCat(10, templateCat, templateSort)
                if not catTitle or not catTitle.exists then
+
end
                    self:addCat('main', 'Articles with invalid date parameter in template')
+
end
                end
 
            elseif mainCat and not date then
 
                self:addCat('main', mainCat)
 
            end
 
            if allCat then
 
                self:addCat('main', allCat)
 
            end
 
        end
 
    end
 
  
    -- Add template-namespace categories.
+
function MessageBox:setAllNamespaceCategories()
    self.isTemplatePage = type(self.name) == 'string' and self.title.prefixedText == ('Template:' .. self.name)
+
-- Set categories for all namespaces.
    if cfg.templateCategory then
+
if self.invalidTypeError then
        if self.name then
+
local allSort = (self.title.namespace == 0 and 'Main:' or '') .. self.title.prefixedText
            if self.isTemplatePage then
+
self:addCat('all', 'Wikipedia message box parameter needs fixing', allSort)
                self:addCat('template', cfg.templateCategory)
+
end
            end
+
if self.isSubstituted then
        elseif not self.title.isSubpage then
+
self:addCat('all', 'Pages with incorrectly substituted templates')
            self:addCat('template', cfg.templateCategory)
+
end
        end
+
end
    end
 
   
 
    -- Add template error category.
 
    if cfg.templateErrorCategory then
 
        local templateErrorCategory = cfg.templateErrorCategory
 
        local templateCat, templateSort
 
        if not self.name and not self.title.isSubpage then
 
            templateCat = templateErrorCategory
 
        elseif type(self.name) == 'string' and self.title.prefixedText == ('Template:' .. self.name) then
 
            local paramsToCheck = cfg.templateErrorParamsToCheck or {}
 
            local count = 0
 
            for i, param in ipairs(paramsToCheck) do
 
                if not args[param] then
 
                    count = count + 1
 
                end
 
            end
 
            if count > 0 then
 
                templateCat = templateErrorCategory
 
                templateSort = tostring(count)
 
            end
 
            if self.categoryNums and #self.categoryNums > 0 then
 
                templateCat = templateErrorCategory
 
                templateSort = 'C'
 
            end
 
        end
 
        self:addCat('template', templateCat, templateSort)
 
    end
 
  
    -- Categories for all namespaces.
+
function MessageBox:setCategories()
    if self.invalidTypeError then
+
if self.title.namespace == 0 then
        local allSort = (nsid == 0 and 'Main:' or '') .. self.title.prefixedText
+
self:setMainspaceCategories()
        self:addCat('all', 'Wikipedia message box parameter needs fixing', allSort)
+
elseif self.title.namespace == 10 then
    end
+
self:setTemplateCategories()
    if self.isSubstituted then
+
end
        self:addCat('all', 'Pages with incorrectly substituted templates')
+
self:setAllNamespaceCategories()
    end
+
end
  
    -- Convert category tables to strings and pass them through [[Module:Category handler]].
+
function MessageBox:renderCategories()
    self.categories = categoryHandler{
+
-- Convert category tables to strings and pass them through
        main = tconcat(self.mainCats or {}),
+
-- [[Module:Category handler]].
        template = tconcat(self.templateCats or {}),
+
return categoryHandler{
        all = tconcat(self.allCats or {}),
+
main = table.concat(self.categories[0] or {}),
        nocat = args.nocat,
+
template = table.concat(self.categories[10] or {}),
        demospace = self.demospace,
+
all = table.concat(self.categories.all or {}),
        page = self.pageTitle and self.pageTitle.prefixedText or nil
+
nocat = self.args.nocat,
    }
+
page = self.args.page
 +
}
 
end
 
end
  
function box:export()
+
function MessageBox:export()
    local root = htmlBuilder.create()
+
local root = mw.html.create()
 +
 
 +
-- Add the subst check error.
 +
if self.isSubstituted and self.name then
 +
root:tag('b')
 +
:addClass('error')
 +
:wikitext(string.format(
 +
'Template <code>%s[[Template:%s|%s]]%s</code> has been incorrectly substituted.',
 +
mw.text.nowiki('{{'), self.name, self.name, mw.text.nowiki('}}')
 +
))
 +
end
 +
 
 +
-- Create the box table.
 +
local boxTable = root:tag('table')
 +
boxTable:attr('id', self.id or nil)
 +
for i, class in ipairs(self.classes or {}) do
 +
boxTable:addClass(class or nil)
 +
end
 +
boxTable
 +
:cssText(self.style or nil)
 +
:attr('role', 'presentation')
 +
 
 +
if self.attrs then
 +
boxTable:attr(self.attrs)
 +
end
  
    -- Add the subst check error.
+
-- Add the left-hand image.
    if self.isSubstituted and self.name then
+
local row = boxTable:tag('tr')
        root
+
if self.imageLeft then
            .tag('b')
+
local imageLeftCell = row:tag('td'):addClass('mbox-image')
                .addClass('error')
+
if self.imageCellDiv then
                .wikitext(format(
+
-- If we are using a div, redefine imageLeftCell so that the image
                    'Template <code>%s[[Template:%s|%s]]%s</code> has been incorrectly substituted.',
+
-- is inside it. Divs use style="width: 52px;", which limits the
                    mw.text.nowiki('{{'), self.name, self.name, mw.text.nowiki('}}')
+
-- image width to 52px. If any images in a div are wider than that,
                ))
+
-- they may overlap with the text or cause other display problems.
    end
+
if self.isSmall then
 +
imageLeftCell = imageLeftCell:tag('div'):css('width', '32px')
 +
else
 +
imageLeftCell = imageLeftCell:tag('div'):css('width', '52px')
 +
end
 +
end
 +
imageLeftCell:wikitext(self.imageLeft or nil)
 +
elseif self.imageEmptyCell then
 +
-- Some message boxes define an empty cell if no image is specified, and
 +
-- some don't. The old template code in templates where empty cells are
 +
-- specified gives the following hint: "No image. Cell with some width
 +
-- or padding necessary for text cell to have 100% width."
 +
row:tag('td')
 +
:addClass('mbox-empty-cell')
 +
:cssText(self.imageEmptyCellStyle or nil)
 +
end
  
    -- Create the box table.
+
-- Add the text.
    local boxTable = root.tag('table')
+
local textCell = row:tag('td'):addClass('mbox-text')
    boxTable
+
if self.useCollapsibleTextFields then
        .attr('id', self.id)
+
-- The message box uses advanced text parameters that allow things to be
    for i, class in ipairs(self.classes or {}) do
+
-- collapsible. At the moment, only ambox uses this.
        boxTable
+
textCell:cssText(self.textstyle or nil)
            .addClass(class)
+
local textCellSpan = textCell:tag('span')
    end
+
textCellSpan
    boxTable
+
:addClass('mbox-text-span')
        .cssText(self.style)
+
:wikitext(self.issue or nil)
        .attr('role', 'presentation')
+
if not self.isSmall then
 +
textCellSpan:tag('span')
 +
:addClass('hide-when-compact')
 +
:wikitext(self.talk and (' ' .. self.talk) or nil)
 +
:wikitext(self.fix and (' ' .. self.fix) or nil)
 +
end
 +
textCellSpan:wikitext(self.date and (' ' .. self.date) or nil)
 +
if not self.isSmall then
 +
textCellSpan
 +
:tag('span')
 +
:addClass('hide-when-compact')
 +
:wikitext(self.info and (' ' .. self.info) or nil)
 +
end
 +
else
 +
-- Default text formatting - anything goes.
 +
textCell
 +
:cssText(self.textstyle or nil)
 +
:wikitext(self.text or nil)
 +
end
  
    -- Add the left-hand image.
+
-- Add the right-hand image.
    local row = boxTable.tag('tr')
+
if self.imageRight then
    if self.imageLeft then
+
local imageRightCell = row:tag('td'):addClass('mbox-imageright')
        local imageLeftCell = row.tag('td').addClass('mbox-image')
+
if self.imageCellDiv then
        if self.imageCellDiv then
+
-- If we are using a div, redefine imageRightCell so that the image
            -- If we are using a div, redefine imageLeftCell so that the image is inside it.
+
-- is inside it.
            -- Divs use style="width: 52px;", which limits the image width to 52px. If any
+
if self.isSmall then
            -- images in a div are wider than that, they may overlap with the text or cause
+
imageRightCell = imageRightCell:tag('div'):css('width', '32px')
            -- other display problems.
+
else
            imageLeftCell = imageLeftCell.tag('div').css('width', '52px')  
+
imageRightCell = imageRightCell:tag('div'):css('width', '52px')
        end
+
end
        imageLeftCell
+
end
            .wikitext(self.imageLeft)
+
imageRightCell
    elseif self.imageEmptyCell then
+
:wikitext(self.imageRight or nil)
        -- Some message boxes define an empty cell if no image is specified, and some don't.
+
end
        -- The old template code in templates where empty cells are specified gives the following hint:
 
        -- "No image. Cell with some width or padding necessary for text cell to have 100% width."
 
        row.tag('td')
 
            .addClass('mbox-empty-cell')  
 
            .cssText(self.imageEmptyCellStyle)
 
    end
 
  
    -- Add the text.
+
-- Add the below row.
    local textCell = row.tag('td').addClass('mbox-text')
+
if self.below then
    if self.useCollapsibleTextFields then
+
boxTable:tag('tr')
        -- The message box uses advanced text parameters that allow things to be collapsible. At the
+
:tag('td')
        -- moment, only ambox uses this.
+
:attr('colspan', self.imageRight and '3' or '2')
        textCell
+
:addClass('mbox-text')
            .cssText(self.textstyle)
+
:cssText(self.textstyle or nil)
        local textCellSpan = textCell.tag('span')
+
:wikitext(self.below or nil)
        textCellSpan
+
end
            .addClass('mbox-text-span')
 
            .wikitext(self.issue)
 
        if not self.isSmall then
 
            textCellSpan
 
                .tag('span')
 
                    .addClass('hide-when-compact')
 
                    .wikitext(self.talk and ' ' .. self.talk)
 
                    .wikitext(self.fix and ' ' .. self.fix)
 
        end
 
        textCellSpan
 
            .wikitext(self.date and ' ' .. self.date)
 
        if not self.isSmall then
 
            textCellSpan
 
                .tag('span')
 
                    .addClass('hide-when-compact')
 
                    .wikitext(self.info and ' ' .. self.info)
 
        end
 
    else
 
        -- Default text formatting - anything goes.
 
        textCell
 
            .cssText(self.textstyle)
 
            .wikitext(self.text)
 
    end
 
  
    -- Add the right-hand image.
+
-- Add error message for invalid type parameters.
    if self.imageRight then
+
if self.invalidTypeError then
        local imageRightCell = row.tag('td').addClass('mbox-imageright')
+
root:tag('div')
        if self.imageCellDiv then
+
:css('text-align', 'center')
            imageRightCell = imageRightCell.tag('div').css('width', '52px') -- If we are using a div, redefine imageRightCell so that the image is inside it.
+
:wikitext(string.format(
        end
+
'This message box is using an invalid "type=%s" parameter and needs fixing.',
        imageRightCell
+
self.type or ''
            .wikitext(self.imageRight)
+
))
    end
+
end
  
    -- Add the below row.
+
-- Add categories.
    if self.below then
+
root:wikitext(self:renderCategories() or nil)
        boxTable.tag('tr')
+
 
            .tag('td')
+
return tostring(root)
                .attr('colspan', self.imageRight and '3' or '2')
+
end
                .addClass('mbox-text')
 
                .cssText(self.textstyle)
 
                .wikitext(self.below)
 
    end
 
  
    -- Add error message for invalid type parameters.
+
--------------------------------------------------------------------------------
    if self.invalidTypeError then
+
-- Exports
        root
+
--------------------------------------------------------------------------------
            .tag('div')
 
                .addClass('error')
 
                .css('text-align', 'center')
 
                .wikitext(format('This message box is using an invalid type parameter (<code>type=%s</code>) and needs fixing.', self.type or ''))
 
    end
 
  
    -- Add categories.
+
local p, mt = {}, {}
    root
 
        .wikitext(self.categories)
 
  
    return tostring(root)
+
function p._exportClasses()
 +
-- For testing.
 +
return {
 +
MessageBox = MessageBox
 +
}
 
end
 
end
  
local function makeBox(boxType, args)
+
function p.main(boxType, args, cfgTables)
    box:setTitle(args)
+
local box = MessageBox.new(boxType, args, cfgTables or mw.loadData('Module:Message box/configuration'))
    local cfg = box:getConfig(boxType)
+
box:setParameters()
    args = box:removeBlankArgs(cfg, args)
+
box:setCategories()
    box:setBoxParameters(cfg, args)
+
return box:export()
    return box:export()
 
 
end
 
end
  
local function makeWrapper(boxType)
+
function mt.__index(t, k)
    return function (frame)
+
return function (frame)
        -- If called via #invoke, use the args passed into the invoking
+
if not getArgs then
        -- template, or the args passed to #invoke if any exist. Otherwise
+
getArgs = require('Module:Arguments').getArgs
        -- assume args are being passed directly in from the debug console
+
end
        -- or from another Lua module.
+
return t.main(k, getArgs(frame, {trim = false, removeBlanks = false}))
        local origArgs
+
end
        if frame == mw.getCurrentFrame() then
 
            origArgs = frame:getParent().args
 
            for k, v in pairs(frame.args) do
 
                origArgs = frame.args
 
                break
 
            end
 
        else
 
            origArgs = frame
 
        end
 
        -- Trim whitespace.
 
        local args = {}
 
        for k, v in pairs(origArgs) do
 
            if type(v) == 'string' then
 
                v = trim(v)
 
            end
 
            args[k] = v
 
        end
 
        return makeBox(boxType, args)
 
    end
 
 
end
 
end
  
return {
+
return setmetatable(p, mt)
    box = box,
 
    makeBox = makeBox,
 
    mbox = makeWrapper('mbox'),
 
    ambox = makeWrapper('ambox'),
 
    cmbox = makeWrapper('cmbox'),
 
    fmbox = makeWrapper('fmbox'),
 
    imbox = makeWrapper('imbox'),
 
    ombox = makeWrapper('ombox'),
 
    tmbox = makeWrapper('tmbox')
 
}
 

Latest revision as of 22:34, 28 May 2018

Documentation for this module may be created at Module:Message box/doc

-- This is a meta-module for producing message box templates, including
-- {{mbox}}, {{ambox}}, {{imbox}}, {{tmbox}}, {{ombox}}, {{cmbox}} and {{fmbox}}.

-- Load necessary modules.
require('Module:No globals')
local getArgs
local categoryHandler = require('Module:Category handler')._main
local yesno = require('Module:Yesno')

-- Get a language object for formatDate and ucfirst.
local lang = mw.language.getContentLanguage()

--------------------------------------------------------------------------------
-- Helper functions
--------------------------------------------------------------------------------

local function getTitleObject(...)
	-- Get the title object, passing the function through pcall
	-- in case we are over the expensive function count limit.
	local success, title = pcall(mw.title.new, ...)
	if success then
		return title
	end
end

local function union(t1, t2)
	-- Returns the union of two arrays.
	local vals = {}
	for i, v in ipairs(t1) do
		vals[v] = true
	end
	for i, v in ipairs(t2) do
		vals[v] = true
	end
	local ret = {}
	for k in pairs(vals) do
		table.insert(ret, k)
	end
	table.sort(ret)
	return ret
end

local function getArgNums(args, prefix)
	local nums = {}
	for k, v in pairs(args) do
		local num = mw.ustring.match(tostring(k), '^' .. prefix .. '([1-9]%d*)$')
		if num then
			table.insert(nums, tonumber(num))
		end
	end
	table.sort(nums)
	return nums
end

--------------------------------------------------------------------------------
-- Box class definition
--------------------------------------------------------------------------------

local MessageBox = {}
MessageBox.__index = MessageBox

function MessageBox.new(boxType, args, cfg)
	args = args or {}
	local obj = {}

	-- Set the title object and the namespace.
	obj.title = getTitleObject(args.page) or mw.title.getCurrentTitle()

	-- Set the config for our box type.
	obj.cfg = cfg[boxType]
	if not obj.cfg then
		local ns = obj.title.namespace
		-- boxType is "mbox" or invalid input
		if ns == 0 then
			obj.cfg = cfg.ambox -- main namespace
		elseif ns == 6 then
			obj.cfg = cfg.imbox -- file namespace
		elseif ns == 14 then
			obj.cfg = cfg.cmbox -- category namespace
		else
			local nsTable = mw.site.namespaces[ns]
			if nsTable and nsTable.isTalk then
				obj.cfg = cfg.tmbox -- any talk namespace
			else
				obj.cfg = cfg.ombox -- other namespaces or invalid input
			end
		end
	end

	-- Set the arguments, and remove all blank arguments except for the ones
	-- listed in cfg.allowBlankParams.
	do
		local newArgs = {}
		for k, v in pairs(args) do
			if v ~= '' then
				newArgs[k] = v
			end
		end
		for i, param in ipairs(obj.cfg.allowBlankParams or {}) do
			newArgs[param] = args[param]
		end
		obj.args = newArgs
	end

	-- Define internal data structure.
	obj.categories = {}
	obj.classes = {}

	return setmetatable(obj, MessageBox)
end

function MessageBox:addCat(ns, cat, sort)
	if not cat then
		return nil
	end
	if sort then
		cat = string.format('[[Category:%s|%s]]', cat, sort)
	else
		cat = string.format('[[Category:%s]]', cat)
	end
	self.categories[ns] = self.categories[ns] or {}
	table.insert(self.categories[ns], cat)
end

function MessageBox:addClass(class)
	if not class then
		return nil
	end
	table.insert(self.classes, class)
end

function MessageBox:setParameters()
	local args = self.args
	local cfg = self.cfg

	-- Get type data.
	self.type = args.type
	local typeData = cfg.types[self.type]
	self.invalidTypeError = cfg.showInvalidTypeError
		and self.type
		and not typeData
	typeData = typeData or cfg.types[cfg.default]
	self.typeClass = typeData.class
	self.typeImage = typeData.image

	-- Find if the box has been wrongly substituted.
	self.isSubstituted = cfg.substCheck and args.subst == 'SUBST'

	-- Find whether we are using a small message box.
	self.isSmall = cfg.allowSmall and (
		cfg.smallParam and args.small == cfg.smallParam
		or not cfg.smallParam and yesno(args.small)
	)

	-- Add attributes, classes and styles.
	self.id = args.id
	self:addClass(
		cfg.usePlainlinksParam and yesno(args.plainlinks or true) and 'plainlinks'
	)
	for _, class in ipairs(cfg.classes or {}) do
		self:addClass(class)
	end
	if self.isSmall then
		self:addClass(cfg.smallClass or 'mbox-small')
	end
	self:addClass(self.typeClass)
	self:addClass(args.class)
	self.style = args.style
	self.attrs = args.attrs

	-- Set text style.
	self.textstyle = args.textstyle

	-- Find if we are on the template page or not. This functionality is only
	-- used if useCollapsibleTextFields is set, or if both cfg.templateCategory
	-- and cfg.templateCategoryRequireName are set.
	self.useCollapsibleTextFields = cfg.useCollapsibleTextFields
	if self.useCollapsibleTextFields
		or cfg.templateCategory
		and cfg.templateCategoryRequireName
	then
		self.name = args.name
		if self.name then
			local templateName = mw.ustring.match(
				self.name,
				'^[tT][eE][mM][pP][lL][aA][tT][eE][%s_]*:[%s_]*(.*)$'
			) or self.name
			templateName = 'Template:' .. templateName
			self.templateTitle = getTitleObject(templateName)
		end
		self.isTemplatePage = self.templateTitle
			and mw.title.equals(self.title, self.templateTitle)
	end

	-- Process data for collapsible text fields. At the moment these are only
	-- used in {{ambox}}.
	if self.useCollapsibleTextFields then
		-- Get the self.issue value.
		if self.isSmall and args.smalltext then
			self.issue = args.smalltext
		else
			local sect
			if args.sect == '' then
				sect = 'This ' .. (cfg.sectionDefault or 'page')
			elseif type(args.sect) == 'string' then
				sect = 'This ' .. args.sect
			end
			local issue = args.issue
			issue = type(issue) == 'string' and issue ~= '' and issue or nil
			local text = args.text
			text = type(text) == 'string' and text or nil
			local issues = {}
			table.insert(issues, sect)
			table.insert(issues, issue)
			table.insert(issues, text)
			self.issue = table.concat(issues, ' ')
		end

		-- Get the self.talk value.
		local talk = args.talk
		-- Show talk links on the template page or template subpages if the talk
		-- parameter is blank.
		if talk == ''
			and self.templateTitle
			and (
				mw.title.equals(self.templateTitle, self.title)
				or self.title:isSubpageOf(self.templateTitle)
			)
		then
			talk = '#'
		elseif talk == '' then
			talk = nil
		end
		if talk then
			-- If the talk value is a talk page, make a link to that page. Else
			-- assume that it's a section heading, and make a link to the talk
			-- page of the current page with that section heading.
			local talkTitle = getTitleObject(talk)
			local talkArgIsTalkPage = true
			if not talkTitle or not talkTitle.isTalkPage then
				talkArgIsTalkPage = false
				talkTitle = getTitleObject(
					self.title.text,
					mw.site.namespaces[self.title.namespace].talk.id
				)
			end
			if talkTitle and talkTitle.exists then
				local talkText = 'Relevant discussion may be found on'
				if talkArgIsTalkPage then
					talkText = string.format(
						'%s [[%s|%s]].',
						talkText,
						talk,
						talkTitle.prefixedText
					)
				else
					talkText = string.format(
						'%s the [[%s#%s|talk page]].',
						talkText,
						talkTitle.prefixedText,
						talk
					)
				end
				self.talk = talkText
			end
		end

		-- Get other values.
		self.fix = args.fix ~= '' and args.fix or nil
		local date
		if args.date and args.date ~= '' then
			date = args.date
		elseif args.date == '' and self.isTemplatePage then
			date = lang:formatDate('F Y')
		end
		if date then
			self.date = string.format(" <small>''(%s)''</small>", date)
		end
		self.info = args.info
	end

	-- Set the non-collapsible text field. At the moment this is used by all box
	-- types other than ambox, and also by ambox when small=yes.
	if self.isSmall then
		self.text = args.smalltext or args.text
	else
		self.text = args.text
	end

	-- Set the below row.
	self.below = cfg.below and args.below

	-- General image settings.
	self.imageCellDiv = true
	self.imageEmptyCell = cfg.imageEmptyCell
	if cfg.imageEmptyCellStyle then
		self.imageEmptyCellStyle = 'border:none;padding:0px;width:1px'
	end

	-- Left image settings.
	local imageLeft = self.isSmall and args.smallimage or args.image
	if cfg.imageCheckBlank and imageLeft ~= 'blank' and imageLeft ~= 'none'
		or not cfg.imageCheckBlank and imageLeft ~= 'none'
	then
		self.imageLeft = imageLeft
		if not imageLeft then
			local imageSize = self.isSmall
				and (cfg.imageSmallSize or '30x30px')
				or '40x40px'
			self.imageLeft = string.format('[[File:%s|%s|link=|alt=]]', self.typeImage
				or 'Imbox notice.png', imageSize)
		end
	end

	-- Right image settings.
	local imageRight = self.isSmall and args.smallimageright or args.imageright
	if not (cfg.imageRightNone and imageRight == 'none') then
		self.imageRight = imageRight
	end
end

function MessageBox:setMainspaceCategories()
	local args = self.args
	local cfg = self.cfg

	if not cfg.allowMainspaceCategories then
		return nil
	end

	local nums = {}
	for _, prefix in ipairs{'cat', 'category', 'all'} do
		args[prefix .. '1'] = args[prefix]
		nums = union(nums, getArgNums(args, prefix))
	end

	-- The following is roughly equivalent to the old {{Ambox/category}}.
	local date = args.date
	date = type(date) == 'string' and date
	local preposition = 'from'
	for _, num in ipairs(nums) do
		local mainCat = args['cat' .. tostring(num)]
			or args['category' .. tostring(num)]
		local allCat = args['all' .. tostring(num)]
		mainCat = type(mainCat) == 'string' and mainCat
		allCat = type(allCat) == 'string' and allCat
		if mainCat and date and date ~= '' then
			local catTitle = string.format('%s %s %s', mainCat, preposition, date)
			self:addCat(0, catTitle)
			catTitle = getTitleObject('Category:' .. catTitle)
			if not catTitle or not catTitle.exists then
				self:addCat(0, 'Articles with invalid date parameter in template')
			end
		elseif mainCat and (not date or date == '') then
			self:addCat(0, mainCat)
		end
		if allCat then
			self:addCat(0, allCat)
		end
	end
end

function MessageBox:setTemplateCategories()
	local args = self.args
	local cfg = self.cfg

	-- Add template categories.
	if cfg.templateCategory then
		if cfg.templateCategoryRequireName then
			if self.isTemplatePage then
				self:addCat(10, cfg.templateCategory)
			end
		elseif not self.title.isSubpage then
			self:addCat(10, cfg.templateCategory)
		end
	end

	-- Add template error categories.
	if cfg.templateErrorCategory then
		local templateErrorCategory = cfg.templateErrorCategory
		local templateCat, templateSort
		if not self.name and not self.title.isSubpage then
			templateCat = templateErrorCategory
		elseif self.isTemplatePage then
			local paramsToCheck = cfg.templateErrorParamsToCheck or {}
			local count = 0
			for i, param in ipairs(paramsToCheck) do
				if not args[param] then
					count = count + 1
				end
			end
			if count > 0 then
				templateCat = templateErrorCategory
				templateSort = tostring(count)
			end
			if self.categoryNums and #self.categoryNums > 0 then
				templateCat = templateErrorCategory
				templateSort = 'C'
			end
		end
		self:addCat(10, templateCat, templateSort)
	end
end

function MessageBox:setAllNamespaceCategories()
	-- Set categories for all namespaces.
	if self.invalidTypeError then
		local allSort = (self.title.namespace == 0 and 'Main:' or '') .. self.title.prefixedText
		self:addCat('all', 'Wikipedia message box parameter needs fixing', allSort)
	end
	if self.isSubstituted then
		self:addCat('all', 'Pages with incorrectly substituted templates')
	end
end

function MessageBox:setCategories()
	if self.title.namespace == 0 then
		self:setMainspaceCategories()
	elseif self.title.namespace == 10 then
		self:setTemplateCategories()
	end
	self:setAllNamespaceCategories()
end

function MessageBox:renderCategories()
	-- Convert category tables to strings and pass them through
	-- [[Module:Category handler]].
	return categoryHandler{
		main = table.concat(self.categories[0] or {}),
		template = table.concat(self.categories[10] or {}),
		all = table.concat(self.categories.all or {}),
		nocat = self.args.nocat,
		page = self.args.page
	}
end

function MessageBox:export()
	local root = mw.html.create()

	-- Add the subst check error.
	if self.isSubstituted and self.name then
		root:tag('b')
			:addClass('error')
			:wikitext(string.format(
				'Template <code>%s[[Template:%s|%s]]%s</code> has been incorrectly substituted.',
				mw.text.nowiki('{{'), self.name, self.name, mw.text.nowiki('}}')
			))
	end

	-- Create the box table.
	local boxTable = root:tag('table')
	boxTable:attr('id', self.id or nil)
	for i, class in ipairs(self.classes or {}) do
		boxTable:addClass(class or nil)
	end
	boxTable
		:cssText(self.style or nil)
		:attr('role', 'presentation')

	if self.attrs then
		boxTable:attr(self.attrs)
	end

	-- Add the left-hand image.
	local row = boxTable:tag('tr')
	if self.imageLeft then
		local imageLeftCell = row:tag('td'):addClass('mbox-image')
		if self.imageCellDiv then
			-- If we are using a div, redefine imageLeftCell so that the image
			-- is inside it. Divs use style="width: 52px;", which limits the
			-- image width to 52px. If any images in a div are wider than that,
			-- they may overlap with the text or cause other display problems.
			if self.isSmall then
				imageLeftCell = imageLeftCell:tag('div'):css('width', '32px')
			else
				imageLeftCell = imageLeftCell:tag('div'):css('width', '52px')
			end
		end
		imageLeftCell:wikitext(self.imageLeft or nil)
	elseif self.imageEmptyCell then
		-- Some message boxes define an empty cell if no image is specified, and
		-- some don't. The old template code in templates where empty cells are
		-- specified gives the following hint: "No image. Cell with some width
		-- or padding necessary for text cell to have 100% width."
		row:tag('td')
			:addClass('mbox-empty-cell')
			:cssText(self.imageEmptyCellStyle or nil)
	end

	-- Add the text.
	local textCell = row:tag('td'):addClass('mbox-text')
	if self.useCollapsibleTextFields then
		-- The message box uses advanced text parameters that allow things to be
		-- collapsible. At the moment, only ambox uses this.
		textCell:cssText(self.textstyle or nil)
		local textCellSpan = textCell:tag('span')
		textCellSpan
			:addClass('mbox-text-span')
			:wikitext(self.issue or nil)
		if not self.isSmall then
			textCellSpan:tag('span')
				:addClass('hide-when-compact')
				:wikitext(self.talk and (' ' .. self.talk) or nil)
				:wikitext(self.fix and (' ' .. self.fix) or nil)
		end
		textCellSpan:wikitext(self.date and (' ' .. self.date) or nil)
		if not self.isSmall then
			textCellSpan
				:tag('span')
				:addClass('hide-when-compact')
				:wikitext(self.info and (' ' .. self.info) or nil)
		end
	else
		-- Default text formatting - anything goes.
		textCell
			:cssText(self.textstyle or nil)
			:wikitext(self.text or nil)
	end

	-- Add the right-hand image.
	if self.imageRight then
		local imageRightCell = row:tag('td'):addClass('mbox-imageright')
		if self.imageCellDiv then
			-- If we are using a div, redefine imageRightCell so that the image
			-- is inside it.
			if self.isSmall then
				imageRightCell = imageRightCell:tag('div'):css('width', '32px')
			else
				imageRightCell = imageRightCell:tag('div'):css('width', '52px')
			end
		end
		imageRightCell
			:wikitext(self.imageRight or nil)
	end

	-- Add the below row.
	if self.below then
		boxTable:tag('tr')
			:tag('td')
				:attr('colspan', self.imageRight and '3' or '2')
				:addClass('mbox-text')
				:cssText(self.textstyle or nil)
				:wikitext(self.below or nil)
	end

	-- Add error message for invalid type parameters.
	if self.invalidTypeError then
		root:tag('div')
			:css('text-align', 'center')
			:wikitext(string.format(
				'This message box is using an invalid "type=%s" parameter and needs fixing.',
				self.type or ''
			))
	end

	-- Add categories.
	root:wikitext(self:renderCategories() or nil)

	return tostring(root)
end

--------------------------------------------------------------------------------
-- Exports
--------------------------------------------------------------------------------

local p, mt = {}, {}

function p._exportClasses()
	-- For testing.
	return {
		MessageBox = MessageBox
	}
end

function p.main(boxType, args, cfgTables)
	local box = MessageBox.new(boxType, args, cfgTables or mw.loadData('Module:Message box/configuration'))
	box:setParameters()
	box:setCategories()
	return box:export()
end

function mt.__index(t, k)
	return function (frame)
		if not getArgs then
			getArgs = require('Module:Arguments').getArgs
		end
		return t.main(k, getArgs(frame, {trim = false, removeBlanks = false}))
	end
end

return setmetatable(p, mt)