McpEditorShutdownCleanup.cs

using System;
using System.Threading.Tasks;
using MCPForUnity.Editor.Constants;
using MCPForUnity.Editor.Helpers;
using MCPForUnity.Editor.Services.Transport;
using UnityEditor;

namespace MCPForUnity.Editor.Services
{
    /// <summary>
    /// Best-effort cleanup when the Unity Editor is quitting.
    /// - Stops active transports so clients don't see a "hung" session longer than necessary.
    /// - If HTTP Local is selected, attempts to stop the local HTTP server (guarded by PID heuristics).
    /// </summary>
    [InitializeOnLoad]
    internal static class McpEditorShutdownCleanup
    {
        static McpEditorShutdownCleanup()
        {
            // Guard against duplicate subscriptions across domain reloads.
            try { EditorApplication.quitting -= OnEditorQuitting; } catch { }
            EditorApplication.quitting += OnEditorQuitting;
        }

        private static void OnEditorQuitting()
        {
            // 1) Stop transports (best-effort, bounded wait).
            try
            {
                var transport = MCPServiceLocator.TransportManager;

                Task stopHttp = transport.StopAsync(TransportMode.Http);
                Task stopStdio = transport.StopAsync(TransportMode.Stdio);

                try { Task.WaitAll(new[] { stopHttp, stopStdio }, 750); } catch { }
            }
            catch (Exception ex)
            {
                // Avoid hard failures on quit.
                McpLog.Warn($"Shutdown cleanup: failed to stop transports: {ex.Message}");
            }

            // 2) Stop local HTTP server if it was Unity-managed (best-effort).
            try
            {
                bool useHttp = EditorConfigurationCache.Instance.UseHttpTransport;
                string scope = string.Empty;
                try { scope = EditorPrefs.GetString(EditorPrefKeys.HttpTransportScope, string.Empty); } catch { }

                bool stopped = false;
                bool httpLocalSelected =
                    useHttp &&
                    (string.Equals(scope, "local", StringComparison.OrdinalIgnoreCase)
                     || string.Equals(scope, "lan", StringComparison.OrdinalIgnoreCase)
                     || (string.IsNullOrEmpty(scope) && MCPServiceLocator.Server.IsLocalUrl()));

                if (httpLocalSelected)
                {
                    // StopLocalHttpServer is already guarded to only terminate processes that look like mcp-for-unity.
                    // If it refuses to stop (e.g. URL was edited away from local), fall back to the Unity-managed stop.
                    stopped = MCPServiceLocator.Server.StopLocalHttpServer();
                }

                // Always attempt to stop a Unity-managed server if one exists.
                // This covers cases where the user switched transports (e.g. to stdio) or StopLocalHttpServer refused.
                if (!stopped)
                {
                    MCPServiceLocator.Server.StopManagedLocalHttpServer();
                }
            }
            catch (Exception ex)
            {
                McpLog.Warn($"Shutdown cleanup: failed to stop local HTTP server: {ex.Message}");
            }
        }
    }
}