#  Custom kick idle player script by mr.smoothy#5824
import time
import ba
from ba._general import Call
import _ba
import ba.internal
import setting
settings = setting.get_settings_data()
INGAME_TIME=settings["afk_remover"]["ingame_idle_time_in_secs"]
LOBBY_KICK=settings['afk_remover']["kick_idle_from_lobby"]
INLOBBY_TIME=settings['afk_remover']["lobby_idle_time_in_secs"]
cIdle = 0 # players/player found idle within very short time
cLastIdle = 0
class checkIdle(object):
    def start(self):
        self.t1=ba.timer(2, ba.Call(self.check),repeat=True)
        self.lobbies={}
    def check(self):
        global cLastIdle
        global cIdle
        current=ba.time(ba.TimeType.REAL,timeformat=ba.TimeFormat.MILLISECONDS)
        if not ba.internal.get_foreground_host_session():
            return
        for player in ba.internal.get_foreground_host_session().sessionplayers:
            last_input=int(player.inputdevice.get_last_input_time())
            afk_time=int((current-last_input)/1000)
            if afk_time in range(INGAME_TIME,INGAME_TIME+20) or afk_time > INGAME_TIME+20:
                if (current - cLastIdle)/1000 < 3:
                    cIdle = cIdle+1
                    if cIdle >= 3:
                        return
                else:
                    cIdle = 0
                cLastIdle = current

            if afk_time in range(INGAME_TIME,INGAME_TIME+20):
                self.warn_player(player.get_v1_account_id(),"Press any button within "+str(INGAME_TIME+20-afk_time)+" secs")
            if afk_time > INGAME_TIME+20:
                player.remove_from_game()
        if LOBBY_KICK:
            current_players=[]
            for player in ba.internal.get_game_roster():
                if player['client_id'] !=-1 and len(player['players']) ==0:
                    current_players.append(player['client_id'])
                    if player['client_id'] not in self.lobbies:
                        self.lobbies[player['client_id']]=current
                    lobby_afk=int((current - self.lobbies[player['client_id']])/1000)
                    if lobby_afk in range(INLOBBY_TIME,INLOBBY_TIME+10):
                        _ba.screenmessage("Join game within "+str(INLOBBY_TIME+10-lobby_afk)+" secs",color=(1,0,0),transient=True,clients=[player['client_id']])
                    if lobby_afk > INLOBBY_TIME+ 10:
                        ba.internal.disconnect_client(player['client_id'],0)
            # clean the lobbies dict
            temp=self.lobbies.copy()
            for clid in temp:
                if clid not in current_players:
                    del self.lobbies[clid]
    def warn_player(self,pbid,msg):
        for player in ba.internal.get_game_roster():
            if player["account_id"]==pbid:
                _ba.screenmessage(msg,color=(1,0,0),transient=True,clients=[player['client_id']])