local export = {}

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

local function otherscript(inflections, args)
	local title = mw.title.getCurrentTitle()
	local sc = lang:findBestScript(title.subpageText)
	
	local other_sc
	
	if sc:getCode() == "Latn" then
		other_sc = "Cyrl"
	elseif sc:getCode() == "Cyrl" then
		other_sc = "Latn"
	end
	
	other_sc = require("Module:scripts").getByCode(other_sc)
	local inflection = {label = other_sc:getCategoryName()}
	
	local heads = args["head"]
	if #heads == 0 then
		heads = {title.subpageText}
	end
	for i, head in ipairs(heads) do
		local tr = args["tr"][i]
		
		if not tr then
			tr = require("Module:sh-translit").tr(require("Module:links").remove_links(head), "sh", sc:getCode())
		end
		
		table.insert(inflection, {term = tr, sc = other_sc})
	end
	
	table.insert(inflections, inflection)
end

local function glossary_link(entry, text)
	text = text or entry
	return "[[ภาคผนวก:อภิธานศัพท์#" .. entry .. "|" .. text .. "]]"
end

function export.basic(frame)
	local params = {
		["head"] = {list = true},
		["tr"] = {list = true, allow_holes = true},
		}
	
	local args = require("Module:parameters").process(frame:getParent().args, params)
	
	local data = {lang = lang, pos_category = frame.args[1], categories = {}, heads = args["head"], genders = {}, inflections = {}}
	
	otherscript(data.inflections, args)
	
	return require("Module:headword").full_headword(data)
end


function export.adjective(frame)
	local params = {
		["comp"] = {},
		["def"] = {},
		["head"] = {list = true},
		["tr"] = {list = true, allow_holes = true},
		}
	
	local args = require("Module:parameters").process(frame:getParent().args, params)
	
	local data = {lang = lang, pos_category = "คำคุณศัพท์", categories = {}, heads = args["head"], genders = {}, inflections = {}}
	
	if args["def"] then
		table.insert(data.inflections, {label = "ชี้เฉพาะ", args["def"]})
	end
	
	if args["comp"] then
		table.insert(data.inflections, {label = "ขั้นกว่า", args["comp"]})
	end
	
	otherscript(data.inflections, args)
	
	return require("Module:headword").full_headword(data)
end


local gender_cats = {
	["m"] = "เพศชาย",
	["f"] = "เพศหญิง",
	["n"] = "เพศกลาง",
	["m-p"] = "เพศชาย",
	["f-p"] = "เพศหญิง",
	["n-p"] = "เพศกลาง",
}

function export.letter(frame)
	local params = {
		["upper"] = {},
		["lower"] = {},
		["head"] = {list = true},
		["tr"] = {list = true, allow_holes = true},
		}
	
	local args = require("Module:parameters").process(frame:getParent().args, params)
	
	local data = {lang = lang, pos_category = "ตัวอักษร", categories = {}, heads = args["head"], genders = {}, inflections = {}}
	
	if args["upper"] then
		table.insert(data.inflections, {label = "อักษรตัวเล็ก", nil})
		table.insert(data.inflections, {label = "อักษรตัวใหญ่", args["upper"]})
	elseif args["lower"] then
		table.insert(data.inflections, {label = "อักษรตัวใหญ่", nil})
		table.insert(data.inflections, {label = "อักษรตัวเล็ก", args["lower"]})
	end
	
	otherscript(data.inflections, args)
	
	return require("Module:headword").full_headword(data)
end


function export.noun(frame)
	local params = {
		["g"] = {list = true, default = "?"},
		["head"] = {list = true},
		["tr"] = {list = true, allow_holes = true},
		["m"] = {list = true},
		["f"] = {list = true},
        ["dim"] = {list = true},
        ["aug"] = {list = true},
        ["adj"] = {list = true}
		}
	
	local args = require("Module:parameters").process(frame:getParent().args, params)
	
	local data = {lang = lang, pos_category = "คำนาม", categories = {}, heads = args["head"], genders = args["g"], inflections = {}}
	
	for i, gender in ipairs(data.genders) do
		if gender_cats[gender] then
			--table.insert(data.categories, lang:getCanonicalName() .. " " .. gender_cats[gender] .. " nouns")
			table.insert(data.categories, "คำนาม" .. gender_cats[gender] .. lang:getCategoryName())
		else
			data.genders[i] = "?"
		end
	end
	
	otherscript(data.inflections, args)
	
	-- Add parameters
	for _, val in pairs {{ "f", "เพศหญิง" }, { "m", "เพศชาย" }, {"dim", "บอกความเล็ก"}, {"aug", "บอกความใหญ่"}, {"adj", "คำคุณศัพท์สัมพันธ์"}} do
		local param_name, label = unpack(val)
		local forms = args[param_name]
		if forms[1] then
			if param_name == "adj" then
				forms.label = glossary_link("สัมพันธ์", label)
			else
				forms.label = glossary_link(label)
			end
			table.insert(data.inflections, forms)
		end
	end
	
	return require("Module:headword").full_headword(data)
end


function export.propernoun(frame)
	local params = {
		["g"] = {list = true, default = "?"},
		["head"] = {list = true},
		["tr"] = {list = true, allow_holes = true},
		}
	
	local args = require("Module:parameters").process(frame:getParent().args, params)
	
	local data = {lang = lang, pos_category = "คำวิสามานยนาม", categories = {}, heads = args["head"], genders = args["g"], inflections = {}}
	
	for i, gender in ipairs(data.genders) do
		if gender_cats[gender] then
			--table.insert(data.categories, lang:getCanonicalName() .. " " .. gender_cats[gender] .. " nouns")
			table.insert(data.categories, "คำนาม" .. gender_cats[gender] .. lang:getCategoryName())
		else
			data.genders[i] = "?"
		end
	end
	
	otherscript(data.inflections, args)
	
	return require("Module:headword").full_headword(data)
end


function export.verb(frame)
	local params = {
		["a"] = {},
		["head"] = {list = true},
		["tr"] = {list = true, allow_holes = true},
		["pf"] = {list = true},
		["impf"] = {list = true}
		}
	
	local args = require("Module:parameters").process(frame:getParent().args, params)
	
	local data = {lang = lang, pos_category = "คำกริยา", categories = {}, heads = args["head"], genders = {}, inflections = {}}
	
	if args["a"] == "impf" or args["a"] == "pf" then
		table.insert(data.genders, args["a"])
	elseif args["a"] == "impf-pf" or args["a"] == "pf-impf" or args["a"] == "dual" or args["a"] == "ip" then
		table.insert(data.genders, "impf")
		table.insert(data.genders, "pf")
	else
		table.insert(data.genders, "?")
	end
	
	otherscript(data.inflections, args)
	
	--add perfective equivalent
	local pf = args["pf"]
	if #pf > 0 then
		pf.label = "สมบูรณ์"
		table.insert(data.inflections, pf)
	end
	
	--add imperfective equivalent
	local impf = args["impf"]
	if #impf > 0 then
		impf.label = "ไม่สมบูรณ์"
		table.insert(data.inflections, impf)
	end
	
	return require("Module:headword").full_headword(data)
end


return export