local lang = require("Module:languages").getByCode("aii")
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 = true
end
return set
end
-- version of mw.ustring.gsub() that discards all but the first return value
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
function track(page)
trackfn("tru-headword/" .. page)
return true
end
local function append_cat(data, pos)
table.insert(data.categories, lang:getCanonicalName() .. " " .. pos)
end
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 = nil end
return t
end,
__newindex = function(self, key, value) t = 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 or error("Part of speech has not been specified. Please pass parameter 1 to the module invocation.")
local params = {
= {list = "head", allow_holes = true, default = ""},
= {default = ""},
= {list = true, allow_holes = true},
}
local args, unused_keys = make_unused_key_tracker(frame:getParent().args)
-- Gather parameters
local data = {lang = lang, pos_category = poscat, categories = {}, heads = {}, translits = {}, genders = {}, inflections = { enable_auto_translit = true }}
local head = args or args or PAGENAME or ""
local translit = ine(args)
local i = 1
local irreg_translit = false
while head do
table.insert(data.heads, head)
data.translits = translit
i = i + 1
head = ine(args)
translit = ine(args)
end
if pos_functions then
pos_functions.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
end):concat("\n")
error("Unused arguments: " .. unused_key_string)
end
return require("Module:headword").full_headword(data)
end
local function getargs(args, argpref, defgender)
-- Gather parameters
local forms = {}
if ine(args) then
form = ine(args)
else
form = ine(args)
end
local translit = ine(args)
local gender = ine(args)
local gender2 = ine(args)
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, translit = translit, genders = genderlist})
i = i + 1
form = ine(args)
translit = ine(args)
gender = ine(args)
gender2 = ine(args)
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 .. " "
end
-- Handle the case where pl=-, indicating an uncountable noun.
local function handle_noun_plural(args, data)
if args == "-" then
table.insert(data.inflections, {label = "usually ]"})
append_cat(data, "uncountable nouns")
else
handle_infl(args, data, "pl", "plural")
handle_infl(args, data, "pauc", "paucal")
end
end
local valid_genders = list_to_set(
{"s",
"m", "m-s",
"f", "f-s",
"m-p", "f-p", "p",
"mf"
})
local function is_masc_sg(g)
return g == "m" or g == "m-s" or g == "m-p" or g == "mf"
end
local function is_fem_sg(g)
return g == "f" or g == "f-s" or g == "f-p" or g == "mf"
end
local function handle_gender(args, data, default, nonlemma)
local PAGENAME = mw.title.getCurrentTitle().text
local g = ine(args) or ine(args) or default
local g2 = ine(args)
local function process_gender(g)
if not g then
table.insert(data.genders, "?")
elseif valid_genders 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
end
-- Part-of-speech functions
function handle_noun_infls(args, data, singonly)
handle_all_infl(args, data, "", "")
if not singonly then
handle_noun_plural(args, data)
handle_all_infl(args, data, "pl", "plural", "nobase")
end
handle_all_infl(args, data, "f", "feminine")
handle_all_infl(args, data, "m", "masculine")
end
--- Lemma forms
pos_functions = {
func = function(args, data)
handle_gender(args, data)
handle_noun_infls(args, data)
local g = ine(args) or default
local g2 = ine(args)
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 = {
func = function(args, data)
append_cat(data, "cardinal numbers")
handle_gender(args, data)
handle_noun_infls(args, data)
handle_all_infl(args, data, "cstr", "construct")
end
}
pos_functions = {
func = function(args, data)
handle_gender(args, data)
end
}
pos_functions = {
params = {
= {},
= {}
},
func = function(args, data)
if ine(args) or ine(args) then
handle_gender(args, data)
end
handle_all_infl(args, data, "s", "singular")
handle_all_infl(args, data, "m", "masculine")
handle_all_infl(args, data, "ms", "masculine singular")
handle_all_infl(args, data, "f", "feminine")
handle_all_infl(args, data, "fs", "feminine singular")
handle_all_infl(args, data, "pl", "plural")
handle_all_infl(args, data, "mpl", "masculine plural")
handle_all_infl(args, data, "fpl", "feminine plural")
end
}
pos_functions = {
params = {
= {},
= {}
},
func = function(args, data)
if ine(args) or ine(args) then
handle_gender(args, data)
end
handle_all_infl(args, data, "s", "singular")
handle_all_infl(args, data, "m", "masculine")
handle_all_infl(args, data, "ms", "masculine singular")
handle_all_infl(args, data, "f", "feminine")
handle_all_infl(args, data, "fs", "feminine singular")
handle_all_infl(args, data, "pl", "plural")
handle_all_infl(args, data, "mpl", "masculine plural")
handle_all_infl(args, data, "fpl", "feminine plural")
end
}
pos_functions = {
params = {
= {},
= {}
},
func = function(args, data)
if ine(args) or ine(args) then
handle_gender(args, data)
end
handle_all_infl(args, data, "f", "feminine")
handle_all_infl(args, data, "pl", "plural")
end
}
pos_functions = {
func = function(args, data)
--
end
}
pos_functions = {
func = function(args, data)
handle_all_infl(args, data, "f", "feminine")
handle_all_infl(args, data, "pl", "plural")
handle_all_infl(args, data, "mpl", "masculine plural")
handle_all_infl(args, data, "fpl", "feminine plural")
end
}
pos_functions = {
params = {
= {},
= {}
},
func = function(args, data)
if ine(args) or ine(args) then
handle_gender(args, data)
end
handle_all_infl(args, data, "f", "feminine")
handle_all_infl(args, data, "pl", "plural")
end
}
--- Non-lemma forms
pos_functions = {
func = function(args, data)
--
end
}
pos_functions = {
func = function(args, data)
handle_all_infl(args, data, "f", "feminine")
handle_all_infl(args, data, "pl", "plural")
end
}
return export