Module:ISO 3166

Documentation for this module may be created at Module:ISO 3166/doc

local p = {}

local data = {
  national = {
    ["AF"] = {alpha3="AFG",numeric="004",name="Afghanistan"},
    ["AL"] = {alpha3="ALB",numeric="008",name="Albania"},
    ["BO"] = {alpha3="BOL",numeric="068",name="Bolivia",isoname="Bolivia, Plurinational State of"},
    ["NL"] = {alpha3="NLD",numeric="528",name="Netherlands"},
    ["GB"] = {alpha3="GBR",numeric="826",name="United Kingdom",altnames={"UK"}}
  },
  NL = {
    ["DR"] = {name="Drenthe"},
    ["FL"] = {name="Flevoland"},
    ["FR"] = {name="Friesland",altnames={"Fryslân","Fryslan"}},
    ["GE"] = {name="Gelderland"}
  },
  GB = {
    ["ENG"] = {name="England"},
    ["NIR"] = {name="Northern Ireland"},
    ["SCT"] = {name="Scotland"},
    ["WLS"] = {name="Wales"}
  }
}

local function cerror(text)
  return mw.html.create("span"):addClass("error"):wikitext(text)
end

local function strip(text)
  text = string.lower(text)
  text = string.gsub(text,"[%s%-%,%.]","")
  --text = string.gsub(text,"[\768-\879]","") --Unicode deaccenting doesn't work
  return text
end

local function findname(qry,cdata)
  local testnames = cdata["altnames"] or {}
  if cdata["isoname"] then table.insert(testnames,1,cdata["isoname"]) end
  table.insert(testnames,1,cdata["name"])
  for _,tname in ipairs(testnames) do
    if strip(qry)==strip(tname) then
      return true
    end
  end
  return false
end

function p.lua_code2name(args)

if not args[1] then
  return cerror("No parameter given")
end

local code1 = string.upper(args[1] or "")
local code2 = string.upper(args[2] or "")
if string.find(code1,"%-") then
  code1, code2 = string.match(code1,"^([^%-]*)%-(.*)$")
end

if --Check if valid code
  --No non-alphanumeric characters allowed
  string.find(code1..code2,"[^A-Z0-9]")
  --3166-1 codes can be two or three letters or three digits;
  --3166-2 codes must be two letters (first part) and 1-3 letters or digits (second part)
  or not (string.find(code1,"^%u%u%u?$") or string.find(code1,"^%d%d%d$"))
  or not (string.find(code2,"^%u?%u?%u?$") or string.find(code2,"^%d?%d?%d?$"))
  or not (string.find(code1,"^%u%u$") or code2=="")
then
  return cerror("Invalid code "..code1..(code2~="" and "-"..code2 or ""))
end

if string.find(code1,"^%u%u$") then
  if code2=="" then
    --3166-1 alpha-2 code
    if data["national"][code1] then
      return args.isoname and data["national"][code1]["isoname"] or data["national"][code1]["name"]
    else
      return cerror("Unknown code "..code1..(code2~="" and "-"..code2 or ""))
    end
  else
    --3166-2 code
    if data[code1] and data[code1][code2] then
      return args.isoname and data[code1][code2]["isoname"] or data[code1][code2]["name"]
    else
      return cerror("Unknown code "..code1..(code2~="" and "-"..code2 or ""))
    end
  end
else
  --3166-1 alpha-3 or numeric code
  local codetype = string.find(code1,"%d") and "numeric" or "alpha3"
  for alpha2,cdata in pairs(data["national"]) do
    if cdata[codetype]==code1 then
      return args.isoname and cdata["isoname"] or cdata["name"]
    end
  end
end

return cerror("Unknown code "..code1..(code2~="" and "-"..code2 or ""))

end

function p.code2name(frame)

local args = require("Module:Arguments").getArgs(frame)
return p.lua_code2name(args)

end

function p.lua_name2code(args)

if not args[1] then
  return cerror("No parameter given")
end

if not args[2] then
  --3166-1 code
  for alpha2,cdata in pairs(data["national"]) do
    if findname(args[1],cdata) then
      if args["codetype"]=="numeric" or args["codetype"]=="alpha3" then
        return data["national"][alpha2][args["codetype"]]
      else
        return alpha2
      end
    end
  end
  return cerror("Unknown name "..args[1])
else
  --3166-2 code
  for alpha2,cdata in pairs(data["national"]) do
    if findname(args[1],cdata) then
      if not data[alpha2] then
        return cerror("No subdivision codes for "..args[1])
      end
      for scode,sdata in pairs(data[alpha2]) do
        if findname(args[2],sdata) then
          return alpha2.."-"..scode
        end
      end
      return cerror("Unknown name "..args[2])
    end
  end
  return cerror("Unknown name "..args[1])
end

end

function p.name2code(frame)

local args = require("Module:Arguments").getArgs(frame)
return p.lua_name2code(args)

end

return p