using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Logging;

namespace ConfectioneryContracts.DI
{
    public class ServiceDependencyContainer : IDependencyContainer
    {
        private ServiceProvider? _serviceProvider;

        private readonly ServiceCollection _serviceCollection;

        public ServiceDependencyContainer()
        {
            _serviceCollection = new ServiceCollection();
        }

        public void AddLogging(Action<ILoggingBuilder> configure)
        {
            _serviceCollection.AddLogging(configure);
        }

        public void RegisterType<T, U>(bool isSingle) where U : class, T where T : class
        {
            if (isSingle)
            {
                _serviceCollection.AddSingleton<T, U>();
            }
            else
            {
                _serviceCollection.AddTransient<T, U>();
            }
            _serviceProvider = null;
        }

        public void RegisterType<T>(bool isSingle) where T : class
        {
            if (isSingle)
            {
                _serviceCollection.AddSingleton<T>();
            }
            else
            {
                _serviceCollection.AddTransient<T>();
            }
            _serviceProvider = null;
        }

        public T Resolve<T>()
        {
            if (_serviceProvider == null)
            {
                _serviceProvider = _serviceCollection.BuildServiceProvider();
            }
            return _serviceProvider.GetService<T>()!;
        }
    }
}