Module:MapRewards

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

local U = require('Module:Core')

local template = {
	footer = "|}",
	header = [[{|style="text-align:center;width:100%%"
!colspan="5" style="background-color:#3baef5;color:white;padding:10px"|Rewards<span class="plainlinks" style="float:right">&#x5b;[https://en.kancollewiki.net/Module:Data/Map/%s?action=edit edit]&#x5d;</span>]],
	ship_row = [=[|-
!style="background-color:#3baef5;color:white;width:40%%"|Ship
|colspan="4" style="background-color:#eef2f7"|%s<br>[[%s]]]=],
	choice = [=[|-
!colspan="5" style="background-color:#3baef5;color:red"|OR]=],
	-- TODO: class="sortable"
	item_header = [[{|style="text-align:center;width:100%"
!style="background-color:#3baef5;color:white;width:40%"|Item/Equipment
!class="unsortable" style="background-color:#3baef5;color:white;width:15%"|丁<br>Casual
!class="unsortable" style="background-color:#3baef5;color:white;width:15%"|丙<br>Easy
!class="unsortable" style="background-color:#3baef5;color:white;width:15%"|乙<br>Medium
!class="unsortable" style="background-color:#3baef5;color:white;width:15%"|甲<br>Hard]],
	item_row = [=[|-
|data-sort-value="%s" style="background-color:#eef2f7;font-size:12px;line-height:14px"|%s<br>[[%s]]
|style="background-color:#eef2f7;font-size:18px"|%s
|style="background-color:#eef2f7;font-size:18px"|%s
|style="background-color:#eef2f7;font-size:18px"|%s
|style="background-color:#eef2f7;font-size:18px"|%s]=],
}

local assetTypes = { item = 'Item Icon', equipment = 'Equipment Card', ship = 'Ship Banner' }

local assetSizes = { item = '75px', equipment = '75px', ship = '240px' }

local function render(frame)
	local event = frame.args[1] or "??"
	local map = frame.args[2] or "??"
	local success, data = U.loadData("Data/Map/" .. event)
	if not success then return "Unknown event: " .. event end
	data = data[map]
	if not data then return "Unknown map: " .. map end
	local index = {}
	local tbl = {}
	local types = {}
	local choices = {}
	-- TODO: assuming 0-1 single ship reward on all difficulties
	local ship = false
	local function setDataItem(diff, v, choice, last)
		local name = v.item or v.equipment or v.ship or '??'
		types[name] = v.item and 'item' or v.equipment and 'equipment' or v.ship and 'ship' or '??'
		if choice then
			if not U.ifind(index, name) then table.insert(index, name) end
			tbl[name] = tbl[name] or {}
			tbl[name][diff] = { count = v.count or 1, level = v.level }
			if not last then
				choices[name] = true
			end
		else
			if v.ship then
				ship = v.ship
			else
				if not U.ifind(index, name) then table.insert(index, name) end
				tbl[name] = tbl[name] or {}
				tbl[name][diff] = { count = v.count or 1, level = v.level }
			end
		end
	end
	local function setData(diff)
		for _, v in ipairs(data[diff] or {}) do
			if v.choice then
				for i, v in ipairs(v.choice) do
					setDataItem(diff, v, true, i == #v.choice)
				end
			else
				setDataItem(diff, v)
			end
		end
	end
	local function getCount(name, diff)
		return ((tbl[name] or {})[diff] or {}).count or 0
	end
	local function getLevel(name, diff)
		local level = ((tbl[name] or {})[diff] or {}).level
		return level and string.format(" (★%d)", level) or ''
	end
	local function getCell(name, diff)
		local count = getCount(name, diff)
		return (count == 0 and '' or count) .. getLevel(name, diff)
	end
	local function getAsset(name)
		return string.format("[[File:%s %s.png|%s|link=%s]]", assetTypes[types[name]] or '??', name, assetSizes[types[name]] or '??', name)
	end
	setData('Hard')
	setData('Medium')
	setData('Easy')
	setData('Casual')
	local result = { string.format(template.header, mw.uri.encode(event, 'WIKI')) }
	if ship then
		table.insert(result, string.format(template.ship_row, getAsset(ship), ship))
	end
	table.insert(result, template.footer)
	table.insert(result, template.item_header)
	local sortedIndex = U.imap(index, function(name, i) return { name = name, i = i, typ = types[name] } end)
	table.sort(sortedIndex, function (x, y)
		if x.typ ~= y.typ then return x.typ > y.typ end
		if getCount(x.name, 'Casual') ~= getCount(y.name, 'Casual') then return getCount(x.name, 'Casual') > getCount(y.name, 'Casual') end
		if getCount(x.name, 'Easy') ~= getCount(y.name, 'Easy') then return getCount(x.name, 'Easy') > getCount(y.name, 'Easy') end
		if getCount(x.name, 'Medium') ~= getCount(y.name, 'Medium') then return getCount(x.name, 'Medium') > getCount(y.name, 'Medium') end
		if getCount(x.name, 'Hard') ~= getCount(y.name, 'Hard') then return getCount(x.name, 'Hard') > getCount(y.name, 'Hard') end
		return x.i < y.i
	end)
	local sortValues = {}
	for i, e in ipairs(sortedIndex) do sortValues[e.name] = i end
	for i, name in ipairs(index) do
		table.insert(result, string.format(template.item_row, sortValues[name], getAsset(name), name,
			getCell(name, 'Casual'), getCell(name, 'Easy'), getCell(name, 'Medium'), getCell(name, 'Hard')))
		if choices[name] then
			table.insert(result, template.choice)
		end
	end
	table.insert(result, template.footer)
	return table.concat(result, "\n")
end

return { render = render }