Editing Module:Infobox Variant Category

local p = {}

local Infobox = require('Module:Infobox')
local parse = require('Module:Param Parse')

function p.main(frame)
	local args = frame:getParent().args
	
	local config = {
		infobox_name = 'Variant Category',
		class = {Infobox.smw_param('episode')},
	}
	
	local params = {
		parse.name,
		parse.image,
		parse.release,
		parse.premium,
		parse.episode,
		parse.additional_episode,
		parse.variant_type,
	}
	
	local infobox = Infobox.new(config, params, args)
	infobox
		:add_row{
			{tag='th', content=Infobox.param('name'), class='infobox-header', colspan='20'},
		}
		:add_row{
			{tag='td', content=Infobox.param('image'), class='infobox-image', colspan='20'},
		}
		:add_row{
			{tag='th', content='Release', colspan="6"},
			{tag='td', content=Infobox.param('release'), colspan="14"},
		}
	local additional_episode = infobox:is_param_defined(Infobox.param('additional_episode')) > 0
	if additional_episode then
		infobox:add_row{
			{tag='th', content='[[Episodes]]', colspan="6", rowspan="2"},
			{tag='td', content=Infobox.param('episode'), colspan="14"},
		}
		:add_row{
			{tag='td', content=Infobox.param('additional_episode'), colspan="14"},
		}
		local additional_episode_class = 'infobox-'..infobox:get_param(Infobox.smw_param('additional_episode'), 0)..'-additional'
		additional_episode_class = mw.ustring.gsub(additional_episode_class, '%s', '_')
		infobox:addClass(additional_episode_class)
	else
		infobox:add_row{
			{tag='th', content='[[Episode]]', colspan="6"},
			{tag='td', content=Infobox.param('episode'), colspan="14"},
		}
	end
	infobox
		:add_row{
			{tag='th', content='[[Premium Pass|Premium]]', colspan="6"},
			{tag='td', content=Infobox.param('premium'), colspan="14"},
		}
		:add_row{
			{tag='th', content='[[Variant]] type', colspan="6"},
			{tag='td', content=Infobox.param('variant_type'), colspan="14"},
		}
		:pad('20')
	
	return infobox
end

return p