Module:ISO 3166

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 ipairs(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

function p.subpagetest(args)

local code = string.upper(args[1])

local sdata if not data[code1] then return "" --Unassigned or reserved 3166-1 code else local sname = "Module:ISO 3166/data/"..code1 if not mw.title.new(sname).exists then return "-" --Data subpage doesn't exist else sdata = mw.loadData(sname) end end for _,_ in pairs(sdata) do return "n" --Non-empty 3166-2 entry end return "e" --Empty 3166-2 entry

end

return p