Rename project and namespaces

src/ModelContextProtocol/Utils/ProcessHelper.cs

@@ -0,0 +1,108 @@
+using System.Diagnostics;
+using System.Runtime.InteropServices;
+
+namespace ModelContextProtocol.Utils;
+
+/// <summary>
+/// Helper class for working with processes.
+/// </summary>
+internal static class ProcessHelper
+{
+    private static readonly bool _isWindows = RuntimeInformation.IsOSPlatform(OSPlatform.Windows);
+    private static readonly TimeSpan _defaultTimeout = TimeSpan.FromSeconds(30);
+
+    public static void KillTree(this Process process) => process.KillTree(_defaultTimeout);
+
+    public static void KillTree(this Process process, TimeSpan timeout)
+    {
+        var pid = process.Id;
+        if (_isWindows)
+        {
+            RunProcessAndWaitForExit(
+                "taskkill",
+                $"/T /F /PID {pid}",
+                timeout,
+                out var _);
+        }
+        else
+        {
+            var children = new HashSet<int>();
+            GetAllChildIdsUnix(pid, children, timeout);
+            foreach (var childId in children)
+            {
+                KillProcessUnix(childId, timeout);
+            }
+            KillProcessUnix(pid, timeout);
+        }
+
+        // wait until the process finishes exiting/getting killed. 
+        // We don't want to wait forever here because the task is already supposed to be dieing, we just want to give it long enough
+        // to try and flush what it can and stop. If it cannot do that in a reasonable time frame then we will just ignore it.
+        process.WaitForExit((int)timeout.TotalMilliseconds);
+    }
+
+    private static void GetAllChildIdsUnix(int parentId, ISet<int> children, TimeSpan timeout)
+    {
+        var exitcode = RunProcessAndWaitForExit(
+            "pgrep",
+            $"-P {parentId}",
+            timeout,
+            out var stdout);
+
+        if (exitcode == 0 && !string.IsNullOrEmpty(stdout))
+        {
+            using var reader = new StringReader(stdout);
+            while (true)
+            {
+                var text = reader.ReadLine();
+                if (text == null)
+                    return;
+
+                if (int.TryParse(text, out var id))
+                {
+                    children.Add(id);
+                    // Recursively get the children
+                    GetAllChildIdsUnix(id, children, timeout);
+                }
+            }
+        }
+    }
+
+    private static void KillProcessUnix(int processId, TimeSpan timeout)
+    {
+        RunProcessAndWaitForExit(
+            "kill",
+            $"-TERM {processId}",
+            timeout,
+            out var _);
+    }
+
+    private static int RunProcessAndWaitForExit(string fileName, string arguments, TimeSpan timeout, out string? stdout)
+    {
+        var startInfo = new ProcessStartInfo
+        {
+            FileName = fileName,
+            Arguments = arguments,
+            RedirectStandardOutput = true,
+            RedirectStandardError = true,
+            UseShellExecute = false,
+        };
+
+        stdout = null;
+
+        var process = Process.Start(startInfo);
+        if (process == null)
+            return -1;
+
+        if (process.WaitForExit((int)timeout.TotalMilliseconds))
+        {
+            stdout = process.StandardOutput.ReadToEnd();
+        }
+        else
+        {
+            process.Kill();
+        }
+
+        return process.ExitCode;
+    }
+}