Module:Rescue Team DX Pokémon Locations table

Documentation for this module may be created at Module:Rescue Team DX Pokémon Locations table/doc

--------------------------------------------------------------------------------
--
--				Module:Rescue Team Pokémon Locations table
--
--------------------------------------------------------------------------------

local p = {}
local mw = require('mw')

function p.main(frame)
	local args = frame.args
	local output = {}
	-- Header row
	table.insert(output, '<table class="templateTheme" style="text-align: center; margin: auto; width: 50%;">')
	table.insert(output, '<tr>')
	table.insert(output, '<th class="templatePrimary" rowspan="2" style="width: 20%;">Location</th>')
	table.insert(output, '<th class="templatePrimary" rowspan="2" style="width: 20%;">Floors</th>')
	table.insert(output, '</tr>')
	-- Data rows
	for i = 1, 100 do
		local locationArg = args["location_" .. i]
		local floorsArg = args["floors_" .. i]
		if not locationArg and not floorsArg then
			break -- exit loop when no more rows are found
		end
		local rowData = {}
		table.insert(rowData, '<tr class="locationsTableRow">')
		if locationArg == nil or locationArg == '' then
			table.insert(rowData, '<th class="templatePrimary">' .. frame:expandTemplate{title = "Research"} .. '</th>')
			table.insert(rowData, '<td class="templateSecondary">' .. frame:expandTemplate{title = "Research"} .. '</td>')
			table.insert(rowData, '<td class="templateSecondary">' .. frame:expandTemplate{title = "Research"} .. '</td>')
		else
			table.insert(rowData, '<td class="templateSecondary">' .. frame:expandTemplate{title = "RescueTeamDX", args = {locationArg}} .. '</td>')
			table.insert(rowData, '<td class="templateSecondary">' .. frame:preprocess('{{#invoke:Rescue Team DX Trap Data Cell|rescueTeamDXTrapDataCell|name=' .. trapArg .. '}}') .. '</td>')
		end
		table.insert(rowData, '<td class="templateSecondary">' .. (locationArg or frame:expandTemplate{title = "Research"}) .. '</td>')
		table.insert(rowData, '<td class="templateSecondary">' .. (floorsArg or frame:expandTemplate{title = "Research"}) .. '</td>')
		table.insert(rowData, '</tr>')
		table.insert(output, table.concat(rowData))
	end
	table.insert(output, '</table>')
	return table.concat(output)
end

return p