using System; using AsbCloudApp.Data.Trajectory; using AsbCloudApp.Repositories; using AsbCloudApp.Services; using AsbCloudInfrastructure.Services.Trajectory.Export; using Microsoft.AspNetCore.Authorization; using Microsoft.AspNetCore.Http; using Microsoft.AspNetCore.Mvc; using System.Collections.Generic; using System.IO; using System.Threading; using System.Threading.Tasks; using AsbCloudApp.Data; using AsbCloudApp.Requests.ParserOptions; using AsbCloudInfrastructure.Services.Parser; using AsbCloudInfrastructure.Services.Trajectory.Parser; namespace AsbCloudWebApi.Controllers.Trajectory { /// /// Плановые и фактические траектории (загрузка и хранение) /// [ApiController] [Authorize] public abstract class TrajectoryEditableController : TrajectoryController where TDto : TrajectoryGeoDto { private readonly TrajectoryParser parserService; private readonly ITrajectoryEditableRepository trajectoryRepository; protected TrajectoryEditableController(IWellService wellService, TrajectoryParser parserService, TrajectoryExportService trajectoryExportService, ITrajectoryEditableRepository trajectoryRepository) : base(wellService, trajectoryExportService, trajectoryRepository) { this.parserService = parserService; this.trajectoryRepository = trajectoryRepository; } /// /// Возвращает excel шаблон для заполнения строк траектории /// /// Запрашиваемый файл [HttpGet("template")] [AllowAnonymous] [ProducesResponseType(typeof(PhysicalFileResult), (int)System.Net.HttpStatusCode.OK, "application/octet-stream")] [ProducesResponseType(StatusCodes.Status204NoContent)] public IActionResult GetTemplate() { var stream = parserService.GetTemplateFile(); return File(stream, "application/octet-stream", fileName); } /// /// Импортирует координаты из excel (xlsx) файла /// /// id скважины /// Коллекция из одного файла xlsx /// Токен отмены задачи /// количество успешно записанных строк в БД [HttpPost("parse")] [ProducesResponseType((int)System.Net.HttpStatusCode.OK)] [ProducesResponseType(typeof(ValidationProblemDetails), (int)System.Net.HttpStatusCode.BadRequest)] public async Task>> Parse(int idWell, [FromForm] IFormFileCollection files, CancellationToken token) { var idUser = User.GetUserId(); if (!idUser.HasValue) return Forbid(); if (!await CanUserAccessToWellAsync(idWell, token)) return Forbid(); var stream = files.GetExcelFile(); try { var options = new WellRelatedParserRequest(idWell); var dto = parserService.Parse(stream, options); return Ok(dto); } catch (FileFormatException ex) { return this.ValidationBadRequest(nameof(files), ex.Message); } } /// /// Добавление /// /// /// /// /// [HttpPost] [ProducesResponseType(typeof(int), (int)System.Net.HttpStatusCode.OK)] public async Task InsertRangeAsync(int idWell, [FromBody] IEnumerable dtos, CancellationToken token) { if (!await CanUserAccessToWellAsync(idWell, token)) return Forbid(); var idUser = User.GetUserId(); if (!idUser.HasValue) return Forbid(); foreach (var dto in dtos) { dto.IdUser = idUser.Value; dto.IdWell = idWell; } var result = await trajectoryRepository.AddRangeAsync(dtos, token); return Ok(result); } /// /// Удалить все по скважине и добавить новые /// /// /// /// /// [HttpPost("replace")] [ProducesResponseType(typeof(int), (int)System.Net.HttpStatusCode.OK)] public async Task ClearAndInsertRangeAsync(int idWell, [FromBody] IEnumerable dtos, CancellationToken token) { //TODO: это вся радость требует рефакторинга. //Удаление с добавлением новых записей должно происходить в рамках одной транзакции, да и вообще должно быть реализовано на уровне репозиторий. //Рефакторинг будет когда доберёмся до журнала изменений для траекторий. if (!await CanUserAccessToWellAsync(idWell, token)) return Forbid(); var idUser = User.GetUserId(); if (!idUser.HasValue) return Forbid(); foreach (var dto in dtos) { dto.IdUser = idUser.Value; dto.IdWell = idWell; } await trajectoryRepository.DeleteByIdWellAsync(idWell, token); var result = await trajectoryRepository.AddRangeAsync(dtos, token); return Ok(result); } /// /// Изменить выбранную строку с координатами /// /// /// /// /// /// количество успешно обновленных строк в БД [HttpPut("{idRow}")] [ProducesResponseType(typeof(int), (int)System.Net.HttpStatusCode.OK)] public async Task UpdateAsync(int idWell, int idRow, [FromBody] TDto row, CancellationToken token) { if (!await CanUserAccessToWellAsync(idWell, token).ConfigureAwait(false)) return Forbid(); int? idUser = User.GetUserId(); if (!idUser.HasValue) return Forbid(); row.Id = idRow; row.IdUser = idUser.Value; row.IdWell = idWell; var result = await trajectoryRepository.UpdateAsync(row, token); return Ok(result); } /// /// Удалить выбранную строку с координатами /// /// /// /// /// количество успешно удаленных строк из БД [HttpDelete("{idRow}")] [ProducesResponseType(typeof(int), (int)System.Net.HttpStatusCode.OK)] public async Task DeleteAsync(int idWell, int idRow, CancellationToken token) { if (!await CanUserAccessToWellAsync(idWell, token).ConfigureAwait(false)) return Forbid(); var result = await trajectoryRepository.DeleteRangeAsync(new int[] { idRow }, token); return Ok(result); } } }