ViewModelLocator.cs

Go to the documentation of this file.

using Dreamine.MVVM.Interfaces.Locators;
using System;
using System.Diagnostics;
using System.Reflection;
using System.Windows;
 
namespace Dreamine.MVVM.Locators
{
    public static class ViewModelLocator
    {
        private static IViewModelResolver? _resolver;
        private static readonly Dictionary<Type, Type> _map = new();

        public static void Register(Type viewType, Type viewModelType)
        {
            if (!_map.ContainsKey(viewType))
                _map[viewType] = viewModelType;
        }

        public static void RegisterResolver(IViewModelResolver resolver)
        {
            _resolver = resolver;
        }

        public static object? Resolve(Type viewType)
        {
            if (_map.TryGetValue(viewType, out var mappedType))
            {
                return _resolver?.Resolve(mappedType) ?? Activator.CreateInstance(mappedType);
            }
 
            var viewName = viewType.FullName;
            if (viewName == null)
                return null;
 
            var asm = viewType.Assembly.FullName;
 
            // 🧠 View → ViewModel 매핑 규칙: .Views → .ViewModels, + "ViewModel" suffix
            if (!viewName.Contains(".Views."))
                return null;
 
            // 예: DreamineApp.Views.Login.MainWindow → DreamineApp.ViewModels.Login.MainWindowViewModel
            var viewModelName = viewName.Replace(".Views.", ".ViewModels.") + "ViewModel";
 
            var vmType = Type.GetType($"{viewModelName}, {asm}");
            return vmType != null
                ? _resolver?.Resolve(vmType) ?? Activator.CreateInstance(vmType)
                : null;
        }

        public static object? ResolveView(Type viewModelType)
        {
            if (viewModelType == null) return null;
 
            // ViewModel 네임스페이스를 기준으로 View 네임스페이스 유추
            string? viewTypeName = viewModelType.FullName?
                .Replace(".ViewModels.", ".Views.")
                .Replace("ViewModel", "View");
 
            if (viewTypeName == null) return null;
 
            // ViewModel과 동일 Assembly 기준으로 View Type 찾기
            var viewType = viewModelType.Assembly.GetType(viewTypeName);
 
            // 그래도 못 찾으면 전체 어셈블리 순회 (Fallback)
            if (viewType == null)
            {
                viewType = AppDomain.CurrentDomain.GetAssemblies()
                    .Where(a => !a.IsDynamic)
                    .Select(a => a.GetType(viewTypeName ?? ""))
                    .FirstOrDefault(t => t != null);
            }
 
            return viewType != null
                ? Activator.CreateInstance(viewType)
                : null;
        }

        public static void RegisterAll(Assembly assembly)
        {
            var viewTypes = assembly.GetTypes()
                .Where(t => t.IsClass && !t.IsAbstract &&
                    (t.Namespace?.Contains(".Views") == true || t.Namespace?.Contains(".Pages") == true));
 
            var allTypes = assembly.GetTypes()
                .Where(t => t.IsClass && !t.IsAbstract);
 
            foreach (var viewType in viewTypes)
            {
                var viewName = viewType.Name;
                var viewNs = viewType.Namespace ?? "";
 
                var candidateNames = new[]
                {
                    viewName + "ViewModel",
                    viewName + ".ViewModel",
                    viewName + "_ViewModel",
                };
 
                var candidateNamespaces = new[]
                {
                    viewNs.Replace(".Views", ".ViewModels"),
                    viewNs.Replace(".Pages", ".ViewModels"),
                    viewNs,
                };
 
                foreach (var ns in candidateNamespaces)
                {
                    foreach (var name in candidateNames)
                    {
                        var fullName = $"{ns}.{name}";
                        var vmType = allTypes.FirstOrDefault(t => t.FullName == fullName);
 
                        if (vmType != null)
                        {
                            ViewModelLocator.Register(viewType, vmType);                            
                            goto NEXT_VIEW;
                        }
                    }
                }
 
            NEXT_VIEW:
                continue;
            }
        }
    }
}