Remotely/Remotely_ScreenCast/Program.cs

using Microsoft.AspNetCore.SignalR.Client;
using Microsoft.Extensions.DependencyInjection;
using Remotely_ScreenCast;
using Remotely_ScreenCast.Capture;
using Remotely_ScreenCast.Enums;
using Remotely_ScreenCast.Models;
using Remotely_ScreenCast.Sockets;
using Remotely_ScreenCast.Utilities;
using System;
using System.Collections.Concurrent;
using System.Collections.Generic;
using System.Diagnostics;
using System.Drawing;
using System.IO;
using System.Linq;
using System.Reflection;
using System.Text;
using System.Threading.Tasks;
using System.Windows.Forms;
using Win32;

namespace Remotely_ScreenCast
{
	public class Program
	{
        public static AppMode Mode { get; private set; }
        public static string RequesterID { get; private set; }
        public static string ServiceID { get; private set; }
        public static string Host { get; private set; }
        public static HubConnection Connection { get; private set; }
        public static OutgoingMessages OutgoingMessages { get; private set; }
        public static string CurrentDesktopName { get; set; }
        public static ConcurrentDictionary<string, Viewer> Viewers { get; } = new ConcurrentDictionary<string, Viewer>();

        static void Main(string[] args)
        {
            try
            {
                AppDomain.CurrentDomain.UnhandledException += CurrentDomain_UnhandledException;

                var argDict = ProcessArgs(args);
                Mode = (AppMode)Enum.Parse(typeof(AppMode), argDict["mode"]);
                RequesterID = argDict["requester"];
                Host = argDict["host"];
                CurrentDesktopName = argDict["desktop"];
                ServiceID = argDict["serviceid"];

                Connection = new HubConnectionBuilder()
                    .WithUrl($"{Host}/RCDeviceHub")
                    .AddMessagePackProtocol()
                    .Build();

                Connection.StartAsync().Wait();

                OutgoingMessages = new OutgoingMessages(Connection);

                OutgoingMessages.SendServiceID(ServiceID).Wait();

                var desktopName = Win32Interop.GetCurrentDesktop();
                if (desktopName.ToLower() != CurrentDesktopName.ToLower())
                {
                    CurrentDesktopName = desktopName;
                    Logger.Write($"Relaunching in desktop {desktopName}.");
                    argDict["desktop"] = desktopName;
                    var openProcessString = Assembly.GetExecutingAssembly().Location;
                    foreach (var arg in argDict)
                    {
                        openProcessString += $" -{arg.Key} {arg.Value}";
                    }
                    var result = Win32Interop.OpenInteractiveProcess(openProcessString, desktopName, true, out _);
                    if (!result)
                    {
                        Logger.Write($"Desktop relaunch to {desktopName} failed.");
                    }
                    Environment.Exit(0);
                }

                MessageHandlers.ApplyConnectionHandlers(Connection, OutgoingMessages);

                CursorIconWatcher.Current.OnChange += CursorIconWatcher_OnChange;

                if (argDict.ContainsKey("relaunch"))
                {
                    var viewersString = argDict["viewers"];
                    var viewerIDs = viewersString.Split(",".ToCharArray());
                    OutgoingMessages.NotifyViewersRelaunchedScreenCasterReady(viewerIDs).Wait();
                }
                else
                {
                    OutgoingMessages.NotifyRequesterUnattendedReady(RequesterID).Wait();
                }

                StartWaitForViewerTimer();

                while (true)
                {
                    if (Mode == AppMode.Unattended)
                    {
                        desktopName = Win32Interop.GetCurrentDesktop();
                        if (desktopName.ToLower() != CurrentDesktopName.ToLower() && Viewers.Count > 0)
                        {
                            CurrentDesktopName = desktopName;
                            Logger.Write($"Switching desktops to {desktopName}.");
                            // TODO: SetThradDesktop causes issues with input after switching.
                            //var inputDesktop = Win32Interop.OpenInputDesktop();
                            //User32.SetThreadDesktop(inputDesktop);
                            //User32.CloseDesktop(inputDesktop);
                            Connection.InvokeAsync("SwitchingDesktops", Viewers.Keys.ToList()).Wait();
                            var result = Win32Interop.OpenInteractiveProcess(Assembly.GetExecutingAssembly().Location + $" -mode {Mode.ToString()} -requester {RequesterID} -serviceid {ServiceID} -host {Host} -relaunch true -desktop {desktopName} -viewers {String.Join(",", Viewers.Keys.ToList())}", desktopName, true, out _);
                            if (!result)
                            {
                                Logger.Write($"Desktop switch to {desktopName} failed.");
                                OutgoingMessages.SendConnectionFailedToViewers(Viewers.Keys.ToList()).Wait();
                            }
                        }
                        System.Threading.Thread.Sleep(100);
                    }
                    else
                    {
                        Console.Read();
                    }
                }
            }
            catch (Exception ex)
            {
                Logger.Write(ex);
            }
        }

        private static async void CursorIconWatcher_OnChange(object sender, string cursor)
        {
            await OutgoingMessages.SendCursorChange(cursor, Viewers.Keys.ToList());
        }

        private static 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();
        }

        private static void CurrentDomain_UnhandledException(object sender, UnhandledExceptionEventArgs e)
        {
            Logger.Write((Exception)e.ExceptionObject);
        }

        private static Dictionary<string, string> ProcessArgs(string[] args)
        {
            var argDict = new Dictionary<string, string>();

            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();
                    }
                }

            }
            return argDict;
        }

    
    }
}