Editing Module:Infobox Monster

local p = {}

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

local attack_styles = {
    ['none'] = 'None',
    ['impact'] = 'Impact',
    ['cryonae'] = 'Cryonae',
    ['arborae'] = 'Arborae',
    ['tempestae'] = 'Tempestae',
    ['infernae'] = 'Infernae',
    ['necromae'] = 'Necromae',
}

function attack_style_func(style)
    local attack_style = attack_styles[string.lower(style or '')]
    if attack_style then
        if attack_style == 'None' then
            return 'None'
        else
            return '[[File:'..attack_style..' damage icon.png|18px|link='..attack_style..']] [['..attack_style..']]'
        end
    end
    return nil
end

function p.main(frame)
	local args = frame:getParent().args
	
	local config = {
		infobox_name = 'Monster',
		class = {Infobox.smw_param('episode')},
	}
	
	local params = {
		parse.name,
		parse.image_size300,
		parse.release,
		parse.premium,
		parse.episode,
		{name = 'level', func = parse.number},
		{name = 'hp', func = parse.number}, -- TODO - move to Module:Param Parse once more details are available, and create smw_property
		{name = 'xp', func = parse.number}, -- TODO - same as above
		{name = 'att_style', func = attack_style_func},
		{name = 'immune_to', func = attack_style_func},
		{name = 'vulnerable_to', func = attack_style_func},
		parse.variant,
		parse.passive,
		parse.aggressive,
		parse.examine,
		parse.profession_bubble_a,
		parse.profession_bubble_b,
		-- Include these next params just to export data to smw
		parse.profession_a_smw,
		parse.profession_a_level_smw,
		parse.profession_b_smw,
		parse.profession_b_level_smw
	}
	
	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="7"},
			{tag='td', content=Infobox.param('release'), colspan="13"},
		}
		:add_row{
			{tag='th', content='[[Episode]]', colspan="7"},
			{tag='td', content=Infobox.param('episode'), colspan="13"},
		}
		:add_row{
			{tag='th', content='[[Premium Pass|Premium]]', colspan="7"},
			{tag='td', content=Infobox.param('premium'), colspan="13"},
		}
		:add_row{
			{tag='th', content='Level', colspan="7"},
			{tag='td', content=Infobox.param('level'), colspan="13"},
		}
		:add_row{
			{tag='th', content='[[Health]]', colspan="7"},
			{tag='td', content=Infobox.param('hp'), colspan="13"},
		}
		:add_row{
			{tag='th', content='Experience', colspan="7"},
			{tag='td', content=Infobox.param('xp'), colspan="13"},
		}
	if infobox:is_param_defined(Infobox.param('att_style')) > 0 then
		infobox:add_row{
			{tag='th', content='[[Attack Style]]', colspan="7"},
			{tag='td', content=Infobox.param('att_style'), colspan="13"},
		}
	end
	if infobox:is_param_defined(Infobox.param('immune_to')) > 0 then
		infobox:add_row{
			{tag='th', content='Immune to', colspan="7"},
			{tag='td', content=Infobox.param('immune_to'), colspan="13"},
		}
	end
	if infobox:is_param_defined(Infobox.param('vulnerable_to')) > 0 then
		infobox:add_row{
			{tag='th', content='Vulnerable to', colspan="7"},
			{tag='td', content=Infobox.param('vulnerable_to'), colspan="13"},
		}
	end
	if infobox:is_param_defined(Infobox.param('variant')) > 0 and infobox:get_param(Infobox.param('variant'), 0) ~= 'N/A' then
		infobox:add_row{
			{tag='th', content='[[Variant]] of', colspan="7"},
			{tag='td', content=Infobox.param('variant'), colspan="13"},
		}
	end
	infobox:add_row{
			{tag='th', content='[[File:Passive activity icon.png|14px|link=Passive activity]] [[Passive activity|Passive]]', colspan="7"},
			{tag='td', content=Infobox.param('passive'), colspan="13"},
		}
		:add_row{
			{tag='th', content='[[File:Aggressive activity icon.png|3px|link=Aggressive]] [[Aggressive]]', colspan="7"},
			{tag='td', content=Infobox.param('aggressive'), colspan="13"},
		}
		:add_row{
			{tag='th', content='Examine', colspan="7"},
			{tag='td', content=Infobox.param('examine'), colspan="13"}
		}
	local bubble_a = infobox:is_param_defined(Infobox.param('profession_bubble_a')) > 0
	local bubble_b = infobox:is_param_defined(Infobox.param('profession_bubble_b')) > 0
	if bubble_a and bubble_b then
		infobox:add_row{
			{tag='td', content=Infobox.param('profession_bubble_a'), class='infobox-bubble-item', colspan="10"},
			{tag='td', content=Infobox.param('profession_bubble_b'), class='infobox-bubble-item', colspan="10"},
			addClass = 'infobox-bubble-row'
		}
	elseif bubble_a then
		infobox:add_row{
			{tag='td', content=Infobox.param('profession_bubble_a'), class='infobox-bubble-item', colspan="20"},
			addClass = 'infobox-bubble-row'
		}
	elseif bubble_b then
		infobox:add_row{
			{tag='td', content=Infobox.param('profession_bubble_b'), class='infobox-bubble-item', colspan="20"},
			addClass = 'infobox-bubble-row'
		}
	else
		infobox:pad('20')
	end
	
	return infobox
end

return p