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

using AsbCloudApp.Data;
using AsbCloudApp.Exceptions;
using AsbCloudApp.Services;
using AsbCloudDb.Model;
using Microsoft.AspNetCore.Authorization;
using Microsoft.AspNetCore.Mvc;
using System.Collections.Generic;
using System.ComponentModel.DataAnnotations;
using System.Threading;
using System.Threading.Tasks;

namespace AsbCloudWebApi.Controllers;

/// <summary>
/// Инфо о скважине
/// </summary>
[Route("api/well")]
[ApiController]
[Authorize]
public class WellController : ControllerBase
{
    private readonly IWellService wellService;

    public WellController(IWellService wellService)
    {
        this.wellService = wellService;
    }

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

        if (idCompany is null)
            return NoContent();

        var wells = await wellService.GetAsync(new() { IdCompany = idCompany },
            token).ConfigureAwait(false);

        return Ok(wells);
    }

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

        if (idCompany is null)
            return NoContent();

        var wells = await wellService.GetWellTreeAsync((int)idCompany,
            token).ConfigureAwait(false);

        return Ok(wells);
    }

    /// <summary>
    /// Возвращает информацию о требуемой скважине 
    /// </summary>
    /// <param name="idWell"> Id требуемой скважины </param>
    /// <param name="token"> Токен отмены задачи </param>
    /// <returns>Информация о требуемой скважине </returns>
    [HttpGet("{idWell}")]
    [Permission]
    [ProducesResponseType(typeof(WellMapInfoWithTelemetryStat), (int)System.Net.HttpStatusCode.OK)]
    [ProducesResponseType((int)System.Net.HttpStatusCode.NoContent)]
    public async Task<IActionResult> GetAsync(int idWell, CancellationToken token)
    {
        var idCompany = User.GetCompanyId();

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

        var well = await wellService.GetOrDefaultStatAsync(idWell, token);

        return Ok(well);
    }

    /// <summary>
    /// Редактирует указанные поля скважины
    /// </summary>
    /// <param name="dto"> Объект параметров скважины.
    /// IdWellType: 1 - Наклонно-направленная, 2 - Горизонтальная.
    /// State: 0 - Неизвестно, 1 - В работе, 2 - Завершена.</param>
    /// <param name="token"> Токен отмены задачи </param>
    /// <returns></returns>
    [HttpPut]
    [Permission]
    [ProducesResponseType(typeof(int), (int)System.Net.HttpStatusCode.OK)]
    public async Task<IActionResult> UpdateWellAsync(WellDto dto,
        CancellationToken token)
    {
        var idCompany = User.GetCompanyId();

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

        var result = await wellService.UpdateAsync(dto, token)
            .ConfigureAwait(false);

        return Ok(result);

    }

    /// <summary>
    /// Обновляет статус скважины
    /// </summary>
    /// <param name="idWell">ключ скважины</param>
    /// <param name="idState">статус: 0 - Неизвестно, 1 - В работе, 2 - Завершена.</param>
    /// <param name="token"></param>
    /// <returns></returns>
    [HttpPut("{idWell}/state")]
    [Permission]
    [ProducesResponseType(typeof(int), (int)System.Net.HttpStatusCode.OK)]
    public async Task<IActionResult> UpdateWellStateAsync(int idWell,
        [Range(0, 2, ErrorMessage = "Статус некорректен")] int idState,
        CancellationToken token)
    {
        var idCompany = User.GetCompanyId();

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

        var dto = await wellService.GetOrDefaultAsync(idWell, token);
        
        if (dto is null)
            return this.ValidationBadRequest(nameof(idWell), $"Скважина с id: {idWell} недоступна");

        dto.IdState = idState;

        var result = await wellService.UpdateAsync(dto, token)
            .ConfigureAwait(false);

        return Ok(result);

    }
}