@mcp-z/client 1.0.0

package/dist/cjs/spawn/spawn-servers.js

@@ -0,0 +1,911 @@
+/**
+ * High-level multi-server registry management.
+ * Starts multiple servers from a servers configuration object.
+ * Supports stdio, http, and ws transports.
+ * Implements Claude Code-compatible configuration with start extension support.
+ */ "use strict";
+Object.defineProperty(exports, "__esModule", {
+    value: true
+});
+Object.defineProperty(exports, "createServerRegistry", {
+    enumerable: true,
+    get: function() {
+        return createServerRegistry;
+    }
+});
+var _fs = /*#__PURE__*/ _interop_require_wildcard(require("fs"));
+var _process = /*#__PURE__*/ _interop_require_wildcard(require("process"));
+var _clienthelpersts = require("../client-helpers.js");
+var _validateconfigts = require("../config/validate-config.js");
+var _connectclientts = require("../connection/connect-client.js");
+var _indexts = require("../search/index.js");
+var _loggerts = require("../utils/logger.js");
+var _spawnserverts = require("./spawn-server.js");
+function _array_like_to_array(arr, len) {
+    if (len == null || len > arr.length) len = arr.length;
+    for(var i = 0, arr2 = new Array(len); i < len; i++)arr2[i] = arr[i];
+    return arr2;
+}
+function _array_with_holes(arr) {
+    if (Array.isArray(arr)) return arr;
+}
+function asyncGeneratorStep(gen, resolve, reject, _next, _throw, key, arg) {
+    try {
+        var info = gen[key](arg);
+        var value = info.value;
+    } catch (error) {
+        reject(error);
+        return;
+    }
+    if (info.done) {
+        resolve(value);
+    } else {
+        Promise.resolve(value).then(_next, _throw);
+    }
+}
+function _async_to_generator(fn) {
+    return function() {
+        var self = this, args = arguments;
+        return new Promise(function(resolve, reject) {
+            var gen = fn.apply(self, args);
+            function _next(value) {
+                asyncGeneratorStep(gen, resolve, reject, _next, _throw, "next", value);
+            }
+            function _throw(err) {
+                asyncGeneratorStep(gen, resolve, reject, _next, _throw, "throw", err);
+            }
+            _next(undefined);
+        });
+    };
+}
+function _define_property(obj, key, value) {
+    if (key in obj) {
+        Object.defineProperty(obj, key, {
+            value: value,
+            enumerable: true,
+            configurable: true,
+            writable: true
+        });
+    } else {
+        obj[key] = value;
+    }
+    return obj;
+}
+function _instanceof(left, right) {
+    if (right != null && typeof Symbol !== "undefined" && right[Symbol.hasInstance]) {
+        return !!right[Symbol.hasInstance](left);
+    } else {
+        return left instanceof right;
+    }
+}
+function _getRequireWildcardCache(nodeInterop) {
+    if (typeof WeakMap !== "function") return null;
+    var cacheBabelInterop = new WeakMap();
+    var cacheNodeInterop = new WeakMap();
+    return (_getRequireWildcardCache = function(nodeInterop) {
+        return nodeInterop ? cacheNodeInterop : cacheBabelInterop;
+    })(nodeInterop);
+}
+function _interop_require_wildcard(obj, nodeInterop) {
+    if (!nodeInterop && obj && obj.__esModule) {
+        return obj;
+    }
+    if (obj === null || typeof obj !== "object" && typeof obj !== "function") {
+        return {
+            default: obj
+        };
+    }
+    var cache = _getRequireWildcardCache(nodeInterop);
+    if (cache && cache.has(obj)) {
+        return cache.get(obj);
+    }
+    var newObj = {
+        __proto__: null
+    };
+    var hasPropertyDescriptor = Object.defineProperty && Object.getOwnPropertyDescriptor;
+    for(var key in obj){
+        if (key !== "default" && Object.prototype.hasOwnProperty.call(obj, key)) {
+            var desc = hasPropertyDescriptor ? Object.getOwnPropertyDescriptor(obj, key) : null;
+            if (desc && (desc.get || desc.set)) {
+                Object.defineProperty(newObj, key, desc);
+            } else {
+                newObj[key] = obj[key];
+            }
+        }
+    }
+    newObj.default = obj;
+    if (cache) {
+        cache.set(obj, newObj);
+    }
+    return newObj;
+}
+function _iterable_to_array_limit(arr, i) {
+    var _i = arr == null ? null : typeof Symbol !== "undefined" && arr[Symbol.iterator] || arr["@@iterator"];
+    if (_i == null) return;
+    var _arr = [];
+    var _n = true;
+    var _d = false;
+    var _s, _e;
+    try {
+        for(_i = _i.call(arr); !(_n = (_s = _i.next()).done); _n = true){
+            _arr.push(_s.value);
+            if (i && _arr.length === i) break;
+        }
+    } catch (err) {
+        _d = true;
+        _e = err;
+    } finally{
+        try {
+            if (!_n && _i["return"] != null) _i["return"]();
+        } finally{
+            if (_d) throw _e;
+        }
+    }
+    return _arr;
+}
+function _non_iterable_rest() {
+    throw new TypeError("Invalid attempt to destructure non-iterable instance.\\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method.");
+}
+function _object_spread(target) {
+    for(var i = 1; i < arguments.length; i++){
+        var source = arguments[i] != null ? arguments[i] : {};
+        var ownKeys = Object.keys(source);
+        if (typeof Object.getOwnPropertySymbols === "function") {
+            ownKeys = ownKeys.concat(Object.getOwnPropertySymbols(source).filter(function(sym) {
+                return Object.getOwnPropertyDescriptor(source, sym).enumerable;
+            }));
+        }
+        ownKeys.forEach(function(key) {
+            _define_property(target, key, source[key]);
+        });
+    }
+    return target;
+}
+function ownKeys(object, enumerableOnly) {
+    var keys = Object.keys(object);
+    if (Object.getOwnPropertySymbols) {
+        var symbols = Object.getOwnPropertySymbols(object);
+        if (enumerableOnly) {
+            symbols = symbols.filter(function(sym) {
+                return Object.getOwnPropertyDescriptor(object, sym).enumerable;
+            });
+        }
+        keys.push.apply(keys, symbols);
+    }
+    return keys;
+}
+function _object_spread_props(target, source) {
+    source = source != null ? source : {};
+    if (Object.getOwnPropertyDescriptors) {
+        Object.defineProperties(target, Object.getOwnPropertyDescriptors(source));
+    } else {
+        ownKeys(Object(source)).forEach(function(key) {
+            Object.defineProperty(target, key, Object.getOwnPropertyDescriptor(source, key));
+        });
+    }
+    return target;
+}
+function _sliced_to_array(arr, i) {
+    return _array_with_holes(arr) || _iterable_to_array_limit(arr, i) || _unsupported_iterable_to_array(arr, i) || _non_iterable_rest();
+}
+function _unsupported_iterable_to_array(o, minLen) {
+    if (!o) return;
+    if (typeof o === "string") return _array_like_to_array(o, minLen);
+    var n = Object.prototype.toString.call(o).slice(8, -1);
+    if (n === "Object" && o.constructor) n = o.constructor.name;
+    if (n === "Map" || n === "Set") return Array.from(n);
+    if (n === "Arguments" || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(n)) return _array_like_to_array(o, minLen);
+}
+function _ts_generator(thisArg, body) {
+    var f, y, t, _ = {
+        label: 0,
+        sent: function() {
+            if (t[0] & 1) throw t[1];
+            return t[1];
+        },
+        trys: [],
+        ops: []
+    }, g = Object.create((typeof Iterator === "function" ? Iterator : Object).prototype), d = Object.defineProperty;
+    return d(g, "next", {
+        value: verb(0)
+    }), d(g, "throw", {
+        value: verb(1)
+    }), d(g, "return", {
+        value: verb(2)
+    }), typeof Symbol === "function" && d(g, Symbol.iterator, {
+        value: function() {
+            return this;
+        }
+    }), g;
+    function verb(n) {
+        return function(v) {
+            return step([
+                n,
+                v
+            ]);
+        };
+    }
+    function step(op) {
+        if (f) throw new TypeError("Generator is already executing.");
+        while(g && (g = 0, op[0] && (_ = 0)), _)try {
+            if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
+            if (y = 0, t) op = [
+                op[0] & 2,
+                t.value
+            ];
+            switch(op[0]){
+                case 0:
+                case 1:
+                    t = op;
+                    break;
+                case 4:
+                    _.label++;
+                    return {
+                        value: op[1],
+                        done: false
+                    };
+                case 5:
+                    _.label++;
+                    y = op[1];
+                    op = [
+                        0
+                    ];
+                    continue;
+                case 7:
+                    op = _.ops.pop();
+                    _.trys.pop();
+                    continue;
+                default:
+                    if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) {
+                        _ = 0;
+                        continue;
+                    }
+                    if (op[0] === 3 && (!t || op[1] > t[0] && op[1] < t[3])) {
+                        _.label = op[1];
+                        break;
+                    }
+                    if (op[0] === 6 && _.label < t[1]) {
+                        _.label = t[1];
+                        t = op;
+                        break;
+                    }
+                    if (t && _.label < t[2]) {
+                        _.label = t[2];
+                        _.ops.push(op);
+                        break;
+                    }
+                    if (t[2]) _.ops.pop();
+                    _.trys.pop();
+                    continue;
+            }
+            op = body.call(thisArg, _);
+        } catch (e) {
+            op = [
+                6,
+                e
+            ];
+            y = 0;
+        } finally{
+            f = t = 0;
+        }
+        if (op[0] & 5) throw op[1];
+        return {
+            value: op[0] ? op[1] : void 0,
+            done: true
+        };
+    }
+}
+/**
+ * Infer transport type: explicit type > URL protocol > default 'stdio'
+ */ function inferTransportType(config) {
+    if (config.type) {
+        return config.type;
+    }
+    if (config.url) {
+        var url = new URL(config.url);
+        if (url.protocol === 'http:' || url.protocol === 'https:') {
+            return 'http';
+        }
+    }
+    return 'stdio';
+}
+/**
+ * Helper to filter undefined values from environment
+ */ function filterEnv(env) {
+    var filtered = {};
+    var _iteratorNormalCompletion = true, _didIteratorError = false, _iteratorError = undefined;
+    try {
+        for(var _iterator = Object.entries(env)[Symbol.iterator](), _step; !(_iteratorNormalCompletion = (_step = _iterator.next()).done); _iteratorNormalCompletion = true){
+            var _step_value = _sliced_to_array(_step.value, 2), key = _step_value[0], value = _step_value[1];
+            if (value !== undefined) {
+                filtered[key] = value;
+            }
+        }
+    } catch (err) {
+        _didIteratorError = true;
+        _iteratorError = err;
+    } finally{
+        try {
+            if (!_iteratorNormalCompletion && _iterator.return != null) {
+                _iterator.return();
+            }
+        } finally{
+            if (_didIteratorError) {
+                throw _iteratorError;
+            }
+        }
+    }
+    return filtered;
+}
+/**
+ * Determine spawn behavior based on dialects and config structure.
+ *
+ * Dialects:
+ * - ['servers'] (default): Claude Code compatible - ignores start blocks, stdio only
+ * - ['start']: Only spawn servers with start blocks (HTTP server testing)
+ * - ['servers', 'start']: Spawn both start blocks and stdio servers (full MCP-Z extension support)
+ *
+ * Environment merging (when baseEnv provided):
+ * - HTTP servers (start block): { ...baseEnv, ...entry.start.env }
+ * - Stdio servers: { ...baseEnv, ...entry.env }
+ *
+ * When baseEnv is not provided, process.env is used as the base.
+ */ function getSpawnConfig(entry, dialects, baseEnv) {
+    var transportType = inferTransportType(entry);
+    var hasServers = dialects.includes('servers');
+    var hasStart = dialects.includes('start');
+    // If only 'servers' dialect: Claude Code compatible (ignore start blocks, stdio only)
+    if (hasServers && !hasStart) {
+        if (transportType === 'stdio' && entry.command) {
+            return {
+                shouldSpawn: true,
+                command: entry.command,
+                args: entry.args || [],
+                env: filterEnv(_object_spread({}, baseEnv, entry.env))
+            };
+        }
+        return {
+            shouldSpawn: false
+        };
+    }
+    // If only 'start' dialect: Only spawn servers with start blocks
+    if (hasStart && !hasServers) {
+        if (entry.start) {
+            return {
+                shouldSpawn: true,
+                command: entry.start.command,
+                args: entry.start.args || [],
+                env: filterEnv(_object_spread({}, baseEnv, entry.start.env))
+            };
+        }
+        return {
+            shouldSpawn: false
+        };
+    }
+    // Both dialects: Spawn both start blocks and stdio servers
+    // Priority: start blocks first, then stdio
+    if (entry.start) {
+        return {
+            shouldSpawn: true,
+            command: entry.start.command,
+            args: entry.start.args || [],
+            env: filterEnv(_object_spread({}, baseEnv, entry.start.env))
+        };
+    }
+    if (transportType === 'stdio' && entry.command) {
+        return {
+            shouldSpawn: true,
+            command: entry.command,
+            args: entry.args || [],
+            env: filterEnv(_object_spread({}, baseEnv, entry.env))
+        };
+    }
+    return {
+        shouldSpawn: false
+    };
+}
+function createServerRegistry(serversConfig, options) {
+    var _ref, _ref1, _ref2;
+    var cwd = (_ref = options === null || options === void 0 ? void 0 : options.cwd) !== null && _ref !== void 0 ? _ref : _process.cwd();
+    var dialects = (_ref1 = options === null || options === void 0 ? void 0 : options.dialects) !== null && _ref1 !== void 0 ? _ref1 : [
+        'servers'
+    ];
+    // Determine base environment:
+    // - If options.env provided, use it (process.env NOT included)
+    // - If options.env omitted, use process.env as base
+    var baseEnv = (_ref2 = options === null || options === void 0 ? void 0 : options.env) !== null && _ref2 !== void 0 ? _ref2 : _process.env;
+    // Validate working directory exists (fail fast for configuration errors)
+    if (!_fs.existsSync(cwd)) {
+        throw new Error("Cannot start servers: working directory '".concat(cwd, "' does not exist"));
+    }
+    // Validate configuration (fail fast with clear errors)
+    var validation = (0, _validateconfigts.validateServers)(serversConfig);
+    if (!validation.valid) {
+        var _ref3;
+        var _validation_errors;
+        throw new Error("Invalid servers configuration:\n".concat((_ref3 = (_validation_errors = validation.errors) === null || _validation_errors === void 0 ? void 0 : _validation_errors.join('\n')) !== null && _ref3 !== void 0 ? _ref3 : 'Unknown validation error'));
+    }
+    // Log validation warnings (non-blocking)
+    if (validation.warnings && validation.warnings.length > 0) {
+        var _iteratorNormalCompletion = true, _didIteratorError = false, _iteratorError = undefined;
+        try {
+            for(var _iterator = validation.warnings[Symbol.iterator](), _step; !(_iteratorNormalCompletion = (_step = _iterator.next()).done); _iteratorNormalCompletion = true){
+                var warning = _step.value;
+                _loggerts.logger.warn(warning);
+            }
+        } catch (err) {
+            _didIteratorError = true;
+            _iteratorError = err;
+        } finally{
+            try {
+                if (!_iteratorNormalCompletion && _iterator.return != null) {
+                    _iterator.return();
+                }
+            } finally{
+                if (_didIteratorError) {
+                    throw _iteratorError;
+                }
+            }
+        }
+    }
+    var servers = new Map();
+    var clients = new Set();
+    var sharedStdioClients = new Map();
+    var _iteratorNormalCompletion1 = true, _didIteratorError1 = false, _iteratorError1 = undefined;
+    try {
+        // Start each server in the configuration
+        for(var _iterator1 = Object.entries(serversConfig)[Symbol.iterator](), _step1; !(_iteratorNormalCompletion1 = (_step1 = _iterator1.next()).done); _iteratorNormalCompletion1 = true){
+            var _step_value = _sliced_to_array(_step1.value, 2), name = _step_value[0], entry = _step_value[1];
+            // Infer transport type from config
+            var transportType = inferTransportType(entry);
+            // Determine spawn behavior based on dialects
+            var spawnConfig = getSpawnConfig(entry, dialects, baseEnv);
+            // Check if we should spawn this server
+            if (!spawnConfig.shouldSpawn) {
+                // External server - just log, no spawn needed
+                if (entry.url) {
+                    _loggerts.logger.info("[".concat(name, "] external ").concat(transportType, " server (url: ").concat(entry.url, ")"));
+                } else {
+                    _loggerts.logger.warn("[".concat(name, "] skipping: no spawn configuration (missing start or command) and no url for external server"));
+                }
+                continue;
+            }
+            try {
+                // Validate spawn config
+                if (!spawnConfig.command) {
+                    throw new Error('Server "'.concat(name, '" missing command field'));
+                }
+                // All servers use the same working directory (cwd from options)
+                var resolvedCwd = cwd;
+                // Start the server
+                _loggerts.logger.info("[".concat(name, "] starting ").concat(transportType, " server (").concat(spawnConfig.command, " ").concat((spawnConfig.args || []).join(' '), ")"));
+                // stdio servers need 'pipe' for MCP communication over stdin/stdout
+                // network servers use 'inherit' so we see their logs
+                var stdio = transportType === 'stdio' ? 'pipe' : 'inherit';
+                var handle = (0, _spawnserverts.spawnProcess)(_object_spread_props(_object_spread(_object_spread_props(_object_spread({
+                    name: name,
+                    command: spawnConfig.command
+                }, spawnConfig.args !== undefined && {
+                    args: spawnConfig.args
+                }), {
+                    cwd: resolvedCwd
+                }), spawnConfig.env && Object.keys(spawnConfig.env).length > 0 && {
+                    env: spawnConfig.env
+                }), {
+                    stdio: stdio
+                }));
+                // Add server to registry (starting is fast, readiness is lazy)
+                servers.set(name, handle);
+                _loggerts.logger.info("[".concat(name, "] started successfully"));
+            } catch (e) {
+                _loggerts.logger.info("[".concat(name, "] start ERROR: ").concat(_instanceof(e, Error) ? e.message : String(e)));
+            }
+        }
+    } catch (err) {
+        _didIteratorError1 = true;
+        _iteratorError1 = err;
+    } finally{
+        try {
+            if (!_iteratorNormalCompletion1 && _iterator1.return != null) {
+                _iterator1.return();
+            }
+        } finally{
+            if (_didIteratorError1) {
+                throw _iteratorError1;
+            }
+        }
+    }
+    // Create connect function that tracks clients
+    var connect = function(name, options) {
+        return _async_to_generator(function() {
+            var serverEntry, available, transportType, entry, released, lease, registryLike, rawClient, decorated;
+            return _ts_generator(this, function(_state) {
+                switch(_state.label){
+                    case 0:
+                        serverEntry = serversConfig[name];
+                        if (!serverEntry) {
+                            available = Object.keys(serversConfig).join(', ');
+                            throw new Error("Server '".concat(name, "' not found in config. Available servers: ").concat(available || 'none'));
+                        }
+                        transportType = inferTransportType(serverEntry);
+                        if (!(transportType === 'stdio')) return [
+                            3,
+                            3
+                        ];
+                        // Stdio is a single logical connection; reuse one client and lease references.
+                        entry = sharedStdioClients.get(name);
+                        if (!entry) {
+                            entry = {
+                                refs: 0
+                            };
+                            sharedStdioClients.set(name, entry);
+                        }
+                        if (!!entry.client) return [
+                            3,
+                            2
+                        ];
+                        if (!entry.connecting) {
+                            entry.connecting = (function() {
+                                return _async_to_generator(function() {
+                                    var registryLike, rawClient, decorated;
+                                    return _ts_generator(this, function(_state) {
+                                        switch(_state.label){
+                                            case 0:
+                                                // Pass minimal RegistryLike object to connectMcpClient
+                                                registryLike = {
+                                                    config: serversConfig,
+                                                    servers: servers
+                                                };
+                                                return [
+                                                    4,
+                                                    (0, _connectclientts.connectMcpClient)(registryLike, name, options)
+                                                ];
+                                            case 1:
+                                                rawClient = _state.sent();
+                                                decorated = (0, _clienthelpersts.decorateClient)(rawClient, {
+                                                    serverName: name
+                                                });
+                                                entry.client = decorated;
+                                                entry.connecting = undefined;
+                                                return [
+                                                    2,
+                                                    decorated
+                                                ];
+                                        }
+                                    });
+                                })();
+                            })().catch(function(error) {
+                                sharedStdioClients.delete(name);
+                                throw error;
+                            });
+                        }
+                        return [
+                            4,
+                            entry.connecting
+                        ];
+                    case 1:
+                        _state.sent();
+                        _state.label = 2;
+                    case 2:
+                        if (!entry.client) {
+                            throw new Error("Failed to connect to stdio server '".concat(name, "'"));
+                        }
+                        entry.refs += 1;
+                        released = false;
+                        lease = new Proxy(entry.client, {
+                            get: function get(target, prop) {
+                                if (prop === 'close') {
+                                    return function() {
+                                        return _async_to_generator(function() {
+                                            return _ts_generator(this, function(_state) {
+                                                switch(_state.label){
+                                                    case 0:
+                                                        if (released) return [
+                                                            2
+                                                        ];
+                                                        released = true;
+                                                        clients.delete(lease);
+                                                        entry.refs = Math.max(0, entry.refs - 1);
+                                                        if (!(entry.refs === 0)) return [
+                                                            3,
+                                                            2
+                                                        ];
+                                                        sharedStdioClients.delete(name);
+                                                        return [
+                                                            4,
+                                                            target.close()
+                                                        ];
+                                                    case 1:
+                                                        _state.sent();
+                                                        _state.label = 2;
+                                                    case 2:
+                                                        return [
+                                                            2
+                                                        ];
+                                                }
+                                            });
+                                        })();
+                                    };
+                                }
+                                var value = Reflect.get(target, prop, target);
+                                if (typeof value === 'function') {
+                                    return value.bind(target);
+                                }
+                                return value;
+                            }
+                        });
+                        clients.add(lease);
+                        return [
+                            2,
+                            lease
+                        ];
+                    case 3:
+                        // Pass minimal RegistryLike object to connectMcpClient
+                        registryLike = {
+                            config: serversConfig,
+                            servers: servers
+                        };
+                        return [
+                            4,
+                            (0, _connectclientts.connectMcpClient)(registryLike, name, options)
+                        ];
+                    case 4:
+                        rawClient = _state.sent();
+                        decorated = (0, _clienthelpersts.decorateClient)(rawClient, {
+                            serverName: name
+                        });
+                        clients.add(decorated);
+                        return [
+                            2,
+                            decorated
+                        ];
+                }
+            });
+        })();
+    };
+    // Create close function that stops all clients and servers
+    var close = function() {
+        var signal = arguments.length > 0 && arguments[0] !== void 0 ? arguments[0] : 'SIGINT', opts = arguments.length > 1 && arguments[1] !== void 0 ? arguments[1] : {};
+        return _async_to_generator(function() {
+            var clientClosePromises, closeResults, timedOut, killedCount;
+            return _ts_generator(this, function(_state) {
+                switch(_state.label){
+                    case 0:
+                        _loggerts.logger.info("[registry] closing (".concat(signal, ")"));
+                        // First, close all tracked clients
+                        clientClosePromises = Array.from(clients).map(function(client) {
+                            return _async_to_generator(function() {
+                                var unused;
+                                return _ts_generator(this, function(_state) {
+                                    switch(_state.label){
+                                        case 0:
+                                            _state.trys.push([
+                                                0,
+                                                2,
+                                                ,
+                                                3
+                                            ]);
+                                            return [
+                                                4,
+                                                client.close()
+                                            ];
+                                        case 1:
+                                            _state.sent();
+                                            return [
+                                                3,
+                                                3
+                                            ];
+                                        case 2:
+                                            unused = _state.sent();
+                                            return [
+                                                3,
+                                                3
+                                            ];
+                                        case 3:
+                                            return [
+                                                2
+                                            ];
+                                    }
+                                });
+                            })();
+                        });
+                        return [
+                            4,
+                            Promise.all(clientClosePromises)
+                        ];
+                    case 1:
+                        _state.sent();
+                        clients.clear();
+                        // Then close all server processes
+                        if (servers.size === 0) {
+                            return [
+                                2,
+                                {
+                                    timedOut: false,
+                                    killedCount: 0
+                                }
+                            ];
+                        }
+                        return [
+                            4,
+                            Promise.all(Array.from(servers.values()).map(function(server) {
+                                return server.close(signal, opts);
+                            }))
+                        ];
+                    case 2:
+                        closeResults = _state.sent();
+                        // Check if any timed out and count how many were force-killed
+                        timedOut = closeResults.some(function(result) {
+                            return result.timedOut;
+                        });
+                        killedCount = closeResults.filter(function(result) {
+                            return result.killed;
+                        }).length;
+                        return [
+                            2,
+                            {
+                                timedOut: timedOut,
+                                killedCount: killedCount
+                            }
+                        ];
+                }
+            });
+        })();
+    };
+    var searchFromRegistry = function(query) {
+        var _$options = arguments.length > 1 && arguments[1] !== void 0 ? arguments[1] : {};
+        return _async_to_generator(function() {
+            var _options_servers, requestedServers, unknownServers, capabilityClients, failures, ensureClient, failureDetails, index;
+            return _ts_generator(this, function(_state) {
+                switch(_state.label){
+                    case 0:
+                        requestedServers = (_options_servers = _$options.servers) !== null && _options_servers !== void 0 ? _options_servers : Object.keys(serversConfig);
+                        if (requestedServers.length === 0) {
+                            throw new Error('Cannot search capabilities: registry has no configured servers');
+                        }
+                        unknownServers = requestedServers.filter(function(name) {
+                            return !(name in serversConfig);
+                        });
+                        if (unknownServers.length > 0) {
+                            throw new Error("Cannot search capabilities: unknown server(s) [".concat(unknownServers.join(', '), "]"));
+                        }
+                        capabilityClients = new Map();
+                        failures = [];
+                        ensureClient = function(serverName) {
+                            return _async_to_generator(function() {
+                                var _iteratorNormalCompletion, _didIteratorError, _iteratorError, _iterator, _step, client;
+                                return _ts_generator(this, function(_state) {
+                                    _iteratorNormalCompletion = true, _didIteratorError = false, _iteratorError = undefined;
+                                    try {
+                                        for(_iterator = clients[Symbol.iterator](); !(_iteratorNormalCompletion = (_step = _iterator.next()).done); _iteratorNormalCompletion = true){
+                                            client = _step.value;
+                                            if (client.serverName === serverName) {
+                                                return [
+                                                    2,
+                                                    client
+                                                ];
+                                            }
+                                        }
+                                    } catch (err) {
+                                        _didIteratorError = true;
+                                        _iteratorError = err;
+                                    } finally{
+                                        try {
+                                            if (!_iteratorNormalCompletion && _iterator.return != null) {
+                                                _iterator.return();
+                                            }
+                                        } finally{
+                                            if (_didIteratorError) {
+                                                throw _iteratorError;
+                                            }
+                                        }
+                                    }
+                                    return [
+                                        2,
+                                        connect(serverName)
+                                    ];
+                                });
+                            })();
+                        };
+                        return [
+                            4,
+                            Promise.all(requestedServers.map(function(serverName) {
+                                return _async_to_generator(function() {
+                                    var managed, error;
+                                    return _ts_generator(this, function(_state) {
+                                        switch(_state.label){
+                                            case 0:
+                                                _state.trys.push([
+                                                    0,
+                                                    2,
+                                                    ,
+                                                    3
+                                                ]);
+                                                return [
+                                                    4,
+                                                    ensureClient(serverName)
+                                                ];
+                                            case 1:
+                                                managed = _state.sent();
+                                                capabilityClients.set(serverName, managed.nativeClient);
+                                                return [
+                                                    3,
+                                                    3
+                                                ];
+                                            case 2:
+                                                error = _state.sent();
+                                                failures.push({
+                                                    server: serverName,
+                                                    reason: _instanceof(error, Error) ? error.message : String(error)
+                                                });
+                                                return [
+                                                    3,
+                                                    3
+                                                ];
+                                            case 3:
+                                                return [
+                                                    2
+                                                ];
+                                        }
+                                    });
+                                })();
+                            }))
+                        ];
+                    case 1:
+                        _state.sent();
+                        if (capabilityClients.size === 0) {
+                            failureDetails = failures.length > 0 ? " Connection failures: ".concat(failures.map(function(f) {
+                                return "".concat(f.server, " (").concat(f.reason, ")");
+                            }).join('; ')) : '';
+                            throw new Error("Cannot search capabilities: unable to connect to any requested servers.".concat(failureDetails));
+                        }
+                        if (failures.length > 0) {
+                            throw new Error("Cannot search capabilities: failed to connect to server(s) [".concat(failures.map(function(f) {
+                                return f.server;
+                            }).join(', '), "]. Reasons: ").concat(failures.map(function(f) {
+                                return "".concat(f.server, ": ").concat(f.reason);
+                            }).join('; ')));
+                        }
+                        return [
+                            4,
+                            (0, _indexts.buildCapabilityIndex)(capabilityClients)
+                        ];
+                    case 2:
+                        index = _state.sent();
+                        return [
+                            2,
+                            (0, _indexts.searchCapabilities)(index, query, _$options)
+                        ];
+                }
+            });
+        })();
+    };
+    // Async dispose for `await using` pattern
+    var asyncDispose = function() {
+        return _async_to_generator(function() {
+            return _ts_generator(this, function(_state) {
+                switch(_state.label){
+                    case 0:
+                        return [
+                            4,
+                            close()
+                        ];
+                    case 1:
+                        _state.sent();
+                        return [
+                            2
+                        ];
+                }
+            });
+        })();
+    };
+    return _define_property({
+        config: serversConfig,
+        servers: servers,
+        clients: clients,
+        connect: connect,
+        close: close,
+        searchCapabilities: searchFromRegistry
+    }, Symbol.asyncDispose, asyncDispose);
+}
+/* CJS INTEROP */ if (exports.__esModule && exports.default) { try { Object.defineProperty(exports.default, '__esModule', { value: true }); for (var key in exports) { exports.default[key] = exports[key]; } } catch (_) {}; module.exports = exports.default; }