Module:descendants

Underlies {{descendant}} and {{descendant tree}}.

---

local export = {}
local getArgs = require("Module:Arguments").getArgs
local m_links = require("Module:links")
local m_languages = require("Module:languages")
local m_inline = require("Module:inline")

function export.descendant(frame)
	local args = getArgs(frame)
	local out = ""
	local lang = args[1]
	if mw.ustring.find(lang, "[%>%_]") then
		lang = mw.ustring.match(lang, "[%>%_]([^%>%_]+)$")
	end
	if mw.ustring.find(lang, "[0-9]+$") then
		lang = mw.ustring.gsub(lang, "[0-9]+$", "")
	end
	lang = m_languages.get_by_code(lang)
	local borrowed = args["borrowed"] or args["bor"] or args["b"]
	local see_desc = args["see"]
	local noname = args["noname"]
	local lostcap = args["lost"] or "lost"
	
	if borrowed then out = "<span class=\"desc-arrow\" title=\"borrowed\">→</span>" .. out end
	if not noname then out = out .. lang.name .. ":&nbsp;" end
	
	if args[2] then
		local terms = {}
		local i = 2
		while args[i] do
			local term, term_args = m_inline.parse(args[i])
			table.insert(terms, m_links.full_link({
					language = lang,
					term = term,
					alt = term_args.alt,
					anchor = term_args.anchor or term_args.a,
					gloss = term_args.t,
					pos = term_args.pos,
					nobold = true,
				})
			)
			i = i + 1	
		end
		out = out .. table.concat(terms, ", ")
		if see_desc then out = out .. " " .. frame:expandTemplate{ title = 'see descendants' , args = {}} end
	else
		out = out .. "— (''" .. lostcap .. "'')"
	end
	
	return out
end

function export.langtree(list)
	
	-- build in-set ancestry
	local ancestors = {}
	for _, lang in ipairs(list) do
		ancestors[lang] = {}
		if mw.ustring.find(lang, "[%>%_]") then
			local ancestor_temp = ""
			for piece in mw.ustring.gmatch(lang, "([^%>%_]+[%>%_])") do
				ancestor_temp = ancestor_temp .. piece
				ancestor_add = mw.ustring.gsub(ancestor_temp, "[%>%_]$", "")
				table.insert(ancestors[lang], ancestor_add)
			end
		elseif mw.ustring.find(lang, "[0-9]+$") then
			local lang_temp, index = mw.ustring.match(lang, "^([^0-9]+)([0-9]+)$")
			for _, test in ipairs(list) do
				if mw.ustring.find(test, index .. "$") then
					local test_temp = mw.ustring.gsub(test, index .. "$", "")
					if m_languages.get_by_code(lang_temp).lineage[test_temp] then
						table.insert(ancestors[lang], test)
					end
				elseif m_languages.get_by_code(lang_temp).lineage[test] then
					local nope = false
					for _, contain_test in ipairs(list) do
						if contain_test == test .. index then nope = true end	
					end
					if not nope then table.insert(ancestors[lang], test) end
				end
			end
		else
			for _, test in ipairs(list) do
				if m_languages.get_by_code(lang).lineage[test] then
					table.insert(ancestors[lang], test)
				end
			end
		end
	end
	
	local structure = {}
	for lang, _ in pairs(ancestors) do
		if #ancestors[lang] == 0 then
			structure[lang] = {}
			ancestors[lang] = nil
		end
	end
	
	local swept = {}
	local function search_and_sort(superordinate)
		for mother, it in pairs(superordinate) do
			for lang, lang_ancestors in pairs(ancestors) do
				for i, lang_ancestor in ipairs(ancestors[lang]) do
					if swept[lang_ancestor] then table.remove(ancestors[lang], i) end
				end
			end
			for lang, lang_ancestors in pairs(ancestors) do
				if #lang_ancestors == 1 and lang_ancestors[1] == mother then
					superordinate[mother][lang] = {}
					swept[mother] = true
					ancestors[lang] = nil
				end
			end
			search_and_sort(it)
		end
	end
	
	search_and_sort(structure)
	return structure

end

function export.desctree(frame)
	local args = getArgs(frame)
	local out = ""
	local current_lang, code
	local terms = {}
	local langs = {}
	local indenter = args["indent"] or args["in"] or ""
	
	for i, term in ipairs(args) do --ryn:term1|term2|term3|rad:term1|term2 ...
		local lang_change = mw.ustring.match(term, "^([^%:%<]+)%:")
		local lang_change_arged = mw.ustring.match(term, "^(.+%>)%:")
		local from_lang, index
		if i == 1 and not (lang_change or lang_change_arged) then error("Language code needs to be specified in the first parameter as xyz(<bor>):term₁") end
		if lang_change or lang_change_arged then
			if lang_change then
				current_lang = lang_change
				term = mw.ustring.gsub(term, "^[^%:]+%:%s*", "")
			elseif lang_change_arged then
				current_lang = mw.ustring.match(lang_change_arged, "^[^%<]+")
				term = mw.ustring.gsub(term, "^.+%>%:%s*", "")
			end
			if mw.ustring.find(current_lang, "[%>%_][^%:]") then
				from_lang, current_lang = mw.ustring.match(current_lang, "^(.*[%>%_])([^%>%_%:]+)$")
				if mw.ustring.match(from_lang, "%>$") then
					if lang_change_arged then
						lang_change_arged = lang_change_arged .. "<bor>"
					else
						lang_change_arged = "<bor>"
					end
				end
			end
			if mw.ustring.find(current_lang, "[0-9]+$") then
				current_lang, index = mw.ustring.match(current_lang, "^(.*)([0-9]+)$")
			end
			current_lang = m_languages.get_by_code(current_lang)
			code = current_lang.code
			if from_lang then code = from_lang .. code end
			if index then code = code .. index end
			if terms[code] == nil then
				terms[code] = {}
				table.insert(langs, code)
			end
		end
		if lang_change_arged then
			for lang_arg in mw.ustring.gmatch(lang_change_arged, "%<([^%>]+)%>") do
				if mw.ustring.find(lang_arg, "%:") then
					local before, after = mw.ustring.match(lang_arg, "^([^%:]+)%:([^%:]+)$")
					terms[code][before] = after
				else
					terms[code][lang_arg] = true
				end
			end
		end
		if from_lang then terms[code]["from"] = from_lang end
		table.insert(terms[code], term)
	end
	
	local items = {}
	
	for lang, lang_args in pairs(terms) do
		table.insert(lang_args, 1, lang)
		items[lang] = export.descendant(lang_args)
	end
	
	-- BEGIN TRIAGE --
	
	local structure = export.langtree(langs)
	
	local outs = {}
	
	local it = 0
	function process_structure(tab)
		it = it + 1
		
		local sorter = {}
		local sorter_proto = {}
		local sorter_bor = {}
		local sorter_bor_proto = {}
		for key, _ in pairs(tab) do
			if mw.ustring.find(key, "%-pro[0-9]*$") then
				if mw.ustring.find(key, "%>[^%>%_%:]+$") then
					table.insert(sorter_bor_proto, key)
				else
					table.insert(sorter_proto, key)
				end
			else
				if mw.ustring.find(key, "%>[^%>%_%:]+$") then
					table.insert(sorter_bor, key)
				else
					table.insert(sorter, key)
				end
			end
		end
		table.sort(sorter)
		table.sort(sorter_proto)
		table.sort(sorter_bor)
		table.sort(sorter_bor_proto)
		for _, key in ipairs(sorter_bor) do table.insert(sorter, key) end
		for _, key in ipairs(sorter_proto) do table.insert(sorter, key) end
		for _, key in ipairs(sorter_bor_proto) do table.insert(sorter, key) end
		
		for _, key in ipairs(sorter) do
			local out = " " .. items[key]
			for i = 1, it do
				out = "*" .. out	
			end
			out = indenter .. out
			table.insert(outs, out)
			process_structure(tab[key])
		end
		it = it - 1
	end
	
	process_structure(structure)
	
	local render = table.concat(outs, "\n")
	
	render = "<div class=\"ul-columns\">\n" .. render .. "\n</div>"
	
	return render
	
end

return export

--[[
Debug console test string:
=p.descendant(mw.getCurrentFrame():newChild{title="whatever",args={"rad"}})
=p.desctree(mw.getCurrentFrame():newChild{title="whatever",args={"ryn>rad-o:1", "aeg:2", "3", "ryn:4", "ryn>rad-o_rad:5", ">rad-pro:6", "hrd-pro:7", "ryn-o:8", "lfv-pro:9", "lfv-o:10", "lfv:11", "rad-pro:12"}})
]]--