Modul:ru-headword

---

local m_common = require("Module:ru-common")

local export = {}
local pos_functions = {}

local lang = require("Module:languages").getByCode("ru")

-- The main entry point.
-- This is the only function that can be invoked from a template.
function export.show(frame)
	local args = frame:getParent().args
	PAGENAME = mw.title.getCurrentTitle().text
	
	local poscat = frame.args[1] or error("Part of speech has not been specified. Please pass parameter 1 to the module invocation.")
	
	local genders = {}
	local inflections = {}
	local categories = {"Ruschada " .. poscat}
	local tracking_categories = {}
	
	-- Get the head parameters
	-- First get the 1st parameter. The remainder is named head2=, head3= etc.
	local heads = {}
	local head = args[1]; if head == "" then head = nil end
	local i = 2
	
	while head do
		if m_common.needs_accents(head) then
			if args["notrcat"] == nil or args["notrcat"] == "" then
				table.insert(categories, "Urgʻu muhtoj rus shartlari")
			end
		end
		
		table.insert(heads, head)
		head = args["head" .. i]; if head == "" then head = nil end
		i = i + 1
	end
	
	if #heads == 0 and m_common.needs_accents(PAGENAME) then
		table.insert(categories, "Urgʻu muhtoj rus shartlari")
	end
	
	-- Get transliteration
	local head = heads[1] and require("Module:links").remove_links(heads[1]) or PAGENAME
	local head_noaccents = mw.ustring.gsub(head, "\204\129", "")
	local tr_gen = mw.ustring.toNFC(lang:transliterate(head, nil))
	local tr_gen_noaccents = mw.ustring.toNFC(lang:transliterate(head_noaccents, nil))
	
	local tr = args["tr"]; if tr == "" then tr = nil end
	
	if tr then
		if args["notrcat"] == nil or args["notrcat"] == "" then
			table.insert(categories, "Tartibsizlik talaffuz bilan rus atamalar")
		end
		local tr_fixed = tr
		tr_fixed = mw.ustring.gsub(tr_fixed, "ɛ", "e")
		tr_fixed = mw.ustring.gsub(tr_fixed, "([eoéó])v([oó])$", "%1g%2")
		tr_fixed = mw.ustring.gsub(tr_fixed, "([eoéó])v([oó][- ])", "%1g%2")
		tr_fixed = mw.ustring.toNFC(tr_fixed)
		
		if tr == tr_gen or tr == tr_gen_noaccents then
			table.insert(tracking_categories, "ru headword with tr/redundant")
		elseif tr_fixed == tr_gen then
			table.insert(tracking_categories, "ru headword with tr/with manual adjustment")
		elseif mw.ustring.find(tr, ",") then
			table.insert(tracking_categories, "ru headword with tr/comma")
		elseif head_noaccents == PAGENAME then
			if args["notrcat"] == nil or args["notrcat"] == "" then
				table.insert(tracking_categories, "ru headword with tr/headword is pagename")
			end
		else
			table.insert(tracking_categories, "ru headword with tr/headword not pagename")
		end	
	end
	
	if not tr then
		tr = tr_gen
	end
	
	if pos_functions[poscat] then
		pos_functions[poscat](args, heads, genders, inflections, categories)
	end
	
	return require("Module:headword").full_headword(lang, nil, heads, tr, genders, inflections, categories, nil) ..
		require("Module:utilities").format_categories(tracking_categories, lang, nil)
end

pos_functions["proper nouns"] = function(args, heads, genders, inflections, categories)
	pos_functions["nouns"](args, heads, genders, inflections, categories, true)
end

-- Display additional inflection information for a noun
pos_functions["nouns"] = function(args, heads, genders, inflections, categories, no_plural)
	-- Iterate over all gn parameters (g2, g3 and so on) until one is empty
	local g = args[2]; if g == "" then g = nil end
	local i = 2
	
	while g do
		table.insert(genders, g)
		g = args["g" .. i]; if g == "" then g = nil end
		i = i + 1
	end
	
	if #genders == 0 then
		table.insert(genders, "?")
	elseif #genders > 1 then
		table.insert(categories, "Russian nouns with multiple genders")
	end
	
	-- Get the genitive parameters
	-- First get the 3rd parameter. The remainder is named gen2=, gen3= etc.
	local genitives = {}
	local form = args[3]; if form == "" then form = nil end
	local i = 2
	
	while form do
		table.insert(genitives, form)
		form = args["gen" .. i]; if form == "" then form = nil end
		i = i + 1
	end
	
	-- Get the plural parameters
	-- First get the 4th parameter. The remainder is named pl2=, pl3= etc.
	local plurals = {}
	local form = args[4]; if form == "" then form = nil end
	local i = 2
	
	while form do
		table.insert(plurals, form)
		form = args["pl" .. i]; if form == "" then form = nil end
		i = i + 1
	end
	
	-- Get the feminine parameters
	-- First get the f= parameter. The remainder is named f2=, f3= etc.
	local feminines = {}
	local form = args["f"]; if form == "" then form = nil end
	local i = 2
	
	while form do
		table.insert(feminines, form)
		form = args["f" .. i]; if form == "" then form = nil end
		i = i + 1
	end

	-- Get the masculine parameters
	-- First get the m= parameter. The remainder is named m2=, m3= etc.
	local masculines = {}
	local form = args["m"]; if form == "" then form = nil end
	local i = 2
	
	while form do
		table.insert(masculines, form)
		form = args["m" .. i]; if form == "" then form = nil end
		i = i + 1
	end
	
	-- Process the genders
	local singular_genders = {
		["m"] = true,
		["m-?"] = true,
		["m-an"] = true,
		["m-in"] = true,
		
		["f"] = true,
		["f-?"] = true,
		["f-an"] = true,
		["f-in"] = true,
		
		["n"] = true,
		["n-an"] = true,
		["n-in"] = true}
	
	local plural_genders = {
		["p"] = true,  -- This is needed because some invariant plurale tantums have no gender to speak of
		
		["m-p"] = true,
		["m-?-p"] = true,
		["m-an-p"] = true,
		["m-in-p"] = true,

		["f-p"] = true,
		["f-?-p"] = true,
		["f-an-p"] = true,
		["f-in-p"] = true,
		
		["n-p"] = true,
		["n-?-p"] = true,
		["n-an-p"] = true,
		["n-in-p"] = true }
	
	for i, g in ipairs(genders) do
		if g == "m" then
			g = "m-?"
		elseif g == "m-p" then
			g = "m-?-p"
		elseif g == "f" and plurals[1] ~= "-" and not no_plural then
			g = "f-?"
		elseif g == "f-p" then
			g = "f-?-p"
		end
		
		if not singular_genders[g] and not plural_genders[g] then
			g = "?"
		end
		
		genders[i] = g
		
		-- Categorize by gender
		if g:sub(1,1) == "m" then
			table.insert(categories, "Russian masculine nouns")
		elseif g:sub(1,1) == "f" then
			table.insert(categories, "Russian feminine nouns")
		elseif g:sub(1,1) == "n" then
			table.insert(categories, "Russian neuter nouns")
		end
		
		-- Categorize by animacy
		if g:sub(3,4) == "an" then
			table.insert(categories, "Russian animate nouns")
		elseif g:sub(3,4) == "in" then
			table.insert(categories, "Russian inanimate nouns")
		end
		
		-- Categorize by number
		if plural_genders[g] then
			table.insert(categories, "Russian pluralia tantum")
			
			if g == "p" then
				table.insert(categories, "Russian pluralia tantum with incomplete gender")
			end
		end
	end
	
	-- Add the genitive forms
	if genitives[1] == "-" then
		table.insert(inflections, {label = "[[Appendix:Glossary#indeclinable|indeclinable]]"})
		table.insert(categories, "Russian indeclinable nouns")
	elseif #genitives > 0 then
		local gen_parts = {label = "genitive"}
		
		for i, form in ipairs(genitives) do
			table.insert(gen_parts, form)
			
			if m_common.needs_accents(form) then
				table.insert(categories, "Russian noun inflections needing accents")
			end
		end
		
		table.insert(inflections, gen_parts)
	end
	
	-- Add the plural forms
	-- If the noun is a plurale tantum, then ignore the 4th parameter altogether
	if no_plural or genitives[1] == "-" then
		-- do nothing
	elseif plural_genders[genders[1]] then
		table.insert(inflections, {label = "[[Appendix:Glossary#plurale tantum|plurale tantum]]"})
	elseif plurals[1] == "-" then
		table.insert(inflections, {label = "[[Appendix:Glossary#uncountable|uncountable]]"})
		table.insert(categories, "Russian uncountable nouns")
	elseif #plurals > 0 then
		local pl_parts = {label = "nominative plural"}
		
		for i, form in ipairs(plurals) do
			table.insert(pl_parts, form)
			
			if m_common.needs_accents(form) then
				table.insert(categories, "Russian noun inflections needing accents")
			end
		end
		if plural and not mw.title.new(plural).exists then
				table.insert(categories, "Russian nouns with missing plurals")
			end
			if plural2 and not mw.title.new(plural2).exists then
				table.insert(categories, "Russian nouns with missing plurals")
			end
			
		table.insert(inflections, pl_parts)
	end
	
	-- Add the feminine forms
	if #feminines > 0 then
		local f_parts = {label = "feminine"}
		
		for i, form in ipairs(feminines) do
			table.insert(f_parts, form)
			
			if m_common.needs_accents(form) then
				table.insert(categories, "Russian noun inflections needing accents")
			end
		end
		
		table.insert(inflections, f_parts)
	end
	
	-- Add the masculine forms
	if #masculines > 0 then
		local m_parts = {label = "masculine"}
		
		for i, form in ipairs(masculines) do
			table.insert(m_parts, form)
			
			if m_common.needs_accents(form) then
				table.insert(categories, "Russian noun inflections needing accents")
			end
		end
		
		table.insert(inflections, m_parts)
	end
end

-- Display additional inflection information for an adjective
pos_functions["adjectives"] = function(args, heads, genders, inflections, categories)
	local comp = args[2]; if comp == "" then comp = nil end
	local sup = args[3]; if sup == "" then sup = nil end
	
	local comp2 = args["comp2"]; if comp2 == "" then comp2 = nil end
	local comp3 = args["comp3"]; if comp3 == "" then comp3 = nil end
	local sup2 = args["sup2"]; if sup2 == "" then sup2 = nil end
	local sup3 = args["sup3"]; if sup3 == "" then sup3 = nil end
	
	if comp then
		local comp_parts = {label = "comparative"}
		
		if comp == "peri" then
			table.insert(comp_parts, "[[бо́лее]] " .. PAGENAME)
		else
			table.insert(comp_parts, comp)
			
			if m_common.needs_accents(comp) then
				table.insert(categories, "Russian adjective inflections needing accents")
			end
		end
		
		if comp2 then
			table.insert(comp_parts, comp2)
			
			if m_common.needs_accents(comp2) then
				table.insert(categories, "Russian adjective inflections needing accents")
			end
		end
		
		if comp3 then
			table.insert(comp_parts, comp3)
			
			if m_common.needs_accents(comp3) then
				table.insert(categories, "Russian adjective inflections needing accents")
			end
		end
		
		table.insert(inflections, comp_parts)
	end
	
	if sup then
		local sup_parts = {label = "superlative"}
		
		if sup == "peri" then
			table.insert(sup_parts, "[[са́мый]] " .. PAGENAME)
		else
			table.insert(sup_parts, sup)
			
			if m_common.needs_accents(sup) then
				table.insert(categories, "Russian adjective inflections needing accents")
			end
		end
		
		if sup2 then
			table.insert(sup_parts, sup2)
			
			if m_common.needs_accents(sup2) then
				table.insert(categories, "Russian adjective inflections needing accents")
			end
		end
		
		if sup3 then
			table.insert(sup_parts, sup3)
			
			if m_common.needs_accents(sup3) then
				table.insert(categories, "Russian adjective inflections needing accents")
			end
		end
		
		table.insert(inflections, sup_parts)
	end
end

-- Display additional inflection information for an adverb
pos_functions["adverbs"] = function(args, heads, genders, inflections, categories)
	local comp = args[2]; if comp == "" then comp = nil end

	local comp2 = args["comp2"]; if comp2 == "" then comp2 = nil end
	local comp3 = args["comp3"]; if comp3 == "" then comp3 = nil end
	
	if comp then
		local encoded_head = ""
 		
		if heads[1] ~= "" then
			-- This is decoded again by [[WT:ACCEL]].
			encoded_head = " origin-" .. heads[1]:gsub("%%", "."):gsub(" ", "_")
		end
 		
 		local comp_parts = {label = "comparative", accel = "comparative-form-of" .. encoded_head}
		table.insert(comp_parts, comp)
		
		if m_common.needs_accents(comp) then
				table.insert(categories, "Russian adverb comparatives needing accents")
		end
		
		if comp2 then
			table.insert(comp_parts, comp2)
 
			if m_common.needs_accents(comp2) then
				table.insert(categories, "Russian adverb comparatives needing accents")
			end
		end
		
		if comp3 then
			table.insert(comp_parts, comp3)
 
			if m_common.needs_accents(comp3) then
				table.insert(categories, "Russian adverb comparatives needing accents")
			end
		end
 
		table.insert(inflections, comp_parts)
	end
end

-- Display additional inflection information for a verb
pos_functions["verbs"] = function(args, heads, genders, inflections, categories)
	-- Aspect
	local aspect = args[2]; if aspect == "" then aspect = nil end
	
	if aspect == "impf" then
		table.insert(genders, "impf")
		table.insert(categories, "Russian imperfective verbs")
	elseif aspect == "pf" then
		table.insert(genders, "pf")
		table.insert(categories, "Russian perfective verbs")
	elseif aspect == "both" then
		table.insert(genders, "impf")
		table.insert(genders, "pf")
		table.insert(categories, "Russian imperfective verbs")
		table.insert(categories, "Russian perfective verbs")
		table.insert(categories, "Russian biaspectual verbs")
	else
		table.insert(genders, "?")
		table.insert(categories, "Russian verbs needing aspect")
	end
	
	-- Get the imperfective parameters
	-- First get the impf= parameter. The remainder is named impf2=, impf3= etc.
	local imperfectives = {}
	local form = args["impf"]; if form == "" then form = nil end
	local i = 2
	
	while form do
		table.insert(imperfectives, form)
		form = args["impf" .. i]; if form == "" then form = nil end
		i = i + 1
	end
	
	-- Get the perfective parameters
	-- First get the pf= parameter. The remainder is named pf2=, pf3= etc.
	local perfectives = {}
	local form = args["pf"]; if form == "" then form = nil end
	local i = 2
	
	while form do
		table.insert(perfectives, form)
		form = args["pf" .. i]; if form == "" then form = nil end
		i = i + 1
	end

	-- Add the imperfective forms
	if #imperfectives > 0 then
		local impf_parts = {label = "imperfective"}
		
		for i, form in ipairs(imperfectives) do
			table.insert(impf_parts, form)
			
			if m_common.needs_accents(form) then
				table.insert(categories, "Russian verb inflections needing accents")
			end
		end
		
		table.insert(inflections, impf_parts)
	end
	
	-- Add the perfective forms
	if #perfectives > 0 then
		local pf_parts = {label = "perfective"}
		
		for i, form in ipairs(perfectives) do
			table.insert(pf_parts, form)
			
			if m_common.needs_accents(form) then
				table.insert(categories, "Russian verb inflections needing accents")
			end
		end
		
		table.insert(inflections, pf_parts)
	end
end

return export