Module:kmr-headword

local export = {}
local pos_functions = {}

local m_scripts = require("Module:scripts")

local lang = require("Module:languages").getByCode("kmr")
local langname = lang:getCanonicalName()
local ku_sc = m_scripts.getByCode("ku-Arab")

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("kmr-headword/" .. page)
	return true
end

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

local function process_arabic_spellings(spellings)
	local inflection = {}
	for _, spelling in ipairs(spellings) do
		table.insert(inflection, {term = spelling, sc = ku_sc})
	end
	inflection.label = "Arabic spelling"
	return inflection
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("Plural part of speech e.g. 'nouns' has not been specified. Please pass parameter 1 to the module invocation.")
	
	local params = {
		["head"] = {list = true},
		["tr"] = {list = true, allow_holes = true},
		["id"] = {},
		["ar"] = {list = true},
		["suff"] = {type = "boolean"},
		["c"] = {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"],
		translits = args["tr"],
		genders = {},
		inflections = {},
		id = args["id"],
		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
	
	if pos_functions[poscat] then
		pos_functions[poscat].func(args, data, tracking_categories)
	end
	
	if #args["ar"] > 0 then
		table.insert(data.inflections, process_arabic_spellings(args["ar"]))
	end

	return require("Module:headword").full_headword(data)
		.. require("Module:utilities").format_categories(tracking_categories, lang)
end

pos_functions["verbs"] = {
	params = {
		[1] = {},
		[2] = {},
	},
	func = function(args, data, tracking_categories)
		if args[1] then
			table.insert(data.inflections, {label = "third-person singular simple present", args[1], accel = {form = '3|s|pres'}})
		end
		if args[2] then
			table.insert(data.inflections, {label = "past tense", args[2], accel = {form = 'past'}})
		end
	end
}

pos_functions["adjectives"] = {
	params = {
		[1] = {alias_of = "comparative"},
		["comparative"] = {list = true}, --comparative(s)
		[2] = {alias_of = "superlative"},
		["superlative"] = {list = true}, --superlative(s)
		["c"] = {type = "boolean"},
	},
	func = function(args, data, tracking_categories)
		if args.comparative[1] == "-" then
			table.insert(data.inflections, {label = "not comparable"})
			table.insert(data.categories, langname .. " uncomparable adjectives")
		else
			if #args.comparative > 0 then
				args.comparative.label = glossary_link("comparative")
				args.comparative.accel = {form = "comparative"}
				table.insert(data.inflections, args.comparative)
			else 
				if args["c"] then
					table.insert(data.inflections, {label = glossary_link("comparative"), accel = {form = 'comparative'}, {term = PAGENAME ..'tir'}})
				end
			end
		
			if #args.superlative > 0 then
				args.superlative.label = glossary_link("superlative")
				args.superlative.accel = {form = "superlative"}
				table.insert(data.inflections, args.superlative)
			else
				if args["c"] then
					table.insert(data.inflections, {label = glossary_link("superlative"), accel = {form = 'superlative'}, {term = '[[herî]] ' .. PAGENAME}, {term = PAGENAME ..'tirîn'}})
				end
			end
		end
	end
}

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

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

local function do_nouns(pos, args, data, tracking_categories)
	local genders = {}
	for _, g in ipairs(args.g) do
		if not allowed_genders[g] then
			error("Unrecognized gender: " .. g)
		end
		if g == "mf" then
			table.insert(genders, "m")
			table.insert(genders, "f")
		else
			table.insert(genders, g)
		end
	end

	if #genders > 0 then
		data.genders = genders
	else
		data.genders = {"?"}
	end
	
	local plpos = require("Module:string utilities").pluralize(pos)

	-- 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] == "~" 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 = glossary_link("uncountable")})
	elseif mode == "-~" then
		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 enough nouns properly use - in the second argument
		-- 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.f.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

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