space-station-14/Content.Server/Database/BanDef.cs

using System.Collections.Immutable;
using System.Linq;
using System.Net;
using Content.Shared.CCVar;
using Content.Shared.Database;
using Robust.Shared.Configuration;
using Robust.Shared.Network;


namespace Content.Server.Database
{
    public sealed class BanDef
    {
        public int? Id { get; }
        public BanType Type { get; }
        public ImmutableArray<NetUserId> UserIds { get; }
        public ImmutableArray<(IPAddress address, int cidrMask)> Addresses { get; }
        public ImmutableArray<ImmutableTypedHwid> HWIds { get; }

        public DateTimeOffset BanTime { get; }
        public DateTimeOffset? ExpirationTime { get; }
        public ImmutableArray<int> RoundIds { get; }
        public TimeSpan PlaytimeAtNote { get; }
        public string Reason { get; }
        public NoteSeverity Severity { get; set; }
        public NetUserId? BanningAdmin { get; }
        public UnbanDef? Unban { get; }
        public ServerBanExemptFlags ExemptFlags { get; }

        public ImmutableArray<BanRoleDef>? Roles { get; }

        public BanDef(
            int? id,
            BanType type,
            ImmutableArray<NetUserId> userIds,
            ImmutableArray<(IPAddress address, int cidrMask)> addresses,
            ImmutableArray<ImmutableTypedHwid> hwIds,
            DateTimeOffset banTime,
            DateTimeOffset? expirationTime,
            ImmutableArray<int> roundIds,
            TimeSpan playtimeAtNote,
            string reason,
            NoteSeverity severity,
            NetUserId? banningAdmin,
            UnbanDef? unban,
            ServerBanExemptFlags exemptFlags = default,
            ImmutableArray<BanRoleDef>? roles = null)
        {
            if (userIds.Length == 0 && addresses.Length == 0 && hwIds.Length == 0)
            {
                throw new ArgumentException("Must have at least one of banned user, banned address or hardware ID");
            }

            addresses = addresses.Select(address =>
                {
                    if (address is { address.IsIPv4MappedToIPv6: true } addr)
                    {
                        // Fix IPv6-mapped IPv4 addresses
                        // So that IPv4 addresses are consistent between separate-socket and dual-stack socket modes.
                        address = (addr.address.MapToIPv4(), addr.cidrMask - 96);
                    }

                    return address;
                })
                .ToImmutableArray();

            Id = id;
            Type = type;
            UserIds = userIds;
            Addresses = addresses;
            HWIds = hwIds;
            BanTime = banTime;
            ExpirationTime = expirationTime;
            RoundIds = roundIds;
            PlaytimeAtNote = playtimeAtNote;
            Reason = reason;
            Severity = severity;
            BanningAdmin = banningAdmin;
            Unban = unban;
            ExemptFlags = exemptFlags;

            switch (Type)
            {
                case BanType.Server:
                    if (roles != null)
                        throw new ArgumentException("Cannot specify roles for server ban types", nameof(roles));
                    break;

                case BanType.Role:
                    if (roles is not { Length: > 0 })
                        throw new ArgumentException("Must specify roles for server ban types", nameof(roles));
                    if (exemptFlags != 0)
                        throw new ArgumentException("Role bans cannot have exempt flags", nameof(exemptFlags));
                    break;

                default:
                    throw new ArgumentOutOfRangeException(nameof(type));
            }

            Roles = roles;
        }

        public string FormatBanMessage(IConfigurationManager cfg, ILocalizationManager loc)
        {
            string expires;
            if (ExpirationTime is { } expireTime)
            {
                var duration = expireTime - BanTime;
                var utc = expireTime.ToUniversalTime();
                expires = loc.GetString("ban-expires", ("duration", duration.TotalMinutes.ToString("N0")), ("time", utc.ToString("f")));
            }
            else
            {
                var appeal = cfg.GetCVar(CCVars.InfoLinksAppeal);
                expires = !string.IsNullOrWhiteSpace(appeal)
                    ? loc.GetString("ban-banned-permanent-appeal", ("link", appeal))
                    : loc.GetString("ban-banned-permanent");
            }

            return $"""
                   {loc.GetString("ban-banned-1")}
                   {loc.GetString("ban-banned-2", ("reason", Reason))}
                   {expires}
                   {loc.GetString("ban-banned-3")}
                   """;
        }
    }
}