Remotely/Server/Services/RcImplementations/HubEventHandler.cs

using Immense.RemoteControl.Server.Abstractions;
using Immense.RemoteControl.Server.Models;
using Immense.RemoteControl.Shared.Enums;
using Microsoft.AspNetCore.SignalR;
using Microsoft.Build.Framework;
using Microsoft.Extensions.Logging;
using NuGet.Protocol.Core.Types;
using Remotely.Server.Hubs;
using Remotely.Server.Models;
using Remotely.Shared.Enums;
using Remotely.Shared.Interfaces;
using Remotely.Shared.Models;
using System;
using System.Collections.Concurrent;
using System.Collections.Generic;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;

namespace Remotely.Server.Services.RcImplementations;

public class HubEventHandler : IHubEventHandler
{
    private readonly IHubContext<AgentHub, IAgentHubClient> _serviceHub;
    private readonly ILogger<HubEventHandler> _logger;

    public HubEventHandler(
        IHubContext<AgentHub, IAgentHubClient> serviceHub,
        ILogger<HubEventHandler> logger)
    {
        _serviceHub = serviceHub;
        _logger = logger;
    }

    public Task ChangeWindowsSession(RemoteControlSession session, string viewerConnectionId, int targetWindowsSession)
    {
        if (session is not RemoteControlSessionEx ex)
        {
            _logger.LogError("Event should have been for RemoteControlSessionEx.");
            return Task.CompletedTask;
        }   

        return _serviceHub.Clients
            .Client(ex.AgentConnectionId)
            .ChangeWindowsSession(
                viewerConnectionId,
                $"{ex.UnattendedSessionId}",
                ex.AccessKey,
                ex.UserConnectionId,
                ex.RequesterUserName,
                ex.OrganizationName,
                ex.OrganizationId,
                targetWindowsSession);
    }

    public Task InvokeCtrlAltDel(RemoteControlSession session, string viewerConnectionId)
    {
        if (session is not RemoteControlSessionEx ex)
        {
            _logger.LogError("Event should have been for RemoteControlSessionEx.");
            return Task.CompletedTask;
        }

        return _serviceHub.Clients.Client(ex.AgentConnectionId).InvokeCtrlAltDel();
    }

    public Task NotifyDesktopSessionAdded(RemoteControlSession sessionInfo)
    {
        return Task.CompletedTask;
    }

    public Task NotifyDesktopSessionDisposed(RemoteControlSession sessionInfo)
    {
        return Task.CompletedTask;
    }

    public Task NotifyDesktopSessionRemoved(RemoteControlSession sessionInfo)
    {
        return Task.CompletedTask;
    }

    public Task NotifyRemoteControlEnded(RemoteControlSession sessionInfo)
    {
        return Task.CompletedTask;
    }

    public Task NotifyRemoteControlStarted(RemoteControlSession sessionInfo)
    {
        return Task.CompletedTask;
    }

    public Task NotifySessionChanged(RemoteControlSession session, SessionSwitchReasonEx reason, int currentSessionId)
    {
        if (session is not RemoteControlSessionEx ex)
        {
            _logger.LogError("Event should have been for RemoteControlSessionEx.");
            return Task.CompletedTask;
        }

        if (ex.RequireConsent)
        {
            // Don't restart if consent wasn't granted on the first request.
            return Task.CompletedTask;
        }

        _logger.LogDebug("Windows session changed during remote control.  " +
            "Reason: {reason}.  " +
            "Current Session ID: {sessionId}.  " +
            "Session Info: {@sessionInfo}",
            reason,
            currentSessionId,
            session);

        return _serviceHub.Clients
            .Client(ex.AgentConnectionId)
            .RestartScreenCaster(
                ex.ViewerList.ToArray(),
                $"{ex.UnattendedSessionId}",
                ex.AccessKey,
                ex.UserConnectionId,
                ex.RequesterUserName,
                ex.OrganizationName,
                ex.OrganizationId);
    }

    public Task RestartScreenCaster(RemoteControlSession session, HashSet<string> viewerList)
    {

        if (session is not RemoteControlSessionEx ex)
        {
            _logger.LogError("Event should have been for RemoteControlSessionEx.");
            return Task.CompletedTask;
        }

        if (ex.RequireConsent)
        {
            // Don't restart if consent wasn't granted on the first request.
            return Task.CompletedTask;
        }

        return _serviceHub.Clients
                 .Client(ex.AgentConnectionId)
                 .RestartScreenCaster(
                        viewerList.ToArray(),
                        $"{ex.UnattendedSessionId}",
                        ex.AccessKey,
                        ex.UserConnectionId,
                        ex.RequesterName,
                        ex.OrganizationName,
                        ex.OrganizationId);
    }
}