Module:Message box

-- This is a meta-module for producing message box templates, including , Template:Ambox, Template:Imbox, Template:Tmbox, , Template:Cmbox and Template:Fmbox.

-- Require necessary modules. local htmlBuilder = require('Module:HtmlBuilder') 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()

-- Set aliases for often-used functions to reduce table lookups. local format = mw.ustring.format local tinsert = table.insert local tconcat = table.concat

local box = {}

local function getTitleObject(page)

   if type(page) == 'string' then
       -- 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, page)
       if success then
           return title
       end
   end

end

local function presentButBlank(s)

   if type(s) ~= 'string' then return end
   if s and not mw.ustring.find(s, '%S') then
       return true
   else
       return false
   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
       tinsert(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
           tinsert(nums, tonumber(num))
       end
   end
   table.sort(nums)
   return nums

end

function box.getNamespaceId(ns)

   if not ns then return end
   if type(ns) == 'string' then
       ns = lang:ucfirst(mw.ustring.lower(ns))
       if ns == 'Main' then
           ns = 0
       end
   end
   local nsTable = mw.site.namespaces[ns]
   if nsTable then
       return nsTable.id
   end

end

function box.getMboxType(nsid)

   -- Gets the mbox type from a namespace number.
   if nsid == 0 then
       return 'ambox' -- main namespace
   elseif nsid == 6 then
       return 'imbox' -- file namespace
   elseif nsid == 14 then
       return 'cmbox' -- category namespace
   else
       local nsTable = mw.site.namespaces[nsid]
       if nsTable and nsTable.isTalk then
           return 'tmbox' -- any talk namespace
       else
           return 'ombox' -- other namespaces or invalid input
       end
   end

end

function box:addCat(ns, cat, sort)

   if type(cat) ~= 'string' then return end
   local nsVals = {'main', 'template', 'all'}
   local tname
   for i, val in ipairs(nsVals) do
       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(, cat, sort))
   else
       tinsert(self[tname], format(, cat))
   end

end

function box:addClass(class)

   if type(class) ~= 'string' then return end
   self.classes = self.classes or {}
   tinsert(self.classes, class)

end

function box:setTitle(args)

   -- Get the title object and the namespace.
   local pageTitle = getTitleObject(args.page ~=  and args.page)
   self.title = pageTitle or mw.title.getCurrentTitle()
   local demospace = box.getNamespaceId(args.demospace ~=  and args.demospace)
   self.nsid = demospace or self.title.namespace

end

function box:getConfig(boxType)

   -- Get the box config data from the data page.
   if boxType == 'mbox' then
       boxType = box.getMboxType(self.nsid)
   end
   local cfgTables = mw.loadData('Module:Message box/configuration')
   local cfg = cfgTables[boxType]
   if not cfg then
       local boxTypes = {}
       for k, v in pairs(dataTables) do
           tinsert(boxTypes, format('"%s"', k))
       end
       tinsert(boxTypes, '"mbox"')
       error(format('invalid message box type "%s"; valid types are %s', tostring(boxType), mw.text.listToText(boxTypes)), 2)
   end
   return cfg

end

function box:removeBlankArgs(cfg, args)

   -- Only allow blank arguments for the parameter names listed in cfg.allowBlankParams.
   local newArgs = {}
   for k, v in pairs(args) do
       for i, param in ipairs(cfg.allowBlankParams or {}) do
           if v ~=  or k == param then
               newArgs[k] = v
           end
       end
   end
   return newArgs

end

function box:setBoxParameters(cfg, args)

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

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

   -- Find whether we are using a small message box.
   self.isSmall = cfg.allowSmall and (args.small == 'yes' or args.small == true) and true or false

   -- 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.style = args.style

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

   -- Process data for collapsible text fields. At the moment these are only used in Template:Ambox.
   self.useCollapsibleTextFields = cfg.useCollapsibleTextFields 
   if self.useCollapsibleTextFields then
       self.name = args.name
       local nameTitle = getTitleObject(name)
       self.isTemplatePage = nameTitle and title.prefixedText == ('Template:' .. nameTitle.text) and true or false

       -- Get the self.issue value.
       local sect = args.sect
       if presentButBlank(sect) then
           sect = 'This ' .. (cfg.sectionDefault or 'page')
       elseif type(sect) == 'string' then
           sect = 'This ' .. sect
       else
           sect = nil
       end
       local issue = args.issue
       issue = type(issue) == 'string' and issue or nil
       local text = args.text
       text = type(text) == 'string' and text or nil
       local issues = {}
       tinsert(issues, sect)
       tinsert(issues, issue)
       tinsert(issues, text)
       self.issue = tconcat(issues, ' ')

       -- Get the self.talk value.
       local talk = args.talk
       if presentButBlank(talk) and self.isTemplatePage then
           talk = '#'
       end
       if talk then
           -- 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(title.talkPageTitle, 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.', talkText, talk, talkTitle.prefixedText)
               else
                   talkText = format('%s the talk page.', talkText, talkTitle.prefixedText, talk)
               end
               self.talk = talkText
           end
       end

       -- Get other values.
       self.fix = args.fix
       local date = args.date
       self.date = date and format(" (%s)", date)
       if presentButBlank(self.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,
   -- and also by ambox when small=yes.
   if self.isSmall then
       if self.useCollapsibleTextFields then
           self.text = args.smalltext or self.issue
       else
           self.text = args.smalltext or args.text
       end
   else
       self.text = args.text
   end

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

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

   -- Left image settings.
   local imageCheckBlank = cfg.imageCheckBlank
   local imageLeft = self.isSmall and args.smallimage or args.image
   if imageLeft ~= 'none' and not imageCheckBlank or imageLeft ~= 'none' and imageCheckBlank and image ~= 'blank' then
       self.imageLeft = imageLeft
       if not imageLeft then
           local imageSize = self.isSmall and (cfg.imageSmallSize or '30x30px') or '40x40px'
           self.imageLeft = format('%s', 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

   -- Add mainspace categories. At the moment these are only used in Template:Ambox.
   if cfg.allowMainspaceCategories then
       if args.cat then
           args.cat1 = args.cat
       end
       self.catNums = getArgNums(args, 'cat')
       if args.category then
           args.category1 = args.category
       end
       self.categoryNums = getArgNums(args, 'category')
       self.categoryParamNums = union(self.catNums, self.categoryNums)
       -- The following is roughly equivalent to the old Template:Ambox/category.
       local date = args.date
       date = type(date) == 'string' and date
       local preposition = 'from'
       for _, num in ipairs(self.categoryParamNums) 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 then
               local catTitle = format('%s %s %s', mainCat, preposition, date)
               self:addCat('main', catTitle)
               catTitle = getTitleObject('Category:' .. catTitle)
               if not catTitle or not catTitle.exists then
                   self:addCat('main', 'Articles with invalid date parameter in template')
               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.
   self.isTemplatePage = type(self.name) == 'string' and title.prefixedText == ('Template:' .. self.name)
   if cfg.templateCategory then
       if self.name then
           if self.isTemplatePage then
               self:addCat('template', cfg.templateCategory)
           end
       elseif not self.title.isSubpage then
           self:addCat('template', cfg.templateCategory)
       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 title.prefixedText == ('Template:' .. 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.
   if self.invalidType then
       local allSort = (nsid == 0 and 'Main:' or ) .. 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

   -- Convert category tables to strings and pass them through Module:Category handler.
   self.categories = categoryHandler{
       main = tconcat(self.mainCats or {}),
       template = tconcat(self.templateCats or {}),
       all = tconcat(self.allCats or {}),
       nocat = args.nocat,
       demospace = self.demospace and args.demospace or nil,
       page = self.pageTitle and pageTitle.prefixedText or nil
   }

end

function box:export()

   local root = htmlBuilder.create()

   -- Add the subst check error.
   if self.isSubstituted and self.name then
       root
           .tag('b')
               .addClass('error')
               .wikitext(format(
                   'Template %s%s%s has been incorrectly substituted.',
                   mw.text.nowiki('Template:'), self.name, mw.text.nowiki('')
               ))
   end

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

   -- 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.
           -- Not sure why only some box types use divs, but it probably has something to do
           -- with that style="width: 52px;". @TODO: find out exactly what this does and fix this comment.
           imageLeftCell = imageLeftCell.tag('div').css('width', '52px') 
       end
       imageLeftCell
           .wikitext(self.imageLeft)
   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)
   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)
       local textCellSpan = textCell.tag('span')
       textCellSpan
           .addClass('mbox-text-span')
           .wikitext(self.issue)
       if not 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 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.
   if self.imageRight then
       local imageRightCell = row.tag('td').addClass('mbox-imageright')
       if not self.imageCellDiv then
           imageRightCell = imageRightCell.tag('div').css('width', '52px') -- If we are using a div, redefine imageRightCell so that the image is inside it.
       end
       imageRightCell
           .wikitext(self.imageRight)
   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)
               .wikitext(self.below)
   end

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

   -- Add categories.
   root
       .wikitext(self.categories)

   return tostring(root)

end

local function makeBox(boxType, args)

   box:setTitle(args)
   local cfg = box:getConfig(boxType)
   args = box:removeBlankArgs(cfg, args)
   box:setBoxParameters(cfg, args)
   return box:export()

end

local function makeWrapper(boxType)

   return function (frame)
       -- If called via #invoke, use the args passed into the invoking
       -- template, or the args passed to #invoke if any exist. Otherwise
       -- assume args are being passed directly in from the debug console
       -- or from another Lua module.
       local args
       if frame == mw.getCurrentFrame() then
           args = frame:getParent().args
           for k, v in pairs(frame.args) do
               args = frame.args
               break
           end
       else
           args = frame
       end
       return makeBox(boxType, args)
   end

end

return {

   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')

}