Module:Infobox: Difference between revisions

Jump to navigation Jump to search
imported>Frietjes
(fix for categories triggering blank data cells per talk page (and spaces to tabs per edits by RexxS))
imported>Galobtter
(Undid revision 956509357 by Frietjes (talk) breaks headers in infobox)
Line 27: Line 27:
s = mw.ustring.gsub(s, '^(%{|[^\r\n]*[\r\n]%s*)' .. marker, '%1')
s = mw.ustring.gsub(s, '^(%{|[^\r\n]*[\r\n]%s*)' .. marker, '%1')
s = mw.ustring.gsub(s, '([\r\n]%{|[^\r\n]*[\r\n]%s*)' .. marker, '%1')
s = mw.ustring.gsub(s, '([\r\n]%{|[^\r\n]*[\r\n]%s*)' .. marker, '%1')
s = mw.ustring.gsub(s, marker .. '(%s*</[Tt][Aa][Bb][Ll][Ee]%s*>)', '%1')
s = mw.ustring.gsub(s, marker .. '(%s*</[Tt][Aa][Bb][Ll][Ee]%s*>)', '%1')
s = mw.ustring.gsub(s, marker .. '(%s*\n|%})', '%1')
s = mw.ustring.gsub(s, marker .. '(%s*\n|%})', '%1')
end
end
if s:match(marker) then
if s:match(marker) then
Line 63: Line 63:


local function union(t1, t2)
local function union(t1, t2)
-- Returns the union of the values of two tables, as a sequence.
    -- Returns the union of the values of two tables, as a sequence.
local vals = {}
    local vals = {}
for k, v in pairs(t1) do
    for k, v in pairs(t1) do
vals[v] = true
        vals[v] = true
end
    end
for k, v in pairs(t2) do
    for k, v in pairs(t2) do
vals[v] = true
        vals[v] = true
end
    end
local ret = {}
    local ret = {}
for k, v in pairs(vals) do
    for k, v in pairs(vals) do
table.insert(ret, k)
        table.insert(ret, k)
end
    end
return ret
    return ret
end
end


local function getArgNums(prefix)
local function getArgNums(prefix)
-- Returns a table containing the numbers of the arguments that exist
    -- Returns a table containing the numbers of the arguments that exist
-- for the specified prefix. For example, if the prefix was 'data', and
    -- for the specified prefix. For example, if the prefix was 'data', and
-- 'data1', 'data2', and 'data5' exist, it would return {1, 2, 5}.
    -- 'data1', 'data2', and 'data5' exist, it would return {1, 2, 5}.
local nums = {}
    local nums = {}
for k, v in pairs(args) do
    for k, v in pairs(args) do
local num = tostring(k):match('^' .. prefix .. '([1-9]%d*)$')
        local num = tostring(k):match('^' .. prefix .. '([1-9]%d*)$')
if num then table.insert(nums, tonumber(num)) end
        if num then table.insert(nums, tonumber(num)) end
end
    end
table.sort(nums)
    table.sort(nums)
return nums
    return nums
end
end


local function addRow(rowArgs)
local function addRow(rowArgs)
-- Adds a row to the infobox, with either a header cell
    -- Adds a row to the infobox, with either a header cell
-- or a label/data cell combination.
    -- or a label/data cell combination.
if rowArgs.header then
    if rowArgs.header then
root
        root
:tag('tr')
            :tag('tr')
:addClass(rowArgs.rowclass)
                :addClass(rowArgs.rowclass)
:cssText(rowArgs.rowstyle)
                :cssText(rowArgs.rowstyle)
:attr('id', rowArgs.rowid)
                :attr('id', rowArgs.rowid)
:tag('th')
                :tag('th')
:attr('colspan', 2)
                    :attr('colspan', 2)
:attr('id', rowArgs.headerid)
                    :attr('id', rowArgs.headerid)
:addClass(rowArgs.class)
                    :addClass(rowArgs.class)
:addClass(args.headerclass)
                    :addClass(args.headerclass)
:css('text-align', 'center')
                    :css('text-align', 'center')
:cssText(args.headerstyle)
                    :cssText(args.headerstyle)
:cssText(rowArgs.rowcellstyle)
                    :cssText(rowArgs.rowcellstyle)
:wikitext(fixChildBoxes(rowArgs.header, 'th'))
                    :wikitext(fixChildBoxes(rowArgs.header, 'th'))
if rowArgs.data then
        if rowArgs.data then
root:wikitext('[[Category:Pages which use infobox templates with ignored data cells]]')
        root:wikitext('[[Category:Pages which use infobox templates with ignored data cells]]')
end
      end
elseif rowArgs.data then
    elseif rowArgs.data then
local row = root:tag('tr')
        local row = root:tag('tr')
row:addClass(rowArgs.rowclass)
        row:addClass(rowArgs.rowclass)
row:cssText(rowArgs.rowstyle)
        row:cssText(rowArgs.rowstyle)
row:attr('id', rowArgs.rowid)
        row:attr('id', rowArgs.rowid)
if rowArgs.label then
        if rowArgs.label then
row
            row
:tag('th')
                :tag('th')
:attr('scope', 'row')
                    :attr('scope', 'row')
:attr('id', rowArgs.labelid)
                    :attr('id', rowArgs.labelid)
:cssText(args.labelstyle)
                    :cssText(args.labelstyle)
:cssText(rowArgs.rowcellstyle)
                    :cssText(rowArgs.rowcellstyle)
:wikitext(rowArgs.label)
                    :wikitext(rowArgs.label)
:done()
                    :done()
end
        end
 
       
local dataCell = row:tag('td')
        local dataCell = row:tag('td')
if not rowArgs.label then
        if not rowArgs.label then  
dataCell
            dataCell
:attr('colspan', 2)
                :attr('colspan', 2)
:css('text-align', 'center')
                :css('text-align', 'center')  
end
        end
dataCell
        dataCell
:attr('id', rowArgs.dataid)
            :attr('id', rowArgs.dataid)
:addClass(rowArgs.class)
            :addClass(rowArgs.class)
:cssText(rowArgs.datastyle)
            :cssText(rowArgs.datastyle)
:cssText(rowArgs.rowcellstyle)
            :cssText(rowArgs.rowcellstyle)
:wikitext(fixChildBoxes(rowArgs.data, 'td'))
            :wikitext(fixChildBoxes(rowArgs.data, 'td'))
end
    end
end
end


local function renderTitle()
local function renderTitle()
if not args.title then return end
    if not args.title then return end


root
    root
:tag('caption')
        :tag('caption')
:addClass(args.titleclass)
            :addClass(args.titleclass)
:cssText(args.titlestyle)
            :cssText(args.titlestyle)
:wikitext(args.title)
            :wikitext(args.title)
end
end


local function renderAboveRow()
local function renderAboveRow()
if not args.above then return end
    if not args.above then return end
 
   
root
    root
:tag('tr')
        :tag('tr')
:tag('th')
            :tag('th')
:attr('colspan', 2)
                :attr('colspan', 2)
:addClass(args.aboveclass)
                :addClass(args.aboveclass)
:css('text-align', 'center')
                :css('text-align', 'center')
:css('font-size', '125%')
                :css('font-size', '125%')
:css('font-weight', 'bold')
                :css('font-weight', 'bold')
:cssText(args.abovestyle)
                :cssText(args.abovestyle)
:wikitext(fixChildBoxes(args.above,'th'))
                :wikitext(fixChildBoxes(args.above,'th'))
end
end


local function renderBelowRow()
local function renderBelowRow()
if not args.below then return end
    if not args.below then return end
 
   
root
    root
:tag('tr')
        :tag('tr')
:tag('td')
            :tag('td')
:attr('colspan', '2')
                :attr('colspan', '2')
:addClass(args.belowclass)
                :addClass(args.belowclass)
:css('text-align', 'center')
                :css('text-align', 'center')
:cssText(args.belowstyle)
                :cssText(args.belowstyle)
:wikitext(fixChildBoxes(args.below,'td'))
                :wikitext(fixChildBoxes(args.below,'td'))
end
end


local function renderSubheaders()
local function renderSubheaders()
if args.subheader then
    if args.subheader then
args.subheader1 = args.subheader
        args.subheader1 = args.subheader
end
    end
if args.subheaderrowclass then
    if args.subheaderrowclass then
args.subheaderrowclass1 = args.subheaderrowclass
        args.subheaderrowclass1 = args.subheaderrowclass
end
    end
local subheadernums = getArgNums('subheader')
    local subheadernums = getArgNums('subheader')
for k, num in ipairs(subheadernums) do
    for k, num in ipairs(subheadernums) do
addRow({
        addRow({
data = args['subheader' .. tostring(num)],
            data = args['subheader' .. tostring(num)],
datastyle = args.subheaderstyle,
            datastyle = args.subheaderstyle,
rowcellstyle = args['subheaderstyle' .. tostring(num)],
            rowcellstyle = args['subheaderstyle' .. tostring(num)],
class = args.subheaderclass,
            class = args.subheaderclass,
rowclass = args['subheaderrowclass' .. tostring(num)]
            rowclass = args['subheaderrowclass' .. tostring(num)]
})
        })
end
    end
end
end


local function renderImages()
local function renderImages()
if args.image then
    if args.image then
args.image1 = args.image
        args.image1 = args.image
end
    end
if args.caption then
    if args.caption then
args.caption1 = args.caption
        args.caption1 = args.caption
end
    end
local imagenums = getArgNums('image')
    local imagenums = getArgNums('image')
for k, num in ipairs(imagenums) do
    for k, num in ipairs(imagenums) do
local caption = args['caption' .. tostring(num)]
        local caption = args['caption' .. tostring(num)]
local data = mw.html.create():wikitext(args['image' .. tostring(num)])
        local data = mw.html.create():wikitext(args['image' .. tostring(num)])
if caption then
        if caption then
data
            data
:tag('div')
                :tag('div')
:cssText(args.captionstyle)
                    :cssText(args.captionstyle)
:wikitext(caption)
                    :wikitext(caption)
end
        end
addRow({
        addRow({
data = tostring(data),
            data = tostring(data),
datastyle = args.imagestyle,
            datastyle = args.imagestyle,
class = args.imageclass,
            class = args.imageclass,
rowclass = args['imagerowclass' .. tostring(num)]
            rowclass = args['imagerowclass' .. tostring(num)]
})
        })
end
    end
end
end


local function renderRows()
local function renderRows()
-- Gets the union of the header and data argument numbers,
    -- Gets the union of the header and data argument numbers,
-- and renders them all in order using addRow.
    -- and renders them all in order using addRow.
local rownums = union(getArgNums('header'), getArgNums('data'))
    local rownums = union(getArgNums('header'), getArgNums('data'))
table.sort(rownums)
    table.sort(rownums)
local row_style
    for k, num in ipairs(rownums) do
for k, num in ipairs(rownums) do
        addRow({
local dval = args['data' .. tostring(num)]
            header = args['header' .. tostring(num)],
if dval and
            label = args['label' .. tostring(num)],
dval:gsub('%[%[%s*[Cc][Aa][Tt][Ee][Gg][Oo][Rr][Yy]%s*:[^]]*]]', ''):match('%S')
            data = args['data' .. tostring(num)],
then
            datastyle = args.datastyle,
row_style = args['rowstyle' .. tostring(num)]
            class = args['class' .. tostring(num)],
else
            rowclass = args['rowclass' .. tostring(num)],
row_style = 'display:none'
            rowstyle = args['rowstyle' .. tostring(num)],
end
            rowcellstyle = args['rowcellstyle' .. tostring(num)],
addRow({
            dataid = args['dataid' .. tostring(num)],
header = args['header' .. tostring(num)],
            labelid = args['labelid' .. tostring(num)],
label = args['label' .. tostring(num)],
            headerid = args['headerid' .. tostring(num)],
data = args['data' .. tostring(num)],
            rowid = args['rowid' .. tostring(num)]
datastyle = args.datastyle,
        })
class = args['class' .. tostring(num)],
    end
rowclass = args['rowclass' .. tostring(num)],
rowstyle = row_style,
rowcellstyle = args['rowcellstyle' .. tostring(num)],
dataid = args['dataid' .. tostring(num)],
labelid = args['labelid' .. tostring(num)],
headerid = args['headerid' .. tostring(num)],
rowid = args['rowid' .. tostring(num)]
})
end
end
end


local function renderNavBar()
local function renderNavBar()
if not args.name then return end
    if not args.name then return end
 
   
root
    root
:tag('tr')
        :tag('tr')
:tag('td')
            :tag('td')
:attr('colspan', '2')
                :attr('colspan', '2')
:css('text-align', 'right')
                :css('text-align', 'right')
:wikitext(navbar{
                :wikitext(navbar{
args.name,
                    args.name,
mini = 1,
                    mini = 1,
})
                })
end
end


local function renderItalicTitle()
local function renderItalicTitle()
local italicTitle = args['italic title'] and mw.ustring.lower(args['italic title'])
    local italicTitle = args['italic title'] and mw.ustring.lower(args['italic title'])
if italicTitle == '' or italicTitle == 'force' or italicTitle == 'yes' then
    if italicTitle == '' or italicTitle == 'force' or italicTitle == 'yes' then
root:wikitext(mw.getCurrentFrame():expandTemplate({title = 'italic title'}))
        root:wikitext(mw.getCurrentFrame():expandTemplate({title = 'italic title'}))
end
    end
end
end


local function renderTrackingCategories()
local function renderTrackingCategories()
if args.decat ~= 'yes' then
    if args.decat ~= 'yes' then
if args.child == 'yes' then
    if args.child == 'yes' then
if args.title then
        if args.title then
root:wikitext('[[Category:Pages which use embedded infobox templates with the title parameter]]')
            root:wikitext('[[Category:Pages which use embedded infobox templates with the title parameter]]')
end
        end
elseif #(getArgNums('data')) == 0 and mw.title.getCurrentTitle().namespace == 0 then
        elseif #(getArgNums('data')) == 0 and mw.title.getCurrentTitle().namespace == 0 then
root:wikitext('[[Category:Articles which use infobox templates with no data rows]]')
            root:wikitext('[[Category:Articles which use infobox templates with no data rows]]')
end
        end
end
    end
end
end


local function _infobox()
local function _infobox()
-- Specify the overall layout of the infobox, with special settings
    -- Specify the overall layout of the infobox, with special settings
-- if the infobox is used as a 'child' inside another infobox.
    -- if the infobox is used as a 'child' inside another infobox.
if args.child ~= 'yes' then
    if args.child ~= 'yes' then
root = mw.html.create('table')
        root = mw.html.create('table')
 
       
root
        root
:addClass((args.subbox ~= 'yes') and 'infobox' or nil)
            :addClass((args.subbox ~= 'yes') and 'infobox' or nil)
:addClass(args.bodyclass)
            :addClass(args.bodyclass)
 
           
if args.subbox == 'yes' then
            if args.subbox == 'yes' then
root
                root
:css('padding', '0')
                    :css('padding', '0')
:css('border', 'none')
                    :css('border', 'none')
:css('margin', '-3px')
                    :css('margin', '-3px')
:css('width', 'auto')
                    :css('width', 'auto')
:css('min-width', '100%')
                    :css('min-width', '100%')
:css('font-size', '100%')
                    :css('font-size', '100%')
:css('clear', 'none')
                    :css('clear', 'none')
:css('float', 'none')
                    :css('float', 'none')
:css('background-color', 'transparent')
                    :css('background-color', 'transparent')
else
            else
root
                root
:css('width', '22em')
                    :css('width', '22em')
end
            end
root
        root
:cssText(args.bodystyle)
            :cssText(args.bodystyle)
 
   
renderTitle()
        renderTitle()
renderAboveRow()
        renderAboveRow()
else
    else
root = mw.html.create()
        root = mw.html.create()
       
        root
            :wikitext(args.title)
    end


root
    renderSubheaders()
:wikitext(args.title)
    renderImages()  
end
    renderRows()  
 
    renderBelowRow()
renderSubheaders()
    renderNavBar()
renderImages()
    renderItalicTitle()
renderRows()
    renderTrackingCategories()
renderBelowRow()
   
renderNavBar()
    return tostring(root)
renderItalicTitle()
renderTrackingCategories()
 
return tostring(root)
end
end


local function preprocessSingleArg(argName)
local function preprocessSingleArg(argName)
-- If the argument exists and isn't blank, add it to the argument table.
    -- If the argument exists and isn't blank, add it to the argument table.
-- Blank arguments are treated as nil to match the behaviour of ParserFunctions.
    -- Blank arguments are treated as nil to match the behaviour of ParserFunctions.
if origArgs[argName] and origArgs[argName] ~= '' then
    if origArgs[argName] and origArgs[argName] ~= '' then
args[argName] = origArgs[argName]
        args[argName] = origArgs[argName]
end
    end
end
end


local function preprocessArgs(prefixTable, step)
local function preprocessArgs(prefixTable, step)
-- Assign the parameters with the given prefixes to the args table, in order, in batches
    -- Assign the parameters with the given prefixes to the args table, in order, in batches
-- of the step size specified. This is to prevent references etc. from appearing in the
    -- of the step size specified. This is to prevent references etc. from appearing in the
-- wrong order. The prefixTable should be an array containing tables, each of which has
    -- wrong order. The prefixTable should be an array containing tables, each of which has
-- two possible fields, a "prefix" string and a "depend" table. The function always parses
    -- two possible fields, a "prefix" string and a "depend" table. The function always parses
-- parameters containing the "prefix" string, but only parses parameters in the "depend"
    -- parameters containing the "prefix" string, but only parses parameters in the "depend"
-- table if the prefix parameter is present and non-blank.
    -- table if the prefix parameter is present and non-blank.
if type(prefixTable) ~= 'table' then
    if type(prefixTable) ~= 'table' then
error("Non-table value detected for the prefix table", 2)
        error("Non-table value detected for the prefix table", 2)
end
    end
if type(step) ~= 'number' then
    if type(step) ~= 'number' then
error("Invalid step value detected", 2)
        error("Invalid step value detected", 2)
end
    end
   
    -- Get arguments without a number suffix, and check for bad input.
    for i,v in ipairs(prefixTable) do
        if type(v) ~= 'table' or type(v.prefix) ~= "string" or (v.depend and type(v.depend) ~= 'table') then
            error('Invalid input detected to preprocessArgs prefix table', 2)
        end
        preprocessSingleArg(v.prefix)
        -- Only parse the depend parameter if the prefix parameter is present and not blank.
        if args[v.prefix] and v.depend then
            for j, dependValue in ipairs(v.depend) do
                if type(dependValue) ~= 'string' then
                    error('Invalid "depend" parameter value detected in preprocessArgs')
                end
                preprocessSingleArg(dependValue)
            end
        end
    end


-- Get arguments without a number suffix, and check for bad input.
    -- Get arguments with number suffixes.
for i,v in ipairs(prefixTable) do
    local a = 1 -- Counter variable.
if type(v) ~= 'table' or type(v.prefix) ~= "string" or (v.depend and type(v.depend) ~= 'table') then
    local moreArgumentsExist = true
error('Invalid input detected to preprocessArgs prefix table', 2)
    while moreArgumentsExist == true do
end
        moreArgumentsExist = false
preprocessSingleArg(v.prefix)
        for i = a, a + step - 1 do
-- Only parse the depend parameter if the prefix parameter is present and not blank.
            for j,v in ipairs(prefixTable) do
if args[v.prefix] and v.depend then
                local prefixArgName = v.prefix .. tostring(i)
for j, dependValue in ipairs(v.depend) do
                if origArgs[prefixArgName] then
if type(dependValue) ~= 'string' then
                    moreArgumentsExist = true -- Do another loop if any arguments are found, even blank ones.
error('Invalid "depend" parameter value detected in preprocessArgs')
                    preprocessSingleArg(prefixArgName)
end
                end
preprocessSingleArg(dependValue)
                -- Process the depend table if the prefix argument is present and not blank, or
end
                -- we are processing "prefix1" and "prefix" is present and not blank, and
end
                -- if the depend table is present.
end
                if v.depend and (args[prefixArgName] or (i == 1 and args[v.prefix])) then
 
                    for j,dependValue in ipairs(v.depend) do
-- Get arguments with number suffixes.
                        local dependArgName = dependValue .. tostring(i)
local a = 1 -- Counter variable.
                        preprocessSingleArg(dependArgName)
local moreArgumentsExist = true
                    end
while moreArgumentsExist == true do
                end
moreArgumentsExist = false
            end
for i = a, a + step - 1 do
        end
for j,v in ipairs(prefixTable) do
        a = a + step
local prefixArgName = v.prefix .. tostring(i)
    end
if origArgs[prefixArgName] then
moreArgumentsExist = true -- Do another loop if any arguments are found, even blank ones.
preprocessSingleArg(prefixArgName)
end
-- Process the depend table if the prefix argument is present and not blank, or
-- we are processing "prefix1" and "prefix" is present and not blank, and
-- if the depend table is present.
if v.depend and (args[prefixArgName] or (i == 1 and args[v.prefix])) then
for j,dependValue in ipairs(v.depend) do
local dependArgName = dependValue .. tostring(i)
preprocessSingleArg(dependArgName)
end
end
end
end
a = a + step
end
end
end
 
function p.infobox(frame)
function p.infobox(frame)
-- If called via #invoke, use the args passed into the invoking template.
    -- If called via #invoke, use the args passed into the invoking template.
-- Otherwise, for testing purposes, assume args are being passed directly in.
    -- Otherwise, for testing purposes, assume args are being passed directly in.
if frame == mw.getCurrentFrame() then
    if frame == mw.getCurrentFrame() then
origArgs = frame:getParent().args
        origArgs = frame:getParent().args
else
    else
origArgs = frame
        origArgs = frame
end
    end
 
   
-- Parse the data parameters in the same order that the old {{infobox}} did, so that
    -- Parse the data parameters in the same order that the old {{infobox}} did, so that
-- references etc. will display in the expected places. Parameters that depend on
    -- references etc. will display in the expected places. Parameters that depend on
-- another parameter are only processed if that parameter is present, to avoid
    -- another parameter are only processed if that parameter is present, to avoid
-- phantom references appearing in article reference lists.
    -- phantom references appearing in article reference lists.
preprocessSingleArg('child')
    preprocessSingleArg('child')
preprocessSingleArg('bodyclass')
    preprocessSingleArg('bodyclass')
preprocessSingleArg('subbox')
    preprocessSingleArg('subbox')
preprocessSingleArg('bodystyle')
    preprocessSingleArg('bodystyle')
preprocessSingleArg('title')
    preprocessSingleArg('title')
preprocessSingleArg('titleclass')
    preprocessSingleArg('titleclass')
preprocessSingleArg('titlestyle')
    preprocessSingleArg('titlestyle')
preprocessSingleArg('above')
    preprocessSingleArg('above')
preprocessSingleArg('aboveclass')
    preprocessSingleArg('aboveclass')
preprocessSingleArg('abovestyle')
    preprocessSingleArg('abovestyle')
preprocessArgs({
    preprocessArgs({
{prefix = 'subheader', depend = {'subheaderstyle', 'subheaderrowclass'}}
        {prefix = 'subheader', depend = {'subheaderstyle', 'subheaderrowclass'}}
}, 10)
    }, 10)
preprocessSingleArg('subheaderstyle')
    preprocessSingleArg('subheaderstyle')
preprocessSingleArg('subheaderclass')
    preprocessSingleArg('subheaderclass')
preprocessArgs({
    preprocessArgs({
{prefix = 'image', depend = {'caption', 'imagerowclass'}}
        {prefix = 'image', depend = {'caption', 'imagerowclass'}}
}, 10)
    }, 10)
preprocessSingleArg('captionstyle')
    preprocessSingleArg('captionstyle')
preprocessSingleArg('imagestyle')
    preprocessSingleArg('imagestyle')
preprocessSingleArg('imageclass')
    preprocessSingleArg('imageclass')
preprocessArgs({
    preprocessArgs({
{prefix = 'header'},
        {prefix = 'header'},
{prefix = 'data', depend = {'label'}},
        {prefix = 'data', depend = {'label'}},
{prefix = 'rowclass'},
        {prefix = 'rowclass'},
{prefix = 'rowstyle'},
        {prefix = 'rowstyle'},
{prefix = 'rowcellstyle'},
        {prefix = 'rowcellstyle'},
{prefix = 'class'},
        {prefix = 'class'},
{prefix = 'dataid'},
        {prefix = 'dataid'},
{prefix = 'labelid'},
        {prefix = 'labelid'},
{prefix = 'headerid'},
        {prefix = 'headerid'},
{prefix = 'rowid'}
        {prefix = 'rowid'}
}, 50)
    }, 50)
preprocessSingleArg('headerclass')
    preprocessSingleArg('headerclass')
preprocessSingleArg('headerstyle')
    preprocessSingleArg('headerstyle')
preprocessSingleArg('labelstyle')
    preprocessSingleArg('labelstyle')
preprocessSingleArg('datastyle')
    preprocessSingleArg('datastyle')
preprocessSingleArg('below')
    preprocessSingleArg('below')
preprocessSingleArg('belowclass')
    preprocessSingleArg('belowclass')
preprocessSingleArg('belowstyle')
    preprocessSingleArg('belowstyle')
preprocessSingleArg('name')
    preprocessSingleArg('name')
args['italic title'] = origArgs['italic title'] -- different behaviour if blank or absent
    args['italic title'] = origArgs['italic title'] -- different behaviour if blank or absent
preprocessSingleArg('decat')
    preprocessSingleArg('decat')
 
return _infobox()
    return _infobox()
end
end
 
return p
return p

Revision as of 19:49, 13 May 2020

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

--
-- This module implements {{Infobox}}
--

local p = {}

local navbar = require('Module:Navbar')._navbar

local args = {}
local origArgs
local root

local function notempty( s ) return s and s:match( '%S' ) end

local function fixChildBoxes(sval, tt)
	if notempty(sval) then
		local marker = '<span class=special_infobox_marker>'
		local s = sval
		s = mw.ustring.gsub(s, '(<%s*[Tt][Rr])', marker .. '%1')
		s = mw.ustring.gsub(s, '(</[Tt][Rr]%s*>)', '%1' .. marker)
		if s:match(marker) then
			s = mw.ustring.gsub(s, marker .. '%s*' .. marker, '')
			s = mw.ustring.gsub(s, '([\r\n]|-[^\r\n]*[\r\n])%s*' .. marker, '%1')
			s = mw.ustring.gsub(s, marker .. '%s*([\r\n]|-)', '%1')
			s = mw.ustring.gsub(s, '(</[Cc][Aa][Pp][Tt][Ii][Oo][Nn]%s*>%s*)' .. marker, '%1')
			s = mw.ustring.gsub(s, '(<%s*[Tt][Aa][Bb][Ll][Ee][^<>]*>%s*)' .. marker, '%1')
			s = mw.ustring.gsub(s, '^(%{|[^\r\n]*[\r\n]%s*)' .. marker, '%1')
			s = mw.ustring.gsub(s, '([\r\n]%{|[^\r\n]*[\r\n]%s*)' .. marker, '%1')
			s = mw.ustring.gsub(s,  marker .. '(%s*</[Tt][Aa][Bb][Ll][Ee]%s*>)', '%1')
			s = mw.ustring.gsub(s,  marker .. '(%s*\n|%})', '%1')
		end
		if s:match(marker) then
			local subcells = mw.text.split(s, marker)
			s = ''
			for k = 1, #subcells do
				if k == 1 then
					s = s .. subcells[k] .. '</' .. tt .. '></tr>'
				elseif k == #subcells then
					local rowstyle = ' style="display:none"'
					if notempty(subcells[k]) then rowstyle = ''	end
					s = s .. '<tr' .. rowstyle ..'><' .. tt .. ' colspan=2>\n' .. subcells[k]
				elseif notempty(subcells[k]) then
					if (k % 2) == 0 then
						s = s .. subcells[k]
					else
						s = s .. '<tr><' .. tt .. ' colspan=2>\n' .. subcells[k] .. '</' .. tt .. '></tr>'
					end
				end
			end
		end
		-- the next two lines add a newline at the end of lists for the PHP parser
		-- https://en.wikipedia.org/w/index.php?title=Template_talk:Infobox_musical_artist&oldid=849054481
		-- remove when [[:phab:T191516]] is fixed or OBE
		s = mw.ustring.gsub(s, '([\r\n][%*#;:][^\r\n]*)$', '%1\n')
		s = mw.ustring.gsub(s, '^([%*#;:][^\r\n]*)$', '%1\n')
		s = mw.ustring.gsub(s, '^([%*#;:])', '\n%1')
		s = mw.ustring.gsub(s, '^(%{%|)', '\n%1')
		return s
	else
		return sval
	end
end

local function union(t1, t2)
    -- Returns the union of the values of two tables, as a sequence.
    local vals = {}
    for k, v in pairs(t1) do
        vals[v] = true
    end
    for k, v in pairs(t2) do
        vals[v] = true
    end
    local ret = {}
    for k, v in pairs(vals) do
        table.insert(ret, k)
    end
    return ret
end

local function getArgNums(prefix)
    -- Returns a table containing the numbers of the arguments that exist
    -- for the specified prefix. For example, if the prefix was 'data', and
    -- 'data1', 'data2', and 'data5' exist, it would return {1, 2, 5}.
    local nums = {}
    for k, v in pairs(args) do
        local num = tostring(k):match('^' .. prefix .. '([1-9]%d*)$')
        if num then table.insert(nums, tonumber(num)) end
    end
    table.sort(nums)
    return nums
end

local function addRow(rowArgs)
    -- Adds a row to the infobox, with either a header cell
    -- or a label/data cell combination.
    if rowArgs.header then
        root
            :tag('tr')
                :addClass(rowArgs.rowclass)
                :cssText(rowArgs.rowstyle)
                :attr('id', rowArgs.rowid)
                :tag('th')
                    :attr('colspan', 2)
                    :attr('id', rowArgs.headerid)
                    :addClass(rowArgs.class)
                    :addClass(args.headerclass)
                    :css('text-align', 'center')
                    :cssText(args.headerstyle)
                    :cssText(rowArgs.rowcellstyle)
                    :wikitext(fixChildBoxes(rowArgs.header, 'th'))
        if rowArgs.data then
        	root:wikitext('[[Category:Pages which use infobox templates with ignored data cells]]')
       	end
    elseif rowArgs.data then
        local row = root:tag('tr')
        row:addClass(rowArgs.rowclass)
        row:cssText(rowArgs.rowstyle)
        row:attr('id', rowArgs.rowid)
        if rowArgs.label then
            row
                :tag('th')
                    :attr('scope', 'row')
                    :attr('id', rowArgs.labelid)
                    :cssText(args.labelstyle)
                    :cssText(rowArgs.rowcellstyle)
                    :wikitext(rowArgs.label)
                    :done()
        end
        
        local dataCell = row:tag('td')
        if not rowArgs.label then 
            dataCell
                :attr('colspan', 2)
                :css('text-align', 'center') 
        end
        dataCell
            :attr('id', rowArgs.dataid)
            :addClass(rowArgs.class)
            :cssText(rowArgs.datastyle)
            :cssText(rowArgs.rowcellstyle)
            :wikitext(fixChildBoxes(rowArgs.data, 'td'))
    end
end

local function renderTitle()
    if not args.title then return end

    root
        :tag('caption')
            :addClass(args.titleclass)
            :cssText(args.titlestyle)
            :wikitext(args.title)
end

local function renderAboveRow()
    if not args.above then return end
    
    root
        :tag('tr')
            :tag('th')
                :attr('colspan', 2)
                :addClass(args.aboveclass)
                :css('text-align', 'center')
                :css('font-size', '125%')
                :css('font-weight', 'bold')
                :cssText(args.abovestyle)
                :wikitext(fixChildBoxes(args.above,'th'))
end

local function renderBelowRow()
    if not args.below then return end
    
    root
        :tag('tr')
            :tag('td')
                :attr('colspan', '2')
                :addClass(args.belowclass)
                :css('text-align', 'center')
                :cssText(args.belowstyle)
                :wikitext(fixChildBoxes(args.below,'td'))
end

local function renderSubheaders()
    if args.subheader then
        args.subheader1 = args.subheader
    end
    if args.subheaderrowclass then
        args.subheaderrowclass1 = args.subheaderrowclass
    end
    local subheadernums = getArgNums('subheader')
    for k, num in ipairs(subheadernums) do
        addRow({
            data = args['subheader' .. tostring(num)],
            datastyle = args.subheaderstyle,
            rowcellstyle = args['subheaderstyle' .. tostring(num)],
            class = args.subheaderclass,
            rowclass = args['subheaderrowclass' .. tostring(num)]
        })
    end
end

local function renderImages()
    if args.image then
        args.image1 = args.image
    end
    if args.caption then
        args.caption1 = args.caption
    end
    local imagenums = getArgNums('image')
    for k, num in ipairs(imagenums) do
        local caption = args['caption' .. tostring(num)]
        local data = mw.html.create():wikitext(args['image' .. tostring(num)])
        if caption then
            data
                :tag('div')
                    :cssText(args.captionstyle)
                    :wikitext(caption)
        end
        addRow({
            data = tostring(data),
            datastyle = args.imagestyle,
            class = args.imageclass,
            rowclass = args['imagerowclass' .. tostring(num)]
        })
    end
end

local function renderRows()
    -- Gets the union of the header and data argument numbers,
    -- and renders them all in order using addRow.
    local rownums = union(getArgNums('header'), getArgNums('data'))
    table.sort(rownums)
    for k, num in ipairs(rownums) do
        addRow({
            header = args['header' .. tostring(num)],
            label = args['label' .. tostring(num)],
            data = args['data' .. tostring(num)],
            datastyle = args.datastyle,
            class = args['class' .. tostring(num)],
            rowclass = args['rowclass' .. tostring(num)],
            rowstyle = args['rowstyle' .. tostring(num)],
            rowcellstyle = args['rowcellstyle' .. tostring(num)],
            dataid = args['dataid' .. tostring(num)],
            labelid = args['labelid' .. tostring(num)],
            headerid = args['headerid' .. tostring(num)],
            rowid = args['rowid' .. tostring(num)]
        })
    end
end

local function renderNavBar()
    if not args.name then return end
    
    root
        :tag('tr')
            :tag('td')
                :attr('colspan', '2')
                :css('text-align', 'right')
                :wikitext(navbar{
                    args.name,
                    mini = 1,
                })
end

local function renderItalicTitle()
    local italicTitle = args['italic title'] and mw.ustring.lower(args['italic title'])
    if italicTitle == '' or italicTitle == 'force' or italicTitle == 'yes' then
        root:wikitext(mw.getCurrentFrame():expandTemplate({title = 'italic title'}))
    end
end

local function renderTrackingCategories()
    if args.decat ~= 'yes' then
    	if args.child == 'yes' then
        	if args.title then
            	root:wikitext('[[Category:Pages which use embedded infobox templates with the title parameter]]')
        	end
        elseif #(getArgNums('data')) == 0 and mw.title.getCurrentTitle().namespace == 0 then
            root:wikitext('[[Category:Articles which use infobox templates with no data rows]]')
        end
    end
end

local function _infobox()
    -- Specify the overall layout of the infobox, with special settings
    -- if the infobox is used as a 'child' inside another infobox.
    if args.child ~= 'yes' then
        root = mw.html.create('table')
        
        root
            :addClass((args.subbox ~= 'yes') and 'infobox' or nil)
            :addClass(args.bodyclass)
            
            if args.subbox == 'yes' then
                root
                    :css('padding', '0')
                    :css('border', 'none')
                    :css('margin', '-3px')
                    :css('width', 'auto')
                    :css('min-width', '100%')
                    :css('font-size', '100%')
                    :css('clear', 'none')
                    :css('float', 'none')
                    :css('background-color', 'transparent')
            else
                root
                    :css('width', '22em')
            end
        root
            :cssText(args.bodystyle)
    
        renderTitle()
        renderAboveRow()
    else
        root = mw.html.create()
        
        root
            :wikitext(args.title)
    end

    renderSubheaders()
    renderImages() 
    renderRows() 
    renderBelowRow()  
    renderNavBar()
    renderItalicTitle()
    renderTrackingCategories()
    
    return tostring(root)
end

local function preprocessSingleArg(argName)
    -- If the argument exists and isn't blank, add it to the argument table.
    -- Blank arguments are treated as nil to match the behaviour of ParserFunctions.
    if origArgs[argName] and origArgs[argName] ~= '' then
        args[argName] = origArgs[argName]
    end
end

local function preprocessArgs(prefixTable, step)
    -- Assign the parameters with the given prefixes to the args table, in order, in batches
    -- of the step size specified. This is to prevent references etc. from appearing in the
    -- wrong order. The prefixTable should be an array containing tables, each of which has
    -- two possible fields, a "prefix" string and a "depend" table. The function always parses
    -- parameters containing the "prefix" string, but only parses parameters in the "depend"
    -- table if the prefix parameter is present and non-blank.
    if type(prefixTable) ~= 'table' then
        error("Non-table value detected for the prefix table", 2)
    end
    if type(step) ~= 'number' then
        error("Invalid step value detected", 2)
    end
    
    -- Get arguments without a number suffix, and check for bad input.
    for i,v in ipairs(prefixTable) do
        if type(v) ~= 'table' or type(v.prefix) ~= "string" or (v.depend and type(v.depend) ~= 'table') then
            error('Invalid input detected to preprocessArgs prefix table', 2)
        end
        preprocessSingleArg(v.prefix)
        -- Only parse the depend parameter if the prefix parameter is present and not blank.
        if args[v.prefix] and v.depend then
            for j, dependValue in ipairs(v.depend) do
                if type(dependValue) ~= 'string' then
                    error('Invalid "depend" parameter value detected in preprocessArgs')
                end
                preprocessSingleArg(dependValue)
            end
        end
    end

    -- Get arguments with number suffixes.
    local a = 1 -- Counter variable.
    local moreArgumentsExist = true
    while moreArgumentsExist == true do
        moreArgumentsExist = false
        for i = a, a + step - 1 do
            for j,v in ipairs(prefixTable) do
                local prefixArgName = v.prefix .. tostring(i)
                if origArgs[prefixArgName] then
                    moreArgumentsExist = true -- Do another loop if any arguments are found, even blank ones.
                    preprocessSingleArg(prefixArgName)
                end
                -- Process the depend table if the prefix argument is present and not blank, or
                -- we are processing "prefix1" and "prefix" is present and not blank, and
                -- if the depend table is present.
                if v.depend and (args[prefixArgName] or (i == 1 and args[v.prefix])) then
                    for j,dependValue in ipairs(v.depend) do
                        local dependArgName = dependValue .. tostring(i)
                        preprocessSingleArg(dependArgName)
                    end
                end
            end
        end
        a = a + step
    end
end
 
function p.infobox(frame)
    -- If called via #invoke, use the args passed into the invoking template.
    -- Otherwise, for testing purposes, assume args are being passed directly in.
    if frame == mw.getCurrentFrame() then
        origArgs = frame:getParent().args
    else
        origArgs = frame
    end
    
    -- Parse the data parameters in the same order that the old {{infobox}} did, so that
    -- references etc. will display in the expected places. Parameters that depend on
    -- another parameter are only processed if that parameter is present, to avoid
    -- phantom references appearing in article reference lists.
    preprocessSingleArg('child')
    preprocessSingleArg('bodyclass')
    preprocessSingleArg('subbox')
    preprocessSingleArg('bodystyle')
    preprocessSingleArg('title')
    preprocessSingleArg('titleclass')
    preprocessSingleArg('titlestyle')
    preprocessSingleArg('above')
    preprocessSingleArg('aboveclass')
    preprocessSingleArg('abovestyle')
    preprocessArgs({
        {prefix = 'subheader', depend = {'subheaderstyle', 'subheaderrowclass'}}
    }, 10)
    preprocessSingleArg('subheaderstyle')
    preprocessSingleArg('subheaderclass')
    preprocessArgs({
        {prefix = 'image', depend = {'caption', 'imagerowclass'}}
    }, 10)
    preprocessSingleArg('captionstyle')
    preprocessSingleArg('imagestyle')
    preprocessSingleArg('imageclass')
    preprocessArgs({
        {prefix = 'header'},
        {prefix = 'data', depend = {'label'}},
        {prefix = 'rowclass'},
        {prefix = 'rowstyle'},
        {prefix = 'rowcellstyle'},
        {prefix = 'class'},
        {prefix = 'dataid'},
        {prefix = 'labelid'},
        {prefix = 'headerid'},
        {prefix = 'rowid'}
    }, 50)
    preprocessSingleArg('headerclass')
    preprocessSingleArg('headerstyle')
    preprocessSingleArg('labelstyle')
    preprocessSingleArg('datastyle')
    preprocessSingleArg('below')
    preprocessSingleArg('belowclass')
    preprocessSingleArg('belowstyle')
    preprocessSingleArg('name')
    args['italic title'] = origArgs['italic title'] -- different behaviour if blank or absent
    preprocessSingleArg('decat')
 
    return _infobox()
end
 
return p