LidgrenEndpoint.cs

#nullable enable
using System.Linq;
using System.Net;
 
namespace Barotrauma.Networking
{
    sealed class LidgrenEndpoint : Endpoint
    {
        public readonly IPEndPoint NetEndpoint;
 
        public int Port => NetEndpoint.Port;
 
        public override string StringRepresentation
            => NetEndpoint.ToString();
 
        public override LocalizedString ServerTypeString { get; } = TextManager.Get("DedicatedServer");
 
        public LidgrenEndpoint(IPAddress address, int port) : this(new IPEndPoint(address, port)) { }
 
        public LidgrenEndpoint(IPEndPoint netEndpoint) : base(new LidgrenAddress(netEndpoint.Address))
        {
            NetEndpoint = new IPEndPoint((Address as LidgrenAddress)!.NetAddress, netEndpoint.Port);
        }
 
        public new static Option<LidgrenEndpoint> Parse(string endpointStr)
        {
            return ParseFromWithHostNameCheck(endpointStr, tryParseHostName: false);
        }
 
        public static Option<LidgrenEndpoint> ParseFromWithHostNameCheck(string endpointStr, bool tryParseHostName)
        {
            string hostName = endpointStr;
            int port = NetConfig.DefaultPort;
            if (endpointStr.Count(c => c == ':') == 1)
            {
                string[] split = endpointStr.Split(':');
                hostName = split[0];
                port = int.TryParse(split[1], out var tmpPort) ? tmpPort : port;
            }
            
            if (LidgrenAddress.Parse(hostName).TryUnwrap(out var adr) || 
                (tryParseHostName && LidgrenAddress.ParseHostName(hostName).TryUnwrap(out adr)))
            {
                return Option<LidgrenEndpoint>.Some(new LidgrenEndpoint(adr.NetAddress, port));
            }
            
            return IPEndPoint.TryParse(endpointStr, out IPEndPoint? netEndpoint)
                ? Option<LidgrenEndpoint>.Some(new LidgrenEndpoint(netEndpoint))
                : Option<LidgrenEndpoint>.None();
        }
 
        public override bool Equals(object? obj)
            => obj switch
            {
                LidgrenEndpoint otherEndpoint => this == otherEndpoint,
                _ => false
            };
 
        public override int GetHashCode()
            => NetEndpoint.GetHashCode();
 
        public static bool operator ==(LidgrenEndpoint a, LidgrenEndpoint b)
            => a.NetEndpoint.EquivalentTo(b.NetEndpoint);
 
        public static bool operator !=(LidgrenEndpoint a, LidgrenEndpoint b)
            => !(a == b);
    }
}