Module:tmh-headword

local list_to_set = require("Module:table").listToSet

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

local export = {}
local pos_functions = {}

local dialects = {
	 = { "Ghat" },
	 = { "Tamahaq" },
	 = { "Tamasheq" },
	 = { "Tawellemmet" },
	 = { "Tayert" },
}

local function parse_qualifiers(quals)
	if quals == nil then
		return nil
	end

	local list = mw.text.split(quals, ',%s*')
	local qualifiers = { }

	for _, qual in ipairs(list) do
		if dialects then
			table.insert(qualifiers, dialects)
		else
			table.insert(qualifiers, qual)
		end
	end

	return table.concat(qualifiers, ", ")
end

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

-- If Not Empty
local function ine(arg)
	if arg == "" then
		return nil
	else
		return arg
	end
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 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 = list_to_set(
			require "Module:table".keysToList(t))
	local mt = {
		__index = function(_, key)
			if key ~= nil then
				unused_keys = nil
			end
			return t
		end,
		__newindex = function(_, 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 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 PAGENAME or ""
	local translit = ine(args)
	local i = 1

	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, position)
	-- Gather parameters
	local forms = {}

	local form

	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 qualifiers = 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, gender = genderlist, qualifiers = parse_qualifiers(qualifiers) })

		i = i + 1
		form = ine(args)
		translit = ine(args)
		gender = ine(args)
		gender2 = ine(args)
		qualifiers = 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 .. " "
	handle_infl(args, data, argpref .. "cons", labelsp .. "construct state")
end

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")
	end
end

local valid_genders = list_to_set{ "m", "f", "m-p", "f-p", "p" }

local function handle_gender(args, data, default, nonlemma)
	local g = ine(args) or default
	local g2 = ine(args)

	local function process_gender(gender)
		if not gender then
			table.insert(data.genders, "?")
		elseif valid_genders then
			table.insert(data.genders, gender)
		else
			error("Unrecognized gender: " .. gender)
		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 = {
	func = function(args, data)
		handle_all_infl(args, data, "", "")
		handle_all_infl(args, data, "f", "feminine")
		handle_all_infl(args, data, "pl", "masculine plural")
		handle_all_infl(args, data, "fpl", "feminine plural")
	end
}

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

pos_functions = {
	func = function(args, data)
		handle_gender(args, data)
		handle_noun_infls(args, data)
	end
}

pos_functions = {
	func = function(args, data)
		data.pos_category = "verbs"
		handle_all_infl(args, data, "imperf", "imperfective", nil, 1)
		handle_all_infl(args, data, "perf", "perfective", nil, 2)
	end
}

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

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

pos_functions = {
	params = {  = {} },
	func = function(args, data)
		handle_gender(args, data)
		handle_all_infl(args, data, "sa", "subject affix")
	end
}

return export