using AsbCloudApp.Data; using AsbCloudApp.Services; using Microsoft.AspNetCore.Authorization; using Microsoft.AspNetCore.Http; using Microsoft.AspNetCore.Mvc; using System; using System.Collections.Generic; using System.IO; using System.Threading; using System.Threading.Tasks; namespace AsbCloudWebApi.Controllers { /// /// Контроллер вручную внесенных операций на скважине /// [Route("api/well/{idWell}/wellOperations")] [ApiController] [Authorize] public class WellOperationController : ControllerBase { private readonly IWellOperationService operationService; private readonly IWellService wellService; private readonly IWellOperationImportService wellOperationImportService; public WellOperationController(IWellOperationService operationService, IWellService wellService, IWellOperationImportService wellOperationImportService) { this.operationService = operationService; this.wellService = wellService; this.wellOperationImportService = wellOperationImportService; } /// /// Возвращает список имен типов операций на скважине /// /// [HttpGet] [Route("categories")] [ProducesResponseType(typeof(IEnumerable), (int)System.Net.HttpStatusCode.OK)] public IActionResult GetCategories() { var result = operationService.GetCategories(); return Ok(result); } /// /// Отфильтрованный список операций на скважине. Если не применять фильтр, то вернется весь список. Сортированный по глубине затем по дате /// /// id скважины /// фильтр по план = 0, факт = 1 /// фильтр по списку id конструкций секции /// фильтр по списку id категорий операции /// фильтр по началу операции /// фильтр по окончанию операции /// фильтр по минимальной глубине скважины /// фильтр по максимальной глубине скважины /// /// /// /// Список операций на скважине в контейнере для постраничного просмотра [HttpGet] [ProducesResponseType(typeof(PaginationContainer), (int)System.Net.HttpStatusCode.OK)] public async Task GetOperationsAsync( [FromRoute] int idWell, [FromQuery] int? opertaionType = default, [FromQuery] IEnumerable sectionTypeIds = default, [FromQuery] IEnumerable operationCategoryIds = default, [FromQuery] DateTime begin = default, [FromQuery] DateTime end = default, [FromQuery] double minDepth = double.MinValue, [FromQuery] double maxDepth = double.MaxValue, [FromQuery] int skip = 0, [FromQuery] int take = 32, CancellationToken token = default) { if (!await CanUserAccessToWellAsync(idWell, token).ConfigureAwait(false)) return Forbid(); var result = await operationService.GetOperationsAsync( idWell, opertaionType, sectionTypeIds, operationCategoryIds, begin, end, minDepth, maxDepth, skip, take, token) .ConfigureAwait(false); return Ok(result); } /// /// Возвращает нужную операцию на скважине /// /// id скважины /// id нужной операции /// Токен отмены задачи /// Нужную операцию на скважине [HttpGet] [Route("{idOperation}")] [ProducesResponseType(typeof(WellOperationDto), (int)System.Net.HttpStatusCode.OK)] public async Task GetAsync(int idWell, int idOperation, CancellationToken token = default) { if (!await CanUserAccessToWellAsync(idWell, token).ConfigureAwait(false)) return Forbid(); var result = await operationService.GetAsync(idOperation, token).ConfigureAwait(false); return Ok(result); } /// /// Добавляет новые операции на скважине /// /// id скважины /// Данные о добавляемых операциях /// Токен отмены задачи /// Количество добавленых в БД строк [HttpPost] [ProducesResponseType(typeof(IEnumerable), (int)System.Net.HttpStatusCode.OK)] public async Task InsertRangeAsync(int idWell, [FromBody] IEnumerable values, CancellationToken token = default) { if (!await CanUserAccessToWellAsync(idWell, token).ConfigureAwait(false)) return Forbid(); var result = await operationService.InsertRangeAsync(idWell, values, token) .ConfigureAwait(false); return Ok(result); } /// /// Обновляет выбранную операцию на скважине /// /// id скважины /// id выбраной операции /// Новые данные для выбраной операции /// Токен отмены задачи /// Количество обновленных в БД строк [HttpPut("{idOperation}")] [ProducesResponseType(typeof(WellOperationDto), (int)System.Net.HttpStatusCode.OK)] public async Task UpdateAsync(int idWell, int idOperation, [FromBody] WellOperationDto value, CancellationToken token = default) { if (!await CanUserAccessToWellAsync(idWell, token).ConfigureAwait(false)) return Forbid(); var result = await operationService.UpdateAsync(idWell, idOperation, value, token) .ConfigureAwait(false); return Ok(result); } /// /// Удаляет выбраную операцию на скважине /// /// id скважины /// id выбраной операции /// Токен отмены задачи /// Количество удаленных из БД строк [HttpDelete("{idOperation}")] [ProducesResponseType(typeof(int), (int)System.Net.HttpStatusCode.OK)] public async Task DeleteAsync(int idWell, int idOperation, CancellationToken token = default) { if (!await CanUserAccessToWellAsync(idWell, token).ConfigureAwait(false)) return Forbid(); var result = await operationService.DeleteAsync(new int[] { idOperation }, token) .ConfigureAwait(false); return Ok(result); } /// /// Импортирует операции из excel (xlsx) файла /// /// id скважины /// Коллекция из одного файла xlsx /// Удалить операции перед импортом = 1, если фал валидный /// Токен отмены задачи /// [HttpPost] [Route("import/{options}")] public async Task ImportAsync(int idWell, [FromForm] IFormFileCollection files, int options = 0, 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(); if (files.Count < 1) return BadRequest("нет файла"); var file = files[0]; if(Path.GetExtension( file.FileName).ToLower() != ".xlsx") return BadRequest("Требуется xlsx файл."); using Stream stream = file.OpenReadStream(); try { wellOperationImportService.Import(idWell, stream, (options & 1) > 0); } catch(FileFormatException ex) { return BadRequest(ex.Message); } return Ok(); } /// /// Создает excel файл с операциями по скважине /// /// id скважины /// Токен отмены задачи /// Запрашиваемый файл [HttpGet] [Route("export")] [ProducesResponseType(typeof(PhysicalFileResult), (int)System.Net.HttpStatusCode.OK)] public async Task ExportAsync([FromRoute] int idWell, CancellationToken token = default) { int? idCompany = User.GetCompanyId(); if (idCompany is null) return Forbid(); if (!await wellService.IsCompanyInvolvedInWellAsync((int)idCompany, idWell, token).ConfigureAwait(false)) return Forbid(); var stream = wellOperationImportService.Export(idWell); var fileName = await wellService.GetWellCaptionByIdAsync(idWell, token) + "_operations.xlsx"; return File(stream, "application/octet-stream", fileName); } /// /// Возвращает шаблон файла импорта /// /// Запрашиваемый файл [HttpGet] [Route("tamplate")] [AllowAnonymous] [ProducesResponseType(typeof(PhysicalFileResult), (int)System.Net.HttpStatusCode.OK)] public IActionResult GetTamplate() { var stream = wellOperationImportService.GetExcelTemplateStream(); var fileName = "ЕЦП_шаблон_файла_операций.xlsx"; return File(stream, "application/octet-stream", fileName); } private async Task CanUserAccessToWellAsync(int idWell, CancellationToken token = default) { int? idCompany = User.GetCompanyId(); return idCompany is not null && await wellService.IsCompanyInvolvedInWellAsync((int)idCompany, idWell, token).ConfigureAwait(false); } } }