#include maps\mp\gametypes_zm\_globallogic_ui;
#include maps\mp\_utility;
#include maps\mp\gametypes_zm\_hud_util;
#include maps\mp\gametypes_zm\_globallogic;
#include maps\mp\gametypes_zm\_spectating;
#include maps\mp\gametypes_zm\_globallogic_player;

menuautoassign( comingfrommenu )
{
    teamkeys = getarraykeys( level.teams );
    assignment = self get_assigned_team();
    self closemenus();

    self.pers["team"] = assignment;
    self.team = assignment;
    self.pers["class"] = undefined;
    self.class = undefined;
    self.pers["weapon"] = undefined;
    self.pers["savedmodel"] = undefined;
    self updateobjectivetext();

    if ( level.teambased )
        self.sessionteam = assignment;
    else
    {
        self.sessionteam = "none";
        self.ffateam = assignment;
    }

    self notify( "joined_team" );
    level notify( "joined_team" );
    self notify( "end_respawn" );
    self beginclasschoice();
    self setclientscriptmainmenu( game["menu_class"] );
}

get_assigned_team()
{
	teamplayers = [];
	teamplayers["axis"] = 0;
	teamplayers["allies"] = 0;

    players = get_players();
    foreach ( player in players )
    {
        if ( !isDefined(player.team) || (player.team != "axis" && player.team != "allies") )
        {
            continue;
        }

        if ( player == self )
        {
            continue;
        }

        teamplayers[player.team]++;
    }

    if ( teamplayers["axis"] <= teamplayers["allies"] )
    {
        guids = strTok(getDvar("team_axis"), " ");
        foreach ( guid in guids )
        {
            if ( self getguid() == int(guid) )
            {
                arrayRemoveValue(guids, guid);

                guid_text = "";
                foreach (guid in guids)
                {
                    guid_text += guid + " ";
                }

                setDvar("team_axis", guid_text);

                return "axis";
            }
        }
    }

    if ( teamplayers["allies"] <= teamplayers["axis"] )
    {
        guids = strTok(getDvar("team_allies"), " ");
        foreach ( guid in guids )
        {
            if ( self getguid() == int(guid) )
            {
                arrayRemoveValue(guids, guid);

                guid_text = "";
                foreach (guid in guids)
                {
                    guid_text += guid + " ";
                }

                setDvar("team_allies", guid_text);

                return "allies";
            }
        }
    }

	if ( teamplayers["allies"] == teamplayers["axis"] )
	{
		if ( randomint( 100 ) >= 50 )
		{
			return "axis";
		}
		else
		{
			return "allies";
		}
	}
	else
	{
		if ( teamplayers["allies"] > teamplayers["axis"] )
		{
			return "axis";
		}
		else
		{
			return "allies";
		}
	}
}