// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
using System.IO.Pipelines;
 
#nullable enable
 
namespace Microsoft.AspNetCore.Connections;
 
internal sealed class CompletionPipeWriter : PipeWriter
{
    private readonly PipeWriter _inner;
 
    public bool IsCompleted { get; private set; }
    public Exception? CompleteException { get; private set; }
    public bool IsCompletedSuccessfully => IsCompleted && CompleteException == null;
 
    public CompletionPipeWriter(PipeWriter inner)
    {
        _inner = inner;
    }
 
    public override void Advance(int bytes)
    {
        _inner.Advance(bytes);
    }
 
    public override void CancelPendingFlush()
    {
        _inner.CancelPendingFlush();
    }
 
    public override void Complete(Exception? exception = null)
    {
        IsCompleted = true;
        CompleteException = exception;
        _inner.Complete(exception);
    }
 
    public override ValueTask CompleteAsync(Exception? exception = null)
    {
        IsCompleted = true;
        CompleteException = exception;
        return _inner.CompleteAsync(exception);
    }
 
    public override ValueTask<FlushResult> WriteAsync(ReadOnlyMemory<byte> source, CancellationToken cancellationToken = default)
    {
        return _inner.WriteAsync(source, cancellationToken);
    }
 
    public override ValueTask<FlushResult> FlushAsync(CancellationToken cancellationToken = default)
    {
        return _inner.FlushAsync(cancellationToken);
    }
 
    public override Memory<byte> GetMemory(int sizeHint = 0)
    {
        return _inner.GetMemory(sizeHint);
    }
 
    public override Span<byte> GetSpan(int sizeHint = 0)
    {
        return _inner.GetSpan(sizeHint);
    }
 
    public override bool CanGetUnflushedBytes => _inner.CanGetUnflushedBytes;
 
    public override long UnflushedBytes => _inner.UnflushedBytes;
 
    public void Reset()
    {
        IsCompleted = false;
        CompleteException = null;
    }
}