player_limit

# =============================================================================
# >> IMPORTS
# =============================================================================
# Source.Python
import core

from commands.server import ServerCommand
from filters.players import PlayerIter
from listeners import ClientConnect
from engines.server import global_vars


# =============================================================================
# >> GLOBAL VARIABLES
# =============================================================================
max_players = global_vars.max_clients


# =============================================================================
# >> LISTENERS
# =============================================================================
@ServerCommand('pl_change_max_players')
def on_change_max_players(command):
    invalid_usage = 'pl_change_max_players <max_players>'
    if command.get_arg_count() != 2:
        core.echo_console(invalid_usage)
        return

    global max_players
    try:
        max_players = int(command[1])
    except:
        core.echo_console(invalid_usage)

@ClientConnect
def on_client_connect(allow_connect_ptr, entity, name, address,
        reject_msg_ptr, max_reject_length):
    """Called when a player tries to connect."""
    if len(PlayerIter()) < max_players:
        return

    reject_message = 'Player limit is set to "{0}".'.format(max_players)
    reject_msg_ptr.set_string_array(reject_message[:max_reject_length])
    allow_connect_ptr.set_bool(False)