Module:ISO 3166

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

local p = {}

local data = mw.loadData("Module:ISO 3166/data/National")

local function cerror(eot,errortext,inputtext)
  if eot=="empty" then
    return ""
  elseif eot=="input" then
    return inputtext
  else
    return mw.html.create("span"):addClass("error"):wikitext(errortext)
  end
end

local function strip(text)
  text = string.upper(text)                         --Case insensitivity
  text = string.gsub(text,"^THE ","")               --Remove definite article
  text = string.gsub(text,"[%s%-%,%.%(%)%/%\']","") --Remove spacing and punctuation
  --text = string.gsub(text,"[\768-\879]","") --Unicode deaccenting doesn't work
  return text
end

local function findname(code,cdata,qry)
  local sqry = strip(qry)
  if cdata["name"] and sqry==strip(cdata["name"])
    or cdata["isoname"] and sqry==strip(cdata["isoname"])
    or sqry==code or sqry==cdata["alpha3"] or sqry==cdata["numeric"]
  then
    return true
  end
  for _,tname in pairs(cdata["altnames"] or {}) do
    if sqry==strip(tname) then
      return true
    end
  end
  return false
end

function p.luaname(args)

local eot = args.error

if not args[1] then
  return cerror(eot,"No parameter given",args[2] or "")
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,"^%w?%w?%w?$")
  or not (string.find(code1,"^%u%u$") or code2=="")
then
  if code2=="" then
    return cerror(eot,"Invalid ISO 3166-1 code "..code1,code1)
  else
    return cerror(eot,"Invalid ISO 3166-2 code "..code1.."-"..code2,code1.."-"..code2)
  end
end

if string.find(code1,"^%u%u$") then
  if code2=="" then
    --3166-1 alpha-2 code
    if data[code1] then
      return args.isoname and data[code1]["isoname"] or data[code1]["name"]
    else
      return cerror(eot,"Unknown ISO 3166-1 code "..code1,code1)
    end
  else
    --3166-2 code
    local sdata
    if data[code1] then
      sdata = mw.loadData("Module:ISO 3166/data/"..code1)
    else
      return cerror(eot,"Unknown ISO 3166-1 code "..code1,code1.."-"..code2)
    end
    if sdata[code2] then
      return args.isoname and sdata[code2]["isoname"] or sdata[code2]["name"]
    else
      for _,_ in pairs(sdata) do
        return cerror(eot,"Unknown ISO 3166-2 code "..code1.."-"..code2,code1.."-"..code2)
      end
      return cerror(eot,"No subdivision codes for "..data[code1]["name"],code1.."-"..code2)
    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) do
    if cdata[codetype]==code1 then
      return args.isoname and cdata["isoname"] or cdata["name"]
    end
  end
  return cerror(eot,"Unknown ISO 3166-1 code "..code1,code1)
end

end

function p.name(frame)

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

end

function p.luacode(args)

local eot = args.error

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

if not args[2] then
  --3166-1 code
  for alpha2,cdata in pairs(data) do
    if findname(alpha2,cdata,args[1]) then
      if args["codetype"]=="numeric" or args["codetype"]=="alpha3" then
        return cdata[args["codetype"]]
      else
        return alpha2
      end
    end
  end
  return cerror(eot,"Unknown country name "..args[1],args[1])
else
  --3166-2 code
  for alpha2,cdata in pairs(data) do
    if findname(alpha2,cdata,args[1]) then
      local sdata = mw.loadData("Module:ISO 3166/data/"..alpha2)
      local empty = true
      for scode,scdata in pairs(sdata) do
        empty = false
        if findname(scode,scdata,args[2]) then
          return alpha2.."-"..scode
        end
      end
      if empty then
        return cerror(eot,"No subdivision codes for "..args[1],args[2])
      else
        return cerror(eot,"Unknown subdivision name "..args[2],args[2])
      end
    end
  end
  return cerror(eot,"Unknown country name "..args[1],args[2])
end

end

function p.code(frame)

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

end

return p