FetchOptionsExtensions.cs

Go to the documentation of this file.

using System;
using System.Threading;
 
using LibGit2Sharp;
using LibGit2Sharp.Handlers;
 
using Microsoft.Extensions.Logging;
 
using Tgstation.Server.Host.Jobs;
 
namespace Tgstation.Server.Host.Extensions
{
    static class FetchOptionsExtensions
    {
        public static FetchOptions Hydrate(
            this FetchOptions fetchOptions,
            ILogger logger,
            JobProgressReporter progressReporter,
            CredentialsHandler credentialsHandler,
            CancellationToken cancellationToken)
        {
            ArgumentNullException.ThrowIfNull(fetchOptions);
            ArgumentNullException.ThrowIfNull(logger);
 
            fetchOptions.OnProgress = _ => !cancellationToken.IsCancellationRequested;
            fetchOptions.OnTransferProgress = transferProgress =>
            {
                if (progressReporter != null)
                {
                    var percentage = ((double)transferProgress.IndexedObjects + transferProgress.ReceivedObjects) / (transferProgress.TotalObjects * 2);
                    progressReporter.ReportProgress(percentage);
                }
 
                return !cancellationToken.IsCancellationRequested;
            };
            fetchOptions.OnUpdateTips = (_, _, _) => !cancellationToken.IsCancellationRequested;
            fetchOptions.CredentialsProvider = credentialsHandler;
            fetchOptions.RepositoryOperationStarting = _ => !cancellationToken.IsCancellationRequested;
            fetchOptions.OnTransferProgress = TransferProgressHandler(
                logger,
                progressReporter,
                cancellationToken);
 
            return fetchOptions;
        }
 
        static TransferProgressHandler TransferProgressHandler(ILogger logger, JobProgressReporter progressReporter, CancellationToken cancellationToken) => transferProgress =>
        {
            double? percentage;
            var totalObjectsToProcess = transferProgress.TotalObjects * 2;
            var processedObjects = transferProgress.IndexedObjects + transferProgress.ReceivedObjects;
            if (totalObjectsToProcess < processedObjects || totalObjectsToProcess == 0)
                percentage = null;
            else
            {
                percentage = (double)processedObjects / totalObjectsToProcess;
                if (percentage < 0)
                    percentage = null;
            }
 
            if (percentage == null)
                logger.LogDebug(
                    "Bad transfer progress values (Please tell Cyberboss)! Indexed: {indexed}, Received: {received}, Total: {total}",
                    transferProgress.IndexedObjects,
                    transferProgress.ReceivedObjects,
                    transferProgress.TotalObjects);
 
            progressReporter?.ReportProgress(percentage);
            return !cancellationToken.IsCancellationRequested;
        };
    }
}