-- Slmod PingCheck
---- autokicks players with a high pÃ­ng
 
slmod = slmod or {}
do
    slmod.pingCheck = {}
    local pingCheckConfig = slmod.config.pingcheck_conf
    local lastCheck = 0

    
    local exemptList = slmod.config.autoAdmin.exemptionList
    
    function slmod.loadPingExemptList()
        if slmod.exemptAll then -- append external exemption all list to the local one
            for id, data in pairs(slmod.exemptAll) do
                if not exemptList[id] then
                    exemptList[id] = data
                end
            end
        end
        
        if slmod.exemptPing then
            for id, data in pairs(slmod.exemptPing) do -- check if anyone is only on the exempt autoAdmin list
                if not exemptList[id] then
                    exemptList[id] = data
                end
            end
        end
    end
    function slmod.pingCheck.addClient(id)
        local ucid = net.get_player_info(id, 'ucid')
        if slmod.isAdmin(ucid) == false and (not exemptList[ucid]) and (not (id == 1)) then -- cant be an admin or on the exemption list
			slmod.clients[id].pingCheck = {}
            slmod.clients[id].pingCheck['ping'] = 0
			slmod.clients[id].pingCheck['avgPing'] = 0
            slmod.clients[id].pingCheck['warnings'] = 0
            slmod.clients[id].pingCheck['lastWarnTime'] = 0
            slmod.clients[id].pingCheck['connectTime'] = os.time()
            slmod.clients[id].pingCheck['checkActive'] = false
        end

        return
    end
    
    function slmod.pingCheck.setActive(id)
        if slmod.clients[id] and slmod.clients[id].pingCheck and slmod.clients[id].pingCheck.checkActive == false then
			slmod.clients[id].pingCheck.checkActive = true
        end
        return
    end
    local function getClients()
        -- check for disconnected clients
        local count = 0
        for id, client in pairs(slmod.clients) do
            if slmod.clients[id].pingCheck then
               count = count + 1
            end
        end
        return count
    end
 
    --- warns and eventually kicks a player if his ping is too high
    local function warnClient(id)
		local client = slmod.clients[id]
        local show_scope = {}
        show_scope["clients"] = {id}
 
        -- get ping/warning limit configuration
        local maxPing = pingCheckConfig.max_ping
        local warnLimit = pingCheckConfig.warning_limit
 
        -- set warning count
        client.pingCheck.warnings = client.pingCheck.warnings + 1
 
        if client.pingCheck.warnings > warnLimit then
            net.kick(client.id, "Your ping is too high (".. client.pingCheck.avgPing ..", max: ".. maxPing ..")")
            kickMsg = "Player ".. client.name .." got kicked for a too high ping (current: ".. client.pingCheck.ping ..", average: ".. client.pingCheck.avgPing ..", max: ".. maxPing ..")"
            slmod.info(kickMsg)

            if pingCheckConfig.warning_msg then
                slmod.scopeMsg(kickMsg, 10, 'chat', nil)
            end
        else
            if pingCheckConfig.warning_msg then
                slmod.scopeMsg("Your ping is too high (current: ".. client.pingCheck.ping ..", average: ".. client.pingCheck.avgPing ..", max: ".. maxPing ..")", 10, 'chat', show_scope)
                slmod.scopeMsg("Warning ".. client.pingCheck.warnings .."/".. warnLimit .." before getting kicked", 10, 'chat', show_scope)
            end
        end
 
        slmod.clients[id].pingCheck.lastWarnTime = DCS.getModelTime()
    end
 
    -- gets called every simulation frame
    function slmod.pingCheck.heartbeat()
        local doCheck = false
 
       
        -- check if elapsed time since last check is > 10 secs
        if DCS.getModelTime() - lastCheck > pingCheckConfig.wait_time then
            lastCheck = DCS.getModelTime()
            doCheck = true
        end
 
        -- dont check if pingCheck is disabled
        if pingCheckConfig.enabled == false then
            doCheck = false
        end
 
        if doCheck == false then
            return
        end
        -- get clients and check ping values
        
        if getClients() > 0 then
            for id, client in pairs(slmod.clients) do
                if client.pingCheck then
					if client.pingCheck.checkActive == true then
                        local curPing = net.get_player_info(id, 'ping')
						slmod.clients[id].pingCheck.avgPing = math.floor((client.pingCheck.ping + curPing + slmod.clients[id].pingCheck.avgPing)/3)
                        slmod.clients[id].pingCheck.ping = curPing
                        local warnRepeatTime = pingCheckConfig.warning_repeat_time
             
                        -- check client's ping and warn if the repeat time has elapsed
                        if (slmod.clients[id].pingCheck.avgPing > pingCheckConfig.max_ping) and (DCS.getModelTime() - slmod.clients[id].pingCheck.lastWarnTime > warnRepeatTime) then
                            warnClient(id)
                        elseif (slmod.clients[id].pingCheck.avgPing < pingCheckConfig.max_ping) then
							slmod.clients[id].pingCheck.warnings = 0
                        end
                    else
                        if os.time() > slmod.clients[id].pingCheck.connectTime + 300 then
						   slmod.clients[id].pingCheck.checkActive = true
                        end
                    end
                end
            end
        end
        return
    end
end
slmod.info('SlmodPingCheck.lua loaded.')