View source for Module:pollasena-roots

local m_put = require("Module:parse utilities")
local m_families = require("Module:families")

local gsub = mw.ustring.gsub
local PAGENAME = gsub(mw.title.getCurrentTitle().text, "%s", "&nbsp;")
local NAMESPACE = mw.title.getCurrentTitle().nsText
local match = mw.ustring.match
local gmatch = mw.ustring.gmatch

local export = {}

local function ncategories(categories)
	local out_categories = {}
	for key, cat in ipairs(categories) do
		out_categories[key] = "[[Category:" .. cat .. "]]"
	end

	return NAMESPACE == "Template" and "" or table.concat(out_categories, "")
end

local function link(lang, term, alt)
	local natlang = lang:getWikidataItem() or false
	-- true if the language declaration has a wikidata item, a marginal amount of conlangs have it
	
	if lang:hasType("reconstructed") and not term:match("^%*") then
		term = "*" .. term
	end
	local data = {term = term, alt = alt or term, lang = lang, to_wikt = natlang}
	return require('Module:links').full_link(data)
end

function export.getLink(frame)
	local params = {}
	local a = 0
	
	if frame.args["offset"] or frame.args["older"] then
		local offset = frame.args["older"] and 1 or tonumber(frame.args["offset"])
		
		for i=1,offset do
			params[i] = {}
		end
		params[offset+1] = {list = true}
		a = offset + 1
	else
		params[1] = {list = true}
		a = 1
	end
	
	-- exceptions
	params["and"], params["see"] = {}, {}
	
	local args = require("Module:parameters").process(frame:getParent().args, params)
	
	local fam, lang = m_families.getByCode(frame.args[1]), require("Module:languages").getByCode(frame.args[2])
	local proto = fam:getProtoLanguage()

	local links = {}; local categories = {}; local alts = {};
	-- WIP
	for i, item in ipairs(args[a]) do
		if item == "+" then
			table.insert(links,"&nbsp;")
		elseif item == "-" then
			table.insert(links, item)
		elseif item:find("<") and not item:find("^[^<]*<[a-z]*[^a-z:]") then
			local run = m_put.parse_balanced_segment_run(item, "<", ">")
			local orig_param = a + i - 1
			local function parse_err(msg)
				error(msg .. ": " .. orig_param .. "= " .. table.concat(run))
			end
			local term = run[1]:gsub("%*","")

			for j = 2, #run - 1, 2 do
				local modtext = run[j]:match("^<(.*)>$")
				local prefix, arg = modtext:match("^([a-z]+):(.*)$")
				
				if item:match("><") then
					if prefix == "a" then
						table.insert(links, link(proto, term, arg))
						table.insert(alts, arg)
					elseif prefix == "c" then
						table.insert(categories, lang:getCanonicalName() .. " terms from " .. proto:getCanonicalName() .. " *" .. arg)
						table.insert(alts, term)
					end
				elseif prefix == "a" then
					table.insert(links, link(proto, term, arg))
					table.insert(categories, lang:getCanonicalName() .. " terms from " .. proto:getCanonicalName() .. " *" .. term)
					table.insert(alts, arg)
				elseif prefix == "c" then
					table.insert(links, link(proto, term))
					table.insert(categories, lang:getCanonicalName() .. " terms from " .. proto:getCanonicalName() .. " *" .. arg)
					table.insert(alts, term)
				end
			end
		else
			table.insert(links, link(proto, args[a][i]))
			table.insert(categories, lang:getCanonicalName() .. " terms from " .. proto:getCanonicalName() .. " *" .. args[a][i])
			table.insert(alts, args[a][i])
		end
		
		if i~=1 and links[i-1] == "-" then links[i] = links[i]:gsub("(%|?)%*","%1") end
		if i~=1 and links[i-1] ~= "&nbsp;" then links[i] = links[i]:gsub("(%|?)%*%-","%1-") end
		--if i~=1 and alts[i-1]:match("%-$") then links[i] = links[i]:gsub("%|%*","|") end
	end
	
	return table.concat(links) .. ncategories(categories)
end


return export