Módulo:Cat principal/Testes

-- Este módulo implementa {{cat principal}}.

local mHatnote = require('Módulo:Hatnote')
local yesno = require('Módulo:Yesno')
local mTableTools -- lazily initialise
local mArguments -- lazily initialise

local p = {}

function p.catMain(frame)
    mTableTools = require('Módulo:TableTools')
    mArguments = require('Módulo:Arguments')
    local args = mArguments.getArgs(frame, {wrappers = 'Predefinição:Cat principal'})
    local pages = mTableTools.compressSparseArray(args)
    local options = {
        article = args.article,
        selfref = args.selfref
    }
    return p._catMain(options, unpack(pages))
end

function p._catMain(options, ...)
    options = options or {}

    -- Get the links table.
    local links = mHatnote.formatPages(...)
    if not links[1] then
        local page = mw.title.getCurrentTitle().text
        links[1] = mHatnote._formatLink(page)
    end
    for i, link in ipairs(links) do
        links[i] = string.format("'''%s'''", link)
    end

    -- Get the pagetype.
    local pagetype
    -- Work out whether we need to be singular or plural.
    local stringToFormat
    if yesno(options.article) ~= false then
        pagetype = 'artigo'
    if #links > 1 then
       stringToFormat = 'Os %ss principais para esta [[Ajuda:Categorias|categoria]] são %s.'
    else
           stringToFormat = 'O %s principal para esta [[Ajuda:Categoria|categoria]] é %s.'
        end
    else
        pagetype = 'página'
    if #links > 1 then
           stringToFormat = 'As %ss principais para esta [[Ajuda:Categorias|categoria]] são %s.'
        else
       stringToFormat = 'A %s principal para esta [[Ajuda:Categoria|categoria]] é %s.'
    end
    end

    -- Get the text.
    local text = string.format(
        stringToFormat,
        pagetype,
        mw.text.listToText(links)
    )
   
    -- Pass it through to Module:Hatnote.
    local hnOptions = {}
    hnOptions.selfref = options.selfref
    hnOptions.extraclasses = 'relarticle mainarticle'

    return mHatnote._hatnote(text, hnOptions)
end

return p