import es
from playerlib import getPlayer
from usermsg import hudhint

class Aimbot(object):
    class Player(object):
        def __init__(self, userid):
            self.team   = 0
            self.userid = userid
            self.enable = False
            self.player = getPlayer(userid)

        def changeTeam(self, team):
            self.team = 5 - team

        def setActive(self):
            if self.enable:
                self.enable = False
                es.tell(self.userid, "#multi", "#green[#lightgreenAimbot#green] #defaultTurned off.")
            else:
                self.enable = True
                es.tell(self.userid, "#multi", "#green[#lightgreenAimbot#green] #defaultTurned on.")

        def tryTurnOff(self):
            if self.enable:
                self.setActive()

        def tryAimOnPlayer(self):
            if self.enable:
                distance, closest = self.player.getClosestPlayer(self.team)
                if closest is not None:
                    self.player.viewPlayer(closest)
                    hudhint(self.userid, "BackRaw's AimBot\n--------------\n\nAiming at %s (Distance: %.2f meters)" % (es.getplayername(closest), distance * 0.0254))

    def __init__(self):
        self.players = {}

    def insertUserid(self, userid):
        self.players[userid] = self.Player(userid)

    def removeUserid(self, userid=0, clear=False):
        if not clear:
            del self.players[userid]
        else:
            self.players.clear()

    def roundEnd(self):
        for userid in self.players:
            self.players[userid].tryTurnOff()

    def playerDeath(self, userid):
        self.players[userid].tryTurnOff()

    def changeTeam(self, userid, team):
        if userid in self.players:
            self.players[userid].changeTeam(team)

    def sayCommand(self, userid):
        if userid in self.players:
            self.players[userid].setActive()

    def scriptLoad(self):
        for userid in map(str, es.getUseridList()):
            self.insertUserid(userid)
        es.regsaycmd("!ab", "aimbot/saycommand")
        es.addons.registerTickListener(self._tickListener)

    def scriptUnload(self):
        self.removeUserid(clear=True)
        es.unregsaycmd("!ab")
        es.addons.unregisterTickListener(self._tickListener)

    def _tickListener(self):
        if self.players:
            for userid in self.players:
                self.players[userid].tryAimOnPlayer()

aimbot = Aimbot()

def round_end(ev):
    aimbot.roundEnd()

def player_activate(ev):
    aimbot.insertUserid(ev["userid"])

def player_death(ev):
    aimbot.playerDeath(ev["userid"])

def player_team(ev):
    if not int(ev["disconnect"]):
        aimbot.changeTeam(ev["userid"], int(ev["team"]))

def player_disconnect(ev):
    aimbot.removeUserid(ev["userid"])

def saycommand():
    aimbot.sayCommand(str(es.getcmduserid()))

def load():
    aimbot.scriptLoad()

def unload():
    aimbot.scriptUnload()