78 lines
3.1 KiB
C#
78 lines
3.1 KiB
C#
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 = EditorPrefs.GetBool(EditorPrefKeys.UseHttpTransport, true);
|
|
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.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}");
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|