Cae: created page

2026-07-18T13:46:55Z

created page

New page

local p = {}

local PokemonForms = require('Module:PokemonForms')
local FORM_SUFFIXES = PokemonForms.FORM_SUFFIXES
local RAW_LIST_ALIASES = PokemonForms.RAW_LIST_ALIASES
local MOB_PREFIXES = PokemonForms.MOB_PREFIXES

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 function blankToNil(v)
if v == nil or v == '' then
return nil
end
return v
end

local moveDataCache = nil

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',
},
})

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',
}
end
end

return moveDataCache
end

local pokemonDataCache = nil

local function loadPokemonData()
if pokemonDataCache then
return pokemonDataCache
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',
number = 'Pokedex Number',
Type1 = 'Type1',
Type2 = 'Type2',
},
})

pokemonDataCache = {}
if csvData then
for _, row in ipairs(csvData) do
pokemonDataCache[row.Name] = {
number = blankToNil(row.number),
type1 = row.Type1 or '',
type2 = row.Type2 or '',
}
end
end
return pokemonDataCache
end

local function loadMoveLearnset(moveName)
local jsonPath = '$[' .. mw.text.jsonEncode(moveName) .. ']'
local ok, result, errors = pcall(function()
return mw.ext.externalData.getFileData({
source = 'data',
['file name'] = 'Move_Learnsets.json',
format = 'JSON',
['use jsonpath'] = true,
data = { move = 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.move
if type(raw) ~= 'table' then
return nil, 'No learnability data found for ' .. moveName
end
return raw, nil
end

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 function hyphenSpaceVariant(s)
if mw.ustring.find(s, '-', 1, true) then
return (mw.ustring.gsub(s, '%-', ' '))
end
return (mw.ustring.gsub(s, ' ', '-'))
end

-- Learnset-style name -> exact PokemonRawList name, or nil.
local function resolveDirect(name, data)
local aliased = RAW_LIST_ALIASES[name]
if aliased and data[aliased] then
return aliased
end
if data[name] then
return name
end
local variant = hyphenSpaceVariant(name)
if data[variant] then
return variant
end

for _, form in ipairs(FORM_SUFFIXES) do
local base
if form.learnsetStyle == 'prefix-space' then
local prefix = form.suffix .. ' '
if mw.ustring.sub(name, 1, mw.ustring.len(prefix)) == prefix then
base = mw.ustring.sub(name, mw.ustring.len(prefix) + 1)
end
else
local tail = '-' .. form.suffix
if mw.ustring.sub(name, -mw.ustring.len(tail)) == tail then
base = mw.ustring.sub(name, 1, -mw.ustring.len(tail) - 1)
end
end

if base then
local rawName
if form.rawList == 'suffix' then
rawName = name
elseif form.rawList == 'prefix-hyphen' then
rawName = form.label .. '-' .. base
elseif form.rawList == 'suffix-label' then
rawName = base .. '-' .. form.label
elseif form.rawList == 'suffix-space' then
rawName = base .. ' ' .. form.label
else
rawName = form.label .. ' ' .. base
end
if data[rawName] then
return rawName
end
local v2 = hyphenSpaceVariant(rawName)
if data[v2] then
return v2
end
end
end

return nil
end

-- Full resolution: direct match first; then special-mob prefix strip
-- ("Crystal Onix" -> Onix, keeping its own name and page link); then
-- progressive suffix strip for forms missing from the raw list
-- ("Arceus-Fire" -> Arceus, piped link to the base page).
local function resolvePokemon(name)
local data = loadPokemonData()

local rawName = resolveDirect(name, data)
if rawName then
return { raw = data[rawName], rawName = rawName, display = rawName, link = rawName, direct = true }
end

for _, prefix in ipairs(MOB_PREFIXES) do
local head = prefix .. ' '
if mw.ustring.sub(name, 1, mw.ustring.len(head)) == head then
rawName = resolveDirect(mw.ustring.sub(name, mw.ustring.len(head) + 1), data)
if rawName then
return { raw = data[rawName], rawName = rawName, display = name, link = name, direct = false, mob = true }
end
end
end

local base = name
while mw.ustring.find(base, '-', 1, true) do
base = mw.ustring.gsub(base, '%-[^%-]*$', '')
rawName = resolveDirect(base, data)
if rawName then
return { raw = data[rawName], rawName = rawName, display = name, link = rawName, direct = false }
end
end

return { raw = nil, rawName = nil, display = name, link = name, direct = false }
end

-- Resolve, dedupe, and dex-sort one section's entries.
-- entries: list of { name = learnsetName, level = number|nil }.
local function prepareRows(entries)
local rows = {}
local byKey = {}

for _, e in ipairs(entries) do
local row = resolvePokemon(e.name)
row.level = e.level

local key = (row.rawName or row.display) .. '\0' .. tostring(e.level or '')
local existing = byKey[key]
if not existing then
byKey[key] = row
table.insert(rows, row)
elseif row.direct and not existing.direct then
for k, v in pairs(row) do
existing[k] = v
end
existing.collapsed = nil
elseif not row.direct and not existing.direct and not existing.mob then
-- several raw-list-less forms of the same base; show the base once
existing.collapsed = true
end
end

table.sort(rows, function(a, b)
local an = a.raw and tonumber(mw.ustring.match(a.raw.number or '', '^%d+')) or math.huge
local bn = b.raw and tonumber(mw.ustring.match(b.raw.number or '', '^%d+')) or math.huge
if an ~= bn then
return an < bn
end
local as = a.raw and a.raw.number or ''
local bs = b.raw and b.raw.number or ''
if as ~= bs then
return as < bs
end
return (a.display or '') < (b.display or '')
end)

return rows
end

local function padNumber(num)
if mw.ustring.len(num) < 3 then
return string.rep('0', 3 - mw.ustring.len(num)) .. num
end
return num
end

local function addTypeCell(frame, tr, pokeType)
local td = tr:tag('td')
if pokeType and pokeType ~= '' then
local c = typeColor(frame, pokeType)
td:css('background-color', '#' .. c.base)
:css('border', '2px solid #' .. c.dark)
:wikitext('[[' .. pokeType .. '_(type)|' .. pokeType .. ']]')
else
td:wikitext('—')
end
end

local function renderTable(frame, entries, moveType, isDamaging, colors, withLevel)
local rows = prepareRows(entries)

local tbl = mw.html.create('table')
:attr('cellspacing', '3')
:css('width', withLevel and '62%' or '60%')
:css('background-color', '#' .. colors.base)
:css('border', '4px solid #' .. colors.dark)
:css('padding', '4px')
:css('border-radius', '25px')
:css('text-align', 'center')

local header = tbl:tag('tr')
:css('color', '#' .. colors.dark)
:css('background-color', '#' .. colors.light)
local function th(text)
return header:tag('th'):wikitext(text):css('border', '1px solid #' .. colors.dark)
end
th('No.'):attr('colspan', '2'):css('width', '17%'):css('border-top-left-radius', '25px')
th('Pokémon'):css('width', '27%')
local t1 = th('Type 1'):css('width', '20%')
local t2 = th('Type 2'):css('width', '20%')
if withLevel then
th('Level'):css('width', '13%'):css('border-top-right-radius', '25px')
else
t2:css('border-top-right-radius', '25px')
end

for _, row in ipairs(rows) do
local tr = tbl:tag('tr'):css('background-color', '#FFFFFF')

local num = row.raw and row.raw.number
tr:tag('td'):wikitext(num and padNumber(num) or '—')
tr:tag('td'):wikitext(num and ('[[File:' .. num .. 'Icon.png]]') or '')

local display = row.collapsed and row.rawName or row.display
local target = row.collapsed and row.rawName or row.link
local link
if target == display then
link = '[[' .. display .. ']]'
else
link = '[[' .. target .. '|' .. display .. ']]'
end
if isDamaging and row.raw and (row.raw.type1 == moveType or row.raw.type2 == moveType) then
link = "'''" .. link .. "'''"
end
tr:tag('td'):wikitext(link)

addTypeCell(frame, tr, row.raw and row.raw.type1)
addTypeCell(frame, tr, row.raw and row.raw.type2)

if withLevel then
tr:tag('td'):wikitext(row.level and tostring(row.level) or '—')
end
end

local footer = tbl:tag('tr')
footer:tag('td')
:attr('colspan', withLevel and '6' or '5')
:css('text-align', 'left')
:css('border', '1px solid #' .. colors.dark)
:css('border-radius', '1px 1px 25px 25px')
:css('background-color', '#' .. colors.light)
:css('color', '#' .. colors.dark)
:css('padding-left', '10px')
:wikitext('\'\'\'Emboldened\'\'\' Pokémon avail from a STAB bonus for this move.')

return tostring(tbl) .. ' '
end

function p.main(frame)
local args = getArgs(frame)
local moveName = args[1] or args.move or ''

if moveName == '' then
local title = mw.title.getCurrentTitle()
moveName = title and title.text or ''
end
if moveName == '' then
return "Usage: {{MoveLearnability|MoveName}}"
end

local entry, err = loadMoveLearnset(moveName)
if not entry then
return "Data load error: " .. (err or 'unknown') .. ""
end

local moveInfo = loadMoveData()[moveName] or {}
local moveType = moveInfo.type or 'Normal'
local isDamaging = (moveInfo.category or 'Status') ~= 'Status'
local colors = typeColor(frame, moveType)

local sectionDefs = {
{ heading = 'Level-Up/Instinctual', key = 'level_up', withLevel = true },
{ heading = '[[TMs]]', key = 'tms' },
{ heading = '[[Move tutor]]', key = 'move_tutors' },
{ heading = '[[Egg-Inheritable Moves|Egg Moves]]', key = 'egg_moves' },
}

local sections = {}
for _, def in ipairs(sectionDefs) do
local raw = entry[def.key] or {}
if #raw > 0 then
local entries = {}
for _, e in ipairs(raw) do
if type(e) == 'table' then
table.insert(entries, { name = e.pokemon, level = e.level })
else
table.insert(entries, { name = e })
end
end
table.insert(sections, '===' .. def.heading .. '===')
table.insert(sections, renderTable(frame, entries, moveType, isDamaging, colors, def.withLevel))
end
end

if #sections == 0 then
return ''
end

return '==Learnability==\n' .. table.concat(sections, '\n')
end

return p

Module:MoveLearnability - Revision history

Cae: created page