Modul:Jednina i množina

local p = {}

--  Funkcija koja detektira množinu 
local function isPlural(text)
    if not text then return false end

    -- Potraži predloške za popise
    local pluralIndicators = {'ubl', 'unbulleted list', 'hlist', 'plainlist'}
    for _, pattern in ipairs(pluralIndicators) do
        if mw.ustring.find(text, pattern) then
            return true
        end
    end

    return false
end

--  Glavna funkcija
function p.main(frame)
    local args = frame:getParent().args
    local input = args[1] or ''

    local jednina = args['jednina'] or args['jd'] or ''
    local mnozina = args['mnozina'] or args['mn'] or ''

    -- Ako su jednina ili mnozina prazni
    if jednina == '' and mnozina == '' then
        return ''  
    end

    if isPlural(input) then
        return mnozina
    else
        return jednina
    end
end

p[""] = p.main
p["main"] = p.main
p.nonexistent = p.main


return p