Module:head

Documentation for this module may be created at Module:head/documentation

local export = {}

local m_links = require("Module:links")
local m_gender_and_number = require("Module:gender and number")
local m_formatting = require("Module:formatting")

local function is_non_lemma(pos)
	return string.find(pos, "%sform") and true or false
end

local function pluralize(word)
	local ending = string.sub(word, -1)
	if ending == "h" and string.sub(word, -2, -1) == "s" then ending = "sh" end
	if ending == "s" or ending == "z" or ending == "x" or ending == "sh" then
		return word .. "es"
	end
	return word .. "s"
end

local function format_headword(data)
	local cats = ""
	if is_non_lemma(data.pos) then
		cats = cats .. "[[Category:" .. data.language.name .. " non-lemmas]]"
	else
		cats = cats .. "[[Category:" .. data.language.name .. " lemmas]]"
	end
	cats = cats .. "[[Category:" .. data.language.name .. " " .. pluralize(data.pos) .. "]]"
	local head = data.term
	if data.head then
		head = data.head
	elseif string.find(head, " ") then
		local pieces = {}
		for piece in string.gmatch(head, "%S+") do
			table.insert(pieces, "[[" .. piece .. "]]")
		end
		head = table.concat(pieces, " ")
	end
	if string.find(head, " ") then
		cats = cats .. "[[Category:" .. data.language.name .. " multiword terms]]"
	end
	return m_formatting.wrap_face(head, data.language, "head"), cats
end

local function format_genders(data)
	if (not data.genders) or (type(data.genders) ~= "table") then return nil end
	return m_gender_and_number.format_genders(data.language, data.pos, data.genders)
end

local function format_inflections(data)
	if (not data.inflections) or (#data.inflections < 1) then return nil end
	local out = {}
	for _, inflection in pairs(data.inflections) do
		if not inflection.label then error("headword inflection must have a label") end
		local i_forms = {}
		local i = 1
		while inflection[i] do
			table.insert(i_forms, m_links.full_link({term = inflection[i], language = data.language}, "bold"))
			i = i + 1
		end
		if #i_forms > 0 then
			table.insert(out, "''" .. inflection.label .. "'' " .. table.concat(i_forms, ", "))
		end
	end
	if #out < 1 then return nil end
	return "(" .. table.concat(out, ", ") .. ")"
end

function export.full_head(data)
	local out = {}
	local cat = ""
	local head_out, head_cat = format_headword(data)
	if head_out then table.insert(out, head_out) end
	if head_cat and head_cat ~= "" then cat = cat .. head_cat end
	local genders_out, genders_cat = format_genders(data)
	if genders_out then table.insert(out, genders_out) end
	if genders_cat and genders_cat ~= "" then cat = cat .. genders_cat end
	local inflections = format_inflections(data)
	if inflections then table.insert(out, inflections) end
	return table.concat(out, " ") .. (data.nocat and "" or cat)
end

return export