RegistryCacheReport.cs

using System;
using System.Collections.Generic;
using Microsoft.Extensions.Options;

namespace UnityNuGet.Server
{
    /// <summary>
    /// Main class that stores relevant messages that may appear during the build of the Unity packages with <see cref="RegistryCache"/>.
    /// </summary>
    public class RegistryCacheReport(RegistryCacheSingleton registryCacheSingleton, IOptions<RegistryOptions> registryOptionsAccessor)
    {
        private readonly RegistryCacheSingleton _registryCacheSingleton = registryCacheSingleton;
        private readonly RegistryOptions _registryOptions = registryOptionsAccessor.Value;

        private readonly List<string> _informationMessages = [];
        private readonly List<string> _warningMessages = [];
        private readonly List<string> _errorMessages = [];

        private DateTime? _lastUpdate;

        public IEnumerable<string> InformationMeessages => _informationMessages;

        public IEnumerable<string> WarningMessages => _warningMessages;

        public IEnumerable<string> ErrorMessages => _errorMessages;

        public bool Running { get; private set; }

        public double Progress
        {
            get
            {
                int currentIndex = _registryCacheSingleton.ProgressPackageIndex;
                int totalCount = _registryCacheSingleton.ProgressTotalPackageCount;
                double percent = totalCount != 0 ? (double)currentIndex * 100 / totalCount : 0;

                return percent;
            }
        }

        public TimeSpan? TimeRemainingForNextUpdate
        {
            get
            {
                if (_errorMessages.Count == 0)
                {
                    return _lastUpdate != null ? _lastUpdate.Value.Add(_registryOptions.UpdateInterval) - DateTime.UtcNow : null;
                }
                else
                {
                    return TimeSpan.FromSeconds(0);
                }
            }
        }

        public void AddInformation(string message) => _informationMessages.Add(message);

        public void AddWarning(string message) => _warningMessages.Add(message);

        public void AddError(string message) => _errorMessages.Add(message);

        public void Start()
        {
            Running = true;

            _informationMessages.Clear();
            _warningMessages.Clear();
            _errorMessages.Clear();
        }

        public void Complete()
        {
            Running = false;

            _lastUpdate = DateTime.UtcNow;
        }
    }
}