Module:Equipment/Recommended

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

local p = {}

local params = require('Module:Shared/Paramtest')

local slots = {
	{ placement = 1, name = 'helm', icon = 'Slot_head', txt = 'Head', link = 'Head slot table' },
	{ placement = 2, name = 'neck', icon = 'Slot_neck', txt = 'Neck', link = 'Neck slot table' },
	{ placement = 3, name = 'cape', icon = 'Slot_cape', txt = 'Back', link = 'Cape slot table' },
	{ placement = 4, name = 'body', icon = 'Slot_body', txt = 'Body', link = 'Body slot table' },
	{ placement = 5, name = 'legs', icon = 'Slot_legs', txt = 'Legs', link = 'Legs slot table' },
	{ placement = 6, name = 'weapon', icon = 'Slot_weapon', txt = 'Weapon', link = 'Weapon slot table' },
	{ placement = 7, name = 'shield', icon = 'Slot_shield', txt = 'Shield', link = 'Shield slot table' },
	{ placement = 9, name = 'ammo', icon = 'Slot_ammo', txt = 'Ammo/Spell', link = 'Ammunition slot table' },
	{ placement = 10, name = 'gloves', icon = 'Slot_hands', txt = 'Gloves', link = 'Hand slot table' },
	{ placement = 11, name = 'boots', icon = 'Slot_feet', txt = 'Boots', link = 'Feet slot table' },
	{ placement = 12, name = 'ring', icon = 'Slot_ring', txt = 'Ring', link = 'Ring slot table' },
	{ placement = 13, name = 'passive', icon = '', txt = 'Special attack', link = '' },
	{ placement = 14, name = 'Familiar #1', icon = 'Slot_summon', txt = 'Familiar', link = '' },
	{ placement = 15, name = 'Familiar #2', icon = 'Slot_summon', txt = 'Familiar', link = '' }
	
}

function p.main(frame)
	local args = frame:getParent().args	
	-- Dynamic colspan and N/A generation value
	local greatest_row_size = 0
	-- Number of choices each slot can have
	local number_of_possible_choices = 5
	-- Have to sort this table or else the order is messed up when you use it
	table.sort(slots, function(a, b) return a.placement < b.placement end)
	
	local function make_row(slot, data)
		local tr = mw.html.create('tr')
		tr:tag('td'):wikitext(string.format('[[File:%s.png|%s|link=%s]]', slot.icon, slot.txt, slot.link))
		for _, v in ipairs(data) do
			tr:tag('td'):wikitext(v)
		end
		
		-- If the data size is smaller than GRS, then we need to fill up the remaining td's with N/As
		if #data < greatest_row_size then
			local difference = greatest_row_size - #data
			for i = 1, difference, 1 do
				tr:tag('td'):addClass('table-na'):wikitext('N/A')	
			end
		end
		return tr
	end
	
	-- Find the greatest row count
	for _, v in next, slots, nil do
		local grs = 0
		for i = 1, number_of_possible_choices, 1 do
			local check = args[v.name .. i]
			if check and params.has_content(check) then
				grs = grs + 1	
			end
		end
		
		if greatest_row_size < grs then
			greatest_row_size = grs	
		end
	end
	
	local parent = mw.html.create('table'):addClass('wikitable')
	
	-- If style is passed in, apply it above the table
	if args.style and params.has_content(args.style) then
		parent:tag('caption'):wikitext(string.format('Recommended equipment for %s', args.style))
	end
	
	parent:tag('tr')
		:tag('th'):wikitext('Slot'):done()
		:tag('th'):attr('colspan', greatest_row_size):wikitext('Item (most effective → least effective)'):done()
		
	for _, v in next, slots, nil do
		local row_data = {}
		for i = 1, number_of_possible_choices, 1 do
			local gear = args[v.name .. i]
			if gear and params.has_content(gear) then
				table.insert(row_data, gear)
			end
		end
		
		if #row_data > 0 then
			parent:node(make_row(v, row_data))
		end
	end
		
	return tostring(parent)
end

return p