using Unity.Netcode.Transports.UTP;
using Unity.Services.Relay.Models;

namespace PashaBibko.PenguinChase.Extensions
{
    public static class UnityTransportExtensions
    {
        public static void SetHostRelayData(this UnityTransport transport, Allocation allocation)
        {
            transport.SetHostRelayData
            (
                allocation.RelayServer.IpV4,
                (ushort)allocation.RelayServer.Port,
                allocation.AllocationIdBytes,
                allocation.Key,
                allocation.ConnectionData
            );
        }

        public static void SetRelayServerData(this UnityTransport transport, JoinAllocation allocation)
        {
            transport.SetRelayServerData
            (
                allocation.RelayServer.IpV4,
                (ushort)allocation.RelayServer.Port,
                allocation.AllocationIdBytes,
                allocation.Key,
                allocation.ConnectionData,
                allocation.HostConnectionData
            );
        }
    }
}