模組:Language/name

local getArgs = require("Module:Arguments").getArgs
local p, e = {}, {}

local function __compile(...)
	local r,i = "",0
	for _, v in ipairs(arg) do
		i = i + 1
		if i == 1 then
			r = v[1]
		elseif i == 2 then
			r = r .. " (" .. v[1]
		else
			r =  r .. ", " .. v[1]
		end
	end
	if i > 1 then
		r = r .. ")"
	end
	return r
end

function e.simple(args, data)
	return __compile(data.lang[args.code])
end

function e.fuzzy(args, data)
	-- split lang code at '-', starting from the end, one dash at a time
	-- and try to find a match; break from loop when we do
	repeat
		if data.lang[args.code] then
			break
		end
		args.code = args.code:gsub("-[^-]*$", "")
	until not args.code:match("-")

	return __compile(data.lang[args.code])
end

function e.formal(args, data)
	-- split the whole lang code up at every '-' and stick the bits in an array
	local bits = {}; for match in args.code:gmatch("[^-]+") do
		bits[#bits+1] = match
	end

	-- if length of the 2nd array item is 4, assume it's a script code,
	-- otherwise a region code
	if bits[2] and #bits[2] == 4 then
		return __compile(data.lang[bits[1]],
						 data.script[bits[2]],
						 data.region[bits[3]]
						)
	else
		return __compile(data.lang[bits[1]],
						 data.region[bits[2]]
						)
	end
end

for function_name in pairs(e) do
	p[function_name] = function (frame)
		local args = getArgs(frame)
		local data = mw.loadData("Module:Language/name/data")

		args.code = args.code:lower()

		-- if dataset=iana, discard wp language table
		if args.dataset == "iana" then
			data.lang = data.lang_iana
		end

		return e[function_name](args, data)
	end
end

return p