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 mw.ustring.find(pos, "%sform") and true or false
end

local function pluralize(word)
	local ending = mw.ustring.sub(word, -1)
	if ending == "h" and mw.ustring.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-lemma forms]]"
	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 mw.ustring.find(head, " ") then
		local pieces = {}
		for piece in mw.ustring.gmatch(head, "%S+") do
			table.insert(pieces, "[[" .. piece .. "]]")
		end
		head = table.concat(pieces, " ")
	end
	if mw.ustring.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) and (not data.labels) then return nil end
	local out = {}
	if data.inflections then
		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
				if inflection.nolink then
					table.insert(i_forms, "<b>" .. inflection[i] .. "</b>")
				else
					if mw.ustring.find(inflection[i], "%*") then
						local temp_links = {}
						if mw.ustring.find(inflection[i], "%*") then
							local before = mw.ustring.match(inflection[i], "^([^%*]+)%S?%*") or ""
							local after = mw.ustring.match(inflection[i], "%*%S?([^%*]+)$") or ""
							
							if before ~= "" then before = m_links.full_link({term = before, language = data.language}, "bold") end
							if after ~= "" then after = m_links.full_link({term = after, language = data.language}, "bold") end
							local head_temp = mw.ustring.match(inflection[i], "(%S?%*%S?)")
							head_temp = mw.ustring.gsub(head_temp, "%*", "<b>" .. data.term .. "</b>")
							
							table.insert(i_forms, before .. head_temp .. after)
						end
					else
						table.insert(i_forms, m_links.full_link({term = inflection[i], language = data.language}, "bold"))
					end
				end
				i = i + 1
			end
			if #i_forms > 0 then
				table.insert(out, "''" .. inflection.label .. "'' " .. table.concat(i_forms, ", "))
			end
		end
	end
	if data.labels then
		for _, label in ipairs(data.labels) do
			table.insert(out, "''" .. "[[Appendix:Glossary#" .. label .. "|" .. label .. "]]''")
		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