Module:headword/templates

From Linguifex
< Module:headword
Revision as of 11:16, 6 August 2024 by Sware (talk | contribs)
Jump to navigation Jump to search


insert = table.insert
local process_params = require("Module:parameters").process

local export = {}

local function get_args(frame)
	local boolean = {type = "boolean"}
	local boolean_list_allow_holes = {type = "boolean", list = true, allow_holes = true}
	local list_allow_holes = {list = true, allow_holes = true}
	local plain = {}
	return process_params(frame:getParent().args, {
		[1] = {required = true, type = "language", default = "und"},
		["sc"] = {type = "script"},
		["sort"] = plain,

		[2] = {required = true, default = "nouns"},
		["sccat"] = boolean,
		["noposcat"] = boolean,
		["nomultiwordcat"] = boolean,
		["nogendercat"] = boolean,
		["nopalindromecat"] = boolean,
		["nolinkhead"] = boolean,
		["autotrinfl"] = boolean,
		["cat2"] = plain,
		["cat3"] = plain,
		["cat4"] = plain,
		
		["head"] = list_allow_holes,
		["id"] = plain,
		["tr"] = list_allow_holes,
		["ts"] = list_allow_holes,
		["gloss"] = plain,
		["g"] = {list = true},
		["g\1qual"] = list_allow_holes,
		
		[3] = list_allow_holes,

		["f\1request"] = list_allow_holes,
		["f\1alt"] = list_allow_holes,
		["f\1lang"] = {list = true, allow_holes = true, type = "language"},
		["f\1sc"] = {list = true, allow_holes = true, type = "script"},
		["f\1id"] = list_allow_holes,
		["f\1tr"] = list_allow_holes,
		["f\1ts"] = list_allow_holes,
		["f\1g"] = list_allow_holes,
		["f\1qual"] = list_allow_holes,
		["f\1autotr"] = boolean_list_allow_holes,
		["f\1nolink"] = boolean_list_allow_holes,
	})
end

function export.head_t(frame)
	local m_headword = require("Module:headword")

	local args = get_args(frame)

	-- Get language and script information
	local data = {}
	data.lang = require("Module:languages").getByCode(args[1])
	data.sc = args["sc"]
	data.sccat = args["sccat"]
	data.sort_key = args["sort"]
	data.heads = args["head"]
	data.id = args["id"]
	data.translits = args["tr"]
	data.transcriptions = args["ts"]
	data.gloss = args["gloss"]
	data.genders = args["g"]

	-- Part-of-speech category
	local pos_category = args[2]
	data.noposcat = args["noposcat"]
	
	-- Check for headword aliases and then pluralize if the POS term does not have an invariable plural.
	data.pos_category = m_headword.pluralize_pos(
		mw.loadData("Module:headword/data").pos_aliases[pos_category] or pos_category
	)

	-- Additional categories.
	data.categories = {}
	data.whole_page_categories = {}
	data.nomultiwordcat = args["nomultiwordcat"]
	data.nogendercat = args["nogendercat"]
	data.nopalindromecat = args["nopalindromecat"]
    table.insert(data.categories, "Contionary")

	if args["cat2"] then
		insert(data.categories, data.lang:getFullName() .. " " .. args["cat2"])
	end

	if args["cat3"] then
		insert(data.categories, data.lang:getFullName() .. " " .. args["cat3"])
	end

	if args["cat4"] then
		insert(data.categories, data.lang:getFullName() .. " " .. args["cat4"])
	end

	-- Headword linking
	data.nolinkhead = args["nolinkhead"]

	-- Inflected forms
	data.inflections = {enable_auto_translit = args["autotrinfl"]}

	for i = 1, math.ceil(args[3].maxindex / 2) do
		local infl_part = {
			label    = args[3][i * 2 - 1],
			request  = args["frequest"][i],
			enable_auto_translit = args["fautotr"][i],
		}
		
		local form = {
			term          =  args[3][i * 2],
			alt           =  args["falt"][i],
			genders       =  args["fg"][i] and mw.text.split(args["fg"][i], ",") or {},
			id            =  args["fid"][i],
			lang          =  args["flang"][i],
			nolinkinfl    =  args["fnolink"][i],
			q             = {args["fqual"][i]},
			sc            =  args["fsc"][i],
			translit      =  args["ftr"][i],
			transcription =  args["fts"][i],
		}
		
		-- If no term or alt is given, then the label is shown alone.
		if form.term or form.alt then
			insert(infl_part, form)
		else
            table.insert(data.categories, "Contionary stubs")
		end
		
		if infl_part.label == "or" then
			-- Append to the previous inflection part, if one exists
			if #infl_part > 0 and data.inflections[1] then
				insert(data.inflections[#data.inflections], form)
			end
		elseif infl_part.label then
			-- Add a new inflection part
			insert(data.inflections, infl_part)
		end
	end
	
	return m_headword.full_headword(data)
end

return export