处理好可执行程序的查找逻辑

Author: walterlv

src/DotNetCampus.ModelContextProtocol/Transports/Stdio/StdioClientTransport.cs

@@ -169,11 +169,6 @@ private async Task RunLoopAsync(StdioProcessInfo stdio, CancellationToken cancel
             };
         }
 
-        /// <summary>
-        /// 跨平台解析命令路径。<br/>
-        /// 如果命令是完整路径或相对路径，直接返回；<br/>
-        /// 否则在 PATH 环境变量中查找命令。
-        /// </summary>
         static string? ResolveCommandPath(string command)
         {
             // 如果命令包含路径分隔符，说明是路径而非命令名
@@ -189,48 +184,40 @@ private async Task RunLoopAsync(StdioProcessInfo stdio, CancellationToken cancel
                 return null;
             }
 
+            // 尝试提取命令里原本的扩展名（例如用户可能指定的是 npx/dnx 也可能指定的是 npx.cmd/dnx.exe。
+            var extensionInCommand = Path.GetExtension(command);
             var paths = pathEnv.Split(Path.PathSeparator, StringSplitOptions.RemoveEmptyEntries);
-            var extensions = GetExecutableExtensions();
+            var extensions = ExecutableExtensionsLazy.Value;
 
             foreach (var path in paths)
             {
-                // 在 Windows 上，尝试命令名 + 各种扩展名
-                foreach (var extension in extensions)
+                if (extensions.Count is 0)
                 {
-                    var fullPath = Path.Combine(path, command + extension);
+                    // Linux 等无扩展名的可执行程序。
+                    var fullPath = Path.Join(path, command);
                     if (File.Exists(fullPath))
                     {
                         return fullPath;
                     }
+                    continue;
                 }
-            }
-
-            return null;
-        }
 
-        /// <summary>
-        /// 获取可执行文件扩展名列表（跨平台）。
-        /// </summary>
-        static string[] GetExecutableExtensions()
-        {
-            if (OperatingSystem.IsWindows())
-            {
-                // Windows 上从 PATHEXT 环境变量获取可执行扩展名
-                var pathExt = Environment.GetEnvironmentVariable("PATHEXT");
-                if (!string.IsNullOrEmpty(pathExt))
+                // Windows 等带扩展名的可执行程序。
+                foreach (var extension in extensions)
                 {
-                    var extensions = pathExt.Split(Path.PathSeparator, StringSplitOptions.RemoveEmptyEntries);
-                    // 确保包含空扩展名（用于无扩展名的可执行文件）
-                    return [.. extensions];
+                    var fullPath = extensionInCommand?.Equals(extension, StringComparison.OrdinalIgnoreCase) is true
+                        // 如果命令自带的扩展名正好与环境变量里的相同，说明命令本身确实带的是扩展名。
+                        ? Path.Join(path, command)
+                        // 否则，叠加环境变量里的扩展名。
+                        : Path.Join(path, command + extension);
+                    if (File.Exists(fullPath))
+                    {
+                        return fullPath;
+                    }
                 }
-                // 默认 Windows 可执行扩展名
-                return [".exe", ".cmd", ".bat", ".com"];
-            }
-            else
-            {
-                // Unix 系统上可执行文件通常没有扩展名
-                return [""];
             }
+
+            return null;
         }
     }
 
@@ -249,6 +236,25 @@ await Task.Run(() =>
         });
     }
 
+    private static readonly Lazy<IReadOnlyList<string>> ExecutableExtensionsLazy = new Lazy<IReadOnlyList<string>>(() =>
+    {
+        if (!OperatingSystem.IsWindows())
+        {
+            // Unix 系统上可执行文件通常没有扩展名
+            return [""];
+        }
+        // Windows 上从 PATHEXT 环境变量获取可执行扩展名
+        var pathExt = Environment.GetEnvironmentVariable("PATHEXT");
+        if (string.IsNullOrEmpty(pathExt))
+        {
+            return [".exe", ".cmd", ".bat", ".com"];
+        }
+        var extensions = pathExt.Split(Path.PathSeparator, StringSplitOptions.RemoveEmptyEntries);
+        // 确保包含空扩展名（用于无扩展名的可执行文件）
+        return extensions;
+        // 默认 Windows 可执行扩展名
+    }, LazyThreadSafetyMode.None);
+
     private readonly record struct StdioProcessInfo
     {
         public required Process Process { get; init; }