RAGECOOP-V/Client/Scripts/PlayerList.cs

using System.Collections.Generic;
using System.Linq;
using System.Net;
using GTA;
using GTA.Math;
using GTA.Native;
using Lidgren.Network;
using Newtonsoft.Json;
using RageCoop.Client.Menus;
using RageCoop.Client.Scripting;
using RageCoop.Core;

namespace RageCoop.Client
{
    internal static class PlayerList
    {
        private const float LEFT_POSITION = 0.122f;
        private const float RIGHT_POSITION = 0.9f;
        private static readonly Scaleform _mainScaleform = new Scaleform("mp_mm_card_freemode");
        private static ulong _lastUpdate = Util.GetTickCount64();

        public static bool LeftAlign = true;
        public static Dictionary<int, Player> Players = new Dictionary<int, Player>();
        public static ulong Pressed { get; set; }

        public static void Tick()
        {
            if (!Networking.IsOnServer) return;

            if (Util.GetTickCount64() - _lastUpdate >= 1000) Update();

            if (Util.GetTickCount64() - Pressed < 5000 && !Main.MainChat.Focused
#if !NON_INTERACTIVE
                                                       && !CoopMenu.MenuPool.AreAnyVisible
#endif
               )
                Call(DRAW_SCALEFORM_MOVIE, _mainScaleform.Handle,
                    LeftAlign ? LEFT_POSITION : RIGHT_POSITION, 0.3f,
                    0.28f, 0.6f,
                    255, 255, 255, 255, 0);
        }

        private static void Update()
        {
            _lastUpdate = Util.GetTickCount64();

            _mainScaleform.CallFunction("SET_DATA_SLOT_EMPTY", 0);

            var i = 0;

            foreach (var player in Players.Values)
                _mainScaleform.CallFunction("SET_DATA_SLOT", i++, $"{player.Ping * 1000:N0}ms",
                    player.Username + (player.IsHost ? " (Host)" : ""), 116, 0, i - 1, "", "", 2, "", "", ' ');

            _mainScaleform.CallFunction("SET_TITLE", "Player list", $"{Players.Count} players");
            _mainScaleform.CallFunction("DISPLAY_VIEW");
        }

        public static void SetPlayer(int id, string username, float latency = 0)
        {
            Main.Logger.Debug($"{id},{username},{latency}");
            if (Players.TryGetValue(id, out var p))
            {
                p.Username = username;
                p.ID = id;
                p._latencyToServer = latency;
            }
            else
            {
                p = new Player { ID = id, Username = username, _latencyToServer = latency };
                Players.Add(id, p);
            }
        }

        public static void UpdatePlayer(Packets.PlayerInfoUpdate packet)
        {
            var p = GetPlayer(packet.PedID);
            if (p != null)
            {
                p._latencyToServer = packet.Latency;
                p.Position = packet.Position;
                p.IsHost = packet.IsHost;
                API.QueueAction(() =>
                {
                    if (p.FakeBlip?.Exists() != true) p.FakeBlip = World.CreateBlip(p.Position);
                    if (EntityPool.PedExists(p.ID))
                    {
                        p.FakeBlip.DisplayType = BlipDisplayType.NoDisplay;
                    }
                    else
                    {
                        p.FakeBlip.Color = API.Config.BlipColor;
                        p.FakeBlip.Scale = API.Config.BlipScale;
                        p.FakeBlip.Sprite = API.Config.BlipSprite;
                        p.FakeBlip.DisplayType = BlipDisplayType.Default;
                        p.FakeBlip.Position = p.Position;
                    }
                });
            }
        }

        public static Player GetPlayer(int id)
        {
            Players.TryGetValue(id, out var p);
            return p;
        }

        public static Player GetPlayer(SyncedPed p)
        {
            var player = GetPlayer(p.ID);
            if (player != null) player.Character = p;
            return player;
        }

        public static void RemovePlayer(int id)
        {
            if (Players.TryGetValue(id, out var player))
            {
                Players.Remove(id);
                API.QueueAction(() => player.FakeBlip?.Delete());
            }
        }

        public static void Cleanup()
        {
            foreach (var p in Players.Values.ToArray()) p.FakeBlip?.Delete();
            Players = new Dictionary<int, Player>();
        }
    }

    internal class Player
    {
        internal float _latencyToServer;
        internal bool ConnectWhenPunched { get; set; }

        [JsonIgnore]
        public Blip FakeBlip { get; internal set; }
        [JsonIgnore]
        public Vector3 Position { get; internal set; }
        [JsonIgnore]
        public SyncedPed Character { get; internal set; }

        [JsonIgnore]
        public NetConnection Connection { get; internal set; }

        public byte HolePunchStatus { get; internal set; } = 1;
        public bool IsHost { get; internal set; }
        public string Username { get; internal set; }

        /// <summary>
        ///     Universal ped ID.
        /// </summary>
        public int ID { get; internal set; }

        public int EntityHandle => Character?.MainPed?.Handle ?? 0;

        public IPEndPoint InternalEndPoint { get; internal set; }
        public IPEndPoint ExternalEndPoint { get; internal set; }

        /// <summary>
        ///     Player round-trip time in seconds, will be the rtt to server if not using P2P connection.
        /// </summary>
        public float Ping => Main.LocalPlayerID == ID ? Networking.Latency * 2 :
            HasDirectConnection ? Connection.AverageRoundtripTime : _latencyToServer * 2;

        public float PacketTravelTime => HasDirectConnection
            ? Connection.AverageRoundtripTime / 2
            : Networking.Latency + _latencyToServer;

        public bool DisplayNameTag { get; set; } = true;
        public bool HasDirectConnection => Connection?.Status == NetConnectionStatus.Connected;
    }
}