View source for Module:PokemonSpawnData

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

-- Read a parser variable set by {{Trichrome}} (border, background, cell, etc.)
local function getVar(frame, name)
    return frame:callParserFunction('#var', { name }) or ''
end

-- Load an External Data source and return true if at least one row was found.
-- Land and surf use separate external variable names (land_area vs surf_area)
-- so they never share a slot; each can be checked independently via #external_value.
local function loadFileData(frame, params, checkField)
    local callParams = { [1] = 'source=' .. (params.source or 'data') }
    for k, v in pairs(params) do
        if k ~= 'source' then callParams[k] = v end
    end
    frame:callParserFunction('#get_file_data', callParams)
    local val = frame:callParserFunction('#external_value', { checkField })
    return val ~= nil and val ~= ''
end

-- ---------------------------------------------------------------------------
-- Land spawn table
-- ---------------------------------------------------------------------------

local function renderLandTable(frame, border, background, cell)
    local rows = frame:callParserFunction('#display_external_table', {
        [1]  = 'template=PokemonLandDataRow',
        data = 'area=land_area,member=member,number=number,name=name,'
            .. 'morning=morning,day=day,night=night,'
            .. 'item=item,rarity=rarity,minlvl=minlvl,maxlvl=maxlvl',
    })

    local header =
        '<table align="left" style="width: 85%; max-width: 90%; text-align: center; margin: auto;'
        .. ' border-radius: 15px; border: 3px solid ' .. border .. ';'
        .. ' background-color: ' .. background .. '; padding: 7px;">\n'
        .. '<tr style="background-color: ' .. cell .. '; color: ' .. border .. ';'
        .. '"><th scope="col" style="border-top-left-radius: 25px; width: 15%; border: 1px solid ' .. border .. ';">Location</th>'
        .. '<th scope="col" style="width: 7%; border: 1px solid ' .. border .. ';">Levels</th>'
        .. '<th colspan="3" scope="col" style="width: 17%; border: 1px solid ' .. border .. ';">[[Pokétime|<span style="color: ' .. border .. ';">Times</span>]]</th>'
        .. '<th style="width: 20%; border: 1px solid ' .. border .. ';">Held Item</th>'
        .. '<th style="border-top-right-radius: 25px; width: 10%; border: 1px solid ' .. border .. ';">[[List of Pokémon by Rarity Tier|<span style="color: ' .. border .. ';">Rarity Tier</span>]]</th></tr>'

    local footer =
        '<tr><td colspan="8" style="text-align: left; border-radius: 1px 1px 25px 25px;'
        .. ' border: 1px solid ' .. border .. '; background-color: ' .. cell .. ';'
        .. '"><p style="margin-top: 8px; margin-left: 10px;"><ul>\n'
        .. '<li><span style="color:#FF00BF; font-weight: bold;">Pink-colored</span> areas denote that this spawn is [[membership]]-exclusive</li>\n'
        .. '<li>\'\'\'Emboldened\'\'\' levels indicate they are isolatable with the [[Repel trick]]</li>'
        .. '</ul></p></td></tr>\n</table><br clear="all">'

    return header .. rows .. footer
end

-- ---------------------------------------------------------------------------
-- Surf/Fishing spawn table
-- ---------------------------------------------------------------------------

local WATER_DARK  = '{{Water_color_dark}}'
local WATER_BG    = '#{{Water_color}}'
local WATER_LIGHT = '#{{Water_color_light}}'

local function renderSurfTable(frame)
    local rows = frame:callParserFunction('#display_external_table', {
        [1]  = 'template=PokemonSurfingDataRow',
        data = 'area=surf_area,number=number,name=name,time=time,'
            .. 'minlvl=minlvl,maxlvl=maxlvl,item=item,rarity=rarity,'
            .. 'rod=rod,fishable=fishable,member=member',
    })

    local header = [=[
<table align="left" style="width:88%;max-width:100%;text-align:center;margin:auto;border-radius:15px;border:4px solid #{{Water_color_dark}};background-color:#{{Water_color}};padding:7px;">
<tr><td colspan="8"><div style="margin:auto;width:53px;height:53px;border:3px solid #{{Water_color_dark}};border-radius:40px;background-color:#{{Water_color_light}};padding:5px;">[[File:PikachuSurf.png]]</div></td></tr>
<tr style="background-color:#{{Water_color_light}};">
<th scope="col" style="border-radius:15px 1px 1px 1px;width:15%;border:1px solid #{{Water_color_dark}};color:#{{Water_color_dark}};">Location</th>
<th scope="col" style="width:7%;border:1px solid #{{Water_color_dark}};color:#{{Water_color_dark}};">Levels</th>
<th colspan="3" scope="col" style="width:6%;border:1px solid #{{Water_color_dark}};">[[Pokétime|<span style="color:#{{Water_color_dark}};">Times</span>]]</th>
<th style="width:10%;border:1px solid #{{Water_color_dark}};color:#{{Water_color_dark}};">Held Item</th>
<th style="width:7%;border:1px solid #{{Water_color_dark}};color:#{{Water_color_dark}};">Rod</th>
<th style="border-radius:1px 15px 1px 1px;width:10%;border:1px solid #{{Water_color_dark}};">[[List of Pokémon by Rarity Tier|<span style="color:#{{Water_color_dark}};">Rarity Tier</span>]]</th>
</tr>]=]

    local footer = [=[
<tr><td colspan="9" style="text-align:left;border-radius:1px 1px 25px 25px;border:1px solid #{{Water_color_dark}};background-color:#{{Water_color_light}};"><ul style="margin-top:8px;margin-left:12px;padding:5px;">
<li>All rows with a fishing rod indicate that its corresponding Pokémon can also be [[Fishing|fished]] with the rod tier shown</li>
<li><span style="color:#FF00BF;font-weight:bold;">Pink-colored</span> areas denote that this area or inhabiting spawn is strictly [[membership]]-exclusive</li>
<li>'''Emboldened''' levels indicate they are isolatable with the [[Repel trick]]</li>
</ul></td></tr>
</table><br clear="all">]=]

    return header .. rows .. footer
end

-- ---------------------------------------------------------------------------
-- main()
-- ---------------------------------------------------------------------------

function p.main(frame)
    local args        = getArgs(frame)
    local pokemonName = args[1] or args.Name
                     or frame:callParserFunction('PAGENAME', {}) or ''
    local border     = args.border     or ''
    local background = args.background or ''
    local cell       = args.cell       or ''

    local out     = {}
    local hasLand = false
    local hasSurf = false

    -- ── Land spawns ──────────────────────────────────────────────────────────
    hasLand = loadFileData(frame, {
        source        = 'spawns',
        ['file name'] = 'land_spawns.csv',
        format        = 'CSV with header',
        filters       = 'Pokemon=' .. pokemonName,
        data          = 'land_area=Map,member=Member,number=DexID,name=Pokemon,'
                     .. 'morning=Morning,day=Day,night=Night,'
                     .. 'item=Item,rarity=Tier,minlvl=MinLvl,maxlvl=MaxLvl',
    }, 'land_area')
    if hasLand then
        table.insert(out, renderLandTable(frame, border, background, cell))
    end

    -- ── Surf / Fishing spawns ─────────────────────────────────────────────────
    hasSurf = loadFileData(frame, {
        source        = 'spawns',
        ['file name'] = 'surf_spawns.json',
        format        = 'JSON',
        filters       = 'Pokemon=' .. pokemonName .. ',FishingOnly=0',
        data          = 'surf_area=Map,member=MemberOnly,number=MonsterID,name=Pokemon,'
                     .. 'time=Daytime,item=Item,rarity=Tier,'
                     .. 'minlvl=MinLvl,maxlvl=MaxLvl,rod=RequiredRod,fishable=Fishing',
    }, 'surf_area')
    if hasSurf then
        table.insert(out, renderSurfTable(frame))
    end

    -- ── No spawns ─────────────────────────────────────────────────────────────
    if not hasLand and not hasSurf then
        table.insert(out,
            "<div style='padding:8px;font-style:italic;'>"
         .. pokemonName
         .. " has no wild spawn locations.</div>"
        )
    end

    return table.concat(out, '\n')
end

return p