using Microsoft.AspNetCore.SignalR.Client; using Microsoft.Extensions.DependencyInjection; using Remotely_Shared.Models; using Remotely_ScreenCast.Core.Enums; using Remotely_ScreenCast.Core.Input; using Remotely_ScreenCast.Core.Models; using Remotely_ScreenCast.Core.Sockets; using Remotely_ScreenCast.Core.Utilities; using System; using System.Collections.Concurrent; using System.Collections.Generic; using System.Linq; using System.Text; using System.Threading.Tasks; namespace Remotely_ScreenCast.Core { public class Conductor { public event EventHandler ScreenCastInitiated; public event EventHandler ScreenCastRequested; public event EventHandler SessionIDChanged; public event EventHandler ViewerAdded; public event EventHandler ViewerRemoved; public Dictionary ArgDict { get; set; } public HubConnection Connection { get; private set; } public string CurrentDesktopName { get; set; } public string Host { get; private set; } public AppMode Mode { get; private set; } public CasterSocket CasterSocket { get; private set; } public string RequesterID { get; private set; } public string ServiceID { get; private set; } public ConcurrentDictionary Viewers { get; } = new ConcurrentDictionary(); public Task Connect() { Connection = new HubConnectionBuilder() .WithUrl($"{Host}/RCDeviceHub") .AddMessagePackProtocol() .Build(); return Connection.StartAsync(); } public void ProcessArgs(string[] args) { ArgDict = new Dictionary(); for (var i = 0; i < args.Length; i += 2) { var key = args?[i]; if (key != null) { key = key.Trim().Replace("-", "").ToLower(); var value = args?[i + 1]; if (value != null) { ArgDict[key] = args[i + 1].Trim(); } } } Mode = (AppMode)Enum.Parse(typeof(AppMode), ArgDict["mode"]); Host = ArgDict["host"]; if (Mode == AppMode.Unattended) { RequesterID = ArgDict["requester"]; CurrentDesktopName = ArgDict["desktop"]; ServiceID = ArgDict["serviceid"]; } } public void SetMessageHandlers(IKeyboardMouseInput keyboardMouse) { CasterSocket = new CasterSocket(Connection, this, keyboardMouse); } public void StartWaitForViewerTimer() { var timer = new System.Timers.Timer(10000); timer.AutoReset = false; timer.Elapsed += (sender, arg) => { // Shut down if no viewers have connected within 10 seconds. if (Viewers.Count == 0) { Logger.Write("No viewers connected after 10 seconds. Shutting down."); Environment.Exit(0); } }; timer.Start(); } internal void InvokeScreenCastInitiated(ScreenCastRequest viewerIdAndRequesterName) { ScreenCastInitiated?.Invoke(null, viewerIdAndRequesterName); } internal void InvokeScreenCastRequested(ScreenCastRequest viewerIdAndRequesterName) { ScreenCastRequested?.Invoke(null, viewerIdAndRequesterName); } internal void InvokeSessionIDChanged(string sessionID) { SessionIDChanged?.Invoke(null, sessionID); } internal void InvokeViewerAdded(Viewer viewer) { ViewerAdded?.Invoke(null, viewer); } internal void InvokeViewerRemoved(string viewerID) { ViewerRemoved?.Invoke(null, viewerID); } } }