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

using AsbCloudApp.Data;
using AsbCloudApp.Exceptions;
using AsbCloudApp.Services;
using Microsoft.AspNetCore.Authorization;
using Microsoft.AspNetCore.Http;
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/{idWell}/drillingProgram")]
    [ApiController]
    [Authorize]
    public class DrillingProgramController : ControllerBase
    {
        private readonly IDrillingProgramService drillingProgramService;
        private readonly IWellService wellService;

        public DrillingProgramController(IDrillingProgramService drillingProgramService, IWellService wellService)
        {
            this.drillingProgramService = drillingProgramService;
            this.wellService = wellService;
        }

        /// <summary>
        /// Список категорий файлов из которых состоит программа бурения
        /// </summary>
        /// <param name="token"></param>
        /// <returns></returns>
        [HttpGet("categories")]
        [ProducesResponseType(typeof(FileCategoryDto[]), (int)System.Net.HttpStatusCode.OK)]
        public async Task<IActionResult> GetCategories(CancellationToken token)
        {
            var result = await drillingProgramService.GetCategoriesAsync(token);
            return Ok(result);
        }

        /// <summary>
        /// Список пользователей 
        /// </summary>
        /// <param name="idWell">id скважины</param>
        /// <param name="token"></param>
        /// <returns></returns>
        [HttpGet("availableUsers")]
        [ProducesResponseType(typeof(FileCategoryDto[]), (int)System.Net.HttpStatusCode.OK)]
        public async Task<IActionResult> GetAvailableUsers(int idWell, CancellationToken token)
        {
            var result = await drillingProgramService.GetAvailableUsers(idWell, token);
            return Ok(result);
        }

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

            var idUser = User.GetUserId();

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

            var result = await drillingProgramService.GetStateAsync(idWell, (int)idUser, token);
            return Ok(result);
        }

        /// <summary>
        /// Сброс ошибки формирования ПБ
        /// </summary>
        /// <param name="idWell"></param>
        /// <returns></returns>
        [HttpPost("ClearError")]
        [Permission("DrillingProgram.get")]
        public IActionResult ClearError(int idWell)
        {
            drillingProgramService.ClearError(idWell);
            return Ok();
        }

        /// <summary>
        /// Загрузка файла для части программы бурения
        /// </summary>
        /// <param name="idWell"></param>
        /// <param name="idFileCategory">ID части программы. Не путать с категорией файла</param>
        /// <param name="files"></param>
        /// <param name="token"></param>
        /// <returns></returns>
        [HttpPost("part/{idFileCategory}")]
        [Permission("DrillingProgram.get")]
        [ProducesResponseType(typeof(int), (int)System.Net.HttpStatusCode.OK)]
        public async Task<IActionResult> AddFile(
            int idWell,
            int idFileCategory,
            [FromForm] IFormFileCollection files,
            CancellationToken token)
        {
            int? idCompany = User.GetCompanyId();
            int? idUser = User.GetUserId();

            if (idCompany is null || idUser is null)
                return Forbid();

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

            if (files.Count > 1)
                return BadRequest(ArgumentInvalidException.MakeValidationError(nameof(files), "only 1 file can be uploaded"));

            if (files.Count == 0)
                return BadRequest(ArgumentInvalidException.MakeValidationError(nameof(files), "at list 1 file should be uploaded"));

            var fileName = files[0].FileName;
            
            var fileStream = files[0].OpenReadStream();
            var result = await drillingProgramService.AddFile(idWell, idFileCategory, (int)idUser, fileName, fileStream, token);

            return Ok(result);
        }

        /// <summary>
        /// Добавляет разделы программы бурения
        /// </summary>
        /// <param name="idWell"></param>
        /// <param name="idFileCategories">Список категорий файлов, по которым будут добавлены части ПБ</param>
        /// <param name="token"></param>
        /// <returns></returns>
        [HttpPost("part")]
        [Permission]
        [ProducesResponseType(typeof(int), (int)System.Net.HttpStatusCode.OK)]
        public async Task<IActionResult> AddPartsAsync(int idWell, [Required] IEnumerable<int> idFileCategories, CancellationToken token)
        {
            int? idCompany = User.GetCompanyId();
            int? idUser = User.GetUserId();

            if (idCompany is null || idUser is null)
                return Forbid();

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

            var result = await drillingProgramService.AddPartsAsync(idWell, idFileCategories, token);
            return Ok(result);
        }

        /// <summary>
        /// Удаляет разделы программы бурения
        /// </summary>
        /// <param name="idWell"></param>
        /// <param name="idFileCategories">Список категорий файлов, по которым будут удалены части ПБ</param>
        /// <param name="token"></param>
        /// <returns></returns>
        [HttpDelete("part")]
        [Permission]
        [ProducesResponseType(typeof(int), (int)System.Net.HttpStatusCode.OK)]
        public async Task<IActionResult> RemovePartsAsync(int idWell, [Required] IEnumerable<int> idFileCategories, CancellationToken token)
        {
            int? idCompany = User.GetCompanyId();
            int? idUser = User.GetUserId();

            if (idCompany is null || idUser is null)
                return Forbid();

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

            var result = await drillingProgramService.RemovePartsAsync(idWell, idFileCategories, token);
            return Ok(result);
        }

        /// <summary>
        /// Добавить пользователя в список publishers или approvers части программы бурения
        /// </summary>
        /// <param name="idWell"></param>
        /// <param name="idUser"></param>
        /// <param name="idFileCategory"></param>
        /// <param name="idUserRole"></param>
        /// <param name="token"></param>
        /// <returns></returns>
        [HttpPost("part/{idFileCategory}/user")]
        [Permission]
        [ProducesResponseType(typeof(int), (int)System.Net.HttpStatusCode.OK)]
        public async Task<IActionResult> AddUserAsync(int idWell, [Required] int idUser, int idFileCategory, [Required] int idUserRole, CancellationToken token)
        {
            int? idCompany = User.GetCompanyId();
            int? idUserEditor = User.GetUserId();

            if (idCompany is null || idUserEditor is null)
                return Forbid();

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

            var result = await drillingProgramService.AddUserAsync(idWell, idFileCategory, idUser, idUserRole, token);
            return Ok(result);
        }

        /// <summary>
        /// Удалить пользователя из списка publishers или approvers части программы бурения
        /// </summary>
        /// <param name="idWell"></param>
        /// <param name="idUser"></param>
        /// <param name="idFileCategory"></param>
        /// <param name="idUserRole"></param>
        /// <param name="token"></param>
        /// <returns></returns>
        [HttpDelete("part/{idFileCategory}/user/{idUser}")]
        [Permission]
        [ProducesResponseType(typeof(int), (int)System.Net.HttpStatusCode.OK)]
        public async Task<IActionResult> RemoveUserAsync(int idWell, int idUser, int idFileCategory, [Required] int idUserRole, CancellationToken token)
        {
            int? idCompany = User.GetCompanyId();
            int? idUserEditor = User.GetUserId();

            if (idCompany is null || idUserEditor is null)
                return Forbid();

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

            var result = await drillingProgramService.RemoveUserAsync(idWell, idFileCategory, idUser, idUserRole, token);
            return Ok(result);
        }

        /// <summary>
        /// Создает метку для файла входящего в программу бурения, заменить существующую
        /// </summary>
        /// <param name="idWell"> id скважины </param>
        /// <param name="markDto">метка файла</param>
        /// <param name="token"> Токен отмены задачи </param>
        /// <returns></returns>
        [HttpPost("fileMark")]
        [Permission("DrillingProgram.get")]
        public async Task<IActionResult> AddOrReplaceFileMarkAsync(int idWell, FileMarkDto markDto, CancellationToken token)
        {
            var idCompany = User.GetCompanyId();

            var idUser = User.GetUserId();

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

            var result = await drillingProgramService.AddOrReplaceFileMarkAsync(markDto, (int)idUser, token)
                .ConfigureAwait(false);

            return Ok(result);
        }

        /// <summary>
        /// Помечает метку у файла входящего, в программу бурения, как удаленную
        /// </summary>
        /// <param name="idWell"> id скважины </param>
        /// <param name="idMark"> id метки </param>
        /// <param name="token"> Токен отмены задачи </param>
        /// <returns></returns>
        [HttpDelete("fileMark/{idMark}")]
        [Permission("DrillingProgram.get")]
        [ProducesResponseType(typeof(int), (int)System.Net.HttpStatusCode.OK)]
        public async Task<IActionResult> MarkAsDeletedFileMarkAsync(int idWell, int idMark,
            CancellationToken token)
        {
            var idCompany = User.GetCompanyId();

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

            var result = await drillingProgramService.MarkAsDeletedFileMarkAsync(idMark, token)
                .ConfigureAwait(false);

            return Ok(result);
        }
    }
}