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 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' and type(v) == '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 = {}

				if groupImgList['group' .. v] then
					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, image)
					end
				end

				table.insert(
					tabber,
					'|-|' .. group .. '=' .. table.concat(groupImages, '')
				)
			end

			table.insert(
				images,
				f:extensionTag('tabber', table.concat(tabber, ''))
			)
		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

			if image ~= '' then
				table.insert(images, image)
			end
		end

		images = table.concat(images, '')

		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

	local footer = args.footer

	if footer then
		json.footer = footer
		footer = '|-\n| class="infobox-footer" colspan="2" | ' .. footer
	else
		footer = ''
	end

	json.title = title

	local rows = mw.getCurrentFrame():preprocess(args.rows or '')

	local html = {
		'<div class="infobox">',
		'<div class="infobox-title">' .. title .. '</div>',
		imageArea .. extraText,
		'{| class="infobox-rows" cellspacing="1" cellpadding="4"',
		rows,
		footer,
		'|}',
		'</div>'
	}

	return table.concat(html, '')
end

return p