SourceFactory.cs

using System.Collections.Concurrent;
using UniGetUI.PackageEngine.Interfaces;

namespace UniGetUI.PackageEngine.Classes.Manager
{
    public class SourceFactory : ISourceFactory
    {
        private readonly IPackageManager __manager;
        private readonly ConcurrentDictionary<string, IManagerSource> __reference;
        private readonly ConcurrentDictionary<string, IManagerSource> __placeholders = [];
        private readonly Uri __default_uri = new("https://devolutions.net/unigetui");

        public SourceFactory(IPackageManager manager)
        {
            __reference = [];
            __manager = manager;
        }

        public void Reset()
        {
            __reference.Clear();
        }

        /// <summary>
        /// Returns the existing source for the given name, or creates a new one if it does not exist.
        /// </summary>
        /// <param name="name">The name of the source</param>
        /// <returns>A valid ManagerSource</returns>
        public IManagerSource GetSourceOrDefault(string name)
        {
            if (__reference.TryGetValue(name, out IManagerSource? source))
            {
                return source;
            }

            IManagerSource new_source = __placeholders.GetOrAdd(
                name,
                n => new ManagerSource(__manager, n, __default_uri)
            );
            return __reference.GetOrAdd(name, new_source);
        }

        /// <summary>
        /// Returns the existing source for the given name, or null if it does not exist.
        /// </summary>
        public IManagerSource? GetSourceIfExists(string name)
        {
            if (__reference.TryGetValue(name, out IManagerSource? source))
            {
                return source;
            }

            return null;
        }

        public void AddSource(IManagerSource source)
        {
            if (!__reference.TryAdd(source.Name, source))
            {
                IManagerSource existing_source = __reference[source.Name];
                if (existing_source.Url == __default_uri)
                {
                    existing_source.ReplaceUrl(source.Url);
                }
            }

            // Placeholders handed out before a Reset() are still referenced by packages
            if (__placeholders.TryRemove(source.Name, out IManagerSource? placeholder))
            {
                if (placeholder.Url == __default_uri)
                {
                    placeholder.ReplaceUrl(source.Url);
                }
            }
        }

        public IManagerSource[] GetAvailableSources()
        {
            return __reference.Values.ToArray();
        }
    }
}