Jump to content

Module:User:Benwing2/lad-headword

From Wiktionary, the free dictionary

This is a private module sandbox of Benwing2, for their own experimentation. Items in this module may be added and removed at Benwing2's discretion; do not rely on this module's stability.


local export = {}
local pos_functions = {}
local m_scripts = require("Module:scripts")

local lang = require("Module:languages").getByCode("lad")
local langname = lang:getCanonicalName()

local PAGENAME = mw.title.getCurrentTitle().text

local suffix_categories = {
	["adjectives"] = true,
	["adverbs"] = true,
	["nouns"] = true,
	["verbs"] = true,
}

local function track(page)
	require("Module:debug").track("lad-headword/" .. page)
	return true
end

local function glossary_link(entry, text)
	text = text or entry
	return "[[Appendix:Glossary#" .. entry .. "|" .. text .. "]]"
end

-- The main entry point.
-- This is the only function that can be invoked from a template.
function export.show(frame)
	local tracking_categories = {}
	
	local poscat = frame.args[1]
		or error("Part of speech has not been specified. Please pass parameter 1 to the module invocation.")
	
	local params = {
		["head"] = {list = true},
		[1] = {}, --explicit script; ignored; will remove
		["lat"] = {list = true}, --Latin equivalent(s)
		["heb"] = {list = true}, --Hebrew equivalent(s)
		["suff"] = {type = "boolean"},
	}

	if pos_functions[poscat] then
		for key, val in pairs(pos_functions[poscat].params) do
			params[key] = val
		end
	end
	
    local parargs = frame:getParent().args
	local args = require("Module:parameters").process(parargs, params)
	local data = {
		lang = lang,
		pos_category = poscat,
		categories = {},
		heads = args["head"],
		genders = {},
		inflections = {},
		categories = {}
	}
	
	if args["suff"] then
		data.pos_category = "suffixes"
		
		if suffix_categories[poscat] then
			local singular_poscat = poscat:gsub("s$", "")
			table.insert(data.categories, langname .. " " .. singular_poscat .. "-forming suffixes")
		else
			error("No category exists for suffixes forming " .. poscat .. ".")
		end
	end
	
	data.sc = m_scripts.findBestScript(data.heads[1] or PAGENAME, lang)
	data.is_heb = data.sc:getCode() == "Hebr"
	data.is_lat = data.sc:getCode() == "Latn"

	if data.is_heb then
		table.insert(data.inflections, {label = "[[Appendix:Hebrew alphabet|Hebrew spelling]]"})
	elseif data.is_lat then
		table.insert(data.inflections, {label = "[[Appendix:Roman script|Latin spelling]]"})
	end

	if #args.lat > 0 then
		args.lat.label = "[[Appendix:Roman script|Latin spelling]]"
		args.lat.sc = m_scripts.getByCode("Latn")
		table.insert(data.inflections, args.lat)
	end

	if #args.heb > 0 then
		args.heb.label = "[[Appendix:Hebrew alphabet|Hebrew spelling]]"
		args.heb.sc = m_scripts.getByCode("Hebr")
		table.insert(data.inflections, args.heb)
	end

	if pos_functions[poscat] then
		pos_functions[poscat].func(args, data, tracking_categories)
	end
	
	return require("Module:headword").full_headword(data)
		.. require("Module:utilities").format_categories(tracking_categories, lang)
end

pos_functions["adjectives"] = {
	params = {
		["g"] = {list = true}, --gender(s)
	},
	func = function(args, data, tracking_categories)
	end
}

local noun_params = {
	["g"] = {list = true}, --gender(s)
	["pl"] = {list = true}, --plural(s)
	["f"] = {list = true}, --feminine form(s)
	["m"] = {list = true}, --masculine form(s)
}

local allowed_genders = {
	["m"] = true,
	["f"] = true,
	["m-p"] = true,
	["f-p"] = true,
}

local function do_nouns(pos, args, data, tracking_categories)
	for _, g in ipairs(args.g) do
		if not allowed_genders[g] then
			error("Unrecognized gender: " .. g)
		end
	end

	local plpos = require("Module:string utilities").pluralize(pos)

	data.genders = args.g

	-- Check for special plural signals
	local mode = nil
	
	if args.pl[1] == "?" or args.pl[1] == "!" or args.pl[1] == "-" or args.pl[1] == "~" or args.pl[1] == "#" then
		mode = args.pl[1]
		table.remove(args.pl, 1)  -- Remove the mode parameter
	end
	
	if mode == "?" then
		-- Plural is unknown
		table.insert(data.categories, langname .. " " .. plpos .. " with unknown or uncertain plurals")
	elseif mode == "!" then
		-- Plural is not attested
		table.insert(data.inflections, {label = "plural not attested"})
		table.insert(data.categories, langname .. " " .. plpos .. " with unattested plurals")
		return
	elseif mode == "-" then
		-- Uncountable noun; may occasionally have a plural
		table.insert(data.categories, langname .. " uncountable " .. plpos)
		table.insert(data.inflections, {label = "usually " .. glossary_link("uncountable")})
	elseif mode == "~" then
		-- Mixed countable/uncountable noun, always has a plural
		table.insert(data.inflections, {label = glossary_link("countable") .. " and " .. glossary_link("uncountable")})
		table.insert(data.categories, langname .. " uncountable " .. plpos)
		table.insert(data.categories, langname .. " countable " .. plpos)
	elseif mode == "#" or pos == "noun" then
		-- Countable nouns; the default for regular nouns but not proper nouns
		if mode == "#" then
			table.insert(data.inflections, {label = glossary_link("countable")})
		end
		-- Not until we're sure that all nouns properly use pl=-
		-- table.insert(data.categories, langname .. " countable " .. plpos)
	end

	if #args.pl > 0 then
		args.pl.label = "plural"
		args.pl.accel = {form = "p"}
		table.insert(data.inflections, args.pl)
	end

	if #args.f > 0 then
		args.g.label = "feminine"
		table.insert(data.inflections, args.f)
	end

	if #args.m > 0 then
		args.m.label = "masculine"
		table.insert(data.inflections, args.m)
	end
end

-- Display additional inflection information for a noun
pos_functions["nouns"] = {
	params = noun_params,
	func = function(args, data, tracking_categories)
		return do_nouns("noun", args, data, tracking_categories)
	end,
}

pos_functions["proper nouns"] = {
	params = noun_params,
	func = function(args, data, tracking_categories)
		return do_nouns("proper noun", args, data, tracking_categories)
	end,
}

return export