Remotely/Server/API/FileSharingController.cs

using Microsoft.AspNetCore.Mvc;
using Remotely.Server.Auth;
using Remotely.Server.Services;
using Remotely.Shared;
using System;
using System.Collections.Generic;
using System.Threading.Tasks;

namespace Remotely.Server.API;

[Route("api/[controller]")]
[ApiController]
public class FileSharingController : ControllerBase
{
    private readonly IDataService _dataService;

    public FileSharingController(IDataService dataService)
    {
        _dataService = dataService;
    }

    [HttpGet("{id}")]
    [ServiceFilter(typeof(ExpiringTokenFilter))]
    public ActionResult Get(string id)
    {
        var sharedFile = _dataService.GetSharedFiled(id);
        if (sharedFile != null)
        {
            return File(sharedFile.FileContents, sharedFile.ContentType, sharedFile.FileName);
        }
        return NotFound();
    }

    [HttpPost]
    [ServiceFilter(typeof(ExpiringTokenFilter))]
    [RequestSizeLimit(AppConstants.MaxUploadFileSize)]
    public async Task<IEnumerable<string>> Post()
    {
        if (Request?.Form?.Files?.Count !> 0)
        {
            return Array.Empty<string>();
        }

        var fileIDs = new List<string>();
        foreach (var file in Request.Form.Files)
        {
            var orgID = User.Identity.IsAuthenticated ? _dataService.GetUserByNameWithOrg(User.Identity.Name).OrganizationID : null;
            var id = await _dataService.AddSharedFile(file, orgID);
            fileIDs.Add(id);
        }
        return fileIDs;
    }
}