@animalabs/agent-framework 0.1.0

package/dist/src/mcpl/server-registry.js

@@ -0,0 +1,119 @@
+/**
+ * McplServerRegistry — manages all connected MCPL server connections.
+ *
+ * Provides lookup by id, capability filtering, and feature set matching.
+ */
+import { McplServerConnection } from './server-connection.js';
+/**
+ * Container that manages all connected MCPL servers.
+ *
+ * Provides methods to add/remove servers, look up by id, and query
+ * servers by advertised capabilities or feature set names.
+ */
+export class McplServerRegistry {
+    servers = new Map();
+    /**
+     * Connect to an MCPL server and register it.
+     *
+     * Throws if a server with the same id is already registered, or if
+     * the connection/handshake fails.
+     */
+    async addServer(config, hostCapabilities) {
+        if (this.servers.has(config.id)) {
+            throw new Error(`MCPL server "${config.id}" is already registered`);
+        }
+        const connection = config.reconnect
+            ? await McplServerConnection.connectWithReconnect(config, hostCapabilities)
+            : await McplServerConnection.connect(config, hostCapabilities);
+        this.servers.set(config.id, connection);
+        // Auto-remove on unexpected close (unless reconnect will re-add)
+        connection.on('close', () => {
+            if (!config.reconnect) {
+                this.servers.delete(config.id);
+            }
+        });
+        // Re-emit reconnect events for observability
+        connection.on('reconnect', () => {
+            // Connection already in the map — capabilities may have changed
+        });
+        return connection;
+    }
+    /**
+     * Disconnect and remove a server by id.
+     *
+     * No-op if the server is not registered.
+     */
+    async removeServer(id) {
+        const connection = this.servers.get(id);
+        if (!connection) {
+            return;
+        }
+        this.servers.delete(id);
+        await connection.close();
+    }
+    /**
+     * Get a server connection by id, or null if not found.
+     */
+    getServer(id) {
+        return this.servers.get(id) ?? null;
+    }
+    /**
+     * Get all currently connected servers.
+     */
+    getAllServers() {
+        return Array.from(this.servers.values());
+    }
+    /**
+     * Get all servers that advertise a specific capability.
+     *
+     * Supported capability queries:
+     * - `'pushEvents'`                   — `capabilities.pushEvents === true`
+     * - `'contextHooks.beforeInference'`  — `capabilities.contextHooks?.beforeInference === true`
+     * - `'contextHooks.afterInference'`   — `capabilities.contextHooks?.afterInference` is truthy
+     * - `'inferenceRequest'`              — `capabilities.inferenceRequest` is truthy
+     * - `'modelInfo'`                     — `capabilities.modelInfo === true`
+     */
+    getServersWithCapability(cap) {
+        return this.getAllServers().filter((server) => {
+            const caps = server.capabilities;
+            if (!caps)
+                return false;
+            switch (cap) {
+                case 'pushEvents':
+                    return caps.pushEvents === true;
+                case 'contextHooks.beforeInference':
+                    return caps.contextHooks?.beforeInference === true;
+                case 'contextHooks.afterInference':
+                    return !!caps.contextHooks?.afterInference;
+                case 'inferenceRequest':
+                    return !!caps.inferenceRequest;
+                case 'modelInfo':
+                    return caps.modelInfo === true;
+                default:
+                    return false;
+            }
+        });
+    }
+    /**
+     * Get all servers that declare a feature set with the given name.
+     *
+     * Checks `capabilities.featureSets` for a key matching `featureSet`.
+     */
+    getServersForFeatureSet(featureSet) {
+        return this.getAllServers().filter((server) => {
+            const caps = server.capabilities;
+            if (!caps?.featureSets)
+                return false;
+            return featureSet in caps.featureSets;
+        });
+    }
+    /**
+     * Close all server connections and clear the registry.
+     */
+    async closeAll() {
+        const connections = Array.from(this.servers.values());
+        this.servers.clear();
+        await Promise.all(connections.map((c) => c.close()));
+    }
+}
+//# sourceMappingURL=server-registry.js.map

package/dist/src/mcpl/server-registry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"server-registry.js","sourceRoot":"","sources":["../../../src/mcpl/server-registry.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EAAE,oBAAoB,EAAE,MAAM,wBAAwB,CAAC;AAY9D;;;;;GAKG;AACH,MAAM,OAAO,kBAAkB;IACrB,OAAO,GAAG,IAAI,GAAG,EAAgC,CAAC;IAE1D;;;;;OAKG;IACH,KAAK,CAAC,SAAS,CACb,MAAwB,EACxB,gBAAsC;QAEtC,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC;YAChC,MAAM,IAAI,KAAK,CAAC,gBAAgB,MAAM,CAAC,EAAE,yBAAyB,CAAC,CAAC;QACtE,CAAC;QAED,MAAM,UAAU,GAAG,MAAM,CAAC,SAAS;YACjC,CAAC,CAAC,MAAM,oBAAoB,CAAC,oBAAoB,CAAC,MAAM,EAAE,gBAAgB,CAAC;YAC3E,CAAC,CAAC,MAAM,oBAAoB,CAAC,OAAO,CAAC,MAAM,EAAE,gBAAgB,CAAC,CAAC;QACjE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;QAExC,iEAAiE;QACjE,UAAU,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;YAC1B,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;gBACtB,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YACjC,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,6CAA6C;QAC7C,UAAU,CAAC,EAAE,CAAC,WAAW,EAAE,GAAG,EAAE;YAC9B,gEAAgE;QAClE,CAAC,CAAC,CAAC;QAEH,OAAO,UAAU,CAAC;IACpB,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,YAAY,CAAC,EAAU;QAC3B,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACxC,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,OAAO;QACT,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QACxB,MAAM,UAAU,CAAC,KAAK,EAAE,CAAC;IAC3B,CAAC;IAED;;OAEG;IACH,SAAS,CAAC,EAAU;QAClB,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC;IACtC,CAAC;IAED;;OAEG;IACH,aAAa;QACX,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;IAC3C,CAAC;IAED;;;;;;;;;OASG;IACH,wBAAwB,CAAC,GAAwB;QAC/C,OAAO,IAAI,CAAC,aAAa,EAAE,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE;YAC5C,MAAM,IAAI,GAAG,MAAM,CAAC,YAAY,CAAC;YACjC,IAAI,CAAC,IAAI;gBAAE,OAAO,KAAK,CAAC;YAExB,QAAQ,GAAG,EAAE,CAAC;gBACZ,KAAK,YAAY;oBACf,OAAO,IAAI,CAAC,UAAU,KAAK,IAAI,CAAC;gBAClC,KAAK,8BAA8B;oBACjC,OAAO,IAAI,CAAC,YAAY,EAAE,eAAe,KAAK,IAAI,CAAC;gBACrD,KAAK,6BAA6B;oBAChC,OAAO,CAAC,CAAC,IAAI,CAAC,YAAY,EAAE,cAAc,CAAC;gBAC7C,KAAK,kBAAkB;oBACrB,OAAO,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC;gBACjC,KAAK,WAAW;oBACd,OAAO,IAAI,CAAC,SAAS,KAAK,IAAI,CAAC;gBACjC;oBACE,OAAO,KAAK,CAAC;YACjB,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;OAIG;IACH,uBAAuB,CAAC,UAAkB;QACxC,OAAO,IAAI,CAAC,aAAa,EAAE,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE;YAC5C,MAAM,IAAI,GAAG,MAAM,CAAC,YAAY,CAAC;YACjC,IAAI,CAAC,IAAI,EAAE,WAAW;gBAAE,OAAO,KAAK,CAAC;YACrC,OAAO,UAAU,IAAI,IAAI,CAAC,WAAW,CAAC;QACxC,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ;QACZ,MAAM,WAAW,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;QACtD,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACrB,MAAM,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;IACvD,CAAC;CACF"}