using AsbCloudApp.Data; using AsbCloudApp.Repositories; using AsbCloudApp.Requests; using AsbCloudApp.Services; using Microsoft.AspNetCore.Authorization; using Microsoft.AspNetCore.Http; using Microsoft.AspNetCore.Mvc; using System; using System.Collections.Generic; using System.ComponentModel.DataAnnotations; using System.IO; using System.Threading; using System.Threading.Tasks; using AsbCloudApp.Services.WellOperationImport; using AsbCloudApp.Data.WellOperationImport.Options; using AsbCloudApp.Exceptions; namespace AsbCloudWebApi.Controllers { /// /// Буровые операции (вводимые вручную) /// [Route("api/well/{idWell}/wellOperations")] [ApiController] [Authorize] public class WellOperationController : ControllerBase { private readonly IWellOperationRepository operationRepository; private readonly IWellService wellService; private readonly IWellOperationExportService wellOperationExportService; private readonly IWellOperationImportTemplateService wellOperationImportTemplateService; private readonly IWellOperationImportService wellOperationImportService; private readonly IWellOperationExcelParser wellOperationDefaultExcelParser; private readonly IWellOperationExcelParser wellOperationGazpromKhantosExcelParser; private readonly IUserRepository userRepository; public WellOperationController(IWellOperationRepository operationRepository, IWellService wellService, IWellOperationImportTemplateService wellOperationImportTemplateService, IWellOperationExportService wellOperationExportService, IWellOperationImportService wellOperationImportService, IWellOperationExcelParser wellOperationDefaultExcelParser, IWellOperationExcelParser wellOperationGazpromKhantosExcelParser, IUserRepository userRepository) { this.operationRepository = operationRepository; this.wellService = wellService; this.wellOperationImportTemplateService = wellOperationImportTemplateService; this.wellOperationExportService = wellOperationExportService; this.wellOperationImportService = wellOperationImportService; this.wellOperationDefaultExcelParser = wellOperationDefaultExcelParser; this.wellOperationGazpromKhantosExcelParser = wellOperationGazpromKhantosExcelParser; this.userRepository = userRepository; } /// /// Возвращает словарь типов секций /// /// [HttpGet("sectionTypes")] [Permission] [ProducesResponseType(typeof(IEnumerable), (int)System.Net.HttpStatusCode.OK)] public IActionResult GetSectionTypes() { var result = operationRepository.GetSectionTypes(); return Ok(result); } /// /// Возвращает список имен типов операций на скважине /// /// флаг, нужно ли включать родителей в список /// [HttpGet("categories")] [Permission] [ProducesResponseType(typeof(IEnumerable), (int)System.Net.HttpStatusCode.OK)] public IActionResult GetCategories(bool includeParents = true) { var result = operationRepository.GetCategories(includeParents); return Ok(result); } /// /// Возвращает список плановых операций для сопоставления /// /// id скважины /// дата для нахождения последней сопоставленной плановой операции /// /// [HttpGet("operationsPlan")] [ProducesResponseType(typeof(WellOperationPlanDto), (int)System.Net.HttpStatusCode.OK)] public async Task GetOperationsPlanAsync( [FromRoute] int idWell, [FromQuery] DateTime currentDate, CancellationToken token) { if (!await CanUserAccessToWellAsync(idWell, token).ConfigureAwait(false)) return Forbid(); var result = await operationRepository .GetOperationsPlanAsync(idWell, currentDate, token) .ConfigureAwait(false); return Ok(result); } /// /// Отфильтрованный список фактических операций на скважине. /// Если не применять фильтр, то вернется весь список. Сортированный по глубине затем по дате /// /// id скважины /// /// /// Список операций на скважине [HttpGet("fact")] [Permission] [ProducesResponseType(typeof(IEnumerable), (int)System.Net.HttpStatusCode.OK)] public async Task GetPageOperationsFactAsync( [FromRoute] int idWell, [FromQuery] WellOperationRequestBase request, CancellationToken token) { if (!await CanUserAccessToWellAsync(idWell, token).ConfigureAwait(false)) return Forbid(); var requestToService = new WellOperationRequest(request, idWell); var result = await operationRepository.GetAsync( requestToService, token) .ConfigureAwait(false); return Ok(result); } /// /// Отфильтрованный список плановых операций на скважине. /// Если не применять фильтр, то вернется весь список. Сортированный по глубине затем по дате /// /// id скважины /// /// /// Список операций на скважине в контейнере для постраничного просмотра [HttpGet("plan")] [Permission] [ProducesResponseType(typeof(PaginationContainer), (int)System.Net.HttpStatusCode.OK)] public async Task GetPageOperationsPlanAsync( [FromRoute] int idWell, [FromQuery] WellOperationRequestBase request, CancellationToken token) { if (!await CanUserAccessToWellAsync(idWell, token).ConfigureAwait(false)) return Forbid(); var requestToService = new WellOperationRequest(request, idWell); var result = await operationRepository.GetPageAsync( requestToService, token) .ConfigureAwait(false); return Ok(result); } /// /// Статистика операций по скважине, группированая по категориям /// /// id скважины /// /// /// [HttpGet("groupStat")] [Permission] [ProducesResponseType(typeof(IEnumerable), (int)System.Net.HttpStatusCode.OK)] public async Task GetGroupOperationsAsync( [FromRoute] int idWell, [FromQuery] WellOperationRequestBase request, CancellationToken token) { if (!await CanUserAccessToWellAsync(idWell, token).ConfigureAwait(false)) return Forbid(); var requestToService = new WellOperationRequest(request, idWell); var result = await operationRepository.GetGroupOperationsStatAsync( requestToService, token) .ConfigureAwait(false); return Ok(result); } /// /// Возвращает нужную операцию на скважине /// /// id скважины /// id нужной операции /// Токен отмены задачи /// Нужную операцию на скважине [HttpGet("{idOperation}")] [Permission] [ProducesResponseType(typeof(WellOperationDto), (int)System.Net.HttpStatusCode.OK)] public async Task GetOrDefaultAsync(int idWell, int idOperation, CancellationToken token) { if (!await CanUserAccessToWellAsync(idWell, token).ConfigureAwait(false)) return Forbid(); var result = await operationRepository.GetOrDefaultAsync(idOperation, token).ConfigureAwait(false); return Ok(result); } /// /// Добавляет новые операции на скважине /// /// id скважины /// Данные о добавляемых операциях /// Токен отмены задачи /// Количество добавленных в БД строк [HttpPost] [Permission] [ProducesResponseType(typeof(IEnumerable), (int)System.Net.HttpStatusCode.OK)] public async Task InsertRangeAsync( [Range(1, int.MaxValue, ErrorMessage = "Id скважины не может быть меньше 1")] int idWell, [FromBody] IEnumerable values, CancellationToken token) { if (!await CanUserAccessToWellAsync(idWell, token)) return Forbid(); if (!await CanUserEditWellOperationsAsync(idWell, token)) return Forbid(); foreach (var value in values) { value.IdWell = idWell; value.LastUpdateDate = DateTimeOffset.UtcNow; value.IdUser = User.GetUserId(); } var result = await operationRepository.InsertRangeAsync(values, token) .ConfigureAwait(false); return Ok(result); } /// /// Обновляет выбранную операцию на скважине /// /// id скважины /// id выбранной операции /// Новые данные для выбранной операции /// Токен отмены задачи /// Количество обновленных в БД строк [HttpPut("{idOperation}")] [Permission] [ProducesResponseType(typeof(WellOperationDto), (int)System.Net.HttpStatusCode.OK)] public async Task UpdateAsync(int idWell, int idOperation, [FromBody] WellOperationDto value, CancellationToken token) { if (!await CanUserAccessToWellAsync(idWell, token)) return Forbid(); if (!await CanUserEditWellOperationsAsync(idWell, token)) return Forbid(); value.IdWell = idWell; value.Id = idOperation; value.LastUpdateDate = DateTimeOffset.UtcNow; value.IdUser = User.GetUserId(); var result = await operationRepository.UpdateAsync(value, token) .ConfigureAwait(false); return Ok(result); } /// /// Удаляет выбранную операцию на скважине /// /// id скважины /// id выбранной операции /// Токен отмены задачи /// Количество удаленных из БД строк [HttpDelete("{idOperation}")] [Permission] [ProducesResponseType(typeof(int), (int)System.Net.HttpStatusCode.OK)] public async Task DeleteAsync(int idWell, int idOperation, CancellationToken token) { if (!await CanUserAccessToWellAsync(idWell, token)) return Forbid(); if (!await CanUserEditWellOperationsAsync(idWell, token)) return Forbid(); var result = await operationRepository.DeleteAsync(new int[] { idOperation }, token) .ConfigureAwait(false); return Ok(result); } /// /// Импорт операций из excel (xlsx) файла. Стандартный заполненный шаблон /// /// id скважины /// Параметры для парсинга файла /// Коллекция из одного файла xlsx /// Удалить операции перед импортом = 1, если файл валидный /// /// [HttpPost("import/default/{deleteBeforeImport}")] [Permission] public async Task ImportDefaultExcelFileAsync(int idWell, [FromQuery] WellOperationImportDefaultOptionsDto options, [FromForm] IFormFileCollection files, [Range(0, 1, ErrorMessage = "Недопустимое значение. Допустимые: 0, 1")] int deleteBeforeImport, CancellationToken token) { var idUser = User.GetUserId(); if (!idUser.HasValue) throw new ForbidException("Неизвестный пользователь"); await AssertUserHasAccessToImportWellOperationsAsync(idWell, token); if (files.Count < 1) return this.ValidationBadRequest(nameof(files), "Нет файла"); var file = files[0]; if (Path.GetExtension(file.FileName).ToLower() != ".xlsx") return this.ValidationBadRequest(nameof(files), "Требуется xlsx файл."); using Stream stream = file.OpenReadStream(); try { var sheet = wellOperationDefaultExcelParser.Parse(stream, options); await wellOperationImportService.ImportAsync(idWell, idUser.Value, options.IdType, sheet, (deleteBeforeImport & 1) > 0, token); } catch (FileFormatException ex) { return this.ValidationBadRequest(nameof(files), ex.Message); } return Ok(); } /// /// Импорт операций из excel (xlsx) файла. ГПНХ (Хантос) /// /// id скважины /// Параметры для парсинга файла /// Коллекция из одного файла xlsx /// Удалить операции перед импортом = 1, если файл валидный /// /// [HttpPost("import/gazpromKhantos/{deleteBeforeImport}")] [Permission] public async Task ImportGazpromKhantosExcelFileAsync(int idWell, [FromQuery] WellOperationImportGazpromKhantosOptionsDto options, [FromForm] IFormFileCollection files, [Range(0, 1, ErrorMessage = "Недопустимое значение. Допустимые: 0, 1")] int deleteBeforeImport, CancellationToken token) { var idUser = User.GetUserId(); if (!idUser.HasValue) throw new ForbidException("Неизвестный пользователь"); await AssertUserHasAccessToImportWellOperationsAsync(idWell, token); if (files.Count < 1) return this.ValidationBadRequest(nameof(files), "Нет файла"); var file = files[0]; if (Path.GetExtension(file.FileName).ToLower() != ".xlsx") return this.ValidationBadRequest(nameof(files), "Требуется xlsx файл."); using Stream stream = file.OpenReadStream(); try { var sheet = wellOperationGazpromKhantosExcelParser.Parse(stream, options); await wellOperationImportService.ImportAsync(idWell, idUser.Value, options.IdType, sheet, (deleteBeforeImport & 1) > 0, token); } catch (FileFormatException ex) { return this.ValidationBadRequest(nameof(files), ex.Message); } return Ok(); } /// /// Создает excel файл с операциями по скважине /// /// id скважины /// Токен отмены задачи /// Запрашиваемый файл [HttpGet("export")] [Permission] [ProducesResponseType(typeof(PhysicalFileResult), (int)System.Net.HttpStatusCode.OK, "application/octet-stream")] [ProducesResponseType(StatusCodes.Status204NoContent)] public async Task ExportAsync([FromRoute] int idWell, CancellationToken token) { int? idCompany = User.GetCompanyId(); if (idCompany is null) return Forbid(); if (!await wellService.IsCompanyInvolvedInWellAsync((int)idCompany, idWell, token).ConfigureAwait(false)) return Forbid(); var stream = await wellOperationExportService.ExportAsync(idWell, token); var fileName = await wellService.GetWellCaptionByIdAsync(idWell, token) + "_operations.xlsx"; return File(stream, "application/octet-stream", fileName); } /// /// Создает excel файл с "сетевым графиком" /// /// id скважины /// /// Токен отмены задачи /// Запрашиваемый файл [HttpGet("scheduleReport")] [Permission] [ProducesResponseType(typeof(PhysicalFileResult), (int)System.Net.HttpStatusCode.OK)] public async Task ScheduleReportAsync([FromRoute] int idWell, [FromServices] IScheduleReportService scheduleReportService, CancellationToken token) { int? idCompany = User.GetCompanyId(); if (idCompany is null) return Forbid(); if (!await wellService.IsCompanyInvolvedInWellAsync((int)idCompany, idWell, token).ConfigureAwait(false)) return Forbid(); var stream = await scheduleReportService.MakeReportAsync(idWell, token); var fileName = await wellService.GetWellCaptionByIdAsync(idWell, token) + "_ScheduleReport.xlsx"; return File(stream, "application/octet-stream", fileName); } /// /// Возвращает шаблон файла импорта /// /// Запрашиваемый файл [HttpGet("template")] [AllowAnonymous] [ProducesResponseType(typeof(PhysicalFileResult), (int)System.Net.HttpStatusCode.OK, "application/octet-stream")] public IActionResult GetTemplate() { var stream = wellOperationImportTemplateService.GetExcelTemplateStream(); var fileName = "ЕЦП_шаблон_файла_операций.xlsx"; return File(stream, "application/octet-stream", fileName); } private async Task AssertUserHasAccessToImportWellOperationsAsync(int idWell, CancellationToken token) { var idCompany = User.GetCompanyId(); var idUser = User.GetUserId(); if (!idCompany.HasValue || !idUser.HasValue) throw new ForbidException("Неизвестный пользователь"); if (!await CanUserAccessToWellAsync(idWell, token)) throw new ForbidException("Нет доступа к скважине"); if (!await CanUserEditWellOperationsAsync(idWell, token)) throw new ForbidException("Недостаточно прав для редактирования ГГД на завершенной скважине"); if (!await wellService.IsCompanyInvolvedInWellAsync(idCompany.Value, idWell, token)) throw new ForbidException("Скважина недоступна для компании"); } private async Task CanUserEditWellOperationsAsync(int idWell, CancellationToken token) { var idUser = User.GetUserId(); if (!idUser.HasValue) return false; var well = await wellService.GetOrDefaultAsync(idWell, token); if (well is null) return false; return well.IdState != 2 || userRepository.HasPermission(idUser.Value, "WellOperation.editCompletedWell"); } private async Task CanUserAccessToWellAsync(int idWell, CancellationToken token) { int? idCompany = User.GetCompanyId(); return idCompany is not null && await wellService.IsCompanyInvolvedInWellAsync((int)idCompany, idWell, token).ConfigureAwait(false); } } }