---
-- This module implements
---
-- The initial version was created by modifying Module:Infobox
---  
local p = {}  
local navbar = require('Module:Navbar')._navbar  
local args = {}
local origArgs
local root  
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,suffix)
-- 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 .. '([0-9]%d*)' .. suffix .. '$')
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', 4)
:attr('id', rowArgs.headerid)
:addClass(rowArgs.class)
:addClass(args.headerclass)
:css('text-align', '')
:cssText(args.headerstyle)
:cssText(rowArgs.rowcellstyle)
:wikitext(rowArgs.header)
elseif rowArgs.label then
if rowArgs.data then
local row = root:tag('tr')
row:addClass(rowArgs.rowclass)
row:cssText(rowArgs.rowstyle)
row:attr('id', rowArgs.rowid)
row
:tag('th')
:attr('scope', 'row')
:attr('id', rowArgs.labelid)
:cssText(args.labelstyle)
:cssText(rowArgs.rowcellstyle)
:wikitext(rowArgs.label)
:done  
local dataCell = row:tag('td')
dataCell
:attr('colspan', 3)
:attr('id', rowArgs.dataid)
:addClass(rowArgs.class)
:cssText(rowArgs.datastyle)
:cssText(rowArgs.rowcellstyle)
:newline
:wikitext(rowArgs.data)
elseif rowArgs.dataa or rowArgs.datab then
local row = root:tag('tr')
row:addClass(rowArgs.rowclass)
row:cssText(rowArgs.rowstyle)
row:attr('id', rowArgs.rowid)
row
:tag('th')
:attr('scope', 'row')
:attr('id', rowArgs.labelid)
:cssText(args.labelstyle)
:cssText(rowArgs.rowcellstyle)
:wikitext(rowArgs.label)
:done  
local dataCella = row:tag('td')
dataCella
:attr('id', rowArgs.dataaid)
:addClass(rowArgs.classa)
:cssText(rowArgs.dataastyle)
:cssText(rowArgs.rowcellstyle)
:newline
:wikitext(rowArgs.dataa)
if rowArgs.renderb then
local dataCellb = row:tag('td')
dataCellb
:attr('id', rowArgs.databid)
:addClass(rowArgs.classb)
:cssText(rowArgs.databstyle)
:cssText(rowArgs.rowcellstyle)
:newline
:wikitext(rowArgs.datab)
end
if rowArgs.renderc then
local dataCellc = row:tag('td')
dataCellc
:attr('id', rowArgs.datacid)
:addClass(rowArgs.classc)
:cssText(rowArgs.datacstyle)
:cssText(rowArgs.rowcellstyle)
:newline
:wikitext(rowArgs.datac)
end
end
elseif rowArgs.data then
local row = root:tag('tr')
row:addClass(rowArgs.rowclass)
row:cssText(rowArgs.rowstyle)
row:attr('id', rowArgs.rowid)  
local dataCell = row:tag('td')
dataCell
:attr('colspan', 4)
:attr('id', rowArgs.dataid)
:addClass(rowArgs.class)
:css('text-align', '')
:cssText(rowArgs.datastyle)
:cssText(rowArgs.rowcellstyle)
:newline
:wikitext(rowArgs.data)
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', 4)
:addClass(args.aboveclass)
:css('text-align', '')
:css('font-size', '125%')
:css('font-weight', 'bold')
:cssText(args.abovestyle)
:wikitext(args.above)
end  
local function renderBelowRow
if not args.below then return end  
root
:tag('tr')
:tag('td')
:attr('colspan', 4)
:addClass(args.belowclass)
:css('text-align', '')
:cssText(args.belowstyle)
:newline
:wikitext(args.below)
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 or 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','[ab]?'))
local datab_count = #(getArgNums('data','b'))
local datac_count = #(getArgNums('data','c'))  
table.sort(rownums)
for k, num in ipairs(rownums) do
addRow({
renderb = datab_count > 0,
renderc = datac_count > 0,
header = args['header' .. tostring(num)],
label = args['label' .. tostring(num)],
data = args['data' .. tostring(num)],
datastyle = args.datastyle,
class = args['class' .. tostring(num)],
dataa = args['data' .. tostring(num) .. 'a'],
dataastyle = args.datastylea,
classa = args['class' .. tostring(num) .. 'a'],
datab = args['data' .. tostring(num) .. 'b'],
databstyle = args.datastyleb,
classb = args['class' .. tostring(num) .. 'b'],
datac = args['data' .. tostring(num) .. 'c'],
datacstyle = args.datastylec,
classc = args['class' .. tostring(num) .. 'c'],
rowclass = args['rowclass' .. tostring(num)],
rowstyle = args['rowstyle' .. tostring(num)],
rowcellstyle = args['rowcellstyle' .. tostring(num)],
dataid = args['dataid' .. tostring(num)],
dataaid = args['dataid' .. tostring(num) .. 'a'],
dataaib = args['dataid' .. tostring(num) .. 'b'],
dataaic = args['dataid' .. tostring(num) .. 'c'],
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', '4')
:css('text-align', '')
: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 #(getArgNums('data','[abc]?')) == 0 and mw.title.getCurrentTitle.namespace == 0 then
root:wikitext('')
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.  
root = mw.html.create('table')  
root
:addClass('infobox')
:addClass(args.bodyclass)
if args.child == 'yes' or args.subbox == 'yes' then
root
:css('padding', '0')
:css(*, *)
:css('margin', (args.subbox == 'yes') and '-' or 'auto')
:css('width', 'auto')
:css('min-width', '100%')
:css('font-size', '100%')
:css('clear', '')
:css('float', '')
:css('background-color', 'transparent')
else
root
:css('width', '22em')
end  
root:cssText(args.bodystyle)  
renderTitle
renderAboveRow
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.