CounterStrikeSharp/src/core/managers/player_manager.h

/**
 * =============================================================================
 * SourceMod
 * Copyright (C) 2004-2016 AlliedModders LLC.  All rights reserved.
 * =============================================================================
 *
 * This program is free software; you can redistribute it and/or modify it under
 * the terms of the GNU General Public License, version 3.0, as published by the
 * Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
 * details.
 *
 * You should have received a copy of the GNU General Public License along with
 * this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 * As a special exception, AlliedModders LLC gives you permission to link the
 * code of this program (as well as its derivative works) to "Half-Life 2," the
 * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
 * by the Valve Corporation.  You must obey the GNU General Public License in
 * all respects for all other code used.  Additionally, AlliedModders LLC grants
 * this exception to all derivative works.  AlliedModders LLC defines further
 * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
 * or <http://www.sourcemod.net/license.php>.
 *
 * This file has been modified from its original form, under the GNU General
 * Public License, version 3.0.
 */

#pragma once

#include <string>

#include "core/global_listener.h"
#include "core/globals.h"

class CBaseEntity;
class INetChannelInfo;
class IPlayerInfo;
struct edict_t;

namespace counterstrikesharp {
class ScriptCallback;
class CBaseEntityWrapper;

enum ListenOverride
{
    Listen_Default = 0,
    Listen_Mute,
    Listen_Hear
};

enum VoiceFlagValue
{
    Speak_Normal = 0,
    Speak_Muted = 1 << 0,
    Speak_All = 1 << 1,
    Speak_ListenAll = 1 << 2,
    Speak_Team = 1 << 3,
    Speak_ListenTeam = 1 << 4,
};

typedef uint8_t VoiceFlag_t;

class CPlayer
{
    friend class PlayerManager;

  public:
    CPlayer();

  public:
    void Initialize(const char* name, const char* ip, CPlayerSlot slot);
    void Connect();
    void Disconnect();
    IPlayerInfo* GetPlayerInfo() const;
    bool WasCountedAsInGame() const;
    int GetUserId();
    bool IsAuthStringValidated() const;
    void Authorize();

  public:
    const char* GetName() const;
    const CSteamID* GetSteamId();
    void SetSteamId(const CSteamID* steam_id);
    bool IsConnected() const;
    bool IsFakeClient() const;
    bool IsAuthorized() const;
    void PrintToConsole(const char* message) const;
    //    void PrintToChat(const char *message);
    //    void PrintToHint(const char *message);
    //    void PrintToCenter(const char *message);
    QAngle GetAbsAngles() const;
    Vector GetAbsOrigin() const;
    bool IsAlive() const;
    bool IsInGame() const;
    void Kick(const char* kickReason);
    const char* GetWeaponName() const;
    void ChangeTeam(int team) const;
    int GetTeam() const;
    int GetArmor() const;
    int GetFrags() const;
    int GetDeaths() const;
    const char* GetKeyValue(const char* key) const;
    Vector GetMaxSize() const;
    Vector GetMinSize() const;
    int GetMaxHealth() const;
    const char* GetIpAddress() const;
    const char* GetModelName() const;
    int GetUserId() const;
    float GetTimeConnected() const;
    void SetListen(CPlayerSlot slot, ListenOverride listen);
    void SetVoiceFlags(VoiceFlag_t flags);
    VoiceFlag_t GetVoiceFlags();
    ListenOverride GetListen(CPlayerSlot slot) const;

  public:
    std::string m_name;
    IPlayerInfo* m_info = nullptr;
    std::string m_auth_id;
    bool m_is_connected = false;
    bool m_is_fake_client = false;
    bool m_is_in_game = false;
    bool m_is_authorized = false;
    int m_user_id = 1;
    CPlayerSlot m_slot = CPlayerSlot(-1);
    const CSteamID* m_steamId;
    std::string m_ip_address;
    ListenOverride m_listenMap[66] = {};
    VoiceFlag_t m_voiceFlag = 0;
    CPlayerBitVec m_selfMutes[64] = {};
    void SetName(const char* name);
    INetChannelInfo* GetNetInfo() const;
};

class PlayerManager : public GlobalClass
{
    friend class CPlayer;

  public:
    PlayerManager();
    void OnStartup() override;
    void OnAllInitialized() override;
    bool
    OnClientConnect(CPlayerSlot slot, const char* pszName, uint64 xuid, const char* pszNetworkID, bool unk1, CBufferString* pRejectReason);
    bool OnClientConnect_Post(
        CPlayerSlot slot, const char* pszName, uint64 xuid, const char* pszNetworkID, bool unk1, CBufferString* pRejectReason);
    void OnClientPutInServer(CPlayerSlot slot, char const* pszName, int type, uint64 xuid);
    void
    OnClientDisconnect(CPlayerSlot slot, ENetworkDisconnectionReason reason, const char* pszName, uint64 xuid, const char* pszNetworkID);
    void OnClientDisconnect_Post(
        CPlayerSlot slot, ENetworkDisconnectionReason reason, const char* pszName, uint64 xuid, const char* pszNetworkID) const;
    void OnClientVoice(CPlayerSlot slot) const;
    void OnAuthorized(CPlayer* player) const;
    void OnServerActivate(edict_t* pEdictList, int edictCount, int clientMax) const;
    void OnThink(bool last_tick) const;
    void OnShutdown() override;
    void OnLevelEnd() override;
    void OnClientCommand(CPlayerSlot slot, const CCommand& args) const;
    int ListenClient() const;
    void RunAuthChecks();

  public:
    int NumPlayers() const;
    int MaxClients() const;
    CPlayer* GetPlayerBySlot(int client) const;
    CPlayer* GetClientOfUserId(int user_id) const;

  private:
    void InvalidatePlayer(CPlayer* pPlayer) const;

    CPlayer* m_players;
    int m_max_clients = 0;
    int m_player_count = 0;
    int* m_user_id_lookup;
    int m_listen_client;
    bool m_is_listen_server;
    float m_last_auth_check_time = 0;

    ScriptCallback* m_on_client_connect_callback;
    ScriptCallback* m_on_client_put_in_server_callback;
    ScriptCallback* m_on_client_connected_callback;
    ScriptCallback* m_on_client_disconnect_callback;
    ScriptCallback* m_on_client_disconnect_post_callback;
    ScriptCallback* m_on_client_voice_callback;
    ScriptCallback* m_on_client_authorized_callback;
};

} // namespace counterstrikesharp