Module:Infobox

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

local p = {}

function p.infobox(f)
	local args = require('Module:Arguments').getArgs(f)
	local titleObject = mw.title.getCurrentTitle()
	local title = args.title or titleObject.baseText
	local template = f:getParent():getTitle():lower():gsub('template:', '')
	local headerArea = ''
	
	local json = {
		images = {},
		rows = {}
	}
	
	local imageArea = args.imagearea
	if not imageArea and imageArea ~= 'none' then
		local images = {}
		local defaultImageSize = args.defaultimagesize or '150px'
		local defaultImageClass = args.defaultimageclass
		args.image1 = args.image1 or args.image or 'title'
		args.image1size = args.image1size or args.imagesize
		args.image1class = args.image1class or args.imageclass
		args.group1 = args.group1 or args.group
		args.group1size = args.group1size or args.groupsize
		args.group1class = args.group1class or args.groupclass
		
		local imgCount = {}
		local invImgCount = {}
		local groupCount = {}
		local groupImgList = {}
		for k, v in pairs(args) do
			if type(k) == 'string' then
				local image, num = k:match('^(image)(%d+)$')
				local invImage, invNum = k:match('^(invimage)(%d+)$')
				local group, groupNum = k:match('^(group)(%d+)$')
				local groupImg, groupImgNum = k:match('^(%d+)-(%d+)$')
				if v:lower() ~= 'none' then
					if image then
						table.insert(imgCount, tonumber(num))
					elseif invImage then
						table.insert(invImgCount, tonumber(invNum))
					elseif group then
						table.insert(groupCount, tonumber(groupNum))
						if not groupImgList['group' .. groupNum] then
							groupImgList['group' .. groupNum] = {}
						end
					elseif groupImg then
						if not groupImgList['group' .. groupImg] then
							groupImgList['group' .. groupImg] = {}
						end
						table.insert(groupImgList['group' .. groupImg], tonumber(groupImgNum))
					end
				end
			end
		end
		
		local animate
		if #groupCount > 0 then
			table.sort(groupCount)
			local tabber = {}
			for _, v in ipairs(groupCount) do
				local group = args['group' .. v]
				local groupSize = args['group' .. v .. 'size'] or defaultImageSize
				local groupClass = args['group' .. v .. 'class'] or defaultImageClass
				local groupImages = {}
				
				table.sort(groupImgList['group' .. v])
				for _, w in ipairs(groupImgList['group' .. v]) do
					local image = args[v .. '-' .. w]
					local size = args[v .. '-' .. w .. 'size'] or groupSize
					local class = args[v .. '-' .. w .. 'class'] or groupClass
					
					if string.match(image, 'UNIQ%-%-gallery%-') then
						image = image
					elseif image:match(';') then
						if not animate then
							animate = require('Module:Animate').animate
						end
						image = animate{image, size, class = class}
					else
						json.images[#json.images + 1] = image
						local altText = image .. ': Infobox image for ' .. title .. ' the ' .. template .. '.'
						image = '[[File:' .. image .. '|' .. size .. '|class=' .. (class or '') .. '|alt=' .. altText .. ']]'
					end
					
					table.insert(groupImages, '<div>' .. image .. '</div>')
				end
				
				table.insert(tabber, '|-|' .. group .. '=\n' .. table.concat(groupImages, '\n'))
			end
			table.insert(images, '<div>' .. f:extensionTag('tabber', table.concat(tabber, '\n')) .. '</div>')
		end
		
		table.sort(imgCount)
		for _, v in ipairs(imgCount) do
			local image = args['image' .. v]
			local size = args['image' .. v .. 'size'] or defaultImageSize
			local class = args['image' .. v .. 'class'] or defaultImageClass
			
			if image == 'title' then
				local imageTitle = mw.title.new('Media:' .. title .. '.png')
				if #groupCount == 0 and imageTitle and imageTitle.exists then
					json.images[#json.images + 1] = title .. '.png'
					local altText = title .. '.png: Infobox image for ' .. title .. ' the ' .. template .. '.'
					image = '[[File:' .. title .. '.png|' .. size .. '|class=' .. (class or '') .. '|alt=' .. altText .. ']]'
				else
					image = ''
				end
			elseif string.match(image, 'UNIQ%-%-gallery%-') then
				image = image
			else
				json.images[#json.images + 1] = image
				local altText = image .. ': Infobox image for ' .. title .. ' the ' .. template .. '.'
				image = '[[File:' .. image .. '|' .. size .. '|class=' .. (class or '') .. '|alt=' .. altText .. ']]'
			end
			
			table.insert(images, '<div>' .. image .. '</div>')
		end
		
		images = table.concat(images, '\n')
		
		if images ~= '' then
			imageArea = images
		else
			imageArea = 'none'
		end
	end
	
	if imageArea and imageArea ~= 'none' then
		imageArea = '<div class="infobox-imagearea animated-container">' .. imageArea .. '</div>'
	else
		local groupArea = args.grouparea
		if groupArea then
			imageArea = groupArea
		else
			imageArea = ''
		end
	end
	
	local extraText = args.extratext
	
	if extraText and extraText ~= 'none' then
		json.extratext = extraText
		extraText = '<div class="infobox-extratext">' .. extraText .. '</div>'
	else
		extraText = ''
	end
	
	headerArea = imageArea .. extraText
	local footer = args.footer
	if footer then
		json.footer = footer
		footer = '| class="infobox-footer" colspan="2" | ' .. footer
	end
	
	json.title = title
	
	local html = {
		'<div class="infobox">',
			'<div class="infobox-title">' .. title .. '</div>',
			headerArea,
			'{| class="infobox-rows" cellspacing="1" cellpadding="4"',
			'|-',
			args.rows or '',
			footer or '',
			'|}',
		'</div>'
	}
	
	return table.concat(html, '\n')
end

return p