Toolkit2/Toolkit.Foundation/Templates/TemplateFactory.cs

using System.Diagnostics.CodeAnalysis;

namespace Toolkit.Foundation
{
    public class TemplateFactory : ITemplateFactory
    {
        private readonly Dictionary<object, object> cache = new();

        private readonly ITemplateDescriptorProvider provider;
        private readonly IServiceFactory serviceFactory;

        public TemplateFactory(ITemplateDescriptorProvider provider,
            IServiceFactory serviceFactory)
        {
            this.provider = provider;
            this.serviceFactory = serviceFactory;
        }

        public virtual object? Create([MaybeNull] object? data)
        {
            if (data is null)
            {
                return null;
            }

            if (cache.TryGetValue(data, out object? template))
            {
                return template;
            }

            if (provider.Get(data.GetType()) is ITemplateDescriptor descriptor)
            {
                template = serviceFactory.Get<object>(descriptor.TemplateType);
                if (template is ICache cache)
                {
                    this.cache[data] = cache;
                }
            }

            return template;
        }
    }
}