Module:Interwiki extra/data

-- This module processes data from mw.site.interwikiMap for use with
-- [[Module:Interwiki]]. By loading the data with mw.loadData we ensure that
-- this is only done once per page parse.

local prefixes, urls, hosts = {}, {}, {}

local function makeCountedData(t, storeKey, counterKey, itemKey, item)
	t = t or {}
	t[storeKey] = t[storeKey] or {}
	t[storeKey][itemKey] = item
	t[counterKey] = t[counterKey] and t[counterKey] + 1 or 1
	return t
end

for prefix, data in pairs(mw.site.interwikiMap()) do
	local prefixData = {}
	for k, v in pairs(data) do
		prefixData[k] = v
	end
	local url = prefixData.url
	local urlObj = mw.uri.new(url)
	if not urlObj then
		error(string.format(
			"could not make URI object for prefix '%s'; " ..
			"please check your wiki's interwiki data",
			tostring(prefix)
		))
	end
	prefixData.url = url
	prefixData.urlObj = urlObj
	prefixes[prefix] = prefixData

	local urlData = makeCountedData(
		urls[urlText],
		'prefixes',
		'nPrefixes',
		prefix,
		prefixData
	)
	urlData.url = url
	urlData.urlObj = urlObj
	urls[url] = urlData
end

for url, urlData in pairs(urls) do
	local host = urlData.urlObj.host
	if not host then
		error(string.format(
			"could not find a host in URL '%s'; " ..
			"please check your wiki's interwiki data",
			tostring(url)
		))
	end
	local hostData = makeCountedData(
		hosts[host],
		'urls',
		'nUrls',
		url,
		urlData
	)
	hostData.host = host
	hosts[host] = hostData
end

return {
	prefixes = prefixes,
	urls = urls,
	hosts = hosts
}