forked from ddrilling/AsbCloudServer
237 lines
10 KiB
C#
237 lines
10 KiB
C#
using AsbCloudApp.Data;
|
||
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;
|
||
|
||
namespace AsbCloudWebApi.Controllers
|
||
{
|
||
#nullable enable
|
||
/// <summary>
|
||
/// Плановая траектория (загрузка и хранение)
|
||
/// </summary>
|
||
[Route("api/well/{idWell}/plannedTrajectory")]
|
||
[ApiController]
|
||
[Authorize]
|
||
public class PlannedTrajectoryController : ControllerBase
|
||
{
|
||
private readonly IWellService wellService;
|
||
private readonly IPlannedTrajectoryImportService plannedTrajectoryImportService;
|
||
private readonly IPlannedTrajectoryService plannedTrajectoryService;
|
||
|
||
public PlannedTrajectoryController(IWellService wellService, IPlannedTrajectoryImportService plannedTrajectoryImportService, IPlannedTrajectoryService plannedTrajectoryService)
|
||
{
|
||
this.plannedTrajectoryImportService = plannedTrajectoryImportService;
|
||
this.wellService = wellService;
|
||
this.plannedTrajectoryService = plannedTrajectoryService;
|
||
}
|
||
|
||
/// <summary>
|
||
/// Возвращает шаблон для заполнения строк плановой траектории
|
||
/// </summary>
|
||
/// <returns>Запрашиваемый файл</returns>
|
||
[HttpGet]
|
||
[Route("template")]
|
||
[AllowAnonymous]
|
||
[ProducesResponseType(typeof(PhysicalFileResult), (int)System.Net.HttpStatusCode.OK)]
|
||
public IActionResult GetTemplate()
|
||
{
|
||
var stream = plannedTrajectoryImportService.GetTemplateFile();
|
||
var fileName = "ЕЦП_шаблон_файла_плановая_траектория.xlsx";
|
||
return File(stream, "application/octet-stream", fileName);
|
||
}
|
||
|
||
/// <summary>
|
||
/// Формируем excel файл с текущими строками плановой траектории
|
||
/// </summary>
|
||
/// <param name="idWell">id скважины</param>
|
||
/// <param name="token"> Токен отмены задачи </param>
|
||
/// <returns>Запрашиваемый файл</returns>
|
||
[HttpGet]
|
||
[Route("export")]
|
||
[Permission]
|
||
[ProducesResponseType(typeof(PhysicalFileResult), (int)System.Net.HttpStatusCode.OK)]
|
||
public async Task<IActionResult> ExportAsync([FromRoute] int idWell, CancellationToken token = default)
|
||
{
|
||
if (!await CanUserAccessToWellAsync(idWell,
|
||
token).ConfigureAwait(false))
|
||
return Forbid();
|
||
var stream = await plannedTrajectoryImportService.ExportAsync(idWell, token);
|
||
var fileName = await plannedTrajectoryImportService.GetFileNameAsync(idWell, token);
|
||
return File(stream, "application/octet-stream", fileName);
|
||
}
|
||
|
||
/// <summary>
|
||
/// Импортирует координаты из excel (xlsx) файла
|
||
/// </summary>
|
||
/// <param name="idWell">id скважины</param>
|
||
/// <param name="files">Коллекция из одного файла xlsx</param>
|
||
/// <param name="deleteBeforeImport">Удалить операции перед импортом, если фал валидный</param>
|
||
/// <param name="token"> Токен отмены задачи </param>
|
||
/// <returns>количество успешно записанных строк в БД</returns>
|
||
[HttpPost]
|
||
[Permission]
|
||
[Route("import/{deleteBeforeImport}")]
|
||
[ProducesResponseType(typeof(int), (int)System.Net.HttpStatusCode.OK)]
|
||
public async Task<IActionResult> ImportAsync(int idWell,
|
||
[FromForm] IFormFileCollection files,
|
||
bool deleteBeforeImport,
|
||
CancellationToken token)
|
||
{
|
||
int? idUser = User.GetUserId();
|
||
if (!idUser.HasValue)
|
||
return Forbid();
|
||
if (!await CanUserAccessToWellAsync(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
|
||
{
|
||
var result = await plannedTrajectoryImportService.ImportAsync(idWell, idUser.Value, stream, deleteBeforeImport, token);
|
||
return Ok(result);
|
||
}
|
||
catch (FileFormatException ex)
|
||
{
|
||
return BadRequest(ex.Message);
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// Получаем список всех строк координат плановой траектории (для клиента)
|
||
/// </summary>
|
||
/// <param name="idWell">id скважины</param>
|
||
/// <param name="token"> Токен отмены задачи </param>
|
||
/// <returns>Список добавленных координат плановой траектории</returns>
|
||
[HttpGet]
|
||
[Route("getRows")]
|
||
[Permission]
|
||
[ProducesResponseType(typeof(IEnumerable<PlannedTrajectoryDto>), (int)System.Net.HttpStatusCode.OK)]
|
||
public async Task<IActionResult> GetAsync([FromRoute] int idWell, CancellationToken token = default)
|
||
{
|
||
if (!await CanUserAccessToWellAsync(idWell,
|
||
token).ConfigureAwait(false))
|
||
return Forbid();
|
||
var result = await plannedTrajectoryService.GetAsync(idWell, token);
|
||
return Ok(result);
|
||
}
|
||
|
||
/// <summary>
|
||
/// Добавить одну новую строчку координат для плановой траектории
|
||
/// </summary>
|
||
/// <param name="idWell"></param>
|
||
/// <param name="row"></param>
|
||
/// <param name="token"></param>
|
||
/// <returns>количество успешно записанных строк в БД</returns>
|
||
[HttpPost]
|
||
[Route("addRow")]
|
||
[Permission]
|
||
[ProducesResponseType(typeof(int), (int)System.Net.HttpStatusCode.OK)]
|
||
public async Task<IActionResult> AddAsync(int idWell, [FromBody] PlannedTrajectoryDto row,
|
||
CancellationToken token = default)
|
||
{
|
||
if (!await CanUserAccessToWellAsync(idWell, token).ConfigureAwait(false))
|
||
return Forbid();
|
||
var idUser = User.GetUserId();
|
||
if (!idUser.HasValue)
|
||
return Forbid();
|
||
row.IdUser = idUser.Value;
|
||
row.IdWell = idWell;
|
||
var result = await plannedTrajectoryService.AddAsync(row, token);
|
||
return Ok(result);
|
||
}
|
||
|
||
/// <summary>
|
||
/// Добавить массив строчек координат для плановой траектории
|
||
/// </summary>
|
||
/// <param name="idWell"></param>
|
||
/// <param name="rows"></param>
|
||
/// <param name="token"></param>
|
||
/// <returns>количество успешно записанных строк в БД</returns>
|
||
[HttpPost]
|
||
[Route("addRangeRows")]
|
||
[Permission]
|
||
[ProducesResponseType(typeof(int), (int)System.Net.HttpStatusCode.OK)]
|
||
public async Task<IActionResult> AddRangeAsync(int idWell, [FromBody] IEnumerable<PlannedTrajectoryDto> rows,
|
||
CancellationToken token = default)
|
||
{
|
||
if (!await CanUserAccessToWellAsync(idWell, token).ConfigureAwait(false))
|
||
return Forbid();
|
||
int? idUser = User.GetUserId();
|
||
if (!idUser.HasValue)
|
||
return Forbid();
|
||
foreach (var item in rows)
|
||
{
|
||
item.IdUser = idUser.Value;
|
||
item.IdWell = idWell;
|
||
}
|
||
var result = await plannedTrajectoryService.AddRangeAsync(rows, 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}")]
|
||
[Permission]
|
||
[ProducesResponseType(typeof(int), (int)System.Net.HttpStatusCode.OK)]
|
||
public async Task<IActionResult> UpdateAsync(int idWell, int idRow,
|
||
[FromBody] PlannedTrajectoryDto row, CancellationToken token = default)
|
||
{
|
||
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 plannedTrajectoryService.UpdateAsync(row, token);
|
||
return Ok(result);
|
||
}
|
||
|
||
/// <summary>
|
||
/// Удалить выбранную строку с координатами
|
||
/// </summary>
|
||
/// <param name="idWell"></param>
|
||
/// <param name="idRow"></param>
|
||
/// <param name="token"></param>
|
||
/// <returns>количество успешно удаленных строк из БД</returns>
|
||
[HttpDelete("{idRow}")]
|
||
[Permission]
|
||
[ProducesResponseType(typeof(int), (int)System.Net.HttpStatusCode.OK)]
|
||
public async Task<IActionResult> DeleteAsync(int idWell, int idRow, CancellationToken token = default)
|
||
{
|
||
if (!await CanUserAccessToWellAsync(idWell,
|
||
token).ConfigureAwait(false))
|
||
return Forbid();
|
||
|
||
var result = await plannedTrajectoryService.DeleteRangeAsync(new int[] { idRow }, token);
|
||
|
||
return Ok(result);
|
||
}
|
||
|
||
private async Task<bool> CanUserAccessToWellAsync(int idWell, CancellationToken token = default)
|
||
{
|
||
int? idCompany = User.GetCompanyId();
|
||
return idCompany is not null && await wellService.IsCompanyInvolvedInWellAsync((int)idCompany,
|
||
idWell, token).ConfigureAwait(false);
|
||
}
|
||
}
|
||
#nullable disable
|
||
}
|