@theia/ai-external-api 1.76.0-next.33

package/lib/node/external-chat-session-registry.js

@@ -0,0 +1,266 @@
+"use strict";
+// *****************************************************************************
+// Copyright (C) 2026 EclipseSource GmbH.
+//
+// This program and the accompanying materials are made available under the
+// terms of the Eclipse Public License v. 2.0 which is available at
+// http://www.eclipse.org/legal/epl-2.0.
+//
+// This Source Code may also be made available under the following Secondary
+// Licenses when the conditions for such availability set forth in the Eclipse
+// Public License v. 2.0 are satisfied: GNU General Public License, version 2
+// with the GNU Classpath Exception which is available at
+// https://www.gnu.org/software/classpath/license.html.
+//
+// SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
+// *****************************************************************************
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ExternalChatSessionRegistry = void 0;
+const tslib_1 = require("tslib");
+const chat_model_1 = require("@theia/ai-chat/lib/common/chat-model");
+const core_1 = require("@theia/core");
+const logger_1 = require("@theia/core/lib/common/logger");
+const inversify_1 = require("@theia/core/shared/inversify");
+/**
+ * Aggregates the {@link ExternalChatSessionProvider}s of all connected frontends.
+ *
+ * Providers register when a frontend connects and are removed when its connection closes.
+ * Queries fan out to all providers; a failing or unresponsive provider is logged and skipped
+ * so that a broken, busy, or disconnecting frontend does not break the external API. Sessions
+ * reported by more than one provider (e.g. a session restored in several frontends) are
+ * deduplicated, preferring restored over persisted-only, in-progress over finished, and more
+ * recent over older reports. Session actions (open, restore) are routed to the preferred
+ * provider in the same order.
+ */
+let ExternalChatSessionRegistry = class ExternalChatSessionRegistry {
+    constructor() {
+        /**
+         * Timeout in milliseconds after which an unresponsive provider is skipped in the query
+         * fan-outs. Only the read-only queries are bounded: an action call may legitimately take
+         * longer, and abandoning it could send it twice.
+         */
+        this.queryTimeout = 10_000;
+        this.providers = new Set();
+        this.onDidChangeSessionsEmitter = new core_1.Emitter();
+        /** Fired when the set of connected frontends changes or a frontend reports a session change. */
+        this.onDidChangeSessions = this.onDidChangeSessionsEmitter.event;
+    }
+    addProvider(provider) {
+        this.providers.add(provider);
+        this.notifySessionsChanged();
+    }
+    removeProvider(provider) {
+        if (this.providers.delete(provider)) {
+            this.notifySessionsChanged();
+        }
+    }
+    /** Announces that the sessions visible through this registry changed, see {@link onDidChangeSessions}. */
+    notifySessionsChanged() {
+        this.onDidChangeSessionsEmitter.fire();
+    }
+    async getSessions() {
+        const results = await this.queryProviders(provider => provider.getSessions());
+        const merged = new Map();
+        for (const summary of results.flat()) {
+            const existing = merged.get(summary.id);
+            if (!existing || this.prefer(summary, existing)) {
+                merged.set(summary.id, summary);
+            }
+        }
+        return Array.from(merged.values())
+            .sort((a, b) => (b.lastInteraction ?? 0) - (a.lastInteraction ?? 0));
+    }
+    /**
+     * Returns the detail of the session from the preferred frontend that knows it, or
+     * `undefined` if no connected frontend knows it. The frontends are ranked by their session
+     * summaries, so that only the preferred frontend has to report the conversation.
+     */
+    async getSession(sessionId) {
+        for (const provider of await this.getProvidersFor(sessionId)) {
+            try {
+                const session = await this.withQueryTimeout(provider.getSession(sessionId), 'get a session');
+                if (session) {
+                    return session;
+                }
+            }
+            catch (error) {
+                this.logger.warn('Failed to read a session from a frontend, trying the next one.', error);
+            }
+        }
+        return undefined;
+    }
+    /**
+     * Opens the session in the preferred frontend that knows it, restoring it if necessary.
+     * Returns `false` if no connected frontend knows the session.
+     */
+    async openSession(sessionId) {
+        for (const provider of await this.getProvidersFor(sessionId)) {
+            try {
+                if (await provider.openSession(sessionId)) {
+                    return true;
+                }
+            }
+            catch (error) {
+                this.logger.warn('Failed to open a session in a frontend, trying the next one.', error);
+            }
+        }
+        return false;
+    }
+    /**
+     * Restores the session in the preferred frontend that knows it and returns its detail.
+     * Returns `undefined` if no connected frontend knows the session.
+     */
+    async restoreSession(sessionId) {
+        for (const provider of await this.getProvidersFor(sessionId)) {
+            try {
+                const session = await provider.restoreSession(sessionId);
+                if (session) {
+                    return session;
+                }
+            }
+            catch (error) {
+                this.logger.warn('Failed to restore a session in a frontend, trying the next one.', error);
+            }
+        }
+        return undefined;
+    }
+    /**
+     * Sends the prompt to the given session in the preferred frontend that knows it.
+     * The first definitive answer wins: in particular, a `busy` rejection by the preferred
+     * frontend is not retried against less preferred copies of the session.
+     * Returns `undefined` if no connected frontend knows the session.
+     */
+    async sendPrompt(sessionId, prompt) {
+        for (const provider of await this.getProvidersFor(sessionId)) {
+            try {
+                const result = await provider.sendPrompt(sessionId, prompt);
+                if (result) {
+                    return result;
+                }
+            }
+            catch (error) {
+                this.logger.warn('Failed to send a prompt to a frontend, trying the next one.', error);
+            }
+        }
+        return undefined;
+    }
+    /**
+     * Creates a session in a frontend matching the requested workspace. Without a requested
+     * workspace, the session is created in the only connected frontend, and if the connected
+     * frontends have different workspaces open, the request is rejected as ambiguous.
+     * Rejected with `noFrontend` when no frontend is connected or none responds.
+     */
+    async createSession(request) {
+        if (this.providers.size === 0) {
+            return { failure: 'noFrontend' };
+        }
+        const candidates = await this.getProvidersForWorkspace(request.workspace);
+        if (candidates.length === 0) {
+            return { failure: request.workspace ? 'workspaceNotFound' : 'noFrontend' };
+        }
+        if (!request.workspace && new Set(candidates.map(candidate => candidate.workspace)).size > 1) {
+            return { failure: 'ambiguousWorkspace' };
+        }
+        for (const candidate of candidates) {
+            try {
+                return await candidate.provider.createSession(request);
+            }
+            catch (error) {
+                this.logger.warn('Failed to create a session in a frontend, trying the next one.', error);
+            }
+        }
+        throw new Error('All matching frontends failed to create the session.');
+    }
+    /** Returns the providers whose workspace matches the given one, or all providers if none is given. */
+    async getProvidersForWorkspace(workspace) {
+        const requested = workspace === undefined ? undefined : this.normalizeWorkspace(workspace);
+        const candidates = [];
+        await Promise.all(Array.from(this.providers.values(), async (provider) => {
+            try {
+                const providerWorkspace = await this.withQueryTimeout(provider.getWorkspace(), 'query a workspace');
+                const normalized = providerWorkspace === undefined ? undefined : this.normalizeWorkspace(providerWorkspace);
+                if (requested === undefined || requested === normalized) {
+                    candidates.push({ provider, workspace: normalized });
+                }
+            }
+            catch (error) {
+                this.logger.warn('Failed to query a frontend session provider, skipping it.', error);
+            }
+        }));
+        return candidates;
+    }
+    /**
+     * Normalizes a workspace URI for comparison: URIs that denote the same workspace but differ
+     * in their path encoding or in a trailing separator normalize to the same string.
+     */
+    normalizeWorkspace(workspace) {
+        const uri = new core_1.URI(workspace).normalizePath();
+        const path = uri.path.toString();
+        return path.length > 1 && path.endsWith('/') ? uri.withPath(path.slice(0, -1)).toString() : uri.toString();
+    }
+    /** Returns the providers that know the given session, most preferred report first. */
+    async getProvidersFor(sessionId) {
+        const candidates = [];
+        await Promise.all(Array.from(this.providers.values(), async (provider) => {
+            try {
+                const session = await this.withQueryTimeout(provider.getSessionSummary(sessionId), 'query a session summary');
+                if (session) {
+                    candidates.push({ provider, session });
+                }
+            }
+            catch (error) {
+                this.logger.warn('Failed to query a frontend session provider, skipping it.', error);
+            }
+        }));
+        return candidates
+            .sort((a, b) => this.prefer(a.session, b.session) ? -1 : this.prefer(b.session, a.session) ? 1 : 0)
+            .map(candidate => candidate.provider);
+    }
+    async queryProviders(query) {
+        const results = [];
+        await Promise.all(Array.from(this.providers.values(), async (provider) => {
+            try {
+                const result = await this.withQueryTimeout(query(provider), 'query a frontend');
+                if (result !== undefined) {
+                    results.push(result);
+                }
+            }
+            catch (error) {
+                this.logger.warn('Failed to query a frontend session provider, skipping it.', error);
+            }
+        }));
+        return results;
+    }
+    /**
+     * Rejects when the query does not answer within {@link queryTimeout}. Theia's RPC has no
+     * request timeout of its own, so an unresponsive frontend would otherwise stall the
+     * external API request that fans out to it indefinitely.
+     */
+    withQueryTimeout(query, description) {
+        let timer;
+        const timeout = new Promise((_, reject) => {
+            timer = setTimeout(() => reject(new Error(`A frontend did not ${description} within ${this.queryTimeout} ms.`)), this.queryTimeout);
+        });
+        return Promise.race([query, timeout]).finally(() => clearTimeout(timer));
+    }
+    prefer(candidate, existing) {
+        if (candidate.restored !== existing.restored) {
+            return candidate.restored;
+        }
+        const candidateInProgress = chat_model_1.ChatSessionStatus.isInProgress(candidate.status);
+        if (candidateInProgress !== chat_model_1.ChatSessionStatus.isInProgress(existing.status)) {
+            return candidateInProgress;
+        }
+        return (candidate.lastInteraction ?? 0) > (existing.lastInteraction ?? 0);
+    }
+};
+exports.ExternalChatSessionRegistry = ExternalChatSessionRegistry;
+tslib_1.__decorate([
+    (0, inversify_1.inject)(logger_1.ILogger),
+    (0, inversify_1.named)('ai-external-api:ExternalChatSessionRegistry'),
+    tslib_1.__metadata("design:type", Object)
+], ExternalChatSessionRegistry.prototype, "logger", void 0);
+exports.ExternalChatSessionRegistry = ExternalChatSessionRegistry = tslib_1.__decorate([
+    (0, inversify_1.injectable)()
+], ExternalChatSessionRegistry);
+//# sourceMappingURL=external-chat-session-registry.js.map

package/lib/node/external-chat-session-registry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"external-chat-session-registry.js","sourceRoot":"","sources":["../../src/node/external-chat-session-registry.ts"],"names":[],"mappings":";AAAA,gFAAgF;AAChF,yCAAyC;AACzC,EAAE;AACF,2EAA2E;AAC3E,mEAAmE;AACnE,wCAAwC;AACxC,EAAE;AACF,4EAA4E;AAC5E,8EAA8E;AAC9E,6EAA6E;AAC7E,yDAAyD;AACzD,uDAAuD;AACvD,EAAE;AACF,gFAAgF;AAChF,gFAAgF;;;;AAEhF,qEAAyE;AACzE,sCAAkD;AAClD,0DAAwD;AACxD,4DAAyE;AAMzE;;;;;;;;;;GAUG;AAEI,IAAM,2BAA2B,GAAjC,MAAM,2BAA2B;IAAjC;QAEH;;;;WAIG;QACgB,iBAAY,GAAG,MAAM,CAAC;QAKtB,cAAS,GAAG,IAAI,GAAG,EAA+B,CAAC;QAEnD,+BAA0B,GAAG,IAAI,cAAO,EAAQ,CAAC;QACpE,gGAAgG;QACvF,wBAAmB,GAAgB,IAAI,CAAC,0BAA0B,CAAC,KAAK,CAAC;IAwNtF,CAAC;IAtNG,WAAW,CAAC,QAAqC;QAC7C,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC7B,IAAI,CAAC,qBAAqB,EAAE,CAAC;IACjC,CAAC;IAED,cAAc,CAAC,QAAqC;QAChD,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC;YAClC,IAAI,CAAC,qBAAqB,EAAE,CAAC;QACjC,CAAC;IACL,CAAC;IAED,0GAA0G;IAC1G,qBAAqB;QACjB,IAAI,CAAC,0BAA0B,CAAC,IAAI,EAAE,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,WAAW;QACb,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,EAAE,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAC;QAC9E,MAAM,MAAM,GAAG,IAAI,GAAG,EAAsC,CAAC;QAC7D,KAAK,MAAM,OAAO,IAAI,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC;YACnC,MAAM,QAAQ,GAAG,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YACxC,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,QAAQ,CAAC,EAAE,CAAC;gBAC9C,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YACpC,CAAC;QACL,CAAC;QACD,OAAO,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;aAC7B,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,eAAe,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,eAAe,IAAI,CAAC,CAAC,CAAC,CAAC;IAC7E,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,UAAU,CAAC,SAAiB;QAC9B,KAAK,MAAM,QAAQ,IAAI,MAAM,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,EAAE,CAAC;YAC3D,IAAI,CAAC;gBACD,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,eAAe,CAAC,CAAC;gBAC7F,IAAI,OAAO,EAAE,CAAC;oBACV,OAAO,OAAO,CAAC;gBACnB,CAAC;YACL,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACb,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,gEAAgE,EAAE,KAAK,CAAC,CAAC;YAC9F,CAAC;QACL,CAAC;QACD,OAAO,SAAS,CAAC;IACrB,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,WAAW,CAAC,SAAiB;QAC/B,KAAK,MAAM,QAAQ,IAAI,MAAM,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,EAAE,CAAC;YAC3D,IAAI,CAAC;gBACD,IAAI,MAAM,QAAQ,CAAC,WAAW,CAAC,SAAS,CAAC,EAAE,CAAC;oBACxC,OAAO,IAAI,CAAC;gBAChB,CAAC;YACL,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACb,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,8DAA8D,EAAE,KAAK,CAAC,CAAC;YAC5F,CAAC;QACL,CAAC;QACD,OAAO,KAAK,CAAC;IACjB,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,cAAc,CAAC,SAAiB;QAClC,KAAK,MAAM,QAAQ,IAAI,MAAM,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,EAAE,CAAC;YAC3D,IAAI,CAAC;gBACD,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;gBACzD,IAAI,OAAO,EAAE,CAAC;oBACV,OAAO,OAAO,CAAC;gBACnB,CAAC;YACL,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACb,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,iEAAiE,EAAE,KAAK,CAAC,CAAC;YAC/F,CAAC;QACL,CAAC;QACD,OAAO,SAAS,CAAC;IACrB,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,UAAU,CAAC,SAAiB,EAAE,MAA0B;QAC1D,KAAK,MAAM,QAAQ,IAAI,MAAM,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,EAAE,CAAC;YAC3D,IAAI,CAAC;gBACD,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,UAAU,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;gBAC5D,IAAI,MAAM,EAAE,CAAC;oBACT,OAAO,MAAM,CAAC;gBAClB,CAAC;YACL,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACb,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,6DAA6D,EAAE,KAAK,CAAC,CAAC;YAC3F,CAAC;QACL,CAAC;QACD,OAAO,SAAS,CAAC;IACrB,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,aAAa,CAAC,OAAyC;QACzD,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;YAC5B,OAAO,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC;QACrC,CAAC;QACD,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,wBAAwB,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAC1E,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1B,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,YAAY,EAAE,CAAC;QAC/E,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,SAAS,IAAI,IAAI,GAAG,CAAC,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YAC3F,OAAO,EAAE,OAAO,EAAE,oBAAoB,EAAE,CAAC;QAC7C,CAAC;QACD,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;YACjC,IAAI,CAAC;gBACD,OAAO,MAAM,SAAS,CAAC,QAAQ,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;YAC3D,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACb,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,gEAAgE,EAAE,KAAK,CAAC,CAAC;YAC9F,CAAC;QACL,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,sDAAsD,CAAC,CAAC;IAC5E,CAAC;IAED,sGAAsG;IAC5F,KAAK,CAAC,wBAAwB,CAAC,SAA6B;QAClE,MAAM,SAAS,GAAG,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC;QAC3F,MAAM,UAAU,GAAoE,EAAE,CAAC;QACvF,MAAM,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,EAAE,KAAK,EAAC,QAAQ,EAAC,EAAE;YACnE,IAAI,CAAC;gBACD,MAAM,iBAAiB,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,YAAY,EAAE,EAAE,mBAAmB,CAAC,CAAC;gBACpG,MAAM,UAAU,GAAG,iBAAiB,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,iBAAiB,CAAC,CAAC;gBAC5G,IAAI,SAAS,KAAK,SAAS,IAAI,SAAS,KAAK,UAAU,EAAE,CAAC;oBACtD,UAAU,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,SAAS,EAAE,UAAU,EAAE,CAAC,CAAC;gBACzD,CAAC;YACL,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACb,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,2DAA2D,EAAE,KAAK,CAAC,CAAC;YACzF,CAAC;QACL,CAAC,CAAC,CAAC,CAAC;QACJ,OAAO,UAAU,CAAC;IACtB,CAAC;IAED;;;OAGG;IACO,kBAAkB,CAAC,SAAiB;QAC1C,MAAM,GAAG,GAAG,IAAI,UAAG,CAAC,SAAS,CAAC,CAAC,aAAa,EAAE,CAAC;QAC/C,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;QACjC,OAAO,IAAI,CAAC,MAAM,GAAG,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;IAC/G,CAAC;IAED,sFAAsF;IAC5E,KAAK,CAAC,eAAe,CAAC,SAAiB;QAC7C,MAAM,UAAU,GAAqF,EAAE,CAAC;QACxG,MAAM,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,EAAE,KAAK,EAAC,QAAQ,EAAC,EAAE;YACnE,IAAI,CAAC;gBACD,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,iBAAiB,CAAC,SAAS,CAAC,EAAE,yBAAyB,CAAC,CAAC;gBAC9G,IAAI,OAAO,EAAE,CAAC;oBACV,UAAU,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC;gBAC3C,CAAC;YACL,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACb,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,2DAA2D,EAAE,KAAK,CAAC,CAAC;YACzF,CAAC;QACL,CAAC,CAAC,CAAC,CAAC;QACJ,OAAO,UAAU;aACZ,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;aAClG,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;IAC9C,CAAC;IAES,KAAK,CAAC,cAAc,CAAI,KAAwE;QACtG,MAAM,OAAO,GAAQ,EAAE,CAAC;QACxB,MAAM,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,EAAE,KAAK,EAAC,QAAQ,EAAC,EAAE;YACnE,IAAI,CAAC;gBACD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE,kBAAkB,CAAC,CAAC;gBAChF,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;oBACvB,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACzB,CAAC;YACL,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACb,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,2DAA2D,EAAE,KAAK,CAAC,CAAC;YACzF,CAAC;QACL,CAAC,CAAC,CAAC,CAAC;QACJ,OAAO,OAAO,CAAC;IACnB,CAAC;IAED;;;;OAIG;IACO,gBAAgB,CAAI,KAAiB,EAAE,WAAmB;QAChE,IAAI,KAAiC,CAAC;QACtC,MAAM,OAAO,GAAG,IAAI,OAAO,CAAQ,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE;YAC7C,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,sBAAsB,WAAW,WAAW,IAAI,CAAC,YAAY,MAAM,CAAC,CAAC,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;QACxI,CAAC,CAAC,CAAC;QACH,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC;IAC7E,CAAC;IAES,MAAM,CAAC,SAAqC,EAAE,QAAoC;QACxF,IAAI,SAAS,CAAC,QAAQ,KAAK,QAAQ,CAAC,QAAQ,EAAE,CAAC;YAC3C,OAAO,SAAS,CAAC,QAAQ,CAAC;QAC9B,CAAC;QACD,MAAM,mBAAmB,GAAG,8BAAiB,CAAC,YAAY,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAC7E,IAAI,mBAAmB,KAAK,8BAAiB,CAAC,YAAY,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;YAC1E,OAAO,mBAAmB,CAAC;QAC/B,CAAC;QACD,OAAO,CAAC,SAAS,CAAC,eAAe,IAAI,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,eAAe,IAAI,CAAC,CAAC,CAAC;IAC9E,CAAC;CACJ,CAAA;AAxOY,kEAA2B;AAUjB;IADlB,IAAA,kBAAM,EAAC,gBAAO,CAAC;IAAE,IAAA,iBAAK,EAAC,6CAA6C,CAAC;;2DACnC;sCAV1B,2BAA2B;IADvC,IAAA,sBAAU,GAAE;GACA,2BAA2B,CAwOvC"}

package/lib/node/external-chat-session-registry.spec.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=external-chat-session-registry.spec.d.ts.map

package/lib/node/external-chat-session-registry.spec.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"external-chat-session-registry.spec.d.ts","sourceRoot":"","sources":["../../src/node/external-chat-session-registry.spec.ts"],"names":[],"mappings":""}

package/lib/node/external-chat-session-registry.spec.js

@@ -0,0 +1,366 @@
+"use strict";
+// *****************************************************************************
+// Copyright (C) 2026 EclipseSource GmbH.
+//
+// This program and the accompanying materials are made available under the
+// terms of the Eclipse Public License v. 2.0 which is available at
+// http://www.eclipse.org/legal/epl-2.0.
+//
+// This Source Code may also be made available under the following Secondary
+// Licenses when the conditions for such availability set forth in the Eclipse
+// Public License v. 2.0 are satisfied: GNU General Public License, version 2
+// with the GNU Classpath Exception which is available at
+// https://www.gnu.org/software/classpath/license.html.
+//
+// SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
+// *****************************************************************************
+Object.defineProperty(exports, "__esModule", { value: true });
+const mock_logger_1 = require("@theia/core/lib/common/test/mock-logger");
+const chai_1 = require("chai");
+const external_api_test_support_1 = require("@theia/external-api/lib/node/test/external-api-test-support");
+const external_chat_session_registry_1 = require("./external-chat-session-registry");
+describe('ExternalChatSessionRegistry', () => {
+    function createRegistry(queryTimeout) {
+        return external_api_test_support_1.ExternalApiTestSupport.inject(new external_chat_session_registry_1.ExternalChatSessionRegistry(), {
+            logger: new mock_logger_1.MockLogger(),
+            ...queryTimeout === undefined ? {} : { queryTimeout }
+        });
+    }
+    /** A provider whose queries never answer, standing in for an unresponsive frontend. */
+    function unresponsiveProvider() {
+        const hang = () => new Promise(() => { });
+        return {
+            getWorkspace: hang,
+            getSessions: hang,
+            getSessionSummary: hang,
+            getSession: hang,
+            openSession: hang,
+            restoreSession: hang,
+            sendPrompt: hang,
+            createSession: hang
+        };
+    }
+    function summary(id, status = 'idle', lastInteraction, restored = true) {
+        return { id, status, lastInteraction, restored };
+    }
+    function detail(id, status = 'idle', lastInteraction, restored = true) {
+        return { ...summary(id, status, lastInteraction, restored), messages: [{ actor: 'user', text: `question for ${id}` }] };
+    }
+    function provider(sessions, details = [], actions = {}) {
+        return {
+            getWorkspace: async () => undefined,
+            getSessions: async () => sessions,
+            getSessionSummary: async (sessionId) => details.find(candidate => candidate.id === sessionId) ?? sessions.find(candidate => candidate.id === sessionId),
+            getSession: async (sessionId) => details.find(candidate => candidate.id === sessionId),
+            openSession: async () => false,
+            restoreSession: async () => undefined,
+            sendPrompt: async () => undefined,
+            createSession: async () => ({ created: { session: summary('created') } }),
+            ...actions
+        };
+    }
+    function failingProvider() {
+        const fail = async () => { throw new Error('connection lost'); };
+        return {
+            getWorkspace: fail,
+            getSessions: fail,
+            getSessionSummary: fail,
+            getSession: fail,
+            openSession: fail,
+            restoreSession: fail,
+            sendPrompt: fail,
+            createSession: fail
+        };
+    }
+    describe('getSessions', () => {
+        it('returns an empty list without providers', async () => {
+            const registry = createRegistry();
+            (0, chai_1.expect)(await registry.getSessions()).to.deep.equal([]);
+        });
+        it('aggregates sessions of all providers sorted by last interaction', async () => {
+            const registry = createRegistry();
+            registry.addProvider(provider([summary('1', 'idle', 100)]));
+            registry.addProvider(provider([summary('2', 'idle', 300), summary('3', 'idle', 200)]));
+            const sessions = await registry.getSessions();
+            (0, chai_1.expect)(sessions.map(session => session.id)).to.deep.equal(['2', '3', '1']);
+        });
+        it('deduplicates sessions preferring restored reports', async () => {
+            const registry = createRegistry();
+            registry.addProvider(provider([summary('1', 'running', 300, false)]));
+            registry.addProvider(provider([summary('1', 'idle', 100, true)]));
+            const sessions = await registry.getSessions();
+            (0, chai_1.expect)(sessions).to.have.length(1);
+            (0, chai_1.expect)(sessions[0].restored).to.equal(true);
+        });
+        it('deduplicates sessions preferring in-progress reports', async () => {
+            const registry = createRegistry();
+            registry.addProvider(provider([summary('1', 'idle', 200)]));
+            registry.addProvider(provider([summary('1', 'running', 100)]));
+            const sessions = await registry.getSessions();
+            (0, chai_1.expect)(sessions).to.have.length(1);
+            (0, chai_1.expect)(sessions[0].status).to.equal('running');
+        });
+        it('deduplicates sessions preferring more recent reports', async () => {
+            const registry = createRegistry();
+            registry.addProvider(provider([summary('1', 'idle', 100)]));
+            registry.addProvider(provider([summary('1', 'failed', 200)]));
+            const sessions = await registry.getSessions();
+            (0, chai_1.expect)(sessions).to.have.length(1);
+            (0, chai_1.expect)(sessions[0].status).to.equal('failed');
+        });
+        it('no longer reports sessions of removed providers', async () => {
+            const registry = createRegistry();
+            const removed = provider([summary('1')]);
+            registry.addProvider(removed);
+            registry.addProvider(provider([summary('2')]));
+            registry.removeProvider(removed);
+            const sessions = await registry.getSessions();
+            (0, chai_1.expect)(sessions.map(session => session.id)).to.deep.equal(['2']);
+        });
+        it('skips failing providers', async () => {
+            const registry = createRegistry();
+            registry.addProvider(failingProvider());
+            registry.addProvider(provider([summary('1')]));
+            const sessions = await registry.getSessions();
+            (0, chai_1.expect)(sessions.map(session => session.id)).to.deep.equal(['1']);
+        });
+        it('skips unresponsive providers instead of waiting for them', async () => {
+            const registry = createRegistry(20);
+            registry.addProvider(unresponsiveProvider());
+            registry.addProvider(provider([summary('1')]));
+            const sessions = await registry.getSessions();
+            (0, chai_1.expect)(sessions.map(session => session.id)).to.deep.equal(['1']);
+        });
+    });
+    describe('getSession', () => {
+        it('returns undefined for unknown sessions', async () => {
+            const registry = createRegistry();
+            registry.addProvider(provider([]));
+            (0, chai_1.expect)(await registry.getSession('missing')).to.equal(undefined);
+        });
+        it('returns the session from the provider that knows it', async () => {
+            const registry = createRegistry();
+            registry.addProvider(provider([]));
+            registry.addProvider(provider([], [detail('1', 'running')]));
+            const session = await registry.getSession('1');
+            (0, chai_1.expect)(session?.status).to.equal('running');
+            (0, chai_1.expect)(session?.messages).to.deep.equal([{ actor: 'user', text: 'question for 1' }]);
+        });
+        it('prefers the in-progress duplicate', async () => {
+            const registry = createRegistry();
+            registry.addProvider(provider([], [detail('1', 'idle', 200)]));
+            registry.addProvider(provider([], [detail('1', 'awaitingApproval', 100)]));
+            const session = await registry.getSession('1');
+            (0, chai_1.expect)(session?.status).to.equal('awaitingApproval');
+        });
+        it('prefers the restored duplicate', async () => {
+            const registry = createRegistry();
+            registry.addProvider(provider([], [detail('1', 'idle', 100, true)]));
+            registry.addProvider(provider([], [detail('1', 'failed', 200, false)]));
+            const session = await registry.getSession('1');
+            (0, chai_1.expect)(session?.restored).to.equal(true);
+        });
+        it('skips failing providers', async () => {
+            const registry = createRegistry();
+            registry.addProvider(failingProvider());
+            registry.addProvider(provider([], [detail('1')]));
+            const session = await registry.getSession('1');
+            (0, chai_1.expect)(session?.id).to.equal('1');
+        });
+        it('reads the conversation from the preferred frontend only', async () => {
+            const registry = createRegistry();
+            const read = [];
+            const tracking = (name, known) => provider([], [known], { getSession: async (sessionId) => { read.push(name); return sessionId === known.id ? known : undefined; } });
+            registry.addProvider(tracking('idle', detail('1', 'idle', 200)));
+            registry.addProvider(tracking('running', detail('1', 'running', 100)));
+            const session = await registry.getSession('1');
+            (0, chai_1.expect)(session?.status).to.equal('running');
+            (0, chai_1.expect)(read).to.deep.equal(['running']);
+        });
+    });
+    describe('openSession', () => {
+        it('returns false when no provider knows the session', async () => {
+            const registry = createRegistry();
+            registry.addProvider(provider([]));
+            (0, chai_1.expect)(await registry.openSession('missing')).to.equal(false);
+        });
+        it('opens the session in the frontend that knows it', async () => {
+            const registry = createRegistry();
+            const opened = [];
+            registry.addProvider(provider([]));
+            registry.addProvider(provider([], [detail('1')], {
+                openSession: async (sessionId) => { opened.push(sessionId); return true; }
+            }));
+            (0, chai_1.expect)(await registry.openSession('1')).to.equal(true);
+            (0, chai_1.expect)(opened).to.deep.equal(['1']);
+        });
+        it('prefers the frontend with the restored session', async () => {
+            const registry = createRegistry();
+            const opened = [];
+            registry.addProvider(provider([], [detail('1', 'idle', 300, false)], {
+                openSession: async () => { opened.push('persisted-only'); return true; }
+            }));
+            registry.addProvider(provider([], [detail('1', 'idle', 100, true)], {
+                openSession: async () => { opened.push('restored'); return true; }
+            }));
+            (0, chai_1.expect)(await registry.openSession('1')).to.equal(true);
+            (0, chai_1.expect)(opened).to.deep.equal(['restored']);
+        });
+        it('falls back to the next frontend when the preferred one fails', async () => {
+            const registry = createRegistry();
+            registry.addProvider(provider([], [detail('1', 'running', 200, true)], {
+                openSession: async () => { throw new Error('connection lost'); }
+            }));
+            registry.addProvider(provider([], [detail('1', 'idle', 100, true)], {
+                openSession: async () => true
+            }));
+            (0, chai_1.expect)(await registry.openSession('1')).to.equal(true);
+        });
+        it('keeps the registration order when neither report is preferred', async () => {
+            const registry = createRegistry();
+            const opened = [];
+            registry.addProvider(provider([], [detail('1', 'idle', 100, true)], {
+                openSession: async () => { opened.push('first'); return true; }
+            }));
+            registry.addProvider(provider([], [detail('1', 'idle', 100, true)], {
+                openSession: async () => { opened.push('second'); return true; }
+            }));
+            (0, chai_1.expect)(await registry.openSession('1')).to.equal(true);
+            (0, chai_1.expect)(opened).to.deep.equal(['first']);
+        });
+    });
+    describe('restoreSession', () => {
+        it('returns undefined when no provider knows the session', async () => {
+            const registry = createRegistry();
+            registry.addProvider(provider([]));
+            (0, chai_1.expect)(await registry.restoreSession('missing')).to.equal(undefined);
+        });
+        it('restores the session in the frontend that knows it', async () => {
+            const registry = createRegistry();
+            const restored = detail('1', 'idle', 100, true);
+            registry.addProvider(provider([]));
+            registry.addProvider(provider([], [detail('1', 'idle', 100, false)], {
+                restoreSession: async () => restored
+            }));
+            (0, chai_1.expect)(await registry.restoreSession('1')).to.deep.equal(restored);
+        });
+    });
+    describe('sendPrompt', () => {
+        it('returns undefined when no provider knows the session', async () => {
+            const registry = createRegistry();
+            registry.addProvider(provider([]));
+            (0, chai_1.expect)(await registry.sendPrompt('missing', { text: 'hello' })).to.equal(undefined);
+        });
+        it('sends the prompt via the frontend that knows the session', async () => {
+            const registry = createRegistry();
+            const prompts = [];
+            registry.addProvider(provider([]));
+            registry.addProvider(provider([], [detail('1')], {
+                sendPrompt: async (sessionId, prompt) => { prompts.push(prompt.text); return { sent: { sessionId, requestId: 'r1' } }; }
+            }));
+            (0, chai_1.expect)(await registry.sendPrompt('1', { text: 'hello' })).to.deep.equal({ sent: { sessionId: '1', requestId: 'r1' } });
+            (0, chai_1.expect)(prompts).to.deep.equal(['hello']);
+        });
+        it('does not retry a busy rejection on other frontends', async () => {
+            const registry = createRegistry();
+            const prompted = [];
+            registry.addProvider(provider([], [detail('1', 'running', 100, true)], {
+                sendPrompt: async () => { prompted.push('restored'); return { failure: 'busy' }; }
+            }));
+            registry.addProvider(provider([], [detail('1', 'idle', 300, false)], {
+                sendPrompt: async (sessionId) => { prompted.push('persisted-only'); return { sent: { sessionId, requestId: 'r1' } }; }
+            }));
+            (0, chai_1.expect)(await registry.sendPrompt('1', { text: 'hello' })).to.deep.equal({ failure: 'busy' });
+            (0, chai_1.expect)(prompted).to.deep.equal(['restored']);
+        });
+        it('falls back to the next frontend when the preferred one fails', async () => {
+            const registry = createRegistry();
+            registry.addProvider(provider([], [detail('1', 'running', 200, true)], {
+                sendPrompt: async () => { throw new Error('connection lost'); }
+            }));
+            registry.addProvider(provider([], [detail('1', 'idle', 100, true)], {
+                sendPrompt: async (sessionId) => ({ sent: { sessionId, requestId: 'r1' } })
+            }));
+            (0, chai_1.expect)(await registry.sendPrompt('1', { text: 'hello' })).to.deep.equal({ sent: { sessionId: '1', requestId: 'r1' } });
+        });
+    });
+    describe('createSession', () => {
+        it('rejects when no frontend is connected', async () => {
+            const registry = createRegistry();
+            (0, chai_1.expect)(await registry.createSession({})).to.deep.equal({ failure: 'noFrontend' });
+            (0, chai_1.expect)(await registry.createSession({ workspace: 'file:///project' })).to.deep.equal({ failure: 'noFrontend' });
+        });
+        it('creates in the frontend with the requested workspace', async () => {
+            const registry = createRegistry();
+            const creators = [];
+            registry.addProvider(provider([], [], {
+                getWorkspace: async () => 'file:///other',
+                createSession: async () => { creators.push('a'); return { created: { session: summary('a') } }; }
+            }));
+            registry.addProvider(provider([], [], {
+                getWorkspace: async () => 'file:///project',
+                createSession: async () => { creators.push('b'); return { created: { session: summary('b') } }; }
+            }));
+            const result = await registry.createSession({ workspace: 'file:///project', prompt: 'hello' });
+            (0, chai_1.expect)(result).to.deep.equal({ created: { session: summary('b') } });
+            (0, chai_1.expect)(creators).to.deep.equal(['b']);
+        });
+        it('matches workspaces ignoring trailing slashes', async () => {
+            const registry = createRegistry();
+            registry.addProvider(provider([], [], { getWorkspace: async () => 'file:///project/' }));
+            const result = await registry.createSession({ workspace: 'file:///project' });
+            (0, chai_1.expect)('created' in result).to.equal(true);
+        });
+        it('rejects an unknown workspace', async () => {
+            const registry = createRegistry();
+            registry.addProvider(provider([], [], { getWorkspace: async () => 'file:///other' }));
+            (0, chai_1.expect)(await registry.createSession({ workspace: 'file:///project' })).to.deep.equal({ failure: 'workspaceNotFound' });
+        });
+        it('creates in the only connected frontend when no workspace is given', async () => {
+            const registry = createRegistry();
+            registry.addProvider(provider([], [], { getWorkspace: async () => 'file:///project' }));
+            const result = await registry.createSession({});
+            (0, chai_1.expect)('created' in result).to.equal(true);
+        });
+        it('rejects when no workspace is given and the frontends have different workspaces', async () => {
+            const registry = createRegistry();
+            registry.addProvider(provider([], [], { getWorkspace: async () => 'file:///one' }));
+            registry.addProvider(provider([], [], { getWorkspace: async () => 'file:///two' }));
+            (0, chai_1.expect)(await registry.createSession({})).to.deep.equal({ failure: 'ambiguousWorkspace' });
+        });
+        it('creates when no workspace is given and the frontends share a workspace', async () => {
+            const registry = createRegistry();
+            registry.addProvider(provider([], [], { getWorkspace: async () => 'file:///project' }));
+            registry.addProvider(provider([], [], { getWorkspace: async () => 'file:///project/' }));
+            const result = await registry.createSession({});
+            (0, chai_1.expect)('created' in result).to.equal(true);
+        });
+        it('passes provider failures through', async () => {
+            const registry = createRegistry();
+            registry.addProvider(provider([], [], { createSession: async () => ({ failure: 'unknownAgent' }) }));
+            (0, chai_1.expect)(await registry.createSession({ agentId: 'ghost' })).to.deep.equal({ failure: 'unknownAgent' });
+        });
+    });
+    describe('onDidChangeSessions', () => {
+        it('fires when providers are added, removed, or report changes', () => {
+            const registry = createRegistry();
+            let events = 0;
+            registry.onDidChangeSessions(() => events++);
+            const added = provider([]);
+            registry.addProvider(added);
+            (0, chai_1.expect)(events).to.equal(1);
+            registry.notifySessionsChanged();
+            (0, chai_1.expect)(events).to.equal(2);
+            registry.removeProvider(added);
+            (0, chai_1.expect)(events).to.equal(3);
+        });
+        it('does not fire when removing an unknown provider', () => {
+            const registry = createRegistry();
+            let events = 0;
+            registry.onDidChangeSessions(() => events++);
+            registry.removeProvider(provider([]));
+            (0, chai_1.expect)(events).to.equal(0);
+        });
+    });
+});
+//# sourceMappingURL=external-chat-session-registry.spec.js.map