Module:Learners

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

local p = {}
local cargo = mw.ext.cargo
local getArgs = require('Module:Arguments').getArgs

function p.Main( frame )
    local args = getArgs(frame)
    local play = (args[1] ~= ""  and args[1]) or mw.title.getCurrentTitle().text
    local levels = args["level"]
    levels = levels and levels ~= ""
    
    local tables = (levels and "BeastieLearnset") or "BeastieLearnFriend"
    local fields = (levels and "Beastie,Plays") or "Beastie"
    local where = tables .. ".Plays HOLDS LIKE '" .. ((levels and play .. "%") or play) .. "'"
    local results = cargo.query(tables, fields, {where=where})
    local bigHeaderText = (levels and "From Level Up") or "With Friendship"
    
    local tbl = mw.html.create('table'):addClass('wikitable')
    tbl:tag("tr"):tag("th"):attr("colspan", (levels and 2) or 1):wikitext(bigHeaderText)
    local header = tbl:tag("tr")
    :tag("th"):wikitext("Beastie"):done()
    if levels then
        header:tag("th"):wikitext("Level")
    end

    for _, result in ipairs(results) do
        local tr = tbl:tag("tr")
        tr:tag("td"):wikitext("[[File:" .. result.Beastie .. "Idle.png|80x80px]]" .. " " .. result.Beastie)
        if levels then
            for rplay in mw.text.gsplit(result.Plays, ", ", true) do
                local rplaysplit = mw.text.split(rplay, ";", true)
                if rplaysplit[1] == play then
                    tr:tag("td"):wikitext(rplaysplit[2])
                    break
                end
            end
        end
    end
    return tbl
end

return p