namespace Robust.Shared.GameObjects
{
    public static class EventBusExt
    {
        public static void SubscribeSessionEvent<T>(this IEventBus eventBus, EventSource source,
            IEntityEventSubscriber subscriber, EntitySessionEventHandler<T> handler)
        {
            var wrapper = new HandlerWrapper<T>(handler);
            eventBus.SubscribeEvent<EntitySessionMessage<T>>(source, subscriber, wrapper.Invoke);
        }

        private sealed class HandlerWrapper<T>
        {
            public HandlerWrapper(EntitySessionEventHandler<T> handler)
            {
                Handler = handler;
            }

            public EntitySessionEventHandler<T> Handler { get; }

            public void Invoke(EntitySessionMessage<T> msg)
            {
                Handler(msg.Message, msg.EventArgs);
            }

            private bool Equals(HandlerWrapper<T> other)
            {
                return Handler.Equals(other.Handler);
            }

            public override bool Equals(object obj)
            {
                return ReferenceEquals(this, obj) || obj is HandlerWrapper<T> other && Equals(other);
            }

            public override int GetHashCode()
            {
                return Handler.GetHashCode();
            }
        }
    }
}