--[[
Основные функции, касающиеся русского языка
]]
local M={}

function M.o(f)
    return M.obo(f.args[1])
end

function M.obo(phr) -- выбирает предлог «о», «об» или «обо» к фразе, в начале которой может быть пунктуация
    local w = mw.ustring.match(phr,"[%p%s%c]*(.-)[%p%s%c]") or mw.ustring.match(phr,"[%p%s%c]*(.-)$")
    if not w then return nil end
    if string.find(" всей всём всех мне ",' '..mw.ustring.lower(w)..' ',1,true) then return 'обо' end
    local ws=mw.ustring.sub(w,1,2)
    if ws==mw.ustring.upper(ws) then -- abbrev
        if mw.ustring.match(ws,"^[ЙУНФЫАРОЛЭСМИRYUIOASFHLXNMÖÜÄΑΕΟΥΩ]") then return 'об' else return 'о' end
    elseif mw.ustring.match(mw.ustring.upper(w),"^[АОЭИУЫAOIEÖÜÄΑΕΟΥΩ]") then
        return 'об'
    else
        return 'о'
    end
end

local cons='бвгджзйклмнпрстфхцчшщ'; -- all consonants
local cons7='кгхчжшщ'; -- consonants requiring -и
local adjends={['ой']='ые',['ый']='ые',['ий']='ие',['ое']='ые',['ая']='ые',['яя']='ие',['ое']='ые',['ее']='ие'};
local specpl={
    ['город']='города',
    ['болгарин']='болгары',
    ['боярин']='бояре',
    ['брат']='братья',
    ['век']='века',
    ['веко']='веки',
    ['вес']='веса',
    ['грузин']='грузины',
    ['день']='дни',
    ['дерево']='деревья',
    ['директор']='директора',
    ['дитя']='дети',
    ['доктор']='доктора',
    ['дочь']='дочери',
    ['друг']='друзья',
    ['звезда']='звёзды',
    ['знамя']='знамёна',
    ['инспектор']='инспектора',
    ['край'] = 'края',
    ['мать']='матери',
    ['небо']='небеса',
    ['озеро']='озёра',
    ['перо']='перья',
    ['поезд']='поезда',
    ['ребёнок']='дети',
    ['список']='списки',
    ['сын']='сыновья',
    ['татарин']='татары',
    ['церковь']='церкви',
    ['цыган']='цыгане',
    ['человек']='люди',
    ['чудо']='чудеса'
}; -- table of some common special plural cases

function M.s(f)
    return M.so(f.args[1])
end

function M.so(phr) -- выбирает предлог «с» или «со» к фразе, в начале которой может быть пунктуация
    local w = mw.ustring.match(phr, '[%p%s%c]*(.-)[%p%s%c]') or mw.ustring.match(phr, '[%p%s%c]*(.-)$')
    if not w then return nil end
    -- Здесь не отражены аббревиатуры и буквосочетания на других языках — предполагается, что пока это будут прописывать
    -- вручную
    w = mw.language.new('ru'):lcfirst(w)
    -- http://new.gramota.ru/spravka/buro/search-answer?s=с%20(со)
    if mw.ustring.match(w, '^[сзшжрлмв][' .. cons .. ']') or
        mw.ustring.match(w, '^ль[' .. cons .. ']') or
        mw.ustring.match(w, '^щ')
    then
        return 'со'
    else
        return 'с'
    end
end
 
function M.pl(p) -- множественное число, 2-й аргумент 0 (нескл.), 1, 2, 3 или a (adjective - прилагательное)
--Поддержка словосочетаний планируется потом
    local word = p.args[1];
    if specpl[word] then return specpl[word] end;
    local d=p.args[2];
    if d==nil then d=M.guessdecl(p) end;
    if d=='0' then return word
    elseif d=='a' then
        if mw.ustring.match(word,'['..cons7..']..$') then
            return mw.ustring.sub(word,1,-3)..'ие'
        else return mw.ustring.sub(word,1,-3)..adjends[mw.ustring.sub(word,-2)]
        end
elseif d=='1' then
    if  mw.ustring.sub(word,-1)=='я' or mw.ustring.match(word,'['..cons7..']а$')
    then return mw.ustring.sub(word,1,-2)..'и'
    else return mw.ustring.sub(word,1,-2)..'ы' end
elseif d=='2' then
    if mw.ustring.match(word,'о$') then return mw.ustring.sub(word,1,-2)..'а'
    elseif mw.ustring.match(word,'е$') then return mw.ustring.sub(word,1,-2)..'я'
    elseif mw.ustring.match(word,'[ьй]$') then return mw.ustring.sub(word,1,-2)..'и'
    elseif mw.ustring.match(word,'онок$') then return mw.ustring.sub(word,1,-5)..'ата'
    elseif mw.ustring.match(word,'ёнок$') then return mw.ustring.sub(word,1,-5)..'ята'
    elseif mw.ustring.match(word,'анин$') then return mw.ustring.sub(word,1,-5)..'ане'
    elseif mw.ustring.match(word,'янин$') then return mw.ustring.sub(word,1,-5)..'яне'
    elseif mw.ustring.match(word,'ец$') and not mw.ustring.match(word,'['..cons7..']['..cons7..']ец$') then -- !!неправильно для этнохоронимов на -ец типа "ньюйоркцы"
        if mw.ustring.match(word,'лец$') then return mw.ustring.sub(word,1,-3)..'ьцы'
            elseif mw.ustring.match(word,'[аоуыэяёюие]ец$') then return mw.ustring.sub(word,1,-3)..'ьцы'
            else return mw.ustring.sub(word,1,-3)..'цы'
        end
    elseif mw.ustring.match(word,'['..cons7..']$') then return word..'и'
    else return word..'ы'
    end
elseif d=='3' then
    if mw.ustring.match(word,'мя$') then return mw.ustring.sub(word,1,-3)..'ена'
        else return mw.ustring.sub(word,1,-2)..'и'
    end
    else return "<span class=error>Wrong declesion '"..d.."' in RuGrammar::pl()</span>"
    end
end
 
function M.guessdecl(p) -- угадывает тип склонения
    local word=p.args[1];
    if mw.ustring.match(word,'['..cons..'][оыи]й$')
      or mw.ustring.match(word,'['..cons..'][ая]я$')
      or mw.ustring.match(word,'['..cons..'][ое]е$')
    then return 'a' --sometimes wrong
    elseif mw.ustring.match(word,'['..cons..'][оеь]?$') then return '2' --but mb 3, can't guess w/o dict
    elseif mw.ustring.match(word,'['..cons..'][ая]$') then
        if mw.ustring.match(word,'[^'..cons..']мя$') then return '3' else return '1' end
    else return '0'
    end
end
 
function M.locative(p) -- Возможно, будет когда-то заменено на функционал mw.language('ru'):grammar(), которого пока не хватает
	local word=p.args[1];
	local locend={['а']='е', ['я']='е', ['й']='е', ['ы']="ах", ['ь']='и'}; --the last one is disputed: most Russian cities ending with -ь are 3rd declesion, but some foreign ones are 2nd decl., so -е should be returned
	local ec=mw.ustring.sub(word,-1);
	if mw.ustring.match(word,'ия$') then return mw.ustring.sub(word,1,-2)..'и'
	elseif mw.ustring.match(word,'['..cons..']ль$') or mw.ustring.sub(word,-4)=='поль' then return mw.ustring.sub(word,1,-2)..'е'
	elseif mw.ustring.match(word,'ской$') or mw.ustring.match(word,'ское$') or mw.ustring.match(word,'цкое$') then return mw.ustring.sub(word,1,-2)..'м'
	elseif mw.ustring.match(word,'ский$') or mw.ustring.match(word,'цкий$') or mw.ustring.match(word,'ный$') then return mw.ustring.sub(word,1,-3)..'ом'
	elseif mw.ustring.match(word,'ская$') or mw.ustring.match(word,'цкая$') or mw.ustring.match(word,'ная$') then return mw.ustring.sub(word,1,-3)..'ой'
	elseif mw.ustring.match(word,'няя$') then return mw.ustring.sub(word,1,-3)..'ей'
	elseif locend[ec] then return mw.ustring.sub(word,1,-2)..locend[ec]
	elseif mw.ustring.match(ec,'['..cons..']') then return word..'е'
	elseif ec=='о' then
	    if mw.ustring.match(word,'[оеё]во$') or mw.ustring.match(word,'[иы]но$') then return mw.ustring.sub(word,1,-2)..'е'  -- корректно только для топонимов со славянскими корнями
	        else return word
	    end
	else return word end
end;
 
return M