Telegrator/src/Telegrator.Hosting/Providers/HostHandlersCollection.cs

using Microsoft.Extensions.DependencyInjection;
using Telegrator.Core;
using Telegrator.Core.Descriptors;

namespace Telegrator.Providers
{
    /// <inheritdoc/>
    public class HostHandlersCollection(IServiceCollection hostServiceColletion, TelegratorOptions options) : HandlersCollection(options)
    {
        private readonly IServiceCollection Services = hostServiceColletion;

        /// <inheritdoc/>
        protected override bool MustHaveParameterlessCtor => false;

        /// <inheritdoc/>
        public override IHandlersCollection AddDescriptor(HandlerDescriptor descriptor)
        {
            switch (descriptor.Type)
            {
                case DescriptorType.General:
                    {
                        if (descriptor.InstanceFactory != null)
                            Services.AddScoped(descriptor.HandlerType, _ => descriptor.InstanceFactory.Invoke());
                        else
                            Services.AddScoped(descriptor.HandlerType);

                        break;
                    }

                case DescriptorType.Keyed:
                    {
                        if (descriptor.InstanceFactory != null)
                            Services.AddKeyedScoped(descriptor.HandlerType, descriptor.ServiceKey, (_, _) => descriptor.InstanceFactory.Invoke());
                        else
                            Services.AddKeyedScoped(descriptor.HandlerType, descriptor.ServiceKey);

                        break;
                    }

                case DescriptorType.Singleton:
                    {
                        Services.AddSingleton(descriptor.HandlerType, descriptor.SingletonInstance ?? (descriptor.InstanceFactory != null
                            ? descriptor.InstanceFactory.Invoke()
                            : throw new Exception()));

                        break;
                    }

                case DescriptorType.Implicit:
                    {
                        Services.AddKeyedSingleton(descriptor.HandlerType, descriptor.ServiceKey, descriptor.SingletonInstance ?? (descriptor.InstanceFactory != null
                            ? descriptor.InstanceFactory.Invoke()
                            : throw new Exception()));
                        
                        break;
                    }
            }

            return base.AddDescriptor(descriptor);
        }
    }
}