start stop icue server with icue.exe with process monitoring

Author: Aytackydln

Project-Aurora/Project-Aurora/Modules/Icue/AuroraIcueServer.cs

@@ -25,25 +25,28 @@ public sealed class AuroraIcueServer : IDisposable, IAsyncDisposable
 
     public event EventHandler? StatusChanged;
 
-    private IcueServerStatus _serverStatus = IcueServerStatus.Disabled;
-
     public IcueServerStatus ServerStatus
     {
-        get => _serverStatus;
+        get;
         private set
         {
-            if (_serverStatus == value) return;
-            _serverStatus = value;
+            if (field == value) return;
+            field = value;
             StatusChanged?.Invoke(this, EventArgs.Empty);
         }
-    }
+    } = IcueServerStatus.Disabled;
 
     private IcueServer? _icueServer;
     private GameHandler? _gameHandler;
+    private RunningProcessMonitor? _processMonitor;
 
     public void RunServer(RunningProcessMonitor processMonitor)
     {
-        Sdk.RunningProcessMonitor = processMonitor;
+        _processMonitor = processMonitor;
+        
+        processMonitor.ProcessStarted += ProcessMonitorOnProcessStarted;
+        processMonitor.ProcessStopped += ProcessMonitorOnProcessStopped;
+        
         var runApproved = IcueInstallationUtils.IsIcueInstalled() && IcueInstallationUtils.IsIcueAutorunEnabled();
         if (runApproved || processMonitor.IsProcessRunning(IcueInstallationUtils.IcueExe))
         {
@@ -54,6 +57,58 @@ public void RunServer(RunningProcessMonitor processMonitor)
         TryStartServer();
     }
 
+    private void ProcessMonitorOnProcessStarted(object? sender, ProcessStarted e)
+    {
+        if (e.ProcessName != IcueInstallationUtils.IcueExe)
+        {
+            return;
+        }
+
+        switch (ServerStatus)
+        {
+            case IcueServerStatus.Conflicted:
+            case IcueServerStatus.Disabled:
+            case IcueServerStatus.PipesInUse:
+                return;
+            case IcueServerStatus.Waiting:
+                // proceed to close the server
+                break;
+        }
+        ServerStatus = IcueServerStatus.Conflicted;
+            
+        if (_icueServer == null) return;
+        _icueServer.GameConnected -= OnGameConnected;
+        _icueServer.Dispose();
+        _icueServer = null;
+    }
+
+    private void ProcessMonitorOnProcessStopped(object? sender, ProcessStopped e)
+    {
+        if (e.ProcessName != IcueInstallationUtils.IcueExe)
+        {
+            return;
+        }
+        
+        switch (ServerStatus)
+        {
+            case IcueServerStatus.Waiting:
+            case IcueServerStatus.Disabled:
+                return;
+            case IcueServerStatus.Conflicted:
+            case IcueServerStatus.PipesInUse:
+                // Aurora's turn to use the pipes
+                break;
+        }
+
+        var runApproved = IcueInstallationUtils.IsIcueInstalled() && IcueInstallationUtils.IsIcueAutorunEnabled();
+        if (runApproved)
+        {
+            ServerStatus = IcueServerStatus.Conflicted;
+        }
+
+        TryStartServer();
+    }
+
     private void TryStartServer()
     {
         try
@@ -118,23 +173,43 @@ private void OnGameDisconnected(object? sender, EventArgs e)
         Sdk.ClearSdkHandler();
     }
 
-    public void Dispose()
+    public void Stop()
     {
+        if (_processMonitor != null)
+        {
+            _processMonitor.ProcessStarted -= ProcessMonitorOnProcessStarted;
+            _processMonitor.ProcessStopped -= ProcessMonitorOnProcessStopped;
+        }
+ 
         ServerStatus = IcueServerStatus.Disabled;
         if (_icueServer == null) return;
         _icueServer.GameConnected -= OnGameConnected;
         _icueServer.Dispose();
+        _icueServer = null;
     }
 
-    public async ValueTask DisposeAsync()
+    public async Task StopAsync()
     {
-        ServerStatus = IcueServerStatus.Disabled;
-        if (_icueServer == null)
+        if (_processMonitor != null)
         {
-            return;
+            _processMonitor.ProcessStarted -= ProcessMonitorOnProcessStarted;
+            _processMonitor.ProcessStopped -= ProcessMonitorOnProcessStopped;
         }
 
+        ServerStatus = IcueServerStatus.Disabled;
+        if (_icueServer == null) return;
         _icueServer.GameConnected -= OnGameConnected;
         await _icueServer.DisposeAsync();
+        _icueServer = null;
+    }
+
+    public void Dispose()
+    {
+        Stop();
+    }
+
+    public async ValueTask DisposeAsync()
+    {
+        await StopAsync();
     }
 }

Project-Aurora/Project-Aurora/Modules/Icue/IcueSdk.cs

@@ -18,8 +18,6 @@ public class IcueSdk
 
     private SdkHandler? _sdkHandler;
 
-    public RunningProcessMonitor? RunningProcessMonitor { get; set; }
-
     public void SetSdkHandler(SdkHandler gameHandlerSdkHandler, int gamePid)
     {
         IsGameConnected = true;

Project-Aurora/Project-Aurora/Modules/IcueModule.cs

@@ -39,7 +39,7 @@ private void ConfigurationOnPropertyChanged(object? sender, PropertyChangedEvent
         }
         else
         {
-            AuroraIcueServer.Dispose();
+            AuroraIcueServer.Stop();
         }
     }
 

Project-Aurora/Project-Aurora/Project-Aurora.csproj

@@ -73,7 +73,7 @@
           <PrivateAssets>all</PrivateAssets>
           <IncludeAssets>runtime; build; native; contentfiles; analyzers; buildtransitive</IncludeAssets>
         </PackageReference>
-        <PackageReference Include="iCUE-ReverseEngineeredServer" Version="0.2.9" />
+        <PackageReference Include="iCUE-ReverseEngineeredServer" Version="0.3.0" />
         <PackageReference Include="JetBrains.Annotations" Version="2024.3.0" />
         <PackageReference Include="JsonSubTypes" Version="2.0.1" />
         <PackageReference Include="LibreHardwareMonitorLib" Version="0.9.7-pre642" />

Project-Aurora/Project-Aurora/packages.lock.json

@@ -96,9 +96,9 @@
       },
       "iCUE-ReverseEngineeredServer": {
         "type": "Direct",
-        "requested": "[0.2.9, )",
-        "resolved": "0.2.9",
-        "contentHash": "PYiz6DE/z3W+bqqjpArxWsVCeOIbKEqslyxggjC627weEhlDR0SfrAVP31dAC7IEb/ZXx0sW0ztKJlEmu87hkg=="
+        "requested": "[0.3.0, )",
+        "resolved": "0.3.0",
+        "contentHash": "xHuJMKdfMqWBjWBk1P/tZw0YavrY6nCYvwxpRQRhoRvEBt126+9tKKGWGJ6w51Zq5AGpIlwg6r9Pl7kHopxKXw=="
       },
       "JetBrains.Annotations": {
         "type": "Direct",