from lib.enumresponse import EnumResponse, Packet
from lib.constants import Teams, GameTypes, MapID


class GiantsEnumResponseParser:
    def __init__(self):
        super().__init__()
        self.server_name = None

        self.max_players = None
        self.current_players = None

        self.game_type = None
        self.teams = None
        self.version = None
        self.points_per_capture = None
        self.points_per_kill = None
        self.base_level = None

        self.allow_joiners = None
        self.damage_teammates = None
        self.lock_teams = None
        self.is_dedicated = None
        self.vimps_disabled = None
        self.weap_bits = None
        self.high_bandwidth = None
        self.no_voting = None
        self.smartie_difficulty = None
        self.vimp_difficulty = None

        self.capture_prevent_count = None
        self.current_cap_prev_time = None

        self.map_id = None
        self.map_checksum = None
        self.map_name = None
        self.build = None
        self.revision = None

    @classmethod
    def parse(cls, enumresp: EnumResponse):
        g_enumresponse = cls()
        g_enumresponse.max_players = enumresp.MaxPlayers
        g_enumresponse.current_players = enumresp.CurrentPlayers
        g_enumresponse.server_name = enumresp.SessionName

        game_data = enumresp.ApplicationReservedData
        # print(game_data)
        p = Packet(game_data)
        g_enumresponse.map_id = p.get_byte()
        g_enumresponse.game_type = p.get_byte()
        g_enumresponse.teams = p.get_byte()
        g_enumresponse.base_level = p.get_byte()
        g_enumresponse.version = p.get_short()

        bitmask = p.get_short()
        b = bin(bitmask)[2:].zfill(16)
        g_enumresponse.allow_joiners = b[0]
        g_enumresponse.damage_teammates = b[1]
        g_enumresponse.lock_teams = b[2]
        g_enumresponse.is_dedicated = b[3]
        g_enumresponse.vimps_disabled = b[4]
        g_enumresponse.weap_bits = b[5]
        g_enumresponse.high_bandwidth = b[6]
        g_enumresponse.no_voting = b[7]

        g_enumresponse.smartie_difficulty = b[8:12]
        g_enumresponse.vimp_difficulty = b[12:16]

        g_enumresponse.points_per_capture = p.get_short()
        g_enumresponse.points_per_kill = p.get_short()
        g_enumresponse.capture_prevent_count = p.get_short()
        g_enumresponse.current_cap_prev_time = p.get_short()

        g_enumresponse.map_checksum = p.get_ulong()
        g_enumresponse.map_name = p.get_string(32)

        if g_enumresponse.version >= 1498:
            g_enumresponse.build = p.get_byte()
            g_enumresponse.revision = p.get_byte()

        if map_name := MapID.get_name(g_enumresponse.map_id):
            g_enumresponse.map_name = map_name

        return g_enumresponse

    def __str__(self):
        return "[%s] Players:%d/%d, Map: %s, Type: %s, Teams: %s, Version: %d, Build: %s, AllowJoiners: %s" % \
               (self.server_name, self.current_players, self.max_players, self.map_name, GameTypes.get_name_by_id(self.game_type), Teams.get_name_by_id(self.teams), self.version, self.build, self.allow_joiners)