Module:Learnsets
Documentation for this module may be created at Module:Learnsets/doc
local p = {}
local function getArgs(frame)
local args = {}
local parent = frame:getParent()
if parent then
for k, v in pairs(parent.args) do
if v ~= '' then
args[k] = v
end
end
end
for k, v in pairs(frame.args) do
if v ~= '' then
args[k] = v
end
end
return args
end
local moveDataCache = nil
local function blankToNil(v)
if v == nil or v == '' then
return nil
end
return v
end
local function loadMoveData()
if moveDataCache then
return moveDataCache
end
local csvData = mw.ext.externalData.getExternalData({
url = 'https://wiki.pokemonrevolution.net/index.php?title=Special:GetData/RawMovesList',
format = 'CSV with header',
data = {
Name = 'Name',
Type = 'Type',
Category = 'Category',
Power = 'Power',
Accuracy = 'Accuracy',
PP = 'PP',
Broken = 'Broken',
Typeless = 'Typeless',
Priority = 'Priority',
TM = 'TM',
['Move Tutor'] = 'Move Tutor',
Egg = 'Egg',
},
})
-- getExternalData returns an array of row-tables here (csvData[i].Name),
-- not column-arrays (csvData.Name[i]).
moveDataCache = {}
if csvData then
for _, row in ipairs(csvData) do
moveDataCache[row.Name] = {
type = blankToNil(row.Type) or 'Normal',
category = blankToNil(row.Category) or 'Status',
power = blankToNil(row.Power),
accuracy = blankToNil(row.Accuracy),
pp = blankToNil(row.PP),
broken = row.Broken or 'N',
typeless = row.Typeless or 'N',
priority = blankToNil(row.Priority) or '0',
tm = blankToNil(row.TM),
move_tutor = blankToNil(row['Move Tutor']),
egg = blankToNil(row.Egg)
}
end
end
return moveDataCache
end
local function loadLearnset(pokemonName)
local jsonPath = '$[' .. mw.text.jsonEncode(pokemonName) .. ']'
local ok, result, errors = pcall(function()
return mw.ext.externalData.getFileData({
source = 'data',
['file name'] = 'Learnsets.json',
format = 'JSON',
['use jsonpath'] = true,
data = { poke = jsonPath },
})
end)
if not ok then
return nil, 'Lua error: ' .. tostring(result)
end
if errors and #errors > 0 then
return nil, 'ExternalData error: ' .. table.concat(errors, ' | ')
end
local raw = result and result.poke
if type(raw) ~= 'table' then
return nil, 'No learnset data found for ' .. pokemonName
end
return raw, nil
end
local learnsetCache = {}
local learnsetErrorCache = {}
local function getCachedLearnset(pokemonName)
if learnsetCache[pokemonName] == nil and learnsetErrorCache[pokemonName] == nil then
learnsetCache[pokemonName], learnsetErrorCache[pokemonName] = loadLearnset(pokemonName)
end
return learnsetCache[pokemonName], learnsetErrorCache[pokemonName]
end
-- Loads Type1/Type2 for every Pokemon from the PokemonRawList CSV.
local pokemonTypeCache = nil
local function loadPokemonTypes()
if pokemonTypeCache then return pokemonTypeCache end
local csvData = mw.ext.externalData.getExternalData({
url = 'https://wiki.pokemonrevolution.net/index.php?title=Special:GetData/PokemonRawList',
format = 'CSV with header',
data = {
Name = 'Name',
Type1 = 'Type1',
Type2 = 'Type2',
},
})
-- Same row-table shape as loadMoveData: csvData[i].Name, not csvData.Name[i].
pokemonTypeCache = {}
if csvData then
for _, row in ipairs(csvData) do
pokemonTypeCache[row.Name] = {
type1 = row.Type1 or '',
type2 = row.Type2 or '',
}
end
end
return pokemonTypeCache
end
-- Colors come from the wiki's own "{Type} color[ light| dark]" and
-- "{Category} color[ dark]" templates -- the same source Template:M and
-- MoveTableLevelUp use -- so this matches the site's palette instead of
-- guessing hex values. Cached per type/category since the same ones repeat
-- across many move rows on a page.
local typeColorCache = {}
local function typeColor(frame, pokeType)
pokeType = (pokeType and pokeType ~= '') and pokeType or 'Normal'
if not typeColorCache[pokeType] then
typeColorCache[pokeType] = {
light = frame:expandTemplate{ title = pokeType .. ' color light' },
base = frame:expandTemplate{ title = pokeType .. ' color' },
dark = frame:expandTemplate{ title = pokeType .. ' color dark' },
}
end
return typeColorCache[pokeType]
end
local categoryColorCache = {}
local function categoryColor(frame, category)
category = (category and category ~= '') and category or 'Status'
if not categoryColorCache[category] then
categoryColorCache[category] = {
base = frame:expandTemplate{ title = category .. ' color' },
dark = frame:expandTemplate{ title = category .. ' color dark' },
}
end
return categoryColorCache[category]
end
local function isSTAB(moveType, primaryType, secondaryType)
return moveType == primaryType or moveType == secondaryType
end
-- Formats the TM/HM compatibility number as e.g. "TM01" / "HM04",
-- zero-padded to 2 digits (numbers >= 100 are left as-is).
local function formatItemNumber(label, num)
num = tonumber(num)
if not num then
return '—'
end
return label .. string.format('%02d', num)
end
function p.renderLevelUp(frame)
local args = getArgs(frame)
local pokemonName = args[1] or args.pokemon
local primaryType = args.type1 or ""
local secondaryType = args.type2 or ""
local pokemon, lsErr = getCachedLearnset(pokemonName)
local moveData = loadMoveData()
if not pokemon then
return "<span style='color:red'>Data load error: " .. (lsErr or 'unknown') .. "</span>"
end
local moves = pokemon.level_up or {}
local colors = typeColor(frame, primaryType)
local html = mw.html.create('div')
local table = html:tag('table')
:css('width', '77%')
:css('text-align', 'center')
:css('margin', 'auto')
:css('border-radius', '15px')
:css('border', '3px solid #' .. colors.dark)
:css('background-color', '#' .. colors.light)
:css('padding', '7px')
local header = table:tag('tr')
:css('background-color', '#' .. colors.base)
:css('color', '#' .. colors.dark)
header:tag('th'):wikitext('Level'):css('width', '10%'):css('border-radius', '15px 1px 1px 1px')
header:tag('th'):wikitext('Move'):css('width', '15%')
header:tag('th'):wikitext('Type'):css('width', '17%')
header:tag('th'):wikitext('Category'):css('width', '16%')
header:tag('th'):wikitext('Base Power'):css('width', '17%')
header:tag('th'):wikitext('Accuracy'):css('width', '15%')
header:tag('th'):wikitext('PP'):css('width', '10%'):css('border-radius', '1px 15px 1px 1px')
for _, move in ipairs(moves) do
local moveName = move.move
local level = move.level
local moveInfo = moveData[moveName] or {}
local row = table:tag('tr'):css('background-color', '#FFFFFF')
row:tag('td'):wikitext(tostring(level))
local moveLink = '[[' .. moveName .. ']]'
if isSTAB(moveInfo.type, primaryType, secondaryType) and moveInfo.category ~= "Status" then
moveLink = "'''" .. moveLink .. "'''"
end
if moveInfo.broken == "Y" then
moveLink = "''" .. moveLink .. "''"
end
row:tag('td'):wikitext(moveLink)
local moveTypeColors = typeColor(frame, moveInfo.type)
row:tag('td')
:css('background-color', '#' .. moveTypeColors.base)
:css('border', '2px solid #' .. moveTypeColors.dark)
:wikitext('[[' .. (moveInfo.type or 'Normal') .. '_(type)|<span style="color:#FFFFFF;">' .. (moveInfo.type or 'Normal') .. '</span>]]')
local moveCatColors = categoryColor(frame, moveInfo.category)
row:tag('td')
:css('background-color', '#' .. moveCatColors.base)
:css('border', '2px solid #' .. moveCatColors.dark)
:wikitext('[[' .. (moveInfo.category or 'Status') .. ' moves|<span style="color:#FFFFFF;">' .. (moveInfo.category or 'Status') .. '</span>]]')
row:tag('td'):wikitext(moveInfo.power or '—')
row:tag('td'):wikitext(moveInfo.accuracy or '—')
row:tag('td'):wikitext(moveInfo.pp or '—')
end
table:tag('tr'):tag('td'):attr('colspan', '7')
:css('text-align', 'left')
:css('background-color', '#' .. colors.base)
:css('padding', '10px')
:css('border-radius', '1px 1px 25px 25px')
:wikitext([=[
* '''Emboldened''' moves denote this Pokémon avails from a STAB bonus for it.
* ''Italicized'' moves are [[Broken_Moves_%26_Abilities|broken]]; see their individual pages for more information.
* All moves subsumed on this list are relearnable via the [[Move Relearner]]s if lost.
]=])
return tostring(html) .. '<br clear="all">'
end
function p.renderTMsHMs(frame)
local args = getArgs(frame)
local pokemonName = args[1] or args.pokemon
local primaryType = args.type1 or ""
local secondaryType = args.type2 or ""
local pokemon = getCachedLearnset(pokemonName)
local moveData = loadMoveData()
if not pokemon then return '' end
local tms = pokemon.tms or {}
local hms = pokemon.hms or {}
local colors = typeColor(frame, primaryType)
local html = mw.html.create('div')
local table = html:tag('table')
:css('width', '77%')
:css('text-align', 'center')
:css('margin', 'auto')
:css('border-radius', '25px')
:css('border', '3px solid #' .. colors.dark)
:css('background-color', '#' .. colors.light)
:css('padding', '7px')
local header = table:tag('tr')
:css('background-color', '#' .. colors.base)
:css('color', '#' .. colors.dark)
header:tag('th'):attr('colspan', '2'):wikitext('TM'):css('width', '10%'):css('border-top-left-radius', '25px')
header:tag('th'):wikitext('Move'):css('width', '15%')
header:tag('th'):wikitext('Type'):css('width', '17%')
header:tag('th'):wikitext('Category'):css('width', '16%')
header:tag('th'):wikitext('Base Power'):css('width', '17%')
header:tag('th'):wikitext('Accuracy'):css('width', '15%')
header:tag('th'):wikitext('PP'):css('width', '10%'):css('border-top-right-radius', '25px')
for _, moveName in ipairs(tms) do
local moveInfo = moveData[moveName] or {}
local row = table:tag('tr'):css('background-color', '#FFFFFF')
row:tag('td'):attr('colspan', '2'):wikitext(formatItemNumber('TM', moveInfo.tm))
local moveLink = '[[' .. moveName .. ']]'
if isSTAB(moveInfo.type, primaryType, secondaryType) and moveInfo.category ~= "Status" then
moveLink = "'''" .. moveLink .. "'''"
end
if moveInfo.broken == "Y" then
moveLink = "''" .. moveLink .. "''"
end
row:tag('td'):wikitext(moveLink)
local moveTypeColors = typeColor(frame, moveInfo.type)
row:tag('td')
:css('background-color', '#' .. moveTypeColors.base)
:css('border', '2px solid #' .. moveTypeColors.dark)
:wikitext('[[' .. (moveInfo.type or 'Normal') .. '_(type)|<span style="color:#FFFFFF;">' .. (moveInfo.type or 'Normal') .. '</span>]]')
local moveCatColors = categoryColor(frame, moveInfo.category)
row:tag('td')
:css('background-color', '#' .. moveCatColors.base)
:css('border', '2px solid #' .. moveCatColors.dark)
:wikitext('[[' .. (moveInfo.category or 'Status') .. ' moves|<span style="color:#FFFFFF;">' .. (moveInfo.category or 'Status') .. '</span>]]')
row:tag('td'):wikitext(moveInfo.power or '—')
row:tag('td'):wikitext(moveInfo.accuracy or '—')
row:tag('td'):wikitext(moveInfo.pp or '—')
end
for _, moveName in ipairs(hms) do
local moveInfo = moveData[moveName] or {}
local row = table:tag('tr'):css('background-color', '#FFFFCC')
row:tag('td'):attr('colspan', '2'):wikitext(formatItemNumber('HM', moveInfo.tm)):css('font-weight', 'bold')
local moveLink = '[[' .. moveName .. ']]'
if isSTAB(moveInfo.type, primaryType, secondaryType) and moveInfo.category ~= "Status" then
moveLink = "'''" .. moveLink .. "'''"
end
if moveInfo.broken == "Y" then
moveLink = "''" .. moveLink .. "''"
end
row:tag('td'):wikitext(moveLink)
local moveTypeColors = typeColor(frame, moveInfo.type)
row:tag('td')
:css('background-color', '#' .. moveTypeColors.base)
:css('border', '2px solid #' .. moveTypeColors.dark)
:wikitext('[[' .. (moveInfo.type or 'Normal') .. '_(type)|<span style="color:#FFFFFF;">' .. (moveInfo.type or 'Normal') .. '</span>]]')
local moveCatColors = categoryColor(frame, moveInfo.category)
row:tag('td')
:css('background-color', '#' .. moveCatColors.base)
:css('border', '2px solid #' .. moveCatColors.dark)
:wikitext('[[' .. (moveInfo.category or 'Status') .. ' moves|<span style="color:#FFFFFF;">' .. (moveInfo.category or 'Status') .. '</span>]]')
row:tag('td'):wikitext(moveInfo.power or '—')
row:tag('td'):wikitext(moveInfo.accuracy or '—')
row:tag('td'):wikitext(moveInfo.pp or '—')
end
table:tag('tr'):tag('td'):attr('colspan', '8')
:css('text-align', 'left')
:css('background-color', '#' .. colors.base)
:css('padding', '10px')
:css('border-radius', '1px 1px 25px 25px')
:wikitext([=[
* '''Emboldened''' moves denote that this Pokémon avails from a STAB bonus for it.
* ''Italicized'' moves are [[Broken_Moves_%26_Abilities|broken]]; see their individual pages for more information.
* Refer to [[TMs and HMs]] for information on each move's obtainability.
]=])
return tostring(html) .. '<br clear="all">'
end
function p.renderMoveTutors(frame)
local args = getArgs(frame)
local pokemonName = args[1] or args.pokemon
local primaryType = args.type1 or ""
local secondaryType = args.type2 or ""
local pokemon = getCachedLearnset(pokemonName)
local moveData = loadMoveData()
if not pokemon then return '' end
local moves = pokemon.move_tutors or {}
if #moves == 0 then
return ""
end
local colors = typeColor(frame, primaryType)
local html = mw.html.create('div')
local table = html:tag('table')
:css('width', '77%')
:css('text-align', 'center')
:css('margin', 'auto')
:css('border-radius', '25px')
:css('border', '3px solid #' .. colors.dark)
:css('background-color', '#' .. colors.light)
:css('padding', '7px')
local header = table:tag('tr')
:css('background-color', '#' .. colors.base)
:css('color', '#' .. colors.dark)
header:tag('th'):wikitext('Move'):css('width', '20%'):css('border-top-left-radius', '25px')
header:tag('th'):wikitext('Type'):css('width', '17%')
header:tag('th'):wikitext('Category'):css('width', '16%')
header:tag('th'):wikitext('Base Power'):css('width', '17%')
header:tag('th'):wikitext('Accuracy'):css('width', '15%')
header:tag('th'):wikitext('PP'):css('width', '15%'):css('border-top-right-radius', '25px')
for _, moveName in ipairs(moves) do
local moveInfo = moveData[moveName] or {}
local row = table:tag('tr'):css('background-color', '#FFFFFF')
local moveLink = '[[' .. moveName .. ']]'
if isSTAB(moveInfo.type, primaryType, secondaryType) and moveInfo.category ~= "Status" then
moveLink = "'''" .. moveLink .. "'''"
end
if moveInfo.broken == "Y" then
moveLink = "''" .. moveLink .. "''"
end
row:tag('td'):wikitext(moveLink)
local moveTypeColors = typeColor(frame, moveInfo.type)
row:tag('td')
:css('background-color', '#' .. moveTypeColors.base)
:css('border', '2px solid #' .. moveTypeColors.dark)
:wikitext('[[' .. (moveInfo.type or 'Normal') .. '_(type)|<span style="color:#FFFFFF;">' .. (moveInfo.type or 'Normal') .. '</span>]]')
local moveCatColors = categoryColor(frame, moveInfo.category)
row:tag('td')
:css('background-color', '#' .. moveCatColors.base)
:css('border', '2px solid #' .. moveCatColors.dark)
:wikitext('[[' .. (moveInfo.category or 'Status') .. ' moves|<span style="color:#FFFFFF;">' .. (moveInfo.category or 'Status') .. '</span>]]')
row:tag('td'):wikitext(moveInfo.power or '—')
row:tag('td'):wikitext(moveInfo.accuracy or '—')
row:tag('td'):wikitext(moveInfo.pp or '—')
end
return tostring(html) .. '<br clear="all">'
end
function p.renderEggMoves(frame)
local args = getArgs(frame)
local pokemonName = args[1] or args.pokemon
local primaryType = args.type1 or ""
local secondaryType = args.type2 or ""
local pokemon = getCachedLearnset(pokemonName)
local moveData = loadMoveData()
if not pokemon then return '' end
local moves = pokemon.egg_moves or {}
if #moves == 0 then
return ""
end
local colors = typeColor(frame, primaryType)
local html = mw.html.create('div')
local table = html:tag('table')
:css('width', '77%')
:css('text-align', 'center')
:css('margin', 'auto')
:css('border-radius', '25px')
:css('border', '3px solid #' .. colors.dark)
:css('background-color', '#' .. colors.light)
:css('padding', '7px')
local header = table:tag('tr')
:css('background-color', '#' .. colors.base)
:css('color', '#' .. colors.dark)
header:tag('th'):wikitext('Move'):css('width', '20%'):css('border-top-left-radius', '25px')
header:tag('th'):wikitext('Type'):css('width', '17%')
header:tag('th'):wikitext('Category'):css('width', '16%')
header:tag('th'):wikitext('Base Power'):css('width', '17%')
header:tag('th'):wikitext('Accuracy'):css('width', '15%')
header:tag('th'):wikitext('PP'):css('width', '15%'):css('border-top-right-radius', '25px')
for _, moveName in ipairs(moves) do
local moveInfo = moveData[moveName] or {}
local row = table:tag('tr'):css('background-color', '#FFFFFF')
local moveLink = '[[' .. moveName .. ']]'
if isSTAB(moveInfo.type, primaryType, secondaryType) and moveInfo.category ~= "Status" then
moveLink = "'''" .. moveLink .. "'''"
end
if moveInfo.broken == "Y" then
moveLink = "''" .. moveLink .. "''"
end
row:tag('td'):wikitext(moveLink)
local moveTypeColors = typeColor(frame, moveInfo.type)
row:tag('td')
:css('background-color', '#' .. moveTypeColors.base)
:css('border', '2px solid #' .. moveTypeColors.dark)
:wikitext('[[' .. (moveInfo.type or 'Normal') .. '_(type)|<span style="color:#FFFFFF;">' .. (moveInfo.type or 'Normal') .. '</span>]]')
local moveCatColors = categoryColor(frame, moveInfo.category)
row:tag('td')
:css('background-color', '#' .. moveCatColors.base)
:css('border', '2px solid #' .. moveCatColors.dark)
:wikitext('[[' .. (moveInfo.category or 'Status') .. ' moves|<span style="color:#FFFFFF;">' .. (moveInfo.category or 'Status') .. '</span>]]')
row:tag('td'):wikitext(moveInfo.power or '—')
row:tag('td'):wikitext(moveInfo.accuracy or '—')
row:tag('td'):wikitext(moveInfo.pp or '—')
end
table:tag('tr'):tag('td'):attr('colspan', '6')
:css('text-align', 'left')
:css('background-color', '#' .. colors.base)
:css('padding', '10px')
:css('border-radius', '1px 1px 25px 25px')
:wikitext([=[
* '''Emboldened''' moves denote that this Pokémon avails from a STAB bonus for it.
* ''Italicized'' moves are [[Broken_Moves_%26_Abilities|broken]]; see their individual pages for more information.
* For more information on egg-move-tutors, refer to: [[Egg Moves]].
]=])
return tostring(html) .. '<br clear="all">'
end
function p.main(frame)
local args = getArgs(frame)
local pokemonName = args[1] or args.pokemon or ''
if pokemonName == '' then
return "<span style='color:red'>Usage: {{Learnsets|PokemonName}}</span>"
end
local primaryType = args.type1 or ''
local secondaryType = args.type2 or ''
if primaryType == '' then
local typeData = loadPokemonTypes()
local pt = typeData[pokemonName]
if pt then
primaryType = pt.type1 or ''
secondaryType = pt.type2 or ''
end
end
local function callRender(fn)
return fn(frame:newChild({
title = frame:getTitle(),
args = { pokemonName, type1 = primaryType, type2 = secondaryType },
}))
end
local poke, lsErr = getCachedLearnset(pokemonName)
if not poke then
return "<span style='color:red'>Data load error: " .. (lsErr or 'unknown') .. "</span>"
end
local sections = {}
if #(poke.level_up or {}) > 0 then
table.insert(sections, "===Level-up===")
table.insert(sections, callRender(p.renderLevelUp))
end
if #(poke.tms or {}) > 0 or #(poke.hms or {}) > 0 then
table.insert(sections, "===[[TMs and HMs]]===")
table.insert(sections, callRender(p.renderTMsHMs))
end
if #(poke.move_tutors or {}) > 0 then
table.insert(sections, "===[[Move Tutors]]===")
table.insert(sections, callRender(p.renderMoveTutors))
end
if #(poke.egg_moves or {}) > 0 then
table.insert(sections, "===[[Egg Moves]]===")
table.insert(sections, callRender(p.renderEggMoves))
end
if #sections == 0 then
return ''
end
return "==Moves==\n" .. table.concat(sections, '\n')
end
return p