fix: company software check all platforms and non-admin

EasyGUI/MainWindow.xaml

@@ -4,6 +4,7 @@
         xmlns:d="http://schemas.microsoft.com/expression/blend/2008"
         xmlns:mc="http://schemas.openxmlformats.org/markup-compatibility/2006"
         xmlns:controls="clr-namespace:EasyGUI.Controls"
+        Closing="MainWindow_OnClosing"
         mc:Ignorable="d"
         KeyDown="MainWindow_OnKeyDown"
         Title="EasySave" Height="720" Width="1280" Icon="/Assets/easysave-logo.ico">

EasyGUI/MainWindow.xaml.cs

@@ -1,4 +1,5 @@
 using System.Collections.ObjectModel;
+using System.ComponentModel;
 using System.Windows;
 using System.Windows.Input;
 using EasyGUI.Events;
@@ -173,4 +174,10 @@ private void SettingsPopup_OnReloadConfig(object? sender, EventArgs e)
     {
         _jobManager.ReloadConfig();
     }
+
+    private void MainWindow_OnClosing(object? sender, CancelEventArgs e)
+    {
+        // destroy job manager
+        _jobManager.StopCheck();
+    }
 }

EasyLib/Events/ProcessStartEvent.cs

@@ -1,68 +1,109 @@
-using System.Management;
+using System.Diagnostics;
+using System.Management;
+using System.Security.Principal;
 
 namespace EasyLib.Events;
 
 public class ProcessStartEvent
 {
+    public static long NumberOfRunningProcesses;
     private readonly JobManager _jobManager;
-    private readonly ManagementEventWatcher _processStartEvent;
-    private readonly ManagementEventWatcher _processStopEvent;
-    private int _numberOfRunningProcesses;
+    private readonly ManagementEventWatcher? _processStartEvent;
+    private readonly ManagementEventWatcher? _processStopEvent;
+    private CancellationTokenSource _cancellationTokenSource;
 
     /// <summary>
     /// create the start and stop event for a process and bind the event to the job manager
     /// </summary>
     /// <param name="processName"></param>
     /// <param name="jobManager"></param>
     public ProcessStartEvent(string? processName, JobManager jobManager)
     {
         // check if system is running Windows
         _jobManager = jobManager;
+        if ((Environment.OSVersion.Platform == PlatformID.Win32NT) &&
+            IsAdministrator) // only on Windows can run this feature
+        {
 #pragma warning disable CA1416
-        _processStartEvent =
-            new ManagementEventWatcher(
-                new WqlEventQuery($"SELECT * FROM Win32_ProcessStartTrace WHERE ProcessName = '{processName}'"));
-        _processStartEvent.EventArrived += ProcessStarted;
-        // on stop
-        _processStopEvent = new ManagementEventWatcher(new WqlEventQuery(
-            $"SELECT * FROM Win32_ProcessStopTrace WHERE ProcessName = '{processName}'"));
-        _processStopEvent.EventArrived += ProcessStopped;
-        _processStartEvent.Start();
-        _processStopEvent.Start();
-#pragma warning restore CA1416
+            _processStartEvent =
+                new ManagementEventWatcher(
+                    new WqlEventQuery($"SELECT * FROM Win32_ProcessStartTrace WHERE ProcessName = '{processName}'"));
+            _processStartEvent.EventArrived += ProcessStarted;
+            // on stop
+            _processStopEvent = new ManagementEventWatcher(new WqlEventQuery(
+                $"SELECT * FROM Win32_ProcessStopTrace WHERE ProcessName = '{processName}'"));
+            _processStopEvent.EventArrived += ProcessStopped;
+            _processStartEvent.Start();
+            _processStopEvent.Start();
+#pragma warning disable CA1416
+        }
+        else
+        {
+            _cancellationTokenSource = new CancellationTokenSource();
+            var newThread = new Thread(() => _checkProcesses(processName));
+            newThread.Start();
+        }
     }
 
+    public static bool IsAdministrator =>
+        new WindowsPrincipal(WindowsIdentity.GetCurrent()).IsInRole(WindowsBuiltInRole.Administrator);
+
     private void ProcessStarted(object sender, EventArrivedEventArgs e)
     {
-        if (_numberOfRunningProcesses == 0)
+        if (NumberOfRunningProcesses == 0)
         {
             _jobManager.PauseAllJobs();
-            Interlocked.Increment(ref _numberOfRunningProcesses);
+            Interlocked.Increment(ref NumberOfRunningProcesses);
         }
         else
         {
-            Interlocked.Increment(ref _numberOfRunningProcesses);
+            Interlocked.Increment(ref NumberOfRunningProcesses);
         }
     }
 
     private void ProcessStopped(object sender, EventArrivedEventArgs e)
     {
-        if (_numberOfRunningProcesses == 1)
+        if (NumberOfRunningProcesses == 1)
         {
             _jobManager.ResumeAllJobs();
-            Interlocked.Decrement(ref _numberOfRunningProcesses);
+            Interlocked.Decrement(ref NumberOfRunningProcesses);
         }
         else
         {
-            Interlocked.Decrement(ref _numberOfRunningProcesses);
+            Interlocked.Decrement(ref NumberOfRunningProcesses);
         }
     }
 
     public void Stop()
     {
 #pragma warning disable CA1416
-        _processStopEvent.Stop();
-        _processStartEvent.Stop();
+        if (_processStartEvent != null && _processStopEvent != null)
+        {
+            _processStopEvent.Stop();
+            _processStartEvent.Stop();
+        }
+        else
+        {
+            _cancellationTokenSource.Cancel();
+        }
 #pragma warning restore CA1416
     }
+
+    private void _checkProcesses(string processName)
+    {
+        while (true)
+        {
+            if (Process.GetProcessesByName(processName).Length > Interlocked.Read(ref NumberOfRunningProcesses))
+            {
+                ProcessStarted(null, null);
+            }
+            else if (Process.GetProcessesByName(processName).Length < Interlocked.Read(ref NumberOfRunningProcesses))
+            {
+                ProcessStopped(null, null);
+            }
+
+            Thread.Sleep(100);
+            if (_cancellationTokenSource.IsCancellationRequested) return;
+        }
+    }
 }

EasyLib/JobManager.cs

@@ -81,6 +81,11 @@ public void OnJobStateChange(JobState state, Job.Job job)
         StateManager.Instance.WriteJobs(_jobs);
     }
 
+    public void StopCheck()
+    {
+        _pauseJobEvent.Stop();
+    }
+
     [GeneratedRegex(@"^\d+-\d+$", RegexOptions.NonBacktracking)]
     private static partial Regex JobIndexRangeRegex();
 
@@ -362,15 +367,9 @@ public void ResumeAllJobs()
         }
     }
 
-    public bool ReloadConfig()
+    public void ReloadConfig()
     {
-        if (Environment.OSVersion.Platform == PlatformID.Win32NT) // only on Windows can run this feature
-        {
-            _pauseJobEvent.Stop();
-            _pauseJobEvent = new ProcessStartEvent(ConfigManager.Instance.CompanySoftwareProcessPath, this);
-            return true;
-        }
-
-        return false;
+        _pauseJobEvent.Stop();
+        _pauseJobEvent = new ProcessStartEvent(ConfigManager.Instance.CompanySoftwareProcessPath, this);
     }
 }