Module:ISO 3166

local p = {}

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

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

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

local function isoname(data,code,lang) if data[code]["isonames"] then local name = data[code]["isonames"][lang or data[code]["defaultlang"] or data["defaultlang"] or "en"] or data[code]["isonames"][data[code]["defaultlang"] or data["defaultlang"] or "en"] or data[code]["isonames"]["en"] if name then return name end for _,iname in pairs(data[code]["isonames"]) do return iname end return data[code]["isodisplayname"] or data[code]["isoname"] else return data[code]["isodisplayname"] or data[code]["isoname"] end end

function p.strip(text) text = mw.ustring.upper(text)                    --Case insensitivity text = string.gsub(text,"^THE ","")              --Remove definite article text = string.gsub(text,"[%s%-%,%.%(%)%/%\']","") --Remove spacing and punctuation local accents = {["À"]="A",["Á"]="A",["Â"]="A",["Ã"]="A",["Ä"]="A",["Å"]="A",["Ç"]="C",["È"]="E",["É"]="E", ["Ê"]="E",["Ë"]="E",["Ì"]="I",["Í"]="I",["Î"]="I",["Ï"]="I",["Ñ"]="N",["Ò"]="O",["Ó"]="O", ["Ô"]="O",["Õ"]="O",["Ö"]="O",["Ø"]="O",["Ù"]="U",["Ú"]="U",["Û"]="U",["Ü"]="U",["Ý"]="Y"} text = mw.ustring.gsub(text,"[À-Ý]",accents)     --Deaccent return text 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 or args.lang) and isoname(data,code1,args.lang) or (data[code1]["displayname"] 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 or args.lang) and isoname(sdata,code2,args.lang) or (sdata[code2]["displayname"] 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 or args.lang) and isoname(data,alpha2,args.lang) or (cdata["displayname"] 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) or ""

end

function p.luacode(args)

local eot = args.error or ''

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       if type(scdata)=="table" then empty = false if findname(scode,scdata,args[2]) then return alpha2.."-"..scode end 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) or ""

end

return p