TransferController.cs

Go to the documentation of this file.

using System;
using System.ComponentModel.DataAnnotations;
using System.Threading;
using System.Threading.Tasks;
 
using Microsoft.AspNetCore.Mvc;
using Microsoft.Extensions.Logging;
 
using Tgstation.Server.Api;
using Tgstation.Server.Api.Models;
using Tgstation.Server.Api.Models.Response;
using Tgstation.Server.Host.Controllers.Results;
using Tgstation.Server.Host.Database;
using Tgstation.Server.Host.Extensions;
using Tgstation.Server.Host.Security;
using Tgstation.Server.Host.Transfer;
using Tgstation.Server.Host.Utils;
 
namespace Tgstation.Server.Host.Controllers
{
    [Route(Routes.Transfer)]
    [RequestSizeLimit(Limits.MaximumFileTransferSize)]
    public sealed class TransferController : ApiController
    {
        readonly IFileTransferStreamHandler fileTransferService;
 
        public TransferController(
            IDatabaseContext databaseContext,
            IAuthenticationContext authenticationContext,
            IFileTransferStreamHandler fileTransferService,
            ILogger<ApiController> logger,
            IApiHeadersProvider apiHeaders)
            : base(
                  databaseContext,
                  authenticationContext,
                  apiHeaders,
                  logger,
                  true)
        {
            this.fileTransferService = fileTransferService ?? throw new ArgumentNullException(nameof(fileTransferService));
        }
 
        [TgsAuthorize]
        [HttpGet]
        [ProducesResponseType(200, Type = typeof(LimitedStreamResult))]
        [ProducesResponseType(410, Type = typeof(ErrorMessageResponse))]
        public ValueTask<IActionResult> Download([Required, FromQuery] string ticket, CancellationToken cancellationToken)
            => fileTransferService.GenerateDownloadResponse(this, ticket, cancellationToken);
 
        [TgsAuthorize]
        [HttpPut]
        [DisableRequestSizeLimit]
        [ProducesResponseType(204)]
        [ProducesResponseType(410, Type = typeof(ErrorMessageResponse))]
        public async ValueTask<IActionResult> Upload([Required, FromQuery] string ticket, CancellationToken cancellationToken)
        {
            if (ticket == null)
                return BadRequest(new ErrorMessageResponse(ErrorCode.ModelValidationFailure));
 
            var fileTicketResult = new FileTicketResponse
            {
                FileTicket = ticket,
            };
 
            var result = await fileTransferService.SetUploadStream(fileTicketResult, Request.Body, cancellationToken);
            if (result != null)
                return result.ErrorCode == ErrorCode.ResourceNotPresent
                    ? this.Gone()
                    : Conflict(result);
 
            return NoContent();
        }
    }
}