Modul:Infoboks/sandkasse

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

local p = {}

local navbar = require('Modul: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.mellomtittel then
		root
			:tag('tr')
				:addClass(rowArgs.radklasse)
				:cssText(rowArgs.radstil)
				:attr('id', rowArgs.radid)
				:tag('th')
					:attr('colspan', 2)
					:attr('id', rowArgs.mellomtittelid)
					:addClass(rowArgs.klasse)
					:addClass(args.mellomtittelklasse)
					:css('text-align', 'center')
					:cssText(args.mellomtittelstil)
					:cssText(rowArgs.radcellestil)
					:wikitext(fixChildBoxes(rowArgs.mellomtittel, 'th'))
		if rowArgs.data then
			root:wikitext('[[Kategory:Sider som bruker innebydge infoboksmaler med ignorert dataceller]]')
		end
	elseif rowArgs.data then
		if not rowArgs.data:gsub('%[%[%s*[Cc][Aa][Tt][Ee][Gg][Oo][Rr][Yy]%s*:[^]]*]]', ''):match('^%S') then
			rowArgs.radstil = 'display:none'
		end
		local row = root:tag('tr')
		row:addClass(rowArgs.radklasse)
		row:cssText(rowArgs.radstil)
		row:attr('id', rowArgs.radid)
		if rowArgs.etikett then
			row
				:tag('th')
					:attr('scope', 'row')
					:attr('id', rowArgs.etikettid)
					:cssText(args.etikettstil)
					:cssText(rowArgs.radcellestil)
					:wikitext(rowArgs.etikett)
					:done()
		end

		local dataCell = row:tag('td')
		if not rowArgs.etikett then
			dataCell
				:attr('colspan', 2)
				:css('text-align', 'center')
		end
		dataCell
			:attr('id', rowArgs.dataid)
			:addClass(rowArgs.klasse)
			:cssText(rowArgs.datastil)
			:cssText(rowArgs.radcellestil)
			:wikitext(fixChildBoxes(rowArgs.data, 'td'))
	end
end

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

	root
		:tag('caption')
			:addClass(args.tittelklasse)
			:cssText(args.tittelstil)
			:wikitext(args.tittel)
end

local function renderAboveRow()
	if not args.over then return end

	root
		:tag('tr')
			:tag('th')
				:attr('colspan', 2)
				:addClass(args.overklasse)
				:css('text-align', 'center')
				:css('font-size', '125%')
				:css('font-weight', 'bold')
				:cssText(args.overstil)
				:wikitext(fixChildBoxes(args.over,'th'))
end

local function renderBelowRow()
	if not args.under then return end

	root
		:tag('tr')
			:tag('td')
				:attr('colspan', '2')
				:addClass(args.underklasse)
				:css('text-align', 'center')
				:cssText(args.understil)
				:wikitext(fixChildBoxes(args.under,'td'))
end

local function renderSubheaders()
	if args.undertittel then
		args.undertittel1 = args.undertittel
	end
	if args.undertittelradklasse then
		args.undertittelradklasse1 = args.undertittelradklasse
	end
	local subheadernums = getArgNums('undertittel')
	for k, num in ipairs(subheadernums) do
		addRow({
			data = args['undertittel' .. tostring(num)],
			datastil = args.undertittelstil,
			radcellestil = args['undertittelstil' .. tostring(num)],
			klasse = args.undertittelklasse,
			radklasse = args['undertittelradklasse' .. tostring(num)]
		})
	end
end

local function renderImages()
	if args.bilde then
		args.bilde1 = args.bilde
	end
	if args.bildetekst then
		args.bildetekst1 = args.bildetekst
	end
	local imagenums = getArgNums('bilde')
	for k, num in ipairs(imagenums) do
		local caption = args['bildetekst' .. tostring(num)]
		local data = mw.html.create():wikitext(args['bilde' .. tostring(num)])
		if caption then
			data
				:tag('div')
					:cssText(args.bildetekststil)
					:wikitext(caption)
		end
		addRow({
			data = tostring(data),
			datastil = args.bildestil,
			klasse = args.bildeklasse,
			radklasse = args['bilderadklasse' .. 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('mellomtittel'), getArgNums('data'))
	table.sort(rownums)
	for k, num in ipairs(rownums) do
		addRow({
			mellomtittel = args['mellomtittel' .. tostring(num)],
			etikett = args['etikett' .. tostring(num)],
			data = args['data' .. tostring(num)],
			datastil = args.datastil,
			klasse = args['klasse' .. tostring(num)],
			radklasse = args['radklasse' .. tostring(num)],
		        radstil = args['radstil' .. tostring(num)],
			radcellestil = args['radcellestil' .. tostring(num)],
			dataid = args['dataid' .. tostring(num)],
			etikettid = args['etikettid' .. tostring(num)],
			mellomtittelid = args['mellomtittelid' .. tostring(num)],
			radid = args['radid' .. tostring(num)]
		})
	end
end

local function renderNavBar()
	if not args.navn then return end

	root
		:tag('tr')
			:tag('td')
				:attr('colspan', '2')
				:css('text-align', 'right')
				:wikitext(navbar{
					args.navn,
					mini = 1,
				})
end

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

local function renderTrackingCategories()
	if args.decat ~= 'yes' then
		if args.barn == 'yes' then
			if args.tittel then
				root:wikitext('[[Kategori:Sider som bruker innebygde infoboksmaler med tittelparameter]]')
			end
		elseif #(getArgNums('data')) == 0 and mw.title.getCurrentTitle().namespace == 0 then
			root:wikitext('[[Kategori:Artikler som bruker infoboksmaler uten datarader]]')
		end
	end
end

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

		root
			:addClass((args.underboks ~= 'yes') and 'infoboks' or nil)
			:addClass(args.bodyklasse)

			if args.underboks == '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.bodystil)

		renderTitle()
		renderAboveRow()
	else
		root = mw.html.create()

		root
			:wikitext(args.tittel)
	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

local function parseDataParameters()
	-- 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('barn')
	preprocessSingleArg('bodyklasse')
	preprocessSingleArg('underboks')
	preprocessSingleArg('bodystil')
	preprocessSingleArg('tittel')
	preprocessSingleArg('tittelklasse')
	preprocessSingleArg('tittelstil')
	preprocessSingleArg('over')
	preprocessSingleArg('overklasse')
	preprocessSingleArg('overstil')
	preprocessArgs({
		{prefix = 'undertittel', depend = {'undertittelstil', 'undertittelradklasse'}}
	}, 10)
	preprocessSingleArg('undertittelstil')
	preprocessSingleArg('undertittelklasse')
	preprocessArgs({
		{prefix = 'bilde', depend = {'bildetekst', 'bilderadklasse'}}
	}, 10)
	preprocessSingleArg('bildetekststil')
	preprocessSingleArg('bildestil')
	preprocessSingleArg('bildeklasse')
	preprocessArgs({
		{prefix = 'mellomtittel'},
		{prefix = 'data', depend = {'etikett'}},
		{prefix = 'radklasse'},
		{prefix = 'radstil'},
		{prefix = 'radcellestil'},
		{prefix = 'klasse'},
		{prefix = 'dataid'},
		{prefix = 'etikettid'},
		{prefix = 'mellomtittelid'},
		{prefix = 'radid'}
	}, 50)
	preprocessSingleArg('mellomtittelklasse')
	preprocessSingleArg('mellomtittelstil')
	preprocessSingleArg('etikettstil')
	preprocessSingleArg('datastil')
	preprocessSingleArg('under')
	preprocessSingleArg('underklasse')
	preprocessSingleArg('understil')
	preprocessSingleArg('navn')
	args['kursiv tittel'] = origArgs['kursiv tittel'] -- different behaviour if blank or absent
	preprocessSingleArg('decat')
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

	parseDataParameters()

	return _infobox()
end

function p.infoboxTemplate(frame)
	-- For calling via #invoke within a template
	origArgs = {}
	for k,v in pairs(frame.args) do origArgs[k] = mw.text.trim(v) end

	parseDataParameters()

	return _infobox()
end

return p