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

using System.Net;
using System.Net.Http;
using System.Threading;
using System.Threading.Tasks;

namespace Microsoft.Arcade.Common
{
    public class ArcadeHttpMessageHandler : HttpMessageHandler
    {
        public virtual RequestResponseHelper [] RequestResponses { get; set; }

        public Task<HttpResponseMessage> SendAsync(HttpRequestMessage request) => SendAsync(request, CancellationToken.None);

        protected override Task<HttpResponseMessage> SendAsync(HttpRequestMessage request, CancellationToken cancellationToken)
        {
            foreach(var requestResponse in RequestResponses)
            {
                if(request.RequestUri.ToString().StartsWith(requestResponse.RequestMessage.RequestUri.ToString()) &&
                    request.Method.Equals(requestResponse.RequestMessage.Method))
                {
                    return Task.FromResult(new HttpResponseMessage()
                    {
                        StatusCode = requestResponse.ResponseMessage.StatusCode,
                        Content = requestResponse.ResponseMessage.Content
                    });
                }
            }
            return Task.FromResult(new HttpResponseMessage()
            {
                StatusCode = HttpStatusCode.InternalServerError,
                Content = new StringContent($"No response specified in RequestResponses for '({request.Method}) {request.RequestUri}")
            });
        }
    }
}