BetterAmongUs/Modules/ExtendedPlayerControl.cs

using AmongUs.GameOptions;
using Hazel;
using InnerNet;
using TMPro;
using UnityEngine;

namespace BetterAmongUs;

static class ExtendedPlayerControl
{
    // Get players client
    public static ClientData? GetClient(this PlayerControl player)
    {
        try
        {
            var client = AmongUsClient.Instance.allClients.ToArray().FirstOrDefault(cd => cd.Character.PlayerId == player.PlayerId);
            return client;
        }
        catch
        {
            return null;
        }
    }
    // Get players client id
    public static int GetClientId(this PlayerControl player)
    {
        if (player == null) return -1;
        var client = player.GetClient();
        return client == null ? -1 : client.Id;
    }
    // Set players over head text
    public static void SetPlayerTextInfo(this PlayerControl player, string text, bool isBottom = false, bool isInfo = false)
    {
        if (player == null) return;
        GameObject playerbase = player.gameObject;
        string InfoType = isBottom ? "InfoText_B_TMP" : "InfoText_T_TMP";

        if (isInfo)
        {
            InfoType = "InfoText_Info_TMP";
            GameObject TopText = GameObject.Find($"{playerbase.transform.name}/Names/NameText_TMP/InfoText_T_TMP");
            if (TopText != null)
            {
                if (TopText.GetComponent<TextMeshPro>()?.text.Replace("<size=65%>", string.Empty).Replace("</size>", string.Empty).Length < 1)
                {
                    text = "<voffset=-2.25em>" + text + "</voffset>";
                }
            }
        }

        text = "<size=65%>" + text + "</size>";
        GameObject TextObj = GameObject.Find($"{player?.gameObject.transform.name}/Names/NameText_TMP/{InfoType}");
        if (TextObj != null)
        {
            TextObj.GetComponent<TextMeshPro>().text = text;
        }
    }
    // Reset players over head text
    public static void ResetAllPlayerTextInfo(this PlayerControl player)
    {
        if (player == null) return;
        player.SetPlayerTextInfo("", isInfo: true);
        player.SetPlayerTextInfo("");
        player.SetPlayerTextInfo("", isBottom: true);
    }
    // Check if players character has been created and received from the Host
    public static bool DataIsCollected(this PlayerControl player)
    {
        if (player == null) return false;

        if (player.gameObject.transform.Find("Names/NameText_TMP").GetComponent<TextMeshPro>().text
        is "???" or "Player" or "<color=#b5b5b5>Loading</color>" or null
        || player.Data == null
        || string.IsNullOrEmpty(player.Data.Puid)
        || player.Data.PlayerLevel == uint.MaxValue
        || player.CurrentOutfit == null
        || player.CurrentOutfit.ColorId == -1)
        {
            return false;
        }
        return true;
    }

    // RPCs
    public static void RpcSendHostChat(this PlayerControl player, string text, string title = "<color=#ffffff><b>(<color=#00ff44>System Message</color>)</b>", bool sendToBetterUser = true)
    {
        if (player == null) return;
        RPC.SendHostChatToPlayer(player, text, title, sendToBetterUser);
    }
    public static void RpcSetNamePrivate(this PlayerControl player, string name, PlayerControl target)
    {
        if (player == null || target == null) return;
        RPC.SetNamePrivate(player, name, target);
    }
    public static void RpcExile(this PlayerControl player)
    {
        if (player == null) return;
        RPC.ExileAsync(player);
    }
    public static void RpcTeleport(this PlayerControl player, Vector2 position)
    {

        var cancelTeleport = false;

        if (player.inVent || player.MyPhysics.Animations.IsPlayingEnterVentAnimation())
        {
            cancelTeleport = true;
        }
        else if (player.onLadder || player.MyPhysics.Animations.IsPlayingAnyLadderAnimation())
        {
            cancelTeleport = true;
        }
        else if (player.inMovingPlat)
        {
            cancelTeleport = true;
        }

        if (cancelTeleport)
        {
            return;
        }

        var netTransform = player.NetTransform;

        if (AmongUsClient.Instance.AmClient)
        {
            // +328 because lastSequenceId has delay between the host and the vanilla client
            // And this cannot forced teleport the player
            netTransform.SnapTo(position, (ushort)(netTransform.lastSequenceId + 328));
        }

        ushort newSid = (ushort)(netTransform.lastSequenceId + 8);
        MessageWriter messageWriter = AmongUsClient.Instance.StartRpcImmediately(netTransform.NetId, (byte)RpcCalls.SnapTo, SendOption.Reliable);
        NetHelpers.WriteVector2(position, messageWriter);
        messageWriter.Write(newSid);
        AmongUsClient.Instance.FinishRpcImmediately(messageWriter);
    }

    public static bool IsInRoomSelect(this PlayerControl player)
    {
        if (player == null) return false;
        return GameStates.AirshipIsActive && Vector2.Distance(player.GetTruePosition(), new(-25, 40)) < 5f;
    }

    // Get vent Id that the player is in.
    public static int GetPlayerVentId(this PlayerControl player)
    {
        if (player == null) return -1;

        if (!(ShipStatus.Instance.Systems.TryGetValue(SystemTypes.Ventilation, out var systemType) &&
              systemType.TryCast<VentilationSystem>() is VentilationSystem ventilationSystem))
            return 0;

        return ventilationSystem.PlayersInsideVents.TryGetValue(player.PlayerId, out var playerIdVentId) ? playerIdVentId : -1;
    }

    public static Vector2 GetCustomPosition(this PlayerControl player) => new(player.transform.position.x, player.transform.position.y);

    // Check if player is a dev
    public static bool IsDev(this PlayerControl player) => player != null && Main.DevUser.Contains(Utils.GetHashPuid(player));
    // Check if player is alive
    public static bool IsAlive(this PlayerControl player) => player?.Data != null && !player.Data.IsDead;
    // Check if player is in a vent
    public static bool IsInVent(this PlayerControl player) => player != null && (player.inVent || player.walkingToVent || player.MyPhysics?.Animations?.IsPlayingEnterVentAnimation() == true);
    // Check if player role name
    public static string GetRoleName(this PlayerControl player) => player?.Data != null && Main.GetRoleName.TryGetValue((int)player.Data.RoleType, out var roleName) ? roleName : string.Empty;
    // Check if player is Shapeshifting
    public static bool IsInShapeshift(this PlayerControl player) => player != null && (player.shapeshiftTargetPlayerId > -1 || player.shapeshifting);
    // Check if player is in vanish as Phantom
    public static bool IsInVanish(this PlayerControl player)
    {
        if (player != null && player.Data.Role is PhantomRole phantomRole)
        {
            return phantomRole.fading;
        }
        return false;
    }
    // Get hex color for team
    public static string GetTeamHexColor(this PlayerControl player)
    {
        if (player == null) return "#ffffff";

        if (player.IsImpostorTeam())
        {
            return "#f00202";
        }
        else
        {
            return "#8cffff";
        }
    }
    // Check if player is on imposter team
    public static bool IsImpostorTeam(this PlayerControl player) => player?.Data != null && (player.Data.RoleType is RoleTypes.Impostor or RoleTypes.ImpostorGhost or RoleTypes.Shapeshifter or RoleTypes.Phantom);
    // Check if player is a imposter teammate
    public static bool IsImpostorTeammate(this PlayerControl player) =>
        player != null && PlayerControl.LocalPlayer != null &&
        ((player == PlayerControl.LocalPlayer && PlayerControl.LocalPlayer.IsImpostorTeam()) ||
        (PlayerControl.LocalPlayer.IsImpostorTeam() && player.IsImpostorTeam()));
    // Check if player is in the Anti-Cheat list
    public static bool IsCheater(this PlayerControl player) =>
        player != null && (AntiCheat.PlayerData.ContainsKey(Utils.GetHashPuid(player)) ||
                           AntiCheat.SickoData.ContainsKey(Utils.GetHashPuid(player)) ||
                           AntiCheat.AUMData.ContainsKey(Utils.GetHashPuid(player)));
    // Check if player is the host
    public static bool IsHost(this PlayerControl player) => player?.Data != null && GameData.Instance?.GetHost()?.Puid == player.Data.Puid;

}