Module:Infobox story

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

---

--------------------------
-- Module for [[Template:Infobox story]]
------------------------
local getArgs   = require('Module:Arguments').getArgs
local builder   = require('Module:InfoboxBuilder')
local constants = require('Module:Constants')
local utils     = require('Module:Utilities')
local icons     = require('Module:Icons')._main

local p = {}
local categories = ""

local function formatImage(image_name)
	return string.format("[[File:%s|220px]]", image_name)
end

local function formatSplit(val)
	if val then
		local split_vals = utils.split(val, ",")
		for i,v in ipairs(split_vals) do
			split_vals[i] = utils.trim(v)
		end
		
		return table.concat(split_vals, "<br>")
	end
	
	return nil
end

local function formatUnits(units)
	if units and units ~= '' then
		local split_units = mw.text.split(units, ',')
		local res_list = {}
		
		for i,v in ipairs(split_units) do
			local icon = icons({ v, size = '20px' })
			if icon then
				table.insert(res_list, string.format(
					"%s [[%s]]",
					icon,
					constants.get_unit(v)
				))
			end
		end
		
		return table.concat(res_list, '<br>')
	end
	
	return nil
end

local function formatCharacters(characters)
	local characterArr = utils.split(characters, ',')
	local resArr = {}
	for i,v in ipairs(characterArr) do
		table.insert(resArr, icons({ v, size = '30px' }))
	end
	
	return table.concat(resArr, " ")
end



function p.main(frame)
	local args = getArgs(frame)
	local infobox = builder.new()
	
	infobox:setName('Infobox story')
		:setWidth("330px")
		:setParams{
			{ name = 'story name', default = mw.title.getCurrentTitle().text },
			{ name = 'image', func = formatImage },
			{ name = 'japanese' },
			{ name = 'romaji' },
			{ name = 'chapters' },
			{ name = 'unit', func = formatUnits },
			{ name = 'character', func = formatCharacters },
			{ name = 'translator', func = formatSplit },
			{ name = 'proofreader', func = formatSplit },
			{ name = 'source', func = formatSplit }
		}
		:setArgs(args)

	infobox
		:addHeader({ tag = 'argth', content = 'story name' })
		:addImage({
			{ tag = 'argtd', content = 'image' },
		})
		:addRow(
			{
				{ tag = 'th', content = 'Japanese', colspan = 12 },
				{ tag = 'argtd', content = 'japanese', colspan = 18 }
			},
			{ hideIfEmpty = { 'japanese' } }
		)
		:addRow(
			{
				{ tag = 'th', content = 'Romaji', colspan = 12 },
				{ tag = 'argtd', content = 'romaji', colspan = 18 }
			},
			{ hideIfEmpty = { 'romaji' } }
		)
		:addHeader({ tag = 'th', content = 'Story Information' }, { subheader = true })
		:addRow({
			{ tag = 'th', content = 'Chapters', colspan = 12 },
			{ tag = 'argtd', content = 'chapters', colspan = 18 }
		})
		:addRow({
			{ tag = 'th', content = 'Unit(s)', colspan = 12 },
			{ tag = 'argtd', content = 'unit', colspan = 18 }
		})
		:addRow({
			{ tag = 'th', content = 'Character(s)', colspan = 12 },
			{ tag = 'argtd', content = 'character', colspan = 18 }
		})
		:addHeader({ tag = 'th', content = 'Translation Information' }, { subheader = true })
		:addRow({
			{ tag = 'th', content = 'Translator(s)', colspan = 12 },
			{ tag = 'argtd', content = 'translator', colspan = 18 }
		})
		:addRow({
			{ tag = 'th', content = 'Proofreader(s)', colspan = 12 },
			{ tag = 'argtd', content = 'proofreader', colspan = 18 }
		})
		:addRow({
			{ tag = 'th', content = 'Source(s)', colspan = 12 },
			{ tag = 'argtd', content = 'source', colspan = 18 }
		})
	
	if mw.title.getCurrentTitle().namespace == 0 then
		categories = "[[Category:Stories]]" .. categories
		
		local procArgs = infobox:getArgs('processed')
		local rawArgs = infobox:getArgs('raw')
		
		if rawArgs['type'] then
			if rawArgs['type'] == 'main' then
				categories = categories .. '[[Category:Main stories]]'
			elseif rawArgs['type'] == 'event' then
				categories = categories .. '[[Category:Event stories]]'
			elseif rawArgs['type'] == 'side' then
				categories = categories .. '[[Category:Side stories]]'
			end
		end
	end

	return infobox:tostring() .. categories
end

return p