using AsbCloudApp.Data; using AsbCloudApp.Requests; using AsbCloudApp.Services; using Microsoft.AspNetCore.Authorization; using Microsoft.AspNetCore.Http; using Microsoft.AspNetCore.Mvc; using System.IO; using System.Threading; using System.Threading.Tasks; using AsbCloudApp.Repositories; namespace AsbCloudWebApi.Controllers { /// /// Хранение файлов /// [Route("api/well/{idWell}/files")] [ApiController] [Authorize] public class FileController : ControllerBase { private readonly FileService fileService; private readonly IWellService wellService; public FileController(FileService fileService, IWellService wellService) { this.fileService = fileService; this.wellService = wellService; } /// /// Сохраняет переданные файлы и информацию о них /// /// id скважины /// id категории файла /// Коллекция файлов /// dependency /// Токен отмены задачи /// [HttpPost] [Permission] [ProducesResponseType(typeof(int), (int)System.Net.HttpStatusCode.OK)] public async Task SaveFilesAsync(int idWell, int idCategory, [FromForm] IFormFileCollection files, [FromServices] IUserRepository userRepository, CancellationToken token) { int? idCompany = User.GetCompanyId(); int? idUser = User.GetUserId(); if (idCompany is null || idUser is null) return Forbid(); if (!await wellService.IsCompanyInvolvedInWellAsync((int)idCompany, idWell, token).ConfigureAwait(false)) return Forbid(); if (!userRepository.HasPermission((int)idUser, $"File.edit{idCategory}")) return Forbid(); foreach (var file in files) { var fileStream = file.OpenReadStream(); await fileService.SaveAsync(idWell, idUser ?? 0, idCategory, file.FileName, fileStream, token).ConfigureAwait(false); } return Ok(); } /// /// Возвращает информацию о файлах для скважины в выбраной категории /// /// /// Токен отмены задачи /// Список информации о файлах в этой категории [HttpGet] [Route("/api/files")] [Permission] [ProducesResponseType(typeof(PaginationContainer), (int)System.Net.HttpStatusCode.OK)] public async Task GetFilesInfoAsync( [FromQuery] FileRequest request, CancellationToken token) { int? idCompany = User.GetCompanyId(); if (request.IdCategory is null || idCompany is null) return Forbid(); if (!await wellService.IsCompanyInvolvedInWellAsync(idCompany.Value, request.IdWell, token).ConfigureAwait(false)) return Forbid(); var filesInfo = await fileService.GetInfosPaginatedAsync(request, token).ConfigureAwait(false); return Ok(filesInfo); } /// /// Возвращает файл с диска на сервере /// /// id запрашиваемого файла /// Токен отмены задачи /// Запрашиваемый файл [HttpGet] [Route("{idFile}")] [Permission] [ProducesResponseType(typeof(PhysicalFileResult), (int)System.Net.HttpStatusCode.OK)] public async Task GetFileAsync( int idFile, CancellationToken token) { int? idCompany = User.GetCompanyId(); if (idCompany is null) return Forbid(); var fileInfo = await fileService.GetOrDefaultAsync(idFile, token); if (fileInfo is null) return NotFound(idFile); if (!await wellService.IsCompanyInvolvedInWellAsync((int)idCompany, fileInfo.IdWell, token).ConfigureAwait(false)) return Forbid(); var fileStream = fileService.GetFileStream(fileInfo); return File(fileStream, "application/octet-stream", fileInfo.Name); } /// /// Помечает файл как удаленный /// /// id скважины /// id запрашиваемого файла /// dependency /// Токен отмены задачи /// [HttpDelete("{idFile}")] [Permission] [ProducesResponseType(typeof(int), (int)System.Net.HttpStatusCode.OK)] public async Task DeleteAsync(int idWell, int idFile, [FromServices] IUserRepository userRepository, CancellationToken token) { int? idUser = User.GetUserId(); int? idCompany = User.GetCompanyId(); if (idUser is null || idCompany is null || !await wellService.IsCompanyInvolvedInWellAsync((int)idCompany, idWell, token).ConfigureAwait(false)) return Forbid(); var fileInfo = await fileService.GetOrDefaultAsync(idFile, token); if (fileInfo is null) return NotFound(idFile); if (!userRepository.HasPermission((int)idUser, $"File.edit{fileInfo?.IdCategory}")) return Forbid(); var result = await fileService.MarkAsDeletedAsync(idFile, token); return Ok(result); } /// /// Создает метку для файла /// /// id скважины /// метка файла /// Токен отмены задачи /// [HttpPost("fileMark")] [Permission] public async Task CreateFileMarkAsync(int idWell, FileMarkDto markDto, CancellationToken token) { var idCompany = User.GetCompanyId(); var idUser = User.GetUserId(); if (idCompany is null || idUser is null || !await wellService.IsCompanyInvolvedInWellAsync((int)idCompany, idWell, token).ConfigureAwait(false)) return Forbid(); var result = await fileService.CreateFileMarkAsync(markDto, (int)idUser, token) .ConfigureAwait(false); return Ok(result); } /// /// Помечает метку у файла как удаленную /// /// id скважины /// id метки /// Токен отмены задачи /// [HttpDelete("fileMark")] [Permission] [ProducesResponseType(typeof(int), (int)System.Net.HttpStatusCode.OK)] public async Task DeleteFileMarkAsync(int idWell, int idMark, CancellationToken token) { var idCompany = User.GetCompanyId(); if (idCompany is null || !await wellService.IsCompanyInvolvedInWellAsync((int)idCompany, idWell, token).ConfigureAwait(false)) return Forbid(); var result = await fileService.MarkFileMarkAsDeletedAsync(idMark, token) .ConfigureAwait(false); return Ok(result); } /// /// Возвращает информацию о файле /// /// id запрашиваемого файла /// Токен отмены задачи /// Запрашиваемый файл [HttpGet] [Route("/api/files/{idFile}")] [Permission] [ProducesResponseType(typeof(FileInfoDto), (int)System.Net.HttpStatusCode.OK)] public async Task GetFileInfoAsync([FromRoute] int idFile, CancellationToken token) { int? idCompany = User.GetCompanyId(); if (idCompany is null) return Forbid(); try { var fileInfo = await fileService.GetOrDefaultAsync(idFile, token).ConfigureAwait(false); return Ok(fileInfo); } catch (FileNotFoundException ex) { return NotFound(ex.FileName); } } } }