Модул:category tree/poscatboiler/data/rhymes

local labels = {}
local raw_categories = {}
local handlers = {}
local raw_handlers = {}



-----------------------------------------------------------------------------
--                                                                         --
--                                  LABELS                                 --
--                                                                         --
-----------------------------------------------------------------------------


labels["риме"] = {
	description = "List of indexes of {{{langname}}} words by their rhymes.",
	umbrella_parents = "Почетак",
	parents = {{name = "{{{langcat}}}", raw = true}},
}

labels["refractory rhymes"] = {
	description = "Lists of {{{langname}}} words that don't rhyme with other words.",
	umbrella_parents = "Риме подкатегорије по језику",
	parents = {"риме"},
}



-----------------------------------------------------------------------------
--                                                                         --
--                              RAW CATEGORIES                             --
--                                                                         --
-----------------------------------------------------------------------------


raw_categories["Риме подкатегорије по језику"] = {
	description = "Umbrella categories covering topics related to rhymes.",
	additional = "{{{umbrella_meta_msg}}}",
	parents = {
		"Кишобран метакатегорија",
		{name = "риме", is_label = true, sort = " "},
	},
}


raw_categories["Риме"] = {
	description = "Umbrella categories covering rhymes by language.",
	additional = "{{{umbrella_meta_msg}}}",
	parents = {
		"Кишобран метакатегорија",
		{name = "риме", is_label = true, sort = " "},
	},
}


-----------------------------------------------------------------------------
--                                                                         --
--                                 HANDLERS                                --
--                                                                         --
-----------------------------------------------------------------------------


table.insert(handlers, function(data)
	local rhyme_beginning = data.label:match("^риме/(.-)%-$")
	if rhyme_beginning then
		return {
			description = ("{{{langname}}} rhyme indexes beginning with {{IPAchar|/%s/}}."):format(rhyme_beginning),
			additional = "The entries in this category are not dictionary entries, but rather indexes of {{{langname}}} terms that rhyme in specific ways.",
			parents = {
				{name = "риме", sort = rhyme_beginning},
			},
			breadcrumb = {name = ("{{IPAchar|/%s-/}}"):format(rhyme_beginning), nocap = true},
			umbrella = false,
		}
	end
end)


-----------------------------------------------------------------------------
--                                                                         --
--                               RAW HANDLERS                              --
--                                                                         --
-----------------------------------------------------------------------------


table.insert(raw_handlers, function(data)
	local function rhymeTopCategory(lang)
		return {
			lang = lang:getCode(),
			description = "Lists of " .. lang:makeCategoryLink() .. " words by their rhymes.",
			additional = "This is an umbrella category. It contains no dictionary entries, but rather possible rhymes as subcategories.",
			parents = {
				"{{{langcat}}}",
				{name = "Риме", sort = langname},
			},
			breadcrumb = "Риме",
			toctemplateprefix = "rhymetop"
		}
	end
	
	local function linkRhymeIfExists(lang, rhyme) -- {{change new|Though this could be ported to Wiktionary}}
		if mw.title.new('Риме:' .. lang:getCanonicalName() .. "/" .. rhyme).exists then
			return "'''[[Риме:" .. lang:getCanonicalName() .. "/" .. rhyme ..'|<span class="IPA">-' .. rhyme .. "</span>]]'''"
		end
		return '<span class="IPA">-' .. rhyme .. '</span>'
	end
	
	local function rhymeCategory(lang, rhyme)
		return {
			lang = lang:getCode(),
			description = "List of " .. lang:makeCategoryLink() .. ' words with the rhyme ' .. linkRhymeIfExists(lang, rhyme) .. '.',
			parents = {
				{name = "Риме:" .. lang:getCanonicalName(), sort = rhyme},
			},
			breadcrumb = '<span class="IPA">-' .. rhyme .. '</span>',
			displaytitle = 'Риме:' .. lang:getCanonicalName() .. '/<span class="IPA">' .. rhyme .. '</span>'
		}
	end
	
	local function rhymeSyllableCountCategory(lang, rhyme, syllableCount)
		return {
			lang = lang:getCode(),
			description = "List of " .. syllableCount .. "-слог " .. lang:makeCategoryLink() .. ' words with the rhyme ' .. linkRhymeIfExists(lang, rhyme) .. '.',
			parents = {
				{name = "Риме:" .. lang:getCanonicalName() .. "/" .. rhyme, sort = syllableCount},
			},
			breadcrumb = tostring(syllableCount) .. " слогова",
			displaytitle = 'Риме:' .. lang:getCanonicalName() .. '/<span class="IPA">' .. rhyme .. '</span>/' .. tostring(syllableCount) .. "слог" .. (syllableCount > 1 and "ова" or "")
		}
	end
	
	local langname = data.category:match("^Риме:([^/]+)")
	if langname then
		local lang = require("Модул:languages").getByCanonicalName(langname)
		if lang then
			local tokens = mw.text.split(data.category, "/")
			if #tokens > 1 then
				local rhyme = tokens[2]
				if #tokens == 3 then
					-- rhyme + syllable count category
					local syllables = nil
					if tokens[3] == "1 слог" then
						syllables = 1
					else
						local syllablesregex = tokens[3]:match("(%d+) слогова")
						if syllablesregex then
							syllables = tonumber(syllablesregex)
							if syllables < 2 then
								syllables = nil
							end
						end
					end
					if syllables then
						return rhymeSyllableCountCategory(lang, rhyme, syllables)
					end
				elseif #tokens == 2 then
					-- rhyme category
					return rhymeCategory(lang, rhyme)
				end
			else
				-- rhyme language top category
				return rhymeTopCategory(lang)
			end
		end
	end
end)




return {LABELS = labels, RAW_CATEGORIES = raw_categories, HANDLERS = handlers, RAW_HANDLERS = raw_handlers}