Module:Mt-headword

-- Based on Module:ar-headword by: Benwing, CodeCat -- Adapted by Fenakhay

local lang = require("Module:languages").getByCode("mt")

local export = {} local pos_functions = {}

local u = mw.ustring.char

--- -- Utility functions -- ---

-- If Not Empty local function ine(arg) if arg == "" then return nil else return arg end end

local function list_to_set(list) local set = {} for _, item in ipairs(list) do		set[item] = true end return set end

-- version of mw.ustring.gsub that discards all but the first return value local function rsub(term, foo, bar) local retval = mw.ustring.gsub(term, foo, bar) return retval end

local rfind = mw.ustring.find

-- Tracking functions

local trackfn = require("Module:debug").track

local function track(page) trackfn("mt-headword/" .. page) return true end

local function append_cat(data, pos) table.insert(data.categories, lang:getCanonicalName .. " " .. pos) end

local function remove_links(text) text = rsub(text, "%[%^|%*|", "") text = rsub(text, "%[%[", "") text = rsub(text, "%]%]", "") return text end

local function make_unused_key_tracker(t) local unused_keys = require "Module:table".listToSet(require "Module:table".keysToList(t)) local mt = { __index = function(self, key) if key ~= nil then unused_keys[key] = nil end return t[key] end, __newindex = function(self, key, value) t[key] = value end }	local proxy_table = setmetatable({}, mt) return proxy_table, unused_keys end

-- The main entry point. function export.show(frame) local PAGENAME = mw.title.getCurrentTitle.text local poscat = frame.args[1] or error("Part of speech has not been specified. Please pass parameter 1 to the module invocation.") local params = { [1] = {list = "head", allow_holes = true, default = ""}, ["head"] = {default = ""} }	local args, unused_keys = make_unused_key_tracker(frame:getParent.args) -- Gather parameters local data = {lang = lang, pos_category = poscat, categories = {}, heads = {}, genders = {}, inflections = {}} local head = args["head"] or PAGENAME or "" local i = 1 while head do		table.insert(data.heads, head) i = i + 1 head = ine(args["head" .. i]) end if pos_functions[poscat] then pos_functions[poscat].func(args, data) end local unused_key_list = require "Module:table".keysToList(unused_keys) if #unused_key_list > 0 then local unused_key_string = require "Module:array"(unused_key_list) :map(function(key)					return "|" .. key .. "=" .. args[key]			end) :concat("\n") error("Unused arguments: " .. unused_key_string) end return require("Module:headword").full_headword(data) end

local function getargs(args, argpref, defgender, position) -- Gather parameters local forms = {}

local form

if ine(args[position]) then form = ine(args[position]) else form = ine(args[argpref]) end

local gender = ine(args[argpref .. "g"]) local gender2 = ine(args[argpref .. "g2"]) local i = 1 while form do		local genderlist = (gender or gender2) and {gender, gender2} or defgender and {defgender} or nil table.insert(forms, {term = form, gender = genderlist})

i = i + 1 form = ine(args[argpref .. i]) gender = ine(args[argpref .. i .. "g"]) gender2 = ine(args[argpref .. i .. "g2"]) end

return forms end

local function handle_infl(args, data, argpref, label, defgender, position) local newinfls = getargs(args, argpref, defgender, position) newinfls.label = label

if #newinfls > 0 then table.insert(data.inflections, newinfls) end end

local function handle_all_infl(args, data, argpref, label, nobase, position) if not nobase and argpref ~= "" then handle_infl(args, data, argpref, label, nil, position) end local labelsp = label == "" and "" or label .. " "	handle_infl(args, data, argpref .. "cons", labelsp .. "construct state") end

-- Handle the case where p=-, indicating an uncountable noun. local function handle_noun_plural(args, data) if args["p"] == "-" then table.insert(data.inflections, {label = "usually uncountable"}) append_cat(data, "uncountable nouns") else handle_infl(args, data, "p", "plural") handle_infl(args, data, "pauc", "paucal") end end

local valid_genders = list_to_set(	{"m", "m-s",	 "f", "f-s",	 "m-p", "f-p", "p",	 "d", "m-d", "f-d",   })

local function is_masc_sg(g) return g == "m" or g == "m-s" or g == "m-p" or g == "m-d" end local function is_fem_sg(g) return g == "f" or g == "f-s" or g == "f-p" or g == "f-d" end

local function handle_gender(args, data, default, nonlemma) local PAGENAME = mw.title.getCurrentTitle.text local g = ine(args["g"]) or default local g2 = ine(args["g2"])

local function process_gender(g) if not g then table.insert(data.genders, "?") elseif valid_genders[g] then table.insert(data.genders, g)		else error("Unrecognized gender: " .. g)		end end

process_gender(g) if g2 then process_gender(g2) end

if nonlemma then return end if g and g2 then append_cat(data, "terms with multiple genders") end end

-- Part-of-speech functions

pos_functions["adjectives"] = { func = function(args, data) handle_all_infl(args, data, "", "") handle_all_infl(args, data, "f", "feminine singular") handle_all_infl(args, data, "p", "plural") handle_all_infl(args, data, "dim", "diminutive") handle_all_infl(args, data, "comp", "comparative") end }

local function handle_sing_coll_noun_infls(args, data) handle_all_infl(args, data, "", "") handle_all_infl(args, data, "d", "dual") handle_noun_plural(args, data) end

pos_functions["collective nouns"] = { func = function(args, data) data.pos_category = "nouns" append_cat(data, "collective nouns") table.insert(data.inflections, {label = "collective"}) handle_gender(args, data, "m") -- Handle sing= (the corresponding singulative noun) and singg= (its gender) handle_infl(args, data, "sing", "singulative", "f") handle_sing_coll_noun_infls(args, data) end }

pos_functions["singulative nouns"] = { func = function(args, data) data.pos_category = "nouns" append_cat(data, "singulative nouns") table.insert(data.inflections, {label = "singulative"}) handle_gender(args, data, "f") -- Handle coll= (the corresponding collective noun) and collg= (its gender) handle_infl(args, data, "coll", "collective", "m") handle_sing_coll_noun_infls(args, data) end }

local function handle_noun_infls(args, data, singonly) handle_all_infl(args, data, "", "") if not singonly then handle_all_infl(args, data, "d", "dual") handle_noun_plural(args, data) handle_all_infl(args, data, "p", "plural", "nobase") handle_all_infl(args, data, "pauc", "paucal", "nobase") end handle_all_infl(args, data, "f", "feminine") handle_all_infl(args, data, "m", "masculine") if not singonly then handle_all_infl(args, data, "dim", "diminutive") end end

pos_functions["nouns"] = { func = function(args, data) handle_gender(args, data) handle_noun_infls(args, data) local g = ine(args["g"]) or default local g2 = ine(args["g2"]) if is_masc_sg(g) or is_masc_sg(g2) then append_cat(data, "masculine nouns") elseif is_fem_sg(g) or is_fem_sg(g2) then append_cat(data, "feminine nouns") end end }

pos_functions["numerals"] = { func = function(args, data) append_cat(data, "cardinal numbers") handle_gender(args, data) handle_noun_infls(args, data) end }

pos_functions["proper nouns"] = { func = function(args, data) handle_gender(args, data) handle_noun_infls(args, data, "singular only") end }

pos_functions["adjective feminine forms"] = { params = { ["g"] = {}, ["g2"] = {}, ["p"] = {}, ["islemma"] = { type = "boolean" }, },	func = function(args, data) data.pos_category = "adjective forms" handle_noun_plural(args, data) handle_gender(args, data, "f", "nonlemma") end }

pos_functions["adjective plural forms"] = { params = { ["g"] = {}, ["g2"] = {}, },	func = function(args, data) data.pos_category = "adjective forms" handle_gender(args, data, "p", "nonlemma") end }

pos_functions["noun forms"] = { params = { ["g"] = {}, ["g2"] = {}, },	func = function(args, data) handle_gender(args, data, nil, "nonlemma") end }

pos_functions["noun dual forms"] = { params = { ["g"] = {}, ["g2"] = {}, },	func = function(args, data) data.pos_category = "noun forms" append_cat(data, "noun dual forms") handle_gender(args, data, "d", "nonlemma") end }

pos_functions["verb"] = { func = function(args, data) data.pos_category = "verbs" handle_all_infl(args, data, "imper", "imperfect", nil, 1) handle_all_infl(args, data, "pp", "past participle", nil, 2) handle_all_infl(args, data, "ap", "active participle", nil, 3) end }

pos_functions["adverbs"] = { func = function(args, data) data.pos_category = "adverbs" handle_all_infl(args, data, "stem", "stem") end }

pos_functions["numeral forms"] = { func = function(args, data) data.pos_category = "numeral forms" handle_gender(args, data) end }

pos_functions["past participles"] = { func = function(args, data) data.pos_category = "participles" append_cat(data, "past participles") handle_all_infl(args, data, "f", "feminine") handle_all_infl(args, data, "p", "plural") end }

return export