DD.WellWorkover.Cloud/AsbCloudWebApi/Controllers/Trajectory/TrajectoryEditableController.cs

using AsbCloudApp.Data.Trajectory;
using AsbCloudApp.Repositories;
using AsbCloudApp.Services;
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 AsbCloudApp.Services.Parsers;
using AsbCloudInfrastructure.Services.Trajectory.Export;

namespace AsbCloudWebApi.Controllers.Trajectory;


/// <summary>
/// Плановые и фактические траектории (загрузка и хранение)
/// </summary>
[ApiController]
[Authorize]
public abstract class TrajectoryEditableController<TDto> : TrajectoryController<TDto>
  where TDto : TrajectoryGeoDto
{
  private readonly IParserService<TDto, WellRelatedParserRequest> parserService;
  private readonly ITrajectoryEditableRepository<TDto> trajectoryRepository;

  protected TrajectoryEditableController(IWellService wellService,
     IParserService<TDto, WellRelatedParserRequest>  parserService,
     TrajectoryExportService<TDto> trajectoryExportService,
        ITrajectoryEditableRepository<TDto> trajectoryRepository)
        : base(wellService, trajectoryExportService, trajectoryRepository)
  {
     this.parserService = parserService;
     this.trajectoryRepository = trajectoryRepository;
  }
  
  /// <summary>
    /// Возвращает excel шаблон для заполнения строк траектории
    /// </summary>
    /// <returns>Запрашиваемый файл</returns>
    [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", TemplateFileName);
    }

  /// <summary>
  /// Импортирует координаты из excel (xlsx) файла
  /// </summary>
  /// <param name="idWell">id скважины</param>
  /// <param name="files">Коллекция из одного файла xlsx</param>
  /// <param name="token"> Токен отмены задачи </param>
  /// <returns>количество успешно записанных строк в БД</returns>
  [HttpPost("parse")]
  [ProducesResponseType((int)System.Net.HttpStatusCode.OK)]
  [ProducesResponseType(typeof(ValidationProblemDetails), (int)System.Net.HttpStatusCode.BadRequest)]
  public async Task<ActionResult<ParserResultDto<TDto>>> 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);
     }
  }
  
  /// <summary>
  /// Добавление
  /// </summary>
  /// <param name="idWell"></param>
  /// <param name="dtos"></param>
  /// <param name="token"></param>
  /// <returns></returns>
  [HttpPost]
  [ProducesResponseType(typeof(int), (int)System.Net.HttpStatusCode.OK)]
  public async Task<IActionResult> InsertRangeAsync(int idWell, [FromBody] IEnumerable<TDto> 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);
  }

  /// <summary>
  /// Удалить все по скважине и добавить новые
  /// </summary>
  /// <param name="idWell"></param>
  /// <param name="dtos"></param>
  /// <param name="token"></param>
  /// <returns></returns>
  [HttpPost("replace")]
  [ProducesResponseType(typeof(int), (int)System.Net.HttpStatusCode.OK)]
  public async Task<IActionResult> ClearAndInsertRangeAsync(int idWell, [FromBody] IEnumerable<TDto> 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);
  }

    /// <summary>
    /// Изменить выбранную строку с координатами
    /// </summary>
    /// <param name="idWell"></param>
    /// <param name="idRow"></param>
    /// <param name="row"></param>
    /// <param name="token"></param>
    /// <returns>количество успешно обновленных строк в БД</returns>
    [HttpPut("{idRow}")]
    [ProducesResponseType(typeof(int), (int)System.Net.HttpStatusCode.OK)]
    public async Task<IActionResult> 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);
    }

    /// <summary>
    /// Удалить выбранную строку с координатами
    /// </summary>
    /// <param name="idWell"></param>
    /// <param name="idRow"></param>
    /// <param name="token"></param>
    /// <returns>количество успешно удаленных строк из БД</returns>
    [HttpDelete("{idRow}")]
    [ProducesResponseType(typeof(int), (int)System.Net.HttpStatusCode.OK)]
    public async Task<IActionResult> 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);
    }
}