// Copyright (c) .NET Foundation. All rights reserved.
// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.

using System.IO;
using System.Threading.Tasks;
using Microsoft.AspNetCore.Http;

namespace Microsoft.AspNetCore.Testing
{
    public static class TestApp
    {
        public static async Task EchoApp(HttpContext httpContext)
        {
            var request = httpContext.Request;
            var response = httpContext.Response;
            var buffer = new byte[httpContext.Request.ContentLength ?? 0];
            var bytesRead = 0;

            while (bytesRead < buffer.Length)
            {
                var count = await request.Body.ReadAsync(buffer, bytesRead, buffer.Length - bytesRead);
                bytesRead += count;
            }

            if (buffer.Length > 0)
            {
                await response.Body.WriteAsync(buffer, 0, buffer.Length);
            }
        }

        public static async Task EchoAppChunked(HttpContext httpContext)
        {
            var request = httpContext.Request;
            var response = httpContext.Response;
            var data = new MemoryStream();
            await request.Body.CopyToAsync(data);
            var bytes = data.ToArray();

            response.Headers["Content-Length"] = bytes.Length.ToString();
            await response.Body.WriteAsync(bytes, 0, bytes.Length);
        }

        public static Task EmptyApp(HttpContext httpContext)
        {
            return Task.CompletedTask;
        }
    }
}