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

local function to_str(arr)
    commastring = ""
    for i = 1, #arr - 1 do
        commastring = commastring .. ltrim(arr[i]) .. ", "
    end
    commastring = commastring .. ltrim(arr[#countryarray])
    return commastring
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 = to_str(countryarray)

    return commastring
end

return p