DD.WellWorkover.Cloud/AsbCloudWebApi/Controllers/SAUB/TelemetryDataSaubController.cs

using AsbCloudApp.Data.SAUB;
using AsbCloudApp.Services;
using AsbCloudWebApi.SignalR;
using AsbCloudWebApi.SignalR.Clients;
using Microsoft.AspNetCore.Http;
using Microsoft.AspNetCore.Mvc;
using Microsoft.AspNetCore.SignalR;
using System;
using System.Collections.Generic;
using System.Threading;
using System.Threading.Tasks;

namespace AsbCloudWebApi.Controllers.SAUB
{
    /// <summary>
    /// Данные АКБ
    /// </summary>
    [Route("api/[controller]")]
    [ApiController]
    public class TelemetryDataSaubController : TelemetryDataBaseController<TelemetryDataSaubDto>
    {
        private readonly ITelemetryDataSaubService telemetryDataSaubService;

        public TelemetryDataSaubController(
            ITelemetryService telemetryService,
            ITelemetryDataSaubService telemetryDataService,
            IWellService wellService,
            IHubContext<TelemetryHub, ITelemetryHubClient> telemetryHubContext)
            : base(
                  telemetryService,
                  telemetryDataService,
                  wellService,
                  telemetryHubContext)
        {
            telemetryDataSaubService = telemetryDataService;
        }

        /// <summary>
        /// Выгрузка архива. Не более 3-х суток. Формат даты строгий.
        /// </summary>
        /// <param name="idWell">id скважины (из адресной строки)</param>
        /// <param name="beginDate">начало интервала в формате: yyyy-MM-DD HH:mm</param>
        /// <param name="endDate">конец интервала в формате: yyyy-MM-DD HH:mm</param>
        /// <param name="token"></param>
        /// <returns></returns>
        [HttpGet("{idWell}/export/csv")]
        [ProducesResponseType(typeof(PhysicalFileResult), (int)System.Net.HttpStatusCode.OK, "application/octet-stream")]
        [ProducesResponseType(StatusCodes.Status204NoContent)]
        public async Task<IActionResult> GetZippedCsv(int idWell, DateTime beginDate, DateTime endDate, CancellationToken token)
        {
            int? idCompany = User.GetCompanyId();

            if (idCompany is null)
                return Forbid();

            bool isCompanyOwnsWell = await wellService.IsCompanyInvolvedInWellAsync((int)idCompany,
                idWell, token).ConfigureAwait(false);

            if (!isCompanyOwnsWell)
                return Forbid();

            var stream = await telemetryDataSaubService.GetZippedCsv(idWell, beginDate, endDate, token).ConfigureAwait(false);
            var fileName = $"DataSaub idWell{idWell} {beginDate:yyyy-MM-DDTHH-mm} - {endDate:yyyy-MM-DDTHH-mm}.zip";
            return File(stream, "application/octet-stream", fileName);
        }

        protected override Task SignalRNotifyAsync(int idWell, IEnumerable<TelemetryDataSaubDto> dtos, CancellationToken token)
        {
            return telemetryHubContext.Clients.Group($"well_{idWell}").ReceiveDataSaub(dtos, token);
        }
    }
}