// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.

using System;
using System.Threading;
using System.Threading.Tasks;

namespace Microsoft.Arcade.Common;

public interface IRetryHandler
{
    Task<bool> RunAsync(
        Func<int, Task<RetryResult>> actionAsync);

    Task<bool> RunAsync(
        Func<int, Task<RetryResult>> actionAsync,
        CancellationToken cancellationToken);
}

public readonly record struct RetryResult(bool Succeeded, TimeSpan? RetryAfter = null)
{
    public static RetryResult Success => new(true);

    public static RetryResult Retry(TimeSpan? retryAfter = null)
        => new(false, retryAfter);

    public static implicit operator RetryResult(bool succeeded)
        => succeeded ? Success : Retry();
}