DMContainer.cs

Go to the documentation of this file.

using System;
using System.Collections.Generic;
using System.Linq;
using System.Reflection;
 
namespace Dreamine.MVVM.Core
{
    public static partial class DMContainer
    {
        private static readonly Dictionary<Type, Func<object>> _map = new();
        private static readonly Dictionary<Type, object> _singletonCache = new();

        public static void Register<T>(Func<T> factory) where T : class
            => _map[typeof(T)] = () => factory();

        public static void RegisterSingleton<T>(T instance) where T : class
            => _map[typeof(T)] = () => instance;

        public static T Resolve<T>() where T : class
        {
            if (_map.TryGetValue(typeof(T), out var factory))
                return (T)factory();
            throw new InvalidOperationException($"[{typeof(T).Name}] 등록되지 않음.");
        }

        public static object Resolve(Type type)
        {
            if (_map.TryGetValue(type, out var factory))
                return factory();
 
            // Fallback: 자동 생성 시도 (등록 안됐지만 생성 가능한 경우)
            if (!type.IsAbstract && type.IsClass)
            {
                var ctor = type.GetConstructors()
                    .OrderByDescending(c => c.GetParameters().Length)
                    .FirstOrDefault();
 
                if (ctor != null)
                {
                    var args = ctor.GetParameters()
                        .Select(p => Resolve(p.ParameterType))
                        .ToArray();
                    var instance = Activator.CreateInstance(type, args)!;
 
                    // 선택: 한번 생성되면 캐시할 수도 있음
                    _map[type] = () => instance;
 
                    return instance;
                }
            }
 
            throw new InvalidOperationException($"[{type.FullName}] 등록되지 않았고, 생성도 불가능합니다.");
        }
 

        public static void AutoRegisterAll(Assembly rootAssembly)
        {
            var assemblies = AppDomain.CurrentDomain.GetAssemblies()
                .Where(a => !a.IsDynamic && !string.IsNullOrEmpty(a.FullName))
                .Prepend(rootAssembly) // 우선 대상 Assembly 먼저
                .Distinct();
 
            foreach (var assembly in assemblies)
            {
                foreach (var type in SafeGetTypes(assembly))
                {
                    if (!type.IsClass || type.IsAbstract || type.IsGenericType)
                        continue;
 
                    var name = type.Name;
 
                    // DreamineApp + SampleTest 구조 모두 대응
                    bool isTarget =
                        name.EndsWith("Model") ||
                        name.EndsWith("Event") ||
                        name.EndsWith("Manager") ||
                        name.EndsWith("ViewModel") ||
                        name.Contains(".xaml.ViewModel") ||
                        name.Contains(".xaml.Model") ||
                        name.Contains(".xaml.Event") ;
 
                    if (!isTarget)
                        continue;
 
                    if (_map.ContainsKey(type))
                        continue;
 
                    // 생성자 우선 등록
                    var ctor = type.GetConstructors()
                        .OrderByDescending(c => c.GetParameters().Length)
                        .FirstOrDefault();
 
                    if (ctor == null)
                    {
                        // 매개변수 없는 생성자라도 등록 시도
                        if (type.GetConstructor(Type.EmptyTypes) != null)
                        {
                            _map[type] = () => Activator.CreateInstance(type)!;
                        }
                        continue;
                    }
 
                    _map[type] = () =>
                    {
                        if (_singletonCache.TryGetValue(type, out var cached))
                            return cached;
 
                        var args = ctor.GetParameters()
                            .Select(p => Resolve(p.ParameterType))
                            .ToArray();
                        var instance = Activator.CreateInstance(type, args)!;
 
                        _singletonCache[type] = instance; // ✅ 진짜 Type 키로 캐싱
                        return instance;
                    };
                }
            }
        }

        private static IEnumerable<Type> SafeGetTypes(Assembly assembly)
        {
            try
            {
                return assembly.GetTypes();
            }
            catch (ReflectionTypeLoadException ex)
            {
                return ex.Types.Where(t => t != null)!;
            }
        }
    }
}