BarotraumaClient/ClientSource/Networking/BanList.cs

#nullable enable
using Microsoft.Xna.Framework;
using System;
using System.Collections.Generic;
 
namespace Barotrauma.Networking
{
    partial class BannedPlayer
    {
        public BannedPlayer(
            UInt32 uniqueIdentifier,
            string name,
            Either<Address, AccountId> addressOrAccountId,
            string reason,
            Option<SerializableDateTime> expiration)
        {
            this.Name = name;
            this.AddressOrAccountId = addressOrAccountId;
            this.UniqueIdentifier = uniqueIdentifier;
            this.Reason = reason;
            this.ExpirationTime = expiration;
        }
    }
 
    partial class BanList
    {
        public GUIComponent? BanFrame { get; private set; }
 
        public List<UInt32> localRemovedBans = new List<UInt32>();
 
        private void RecreateBanFrame()
        {
            if (BanFrame != null)
            {
                var parent = BanFrame.Parent;
                parent.RemoveChild(BanFrame);
                CreateBanFrame(parent);
            }
        }
 
        public GUIComponent CreateBanFrame(GUIComponent parent)
        {
            BanFrame = new GUIListBox(new RectTransform(Vector2.One, parent.RectTransform, Anchor.Center));
 
            foreach (BannedPlayer bannedPlayer in bannedPlayers)
            {
                if (localRemovedBans.Contains(bannedPlayer.UniqueIdentifier)) { continue; }
 
                var playerFrame = new GUIFrame(new RectTransform(new Vector2(1.0f, 0.2f), ((GUIListBox)BanFrame).Content.RectTransform) { MinSize = new Point(0, 70) }, style: "InnerFrame")
                {
                    UserData = BanFrame
                };
 
                var paddedPlayerFrame = new GUILayoutGroup(new RectTransform(new Vector2(0.95f, 0.85f), playerFrame.RectTransform, Anchor.Center))
                {
                    Stretch = true,
                    RelativeSpacing = 0.05f,
                    CanBeFocused = true
                };
 
                var topArea = new GUILayoutGroup(new RectTransform(new Vector2(1.0f, 0.0f), paddedPlayerFrame.RectTransform), 
                    isHorizontal: true, childAnchor: Anchor.CenterLeft)
                {
                    Stretch = true,
                    RelativeSpacing = 0.02f
                };
 
                var addressOrAccountId = bannedPlayer.AddressOrAccountId;
 
                string nameText = bannedPlayer.Name;
                if (addressOrAccountId.TryCast(out Address address))
                {
                    nameText += $" ({address.StringRepresentation})";
                }
                else if (addressOrAccountId.TryCast(out AccountId accountId))
                {
                    nameText += $" ({accountId.StringRepresentation})";
                }
                GUITextBlock textBlock = new GUITextBlock(new RectTransform(new Vector2(0.5f, 1.0f), topArea.RectTransform), nameText) 
                { 
                    CanBeFocused = true 
                };
                textBlock.RectTransform.MinSize = new Point(0, (int)textBlock.Font.MeasureString(textBlock.Text.SanitizedValue).Y);
 
                var removeButton = new GUIButton(new RectTransform(new Vector2(0.2f, 0.4f), topArea.RectTransform, Anchor.CenterRight), 
                    TextManager.Get("BanListRemove"), style: "GUIButtonSmall")
                {
                    IgnoreLayoutGroups = true,
                    UserData = bannedPlayer,
                    OnClicked = RemoveBan,
                    Enabled = false
                };
                removeButton.OnAddedToGUIUpdateList += (component) =>
                {
                    component.Enabled = GameMain.Client?.HasPermission(ClientPermissions.Unban) ?? false;
                };
                topArea.RectTransform.MinSize = new Point(0, Math.Max(textBlock.RectTransform.MinSize.Y, removeButton.RectTransform.MinSize.Y));
                topArea.RectTransform.IsFixedSize = true;
 
                new GUITextBlock(new RectTransform(new Vector2(1.0f, 0.0f), paddedPlayerFrame.RectTransform),
                    bannedPlayer.ExpirationTime.TryUnwrap(out var expirationTime)
                        ? TextManager.GetWithVariable("BanExpires", "[time]", expirationTime.ToLocalUserString())
                        : TextManager.Get("BanPermanent"),
                    font: GUIStyle.SmallFont);
 
                LocalizedString reason = TextManager.GetServerMessage(bannedPlayer.Reason).Fallback(bannedPlayer.Reason);
                var reasonText = new GUITextBlock(new RectTransform(new Vector2(1.0f, 0.0f), paddedPlayerFrame.RectTransform),
                    TextManager.Get("BanReason") + " " +
                        (string.IsNullOrEmpty(bannedPlayer.Reason) ? TextManager.Get("None") : reason),
                    font: GUIStyle.SmallFont, wrap: true)
                {
                    ToolTip = reason
                };
 
                paddedPlayerFrame.Recalculate();
 
                new GUIFrame(new RectTransform(new Vector2(1.0f, 0.01f), ((GUIListBox)BanFrame).Content.RectTransform), style: "HorizontalLine");
            }
 
            return BanFrame;
        }
 
        private bool RemoveBan(GUIButton button, object obj)
        {
            if (obj is not BannedPlayer banned) { return false; }
 
            localRemovedBans.Add(banned.UniqueIdentifier);
            RecreateBanFrame();
 
            GameMain.Client?.ServerSettings?.ClientAdminWrite(ServerSettings.NetFlags.Properties);
 
            return true;
        }
        
        public void ClientAdminRead(IReadMessage incMsg)
        {
            bool hasPermission = incMsg.ReadBoolean();
            if (!hasPermission)
            {
                incMsg.ReadPadBits();
                return;
            }
 
            bool isOwner = incMsg.ReadBoolean();
            incMsg.ReadPadBits();
 
            bannedPlayers.Clear();
            UInt32 bannedPlayerCount = incMsg.ReadVariableUInt32();
 
            for (int i = 0; i < (int)bannedPlayerCount; i++)
            {
                string name = incMsg.ReadString();
                UInt32 uniqueIdentifier = incMsg.ReadUInt32();
                bool includesExpiration = incMsg.ReadBoolean();
                incMsg.ReadPadBits();
 
                Option<SerializableDateTime> expiration = Option<SerializableDateTime>.None();
                if (includesExpiration)
                {
                    double hoursFromNow = incMsg.ReadDouble();
                    expiration = Option<SerializableDateTime>.Some(SerializableDateTime.LocalNow + TimeSpan.FromHours(hoursFromNow));
                }
 
                string reason = incMsg.ReadString();
 
                Either<Address, AccountId> addressOrAccountId;
                if (isOwner)
                {
                    bool isAddress = incMsg.ReadBoolean();
                    incMsg.ReadPadBits();
                    string str = incMsg.ReadString();
                    if (isAddress && Address.Parse(str).TryUnwrap(out var address))
                    {
                        addressOrAccountId = address;
                    }
                    else if (AccountId.Parse(str).TryUnwrap(out var accountId))
                    {
                        addressOrAccountId = accountId;
                    }
                    else
                    {
                        continue;
                    }
                }
                else
                {
                    addressOrAccountId = new UnknownAddress();
                }
                bannedPlayers.Add(new BannedPlayer(uniqueIdentifier, name, addressOrAccountId, reason, expiration));
            }
 
            if (BanFrame != null)
            {
                var parent = BanFrame.Parent;
                parent.RemoveChild(BanFrame);
                CreateBanFrame(parent);
            }
        }
 
        public void ClientAdminWrite(IWriteMessage outMsg)
        {
            outMsg.WriteVariableUInt32((UInt32)localRemovedBans.Count);
            foreach (UInt32 uniqueId in localRemovedBans)
            {
                outMsg.WriteUInt32(uniqueId);
            }
 
            localRemovedBans.Clear();
        }
    }
}