LimitedStreamResult.cs

Go to the documentation of this file.

using System;
using System.IO;
using System.Net.Mime;
using System.Threading;
using System.Threading.Tasks;
 
using Microsoft.AspNetCore.Mvc;
using Microsoft.AspNetCore.Mvc.Infrastructure;
using Microsoft.Extensions.DependencyInjection;
 
using Tgstation.Server.Host.IO;
 
namespace Tgstation.Server.Host.Controllers.Results
{
    public sealed class LimitedStreamResult : FileResult, IFileStreamProvider
    {
        readonly Stream stream;
 
        public LimitedStreamResult(Stream stream)
            : base(MediaTypeNames.Application.Octet)
        {
            this.stream = stream ?? throw new ArgumentNullException(nameof(stream));
        }
 
        public ValueTask DisposeAsync() => stream.DisposeAsync();
 
        public override Task ExecuteResultAsync(ActionContext context)
        {
            ArgumentNullException.ThrowIfNull(context);
 
            var executor = context
                .HttpContext
                .RequestServices
                .GetRequiredService<IActionResultExecutor<LimitedStreamResult>>();
            return executor.ExecuteAsync(context, this);
        }
 
        public ValueTask<Stream> GetResult(CancellationToken cancellationToken) => ValueTask.FromResult(stream);
    }
}