Module:Infobox

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

local counter

local h = {}
local p = {}

function p.arraymap(frame)
	-- a lua implementation of Page Forms' arraymap
	local args = h.overwrite()
	local items = h.split(args[1], args[2] or ',')
	for i, item in ipairs(items) do
		items[i] = args[4]:gsub(args[3], item)
	end
	return table.concat(items, args[6] or ',')
end

function p.preprocess(frame)
    return frame:preprocess(frame.args[1] or frame:getParent().args[1])
end

function p.main(frame)
	h.increment()
	local args = h.overwrite()
	local sep = args.sep or ','
	h.castArgs(args, sep)
	return h.makeInfobox(args, sep)
end

function h.increment()
	counter = mw.getCurrentFrame():callParserFunction('#var', {'DRUID_INFOBOX_ID'}) or 1
	mw.getCurrentFrame():callParserFunction('#vardefine', {'DRUID_INFOBOX_ID', counter})
end

function h.castArgs(args, sep)
  args.image = args.images or args.image
	args.image_labels = h.split(args.image_labels, sep)
end

function h.makeInfobox(args, sep)
	local out = mw.html.create('table')
		:addClass('druid-container')
		:attr('id', 'druid-' .. counter)
	if args.kind then out:addClass('druid-container-' .. h.escape(args.kind)) end
	if args.title then
		out:tag('tr')
			:tag('th')
				:addClass('druid-title')
				:attr('colspan', 2)
				:wikitext(args.title)
	end
	h.printImages(out, h.split(args.image, sep), args)
	for _, section in ipairs(h.split(args.sections, sep)) do
		-- cannot begin tagging here because we don't know if any applicable args are present
		local cols = args[section .. '_columns']
		if cols then
			out:node(h.makeGridSection(
				section, h.split(args[section], sep), tonumber(cols), args
			))
		else
			out:node(h.makeSection(section, h.split(args[section], sep), args))
		end
	end
	return out
end

function h.printImages(out, images, args)
	if #images == 0 then return end
	local labels = args.image_labels
	-- burden is on the user to format this as an image. this should be done in the infobox template,
	-- with something like |image={{#if:{{{image|}}}|[[File:{{{image|}}}{{!}}300px{{!}}link=]]}}
	local td = out:tag('tr')
		:tag('td')
		:attr('colspan', 2)
	if #images == 1 then
		td:addClass('druid-main-image')
			:wikitext(images[1])
		return
	end
	td:addClass('druid-main-images')
	local labelsContainer = td:tag('div')
		:addClass('druid-main-images-labels')
	local imagesContainer = td:tag('div')
		:addClass('druid-main-images-files')
	for i, item in ipairs(images) do
		local labelText = labels[i] or ('[[Category:Infoboxes missing image labels]]Image ' .. i)
		local label = labelsContainer:tag('div')
			:addClass('druid-main-images-label')
			:attr('data-druid', counter .. '-' .. i)
			:wikitext(labelText)
		local container = imagesContainer:tag('div')
			:addClass('druid-main-images-file')
			:attr('data-druid', counter .. '-' .. i)
			:wikitext(item)
		if args[labelText .. '_caption'] then
			container:tag('div')
				:addClass('druid-main-images-caption')
				:wikitext(args[labelText .. '_caption'])
		end
		if i == 1 then
			label:addClass('focused')
			container:addClass('focused')
		end
	end
end

function h.makeGridSection(section, sectionFields, cols, args)
	local shouldPrint = false
	local node = mw.html.create()
	h.printSectionHeader(node, section, args)
	local grid = node:tag('tr'):tag('td')
		:attr('colspan', 2)
		:addClass('druid-grid-section')
		:addClass('druid-grid-section-' .. h.escape(section))
		:tag('div')
			:addClass('druid-grid')
			:css('grid-template-columns', ('repeat(%s, 1fr)'):format(cols))
	local row = 1
	local col = 1
	local itemContainer
	for _, item in ipairs(sectionFields) do
		if args[item] then
			shouldPrint = true
			itemContainer = grid:tag('div')
				:addClass('druid-grid-item')
				:addClass('druid-grid-item-' .. h.escape(item))
				:css('grid-column', col)
				:css('grid-row', row)
			h.printLabel(itemContainer:tag('div'), item, args)
			h.printData(itemContainer:tag('div'), item, args)
			
			if col == cols then
				row = row + 1
				col = 1
			else
				col = col + 1
			end
		end
	end
	if not shouldPrint then return nil end
	if cols ~= 1 then
		-- this will overwrite the existing `grid-column` value for each
		itemContainer:css('grid-column', ('%s / -1'):format(col - 1))
	end
	return node
end

function h.makeSection(section, sectionFields, args)
	local shouldPrint = false
	local node = mw.html.create()
	h.printSectionHeader(node, section, args)
	for _, item in ipairs(sectionFields) do
		if h.shouldPrint(item, args) then
			shouldPrint = true
			local tr = node:tag('tr')
				:addClass('druid-row')
				:addClass('druid-row-' .. h.escape(item))
			if args[item .. '_wide'] then
				local td = h.printData(tr:tag('td'), item, args)
				td
					:attr('colspan', 2)
					:addClass('druid-data-wide')
			else
				h.printLabel(tr:tag('th'), item, args)
				h.printData(tr:tag('td'), item, args)
			end
		end
	end
	if not shouldPrint then return nil end
	return node
end

function h.shouldPrint(item, args)
	if args[item] then return true end
	for _, key in ipairs(args.image_labels) do
		if args[key .. '_' .. item] then
			return true
		end
	end
	return false
end

function h.printLabel(node, item, args)
	return node
		:addClass('druid-label')
		:addClass('druid-label-' .. h.escape(item))
		:wikitext(args[item .. '_display'] or item)
end

function h.printData(node, item, args)
	if not args.image_labels or #args.image_labels == 0 then
		h.printSimpleData(node, item, args)
		return node
	end
	if not h.hasComplexData(item, args) then
		h.printSimpleData(node, item, args)
		return node
	end
	for i, v in ipairs(args.image_labels) do
		local div = node:tag('div')
			:addClass('druid-toggleable-data')
			:attr('data-druid', counter .. '-' .. i)
			:wikitext(args[v .. '_' .. item] or args[item])
		if i == 1 then div:addClass('focused') end
	end
	return node
end

function h.printSimpleData(node, item, args)
	node:addClass('druid-data')
		:addClass('druid-data-' .. h.escape(item))
		:wikitext(args[item])
end

function h.hasComplexData(item, args)
	for _, v in ipairs(args.image_labels) do
		if args[v .. '_' .. item] then return true end
	end
	return false
end

function h.printSectionHeader(node, section, args)
	if args[section .. '_nolabel'] then return end
	node:tag('tr')
		:tag('th')
		:attr('colspan', 2)
		:wikitext(section)
		:addClass('druid-section')
		:addClass('druid-section-' .. h.escape(section))
end

function h.overwrite()
	-- this is a generic utility function that collects args from the invoke call & the parent template.
	-- normally, you merge args with parent template overwriting the invoke call, but
	-- since we'll be putting markup/formatting into our invoke call,
	-- we actually want to overwrite what the user sent.
	local f = mw.getCurrentFrame()
	local origArgs = f.args
	local parentArgs = f:getParent().args

	local args = {}
	
	for k, v in pairs(parentArgs) do
		v = mw.text.trim(v)
		if v ~= '' then
			args[k] = v
		end
	end
	
	for k, v in pairs(origArgs) do
		v = mw.text.trim(tostring(v))
		if v ~= '' then
			args[k] = v
		end
	end
	
	return args
end

-- generic utility functions
-- these would normally be provided by other modules, but to make installation easy
-- I'm including everything here

function h.split(text, pattern, plain)
	if not text then
		return {}
	end
	local ret = {}
	for m in h.gsplit(text, pattern, plain) do
		ret[#ret+1] = m
	end
	return ret
end

function h.gsplit( text, pattern, plain )
	if not pattern then pattern = ',' end
	if not plain then
		pattern = '%s*' .. pattern .. '%s*'
	end
	local s, l = 1, text:len()
	return function ()
		if s then
			local e, n = text:find( pattern, s, plain )
			local ret
			if not e then
				ret = text:sub( s )
				s = nil
			elseif n < e then
				-- Empty separator!
				ret = text:sub( s, e )
				if e < l then
					s = e + 1
				else
					s = nil
				end
			else
				ret = e > s and text:sub( s, e - 1 ) or ''
				s = n + 1
			end
			return ret
		end
	end, nil, nil
end

function h.escape(s)
	s = s:gsub(' ', '')
		:gsub('"', '')
		:gsub("'", '')
		:gsub("%?", '')
		:gsub("%%", '')
		:gsub("%[", '')
		:gsub("%]", '')
		:gsub("{", '')
		:gsub("}", '')
		:gsub("!", '')
	return s
end

return p