using AsbCloudApp.Data; using AsbCloudApp.Services; using Microsoft.AspNetCore.Authorization; using Microsoft.AspNetCore.Http; using Microsoft.AspNetCore.Mvc; using System; using System.IO; using System.Collections.Generic; using System.Linq; using System.Threading; using System.Threading.Tasks; namespace AsbCloudWebApi.Controllers { [Route("api/well/{idWell}/files")] [ApiController] [Authorize] public class FileController : ControllerBase { private readonly IFileService fileService; private readonly IWellService wellService; public FileController(IFileService fileService, IWellService wellService) { this.fileService = fileService; this.wellService = wellService; } /// /// Сохраняет переданные файлы и информацию о них /// /// id скважины /// id категории файла /// Коллекция файлов /// Токен отмены задачи /// [HttpPost] [ProducesResponseType(typeof(int), (int)System.Net.HttpStatusCode.OK)] public async Task SaveFilesAsync(int idWell, int idCategory, [FromForm] IFormFileCollection files, CancellationToken token = default) { 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(); var fileInfoCollection = files.Select(f => new FileInfoDto { Name = f.FileName, IdCategory = idCategory, UploadDate = DateTime.Now }); var fileNamesAndIds = fileService.SaveFileInfos(idWell, (int)idUser, fileInfoCollection); foreach (var file in files) { var fileExtension = Path.GetExtension(file.FileName); var fileId = fileNamesAndIds[file.FileName]; var fileStream = file.OpenReadStream(); await fileService.SaveFile(idWell, idCategory, fileId, fileExtension, fileStream); } return Ok(); } /// /// Возвращает информацию о файлах для скважины в выбраной категории /// /// id скважины /// id категории файла /// id компаний для фильтрации возвращаемых файлов /// дата начала /// дата окончания /// для пагинации кол-во записей пропустить /// для пагинации кол-во записей взять /// Токен отмены задачи /// Список информации о файлах в этой категории [HttpGet] [ProducesResponseType(typeof(PaginationContainer), (int)System.Net.HttpStatusCode.OK)] public async Task GetFilesInfoAsync([FromRoute] int idWell, int skip = 0, int take = 32, int idCategory = default, IEnumerable companies = default, DateTime begin = default, DateTime end = default, CancellationToken token = default) { int? idCompany = User.GetCompanyId(); if (idCompany is null || !await wellService.IsCompanyInvolvedInWellAsync((int)idCompany, idWell, token).ConfigureAwait(false)) return Forbid(); var filesInfo = await fileService.GetFilesInfoAsync(idWell, idCategory, companies, begin, end, skip, take, token).ConfigureAwait(false); if (filesInfo is null || !filesInfo.Items.Any()) return NoContent(); return Ok(filesInfo); } /// /// Возвращает файл с диска на сервере /// /// id скважины /// id запрашиваемого файла /// Токен отмены задачи /// Запрашиваемый файл [HttpGet] [Route("{fileId}")] [ProducesResponseType(typeof(PhysicalFileResult), (int)System.Net.HttpStatusCode.OK)] public async Task GetFileAsync([FromRoute] int idWell, int fileId, CancellationToken token = default) { try { int? idCompany = User.GetCompanyId(); if (idCompany is null) return Forbid(); if (!await wellService.IsCompanyInvolvedInWellAsync((int)idCompany, idWell, token).ConfigureAwait(false)) return Forbid(); var fileInfo = await fileService.GetFileInfoAsync(fileId, token); if (fileInfo is null) throw new FileNotFoundException(); // TODO: словарь content typoв var relativePath = Path.Combine(fileService.RootPath, $"{idWell}", $"{fileInfo.IdCategory}", $"{fileInfo.Id}" + Path.GetExtension($"{fileInfo.Name}")); return PhysicalFile(Path.GetFullPath(relativePath), "application/octet-stream", fileInfo.Name); } catch (FileNotFoundException ex) { return NotFound($"Файл не найден. Текст ошибки: {ex.Message}"); } } /// /// Удаляет файл с диска на сервере /// /// id скважины /// id запрашиваемого файла /// Токен отмены задачи /// [HttpDelete("{idFile}")] [ProducesResponseType(typeof(int), (int)System.Net.HttpStatusCode.OK)] public async Task DeleteAsync(int idWell, int idFile, CancellationToken token = default) { int? idCompany = User.GetCompanyId(); if (idCompany is null || !await wellService.IsCompanyInvolvedInWellAsync((int)idCompany, idWell, token).ConfigureAwait(false)) return Forbid(); var result = await fileService.DeleteFileAsync(idFile, token); return Ok(result); } } }