ZonyLrcToolsX/src/ZonyLrcTools.Cli/Infrastructure/DependencyInject/AutoDependencyInjectExtensions.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Reflection;
using Microsoft.Extensions.DependencyInjection;
using ZonyLrcTools.Cli.Infrastructure.Extensions;

namespace ZonyLrcTools.Cli.Infrastructure.DependencyInject
{
    public static class AutoDependencyInjectExtensions
    {
        /// <summary>
        /// 开始进行自动依赖注入。
        /// </summary>
        /// <remarks>
        /// 会根据实现了 <see cref="ITransientDependency"/> 或 <see cref="ISingletonDependency"/> 的接口进行自动注入。
        /// </remarks>
        /// <param name="services">服务定义集合。</param>
        /// <typeparam name="TAssemblyType">需要注入的任意类型。</typeparam>
        public static IServiceCollection BeginAutoDependencyInject<TAssemblyType>(this IServiceCollection services)
        {
            var allTypes = typeof(TAssemblyType).Assembly
                .GetTypes()
                .Where(t => t.IsClass && !t.IsAbstract && !t.IsGenericType)
                .ToArray();

            var transientTypes = allTypes.Where(t => typeof(ITransientDependency).IsAssignableFrom(t));
            var singletonTypes = allTypes.Where(t => typeof(ISingletonDependency).IsAssignableFrom(t));

            transientTypes.Foreach(t =>
            {
                foreach (var exposedService in GetDefaultExposedTypes(t))
                {
                    services.Add(CreateServiceDescriptor(t, exposedService, ServiceLifetime.Transient));
                }
            });

            singletonTypes.Foreach(t =>
            {
                foreach (var exposedService in GetDefaultExposedTypes(t))
                {
                    services.Add(CreateServiceDescriptor(t, exposedService, ServiceLifetime.Singleton));
                }
            });

            return services;
        }

        public static List<Type> GetDefaultExposedTypes(Type type)
        {
            var serviceTypes = new List<Type>();

            foreach (var interfaceType in type.GetTypeInfo().GetInterfaces())
            {
                var interfaceName = interfaceType.Name;

                if (interfaceName.StartsWith("I"))
                {
                    interfaceName = interfaceName.Substring(1, interfaceName.Length - 1);
                }

                if (type.Name.EndsWith(interfaceName))
                {
                    serviceTypes.Add(interfaceType);
                    serviceTypes.Add(type);
                }
            }

            return serviceTypes;
        }

        public static ServiceDescriptor CreateServiceDescriptor(Type implementationType,
            Type exposingServiceType,
            ServiceLifetime lifetime)
        {
            return new ServiceDescriptor(exposingServiceType, implementationType, lifetime);
        }
    }
}