Module:languages

Revision as of 22:53, 6 August 2024 by TheNightAvl (talk | contribs)
local export = {}

function export.get_by_code(code)
	local lang = export.find_by_code(code)
	if lang == nil then error("No such language (" .. code .. ")") end
	return lang
end

function export.find_by_code(code)
	if type(code) ~= "string" then
		error("Language code must be string. Got " .. type(code))
	end
	return mw.loadData("Module:languages/data")[code]
end

function export.get_by_name(name)
	local lang = export.find_by_canonical_name(name)
	if lang == nil then error("No such language (" .. name .. ")") end
	return lang
end

function export.find_by_name(name)
	local code = mw.loadData("Module:languages/names")[name]
	if code == nil then return nil end
	return export.find_by_code(code)
end

function export.get_field_by_code(code, field)
	if type(field) ~= "string" then
		error("Desired field must be a string")
	end
	local l = export.get_by_code(code)
	return l[field]
end

function export.is_ancestor(language, code)
	if language.lineage == nil then return false end
	return language.lineage[code] ~= nil
end

function export.distance(a, b)
	a, b = export.get_by_code(a), export.get_by_code(b)
	local i = 0
	local count = 0
	
	if a.lineage == nil or b.lineage == nil then return nil end
	
	for ancestor, _ in pairs(a.lineage) do
		count = count + 1
		if b.lineage[ancestor] then i = i + 1 end
	end
	
	if i > 0 then return count - i else return nil end 
end

function export.getByCode(frame)
	local code = frame.args[1]
	local field = frame.args[2]
	if type(field) == "string" then
		return export.get_field_by_code(code, field)
	end
	return export.get_by_code(code)
end

return export