Module:Infobox unit

To generate {{Infobox unit}}, invoke using the main function.

---

--------------------------
-- Module for [[Template:Infobox unit]]
------------------------
local getArgs   = require('Module:Arguments').getArgs
local InfoboxBuilder      = require('Module:InfoboxBuilder')
local DisplayFunctions    = require('Module:InfoboxBuilderDisplayFunctions')
local ProcessingFunctions = require('Module:InfoboxBuilderProcessingFunctions')
local utils     = require('Module:Utilities')

local p = {}

local function formatMembers(members)
	if #members < 2 then
		return "[[" .. members[1] .. "]]"
	end
	
	local ul = mw.html.create('ul')
	
	for i=1,#members do
		ul:tag('li')
			:wikitext("[[" .. members[i] .. "]]")
	end
	
	return tostring(ul)
end


function p.main(frame)
	local args = getArgs(frame)
	local infobox = builder.new();
	
	infobox:setName('Infobox unit')
		:setParams{
			{ name = 'name', default = mw.title.getCurrentTitle().text },
			{ name = 'image', dFunc = DisplayFunctions.generateImage('300px') },
			{ name = 'japanese' },
			{ name = 'romaji' },
			{ name = 'english' },
			{ name = 'logo', dFunc = DisplayFunctions.generateImage('200px') },
			{ name = 'members', pFunc = ProcessingFunctions.convertStringToArray(','), dFunc = formatMembers },
			{ name = 'virtual singers', pFunc = ProcessingFunctions.convertStringToArray(','), dFunc = formatMembers }
		}
		:setArgs(args)
		:processArgs()

	infobox
		:addHeader({ tag = 'argth', content = 'name' })
		:addImage{{ tag = 'argtd', content = 'image' }}
		:addRow(
			{
				{ tag = 'th', content = 'Japanese' },
				{ tag = 'argtd', content = 'japanese' }
			},
			{ hideIfEmpty = { 'japanese' } }
		)
		:addRow(
			{
				{ tag = 'th', content = 'Romaji' },
				{ tag = 'argtd', content = 'romaji' }
			},
			{ hideIfEmpty = { 'romaji' } }
		)
		:addRow(
			{
				{ tag = 'th', content = 'English' },
				{ tag = 'argtd', content = 'english' }
			},
			{ hideIfEmpty = { 'english' } }
		)
		:addHeader({ tag = 'th', content = 'Logo'}, { subheader = true })
		:addImage{{ tag = 'argtd', content = 'logo' }}
		:addHeader({ tag = 'th', content = 'Information' }, { subheader = true })
		:addRow(
			{
				{ tag = 'th', content = 'Members' },
				{ tag = 'argtd', content = 'members' }
			},
			{ hideIfEmpty = { 'members' } }
		)
		:addRow(
			{
				{ tag = 'th', content = 'Virtual Singers' },
				{ tag = 'argtd', content = 'virtual singers' }
			},
			{ hideIfEmpty = { 'virtual singers' } }
		)
		
	local categories = ""
	if mw.title.getCurrentTitle().namespace == 0 then
		categories = "[[Category:Units]]" .. categories
	end

	return infobox:tostring() .. categories
end

return p