添加项目文件。

2026-08-14 18:00:58 +08:00
1 parent 6a017c9547
commit 58abbb5b4f
@@ -0,0 +1,122 @@
+using MAX.Core.DependencyInjection;
+using MAX.Core.Infrastructure;
+using Microsoft.Extensions.DependencyInjection;
+using System.Diagnostics;
+using System.Reflection;
+
+namespace StoreHub.Framework.DependencyInjection
+{
+	/// <summary>
+	/// IOC 注册入口（由宿主 Program.cs 调用）
+	/// </summary>
+	public static class ServiceCollectionExtensions
+	{
+		private static readonly Type[] DependencyMarkers =
+		[
+			typeof(ISingletonDependency),
+			typeof(IScopedDependency),
+			typeof(ITransientDependency),
+		];
+
+		/// <summary>
+		/// 注册应用服务：
+		/// <list type="number">
+		/// <item>自动注册所有实现 <see cref="ISingletonDependency"/> / <see cref="IScopedDependency"/> / <see cref="ITransientDependency"/> 的类型；</item>
+		/// <item>发现所有 <see cref="IStartup"/>，按 <see cref="IStartup.Priority"/> 降序执行 Initialize + Register。</item>
+		/// </list>
+		/// </summary>
+		public static IServiceCollection AddStoreHub(this IServiceCollection services)
+		{
+			IEnumerable<Assembly> assemblies = GetApplicationAssemblies();
+
+			RegisterDependencyMarkers(services, assemblies);
+			RunStartups(services, assemblies);
+
+			return services;
+		}
+
+		/// <summary>
+		/// 自动注册标记接口（ISingletonDependency / IScopedDependency / ITransientDependency）的实现类型
+		/// </summary>
+		private static void RegisterDependencyMarkers(IServiceCollection services, IEnumerable<Assembly> assemblies)
+		{
+			foreach (Type type in GetConcreteTypes(assemblies))
+			{
+				Debug.WriteLine(type.Namespace);
+				Type? marker = DependencyMarkers.FirstOrDefault(m => m.IsAssignableFrom(type));
+				if (marker is null)
+				{
+					continue;
+				}
+
+				ServiceLifetime lifetime = marker == typeof(ISingletonDependency) ? ServiceLifetime.Singleton
+					: marker == typeof(IScopedDependency) ? ServiceLifetime.Scoped
+					: ServiceLifetime.Transient;
+
+				// 注册为自身
+				services.Add(new ServiceDescriptor(type, type, lifetime));
+
+				// 注册为实现的接口（排除标记接口与系统接口）
+				foreach (Type iface in type.GetInterfaces())
+				{
+					if (DependencyMarkers.Contains(iface) || IsSystemInterface(iface))
+					{
+						continue;
+					}
+
+					services.Add(new ServiceDescriptor(iface, type, lifetime));
+				}
+			}
+		}
+
+		/// <summary>
+		/// 发现所有 <see cref="IStartup"/> 实现，按 <see cref="IStartup.Priority"/> 降序执行
+		/// </summary>
+		private static void RunStartups(IServiceCollection services, IEnumerable<Assembly> assemblies)
+		{
+			IServiceRegistrar registrar = new ServiceRegistrarProxy(services);
+
+			foreach (IStartup startup in GetConcreteTypes(assemblies)
+				.Where(t => typeof(IStartup).IsAssignableFrom(t))
+				.Select(t => (IStartup)Activator.CreateInstance(t, true)!)
+				.OrderByDescending(s => s.Priority))
+			{
+				startup.Initialize();
+				startup.Register(registrar);
+			}
+		}
+
+		private static IEnumerable<Type> GetConcreteTypes(IEnumerable<Assembly> assemblies)
+		{
+			return assemblies
+				.SelectMany(SafeGetTypes)
+				.Where(t => t.IsClass && !t.IsAbstract && !t.IsGenericTypeDefinition);
+		}
+
+		private static IEnumerable<Type> SafeGetTypes(Assembly assembly)
+		{
+			try
+			{
+				return assembly.GetTypes();
+			}
+			catch (ReflectionTypeLoadException ex)
+			{
+				return ex.Types.OfType<Type>();
+			}
+		}
+
+		private static IEnumerable<Assembly> GetApplicationAssemblies()
+		{
+			// 使用 AppTypeFinder：扫描应用输出目录 + 已加载程序集，
+			return StoreHubAppTypeFinder.Current.GetAssemblies();
+		}
+
+		private static bool IsSystemInterface(Type type)
+		{
+			return type.Namespace is null
+				|| type.Namespace.StartsWith("System", StringComparison.Ordinal)
+				|| type.Namespace.StartsWith("Microsoft", StringComparison.Ordinal);
+		}
+
+	}
+}