Module:Interwiki extra/data

-- This module processes data from mw.site.interwikiMap for use with
-- [[Module:Interwiki extra]]. 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
 prefixes[prefix] = prefixData

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

for url, urlData in pairs(urls) do
 local urlObj = mw.uri.new(url) -- We can't output URI objects as they have metatables
 if not urlObj or not urlObj.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[urlObj.host],
  'urls',
  'nUrls',
  url,
  urlData
 )
 hostData.host = urlObj.host
 hosts[urlObj.host] = hostData
end

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