Module:Link with country/sandbox

Module documentation[view] [edit] [history] [purge]

Usage

{{#invoke:Link with country|countrynames}}

The above documentation is transcluded from Module:Link with country/doc. (edit | history)
Editors can experiment in this module's sandbox (edit | diff) and testcases (edit | run) pages.
Add categories to the /doc subpage. Subpages of this module.

local p = {} 

local ISO_3166 = require('Module:ISO 3166')

local function ltrim(s)
  return s:match'^%s*(.*)'
end

local function split_with_comma(str)
  local fields = {}
  for field in str:gmatch('([^,]+)') do
    fields[#fields+1] = ltrim (field)
  end
  return fields
end

local function to_ISO_3166(country)
  for i = 1, #country do
    shortname = ISO_3166.name({country[i]})
    if shortname ~= "" then country[i] = shortname end
  end
  return country
end

function p.countrynames( frame )
-- local test = frame.args[1] .. ", " .. frame.args.property
--    return test

-- change comma list to array
countryarray = split_with_comma( frame.args[1] )

-- put each array entry {as a table} through Module:ISO 3166. If the result is null, keep original entry
to_ISO_3166(countryarray)

-- change array back to comma string
commastring = ""
for i = 1, #countryarray-1 do
  commastring = commastring .. ltrim(countryarray[i]) .. ", "
end
commastring = commastring .. ltrim(countryarray[#countryarray]) 

return commastring

end

return p