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

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

local export = {}
local pos_functions = {}

local u = mw.ustring.char

local TAM = u(0x0629)
local ALIF = u(0x0627) -- ʾalif = ا

-----------------------
-- 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
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("ary-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

-- 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 = ""},
		["tr"] = {list = true, allow_holes = true},
	}
	
	local args = frame:getParent().args
	
	-- Gather parameters
	local data = {lang = lang, pos_category = poscat, categories = {}, heads = {}, translits = {}, genders = {}, inflections = {}}
	
	local head = args["head"] or PAGENAME or ""
	local translit = ine(args["tr"])
	local i = 1
	
	local irreg_translit = false
	
	while head do
		table.insert(data.heads, head)
		data.translits[#data.heads] = translit
		
		i = i + 1
		head = ine(args["head" .. i])
		translit = ine(args["tr" .. i])
	end
	
	if pos_functions[poscat] then
		pos_functions[poscat].func(args, data)
	end
	
	return require("Module:headword").full_headword(data)
end

-- Get a list of inflections. See handle_infl() for meaning of ARGS, ARGPREF
-- and DEFGENDER.
local function getargs(args, argpref, defgender)
	-- Gather parameters
	local forms = {}

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

	local translit = ine(args[argpref .. "tr"])
	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, translit = translit, genders = genderlist})

		i = i + 1
		form = ine(args[argpref .. i])
		translit = ine(args[argpref .. i .. "tr"])
		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 .. "結構態")
end

-- Handle the case where pl=-, indicating an uncountable noun.
local function handle_noun_plural(args, data)
	if args["pl"] == "-" then
		table.insert(data.inflections, {label = "一般不可數"})
		append_cat(data, "不可數名詞")
	else
		handle_infl(args, data, "pl", "複數")
	end
end

local valid_genders = list_to_set(
	{"s", 
     "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

-- Handle gender in unnamed param 2 and a second gender in param g2, inserting
-- into the list of genders in GENDER. Also, if a lemma, insert categories
-- into CATS if the gender is unexpected for the form of the noun or if multiple
-- genders occur. If gender unspecified, default to DEFAULT, which may be
-- omitted.
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
end

-- Part-of-speech functions

pos_functions["形容詞"] = {
	func = function(args, data)
		handle_all_infl(args, data, "", "")
		handle_all_infl(args, data, "f", "陰性")
		handle_all_infl(args, data, "cpl", "通性複數")
		handle_all_infl(args, data, "pl", "陽性複數")
		handle_all_infl(args, data, "fpl", "陰性複數")
		handle_all_infl(args, data, "dim", "指小詞")
		handle_infl(args, data, "el", "絕對最高級")
	end
}

function handle_sing_coll_noun_infls(args, data)
	handle_all_infl(args, data, "", "")
	handle_all_infl(args, data, "d", "雙數")
	handle_all_infl(args, data, "pauc", "眾數")
	handle_noun_plural(args, data)
	handle_all_infl(args, data, "pl", "複數", "nobase")
end

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

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

function handle_noun_infls(args, data, singonly)
	handle_all_infl(args, data, "", "")
	
	if not singonly then
		handle_all_infl(args, data, "d", "雙數")
		handle_noun_plural(args, data)
		handle_all_infl(args, data, "pl", "複數", "nobase")
	end
	
	handle_all_infl(args, data, "f", "陰性")
	handle_all_infl(args, data, "m", "陽性")
	
	if not singonly then
		handle_all_infl(args, data, "dim", "指小詞")
	end
end

pos_functions["名詞"] = {
	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, "陽性名詞")
		elseif is_fem_sg(g) or is_fem_sg(g2) then
			append_cat(data, "陰性名詞")
		end
	end
}

pos_functions["數詞"] = {
	func = function(args, data)
		append_cat(data, "基數詞")
		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, "僅單數")
	end
}

pos_functions["代詞"] = {
	params = {
		["g"] = {},
		["g2"] = {}
		},
	func = function(args, data)
        if ine(args["g"]) or ine(args["g2"]) then
            handle_gender(args, data)
        end
		handle_all_infl(args, data, "f", "陰性")
	end
}

pos_functions["adjective feminine forms"] = {
	params = {
		["g"] = {},
		["g2"] = {},
		["pl"] = {},
		["islemma"] = { type = boolean },
		},
	func = function(args, data)
		data.pos_category = "形容詞變格形"
		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 = "形容詞變格形"
		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 = "名詞變格形"
		append_cat(data, "名詞雙數")
		handle_gender(args, data, "m-d", "nonlemma")
	end
}

pos_functions["active participles"] = {
	params = {
		[2] = {},
		},
	func = function(args, data)
		data.pos_category = "分詞"
		append_cat(data, "主動分詞")
		handle_all_infl(args, data, "", "")
		handle_all_infl(args, data, "f", "陰性")
		handle_all_infl(args, data, "cpl", "通性複數")
		handle_all_infl(args, data, "pl", "陽性複數")
		handle_all_infl(args, data, "fpl", "陰性複數")
	end
}

pos_functions["passive participles"] = {
	params = {
		[2] = {},
		},
	func = function(args, data)
		data.pos_category = "分詞"
		append_cat(data, "被動分詞")
		handle_all_infl(args, data, "", "")
		handle_all_infl(args, data, "f", "陰性")
		handle_all_infl(args, data, "cpl", "通性複數")
		handle_all_infl(args, data, "pl", "陽性複數")
		handle_all_infl(args, data, "fpl", "陰性複數")
	end
}

pos_functions["動詞"] = {
	func = function(args, data)
		data.pos_category = "動詞"
		handle_all_infl(args, data, "np", "非過去式")
		if ine(args[1]) then
            table.insert(data.inflections, {label = '[[Appendix:阿拉伯語動詞#類 ' .. args[1] .. '|第' .. args[1] .. '類]]'})
		    append_cat(data, "第" .. args[1] .. "類動詞")
        else 
            append_cat(data, "verbs lacking forms")
        end
    end
}

pos_functions["限定詞"] = {
	params = {
		["g"] = {},
		["g2"] = {}
		},
	func = function(args, data)
        if ine(args["g"]) or ine(args["g2"]) then
            handle_gender(args, data)
        end
		handle_all_infl(args, data, "f", "陰性")
        handle_all_infl(args, data, "pl", "複數")
	end
}

return export