Module:Pokemon

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

local p = {}

local DATA_FORMAT = 'CSV with header'

local COLUMNS = {
	number = 'Pokedex Number',
	name = 'Name',
	type1 = 'Type1',
	type2 = 'Type2',
	ability1 = 'Ability 1',
	ability2 = 'Ability 2',
	ha = 'Hidden Ability',
	generation = 'Generation',
	hp = 'HP',
	atk = 'Attack',
	def = 'Defense',
	spatk = 'Special Attack',
	spdef = 'Special Defense',
	spd = 'Speed',
	catch = 'Catch Rate',
	evhp = 'EVHP',
	evatk = 'EVATK',
	evdef = 'EVDEF',
	evspa = 'EVSPA',
	evspd = 'EVSPD',
	evsp = 'EVSP',
	male = 'Male Ratio',
	height = 'Height',
	weight = 'Weight',
	npc = 'NPC',
	exp = 'Base Experience',
	rarity = 'Tier',
}

local GEN_NUMERALS = { 'I', 'II', 'III', 'IV', 'V', 'VI', 'VII', 'VIII', 'IX' }

local STATS = {
	{ key = 'hp', label = 'HP', template = 'HP' },
	{ key = 'atk', label = 'Atk', template = 'Attack' },
	{ key = 'def', label = 'Def', template = 'Defense' },
	{ key = 'spatk', label = 'Sp.Atk', template = 'Special Attack' },
	{ key = 'spdef', label = 'Sp.Def', template = 'Special Defense' },
	{ key = 'spd', label = 'Spd', template = 'Speed' },
}

local EV_KEYS = { 'evhp', 'evatk', 'evdef', 'evspa', 'evspd', 'evsp' }

local function nonEmpty(s)
	if s == nil then return nil end
	s = mw.text.trim(tostring(s))
	if s == '' then return nil end
	return s
end

local function templateArgs(frame)
	local parent = frame:getParent()
	return (parent and parent.args) or frame.args
end

local function padNumber(n)
	n = tonumber(n) or 0
	return string.format('%03d', n)
end

local PokemonForms = require('Module:PokemonForms')
local FORM_SUFFIXES = PokemonForms.FORM_SUFFIXES
local STRIP_SUFFIXES = PokemonForms.STRIP_SUFFIXES
local MEGA_VARIANTS = PokemonForms.MEGA_VARIANTS or { '', ' X', ' Y' }

-- The single data fetch: the whole raw list is loaded once per page render
-- and every lookup (by name, by number, form probing) reads from it.
local allRowsCache = nil
local numberIndexCache = nil

local function loadAllRows()
	if allRowsCache then
		return allRowsCache
	end
	local csvData = mw.ext.externalData.getExternalData{
		url = 'https://wiki.pokemonrevolution.net/index.php?title=Special:GetData/PokemonRawList',
		format = DATA_FORMAT,
		data = COLUMNS,
	}
	allRowsCache = {}
	numberIndexCache = {}
	if csvData then
		for _, row in ipairs(csvData) do
			if nonEmpty(row.name) then
				allRowsCache[row.name] = row
				local num = nonEmpty(row.number)
				-- First occurrence wins so base forms beat Megas sharing a number.
				if num and numberIndexCache[num] == nil then
					numberIndexCache[num] = row
				end
			end
		end
	end
	return allRowsCache
end

local function hyphenSpaceVariant(s)
	if mw.ustring.find(s, '-', 1, true) then
		return (mw.ustring.gsub(s, '%-', ' '))
	end
	return (mw.ustring.gsub(s, ' ', '-'))
end

local function byName(name)
	local rows = loadAllRows()
	return rows[name] or rows[hyphenSpaceVariant(name)]
end

local function byNumber(number)
	loadAllRows()
	return numberIndexCache[tostring(number)]
end

-- Mega evolutions live in a separate CSV without a Special:GetData page, so
-- they are fetched per name through the filtered file source — the same path
-- Template:PokemonInfo/Mega has always used.
local MEGA_COLUMNS = {
	number = 'Pokedex Number',
	name = 'Name',
	type1 = 'Type1',
	type2 = 'Type2',
	ability1 = 'Ability 1',
	ability2 = 'Ability 2',
	ha = 'Hidden Ability',
	generation = 'Generation',
	hp = 'HP',
	atk = 'Attack',
	def = 'Defense',
	spatk = 'Special Attack',
	spdef = 'Special Defense',
	spd = 'Speed',
	evhp = 'EVHP',
	evatk = 'EVATK',
	evdef = 'EVDEF',
	evspa = 'EVSPA',
	evspd = 'EVSPD',
	evsp = 'EVSP',
	height = 'Height',
	weight = 'Weight',
	stone = 'Stone',
}

local MEGA_SPEC
do
	local parts = {}
	for key, header in pairs(MEGA_COLUMNS) do
		parts[#parts + 1] = key .. '=' .. header
	end
	MEGA_SPEC = table.concat(parts, ',')
end

local megaCache = {}

local function megaByName(name)
	if megaCache[name] == nil then
		local ok, values = pcall(function()
			return mw.ext.externalData.getFileData{
				source = 'data',
				['file name'] = 'RawMegaPokemonList.csv',
				format = DATA_FORMAT,
				filters = 'Name=' .. name,
				data = MEGA_SPEC,
			}
		end)
		if ok and values and values.name ~= nil then
			megaCache[name] = values
		else
			megaCache[name] = false
		end
	end
	return megaCache[name] or nil
end

-- Page/raw-list name -> base name the form candidates are derived from
-- ("Alolan Rattata" -> "Rattata", "Eiscue-Ice Face" -> "Eiscue").
local function formStem(name)
	for _, form in ipairs(FORM_SUFFIXES) do
		local head = form.label .. ' '
		if mw.ustring.sub(name, 1, mw.ustring.len(head)) == head then
			return mw.ustring.sub(name, mw.ustring.len(head) + 1)
		end
	end
	for _, suffix in ipairs(STRIP_SUFFIXES) do
		for _, sep in ipairs({ '-', ' ' }) do
			local tail = sep .. suffix
			if mw.ustring.sub(name, -mw.ustring.len(tail)) == tail then
				return mw.ustring.sub(name, 1, -mw.ustring.len(tail) - 1)
			end
		end
	end
	return name
end

local function rawListFormName(form, stem)
	if form.rawList == 'suffix' then
		return stem .. '-' .. form.suffix
	elseif form.rawList == 'prefix-hyphen' then
		return form.label .. '-' .. stem
	elseif form.rawList == 'suffix-label' then
		return stem .. '-' .. form.label
	elseif form.rawList == 'suffix-space' then
		return stem .. ' ' .. form.label
	end
	return form.label .. ' ' .. stem
end

local function typeColors(frame, pokeType)
	return {
		light = frame:expandTemplate{ title = pokeType .. ' color light' },
		base = frame:expandTemplate{ title = pokeType .. ' color' },
		dark = frame:expandTemplate{ title = pokeType .. ' color dark' },
	}
end

local statColorCache = {}
local function statColors(frame, statTemplate)
	if not statColorCache[statTemplate] then
		statColorCache[statTemplate] = {
			base = frame:expandTemplate{ title = statTemplate .. ' color' },
			dark = frame:expandTemplate{ title = statTemplate .. ' color dark' },
		}
	end
	return statColorCache[statTemplate]
end

local function navTableHtml(frame, data, colors, pageName)
	local number = tonumber(data.number)

	local prevCell = ''
	if number and number > 1 then
		local prev = byNumber(number - 1)
		if prev then
			prevCell = ' [[File:' .. prev.number .. 'Icon.png|' .. prev.name .. '|link=' .. prev.name ..
				']]  #' .. padNumber(prev.number) .. ' [[' .. prev.name .. ']]'
		end
	end

	local nextCell = ''
	if number then
		local nxt = byNumber(number + 1)
		if nxt then
			nextCell = '[[' .. nxt.name .. ']] #' .. padNumber(nxt.number) .. '  [[File:' ..
				nxt.number .. 'Icon.png|' .. nxt.name .. '|link=' .. nxt.name .. ']]'
		end
	end

	return '<table style="margin-left: auto; margin-right: auto; margin-bottom: 15px; height: 60px; ' ..
		'border: 4px #' .. colors.dark .. ' solid; font-weight: bold; width: 65%; color: #' .. colors.dark ..
		'; border-radius: 20px; padding: 3px; background: #' .. colors.base .. ';">' ..
		'<tr>' ..
		'<td style="text-align: left; width: 36%; background: #' .. colors.light ..
		'; border-radius: 15px 1px 1px 15px; border: 1px solid #' .. colors.dark .. ';">' .. prevCell .. '</td>' ..
		'<td style="text-align: center; width: 28%; background: #' .. colors.light ..
		'; border: 1px solid #' .. colors.dark .. ';">#' .. padNumber(number) .. '&nbsp;&nbsp;-&nbsp;&nbsp;' ..
		pageName .. '</td>' ..
		'<td style="text-align: right; width: 36%; padding-right: 10px; background: #' .. colors.light ..
		'; border: 1px solid #' .. colors.dark .. '; border-radius: 1px 15px 15px 1px;">' .. nextCell .. '</td>' ..
		'</tr></table>'
end

function p.navTable(frame)
	local args = templateArgs(frame)
	local pageName = mw.title.getCurrentTitle().text
	local name = nonEmpty(args.Name) or pageName

	local data = byName(name)
	if not data then
		return '<strong class="error">PokeNavTable: no data found for "' .. name .. '"</strong>'
	end

	return navTableHtml(frame, data, typeColors(frame, data.type1), pageName)
end

local function infoCell(colors, labelHtml, valueHtml)
	return '<td colspan="3" style="width: 50%; border: 1px solid #' .. colors.dark .. '; background-color: #' .. colors.light ..
		'; border-radius: 15px; padding: 5px;">' ..
		'<p style="margin-bottom: 7px; font-weight: bold; color: #' .. colors.dark .. ';">' .. labelHtml .. '</p>' ..
		'<span style="border-radius: 12px; padding: 4px 10px; border: 1px solid #' .. colors.dark ..
		'; background-color: #FFFFFF;">&nbsp;&nbsp;' .. valueHtml .. '&nbsp;&nbsp;</span></td>'
end

local function abilityCell(colors, label, ability, widthPercent)
	if not ability then return '' end
	local style = 'border-radius: 12px; padding: 4px 10px; border: 1px solid #' .. colors.dark ..
		'; background-color: #FFFFFF;'
	if widthPercent then
		style = 'width: ' .. widthPercent .. '%; ' .. style
	end
	return '<td style="' .. style .. '"><b>' .. label .. '</b><br>[[' .. ability .. '_(ability)|' .. ability ..
		']]</td>'
end

local function abilitiesRow(colors, data)
	local entries = {
		{ 'Ability 1', nonEmpty(data.ability1) },
		{ 'Ability 2', nonEmpty(data.ability2) },
		{ 'Hidden Ability', nonEmpty(data.ha) },
	}
	local present = {}
	for _, entry in ipairs(entries) do
		if entry[2] then present[#present + 1] = entry end
	end
	local widthPercent = (#present > 0) and string.format('%.2f', 100 / #present) or nil
	local cells = {}
	for _, entry in ipairs(present) do
		cells[#cells + 1] = abilityCell(colors, entry[1], entry[2], widthPercent)
	end
	return table.concat(cells)
end

local function statValueCell(frame, stat, value)
	local colors = statColors(frame, stat.template)
	return '<td style="width: 16.66%; border: 1px solid #' .. colors.dark .. '; background-color: #' .. colors.base ..
		'; border-radius: 12px; padding: 4px; font-size: 11px;">\'\'\'' .. stat.label .. '\'\'\'<br>' ..
		value .. '</td>'
end

-- One horizontal stat bar, scaled against the stat ceiling of 255.
-- The fill is colored by the value (red -> yellow -> green -> teal); the
-- label keeps the stat's own color.
local function statBarRow(frame, stat, value)
	local labelColors = statColors(frame, stat.template)
	local pct = math.floor(value / 255 * 100 + 0.5)
	if pct < 2 then pct = 2 end
	if pct > 100 then pct = 100 end
	local hue = math.floor(math.max(0, math.min(180, value * 180 / 255)) + 0.5)
	local fill = 'hsl(' .. hue .. ', 85%, 45%)'
	return '<tr>' ..
		'<td style="text-align: right; padding: 2px 8px 2px 4px; font-weight: bold; font-size: 12px; ' ..
		'color: #' .. labelColors.dark .. '; white-space: nowrap;">' .. stat.label .. '</td>' ..
		'<td style="width: 100%; padding: 2px 0;"><div style="background-color: #FFFFFF; border: 1px solid ' ..
		'#999999; border-radius: 8px; height: 13px; overflow: hidden;">' ..
		'<div style="width: ' .. pct .. '%; height: 100%; background-color: ' .. fill .. ';"></div>' ..
		'</div></td>' ..
		'<td style="text-align: left; padding: 2px 4px 2px 8px; font-weight: bold; font-size: 12px; ' ..
		'width: 34px;">' .. value .. '</td></tr>'
end

local MALE_COLOR = '3B6BF5'
local FEMALE_COLOR = 'F5629C'

local function genderRatioHtml(male)
	if male == 'Genderless' then
		return '<span style="display: inline-block; color: #000000; background-color: #FFFFFF; ' ..
			'border-radius: 12px; border: 1px solid #000000; padding: 3px 12px; font-weight: bold;">' ..
			'Genderless</span>'
	end
	-- ExternalData drops cells whose value is '0' (PHP empty('0') is true),
	-- so a nil/blank ratio on an otherwise-complete row means 0% male.
	local m = tonumber(male)
	if m == nil then
		m = nonEmpty(male) and 100 or 0
	end
	if m < 0 then m = 0 end
	if m > 100 then m = 100 end
	local f = 100 - m

	local segments = ''
	if m > 0 then
		segments = segments .. '<div style="width: ' .. m .. '%; background-color: #' .. MALE_COLOR .. ';"></div>'
	end
	if f > 0 then
		segments = segments .. '<div style="width: ' .. f .. '%; background-color: #' .. FEMALE_COLOR .. ';"></div>'
	end

	local labels = {}
	if m > 0 then
		labels[#labels + 1] = '<span style="color: #' .. MALE_COLOR .. '; font-weight: bold;">' .. m .. '% ♂</span>'
	end
	if f > 0 then
		labels[#labels + 1] = '<span style="color: #' .. FEMALE_COLOR .. '; font-weight: bold;">' .. f .. '% ♀</span>'
	end

	return '<div style="width: 85%; margin: 4px auto 0;">' ..
		'<div style="display: flex; height: 12px; border-radius: 8px; overflow: hidden; ' ..
		'border: 1px solid #555555; background-color: #FFFFFF;">' .. segments .. '</div>' ..
		'<div style="margin-top: 4px; font-size: 12px;">' .. table.concat(labels, ' &nbsp;·&nbsp; ') ..
		'</div></div>'
end

local function categories(data, gen, bstCategory, args)
	local cats = { '[[Category:Pokémon species]]' }
	if gen then
		cats[#cats + 1] = '[[Category:Generation ' .. gen .. ' Pokémon]]'
	end
	cats[#cats + 1] = '[[Category:' .. data.type1 .. '-type Pokémon]]'
	if nonEmpty(data.type2) then
		cats[#cats + 1] = '[[Category:' .. data.type2 .. '-type Pokémon]]'
	end
	cats[#cats + 1] = '[[Category:' .. data.rarity .. ' Pokémon]]'
	if nonEmpty(data.npc) then
		cats[#cats + 1] = '[[Category:NPC-obtainable Pokémon]]'
	end
	if not nonEmpty(args.EvolvesFrom) then
		cats[#cats + 1] = '[[Category:Base-stage Pokémon]]'
	end
	cats[#cats + 1] = '[[Category:' .. bstCategory .. '-plus-BST Pokémon]]'
	return table.concat(cats, ' ')
end

local function infoboxHtml(frame, args, data, displayName, width, pageName, includeNav)
	local colors = typeColors(frame, data.type1)
	-- Regional forms carry values like '1A'/'5A'; use the leading digits.
	local gen = GEN_NUMERALS[tonumber(mw.ustring.match(tostring(data.generation or ''), '^%d+') or '')]

	-- Types pill(s).
	local typesHtml
	if nonEmpty(data.type2) then
		local type2Colors = typeColors(frame, data.type2)
		typesHtml =
			'<td style="width: 50%; background-color: #' .. colors.dark .. '; border-radius: 25px 1px 1px 25px;">' ..
			'[[' .. data.type1 .. '_(type)|<span style="color: #FFFFFF;">' .. data.type1 .. '</span>]]</td>' ..
			'<td style="width: 50%; background-color: #' .. type2Colors.dark .. '; border-radius: 1px 25px 25px 1px;">' ..
			'[[' .. data.type2 .. '_(type)|<span style="color: #FFFFFF;">' .. data.type2 .. '</span>]]</td>'
	else
		typesHtml =
			'<td style="width:100%; background-color: #' .. colors.dark .. '; border-radius: 25px;">' ..
			'[[' .. data.type1 .. '_(type)|<span style="color: #FFFFFF;">' .. data.type1 .. '</span>]]</td>'
	end

	-- Base stats
	local statRows, bst = {}, 0
	for _, stat in ipairs(STATS) do
		local value = tonumber(data[stat.key]) or 0
		bst = bst + value
		statRows[#statRows + 1] = statBarRow(frame, stat, value)
	end
	local bstCategory = math.floor(bst / 100 + 0.5) * 100

	-- EV yield.
	local evCells, evTotal = {}, 0
	for i, stat in ipairs(STATS) do
		local ev = tonumber(data[EV_KEYS[i]]) or 0
		evTotal = evTotal + ev
		evCells[#evCells + 1] = statValueCell(frame, stat, ev)
	end

	-- Catch rate.
	local catchRaw = tonumber(data.catch) or 0
	local catchPercent = string.format('%.2f', catchRaw / 255 / 3 * 100)

	local buf = {}
	if includeNav then
		buf[#buf + 1] = navTableHtml(frame, data, colors, pageName)
	end
	buf[#buf + 1] = '<table align="right" class="informational-box" cellpadding="4" cellspacing="4" ' ..
		'style="width: ' .. width .. '%; border: 4px solid #' .. colors.dark .. '; background-color: #' ..
		colors.base .. '; padding: 4px; border-radius: 15px;">'

	-- Header: dex number + icon, name.
	buf[#buf + 1] = '<tr><td colspan="3" style="width: 50%; border: 1px solid #' .. colors.dark .. '; background-color: #' ..
		colors.light .. '; border-radius: 15px;"><span style="margin-bottom: 7px; font-weight: bold; color:#' ..
		colors.dark .. ';">#' .. padNumber(data.number) .. '</span>&nbsp;&nbsp;&nbsp;[[File:' .. data.number ..
		'Icon.png]]</td><td colspan="3" style="width: 50%; border: 1px solid #' .. colors.dark .. '; background-color: #' ..
		colors.light .. '; border-radius: 15px;"><span style="margin-bottom: 7px; font-weight: bold; color:#' ..
		colors.dark .. ';">' .. displayName .. '</span></td></tr>'

	-- Artwork.
	buf[#buf + 1] = '<tr><td colspan="6" style="border: 1px solid #' .. colors.dark .. '; background-color: #' ..
		colors.light .. '; border-radius: 15px; padding: 5px;"><div style="border-radius: 15px; ' ..
		'text-align: center; padding: 5px; background-color: #FFFFFF;">[[File:' .. data.number .. data.name ..
		'.png|150px]]</div></td></tr>'

	-- Types + Generation.
	local genHtml = '—'
	if gen then
		genHtml = '[[:Category:Generation ' .. gen .. ' Pokémon|<span style="color:#' ..
			colors.dark .. ';">' .. gen .. '</span>]]'
	end
	buf[#buf + 1] = '<tr><td colspan="3" style="width: 50%; border: 1px solid #' .. colors.dark .. '; background-color: #' ..
		colors.light .. '; padding: 11px; border-radius: 15px;"><p style="margin-bottom: 7px; font-weight: bold; ' ..
		'color: #' .. colors.dark .. ';">Types</p><table align="center" style="width: 75%; font-weight: bold; ' ..
		'padding: 3px;"><tr>' .. typesHtml .. '</tr></table></td>' ..
		'<td colspan="3" style="width: 50%; border: 1px solid #' .. colors.dark .. '; background-color: #' .. colors.light ..
		'; border-radius: 15px;"><p style="margin-bottom: 7px; font-weight: bold; color: #' .. colors.dark ..
		';">Generation</p><span style="border-radius: 12px; padding: 4px 10px; border: 1px solid #' .. colors.dark ..
		'; background-color: #FFFFFF;">&nbsp;&nbsp;' .. genHtml .. '&nbsp;&nbsp;</span></td></tr>'

	-- Abilities.
	buf[#buf + 1] = '<tr><td colspan="6" style="border: 1px solid #' .. colors.dark .. '; background-color: #' ..
		colors.light .. '; padding: 11px; border-radius: 15px; width: 100%;"><table width="100%">' ..
		'<tr><th colspan="6"><p style="margin-bottom: 7px; font-weight: bold; color:#' .. colors.dark ..
		';">[[Abilities|<span style="color:#' .. colors.dark .. ';">Abilities</span>]]</p></th></tr>' ..
		'<tr>' .. abilitiesRow(colors, data) .. '</tr></table></td></tr>'

	-- Rarity + Catch rate.
	buf[#buf + 1] = '<tr>' ..
		infoCell(colors, '[[List of Pokémon by Rarity Tier|<span style="color:#' .. colors.dark ..
			';">Rarity Tier</span>]]', '[[:Category:' .. data.rarity .. ' Pokémon|<span style="color:#' ..
			colors.dark .. ';">' .. data.rarity .. '</span>]]') ..
		infoCell(colors, '[[Catch Rates|<span style="color:#' .. colors.dark .. ';">Catch rate</span>]]',
			data.catch .. '/255 (' .. catchPercent .. '%)') ..
		'</tr>'

	-- Base stats.
	buf[#buf + 1] = '<tr><td colspan="6" style="border: 1px solid #' .. colors.dark .. '; background-color: #' ..
		colors.light .. '; border-radius: 15px; padding: 8px; width: 100%">' ..
		'<table width="100%" style="border-collapse: collapse;">' ..
		'<tr><th colspan="3"><p style="margin-bottom: 7px; font-weight: bold; color: #' .. colors.dark ..
		';">[[Base stats|<span style="color: #' .. colors.dark .. ';">Base stats</span>]]</p></th></tr>' ..
		table.concat(statRows) ..
		'<tr><td colspan="3" style="text-align: center; padding-top: 6px;">' ..
		'<div style="display: inline-block; background-color: #FFFFFF; border: 1px solid #' .. colors.dark ..
		'; padding: 4px 14px; border-radius: 12px;">Base-stat total: ' .. bst .. '</div></td></tr>' ..
		'</table></td></tr>'

	-- EV yield.
	buf[#buf + 1] = '<tr><td colspan="6" style="border: 1px solid #' .. colors.dark .. '; background-color: #' ..
		colors.light .. '; border-radius: 15px; padding: 5px; width: 100%"><table width="100%">' ..
		'<tr><th colspan="6">[[EVs|<span style="margin-bottom: 7px; font-weight: bold; color: #' .. colors.dark ..
		';">EV Yield</span>]]</th></tr><tr>' .. table.concat(evCells) .. '</tr>' ..
		'<tr><td colspan="6"><div style="margin: auto; background-color:#FFFFFF; border: 1px solid #' ..
		colors.dark .. '; padding: 4px 14px; border-radius: 12px; margin-bottom: 5px; width: 50%;">' ..
		'EV-yield total: ' .. evTotal .. '</div></td></tr></table></td></tr>'

	-- Base experience + Gender ratio.
	buf[#buf + 1] = '<tr>' ..
		infoCell(colors, '[[Experience System#Base Experience|<span style="color:#' .. colors.dark ..
			';">Base Experience</span>]]', data.exp) ..
		'<td colspan="3" style="width: 50%; border: 1px solid #' .. colors.dark .. '; background-color: #' .. colors.light ..
		'; border-radius: 15px; padding: 5px;"><p style="margin-bottom: 7px; font-weight: bold; color: #' ..
		colors.dark .. ';">Gender Ratio</p>' .. genderRatioHtml(data.male) .. '</td>' ..
		'</tr>'

	-- Height + Weight.
	buf[#buf + 1] = '<tr>' ..
		infoCell(colors, 'Height', data.height .. 'm') ..
		infoCell(colors, 'Weight', data.weight .. 'kg') ..
		'</tr>'

	buf[#buf + 1] = '</table>'
	buf[#buf + 1] = categories(data, gen, bstCategory, args)

	return table.concat(buf)
end

-- Infobox for a Mega evolution row from RawMegaPokemonList.csv: named
-- "Mega <name>", artwork <number><name>-Mega.png, a Mega Stone row, and no
-- nav/catch/rarity/gender/experience sections.
local function megaInfoboxHtml(frame, data, width)
	local colors = typeColors(frame, data.type1)
	local gen = GEN_NUMERALS[tonumber(mw.ustring.match(tostring(data.generation or ''), '^%d+') or '')]
	local displayName = 'Mega ' .. (data.name or '')

	local typesHtml
	if nonEmpty(data.type2) then
		local type2Colors = typeColors(frame, data.type2)
		typesHtml =
			'<td style="width: 50%; background-color: #' .. colors.dark .. '; border-radius: 25px 1px 1px 25px;">' ..
			'[[' .. data.type1 .. '_(type)|<span style="color: #FFFFFF;">' .. data.type1 .. '</span>]]</td>' ..
			'<td style="width: 50%; background-color: #' .. type2Colors.dark .. '; border-radius: 1px 25px 25px 1px;">' ..
			'[[' .. data.type2 .. '_(type)|<span style="color: #FFFFFF;">' .. data.type2 .. '</span>]]</td>'
	else
		typesHtml =
			'<td style="width:100%; background-color: #' .. colors.dark .. '; border-radius: 25px;">' ..
			'[[' .. data.type1 .. '_(type)|<span style="color: #FFFFFF;">' .. data.type1 .. '</span>]]</td>'
	end

	local statRows, bst = {}, 0
	for _, stat in ipairs(STATS) do
		local value = tonumber(data[stat.key]) or 0
		bst = bst + value
		statRows[#statRows + 1] = statBarRow(frame, stat, value)
	end

	local evCells, evTotal = {}, 0
	for i, stat in ipairs(STATS) do
		local ev = tonumber(data[EV_KEYS[i]]) or 0
		evTotal = evTotal + ev
		evCells[#evCells + 1] = statValueCell(frame, stat, ev)
	end

	local buf = {}
	buf[#buf + 1] = '<table align="right" class="informational-box" cellpadding="4" cellspacing="4" ' ..
		'style="width: ' .. width .. '%; border: 4px solid #' .. colors.dark .. '; background-color: #' ..
		colors.base .. '; padding: 4px; border-radius: 15px;">'

	-- Header: name across the whole top line.
	buf[#buf + 1] = '<tr><td colspan="6" style="border: 1px solid #' .. colors.dark .. '; background-color: #' ..
		colors.light .. '; border-radius: 15px;"><span style="margin-bottom: 7px; font-weight: bold; color:#' ..
		colors.dark .. ';">' .. displayName .. '</span></td></tr>'

	-- Artwork.
	buf[#buf + 1] = '<tr><td colspan="6" style="border: 1px solid #' .. colors.dark .. '; background-color: #' ..
		colors.light .. '; border-radius: 15px; padding: 5px;"><div style="border-radius: 15px; ' ..
		'text-align: center; padding: 5px; background-color: #FFFFFF;">[[File:' .. (data.number or '') .. (data.name or '') ..
		'-Mega.png|150px]]</div></td></tr>'

	-- Types + Generation.
	local genHtml = '—'
	if gen then
		genHtml = '[[:Category:Generation ' .. gen .. ' Pokémon|<span style="color:#' ..
			colors.dark .. ';">' .. gen .. '</span>]]'
	end
	buf[#buf + 1] = '<tr><td colspan="3" style="width: 50%; border: 1px solid #' .. colors.dark .. '; background-color: #' ..
		colors.light .. '; padding: 11px; border-radius: 15px;"><p style="margin-bottom: 7px; font-weight: bold; ' ..
		'color: #' .. colors.dark .. ';">Types</p><table align="center" style="width: 75%; font-weight: bold; ' ..
		'padding: 3px;"><tr>' .. typesHtml .. '</tr></table></td>' ..
		'<td colspan="3" style="width: 50%; border: 1px solid #' .. colors.dark .. '; background-color: #' .. colors.light ..
		'; border-radius: 15px;"><p style="margin-bottom: 7px; font-weight: bold; color: #' .. colors.dark ..
		';">Generation</p><span style="border-radius: 12px; padding: 4px 10px; border: 1px solid #' .. colors.dark ..
		'; background-color: #FFFFFF;">&nbsp;&nbsp;' .. genHtml .. '&nbsp;&nbsp;</span></td></tr>'

	-- Abilities.
	buf[#buf + 1] = '<tr><td colspan="6" style="border: 1px solid #' .. colors.dark .. '; background-color: #' ..
		colors.light .. '; padding: 11px; border-radius: 15px; width: 100%;"><table width="100%">' ..
		'<tr><th colspan="6"><p style="margin-bottom: 7px; font-weight: bold; color:#' .. colors.dark ..
		';">[[Abilities|<span style="color:#' .. colors.dark .. ';">Abilities</span>]]</p></th></tr>' ..
		'<tr>' .. abilitiesRow(colors, data) .. '</tr></table></td></tr>'

	-- Mega Stone, linked to its quest page ("Venusaurite (quest)").
	local stone = nonEmpty(data.stone)
	local stoneHtml = '—'
	if stone then
		stoneHtml = '[[' .. stone .. ' (quest)|<span style="color:#' .. colors.dark .. ';">' .. stone .. '</span>]]'
	end
	buf[#buf + 1] = '<tr><td colspan="6" style="border: 1px solid #' .. colors.dark .. '; background-color: #' ..
		colors.light .. '; border-radius: 15px; padding: 5px;">' ..
		'<p style="margin-bottom: 7px; font-weight: bold; color: #' .. colors.dark .. ';">Mega Stone</p>' ..
		'<span style="border-radius: 12px; padding: 4px 10px; border: 1px solid #' .. colors.dark ..
		'; background-color: #FFFFFF;">&nbsp;&nbsp;' .. stoneHtml .. '&nbsp;&nbsp;</span></td></tr>'

	-- Base stats.
	buf[#buf + 1] = '<tr><td colspan="6" style="border: 1px solid #' .. colors.dark .. '; background-color: #' ..
		colors.light .. '; border-radius: 15px; padding: 8px; width: 100%">' ..
		'<table width="100%" style="border-collapse: collapse;">' ..
		'<tr><th colspan="3"><p style="margin-bottom: 7px; font-weight: bold; color: #' .. colors.dark ..
		';">[[Base stats|<span style="color: #' .. colors.dark .. ';">Base stats</span>]]</p></th></tr>' ..
		table.concat(statRows) ..
		'<tr><td colspan="3" style="text-align: center; padding-top: 6px;">' ..
		'<div style="display: inline-block; background-color: #FFFFFF; border: 1px solid #' .. colors.dark ..
		'; padding: 4px 14px; border-radius: 12px;">Base-stat total: ' .. bst .. '</div></td></tr>' ..
		'</table></td></tr>'

	-- EV yield.
	buf[#buf + 1] = '<tr><td colspan="6" style="border: 1px solid #' .. colors.dark .. '; background-color: #' ..
		colors.light .. '; border-radius: 15px; padding: 5px; width: 100%"><table width="100%">' ..
		'<tr><th colspan="6">[[EVs|<span style="margin-bottom: 7px; font-weight: bold; color: #' .. colors.dark ..
		';">EV Yield</span>]]</th></tr><tr>' .. table.concat(evCells) .. '</tr>' ..
		'<tr><td colspan="6"><div style="margin: auto; background-color:#FFFFFF; border: 1px solid #' ..
		colors.dark .. '; padding: 4px 14px; border-radius: 12px; margin-bottom: 5px; width: 50%;">' ..
		'EV-yield total: ' .. evTotal .. '</div></td></tr></table></td></tr>'

	-- Height + Weight.
	buf[#buf + 1] = '<tr>' ..
		infoCell(colors, 'Height', (data.height or '—') .. 'm') ..
		infoCell(colors, 'Weight', (data.weight or '—') .. 'kg') ..
		'</tr>'

	buf[#buf + 1] = '</table>'
	buf[#buf + 1] = '[[Category:Mega-Evolvable Pokémon]]'

	return table.concat(buf)
end

function p.infobox(frame)
	local args = templateArgs(frame)
	local pageName = mw.title.getCurrentTitle().text
	local name = nonEmpty(args.Name) or pageName
	local width = nonEmpty(args.Width) or '45'

	local data = byName(name)
	if not data then
		return '<strong class="error">PokemonInfo: no data found for "' .. name .. '"</strong>'
	end

	-- Downstream templates ({{Tabs:MultiformTab}} etc.) color themselves via
	-- {{#var:type}}, which the old wikitext infobox used to define.
	frame:callParserFunction('#vardefine', 'type', data.type1 or 'Normal')

	return infoboxHtml(frame, args, data, name, width, pageName, true)
end

-- Returns the page Pokémon's primary type as plain text, for use in template
-- wikitext: {{#vardefine:type|{{#invoke:Pokemon|primaryType}}}}. Variables
-- defined from inside Lua do not persist on this wiki, so the template layer
-- has to do the #vardefine itself.
function p.primaryType(frame)
	local args = templateArgs(frame)
	local name = nonEmpty(args.Name) or mw.title.getCurrentTitle().text
	local data = byName(name) or byName(formStem(name))
	return (data and nonEmpty(data.type1)) or 'Normal'
end

-- Name/index plumbing for the template-side tab shell: the index arrives as
-- the invoke's own first argument, while Name comes from the surrounding
-- template call.
local function requestedName(frame)
	local parent = frame:getParent()
	return nonEmpty(frame.args.Name)
		or (parent and nonEmpty(parent.args.Name))
		or mw.title.getCurrentTitle().text
end

local participantsCache = {}

-- All forms of the page's Pokémon, base first. parts[1].label is 'Normal'
-- unless a matched form dictates a baseLabel ('Male').
local function computeParticipants(frame, name)
	if participantsCache[name] ~= nil then
		return participantsCache[name] or nil
	end
	local rows = loadAllRows()
	local function rowFor(n)
		return rows[n] or rows[hyphenSpaceVariant(n)]
	end
	local stem = formStem(name)
	local baseData = rowFor(stem) or rowFor(name)
	if not baseData then
		participantsCache[name] = false
		return nil
	end
	local parts = { { data = baseData, label = 'Normal' } }
	local seen = { [baseData.name] = true }
	for _, form in ipairs(FORM_SUFFIXES) do
		local row = rowFor(rawListFormName(form, stem))
		if row and not seen[row.name] then
			seen[row.name] = true
			parts[#parts + 1] = { data = row, label = form.label }
			if form.baseLabel then
				parts[1].label = form.baseLabel
			end
		end
	end
	for _, variant in ipairs(MEGA_VARIANTS) do
		local row = megaByName(stem .. variant)
		if row then
			parts[#parts + 1] = { data = row, label = 'Mega' .. variant, mega = true }
		end
	end
	participantsCache[name] = parts
	return parts
end

-- {{#invoke:Pokemon|formLabel|N}}: tab label of the Nth participant
-- (0 = base form, 1.. = extra forms); empty when there is no such form or the
-- Pokémon has no extra forms at all.
function p.formLabel(frame)
	local idx = tonumber(frame.args[1]) or 0
	local parts = computeParticipants(frame, requestedName(frame))
	if not parts then
		return ''
	end
	local part = parts[idx + 1]
	return part and part.label or ''
end

-- {{#invoke:Pokemon|rawName}}: the page Pokémon's raw-list name, for the
-- template-side {{#get_file_data:}} filter that feeds downstream
-- {{#external_value:}} consumers.
function p.rawName(frame)
	local name = requestedName(frame)
	local parts = computeParticipants(frame, name)
	return (parts and parts[1].data.name) or name
end

-- {{#invoke:Pokemon|formBox|N}}: the Nth participant's infobox HTML
-- (0 = base form), without the nav table. Width defaults to 100 (inside a
-- tab cell); pass Width=45 for the untabbed fallback.
function p.formBox(frame)
	local idx = tonumber(frame.args[1]) or 0
	local width = nonEmpty(frame.args.Width) or '100'
	local name = requestedName(frame)
	local parts = computeParticipants(frame, name)
	if not parts then
		return '<strong class="error">PokemonInfo: no data found for "' .. name .. '"</strong>'
	end
	local part = parts[idx + 1]
	if not part then
		return ''
	end
	if part.mega then
		return megaInfoboxHtml(frame, part.data, width)
	end
	local parent = frame:getParent()
	local args = (parent and parent.args) or frame.args
	return infoboxHtml(frame, args, part.data, part.data.name, width, mw.title.getCurrentTitle().text, false)
end

-- Temporary diagnostic: {{#invoke:Pokemon|debugForms|Name=Meowth}}
function p.debugForms(frame)
	local args = templateArgs(frame)
	local name = nonEmpty(args.Name) or mw.title.getCurrentTitle().text
	local rows = loadAllRows()
	local count = 0
	for _ in pairs(rows) do count = count + 1 end
	local stem = formStem(name)
	local out = { 'name=' .. name, 'stem=' .. stem, 'rows loaded=' .. count }
	for _, form in ipairs(FORM_SUFFIXES) do
		local candidate = rawListFormName(form, stem)
		if rows[candidate] or rows[hyphenSpaceVariant(candidate)] then
			out[#out + 1] = 'form found: ' .. form.label .. ' -> ' .. candidate
		end
	end
	for _, variant in ipairs(MEGA_VARIANTS) do
		if megaByName(stem .. variant) then
			out[#out + 1] = 'mega found: Mega' .. variant .. ' -> ' .. stem .. variant
		end
	end
	return table.concat(out, '; ')
end

-- Like infobox, but detects alternate forms via Module:PokemonForms and, when
-- any exist, renders one nav table plus a {{TabbedInfoBoxes}} block with one
-- infobox per form. Falls back to the plain single infobox otherwise.
function p.infoboxes(frame)
	local args = templateArgs(frame)
	local pageName = mw.title.getCurrentTitle().text
	local name = nonEmpty(args.Name) or pageName

	local rows = loadAllRows()
	local function rowFor(n)
		return rows[n] or rows[hyphenSpaceVariant(n)]
	end

	local stem = formStem(name)
	local baseData = rowFor(stem) or rowFor(name)
	if not baseData then
		return '<strong class="error">PokemonInfo: no data found for "' .. name .. '"</strong>'
	end
	local baseName = baseData.name

	-- Downstream templates ({{Tabs:MultiformTab}} etc.) color themselves via
	-- {{#var:type}}, which the old wikitext infobox used to define.
	frame:callParserFunction('#vardefine', 'type', baseData.type1 or 'Normal')

	local forms = {}
	local seen = { [baseName] = true }
	for _, form in ipairs(FORM_SUFFIXES) do
		local row = rowFor(rawListFormName(form, stem))
		if row and not seen[row.name] then
			seen[row.name] = true
			forms[#forms + 1] = { label = form.label, data = row, baseLabel = form.baseLabel }
		end
	end

	if #forms == 0 then
		local width = nonEmpty(args.Width) or '45'
		return infoboxHtml(frame, args, baseData, baseName, width, pageName, true)
	end

	local baseLabel = 'Normal'
	for _, f in ipairs(forms) do
		if f.baseLabel then
			baseLabel = f.baseLabel
			break
		end
	end

	-- Native tab shell replicating {{TabbedInfoBoxes}} + {{Tabs:MultiformTab}} +
	-- {{Tabs:MultiformTable}} without Extension:Variables, whose page variables
	-- are not visible inside Scribunto expansions on this wiki. The onclick div
	-- and the name-attributed td must go through {{#tag:html}}, as the original
	-- templates do.
	local shellColors = typeColors(frame, baseData.type1)
	local shellWidth = nonEmpty(args.Width) or '45'

	-- The {{#tag:html}} bits live in Template:PokemonTabHeader and
	-- Template:PokemonTabContent: Extension:NamespaceHTML fatals when the tag
	-- reaches it directly from Scribunto (callParserFunction or a top-level
	-- preprocess string), but survives inside a transcluded template body —
	-- the path {{Tabs:MultiformTab}} has always used.
	local tabHeaders, boxes = {}, {}
	local function addTab(tabName, label, boxHtml)
		tabHeaders[#tabHeaders + 1] = frame:expandTemplate{
			title = 'PokemonTabHeader',
			args = { Tab = tabName, Label = label, Dark = shellColors.dark, Light = shellColors.light },
		}
		boxes[#boxes + 1] = frame:expandTemplate{
			title = 'PokemonTabContent',
			args = { Tab = tabName, Content = boxHtml },
		}
	end

	addTab('Default', baseLabel, infoboxHtml(frame, args, baseData, baseName, '100', pageName, false))
	for _, f in ipairs(forms) do
		addTab(f.label, f.label, infoboxHtml(frame, args, f.data, f.data.name, '100', pageName, false))
	end

	return navTableHtml(frame, baseData, shellColors, pageName) ..
		'<table class="informational-box" style="width: ' .. shellWidth .. '%;">' ..
		'<tr class="tab" style="color: #' .. shellColors.dark .. ';">' .. table.concat(tabHeaders) .. '</tr>' ..
		'<tr>' .. table.concat(boxes) .. '</tr></table>'
end

return p