Editing Module:MonsterVariantsTable

require('strict')
require('Module:Mw.html extension')
local editbutton = require('Module:Edit button')
local p = {}

function p.main(frame)
	local args = frame:getParent().args
	return p._main(args[1] or args.variant or '')
end

local function formatProfessionLevel(profession, level, high)
	profession = profession or 'Unknown profession'
	local level_text = tostring(level or '?')
	if high then
		level_text = ('%s➨%d'):format(level_text, high)
	end
	return ('%s [[File:%s small icon.png|21x21px|link=%s]]'):format(level_text, profession, profession)
end

function p._main(variant)
	local results = mw.smw.ask{
		('[[Variant of::%s]]'):format(variant),
		'?Image#64px;x64px = img',
		'?Name = name',
		'?= page',
		'?Unlock_level = unlock_lvl',
		'?Unlock profession # = unlock_profession',
		'?Profession_Level_A = combat',
		'?Profession A # = profession',
		'?Version default = default',
		'?Variant name = variant',
		'?Version anchor = version',
		'?-Has subobject#- = subobject',
		'sort=Profession Level A',
		'order=asc'
	}
	mw.logObject(results)
	if results == nil or results[1] == nil then
		return ":''No variants found for "..variant.."''"
	end

	local filtered = p.filter_versions(results)
	local sorted = p.sort_by_combat(filtered)

	return p.create_table(sorted)
end

function p.filter_versions(data)
	local filtered = {}
	for _, entry in ipairs(data) do
		if entry.default then
			entry.page = ('[[%s|%s]]'):format(entry.subobject, entry.name)
			table.insert(filtered, entry)
		elseif entry.version == nil then
			table.insert(filtered, entry)
		end
	end
	return filtered
end

function p.sort_by_combat(data)
	table.sort(data, function(a, b)
		-- Determine combat values
		local combatA = a.combat or a.unlock_lvl or math.huge
		local combatB = b.combat or b.unlock_lvl or math.huge
		-- Sort in ascending order
		return combatA < combatB
	end)
	return data
end

function p.insert_row(tbl, entry)
	local edit = editbutton("'''?''' (edit)", entry.name)
	tbl:tr()
		:td{ entry.img, addClass = 'plinkt-image' }:done()
		:td{ entry.page, addClass = 'plinkt-link' }:done()
		:td()
			:IF(entry.unlock_profession)
				:wikitext(entry.unlock_lvl and formatProfessionLevel(entry.unlock_profession, entry.unlock_lvl) or edit)
			:ELSE()
				:addClass('table-na')
				:wikitext('N/A')
			:END()
		:done()
		:td()
			:wikitext(entry.combat and formatProfessionLevel(entry.profession, entry.combat) or edit)
		:done()
	:done()

	return tbl
end

function p.create_table(results)
	local out = mw.html.create('table')
		:addClass('wikitable sortable align-right-3 align-right-4')
		:tr()
			:th{ 'Monster', attr = { colspan = 2 } }:done()
			:th('Unlocked at'):done()
			:th('Combat level'):done()
		:done()

	for _, entry in ipairs(results) do
		out = p.insert_row(out, entry)
	end

	return out
end

return p