Remotely/Remotely_ScreenCast/Program.cs
2021-07-29 07:53:41 -07:00

122 lines
3.8 KiB
C#

using Microsoft.AspNetCore.SignalR.Client;
using Remotely_ScreenCapture;
using Remotely_ScreenCapture.Capture;
using Remotely_ScreenCapture.Sockets;
using Remotely_ScreenCapture.Utilities;
using Remotely_ScreenCast.Capture;
using System;
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 ICapturer Capturer { get; set; }
public static CaptureMode CaptureMode { get; set; }
public static bool DisconnectRequested { get; set; }
public static string Mode { get; set; }
public static string RequesterID { get; set; }
public static string HostName { get; set; }
public static HubConnection Connection { get; set; }
private static OutgoingMessages OutgoingMessages { get; set; }
static void Main(string[] args)
{
var argDict = ProcessArgs(args);
Mode = argDict["mode"];
RequesterID = argDict["requester"];
HostName = argDict["hostname"];
Connection = new HubConnectionBuilder()
.WithUrl($"http://{HostName}/RCDeviceHub")
.Build();
MessageHandlers.ApplyConnectionHandlers(Connection);
Connection.StartAsync().Wait();
OutgoingMessages = new OutgoingMessages(Connection);
try
{
Capturer = new DXCapture();
CaptureMode = CaptureMode.DirectX;
}
catch
{
Capturer = new BitBltCapture();
CaptureMode = CaptureMode.BitBtl;
}
OutgoingMessages.SendScreenCount(
Screen.AllScreens.ToList().IndexOf(Screen.PrimaryScreen),
Screen.AllScreens.Length,
RequesterID).Wait();
Capturer.ScreenChanged += HandleScreenChanged;
OutgoingMessages.SendScreenSize(Capturer.CurrentScreenSize.Width, Capturer.CurrentScreenSize.Height).Wait();
while (!DisconnectRequested)
{
try
{
Capturer.Capture();
var newImage = ImageDiff.GetImageDiff(Capturer.CurrentFrame, Capturer.PreviousFrame, Capturer.CaptureFullscreen);
var img = ImageDiff.EncodeBitmap(newImage);
if (Capturer.CaptureFullscreen)
{
Capturer.CaptureFullscreen = false;
}
if (img?.Length > 0)
{
OutgoingMessages.SendScreenCapture(img).Wait();
}
}
catch (Exception ex)
{
Logger.Write($"Outer Error: {ex.Message}{Environment.NewLine}{ex.StackTrace}");
}
}
}
private static async void HandleScreenChanged(object sender, Size size)
{
await OutgoingMessages.SendScreenSize(size.Width, size.Height);
}
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;
}
}
}