Squirrowse/Squirrowse.Service/Hubs/StreamHub.cs

using System;
using System.Collections.Generic;
using System.Globalization;
using System.Linq;
using System.Threading.Tasks;
using Microsoft.AspNetCore.SignalR;
using Microsoft.Extensions.Logging;
using OpenCvSharp;
using Squirrowse.Core.Models;
using Squirrowse.Core.Services;

namespace Squirrowse.Service.Hubs
{
    public class StreamHub : Hub, IStreamHub //fujka
    {
        private readonly ILogger<StreamHub> logger;
        private readonly IStreamManager manager;


        public StreamHub(ILogger<StreamHub> logger, IStreamManager manager)
        {
            this.logger = logger;
            this.manager = manager;
        }

        public async Task AddUser(string UserName, ConnectionType type)
        {
            await manager.AddUser(Context.ConnectionId, UserName, type);
            switch (type)
            {
                case ConnectionType.Client:
                    await AddToGroup(Core.Models.Groups.normal);
                    break;
                case ConnectionType.Server:
                    await AddToGroup(Core.Models.Groups.superUser);
                    break;
            }

            logger.LogInformation($"{nameof(AddUser)}: {UserName} of {type}");
        }


        public async Task AddToGroup(Groups group, string user = "")
        {
            var connectionId = string.IsNullOrWhiteSpace(user) ? Context.ConnectionId : user;
            await Groups.AddToGroupAsync(connectionId, group.ToString());
            logger.LogInformation($"{nameof(AddToGroup)}: {connectionId} joined to {group}");
        }

        /// <summary>
        ///     USE ONLY FOR DEBUG
        /// </summary>
        /// <param name="act"></param>
        /// <returns>DESTRUCTION</returns>
        public async Task ExecCommandOnAll(string command, object[] act)
        {
            await Clients.All.SendCoreAsync(command, act);
        }

        public async Task RemoveFromGroup(Groups group, string user = "")
        {
            var connectionId = string.IsNullOrWhiteSpace(user) ? Context.ConnectionId : user;
            await Groups.RemoveFromGroupAsync(connectionId, group.ToString());
            logger.LogInformation($"{nameof(AddToGroup)}: {connectionId} joined to {group}");
        }

        public async Task Startstream(string clientId)
        {
            streamOn = true;
            //var client = Clients.Client(clientId);
            await Clients.Groups(Core.Models.Groups.normal.ToString()).SendAsync("Start");
            // await client.SendAsync("Start");
        }


        public async Task StopStream(string clientId)
        {
            streamOn = false;
            //var client = Clients.Client(clientId);
            // await Clients.Groups(Core.Models.Groups.normal.ToString()).SendAsync("Stop");
            //await client.SendAsync("Stop");
        }

        public static bool streamOn { get; set; }

        public async Task UploadByteStream(IAsyncEnumerable<byte[]> stream)
        {
            
            await foreach (var frame in stream)
            {
                //    var frameStamp = frame.ConvertByteToMat();
                //    string timestamp = DateTime.UtcNow.ToString("yyyy-MM-dd HH:mm:ss.fff",
                //        CultureInfo.InvariantCulture);
                //    Cv2.PutText(frameStamp, timestamp, new Point(1, 480 / 2), HersheyFonts.HersheySimplex, 0.5f, Scalar.Red);
                logger.LogInformation($"{this.GetHashCode()}: send frame");

                //logger.LogInformation($"Got frame size: {frame.Length} ");
                //if (!streamOn) continue;
                //  logger.LogInformation($"Send frame of size: {frame.Length}  to {Core.Models.Groups.superUser.ToString()}");
                var framewithTs = await AddTimestamp(frame);
                await Clients.Groups(Core.Models.Groups.superUser.ToString()).SendAsync("RecData",framewithTs );
                logger.LogInformation($"{this.GetHashCode()}: send data");
                //await Task.Delay(100); //leave some delay for debug purpose 
                await Task.Delay(1000 / 30);
            }
        }
        public async Task StartRecord(string connectionId)
        {
            
        }
        public async Task StopRecord(string connectionId)
        {

        }
        public async Task<byte[]> AddTimestamp(byte[] framebytes)
        {

            var frameStamp = framebytes.ConvertByteToMat();
            string timestamp = DateTime.UtcNow.ToString("yyyy-MM-dd HH:mm:ss.fff",
                CultureInfo.InvariantCulture);
            Cv2.PutText(frameStamp, timestamp, new Point(1, 480 / 2), HersheyFonts.HersheySimplex, 0.5f, Scalar.Red);
            return frameStamp.ConvertToJpgByte();
        }

        public async Task<IEnumerable<User>> GetListOfTypeUser(ConnectionType t)
        {
            if (t == ConnectionType.Client) return manager.getClientSideUsers();
            if (t == ConnectionType.Server) return manager.getServerSideUsers();
            return manager.getAllUsers();
        }

        public async Task<IEnumerable<User>> GetAllUsers()
        {
            return manager.getAllUsers();
        }

        public async IAsyncEnumerable<User> GetListOfTypeUserAsync(ConnectionType t)
        {
            var client = await GetListOfTypeUser(t);
            foreach (var va in client) yield return va;
        }
    }
}