Модуль:Песочница/Pok

Для документации этого модуля может быть создана страница Модуль:Песочница/Pok/doc

local p = {}

local rules = {
    {"ие$", "ие", "ия"},  -- здание -> здания
    {"ка$", "ка", "ки"},  -- собака -> собаки
    {"ок$", "ок", "ки"},  -- носок -> носки
    {"ец$", "ец", "цы"},  -- столбец -> столбцы
    {"ье$", "ье", "ья"},  -- зелье -> зелья
    {"е$", "е", "я"},     -- поле -> поля
    {"й$", "й", "и"},     -- край -> края
    {"ь$", "ь", "и"},     -- лошадь -> лошади
    {"о$", "о", "а"},     -- окно -> окна
    {"ый$", "ый", "ые"},  -- малый -> малые, новый -> новые
    {"ий$", "ий", "ие"},  -- синий -> синие
    {"ой$", "ой", "ие"},  -- большой -> большие
    {"ое$", "ое", "ые"},  -- холодное -> холодные
    {"ее$", "ее", "ие"}   -- синее -> синие
}

function p.main(frame)
    local text = frame.args[1] or ""
    local words = {}
    for word in text:gmatch("%S+") do
        -- Если слово заканчивается на "о" или "я", оставляем без изменений
        if word:match("[оя]$") then
            table.insert(words, word)
        else
            local pluralized = nil
            for _, rule in ipairs(rules) do
                if word:match(rule[1]) then
                    pluralized = word:gsub(rule[2].."$", rule[3])
                    break
                end
            end
            if not pluralized then
                pluralized = word .. "ы"
            end
            table.insert(words, pluralized)
        end
    end
    return table.concat(words, " ")
end

return p