Discord.Net/src/Discord.Net.Interactions/Utilities/ModalUtils.cs

using Discord.Interactions.Builders;
using System;
using System.Collections.Concurrent;
using System.Collections.Generic;

namespace Discord.Interactions
{
    internal static class ModalUtils
    {
        private static readonly ConcurrentDictionary<Type, ModalInfo> _modalInfos = new();

        public static IReadOnlyCollection<ModalInfo> Modals => _modalInfos.Values.ToReadOnlyCollection();

        public static ModalInfo GetOrAdd(Type type, InteractionService interactionService)
        {
            if (!typeof(IModal).IsAssignableFrom(type))
                throw new ArgumentException($"Must be an implementation of {nameof(IModal)}", nameof(type));

            return _modalInfos.GetOrAdd(type, ModuleClassBuilder.BuildModalInfo(type, interactionService));
        }

        public static ModalInfo GetOrAdd<T>(InteractionService interactionService) where T : class, IModal
            => GetOrAdd(typeof(T), interactionService);

        public static bool TryGet(Type type, out ModalInfo modalInfo)
        {
            if (!typeof(IModal).IsAssignableFrom(type))
                throw new ArgumentException($"Must be an implementation of {nameof(IModal)}", nameof(type));

            return _modalInfos.TryGetValue(type, out modalInfo);
        }

        public static bool TryGet<T>(out ModalInfo modalInfo) where T : class, IModal
            => TryGet(typeof(T), out modalInfo);

        public static bool TryRemove(Type type, out ModalInfo modalInfo)
        {
            if (!typeof(IModal).IsAssignableFrom(type))
                throw new ArgumentException($"Must be an implementation of {nameof(IModal)}", nameof(type));

            return _modalInfos.TryRemove(type, out modalInfo);
        }

        public static bool TryRemove<T>(out ModalInfo modalInfo) where T : class, IModal
            => TryRemove(typeof(T), out modalInfo);

        public static void Clear() => _modalInfos.Clear();

        public static int Count() => _modalInfos.Count;
    }
}