// 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.Diagnostics.CodeAnalysis;
using System.Net.Http;
using Microsoft.Extensions.Http.Resilience.Internal;
using Microsoft.Shared.DiagnosticIds;
using Microsoft.Shared.Diagnostics;
 
namespace Polly;
 
/// <summary>
/// Provides utility methods for working with <see cref="ResilienceContext"/>.
/// </summary>
[Experimental(diagnosticId: DiagnosticIds.Experiments.Resilience, UrlFormat = DiagnosticIds.UrlFormat)]
public static class HttpResilienceContextExtensions
{
    /// <summary>
    /// Gets the request message from the <see cref="ResilienceContext"/>.
    /// </summary>
    /// <param name="context">The resilience context.</param>
    /// <returns>
    /// The request message.
    /// If the request message is not present in the <see cref="ResilienceContext"/> the method returns <see langword="null"/>.
    /// </returns>
    /// <exception cref="ArgumentNullException"><paramref name="context"/> is <see langword="null"/>.</exception>
    public static HttpRequestMessage? GetRequestMessage(this ResilienceContext context)
    {
        _ = Throw.IfNull(context);
        return context.Properties.GetValue(ResilienceKeys.RequestMessage, default);
    }
 
    /// <summary>
    /// Sets the request message on the <see cref="ResilienceContext"/>.
    /// </summary>
    /// <param name="context">The resilience context.</param>
    /// <param name="requestMessage">The request message.</param>
    /// <exception cref="ArgumentNullException"><paramref name="context"/> is <see langword="null"/>.</exception>
    public static void SetRequestMessage(this ResilienceContext context, HttpRequestMessage? requestMessage)
    {
        _ = Throw.IfNull(context);
        context.Properties.Set(ResilienceKeys.RequestMessage, requestMessage);
    }
}