DD.WellWorkover.Cloud/AsbCloudWebApi/Controllers/DrillParamsController.cs

using AsbCloudApp.Data;
using AsbCloudApp.Services;
using Microsoft.AspNetCore.Mvc;
using System.Collections.Generic;
using System.Threading;
using System.Threading.Tasks;

namespace AsbCloudWebApi.Controllers
{
    /// <summary>
    /// Режимы бурения
    /// </summary>
    [Route("api/well/{idWell}/drillParams/")]
    [ApiController]
    public class DrillParamsController : ControllerBase
    {
        private readonly IDrillParamsService drillParamsService;
        private readonly IWellService wellService;

        public DrillParamsController(IDrillParamsService drillParamsService,
            IWellService wellService)
        {
            this.drillParamsService = drillParamsService;
            this.wellService = wellService;
        }

        /// <summary>
        /// Возвращает автоматически рассчитанные значения для режимов бурения
        /// </summary>
        /// <param name="idWell"> id скважины </param>
        /// <param name="startDepth"> Стартовая глубина </param>
        /// <param name="endDepth"> Конечная глубина </param>
        /// <param name="token"> Токен отмены задачи </param>
        /// <returns> Значения по умолчанию для режимов бурения </returns>
        [HttpGet("autoParams")]
        [Permission]
        [ProducesResponseType(typeof(DrillParamsDto), (int)System.Net.HttpStatusCode.OK)]
        public async Task<IActionResult> GetDefaultAsync(int idWell,
            double startDepth, double endDepth, CancellationToken token = default)
        {
            var idCompany = User.GetCompanyId();

            if (idCompany is null || !await wellService.IsCompanyInvolvedInWellAsync((int)idCompany,
                idWell, token).ConfigureAwait(false))
                return Forbid();

            var drillParamsDo = await drillParamsService.GetDefaultDrillParamsAsync(idWell,
                 startDepth, endDepth, token);

            return Ok(drillParamsDo);
        }

        /// <summary>
        /// Возвращает все значения для режимов бурения на скважине
        /// </summary>
        /// <param name="idWell"> id скважины </param>
        /// <param name="token"> Токен отмены задачи </param>
        /// <returns> Список параметров для режимов бурения на скважине </returns>
        [HttpGet]
        [Permission]
        [ProducesResponseType(typeof(IEnumerable<DrillParamsDto>), (int)System.Net.HttpStatusCode.OK)]
        public async Task<IActionResult> GetAllAsync(int idWell,
            CancellationToken token = default)
        {
            var idCompany = User.GetCompanyId();

            if (idCompany is null || !await wellService.IsCompanyInvolvedInWellAsync((int)idCompany,
                idWell, token).ConfigureAwait(false))
                return Forbid();

            var dto = await drillParamsService.GetAllAsync(idWell, token);

            return Ok(dto);
        }

        /// <summary>
        /// Сохраняет значения для режимов бурения по секции на скважине
        /// </summary>
        /// <param name="idWell"> id скважины </param>
        /// <param name="drillParamsDto"> Параметры режимов бурений для секции</param>
        /// <param name="token"> Токен отмены задачи </param>
        /// <returns></returns>
        [HttpPost]
        [Permission]
        [ProducesResponseType(typeof(int), (int)System.Net.HttpStatusCode.OK)]
        public async Task<IActionResult> InsertAsync(int idWell,
            DrillParamsDto drillParamsDto, CancellationToken token = default)
        {
            var idCompany = User.GetCompanyId();

            if (idCompany is null || !await wellService.IsCompanyInvolvedInWellAsync((int)idCompany,
                idWell, token).ConfigureAwait(false))
                return Forbid();

            var result = await drillParamsService.InsertAsync(idWell, drillParamsDto, token);

            return Ok(result);
        }

        /// <summary>
        /// Добавляет массив объектов режимов бурений
        /// </summary>
        /// <param name="idWell"> id скважины </param>
        /// <param name="drillParams"> Массив объектов параметров режима бурений для секции</param>
        /// <param name="token"> Токен отмены задачи </param>
        /// <returns></returns>
        [HttpPost("range")]
        [Permission]
        [ProducesResponseType(typeof(int), (int)System.Net.HttpStatusCode.OK)]
        public async Task<IActionResult> InsertRangeAsync(int idWell,
            IEnumerable<DrillParamsDto> drillParams, CancellationToken token = default)
        {
            var idCompany = User.GetCompanyId();

            if (idCompany is null || !await wellService.IsCompanyInvolvedInWellAsync((int)idCompany,
                idWell, token).ConfigureAwait(false))
                return Forbid();

            var result = await drillParamsService.InsertRangeAsync(idWell, drillParams,
                token);

            return Ok(result);
        }

        /// <summary>
        /// Удаляет старые режимы бурения по скважине и добавляет новые
        /// </summary>
        /// <param name="idWell">Id скважины для добавления</param>
        /// <param name="drillParams">Новые режимы бурения </param>
        /// <param name="token"></param>
        /// <returns>кол-во затронутых изменениями записей</returns>
        [HttpPost("save")]
        [Permission]
        [ProducesResponseType(typeof(int), (int)System.Net.HttpStatusCode.OK)]
        public async Task<IActionResult> SaveAsync(int idWell, IEnumerable<DrillParamsDto> drillParams, CancellationToken token = default)
        {
            var idCompany = User.GetCompanyId();

            if (idCompany is null || !await wellService.IsCompanyInvolvedInWellAsync((int)idCompany,
                idWell, token).ConfigureAwait(false))
                return Forbid();

            var result = await drillParamsService.SaveAsync(idWell, drillParams,
                token);

            return Ok(result);
        }

        /// <summary>
        /// Изменяет значения выбранного режима бурения
        /// </summary>
        /// <param name="idWell"> id скважины </param>
        /// <param name="dtoId"> id dto для изменения </param>
        /// <param name="drillParamsDto"> Параметры режимов бурений для секции</param>
        /// <param name="token"> Токен отмены задачи </param>
        /// <returns></returns>
        [HttpPut]
        [Permission]
        [ProducesResponseType(typeof(int), (int)System.Net.HttpStatusCode.OK)]
        public async Task<IActionResult> UpdateAsync(int idWell, int dtoId,
            DrillParamsDto drillParamsDto, CancellationToken token = default)
        {
            var idCompany = User.GetCompanyId();

            if (idCompany is null || !await wellService.IsCompanyInvolvedInWellAsync((int)idCompany,
                idWell, token).ConfigureAwait(false))
                return Forbid();

            var result = await drillParamsService.UpdateAsync(idWell, dtoId,
                drillParamsDto, token);

            return Ok(result);
        }

        /// <summary>
        /// Удаляет объект значений выбранного режима бурения
        /// </summary>
        /// <param name="idWell"> id скважины </param>
        /// <param name="drillParamsId"> Id объекта параметров режима бурений для секции</param>
        /// <param name="token"> Токен отмены задачи </param>
        /// <returns></returns>
        [HttpDelete]
        [Permission]
        [ProducesResponseType(typeof(int), (int)System.Net.HttpStatusCode.OK)]
        public async Task<IActionResult> DeleteAsync(int idWell,
            int drillParamsId, CancellationToken token = default)
        {
            var idCompany = User.GetCompanyId();

            if (idCompany is null || !await wellService.IsCompanyInvolvedInWellAsync((int)idCompany,
                idWell, token).ConfigureAwait(false))
                return Forbid();

            var result = await drillParamsService.DeleteAsync(drillParamsId,
                token);

            return Ok(result);
        }

        /// <summary>
        /// Возвращает значения для режимов бурения на композитной скважине
        /// </summary>
        /// <param name="idWell"> id скважины </param>
        /// <param name="token"> Токен отмены задачи </param>
        /// <returns> Список параметров для режимов бурения на композитной скважине </returns>
        [HttpGet("composite")]
        [Permission]
        [ProducesResponseType(typeof(IEnumerable<DrillParamsDto>), (int)System.Net.HttpStatusCode.OK)]
        public async Task<IActionResult> GetCompositeAllAsync(int idWell, CancellationToken token = default)
        {
            var idCompany = User.GetCompanyId();

            if (idCompany is null || !await wellService.IsCompanyInvolvedInWellAsync((int)idCompany,
                idWell, token).ConfigureAwait(false))
                return Forbid();

            var dto = await drillParamsService.GetCompositeAllAsync(idWell, token);

            return Ok(dto);
        }
    }
}