// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
using System.Runtime.Versioning;
using System.Threading;
using System.Threading.Tasks;
 
namespace System.Net.Http
{
    public class HttpMessageInvoker : IDisposable
    {
        private volatile bool _disposed;
        private readonly bool _disposeHandler;
        private readonly HttpMessageHandler _handler;
 
        public HttpMessageInvoker(HttpMessageHandler handler)
            : this(handler, true)
        {
        }
 
        public HttpMessageInvoker(HttpMessageHandler handler, bool disposeHandler)
        {
            ArgumentNullException.ThrowIfNull(handler);
 
            if (NetEventSource.Log.IsEnabled()) NetEventSource.Associate(this, handler);
 
            _handler = handler;
            _disposeHandler = disposeHandler;
        }
 
        [UnsupportedOSPlatformAttribute("browser")]
        public virtual HttpResponseMessage Send(HttpRequestMessage request, CancellationToken cancellationToken)
        {
            ArgumentNullException.ThrowIfNull(request);
 
            ObjectDisposedException.ThrowIf(_disposed, this);
 
            if (ShouldSendWithTelemetry(request))
            {
                HttpTelemetry.Log.RequestStart(request);
 
                HttpResponseMessage? response = null;
                try
                {
                    response = _handler.Send(request, cancellationToken);
                    return response;
                }
                catch (Exception ex) when (LogRequestFailed(ex, telemetryStarted: true))
                {
                    // Unreachable as LogRequestFailed will return false
                    throw;
                }
                finally
                {
                    HttpTelemetry.Log.RequestStop(response);
                }
            }
            else
            {
                return _handler.Send(request, cancellationToken);
            }
        }
 
        public virtual Task<HttpResponseMessage> SendAsync(HttpRequestMessage request, CancellationToken cancellationToken)
        {
            ArgumentNullException.ThrowIfNull(request);
 
            ObjectDisposedException.ThrowIf(_disposed, this);
 
            if (ShouldSendWithTelemetry(request))
            {
                return SendAsyncWithTelemetry(_handler, request, cancellationToken);
            }
 
            return _handler.SendAsync(request, cancellationToken);
 
            static async Task<HttpResponseMessage> SendAsyncWithTelemetry(HttpMessageHandler handler, HttpRequestMessage request, CancellationToken cancellationToken)
            {
                HttpTelemetry.Log.RequestStart(request);
 
                HttpResponseMessage? response = null;
                try
                {
                    response = await handler.SendAsync(request, cancellationToken).ConfigureAwait(false);
                    return response;
                }
                catch (Exception ex) when (LogRequestFailed(ex, telemetryStarted: true))
                {
                    // Unreachable as LogRequestFailed will return false
                    throw;
                }
                finally
                {
                    HttpTelemetry.Log.RequestStop(response);
                }
            }
        }
 
        private static bool ShouldSendWithTelemetry(HttpRequestMessage request) =>
            HttpTelemetry.Log.IsEnabled() &&
            !request.WasSentByHttpClient() &&
            request.RequestUri is Uri requestUri &&
            requestUri.IsAbsoluteUri;
 
        internal static bool LogRequestFailed(Exception exception, bool telemetryStarted)
        {
            if (HttpTelemetry.Log.IsEnabled() && telemetryStarted)
            {
                HttpTelemetry.Log.RequestFailed(exception);
            }
            return false;
        }
 
        public void Dispose()
        {
            Dispose(true);
            GC.SuppressFinalize(this);
        }
 
        protected virtual void Dispose(bool disposing)
        {
            if (disposing && !_disposed)
            {
                _disposed = true;
                if (_disposeHandler)
                {
                    _handler.Dispose();
                }
            }
        }
    }
}