@office-ai/aioncli-core 0.2.3 → 0.18.4

package/dist/src/utils/events.d.ts

@@ -0,0 +1,121 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { EventEmitter } from 'node:events';
+import type { LoadServerHierarchicalMemoryResponse } from './memoryDiscovery.js';
+/**
+ * Defines the severity level for user-facing feedback.
+ * This maps loosely to UI `MessageType`
+ */
+export type FeedbackSeverity = 'info' | 'warning' | 'error';
+/**
+ * Payload for the 'user-feedback' event.
+ */
+export interface UserFeedbackPayload {
+    /**
+     * The severity level determines how the message is rendered in the UI
+     * (e.g. colored text, specific icon).
+     */
+    severity: FeedbackSeverity;
+    /**
+     * The main message to display to the user in the chat history or stdout.
+     */
+    message: string;
+    /**
+     * The original error object, if applicable.
+     * Listeners can use this to extract stack traces for debug logging
+     * or verbose output, while keeping the 'message' field clean for end users.
+     */
+    error?: unknown;
+}
+/**
+ * Payload for the 'fallback-mode-changed' event.
+ */
+export interface FallbackModeChangedPayload {
+    /**
+     * Whether fallback mode is now active.
+     */
+    isInFallbackMode: boolean;
+}
+/**
+ * Payload for the 'model-changed' event.
+ */
+export interface ModelChangedPayload {
+    /**
+     * The new model that was set.
+     */
+    model: string;
+}
+/**
+ * Payload for the 'console-log' event.
+ */
+export interface ConsoleLogPayload {
+    type: 'log' | 'warn' | 'error' | 'debug' | 'info';
+    content: string;
+}
+/**
+ * Payload for the 'output' event.
+ */
+export interface OutputPayload {
+    isStderr: boolean;
+    chunk: Uint8Array | string;
+    encoding?: BufferEncoding;
+}
+/**
+ * Payload for the 'memory-changed' event.
+ */
+export type MemoryChangedPayload = LoadServerHierarchicalMemoryResponse;
+export declare enum CoreEvent {
+    UserFeedback = "user-feedback",
+    FallbackModeChanged = "fallback-mode-changed",
+    ModelChanged = "model-changed",
+    ConsoleLog = "console-log",
+    Output = "output",
+    MemoryChanged = "memory-changed",
+    ExternalEditorClosed = "external-editor-closed"
+}
+export interface CoreEvents {
+    [CoreEvent.UserFeedback]: [UserFeedbackPayload];
+    [CoreEvent.FallbackModeChanged]: [FallbackModeChangedPayload];
+    [CoreEvent.ModelChanged]: [ModelChangedPayload];
+    [CoreEvent.ConsoleLog]: [ConsoleLogPayload];
+    [CoreEvent.Output]: [OutputPayload];
+    [CoreEvent.MemoryChanged]: [MemoryChangedPayload];
+    [CoreEvent.ExternalEditorClosed]: never[];
+}
+export declare class CoreEventEmitter extends EventEmitter<CoreEvents> {
+    private _eventBacklog;
+    private static readonly MAX_BACKLOG_SIZE;
+    constructor();
+    private _emitOrQueue;
+    /**
+     * Sends actionable feedback to the user.
+     * Buffers automatically if the UI hasn't subscribed yet.
+     */
+    emitFeedback(severity: FeedbackSeverity, message: string, error?: unknown): void;
+    /**
+     * Broadcasts a console log message.
+     */
+    emitConsoleLog(type: 'log' | 'warn' | 'error' | 'debug' | 'info', content: string): void;
+    /**
+     * Broadcasts stdout/stderr output.
+     */
+    emitOutput(isStderr: boolean, chunk: Uint8Array | string, encoding?: BufferEncoding): void;
+    /**
+     * Notifies subscribers that fallback mode has changed.
+     * This is synchronous and doesn't use backlog (UI should already be initialized).
+     */
+    emitFallbackModeChanged(isInFallbackMode: boolean): void;
+    /**
+     * Notifies subscribers that the model has changed.
+     */
+    emitModelChanged(model: string): void;
+    /**
+     * Flushes buffered messages. Call this immediately after primary UI listener
+     * subscribes.
+     */
+    drainBacklogs(): void;
+}
+export declare const coreEvents: CoreEventEmitter;

package/dist/src/utils/events.js

@@ -0,0 +1,84 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { EventEmitter } from 'node:events';
+export var CoreEvent;
+(function (CoreEvent) {
+    CoreEvent["UserFeedback"] = "user-feedback";
+    CoreEvent["FallbackModeChanged"] = "fallback-mode-changed";
+    CoreEvent["ModelChanged"] = "model-changed";
+    CoreEvent["ConsoleLog"] = "console-log";
+    CoreEvent["Output"] = "output";
+    CoreEvent["MemoryChanged"] = "memory-changed";
+    CoreEvent["ExternalEditorClosed"] = "external-editor-closed";
+})(CoreEvent || (CoreEvent = {}));
+export class CoreEventEmitter extends EventEmitter {
+    _eventBacklog = [];
+    static MAX_BACKLOG_SIZE = 10000;
+    constructor() {
+        super();
+    }
+    _emitOrQueue(event, ...args) {
+        if (this.listenerCount(event) === 0) {
+            if (this._eventBacklog.length >= CoreEventEmitter.MAX_BACKLOG_SIZE) {
+                this._eventBacklog.shift();
+            }
+            this._eventBacklog.push({ event, args });
+        }
+        else {
+            this.emit(event, ...args);
+        }
+    }
+    /**
+     * Sends actionable feedback to the user.
+     * Buffers automatically if the UI hasn't subscribed yet.
+     */
+    emitFeedback(severity, message, error) {
+        const payload = { severity, message, error };
+        this._emitOrQueue(CoreEvent.UserFeedback, payload);
+    }
+    /**
+     * Broadcasts a console log message.
+     */
+    emitConsoleLog(type, content) {
+        const payload = { type, content };
+        this._emitOrQueue(CoreEvent.ConsoleLog, payload);
+    }
+    /**
+     * Broadcasts stdout/stderr output.
+     */
+    emitOutput(isStderr, chunk, encoding) {
+        const payload = { isStderr, chunk, encoding };
+        this._emitOrQueue(CoreEvent.Output, payload);
+    }
+    /**
+     * Notifies subscribers that fallback mode has changed.
+     * This is synchronous and doesn't use backlog (UI should already be initialized).
+     */
+    emitFallbackModeChanged(isInFallbackMode) {
+        const payload = { isInFallbackMode };
+        this.emit(CoreEvent.FallbackModeChanged, payload);
+    }
+    /**
+     * Notifies subscribers that the model has changed.
+     */
+    emitModelChanged(model) {
+        const payload = { model };
+        this.emit(CoreEvent.ModelChanged, payload);
+    }
+    /**
+     * Flushes buffered messages. Call this immediately after primary UI listener
+     * subscribes.
+     */
+    drainBacklogs() {
+        const backlog = [...this._eventBacklog];
+        this._eventBacklog.length = 0; // Clear in-place
+        for (const item of backlog) {
+            this.emit(item.event, ...item.args);
+        }
+    }
+}
+export const coreEvents = new CoreEventEmitter();
+//# sourceMappingURL=events.js.map

package/dist/src/utils/events.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"events.js","sourceRoot":"","sources":["../../../src/utils/events.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAwE3C,MAAM,CAAN,IAAY,SAQX;AARD,WAAY,SAAS;IACnB,2CAA8B,CAAA;IAC9B,0DAA6C,CAAA;IAC7C,2CAA8B,CAAA;IAC9B,uCAA0B,CAAA;IAC1B,8BAAiB,CAAA;IACjB,6CAAgC,CAAA;IAChC,4DAA+C,CAAA;AACjD,CAAC,EARW,SAAS,KAAT,SAAS,QAQpB;AAmBD,MAAM,OAAO,gBAAiB,SAAQ,YAAwB;IACpD,aAAa,GAAuB,EAAE,CAAC;IACvC,MAAM,CAAU,gBAAgB,GAAG,KAAK,CAAC;IAEjD;QACE,KAAK,EAAE,CAAC;IACV,CAAC;IAEO,YAAY,CAClB,KAAQ,EACR,GAAG,IAAmB;QAEtB,IAAI,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC;YACpC,IAAI,IAAI,CAAC,aAAa,CAAC,MAAM,IAAI,gBAAgB,CAAC,gBAAgB,EAAE,CAAC;gBACnE,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,CAAC;YAC7B,CAAC;YACD,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,IAAI,EAAsB,CAAC,CAAC;QAC/D,CAAC;aAAM,CAAC;YAEJ,IAAI,CAAC,IAIN,CAAC,KAAK,EAAE,GAAG,IAAI,CAAC,CAAC;QACpB,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,YAAY,CACV,QAA0B,EAC1B,OAAe,EACf,KAAe;QAEf,MAAM,OAAO,GAAwB,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;QAClE,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;IACrD,CAAC;IAED;;OAEG;IACH,cAAc,CACZ,IAAiD,EACjD,OAAe;QAEf,MAAM,OAAO,GAAsB,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;QACrD,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;IACnD,CAAC;IAED;;OAEG;IACH,UAAU,CACR,QAAiB,EACjB,KAA0B,EAC1B,QAAyB;QAEzB,MAAM,OAAO,GAAkB,EAAE,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC;QAC7D,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAC/C,CAAC;IAED;;;OAGG;IACH,uBAAuB,CAAC,gBAAyB;QAC/C,MAAM,OAAO,GAA+B,EAAE,gBAAgB,EAAE,CAAC;QACjE,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,mBAAmB,EAAE,OAAO,CAAC,CAAC;IACpD,CAAC;IAED;;OAEG;IACH,gBAAgB,CAAC,KAAa;QAC5B,MAAM,OAAO,GAAwB,EAAE,KAAK,EAAE,CAAC;QAC/C,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;IAC7C,CAAC;IAED;;;OAGG;IACH,aAAa;QACX,MAAM,OAAO,GAAG,CAAC,GAAG,IAAI,CAAC,aAAa,CAAC,CAAC;QACxC,IAAI,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,iBAAiB;QAChD,KAAK,MAAM,IAAI,IAAI,OAAO,EAAE,CAAC;YAEzB,IAAI,CAAC,IAIN,CAAC,IAAI,CAAC,KAAK,EAAE,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC;QAC9B,CAAC;IACH,CAAC;;AAGH,MAAM,CAAC,MAAM,UAAU,GAAG,IAAI,gBAAgB,EAAE,CAAC"}

package/dist/src/utils/events.test.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export {};

package/dist/src/utils/events.test.js

@@ -0,0 +1,212 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+import { CoreEventEmitter, CoreEvent, } from './events.js';
+describe('CoreEventEmitter', () => {
+    let events;
+    beforeEach(() => {
+        events = new CoreEventEmitter();
+    });
+    it('should emit feedback immediately when a listener is present', () => {
+        const listener = vi.fn();
+        events.on(CoreEvent.UserFeedback, listener);
+        const payload = {
+            severity: 'info',
+            message: 'Test message',
+        };
+        events.emitFeedback(payload.severity, payload.message);
+        expect(listener).toHaveBeenCalledTimes(1);
+        expect(listener).toHaveBeenCalledWith(expect.objectContaining(payload));
+    });
+    it('should buffer feedback when no listener is present', () => {
+        const listener = vi.fn();
+        const payload = {
+            severity: 'warning',
+            message: 'Buffered message',
+        };
+        // Emit while no listeners attached
+        events.emitFeedback(payload.severity, payload.message);
+        expect(listener).not.toHaveBeenCalled();
+        // Attach listener and drain
+        events.on(CoreEvent.UserFeedback, listener);
+        events.drainBacklogs();
+        expect(listener).toHaveBeenCalledTimes(1);
+        expect(listener).toHaveBeenCalledWith(expect.objectContaining(payload));
+    });
+    it('should respect the backlog size limit and maintain FIFO order', () => {
+        const listener = vi.fn();
+        const MAX_BACKLOG_SIZE = 10000;
+        for (let i = 0; i < MAX_BACKLOG_SIZE + 10; i++) {
+            events.emitFeedback('info', `Message ${i}`);
+        }
+        events.on(CoreEvent.UserFeedback, listener);
+        events.drainBacklogs();
+        expect(listener).toHaveBeenCalledTimes(MAX_BACKLOG_SIZE);
+        // Verify strictly that the FIRST call was Message 10 (0-9 dropped)
+        expect(listener.mock.calls[0][0]).toMatchObject({ message: 'Message 10' });
+        // Verify strictly that the LAST call was Message 109
+        expect(listener.mock.lastCall?.[0]).toMatchObject({
+            message: `Message ${MAX_BACKLOG_SIZE + 9}`,
+        });
+    });
+    it('should clear the backlog after draining', () => {
+        const listener = vi.fn();
+        events.emitFeedback('error', 'Test error');
+        events.on(CoreEvent.UserFeedback, listener);
+        events.drainBacklogs();
+        expect(listener).toHaveBeenCalledTimes(1);
+        listener.mockClear();
+        events.drainBacklogs();
+        expect(listener).not.toHaveBeenCalled();
+    });
+    it('should include optional error object in payload', () => {
+        const listener = vi.fn();
+        events.on(CoreEvent.UserFeedback, listener);
+        const error = new Error('Original error');
+        events.emitFeedback('error', 'Something went wrong', error);
+        expect(listener).toHaveBeenCalledWith(expect.objectContaining({
+            severity: 'error',
+            message: 'Something went wrong',
+            error,
+        }));
+    });
+    it('should handle multiple listeners correctly', () => {
+        const listenerA = vi.fn();
+        const listenerB = vi.fn();
+        events.on(CoreEvent.UserFeedback, listenerA);
+        events.on(CoreEvent.UserFeedback, listenerB);
+        events.emitFeedback('info', 'Broadcast message');
+        expect(listenerA).toHaveBeenCalledTimes(1);
+        expect(listenerB).toHaveBeenCalledTimes(1);
+    });
+    it('should stop receiving events after off() is called', () => {
+        const listener = vi.fn();
+        events.on(CoreEvent.UserFeedback, listener);
+        events.emitFeedback('info', 'First message');
+        expect(listener).toHaveBeenCalledTimes(1);
+        events.off(CoreEvent.UserFeedback, listener);
+        events.emitFeedback('info', 'Second message');
+        expect(listener).toHaveBeenCalledTimes(1); // Still 1
+    });
+    it('should handle re-entrant feedback emission during draining safely', () => {
+        events.emitFeedback('info', 'Buffered 1');
+        events.emitFeedback('info', 'Buffered 2');
+        const listener = vi.fn((payload) => {
+            // When 'Buffered 1' is received, immediately emit another event.
+            if (payload.message === 'Buffered 1') {
+                events.emitFeedback('warning', 'Re-entrant message');
+            }
+        });
+        events.on(CoreEvent.UserFeedback, listener);
+        events.drainBacklogs();
+        // Expectation with atomic snapshot:
+        // 1. loop starts with ['Buffered 1', 'Buffered 2']
+        // 2. emits 'Buffered 1'
+        // 3. listener fires for 'Buffered 1', calls emitFeedback('Re-entrant')
+        // 4. emitFeedback sees listener attached, emits 'Re-entrant' synchronously
+        // 5. listener fires for 'Re-entrant'
+        // 6. loop continues, emits 'Buffered 2'
+        // 7. listener fires for 'Buffered 2'
+        expect(listener).toHaveBeenCalledTimes(3);
+        expect(listener.mock.calls[0][0]).toMatchObject({ message: 'Buffered 1' });
+        expect(listener.mock.calls[1][0]).toMatchObject({
+            message: 'Re-entrant message',
+        });
+        expect(listener.mock.calls[2][0]).toMatchObject({ message: 'Buffered 2' });
+    });
+    describe('ConsoleLog Event', () => {
+        it('should emit console log immediately when a listener is present', () => {
+            const listener = vi.fn();
+            events.on(CoreEvent.ConsoleLog, listener);
+            const payload = {
+                type: 'info',
+                content: 'Test log',
+            };
+            events.emitConsoleLog(payload.type, payload.content);
+            expect(listener).toHaveBeenCalledTimes(1);
+            expect(listener).toHaveBeenCalledWith(expect.objectContaining(payload));
+        });
+        it('should buffer console logs when no listener is present', () => {
+            const listener = vi.fn();
+            const payload = {
+                type: 'warn',
+                content: 'Buffered log',
+            };
+            // Emit while no listeners attached
+            events.emitConsoleLog(payload.type, payload.content);
+            expect(listener).not.toHaveBeenCalled();
+            // Attach listener and drain
+            events.on(CoreEvent.ConsoleLog, listener);
+            events.drainBacklogs();
+            expect(listener).toHaveBeenCalledTimes(1);
+            expect(listener).toHaveBeenCalledWith(expect.objectContaining(payload));
+        });
+        it('should respect the backlog size limit for console logs', () => {
+            const listener = vi.fn();
+            const MAX_BACKLOG_SIZE = 10000;
+            for (let i = 0; i < MAX_BACKLOG_SIZE + 10; i++) {
+                events.emitConsoleLog('debug', `Log ${i}`);
+            }
+            events.on(CoreEvent.ConsoleLog, listener);
+            events.drainBacklogs();
+            expect(listener).toHaveBeenCalledTimes(MAX_BACKLOG_SIZE);
+            // Verify strictly that the FIRST call was Log 10 (0-9 dropped)
+            expect(listener.mock.calls[0][0]).toMatchObject({ content: 'Log 10' });
+        });
+    });
+    describe('Output Event', () => {
+        it('should emit output immediately when a listener is present', () => {
+            const listener = vi.fn();
+            events.on(CoreEvent.Output, listener);
+            const payload = {
+                isStderr: false,
+                chunk: 'Test output',
+                encoding: 'utf8',
+            };
+            events.emitOutput(payload.isStderr, payload.chunk, payload.encoding);
+            expect(listener).toHaveBeenCalledTimes(1);
+            expect(listener).toHaveBeenCalledWith(expect.objectContaining(payload));
+        });
+        it('should buffer output when no listener is present', () => {
+            const listener = vi.fn();
+            const payload = {
+                isStderr: true,
+                chunk: 'Buffered output',
+            };
+            // Emit while no listeners attached
+            events.emitOutput(payload.isStderr, payload.chunk);
+            expect(listener).not.toHaveBeenCalled();
+            // Attach listener and drain
+            events.on(CoreEvent.Output, listener);
+            events.drainBacklogs();
+            expect(listener).toHaveBeenCalledTimes(1);
+            expect(listener).toHaveBeenCalledWith(expect.objectContaining(payload));
+        });
+        it('should respect the backlog size limit for output', () => {
+            const listener = vi.fn();
+            const MAX_BACKLOG_SIZE = 10000;
+            for (let i = 0; i < MAX_BACKLOG_SIZE + 10; i++) {
+                events.emitOutput(false, `Output ${i}`);
+            }
+            events.on(CoreEvent.Output, listener);
+            events.drainBacklogs();
+            expect(listener).toHaveBeenCalledTimes(MAX_BACKLOG_SIZE);
+            // Verify strictly that the FIRST call was Output 10 (0-9 dropped)
+            expect(listener.mock.calls[0][0]).toMatchObject({ chunk: 'Output 10' });
+        });
+    });
+    describe('ModelChanged Event', () => {
+        it('should emit ModelChanged event with correct payload', () => {
+            const listener = vi.fn();
+            events.on(CoreEvent.ModelChanged, listener);
+            const newModel = 'gemini-2.5-pro';
+            events.emitModelChanged(newModel);
+            expect(listener).toHaveBeenCalledTimes(1);
+            expect(listener).toHaveBeenCalledWith({ model: newModel });
+        });
+    });
+});
+//# sourceMappingURL=events.test.js.map

package/dist/src/utils/events.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"events.test.js","sourceRoot":"","sources":["../../../src/utils/events.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EACL,gBAAgB,EAChB,SAAS,GAEV,MAAM,aAAa,CAAC;AAErB,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,IAAI,MAAwB,CAAC;IAE7B,UAAU,CAAC,GAAG,EAAE;QACd,MAAM,GAAG,IAAI,gBAAgB,EAAE,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;QACrE,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACzB,MAAM,CAAC,EAAE,CAAC,SAAS,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;QAE5C,MAAM,OAAO,GAAG;YACd,QAAQ,EAAE,MAAe;YACzB,OAAO,EAAE,cAAc;SACxB,CAAC;QAEF,MAAM,CAAC,YAAY,CAAC,OAAO,CAAC,QAAQ,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;QAEvD,MAAM,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC1C,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC,CAAC;IAC1E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACzB,MAAM,OAAO,GAAG;YACd,QAAQ,EAAE,SAAkB;YAC5B,OAAO,EAAE,kBAAkB;SAC5B,CAAC;QAEF,mCAAmC;QACnC,MAAM,CAAC,YAAY,CAAC,OAAO,CAAC,QAAQ,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;QACvD,MAAM,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAExC,4BAA4B;QAC5B,MAAM,CAAC,EAAE,CAAC,SAAS,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;QAC5C,MAAM,CAAC,aAAa,EAAE,CAAC;QAEvB,MAAM,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC1C,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC,CAAC;IAC1E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;QACvE,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACzB,MAAM,gBAAgB,GAAG,KAAK,CAAC;QAE/B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,gBAAgB,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;YAC/C,MAAM,CAAC,YAAY,CAAC,MAAM,EAAE,WAAW,CAAC,EAAE,CAAC,CAAC;QAC9C,CAAC;QAED,MAAM,CAAC,EAAE,CAAC,SAAS,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;QAC5C,MAAM,CAAC,aAAa,EAAE,CAAC;QAEvB,MAAM,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,gBAAgB,CAAC,CAAC;QACzD,mEAAmE;QACnE,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC;QAC3E,qDAAqD;QACrD,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;YAChD,OAAO,EAAE,WAAW,gBAAgB,GAAG,CAAC,EAAE;SAC3C,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACzB,MAAM,CAAC,YAAY,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;QAE3C,MAAM,CAAC,EAAE,CAAC,SAAS,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;QAC5C,MAAM,CAAC,aAAa,EAAE,CAAC;QACvB,MAAM,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAE1C,QAAQ,CAAC,SAAS,EAAE,CAAC;QACrB,MAAM,CAAC,aAAa,EAAE,CAAC;QACvB,MAAM,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;QACzD,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACzB,MAAM,CAAC,EAAE,CAAC,SAAS,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;QAE5C,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,gBAAgB,CAAC,CAAC;QAC1C,MAAM,CAAC,YAAY,CAAC,OAAO,EAAE,sBAAsB,EAAE,KAAK,CAAC,CAAC;QAE5D,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CACnC,MAAM,CAAC,gBAAgB,CAAC;YACtB,QAAQ,EAAE,OAAO;YACjB,OAAO,EAAE,sBAAsB;YAC/B,KAAK;SACN,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAC1B,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAE1B,MAAM,CAAC,EAAE,CAAC,SAAS,CAAC,YAAY,EAAE,SAAS,CAAC,CAAC;QAC7C,MAAM,CAAC,EAAE,CAAC,SAAS,CAAC,YAAY,EAAE,SAAS,CAAC,CAAC;QAE7C,MAAM,CAAC,YAAY,CAAC,MAAM,EAAE,mBAAmB,CAAC,CAAC;QAEjD,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC3C,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACzB,MAAM,CAAC,EAAE,CAAC,SAAS,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;QAE5C,MAAM,CAAC,YAAY,CAAC,MAAM,EAAE,eAAe,CAAC,CAAC;QAC7C,MAAM,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAE1C,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;QAC7C,MAAM,CAAC,YAAY,CAAC,MAAM,EAAE,gBAAgB,CAAC,CAAC;QAC9C,MAAM,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,CAAC,UAAU;IACvD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mEAAmE,EAAE,GAAG,EAAE;QAC3E,MAAM,CAAC,YAAY,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;QAC1C,MAAM,CAAC,YAAY,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;QAE1C,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC,OAA4B,EAAE,EAAE;YACtD,iEAAiE;YACjE,IAAI,OAAO,CAAC,OAAO,KAAK,YAAY,EAAE,CAAC;gBACrC,MAAM,CAAC,YAAY,CAAC,SAAS,EAAE,oBAAoB,CAAC,CAAC;YACvD,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,EAAE,CAAC,SAAS,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;QAC5C,MAAM,CAAC,aAAa,EAAE,CAAC;QAEvB,oCAAoC;QACpC,mDAAmD;QACnD,wBAAwB;QACxB,uEAAuE;QACvE,2EAA2E;QAC3E,qCAAqC;QACrC,wCAAwC;QACxC,qCAAqC;QAErC,MAAM,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC1C,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC;QAC3E,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;YAC9C,OAAO,EAAE,oBAAoB;SAC9B,CAAC,CAAC;QACH,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC;IAC7E,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,gEAAgE,EAAE,GAAG,EAAE;YACxE,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YACzB,MAAM,CAAC,EAAE,CAAC,SAAS,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;YAE1C,MAAM,OAAO,GAAG;gBACd,IAAI,EAAE,MAAe;gBACrB,OAAO,EAAE,UAAU;aACpB,CAAC;YAEF,MAAM,CAAC,cAAc,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;YAErD,MAAM,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC1C,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC,CAAC;QAC1E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;YAChE,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YACzB,MAAM,OAAO,GAAG;gBACd,IAAI,EAAE,MAAe;gBACrB,OAAO,EAAE,cAAc;aACxB,CAAC;YAEF,mCAAmC;YACnC,MAAM,CAAC,cAAc,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;YACrD,MAAM,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YAExC,4BAA4B;YAC5B,MAAM,CAAC,EAAE,CAAC,SAAS,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;YAC1C,MAAM,CAAC,aAAa,EAAE,CAAC;YAEvB,MAAM,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC1C,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC,CAAC;QAC1E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;YAChE,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YACzB,MAAM,gBAAgB,GAAG,KAAK,CAAC;YAE/B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,gBAAgB,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC/C,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC;YAC7C,CAAC;YAED,MAAM,CAAC,EAAE,CAAC,SAAS,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;YAC1C,MAAM,CAAC,aAAa,EAAE,CAAC;YAEvB,MAAM,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,gBAAgB,CAAC,CAAC;YACzD,+DAA+D;YAC/D,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAC;QACzE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;YACnE,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YACzB,MAAM,CAAC,EAAE,CAAC,SAAS,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;YAEtC,MAAM,OAAO,GAAG;gBACd,QAAQ,EAAE,KAAK;gBACf,KAAK,EAAE,aAAa;gBACpB,QAAQ,EAAE,MAAwB;aACnC,CAAC;YAEF,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,QAAQ,EAAE,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC;YAErE,MAAM,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC1C,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC,CAAC;QAC1E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YACzB,MAAM,OAAO,GAAG;gBACd,QAAQ,EAAE,IAAI;gBACd,KAAK,EAAE,iBAAiB;aACzB,CAAC;YAEF,mCAAmC;YACnC,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,QAAQ,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;YACnD,MAAM,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YAExC,4BAA4B;YAC5B,MAAM,CAAC,EAAE,CAAC,SAAS,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;YACtC,MAAM,CAAC,aAAa,EAAE,CAAC;YAEvB,MAAM,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC1C,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC,CAAC;QAC1E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YACzB,MAAM,gBAAgB,GAAG,KAAK,CAAC;YAE/B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,gBAAgB,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC/C,MAAM,CAAC,UAAU,CAAC,KAAK,EAAE,UAAU,CAAC,EAAE,CAAC,CAAC;YAC1C,CAAC;YAED,MAAM,CAAC,EAAE,CAAC,SAAS,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;YACtC,MAAM,CAAC,aAAa,EAAE,CAAC;YAEvB,MAAM,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,gBAAgB,CAAC,CAAC;YACzD,kEAAkE;YAClE,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;QAC1E,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YACzB,MAAM,CAAC,EAAE,CAAC,SAAS,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;YAE5C,MAAM,QAAQ,GAAG,gBAAgB,CAAC;YAClC,MAAM,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;YAElC,MAAM,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC1C,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/utils/extensionLoader.d.ts

@@ -0,0 +1,86 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { EventEmitter } from 'node:events';
+import type { Config, GeminiCLIExtension } from '../config/config.js';
+export declare abstract class ExtensionLoader {
+    private readonly eventEmitter?;
+    protected config: Config | undefined;
+    protected startingCount: number;
+    protected startCompletedCount: number;
+    protected stoppingCount: number;
+    protected stopCompletedCount: number;
+    private isStarting;
+    constructor(eventEmitter?: EventEmitter<ExtensionEvents> | undefined);
+    /**
+     * All currently known extensions, both active and inactive.
+     */
+    abstract getExtensions(): GeminiCLIExtension[];
+    /**
+     * Fully initializes all active extensions.
+     *
+     * Called within `Config.initialize`, which must already have an
+     * McpClientManager, PromptRegistry, and GeminiChat set up.
+     */
+    start(config: Config): Promise<void>;
+    /**
+     * Unconditionally starts an `extension` and loads all its MCP servers,
+     * context, custom commands, etc. Assumes that `start` has already been called
+     * and we have a Config object.
+     *
+     * This should typically only be called from `start`, most other calls should
+     * go through `maybeStartExtension` which will only start the extension if
+     * extension reloading is enabled and the `config` object is initialized.
+     */
+    protected startExtension(extension: GeminiCLIExtension): Promise<void>;
+    private maybeRefreshMemories;
+    /**
+     * Refreshes the gemini tools list if it is initialized and the extension has
+     * any excludeTools settings.
+     */
+    private maybeRefreshGeminiTools;
+    /**
+     * If extension reloading is enabled and `start` has already been called,
+     * then calls `startExtension` to include all extension features into the
+     * program.
+     */
+    protected maybeStartExtension(extension: GeminiCLIExtension): Promise<void> | undefined;
+    /**
+     * Unconditionally stops an `extension` and unloads all its MCP servers,
+     * context, custom commands, etc. Assumes that `start` has already been called
+     * and we have a Config object.
+     *
+     * Most calls should go through `maybeStopExtension` which will only stop the
+     * extension if extension reloading is enabled and the `config` object is
+     * initialized.
+     */
+    protected stopExtension(extension: GeminiCLIExtension): Promise<void>;
+    /**
+     * If extension reloading is enabled and `start` has already been called,
+     * then this also performs all necessary steps to remove all extension
+     * features from the rest of the system.
+     */
+    protected maybeStopExtension(extension: GeminiCLIExtension): Promise<void> | undefined;
+    restartExtension(extension: GeminiCLIExtension): Promise<void>;
+}
+export interface ExtensionEvents {
+    extensionsStarting: ExtensionsStartingEvent[];
+    extensionsStopping: ExtensionsStoppingEvent[];
+}
+export interface ExtensionsStartingEvent {
+    total: number;
+    completed: number;
+}
+export interface ExtensionsStoppingEvent {
+    total: number;
+    completed: number;
+}
+export declare class SimpleExtensionLoader extends ExtensionLoader {
+    protected readonly extensions: GeminiCLIExtension[];
+    constructor(extensions: GeminiCLIExtension[], eventEmitter?: EventEmitter<ExtensionEvents>);
+    getExtensions(): GeminiCLIExtension[];
+    loadExtension(extension: GeminiCLIExtension): Promise<void>;
+    unloadExtension(extension: GeminiCLIExtension): Promise<void>;
+}