require('strict');





local p = {} 



local error_msg = '<span style=\"font-size:100%\" class=\"error\"><code style=\"color:inherit; border:inherit; padding:inherit;\">&#124;_template=</code> missing or empty</span>';



-- data for various rankings held in module subpages, e.g. "Module:SportsRankings/data/FIFA World Rankings"

local data = {}      --[[ parameters containing data help in three tables

						data.source = {}     -- parameters for using in cite web (title, url, website)

						data.updated = {}    -- date of latest update (month, day, year)

						data.rankings = {}   -- the rankings list (country code, ranking, movement)

					    data.alias = {}      -- alias list (country code, country name [=key])

					    

					--]]



local  templateArgs = {} -- contains arguments from template involking module





local function getArgs(frame)

	local parents = mw.getCurrentFrame():getParent()

		

	for k,v in pairs(parents.args) do

		--check content

		if v and v ~= "" then

			templateArgsk=v --parents.args[k]

		end

	end

	for k,v in pairs(frame.args) do

		--check content

		if v and v ~= "" then

			templateArgsk=v --parents.args[k]

		end

	end

	-- allow empty caption to blank default

	if parents.args'caption' then templateArgs'caption' = parents.args'caption' end

	if frame.args'caption' then templateArgs'caption' = frame.args'caption' end

end



local function loadData(frame)

    

    local source = frame.args1 -- source of rankings e.g. FIFA World Rankings

    data = require('Module:SportsRankings/data/'.. source);

    

end



local function getDate(option)

   

   local dateTable = data.updated         -- there must be date table (data.updated)

                                          -- TODO add a warning and/or category

   if option == "LAST" then 

   		local lastDateTable = data.previous 

   		if lastDateTable then             -- there might not be a previous data table (data.previous)

   			dateTable = lastDateTable

	   else 

	   		return "No previous date available (data.updated missing)"

       end

   end

   

   if templateArgs'mdy' and templateArgs'mdy' ~= "" then

   	   return dateTable'month' .. " " .. dateTable'day' .. ", " .. dateTable'year'

   else

   	   return dateTable'day' .. " " .. dateTable'month' .. " " .. dateTable'year'

   end

end



local function addCiteWeb(frame)  -- use cite web template

	

	return frame:expandTemplate{ title = 'cite web' , args = {

    		url = data.source'url'],            --"https://www.fifa.com/fifa-world-ranking/ranking-table/men/index.html", 

			title = data.source'title'],        -- "The FIFA/Coca-Cola World Ranking",

			website = data.source'website'],    --"FIFA",

			'date' = getDate(),

			'access-date' = getDate()

			}}

end

local function addReference(frame)

	

	local text = ""

	if data.source'text' then text = data.source'text' end

	

	return frame:expandTemplate{ title = 'refn' , args = {

		name=frame.args1],                                 --ranking used, e.g. "FIFA World Rankings",

	    text .. addCiteWeb(frame)

	}}



end



--[[ the main function returning ranking for one country

      - takes three-letter country code or name of country as parameters

      - displays as rank | movement |date

]]

function p.main(frame)

	

    getArgs(frame) -- returns args table having checked for content

    loadData(frame)

    local outputString = ""

    local validCode = false

    local country = templateArgs2 -- country name or county code passed as parameter

    local rank, move

    

    if string.len( country) ==  3 then -- if we have a three letter country code 

	    for _,u in pairs(data.alias) do  -- run through alias list { 3-letter code, country name }

	    	if u1==country then        -- if code = passed parameter

	       		country = u2           -- set country name as key for ranking table

	       		validCode = true

	       		break

	       	end

	    end    

	    -- if no match of code to country name, set category

	    if not validCode then

	    	outputString="[[Category:Pages using SportsRankings with unknown parameters]]" .. outputString

        end

    end

    for _,v in pairs(data.rankings) do

    	if v1==country then 

       		rank = v2    -- get rank

       		move = v3    -- get move from last ranking

       		break

       	end

    end

    if not rank then -- no ranking found (do we want a tracking for no rank found?)

    	rank = 'NR' 

	    --outputString="[[Category:Pages using SportsRankings with unknown parameters]]" .. outputString

        --outputString="[[Category:Pages using SportsRankings with no ranking]]" .. outputString

	end

	

	if rank ~= 'NR' then

		outputString = outputString .. ' ' .. rank .. ' '

		if move < 0 and math.abs( move ) == math.abs( rank ) then -- new teams in ranking: move = -ranking

			outputString = outputString .. frame:expandTemplate{ title = 'new entry' } 

	    elseif move == 0 then                                    -- if no change in ranking

	    	outputString = outputString .. frame:expandTemplate{ title = 'steady' } 

	    elseif move < 0 then                                 --  if ranking down

	    	outputString = outputString .. frame:expandTemplate{ title = 'decrease' } 

	    	if move ~= -1000 then outputString = outputString .. ' ' .. math.abs(move) end

	    elseif move > 0 then                                 -- if ranking up

	    	outputString = outputString .. frame:expandTemplate{ title = 'increase' } 

	    	if move ~= 1000 then outputString = outputString .. ' ' .. move end

	    end	

    else

    	outputString = outputString .. frame:expandTemplate{ title = 'Abbr', args = { "NR", "Not ranked"}  }

    	--	{{Abbr|NR|Not ranked}} 

	end

	outputString = outputString .. ' (' .. getDate() .. ')'

	outputString = outputString .. addReference(frame)

    

    return outputString

	

end



--[[  outputs a table of the rankings 

        called by list() or list2() 

        positional parameters - |ranking|first|last the ranking to use, fist and last in table

        other parameters: |style=               -- CSS styling

                          |headerN= footerN=    -- displays header and footer rows with additional information

                          |caption=             -- value of caption to display

                                                -- by default it generates a caption

                                                -- this can be suppressed with empty |caption=

]]

local function table(frame, ranking, first,last)



    local styleString = ""

    if templateArgs'style' and templateArgs'style' ~= "" then styleString = templateArgs'style' end

    

    

    local sublist2 = { "ENG", "SCO", "WAL", "IRE", "NIR", "FRA", "England", "France", "Germany" }

    local sublist3 = { "AFG","AUS","BAN","BHR","BHU","BRU","CAM","CHN","GUM","HKG","IDN","IND","IRN","IRQ","JOR",

    	              "JPN","KGZ","KOR","KSA","KUW","LAO","LIB","MAC","MAS","MDV","MNG","MYA","NEP","OMA","PAK",

    	              "PHI","PLE","PRK","QAT","SIN","SRI","SYR","THA","TJK","TKM","TLS","TPE","UAE","UZB","VIE",

    	              "YEM" }

    local lastRank = 0

    local selectCount = 0

    local selectData = nil

    local selectList = nil

    if templateArgs'select' then 

    	if data.confederationtemplateArgs'select']] then

	    	selectList = templateArgs'select'

	    	selectData = data.confederationselectList

	    	selectCount = 1

    	end

    end

    

    -- column header customisation

    local rankHeader = templateArgs'rank_header' or "Rank"

    local selectionHeader = templateArgs'selection_header' or selectList or "Rank"

    local teamHeader = templateArgs'team_header' or "Team"

    local pointsHeader = templateArgs'points_header' or "Points"

    local changeHeader = templateArgs'change_header' or "Change"

    

    --start table

    local outputString = '{| class="wikitable" style="text-align:center;' .. styleString .. '"'

    

    -- add default or custom caption

    local caption = ranking .. ' as of ' .. getDate() .. '.'

    if templateArgs'caption' and templateArgs'caption'  ~= "" then 

    	caption = templateArgs'caption' 

    	caption = p.replaceKeywords(caption)

    end

    if not (templateArgs'caption' and templateArgs'caption'  == "") then 

		outputString = outputString ..	'\n|+' .. caption .. addReference(frame)

    end

    

    -- add header rows (logo, date of update etc)

    local count = 0

    local header = {}

    local tableWidth = 4

    if selectList then tableWidth = 5 end

    while count < 5 do

    	count = count + 1

	    if templateArgs'header'..count then

	    	headercount = templateArgs'header'..count 

	    	headercount = p.replaceKeywords( headercount])

	    	outputString = outputString ..	'\n|-\n| colspan="'.. tableWidth .. '" |' .. headercount

	    end

    end

    

    -- add the add part of the table

    local optionalColumn = ""

    if selectList then

    	optionalColumn = '\n!' .. selectionHeader 

    end

   	outputString = outputString .. '\n|-' .. optionalColumn

    	                        .. '\n!' .. rankHeader .. '\n!' .. changeHeader 

    	                        .. '\n!' .. teamHeader .. '\n!' .. pointsHeader

   

    local change,code = '', ''

    --while i<last do 

    for k,v in pairs(data.rankings) do

	   --v[2] = tonumber(v[2])

	   if v2 >= first and v2 <= last then 



		   for _,u in pairs(data.alias) do  -- get country code from name

		    	if u2==v1 then 

		       		code = u1    -- if alias (country code) then use country name as key

		       		break

		       	end

		    end   

	   	   

	   	    local continue = true

	   	    if selectList then                 -- select from list

	   	   	    continue = false 

	   			for _,u in pairs(selectData) do

	   				if u == v1 or u == code then 

	   					continue = true 

	   					break

	   				end

	   			end

	   	    end

		

			if continue ==true  then 

	   	   

			   local rowString = '\n|-'

			   if selectList then 

			   	    local selectRank = selectCount

			   	    if v2==lastRank then selectRank = selectCount -1 end -- only handles two at same rank

					rowString = rowString ..  '\n|' .. selectRank 

					selectCount = selectCount + 1

			   end

			   rowString = rowString .. '\n|' .. v2  -- rank

			   lastRank = v2

			   

			   local move = v3

			   if move < 0 and math.abs( move ) == math.abs( v2 ) then -- new teams in ranking: move = -ranking

					change = frame:expandTemplate{ title = 'new entry' } 

			   elseif move == 0 then                                    -- if no change in ranking

			    	change = frame:expandTemplate{ title = 'steady' } 

			    elseif move < 0 then                                 --  if ranking down

			    	change = frame:expandTemplate{ title = 'decrease' } .. ' ' .. math.abs(move)

			    elseif move > 0 then                                 -- if ranking up

			    	change = frame:expandTemplate{ title = 'increase' } .. ' ' .. move

			    end	

			   rowString = rowString .. '||' .. change

			   

	--[[		   for _,u in pairs(data.alias) do

			    	if u[2]==v[1] then 

			       		code = u[1]    -- if alias (country code) then use country name as key

			       		break

			       	end

			    end   

	]]		   

			   --TODO reorganise the following with better logic

			   --[[ template to display flag icon and team link (e.g. fb, fbw, bk, ih)

			       e.g.  "FIFA World Rankings" = 'fb', "FIFA Women's World Rankings" 'fbw',

			             "FIBA World Rankings" = 'bk', "IIHF World Ranking"  = 'ih'  

			       tries with country code, then if error, tried with country name]]

			   local countryTemplate = data.templates'flagged_team_link' 

			   local flagVariant = ""

               if data.templates.flagvar and data.templates.flagvarcode then

            		flagVariant = data.templates.flagvarcode 

               end

               local flagCode = code

               if data.templates.flag_codes then

               		flagCode = data.templates.flag_codescode or code

               end

			   local countryIconString = frame:expandTemplate{ title = countryTemplate, args = {flagCode, flagVariant} }    -- country



	 		   local _,test =  string.gsub( countryIconString, "Template:Country data", "") -- page does not exist

	 		   if test == 1 then -- if error try country name

	 		   	  countryIconString = frame:expandTemplate{ title = countryTemplate, args = {v1]} }

	 		   	  --countryIconString = "testing"

	 		   else 

	 		   	 -- countryIconString = "exists"

	 		   end

	 		   	  

	 		   rowString = rowString .. '\n|style="text-align:left"|' .. countryIconString

			   

			   local points = ""

			   if v4 then points = v4 end

			   rowString = rowString ..  '||' .. points       -- country for now, later points

			   outputString = outputString .. rowString

			end

		end

	end

	

    -- add footer rows

    count = 0

    local footer = {}

    while count < 5 do

    	count = count + 1

	    if templateArgs'footer'..count then

	    	footercount = templateArgs'footer'..count 

	    	footercount = p.replaceKeywords(footercount])

	    	outputString = outputString ..	'\n|-\n| colspan="'.. tableWidth .. '" |' .. footercount

	    end

    end





    outputString = outputString .. "\n|}"



    return outputString

	

end



function p.replaceKeywords(keyword)

	keyword =  string.gsub( keyword, "INSERT_UPDATE_DATE", getDate())

    keyword =  string.gsub( keyword, "INSERT_LAST_DATE", getDate("LAST"))

	if string.find(keyword, "INSERT_REFERENCE") then

		keyword =  string.gsub( keyword, "INSERT_REFERENCE", addReference(mw.getCurrentFrame()))

	end

    return keyword

end



--[[ create a table of rankings

       parameters:  |ranking=        -- ranking to display (e.g. FIFA World Rankings)

                    |first= |last=   -- first and last ranking to display (defaults 1-10)

]]

function p.list(frame)



    getArgs(frame) -- returns args table having checked for content

    loadData(frame)	

    local ranking = frame.args1

    

    local first = tonumber(frame.args'2']) or 1

    local last = tonumber(frame.args'3']) or 10

    

    return table(frame, ranking, first, last)

end



--[[ create a particla table of rankings above and below a country 

       parameters:  |ranking |country |span     

                                              -- ranking - the ranking to display (e.g. FIFA World Rankings)

                                              -- country - country table is centred around

                                              -- span=   - rows to display above and below country

]]

function p.list2(frame)



    getArgs(frame) -- returns args table having checked for content

    loadData(frame)	

    local ranking = frame.args1

    local first, last = 1,10

    local country = frame.args2           -- name or code of country to center table around

    local span = frame.args3 or 2         -- number of rows to display above and below country (default:2)

    

    if string.len(country) == 3 then        -- if three letter country code

    		for _,u in pairs(data.alias) do

		    	if u1==country then 

		       		country = u2          -- if country code then use country name 

		       		break

		       	end

		    end   

    end

    for k,v in pairs(data.rankings) do      -- find position of country in rankings

       if v1 == country then

       	  first = v2-span

       	  last = v2+span

       end

    end

    

    return table(frame, ranking, first, last)

end



return p