-- Made by AquaMonkey#1470, updated 27 may 2021
-- Update for configurable vote options and do_vote command by Fl_GUI#5136, 17 april 2022 (also some cleanup)

-- changelog:
-- 0.4: add command to toggle ranking, and add help command. both prefixed with aqm (aqm_help, aqm_enabled)
--    : add aqm_resetmessage to toggle a message stating votes were wiped
-- 0.5: add configurable vote options, usage: aqm_new_option <vote option> <vote value>, aqm_resetmessage to return to default
--      <vote option> is the string that is shown in chat and the message players have to type to vote.
--      <vote value> is a number representing the vote, negative numbers for negative votes, positive numbers for positive votes
--    : add extra command aqm_do_vote to allow for players to bind keys to vote, usage: aqm_do_vote <vote option>
--      <vote option> is the string you would type in chat
-- 1.0: allow players to vote directly after they finished a race by holding either DRIFT (positive) or BRAKE (negative)
--		this will only work with the default voting options for now!!!
-- 1.1: change the blue header above the map's name when a player is done racing to not say LIKED but a random option
--		now also hides the text after a few seconds

local mapVotingActive = false
local mapVotingAnnounced = false
local mapVotesStoredPositive = 0
local mapVotesStoredNegative = 0
local mapID = ''
local mapTitle = ''
local zoneTitle = '' -- why no work -- nvm, it work -- nvm, still sometimes no work
local mapVoteTable = {}         -- Store votes in a table. This one stores nodes,
local mapVoteTableNames = {}    -- while this one stores player names.
local mapVoteTableStore = {}
local mapVoteTableOptions = {}
local mapVoteTableOptionsDisplay = {}
local mapVoteTableOptionsCount = 0
local voteCount = 0
local randomAsk = nil
local randomAskNum = nil

-- configuration --
local defaultMapVoteTableOptions = {
    ["++"] = 1,
    ["11"] = 1,
	["=="] = 1,
    ["--"] = -1,
    ["00"] = -1
}

local defaultMapVoteTableOptionsDisplay = {
  [1] = "++",
  [-1] = "--"
}

local TEXT = "\x88"
local NEG = "\x87"
local POS = "\x83"
local WHITE = "\x80"

local gfxAsks = {
	"ENJOYED",
	"HAD FUN ON",
	"DID YOU LIKE",
	"DID YOU ENJOY"
}


-- CVARS --
local toggleEnabled = CV_RegisterVar({
	name = "aqm_enabled",
	defaultvalue = "on",
	flags = CV_NETVAR,
	PossibleValue = {off=0,on=1},
})

local resetMessage = CV_RegisterVar({
	name = "aqm_resetmessage",
	defaultvalue = "off",
	flags = CV_NETVAR,
	PossibleValue = {off=0,on=1},
})

-- functions --
local StringJoiner = function(sep)
    local factory = {
        text = "",
        sep = sep,
        parts = 0
    }
    function factory:addText(text)
        if self.text == "" then
            self.text = text
        else
            self.text = self.text..self.sep..text
        end
        self.parts = self.parts + 1
    end
    function factory:result()
        return self.text
    end
    function factory:length()
        return self.parts
    end
    return factory;
end

local norm = function(value)
    if value < 0 then
        return -1
    elseif value == 0 then
        return 0
    elseif value > 0 then
        return 1
    end
end

local getVoteOptionTables = function()
    if next(mapVoteTableOptions)
        then -- options set, use that
        return mapVoteTableOptions, mapVoteTableOptionsDisplay
    else -- no options set, use default
        return defaultMapVoteTableOptions, defaultMapVoteTableOptionsDisplay
    end
end

local getLevelFullName = function()
  return mapTitle..(zoneTitle and " "..zoneTitle or "")
end

-- metatable fun
local voteValueToColor = {}

voteValueToColor.prototype = {
    [-1] = NEG,
    [0] = WHITE,
    [1] = POS
}

voteValueToColor.mt = {
    __index = function(t, val)
        return  voteValueToColor.prototype[norm(val)]
    end
}

local mapVoteRecord = function(vote, player)
    -- Record a node's vote together with the associated name
    if #player ~= 0 then
        mapVoteTableNames[#player] = player.name
        mapVoteTable[#player] = vote
    end
end

setmetatable(voteValueToColor, voteValueToColor.mt)

-- console commands --
local consoleHelp = function(player)
    local voteOptions = getVoteOptionTables()
    local votes = {
        [-1] = StringJoiner(" / "),
        [0] = StringJoiner(" / "),
        [1] = StringJoiner(" / "),
    }

    for k,v in pairs(voteOptions) do
        local n = norm(v)
        votes[n]:addText(k)
    end

    local string =
"\136Aqua's Map Ranking Script\128\
After every race, you can vote on whether you enjoyed a map.\
---\
Vote by holding either DRIFT (to vote positive) or BRAKE (to vote negative) after you finished,\
or by typing one of the following options in chat during intermission:"
    if votes[1]:length() ~= 0 then
        string = $.."\n"..POS.."Voting positive: "..WHITE..votes[1]:result()
    end
    if votes[0]:length() ~= 0 then
        string = $.."\n".."Voting neutral: "..votes[0]:result()
    end
    if votes[-1]:length() ~= 0 then
        string = $.."\n"..NEG.."Voting negative: "..WHITE..votes[-1]:result()
    end
    
    -- TODO: allow people to include a link where to see results
    --string = $.."\nResults are posted at \129https://srb2kart.aqua.fyi/maps.html"

    CONS_Printf(player, string)
end

local addVoteOption = function(player, arg1, arg2)
    if not (IsPlayerAdmin(player) or player == server) then return end;
    if arg1 == nil or arg2 == nil then
        CONS_Printf(player, "usage: aqm_new_option <option> <number>")
        return
    end
    local tryParse = tonumber(arg2)
    if tryParse == nil then
        CONS_Printf(player, "Vote value ("..arg2..") is not a number")
        return
    end

    mapVoteTableOptions[arg1] = tryParse
    mapVoteTableOptionsDisplay[tryParse] = arg1
	mapVoteTableOptionsCount = mapVoteTableOptionsCount + 1
end

local clearVoteOptions = function(player)
    if not IsPlayerAdmin(player) then return end;
    mapVoteTableOptions = {}
	mapVoteTableOptionsCount = 0
    CONS_Printf(player, "Vote options have been reset")
end

local makeVote = function(player, vote)
    if vote == nil then return CONS_Printf(player, "usage: aqm_do_vote <vote option>")  end
    local voteTable = getVoteOptionTables()
    if voteTable[vote] == nil then
        CONS_Printf(player, "Invalid vote option")
        local options = StringJoiner(WHITE.." , ")

        for k,v in pairs(voteTable) do
            options:addText(voteValueToColor[v]..k)
        end
        CONS_Printf(player, "Valid options are: "..options:result())
		return
    end
    mapVoteRecord(voteTable[vote], player);
	CONS_Printf(player, "Recorded your "..voteValueToColor[voteTable[vote]]..vote..WHITE.." vote!")
end

COM_AddCommand("aqm_help", consoleHelp)
COM_AddCommand("aqm_do_vote", makeVote)
COM_AddCommand("aqm_new_option", addVoteOption)
COM_AddCommand("aqm_reset_options", clearVoteOptions)

-- file functions --
local iterateLines = function(file)
    local fileIterator = file:lines()
    return function()
        local line = fileIterator()
        if line == nil then return end
        local m1, m2, m3, storedVote = string.match(line, "(%s)(;)(%s)([+-]?%d)$")
        local storedName = line:sub(1, -(#(m1..m2..m3..storedVote) + 1))

        storedVote = tonumber(storedVote)
        if storedVote == nil then return end
        return storedName, storedVote
    end
end

local mapVoteRead = function()

    voteCount = 0 -- nullify this so we dont keep old data
    ---- First load votes if they exist
    local mapVoteTableStoredFile = io.open("maprankings\\"..mapID..".cfg", "r")

    if not (mapVoteTableStoredFile) then
        CONS_Printf(server, "AQM - No previous votes exist for "..mapTitle.." ("..mapID.."), creating empty file")
        local mapConfig = io.open("maprankings\\"..mapID..".cfg", "w")
        mapConfig:close()

    elseif (mapVoteTableStoredFile) then
        -- Extract previous votes from each line in the config
        for _, _ in iterateLines(mapVoteTableStoredFile) do
            -- Count amount of votes
            voteCount = voteCount + 1
        end
        --chatprint("* Got "..voteCount.." votes on the server")

    end

    server.votecount = voteCount
    return voteCount
end

-- Function to store votes into a file
local mapVoteWrite = function(mapVoteTable, mapVoteTableNames)

    -- Make sure only the server handles votes.
    -- !! REQUIRES HOSTMOD FOR THIS SPECIFIC LINE
    if consoleplayer ~= server then return end

    -- Indicate it's time to write stuff to disk
    print("AQM - Writing votes to disk")

    ---- First load votes if they exist
    local mapVoteTableStoredFile = io.open("maprankings\\"..mapID..".cfg", "r")

    if not (mapVoteTableStoredFile) then
        CONS_Printf(server, "AQM - Config didn't exist yet for "..mapTitle.." ("..mapID.."), creating it")
        local mapConfig = io.open("maprankings\\"..mapID..".cfg", "w")
        mapConfig:close()

    elseif (mapVoteTableStoredFile) then
        CONS_Printf(server, "AQM - Config exists for "..mapTitle.." ("..mapID..")")
        -- Extract previous votes from each line in the config
        for storedName, storedVote in iterateLines(mapVoteTableStoredFile) do
            -- Store previous votes in a table
            mapVoteTableStore[storedName] = storedVote
        end
    end

    ---- Save the votes to disk
    -- First check if people voted at all
    if next(mapVoteTable) == nil then
        CONS_Printf(server, "AQM - No ranks were recorded, not saving config for "..mapID)
        return
    else -- assume people voted

        -- Check if mapVoteTableStore exists
        if (mapVoteTableStoredFile) then
            CONS_Printf(server, "AQM - Player ranks this round for "..mapID..": ")
            -- Get a node's current name, then add them to the store
            for node, value in pairs(mapVoteTable) do
                local playerName = mapVoteTableNames[node]
                mapVoteTableStore[playerName] = value

                local _, voteDisplays = getVoteOptionTables()
                local valueStr = voteDisplays[value]

                CONS_Printf(server, "AQM - "..playerName..": "..valueStr.." ("..value..")")
            end
        end

        -- Open the corresponding map's config anew and just write everything to it swag
        -- Open the cfg
        local mapConfig = io.open("maprankings\\"..mapID..".cfg", "w")

        -- Now actually save the votes
        for name, vote in pairs(mapVoteTableStore) do
            mapConfig:write(name..' ; '..vote, '\n')
        end

        -- First initialize variables
        mapVotesStoredPositive = 0
        mapVotesStoredNegative = 0

        -- Clear everything we've used
        mapVoteTableStore = {}

        -- Bye bye, suckers!
        mapConfig:close()

    end

    -- TODO: 1 likes, 1 dislikes
    -- i still dont know how to sync numbers between server and client and i no longer care for i am tired of the wiki and its incomprehensible explanations
    --chatprint("*"..TEXT.." Total ranks: "..POS..mapVotesStoredPositive.." likes, "..NEG..mapVotesStoredNegative.." dislikes")

end

-- Hooks --
local mapVoteEnabled = function()
    if toggleEnabled.value == 1 then
        mapID = G_BuildMapName(gamemap)
        mapTitle = mapheaderinfo[gamemap].lvlttl
        zoneTitle = mapheaderinfo[gamemap].zonttl

        if mapVotingAnnounced == false then
            mapVoteRead()

            local voteOptions = {}
            local voteTableOptions, voteTableOptionsDisplay = getVoteOptionTables();
            local _, from = next(voteTableOptions)
            local _, to = next(voteTableOptions)
            for k,v in pairs(voteTableOptions) do
              voteOptions[v] = k
              if v < from then from = v end
              if to < v then to = v end
            end

            local options = StringJoiner(TEXT..",");
            for i = from,to do
                if voteOptions[i] then
                    options:addText(voteValueToColor[i]..voteTableOptionsDisplay[i])
                end
            end

			for p in players.iterate
				if p.mapRank == nil then
					chatprintf(p, "* "..TEXT.."Map ranking started!", true)
					chatprintf(p, WHITE.."* "..TEXT.."What did you think of "..WHITE..getLevelFullName()..TEXT.."?", false)
					chatprintf(p, WHITE.."* "..TEXT.."Vote with: "..options:result()..WHITE, false)
					if resetMessage.value == 1 then
						chatprintf(p, WHITE.."* "..POS.."Map ranks have recently been reset.", false)
					end
				end
			end
			
			// when we're done
            mapVotingAnnounced = true
        end

        mapVotingActive = true
    end
end

local function mapRankExiting()
	-- allow players to rank a map by just pressing DRIFT/BRAKE
	-- if player holds a button assign p.mapRank which we record
	-- currently hardcoding values 1 and -1 because i'm DUMB
	-- functionality gets disabled if custom vote options are used
	
	// dont run if not recording ranks
	if toggleEnabled.value == 0 then return end
	
	// dont run if not using default vote options
	if mapVoteTableOptionsCount ~= 0 then return end
	
	if randomAskNum == nil then
		randomAskNum = P_RandomRange(1, 4)
	end
	
	for p in players.iterate
		if leveltime == 0 or not p.mapRankSet then
			-- Set up playerdata
			p.mapRank = nil
			p.mapRankSet = true
			p.mapRankHoldTimer = 26
			p.mapRankShowDist = 237
			p.mapRankHideTimer = 70
			p.mapRankHideDist = 0
			//CONS_Printf(p, "Set up player data")
		end
		
		-- if the player doesnt have a rank yet allow input
		if p.mapRank == nil and p.exiting then
			if p.cmd.buttons == 4 then
				if p.mapRankHoldTimer ~= 0 then
					p.mapRankHoldTimer = p.mapRankHoldTimer - 1
				else
					p.mapRank = 1
					mapVoteRecord(p.mapRank, p)
					S_StartSound(nil, sfx_s3k34, p)
				end
			elseif p.cmd.buttons == 8 then
				if p.mapRankHoldTimer ~= 0 then
					p.mapRankHoldTimer = p.mapRankHoldTimer - 1
				else
					if mapheaderinfo[gamemap].lvlttl == "CTGP Mushroom Peaks" then -- :)
						p.mapRank = 1
					else
						p.mapRank = -1
					end
					mapVoteRecord(p.mapRank, p)
					S_StartSound(nil, sfx_s3k34, p)
				end
			else p.mapRankHoldTimer = 26
			end
		end
		
		if p.exiting and p.mapRankShowDist ~= 0 then
			p.mapRankShowDist = p.mapRankShowDist - 3 
		elseif p.exiting and p.mapRank ~= nil and p.mapRankHideDist ~= 237 then
			if p.mapRankHideTimer ~= 0
				p.mapRankHideTimer = p.mapRankHideTimer - 1
			else
				p.mapRankHideDist = p.mapRankHideDist + 3
			end
		end
	end
end

local function drawRankInterface(v)
	-- draw a GUI to vote with buttons
	
	// dont run if not recording ranks
	if toggleEnabled.value == 0 then return end
	
	// dont run if not using default vote options
	if mapVoteTableOptionsCount ~= 0 then return end
	
	// get local player
	local p = displayplayers[0]
	
	-- check if the player is racing
	if p.spectator then return end
	
	-- check if the player finished
	-- technically doesnt matter since the HUD is offscreen until finished but lets save frames
	//if p.exiting == 0 then return end
	
	// get map stuff
	mapTitle = mapheaderinfo[gamemap].lvlttl
    zoneTitle = mapheaderinfo[gamemap].zonttl
	
	// set up HUD vars
	local flags = V_SNAPTOTOP|V_SNAPTOLEFT
	local winWidth = v.width()/v.dupx();
	local winHeight = v.height()/v.dupy();
	local q = "?"
	
	// flashing text when holding button
	local textColor = nil
	if (leveltime % 2 == 0) then
		textColor = 0
	else
		textColor = V_YELLOWMAP
	end
	
	// data
	local mapName = mapTitle..(zoneTitle and " "..zoneTitle or "")
	local ask = gfxAsks[randomAskNum]

	// header
	//v.drawFill((winWidth / 4), 0, v.stringWidth(mapName) + 4, 50, 207|flags|V_50TRANS)
	if p.mapRank == nil then
		v.drawString((winWidth / 2), ((winHeight - 55) + (p.mapRankShowDist + p.mapRankHideDist)), ask, V_SKYMAP|flags, "center")
	else
		q = ""
	end
	v.drawString((winWidth / 2), ((winHeight - 45) + (p.mapRankShowDist + p.mapRankHideDist)), mapName..q, flags, "center")

	if p.mapRank == nil then
		v.drawString(((winWidth / 2) - 20), ((winHeight - 35) + (p.mapRankShowDist + p.mapRankHideDist)), "++", V_GREENMAP|flags, "right")
		v.drawString(((winWidth / 2) - 20), ((winHeight - 25) + (p.mapRankShowDist + p.mapRankHideDist)), "DRIFT", V_GREENMAP|flags, "right")
		v.drawString(((winWidth / 2) + 20), ((winHeight - 35) + (p.mapRankShowDist + p.mapRankHideDist)), "--", V_ORANGEMAP|flags, "left")
		v.drawString(((winWidth / 2) + 20), ((winHeight - 25) + (p.mapRankShowDist + p.mapRankHideDist)), "BRAKE", V_ORANGEMAP|flags, "left")
		
		if p.mapRankHoldTimer ~= 0 and p.mapRankHoldTimer ~= 26 and p.mapRankHoldTimer ~= nil then
			v.drawString((winWidth / 2), ((winHeight - 15) + (p.mapRankShowDist + p.mapRankHideDist)), "KEEP HOLDING...", textColor|flags, "center")
		end
	elseif p.mapRank == 1 then
		v.drawString((winWidth / 2), ((winHeight - 35) + (p.mapRankShowDist + p.mapRankHideDist)), "YOU VOTED ++", V_GREENMAP|flags, "center")
		v.drawString((winWidth / 2), ((winHeight - 25) + (p.mapRankShowDist + p.mapRankHideDist)), "THANKS!", V_GREENMAP|flags, "center")
	elseif p.mapRank == -1 then
		v.drawString((winWidth / 2), ((winHeight - 35) + (p.mapRankShowDist + p.mapRankHideDist)), "YOU VOTED --", V_ORANGEMAP|flags, "center")
		v.drawString((winWidth / 2), ((winHeight - 25) + (p.mapRankShowDist + p.mapRankHideDist)), "THANKS!", V_ORANGEMAP|flags, "center")
	end
	
end
hud.add(drawRankInterface, game)

local mapVoteDisabled = function()

    if mapVotingAnnounced == true then
        chatprint("* "..TEXT.."Map ranking for "..WHITE..getLevelFullName()..TEXT.." ended!")

        local mapVotesPositive = 0
        local mapVotesNegative = 0
        -- Count amount of positive/negative votes
        for _,v in pairs(mapVoteTable)
        do
            if v > 0 then
                mapVotesPositive = mapVotesPositive + 1
            elseif v < 0 then
                mapVotesNegative = mapVotesNegative + 1
            end
        end

        local likeStr = " gamers"
        local dislikeStr = " gamers"

        if mapVotesPositive == 1 then
            likeStr = " like"
        else
            likeStr = " likes"
        end

        if mapVotesNegative == 1 then
            dislikeStr = " dislike"
        else
            dislikeStr = " dislikes"
        end

        chatprint("* "..POS..mapVotesPositive..likeStr..WHITE..", "..NEG..mapVotesNegative..dislikeStr)

        -- Reset the counters back to 0, re-using them in mapVoteWrite
        mapVotesPositive = 0
        mapVotesNegative = 0

        -- Now write the votes to disk
        mapVoteWrite(mapVoteTable, mapVoteTableNames)

        -- Once done, clear out the table
        mapVoteTable = {}
        mapVoteTableNames = {}
        mapVotingAnnounced = false
		
		-- Clear out player data
		for p in players.iterate
			p.mapRank = nil
			p.mapRankSet = true
			p.mapRankHoldTimer = 26
			p.mapRankShowDist = 237
		end

		randomAskNum = nil
    end

    mapVotingActive = false
end


local mapVoteCast = function(source, msgtype, _, msg)

    -- First check if the message was a say
    if msgtype ~= 0 then
        return
    end

    -- Handle the vote options
    local voteTable = getVoteOptionTables();
    local bestVote, bestVoteValue = next(voteTable)
    for k, v in pairs(voteTable) do
        if v < bestVoteValue then
            bestVote = k
        end
    end

    local result = voteTable[msg]
    if result == nil then return false end


    -- Handle the message
    if mapVotingActive == true then

        if mapheaderinfo[gamemap].lvlttl == "CTGP Mushroom Peaks" then -- :)
            -- Person liked map, add it
            chatprintf(source, "* "..POS.."You voted "..bestVote..TEXT..". Thanks!", True)
            mapVoteRecord(bestVoteValue, source)
			source.mapRank = bestVote
            return true
        end

        chatprintf(source, "* "..TEXT.."You voted "..voteValueToColor[result]..msg..TEXT..". Thanks!", True)
        mapVoteRecord(result, source)
		source.mapRank = result

    elseif mapVotingActive == false then
        if toggleEnabled.value == 1 then
            chatprintf(source, "* "..NEG.."NOTE: Map ranking is active during intermission.")
            return false
        elseif toggleEnabled.value == 0 then
            chatprintf(source, "* "..NEG.."NOTE: Map ranking is currently disabled.")
            return false
        end
    end
end

local netvars = function(net)
    mapVoteTableOptions = net(mapVoteTableOptions)
    mapVoteTableOptionsDisplay = net(mapVoteTableOptionsDisplay)
end

addHook("IntermissionThinker", mapVoteEnabled) -- start map ranking when intermission screen pops up
addHook("VoteThinker", mapVoteDisabled) -- disable map ranking when map voting starts
addHook("ThinkFrame", mapVoteDisabled) -- for when the map voting screen is never reached
addHook("ThinkFrame", mapRankExiting)
addHook("PlayerMsg", mapVoteCast)
addHook("NetVars", netvars)
addHook("ThinkFrame", function() -- safeguard against late joiners
	for p in players.iterate do
		if p.mapRankSet == nil then
			-- Set up playerdata
			p.mapRank = nil
			p.mapRankSet = true
			p.mapRankHoldTimer = 26
			p.mapRankShowDist = 237
		end
	end
end)

print("AQM - Map ranking initialized! Have fun arguing :)")