Module:Infobox

-- -- This module implements --

local p = {}

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(args, 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(root, args, 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) :wikitext(rowArgs.header) 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) :css('text-align', 'left') :cssText(args.labelstyle) :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) :newline :wikitext(rowArgs.data) end end

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

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

local function renderAboveRow(root, args) 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(args.above) end

local function renderBelowRow(root, args) 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) :newline :wikitext(args.below) end

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

local function renderImages(root, args) if args.image then args.image1 = args.image end if args.caption then args.caption1 = args.caption end local imagenums = getArgNums(args, '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(root, args, {           data = tostring(data),            datastyle = args.imagestyle,            class = args.imageclass,            rowclass = args['imagerowclass' .. tostring(num)]        }) end end

local function renderRows(root, args) -- Gets the union of the header and data argument numbers, -- and renders them all in order using addRow. local rownums = union(getArgNums(args, 'header'), getArgNums(args, 'data')) table.sort(rownums) for k, num in ipairs(rownums) do       addRow(root, args, {            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)],            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(root, args) if not args.name then return end

root :tag('tr') :tag('td') :attr('colspan', '2') :css('text-align', 'right') :wikitext(require('Module:Navbar')._navbar{                   args.name,                    mini = 1,                }) end

local function renderItalicTitle(root, args) 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(root, args) if args.decat ~= 'yes' then if #(getArgNums(args, 'data')) == 0 and mw.title.getCurrentTitle.namespace == 0 then root:wikitext('') end if args.child == 'yes' and args.title then root:wikitext('') end end end

function p.infobox(frame) local args = require('Module:Arguments').getArgs(frame, { wrappers = 'Template:Infobox', valueFunc = function(k, v)   	if v ~= '' or k == 'italic title' then    		return v    	end    end })

-- Specify the overall layout of the infobox, with special settings -- if the infobox is used as a 'child' inside another infobox. local root if args.child ~= 'yes' then root = mw.html.create('table')

root :addClass('infobox') :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(root, args) renderAboveRow(root, args) else root = mw.html.create

root :wikitext(args.title) end

renderSubheaders(root, args) renderImages(root, args) renderRows(root, args) renderBelowRow(root, args) renderNavBar(root, args) renderItalicTitle(root, args) renderTrackingCategories(root, args)

return tostring(root) end

return p