Empostor/src/Empostor.Server/Service/Reactor/ReactorModExtensions.cs

using System.Runtime.CompilerServices;
using Empostor.Api.Net;

namespace Empostor.Server.Service.Reactor
{

    public static class ReactorModExtensions
    {
        private static readonly ConditionalWeakTable<IHazelConnection, ReactorModInfo> Table = new();

        public static ReactorModInfo? GetReactorMods(this IClient client)
            => client.Connection?.GetReactorMods();

        public static ReactorModInfo? GetReactorMods(this IHazelConnection connection)
            => Table.TryGetValue(connection, out var info) ? info : null;

        internal static void SetReactorMods(this IHazelConnection connection, ReactorModInfo info)
            => Table.AddOrUpdate(connection, info);
    }
}