--[[ v1.00

     Test the string against the list of countries/continents.

     Return the first word which matches a country/continent name ...

     unless the "match=" parameter specifies a different match.

     If there is no match, then return an empty string ... unless

     the "nomatch" parameter specifies something different

]]



local getArgs = require('Module:Arguments').getArgs

local p = {}



-- config

local nomatch = ""

local matchnum = 1



local countryList = {

	'South African',

	'African',

	'Antarctican',

	'Central Asian',

	'South Asian',

	'South East Asian',

	'Southeast Asian',

	'Asian',

	'Caribbean',

	'Eurasian',

	'European',

	'Middle Eastern',

	'Central American',

	'North American',

	'South American',

	'Afghan',

	'Albanian',

	'Algerian',

	'American Samoan',

	'American',

	'Andorran',

	'Angolan',

	'Anguillan',

	'Antarctican',

	'Antigua and Barbuda',

	'Argentine',

	'Armenian',

	'Aruban',

	'Australian',

	'Austrian',

	'Azerbaijani',

	'Bahamian',

	'Bahraini',

	'Bangladeshi',

	'Barbadian',

	'Belarusian',

	'Belgian',

	'Belizean',

	'Beninese',

	'Bermudian',

	'Bhutanese',

	'Bolivian',

	'Bosnia and Herzegovina',

	'Botswanan',

	'Bouvet Island',

	'Brazilian',

	'British Indian Ocean Territory',

	'British Virgin Islands',

	'British',

	'English',

	'Scottish',

	'Welsh',

	'Northern Irish',

	'Bruneian',

	'Bulgarian',

	'Burkinabé',

	'Burmese',

	'Burundian',

	'Cambodian',

	'Cameroonian',

	'Canadian',

	'Cape Verdean',

	'Caymanian Islands',

	'Caymanian',

	'Central African',

	'Chadian',

	'Chilean',

	'Chinese',

	'Christmas Island',

	'Cocos (Keeling) Islands',

	'Colombian',

	'Comorian',

	'Cook Islands',

	'Costa Rican',

	'Croatian',

	'Cuban',

	'Cypriot',

	'Czech',

	'Danish',

	'Democratic Republic of the Congo',

	'Djiboutian',

	'Dominica',

	'Dominican Republic',

	'Dutch',

	'East Timorese',

	'Ecuadorian',

	'Egyptian',

	'Emirati',

	'Eritrean',

	'Estonian',

	'Ethiopian',

	'Falkland Islands',

	'Faroese',

	'Federated States of Micronesia',

	'Fijian',

	'Filipino',

	'Philippine',

	'Finnish',

	'French Polynesian',

	'French Southern Territories',

	'French',

	'Gabonese',

	'Gambian',

	'Georgian',

	'German',

	'Ghanaian',

	'Gibraltarian',

	'Greek',

	'Greenlandic',

	'Grenadian',

	'Guadeloupean',

	'Guam',

	'Guatemalan',

	'Guernsey',

	'Guianese',

	'Bissau-Guinean',

	'Bissau-Guinean',

	'Papua New Guinean',	

	'Equatoguinean',	

	'Guinean',

	'Guyanese',

	'Haitian',

	'Heard Island and McDonald Islands',

	'Honduran',

	'Hong Kong',

	'Hungarian',

	'Icelandic',

	'Indian',

	'Indonesian',

	'Iranian',

	'Iraqi',

	'Irish',

	'Israeli',

	'Italian',

	'Ivorian',

	'Jamaican',

	'Japanese',

	'Jersey',

	'Jordanian',

	'Kazakh',

	'Kenyan',

	'Kiribati',

	'Kuwaiti',

	'Kosovan',

	'Kosovar',

	'Kyrgyzstani',

	'Lao',

	'Latvian',

	'Lebanese',

	'Lesothan',

	'Liberian',

	'Libyan',

	'Liechtensteiner',

	'Liechtenstein',

	'Lithuanian',

	'Luxembourgian',

	'Luxembourg',

	'Macanese',

	'Macedonian',

	'Malagasy',

	'Malawian',

	'Malaysian',

	'Maldivian',

	'Malian',

	'Maltese',

	'Manx',

	'Marshallese',

	'Martiniquan',

	'Mauritanian',

	'Mauritian',

	'Mayotte',

	'Mexican',

	'Moldovan',

	'Mongolian',

	'Montenegrin',

	'Montserratian',

	'Monégasque',

	'Moroccan',

	'Mozambican',

	'Namibian',

	'Nauruan',

	'Nepalese',

	'Netherlands Antillean',

	'New Caledonian',

	'New Caledonia',

	'New Zealand',

	'Nicaraguan',

	'Nigerian',

	'Nigerien',

	'Niuean',

	'Norfolk Island',

	'North Korean',

	'Macedonian',

	'Northern Mariana Islands',

	'Norwegian',

	'Omani',

	'Pakistani',

	'Palauan',

	'Palestinian',

	'Panamanian',

	'Paraguayan',

	'Peruvian',

	'Pitcairn Islands',

	'Polish',

	'Portuguese',

	'Puerto Rican',

	'Qatari',

	'Republic of the Congo',

	'Romanian',

	'Russian',

	'Rwandan',

	'Réunionnais',

	'Sahrawi',

	'Saint Barthélemy',

	'Saint Helenian',

	'Saint Kitts and Nevis',

	'Saint Lucian',

	'Saint Martin',

	'Saint Pierre and Miquelon',

	'Saint Vincent and the Grenadines',

	'Salvadoran',

	'Sammarinese',

	'Samoan',

	'Saudi Arabian',

	'Senegalese',

	'Serbian',

	'Seychellois',

	'Sierra Leonean',

	'Singaporean',

	'Slovak',

	'Slovenian',

	'Solomon Islands',

	'Somalian',

	'South Georgia and the South Sandwich Islands',

	'South Korean',

    'Korean',

	'Spanish',

	'Catalan',

	'Sri Lankan',

	'Sudanese',

	'Surinamese',

	'Svalbard and Jan Mayen',

	'Swazi',

	'Swedish',

	'Swiss',

	'Syrian',

	'São Tomé and Príncipe',

	'Taiwanese',

	'Tajikistani',

	'Tajik',

	'Tanzanian',

	'Thai',

	'Togolese',

	'Tokelauan',

	'Tongan',

	'Trinidad and Tobago',

	'Tunisian',

	'Turkish',

	'Turkmen',

	'Turks and Caicos Islands',

	'Tuvaluan',

	'Ugandan',

	'Ukrainian',

	'United States Minor Outlying Islands',

	'United States Virgin Islands',

	'Uruguayan',

	'Uzbek',

	'Vanuatuan',

	'Vatican City',

	'Venezuelan',

	'Vietnamese',

	'Wallis and Futuna',

	'Yemeni',

	'Zambian',

	'Zimbabwean',

	'Åland',

	'Find demonym/testcases'

}



-- returns the name of a country demonym that is found in the string

-- ... or an empty string if there is no match

function findcountryinstring(str)



	nMatches = 0

	myMatches ={}

	str=" " .. str:gsub("^%s*(.-)%s*$", "%1") .. " "



		-- check agaist each country

		-- if there is a match, then return that country

		for i, thiscountry in ipairs(countryList) do

			local pat = thiscountry

			if pat == "Dominica" then

				pat = "Dominica%f[^a-z]" -- Don't match "dominican republic"; the correct demonym for people from Dominica is just "Dominica people"

			end

			if mw.ustring.find(str, pat) then

				nMatches = nMatches + 1

				myMatchesnMatches = thiscountry

			end

		end





	if (nMatches == 0) then

		-- none of the title words matches a whole country

		return nomatch

	end

	

	if ((matchnum >= 1) and (matchnum <= nMatches)) then

		return myMatchesmatchnum

	end



	if (matchnum < 0) then

		matchnum = matchnum + 1 -- so that -1 is the last match etc

		if ((matchnum + nMatches) >= 1) then

			return myMatchesmatchnum + nMatches

		end

	end

	

	-- if we get here, we have not found a match at the position specified by "matchnum"

	return nomatch

end



function p.main(frame)

	local args = getArgs(frame)

	return p._main(args)

end



function p._main(args)

	if (args'nomatch' ~= nil) then

		nomatch = args'nomatch'

	end

	

	-- by default, we return the first match

	-- but the optional "C" paarmeter sets the "matchnum" variable, which

	-- * for a positive matchnum "n", returns the nth match if it exists

	-- * for a positive matchnum "n", returns (if it exists) the nth match

	--   counting backwards from the end.

	--   So "match=-1" returns the last match

	--   and "match=-3" returns the 3rd-last match

	if (args'match' ~= nil) then

		matchnum = tonumber(args'match'])

		if ((matchnum == nil) or (matchnum == 0)) then

			matchnum = 1

		end

	end

	

	-- by default, we use the current page

	-- but if the "string=" parameters is supplied, we use that

	-- so we try the parameter first

	thispagename = nil

	if ((args'string' ~= nil) and (args'string' ~= "")) then

		-- we have a non-empty "string" parameter, so we use it

		thisstring = args'string'

	else

		-- get the page title

		thispage = mw.title.getCurrentTitle()

		thisstring = thispage.text;

	end

	

	-- now check the pagename to try to find a country

	result = findcountryinstring(thisstring)

	if (result == "") then

		return nomatch

	end

	return result

end



return p