@office-ai/aioncli-core 0.2.3 → 0.18.4

package/dist/src/hooks/types.test.js

@@ -0,0 +1,313 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi } from 'vitest';
+import { createHookOutput, DefaultHookOutput, BeforeToolHookOutput, BeforeModelHookOutput, BeforeToolSelectionHookOutput, AfterModelHookOutput, HookEventName, HookType, } from './types.js';
+import { defaultHookTranslator } from './hookTranslator.js';
+vi.mock('./hookTranslator.js', () => ({
+    defaultHookTranslator: {
+        fromHookLLMResponse: vi.fn((response) => response),
+        fromHookLLMRequest: vi.fn((request, target) => ({
+            ...target,
+            ...request,
+        })),
+        fromHookToolConfig: vi.fn((config) => config),
+    },
+}));
+describe('Hook Types', () => {
+    describe('HookEventName', () => {
+        it('should contain all required event names', () => {
+            const expectedEvents = [
+                'BeforeTool',
+                'AfterTool',
+                'BeforeAgent',
+                'Notification',
+                'AfterAgent',
+                'SessionStart',
+                'SessionEnd',
+                'PreCompress',
+                'BeforeModel',
+                'AfterModel',
+                'BeforeToolSelection',
+            ];
+            for (const event of expectedEvents) {
+                expect(Object.values(HookEventName)).toContain(event);
+            }
+        });
+    });
+    describe('HookType', () => {
+        it('should contain command type', () => {
+            expect(HookType.Command).toBe('command');
+        });
+    });
+});
+describe('Hook Output Classes', () => {
+    describe('createHookOutput', () => {
+        it('should return DefaultHookOutput for unknown event names', () => {
+            const output = createHookOutput('UnknownEvent', {});
+            expect(output).toBeInstanceOf(DefaultHookOutput);
+            expect(output).not.toBeInstanceOf(BeforeModelHookOutput);
+            expect(output).not.toBeInstanceOf(AfterModelHookOutput);
+            expect(output).not.toBeInstanceOf(BeforeToolSelectionHookOutput);
+        });
+        it('should return BeforeModelHookOutput for BeforeModel event', () => {
+            const output = createHookOutput(HookEventName.BeforeModel, {});
+            expect(output).toBeInstanceOf(BeforeModelHookOutput);
+        });
+        it('should return AfterModelHookOutput for AfterModel event', () => {
+            const output = createHookOutput(HookEventName.AfterModel, {});
+            expect(output).toBeInstanceOf(AfterModelHookOutput);
+        });
+        it('should return BeforeToolSelectionHookOutput for BeforeToolSelection event', () => {
+            const output = createHookOutput(HookEventName.BeforeToolSelection, {});
+            expect(output).toBeInstanceOf(BeforeToolSelectionHookOutput);
+        });
+    });
+    describe('DefaultHookOutput', () => {
+        it('should construct with provided data', () => {
+            const data = {
+                continue: false,
+                stopReason: 'test stop',
+                suppressOutput: true,
+                systemMessage: 'test system message',
+                decision: 'block',
+                reason: 'test reason',
+                hookSpecificOutput: { key: 'value' },
+            };
+            const output = new DefaultHookOutput(data);
+            expect(output.continue).toBe(data.continue);
+            expect(output.stopReason).toBe(data.stopReason);
+            expect(output.suppressOutput).toBe(data.suppressOutput);
+            expect(output.systemMessage).toBe(data.systemMessage);
+            expect(output.decision).toBe(data.decision);
+            expect(output.reason).toBe(data.reason);
+            expect(output.hookSpecificOutput).toEqual(data.hookSpecificOutput);
+        });
+        it('should return false for isBlockingDecision if decision is not block or deny', () => {
+            const output1 = new DefaultHookOutput({ decision: 'approve' });
+            expect(output1.isBlockingDecision()).toBe(false);
+            const output2 = new DefaultHookOutput({ decision: undefined });
+            expect(output2.isBlockingDecision()).toBe(false);
+        });
+        it('should return true for isBlockingDecision if decision is block or deny', () => {
+            const output1 = new DefaultHookOutput({ decision: 'block' });
+            expect(output1.isBlockingDecision()).toBe(true);
+            const output2 = new DefaultHookOutput({ decision: 'deny' });
+            expect(output2.isBlockingDecision()).toBe(true);
+        });
+        it('should return true for shouldStopExecution if continue is false', () => {
+            const output = new DefaultHookOutput({ continue: false });
+            expect(output.shouldStopExecution()).toBe(true);
+        });
+        it('should return false for shouldStopExecution if continue is true or undefined', () => {
+            const output1 = new DefaultHookOutput({ continue: true });
+            expect(output1.shouldStopExecution()).toBe(false);
+            const output2 = new DefaultHookOutput({});
+            expect(output2.shouldStopExecution()).toBe(false);
+        });
+        it('should return reason if available', () => {
+            const output = new DefaultHookOutput({ reason: 'specific reason' });
+            expect(output.getEffectiveReason()).toBe('specific reason');
+        });
+        it('should return stopReason if reason is not available', () => {
+            const output = new DefaultHookOutput({ stopReason: 'stop reason' });
+            expect(output.getEffectiveReason()).toBe('stop reason');
+        });
+        it('should return "No reason provided" if neither reason nor stopReason are available', () => {
+            const output = new DefaultHookOutput({});
+            expect(output.getEffectiveReason()).toBe('No reason provided');
+        });
+        it('applyLLMRequestModifications should return target unchanged', () => {
+            const target = {
+                model: 'gemini-pro',
+                contents: [],
+            };
+            const output = new DefaultHookOutput({});
+            expect(output.applyLLMRequestModifications(target)).toBe(target);
+        });
+        it('applyToolConfigModifications should return target unchanged', () => {
+            const target = { toolConfig: {}, tools: [] };
+            const output = new DefaultHookOutput({});
+            expect(output.applyToolConfigModifications(target)).toBe(target);
+        });
+        it('getAdditionalContext should return additionalContext if present', () => {
+            const output = new DefaultHookOutput({
+                hookSpecificOutput: { additionalContext: 'some context' },
+            });
+            expect(output.getAdditionalContext()).toBe('some context');
+        });
+        it('getAdditionalContext should return undefined if additionalContext is not present', () => {
+            const output = new DefaultHookOutput({
+                hookSpecificOutput: { other: 'value' },
+            });
+            expect(output.getAdditionalContext()).toBeUndefined();
+        });
+        it('getAdditionalContext should return undefined if hookSpecificOutput is undefined', () => {
+            const output = new DefaultHookOutput({});
+            expect(output.getAdditionalContext()).toBeUndefined();
+        });
+        it('getBlockingError should return blocked: true and reason if blocking decision', () => {
+            const output = new DefaultHookOutput({
+                decision: 'block',
+                reason: 'blocked by hook',
+            });
+            expect(output.getBlockingError()).toEqual({
+                blocked: true,
+                reason: 'blocked by hook',
+            });
+        });
+        it('getBlockingError should return blocked: false if not blocking decision', () => {
+            const output = new DefaultHookOutput({ decision: 'approve' });
+            expect(output.getBlockingError()).toEqual({ blocked: false, reason: '' });
+        });
+    });
+    describe('BeforeToolHookOutput', () => {
+        it('isBlockingDecision should use permissionDecision from hookSpecificOutput', () => {
+            const output1 = new BeforeToolHookOutput({
+                hookSpecificOutput: { permissionDecision: 'block' },
+            });
+            expect(output1.isBlockingDecision()).toBe(true);
+            const output2 = new BeforeToolHookOutput({
+                hookSpecificOutput: { permissionDecision: 'approve' },
+            });
+            expect(output2.isBlockingDecision()).toBe(false);
+        });
+        it('getEffectiveReason should use permissionDecisionReason from hookSpecificOutput', () => {
+            const output1 = new BeforeToolHookOutput({
+                hookSpecificOutput: { permissionDecisionReason: 'compat reason' },
+            });
+            expect(output1.getEffectiveReason()).toBe('compat reason');
+            const output2 = new BeforeToolHookOutput({
+                reason: 'default reason',
+                hookSpecificOutput: { other: 'value' },
+            });
+            expect(output2.getEffectiveReason()).toBe('default reason');
+        });
+    });
+    describe('BeforeModelHookOutput', () => {
+        it('getSyntheticResponse should return synthetic response if llm_response is present', () => {
+            const mockResponse = { candidates: [] };
+            const output = new BeforeModelHookOutput({
+                hookSpecificOutput: { llm_response: mockResponse },
+            });
+            expect(output.getSyntheticResponse()).toEqual(mockResponse);
+            expect(defaultHookTranslator.fromHookLLMResponse).toHaveBeenCalledWith(mockResponse);
+        });
+        it('getSyntheticResponse should return undefined if llm_response is not present', () => {
+            const output = new BeforeModelHookOutput({});
+            expect(output.getSyntheticResponse()).toBeUndefined();
+        });
+        it('applyLLMRequestModifications should apply modifications if llm_request is present', () => {
+            const target = {
+                model: 'gemini-pro',
+                contents: [{ parts: [{ text: 'original' }] }],
+            };
+            const mockRequest = {
+                messages: [{ role: 'user', content: 'modified' }],
+            };
+            const output = new BeforeModelHookOutput({
+                hookSpecificOutput: { llm_request: mockRequest },
+            });
+            const result = output.applyLLMRequestModifications(target);
+            expect(result).toEqual({ ...target, ...mockRequest });
+            expect(defaultHookTranslator.fromHookLLMRequest).toHaveBeenCalledWith(mockRequest, target);
+        });
+        it('applyLLMRequestModifications should return target unchanged if llm_request is not present', () => {
+            const target = {
+                model: 'gemini-pro',
+                contents: [],
+            };
+            const output = new BeforeModelHookOutput({});
+            expect(output.applyLLMRequestModifications(target)).toBe(target);
+        });
+    });
+    describe('BeforeToolSelectionHookOutput', () => {
+        it('applyToolConfigModifications should apply modifications if toolConfig is present', () => {
+            const target = { tools: [{ functionDeclarations: [] }] };
+            const mockToolConfig = { functionCallingConfig: { mode: 'ANY' } };
+            const output = new BeforeToolSelectionHookOutput({
+                hookSpecificOutput: { toolConfig: mockToolConfig },
+            });
+            const result = output.applyToolConfigModifications(target);
+            expect(result).toEqual({ ...target, toolConfig: mockToolConfig });
+            expect(defaultHookTranslator.fromHookToolConfig).toHaveBeenCalledWith(mockToolConfig);
+        });
+        it('applyToolConfigModifications should return target unchanged if toolConfig is not present', () => {
+            const target = { toolConfig: {}, tools: [] };
+            const output = new BeforeToolSelectionHookOutput({});
+            expect(output.applyToolConfigModifications(target)).toBe(target);
+        });
+        it('applyToolConfigModifications should initialize tools array if not present', () => {
+            const target = {};
+            const mockToolConfig = { functionCallingConfig: { mode: 'ANY' } };
+            const output = new BeforeToolSelectionHookOutput({
+                hookSpecificOutput: { toolConfig: mockToolConfig },
+            });
+            const result = output.applyToolConfigModifications(target);
+            expect(result).toEqual({ tools: [], toolConfig: mockToolConfig });
+        });
+    });
+    describe('AfterModelHookOutput', () => {
+        it('getModifiedResponse should return modified response if llm_response is present and has content', () => {
+            const mockResponse = {
+                candidates: [{ content: { role: 'model', parts: ['modified'] } }],
+            };
+            const output = new AfterModelHookOutput({
+                hookSpecificOutput: { llm_response: mockResponse },
+            });
+            expect(output.getModifiedResponse()).toEqual(mockResponse);
+            expect(defaultHookTranslator.fromHookLLMResponse).toHaveBeenCalledWith(mockResponse);
+        });
+        it('getModifiedResponse should return undefined if llm_response is present but no content', () => {
+            const mockResponse = {
+                candidates: [{ content: { role: 'model', parts: [] } }],
+            };
+            const output = new AfterModelHookOutput({
+                hookSpecificOutput: { llm_response: mockResponse },
+            });
+            expect(output.getModifiedResponse()).toBeUndefined();
+        });
+        it('getModifiedResponse should return undefined if llm_response is not present', () => {
+            const output = new AfterModelHookOutput({});
+            expect(output.getModifiedResponse()).toBeUndefined();
+        });
+        it('getModifiedResponse should return a synthetic stop response if shouldStopExecution is true', () => {
+            const output = new AfterModelHookOutput({
+                continue: false,
+                stopReason: 'stopped by hook',
+            });
+            const expectedResponse = {
+                candidates: [
+                    {
+                        content: {
+                            role: 'model',
+                            parts: ['stopped by hook'],
+                        },
+                        finishReason: 'STOP',
+                    },
+                ],
+            };
+            expect(output.getModifiedResponse()).toEqual(expectedResponse);
+            expect(defaultHookTranslator.fromHookLLMResponse).toHaveBeenCalledWith(expectedResponse);
+        });
+        it('getModifiedResponse should return a synthetic stop response with default reason if shouldStopExecution is true and no stopReason', () => {
+            const output = new AfterModelHookOutput({ continue: false });
+            const expectedResponse = {
+                candidates: [
+                    {
+                        content: {
+                            role: 'model',
+                            parts: ['No reason provided'],
+                        },
+                        finishReason: 'STOP',
+                    },
+                ],
+            };
+            expect(output.getModifiedResponse()).toEqual(expectedResponse);
+            expect(defaultHookTranslator.fromHookLLMResponse).toHaveBeenCalledWith(expectedResponse);
+        });
+    });
+});
+//# sourceMappingURL=types.test.js.map

package/dist/src/hooks/types.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.test.js","sourceRoot":"","sources":["../../../src/hooks/types.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAClD,OAAO,EACL,gBAAgB,EAChB,iBAAiB,EACjB,oBAAoB,EACpB,qBAAqB,EACrB,6BAA6B,EAC7B,oBAAoB,EACpB,aAAa,EACb,QAAQ,GACT,MAAM,YAAY,CAAC;AACpB,OAAO,EAAE,qBAAqB,EAAE,MAAM,qBAAqB,CAAC;AAS5D,EAAE,CAAC,IAAI,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC;IACpC,qBAAqB,EAAE;QACrB,mBAAmB,EAAE,EAAE,CAAC,EAAE,CACxB,CAAC,QAAqB,EAAE,EAAE,CAAC,QAA8C,CAC1E;QACD,kBAAkB,EAAE,EAAE,CAAC,EAAE,CACvB,CAAC,OAAmB,EAAE,MAAiC,EAAE,EAAE,CAAC,CAAC;YAC3D,GAAG,MAAM;YACT,GAAG,OAAO;SACX,CAAC,CACH;QACD,kBAAkB,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,MAAkB,EAAE,EAAE,CAAC,MAAM,CAAC;KAC1D;CACF,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,MAAM,cAAc,GAAG;gBACrB,YAAY;gBACZ,WAAW;gBACX,aAAa;gBACb,cAAc;gBACd,YAAY;gBACZ,cAAc;gBACd,YAAY;gBACZ,aAAa;gBACb,aAAa;gBACb,YAAY;gBACZ,qBAAqB;aACtB,CAAC;YAEF,KAAK,MAAM,KAAK,IAAI,cAAc,EAAE,CAAC;gBACnC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;YACxD,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;YACrC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;YACjE,MAAM,MAAM,GAAG,gBAAgB,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC;YACpD,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,iBAAiB,CAAC,CAAC;YACjD,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,qBAAqB,CAAC,CAAC;YACzD,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,oBAAoB,CAAC,CAAC;YACxD,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,6BAA6B,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;YACnE,MAAM,MAAM,GAAG,gBAAgB,CAAC,aAAa,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;YAC/D,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,qBAAqB,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;YACjE,MAAM,MAAM,GAAG,gBAAgB,CAAC,aAAa,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;YAC9D,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,oBAAoB,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2EAA2E,EAAE,GAAG,EAAE;YACnF,MAAM,MAAM,GAAG,gBAAgB,CAAC,aAAa,CAAC,mBAAmB,EAAE,EAAE,CAAC,CAAC;YACvE,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,6BAA6B,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,IAAI,GAAG;gBACX,QAAQ,EAAE,KAAK;gBACf,UAAU,EAAE,WAAW;gBACvB,cAAc,EAAE,IAAI;gBACpB,aAAa,EAAE,qBAAqB;gBACpC,QAAQ,EAAE,OAAuB;gBACjC,MAAM,EAAE,aAAa;gBACrB,kBAAkB,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE;aACrC,CAAC;YACF,MAAM,MAAM,GAAG,IAAI,iBAAiB,CAAC,IAAI,CAAC,CAAC;YAC3C,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC5C,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAChD,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YACxD,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YACtD,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC5C,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACxC,MAAM,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;QACrE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6EAA6E,EAAE,GAAG,EAAE;YACrF,MAAM,OAAO,GAAG,IAAI,iBAAiB,CAAC,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,CAAC;YAC/D,MAAM,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACjD,MAAM,OAAO,GAAG,IAAI,iBAAiB,CAAC,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,CAAC;YAC/D,MAAM,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wEAAwE,EAAE,GAAG,EAAE;YAChF,MAAM,OAAO,GAAG,IAAI,iBAAiB,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC;YAC7D,MAAM,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAChD,MAAM,OAAO,GAAG,IAAI,iBAAiB,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;YAC5D,MAAM,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;YACzE,MAAM,MAAM,GAAG,IAAI,iBAAiB,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;YAC1D,MAAM,CAAC,MAAM,CAAC,mBAAmB,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8EAA8E,EAAE,GAAG,EAAE;YACtF,MAAM,OAAO,GAAG,IAAI,iBAAiB,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;YAC1D,MAAM,CAAC,OAAO,CAAC,mBAAmB,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAClD,MAAM,OAAO,GAAG,IAAI,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAC1C,MAAM,CAAC,OAAO,CAAC,mBAAmB,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;YAC3C,MAAM,MAAM,GAAG,IAAI,iBAAiB,CAAC,EAAE,MAAM,EAAE,iBAAiB,EAAE,CAAC,CAAC;YACpE,MAAM,CAAC,MAAM,CAAC,kBAAkB,EAAE,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,MAAM,MAAM,GAAG,IAAI,iBAAiB,CAAC,EAAE,UAAU,EAAE,aAAa,EAAE,CAAC,CAAC;YACpE,MAAM,CAAC,MAAM,CAAC,kBAAkB,EAAE,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mFAAmF,EAAE,GAAG,EAAE;YAC3F,MAAM,MAAM,GAAG,IAAI,iBAAiB,CAAC,EAAE,CAAC,CAAC;YACzC,MAAM,CAAC,MAAM,CAAC,kBAAkB,EAAE,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;YACrE,MAAM,MAAM,GAA8B;gBACxC,KAAK,EAAE,YAAY;gBACnB,QAAQ,EAAE,EAAE;aACb,CAAC;YACF,MAAM,MAAM,GAAG,IAAI,iBAAiB,CAAC,EAAE,CAAC,CAAC;YACzC,MAAM,CAAC,MAAM,CAAC,4BAA4B,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;YACrE,MAAM,MAAM,GAAG,EAAE,UAAU,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;YAC7C,MAAM,MAAM,GAAG,IAAI,iBAAiB,CAAC,EAAE,CAAC,CAAC;YACzC,MAAM,CAAC,MAAM,CAAC,4BAA4B,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;YACzE,MAAM,MAAM,GAAG,IAAI,iBAAiB,CAAC;gBACnC,kBAAkB,EAAE,EAAE,iBAAiB,EAAE,cAAc,EAAE;aAC1D,CAAC,CAAC;YACH,MAAM,CAAC,MAAM,CAAC,oBAAoB,EAAE,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kFAAkF,EAAE,GAAG,EAAE;YAC1F,MAAM,MAAM,GAAG,IAAI,iBAAiB,CAAC;gBACnC,kBAAkB,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE;aACvC,CAAC,CAAC;YACH,MAAM,CAAC,MAAM,CAAC,oBAAoB,EAAE,CAAC,CAAC,aAAa,EAAE,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iFAAiF,EAAE,GAAG,EAAE;YACzF,MAAM,MAAM,GAAG,IAAI,iBAAiB,CAAC,EAAE,CAAC,CAAC;YACzC,MAAM,CAAC,MAAM,CAAC,oBAAoB,EAAE,CAAC,CAAC,aAAa,EAAE,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8EAA8E,EAAE,GAAG,EAAE;YACtF,MAAM,MAAM,GAAG,IAAI,iBAAiB,CAAC;gBACnC,QAAQ,EAAE,OAAO;gBACjB,MAAM,EAAE,iBAAiB;aAC1B,CAAC,CAAC;YACH,MAAM,CAAC,MAAM,CAAC,gBAAgB,EAAE,CAAC,CAAC,OAAO,CAAC;gBACxC,OAAO,EAAE,IAAI;gBACb,MAAM,EAAE,iBAAiB;aAC1B,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wEAAwE,EAAE,GAAG,EAAE;YAChF,MAAM,MAAM,GAAG,IAAI,iBAAiB,CAAC,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,CAAC;YAC9D,MAAM,CAAC,MAAM,CAAC,gBAAgB,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;QAC5E,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,0EAA0E,EAAE,GAAG,EAAE;YAClF,MAAM,OAAO,GAAG,IAAI,oBAAoB,CAAC;gBACvC,kBAAkB,EAAE,EAAE,kBAAkB,EAAE,OAAO,EAAE;aACpD,CAAC,CAAC;YACH,MAAM,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAEhD,MAAM,OAAO,GAAG,IAAI,oBAAoB,CAAC;gBACvC,kBAAkB,EAAE,EAAE,kBAAkB,EAAE,SAAS,EAAE;aACtD,CAAC,CAAC;YACH,MAAM,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gFAAgF,EAAE,GAAG,EAAE;YACxF,MAAM,OAAO,GAAG,IAAI,oBAAoB,CAAC;gBACvC,kBAAkB,EAAE,EAAE,wBAAwB,EAAE,eAAe,EAAE;aAClE,CAAC,CAAC;YACH,MAAM,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YAE3D,MAAM,OAAO,GAAG,IAAI,oBAAoB,CAAC;gBACvC,MAAM,EAAE,gBAAgB;gBACxB,kBAAkB,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE;aACvC,CAAC,CAAC;YACH,MAAM,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,kFAAkF,EAAE,GAAG,EAAE;YAC1F,MAAM,YAAY,GAAgB,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;YACrD,MAAM,MAAM,GAAG,IAAI,qBAAqB,CAAC;gBACvC,kBAAkB,EAAE,EAAE,YAAY,EAAE,YAAY,EAAE;aACnD,CAAC,CAAC;YACH,MAAM,CAAC,MAAM,CAAC,oBAAoB,EAAE,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;YAC5D,MAAM,CAAC,qBAAqB,CAAC,mBAAmB,CAAC,CAAC,oBAAoB,CACpE,YAAY,CACb,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6EAA6E,EAAE,GAAG,EAAE;YACrF,MAAM,MAAM,GAAG,IAAI,qBAAqB,CAAC,EAAE,CAAC,CAAC;YAC7C,MAAM,CAAC,MAAM,CAAC,oBAAoB,EAAE,CAAC,CAAC,aAAa,EAAE,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mFAAmF,EAAE,GAAG,EAAE;YAC3F,MAAM,MAAM,GAA8B;gBACxC,KAAK,EAAE,YAAY;gBACnB,QAAQ,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC;aAC9C,CAAC;YACF,MAAM,WAAW,GAAwB;gBACvC,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC;aAClD,CAAC;YACF,MAAM,MAAM,GAAG,IAAI,qBAAqB,CAAC;gBACvC,kBAAkB,EAAE,EAAE,WAAW,EAAE,WAAW,EAAE;aACjD,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,CAAC,4BAA4B,CAAC,MAAM,CAAC,CAAC;YAC3D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,GAAG,MAAM,EAAE,GAAG,WAAW,EAAE,CAAC,CAAC;YACtD,MAAM,CAAC,qBAAqB,CAAC,kBAAkB,CAAC,CAAC,oBAAoB,CACnE,WAAW,EACX,MAAM,CACP,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2FAA2F,EAAE,GAAG,EAAE;YACnG,MAAM,MAAM,GAA8B;gBACxC,KAAK,EAAE,YAAY;gBACnB,QAAQ,EAAE,EAAE;aACb,CAAC;YACF,MAAM,MAAM,GAAG,IAAI,qBAAqB,CAAC,EAAE,CAAC,CAAC;YAC7C,MAAM,CAAC,MAAM,CAAC,4BAA4B,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;QAC7C,EAAE,CAAC,kFAAkF,EAAE,GAAG,EAAE;YAC1F,MAAM,MAAM,GAAG,EAAE,KAAK,EAAE,CAAC,EAAE,oBAAoB,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC;YACzD,MAAM,cAAc,GAAG,EAAE,qBAAqB,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;YAClE,MAAM,MAAM,GAAG,IAAI,6BAA6B,CAAC;gBAC/C,kBAAkB,EAAE,EAAE,UAAU,EAAE,cAAc,EAAE;aACnD,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,CAAC,4BAA4B,CAAC,MAAM,CAAC,CAAC;YAC3D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,GAAG,MAAM,EAAE,UAAU,EAAE,cAAc,EAAE,CAAC,CAAC;YAClE,MAAM,CAAC,qBAAqB,CAAC,kBAAkB,CAAC,CAAC,oBAAoB,CACnE,cAAc,CACf,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0FAA0F,EAAE,GAAG,EAAE;YAClG,MAAM,MAAM,GAAG,EAAE,UAAU,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;YAC7C,MAAM,MAAM,GAAG,IAAI,6BAA6B,CAAC,EAAE,CAAC,CAAC;YACrD,MAAM,CAAC,MAAM,CAAC,4BAA4B,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2EAA2E,EAAE,GAAG,EAAE;YACnF,MAAM,MAAM,GAAG,EAAE,CAAC;YAClB,MAAM,cAAc,GAAG,EAAE,qBAAqB,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;YAClE,MAAM,MAAM,GAAG,IAAI,6BAA6B,CAAC;gBAC/C,kBAAkB,EAAE,EAAE,UAAU,EAAE,cAAc,EAAE;aACnD,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,CAAC,4BAA4B,CAAC,MAAM,CAAC,CAAC;YAC3D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,UAAU,EAAE,cAAc,EAAE,CAAC,CAAC;QACpE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,gGAAgG,EAAE,GAAG,EAAE;YACxG,MAAM,YAAY,GAAgB;gBAChC,UAAU,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,UAAU,CAAC,EAAE,EAAE,CAAC;aAClE,CAAC;YACF,MAAM,MAAM,GAAG,IAAI,oBAAoB,CAAC;gBACtC,kBAAkB,EAAE,EAAE,YAAY,EAAE,YAAY,EAAE;aACnD,CAAC,CAAC;YACH,MAAM,CAAC,MAAM,CAAC,mBAAmB,EAAE,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;YAC3D,MAAM,CAAC,qBAAqB,CAAC,mBAAmB,CAAC,CAAC,oBAAoB,CACpE,YAAY,CACb,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uFAAuF,EAAE,GAAG,EAAE;YAC/F,MAAM,YAAY,GAAgB;gBAChC,UAAU,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,CAAC;aACxD,CAAC;YACF,MAAM,MAAM,GAAG,IAAI,oBAAoB,CAAC;gBACtC,kBAAkB,EAAE,EAAE,YAAY,EAAE,YAAY,EAAE;aACnD,CAAC,CAAC;YACH,MAAM,CAAC,MAAM,CAAC,mBAAmB,EAAE,CAAC,CAAC,aAAa,EAAE,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4EAA4E,EAAE,GAAG,EAAE;YACpF,MAAM,MAAM,GAAG,IAAI,oBAAoB,CAAC,EAAE,CAAC,CAAC;YAC5C,MAAM,CAAC,MAAM,CAAC,mBAAmB,EAAE,CAAC,CAAC,aAAa,EAAE,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4FAA4F,EAAE,GAAG,EAAE;YACpG,MAAM,MAAM,GAAG,IAAI,oBAAoB,CAAC;gBACtC,QAAQ,EAAE,KAAK;gBACf,UAAU,EAAE,iBAAiB;aAC9B,CAAC,CAAC;YACH,MAAM,gBAAgB,GAAgB;gBACpC,UAAU,EAAE;oBACV;wBACE,OAAO,EAAE;4BACP,IAAI,EAAE,OAAO;4BACb,KAAK,EAAE,CAAC,iBAAiB,CAAC;yBAC3B;wBACD,YAAY,EAAE,MAAM;qBACrB;iBACF;aACF,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,mBAAmB,EAAE,CAAC,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;YAC/D,MAAM,CAAC,qBAAqB,CAAC,mBAAmB,CAAC,CAAC,oBAAoB,CACpE,gBAAgB,CACjB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kIAAkI,EAAE,GAAG,EAAE;YAC1I,MAAM,MAAM,GAAG,IAAI,oBAAoB,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;YAC7D,MAAM,gBAAgB,GAAgB;gBACpC,UAAU,EAAE;oBACV;wBACE,OAAO,EAAE;4BACP,IAAI,EAAE,OAAO;4BACb,KAAK,EAAE,CAAC,oBAAoB,CAAC;yBAC9B;wBACD,YAAY,EAAE,MAAM;qBACrB;iBACF;aACF,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,mBAAmB,EAAE,CAAC,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;YAC/D,MAAM,CAAC,qBAAqB,CAAC,mBAAmB,CAAC,CAAC,oBAAoB,CACpE,gBAAgB,CACjB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/ide/constants.d.ts

@@ -4,3 +4,6 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 export declare const GEMINI_CLI_COMPANION_EXTENSION_NAME = "Gemini CLI Companion";
+export declare const IDE_MAX_OPEN_FILES = 10;
+export declare const IDE_MAX_SELECTED_TEXT_LENGTH = 16384;
+export declare const IDE_REQUEST_TIMEOUT_MS: number;

package/dist/src/ide/constants.js

@@ -4,4 +4,7 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 export const GEMINI_CLI_COMPANION_EXTENSION_NAME = 'Gemini CLI Companion';
+export const IDE_MAX_OPEN_FILES = 10;
+export const IDE_MAX_SELECTED_TEXT_LENGTH = 16384; // 16 KiB limit
+export const IDE_REQUEST_TIMEOUT_MS = 10 * 60 * 1000; // 10 minutes
 //# sourceMappingURL=constants.js.map

package/dist/src/ide/constants.js.map

@@ -1 +1 @@
-{"version":3,"file":"constants.js","sourceRoot":"","sources":["../../../src/ide/constants.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,MAAM,CAAC,MAAM,mCAAmC,GAAG,sBAAsB,CAAC"}
+{"version":3,"file":"constants.js","sourceRoot":"","sources":["../../../src/ide/constants.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,MAAM,CAAC,MAAM,mCAAmC,GAAG,sBAAsB,CAAC;AAC1E,MAAM,CAAC,MAAM,kBAAkB,GAAG,EAAE,CAAC;AACrC,MAAM,CAAC,MAAM,4BAA4B,GAAG,KAAK,CAAC,CAAC,eAAe;AAClE,MAAM,CAAC,MAAM,sBAAsB,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,aAAa"}

package/dist/src/ide/detect-ide.d.ts

@@ -3,18 +3,58 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-export declare enum DetectedIde {
-    Devin = "devin",
-    Replit = "replit",
-    VSCode = "vscode",
-    Cursor = "cursor",
-    CloudShell = "cloudshell",
-    Codespaces = "codespaces",
-    FirebaseStudio = "firebasestudio",
-    Trae = "trae"
-}
+export declare const IDE_DEFINITIONS: {
+    readonly devin: {
+        readonly name: "devin";
+        readonly displayName: "Devin";
+    };
+    readonly replit: {
+        readonly name: "replit";
+        readonly displayName: "Replit";
+    };
+    readonly cursor: {
+        readonly name: "cursor";
+        readonly displayName: "Cursor";
+    };
+    readonly cloudshell: {
+        readonly name: "cloudshell";
+        readonly displayName: "Cloud Shell";
+    };
+    readonly codespaces: {
+        readonly name: "codespaces";
+        readonly displayName: "GitHub Codespaces";
+    };
+    readonly firebasestudio: {
+        readonly name: "firebasestudio";
+        readonly displayName: "Firebase Studio";
+    };
+    readonly trae: {
+        readonly name: "trae";
+        readonly displayName: "Trae";
+    };
+    readonly vscode: {
+        readonly name: "vscode";
+        readonly displayName: "VS Code";
+    };
+    readonly vscodefork: {
+        readonly name: "vscodefork";
+        readonly displayName: "IDE";
+    };
+    readonly antigravity: {
+        readonly name: "antigravity";
+        readonly displayName: "Antigravity";
+    };
+};
 export interface IdeInfo {
+    name: string;
     displayName: string;
 }
-export declare function getIdeInfo(ide: DetectedIde): IdeInfo;
-export declare function detectIde(): DetectedIde | undefined;
+export declare function isCloudShell(): boolean;
+export declare function detectIdeFromEnv(): IdeInfo;
+export declare function detectIde(ideProcessInfo: {
+    pid: number;
+    command: string;
+}, ideInfoFromFile?: {
+    name?: string;
+    displayName?: string;
+}): IdeInfo | undefined;

package/dist/src/ide/detect-ide.js

@@ -3,84 +3,70 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-export var DetectedIde;
-(function (DetectedIde) {
-    DetectedIde["Devin"] = "devin";
-    DetectedIde["Replit"] = "replit";
-    DetectedIde["VSCode"] = "vscode";
-    DetectedIde["Cursor"] = "cursor";
-    DetectedIde["CloudShell"] = "cloudshell";
-    DetectedIde["Codespaces"] = "codespaces";
-    DetectedIde["FirebaseStudio"] = "firebasestudio";
-    DetectedIde["Trae"] = "trae";
-})(DetectedIde || (DetectedIde = {}));
-export function getIdeInfo(ide) {
-    switch (ide) {
-        case DetectedIde.Devin:
-            return {
-                displayName: 'Devin',
-            };
-        case DetectedIde.Replit:
-            return {
-                displayName: 'Replit',
-            };
-        case DetectedIde.VSCode:
-            return {
-                displayName: 'VS Code',
-            };
-        case DetectedIde.Cursor:
-            return {
-                displayName: 'Cursor',
-            };
-        case DetectedIde.CloudShell:
-            return {
-                displayName: 'Cloud Shell',
-            };
-        case DetectedIde.Codespaces:
-            return {
-                displayName: 'GitHub Codespaces',
-            };
-        case DetectedIde.FirebaseStudio:
-            return {
-                displayName: 'Firebase Studio',
-            };
-        case DetectedIde.Trae:
-            return {
-                displayName: 'Trae',
-            };
-        default: {
-            // This ensures that if a new IDE is added to the enum, we get a compile-time error.
-            const exhaustiveCheck = ide;
-            return exhaustiveCheck;
-        }
-    }
+export const IDE_DEFINITIONS = {
+    devin: { name: 'devin', displayName: 'Devin' },
+    replit: { name: 'replit', displayName: 'Replit' },
+    cursor: { name: 'cursor', displayName: 'Cursor' },
+    cloudshell: { name: 'cloudshell', displayName: 'Cloud Shell' },
+    codespaces: { name: 'codespaces', displayName: 'GitHub Codespaces' },
+    firebasestudio: { name: 'firebasestudio', displayName: 'Firebase Studio' },
+    trae: { name: 'trae', displayName: 'Trae' },
+    vscode: { name: 'vscode', displayName: 'VS Code' },
+    vscodefork: { name: 'vscodefork', displayName: 'IDE' },
+    antigravity: { name: 'antigravity', displayName: 'Antigravity' },
+};
+export function isCloudShell() {
+    return !!(process.env['EDITOR_IN_CLOUD_SHELL'] || process.env['CLOUD_SHELL']);
 }
-export function detectIde() {
-    // Only VSCode-based integrations are currently supported.
-    if (process.env['TERM_PROGRAM'] !== 'vscode') {
-        return undefined;
+export function detectIdeFromEnv() {
+    if (process.env['ANTIGRAVITY_CLI_ALIAS']) {
+        return IDE_DEFINITIONS.antigravity;
     }
     if (process.env['__COG_BASHRC_SOURCED']) {
-        return DetectedIde.Devin;
+        return IDE_DEFINITIONS.devin;
     }
     if (process.env['REPLIT_USER']) {
-        return DetectedIde.Replit;
+        return IDE_DEFINITIONS.replit;
     }
     if (process.env['CURSOR_TRACE_ID']) {
-        return DetectedIde.Cursor;
+        return IDE_DEFINITIONS.cursor;
     }
     if (process.env['CODESPACES']) {
-        return DetectedIde.Codespaces;
+        return IDE_DEFINITIONS.codespaces;
     }
-    if (process.env['EDITOR_IN_CLOUD_SHELL'] || process.env['CLOUD_SHELL']) {
-        return DetectedIde.CloudShell;
+    if (isCloudShell()) {
+        return IDE_DEFINITIONS.cloudshell;
     }
     if (process.env['TERM_PRODUCT'] === 'Trae') {
-        return DetectedIde.Trae;
+        return IDE_DEFINITIONS.trae;
+    }
+    if (process.env['MONOSPACE_ENV']) {
+        return IDE_DEFINITIONS.firebasestudio;
+    }
+    return IDE_DEFINITIONS.vscode;
+}
+function verifyVSCode(ide, ideProcessInfo) {
+    if (ide.name !== IDE_DEFINITIONS.vscode.name) {
+        return ide;
     }
-    if (process.env['FIREBASE_DEPLOY_AGENT'] || process.env['MONOSPACE_ENV']) {
-        return DetectedIde.FirebaseStudio;
+    if (!ideProcessInfo.command ||
+        ideProcessInfo.command.toLowerCase().includes('code')) {
+        return IDE_DEFINITIONS.vscode;
+    }
+    return IDE_DEFINITIONS.vscodefork;
+}
+export function detectIde(ideProcessInfo, ideInfoFromFile) {
+    if (ideInfoFromFile?.name && ideInfoFromFile.displayName) {
+        return {
+            name: ideInfoFromFile.name,
+            displayName: ideInfoFromFile.displayName,
+        };
+    }
+    // Only VSCode-based integrations are currently supported.
+    if (process.env['TERM_PROGRAM'] !== 'vscode') {
+        return undefined;
     }
-    return DetectedIde.VSCode;
+    const ide = detectIdeFromEnv();
+    return verifyVSCode(ide, ideProcessInfo);
 }
 //# sourceMappingURL=detect-ide.js.map

package/dist/src/ide/detect-ide.js.map

@@ -1 +1 @@
-{"version":3,"file":"detect-ide.js","sourceRoot":"","sources":["../../../src/ide/detect-ide.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,MAAM,CAAN,IAAY,WASX;AATD,WAAY,WAAW;IACrB,8BAAe,CAAA;IACf,gCAAiB,CAAA;IACjB,gCAAiB,CAAA;IACjB,gCAAiB,CAAA;IACjB,wCAAyB,CAAA;IACzB,wCAAyB,CAAA;IACzB,gDAAiC,CAAA;IACjC,4BAAa,CAAA;AACf,CAAC,EATW,WAAW,KAAX,WAAW,QAStB;AAMD,MAAM,UAAU,UAAU,CAAC,GAAgB;IACzC,QAAQ,GAAG,EAAE,CAAC;QACZ,KAAK,WAAW,CAAC,KAAK;YACpB,OAAO;gBACL,WAAW,EAAE,OAAO;aACrB,CAAC;QACJ,KAAK,WAAW,CAAC,MAAM;YACrB,OAAO;gBACL,WAAW,EAAE,QAAQ;aACtB,CAAC;QACJ,KAAK,WAAW,CAAC,MAAM;YACrB,OAAO;gBACL,WAAW,EAAE,SAAS;aACvB,CAAC;QACJ,KAAK,WAAW,CAAC,MAAM;YACrB,OAAO;gBACL,WAAW,EAAE,QAAQ;aACtB,CAAC;QACJ,KAAK,WAAW,CAAC,UAAU;YACzB,OAAO;gBACL,WAAW,EAAE,aAAa;aAC3B,CAAC;QACJ,KAAK,WAAW,CAAC,UAAU;YACzB,OAAO;gBACL,WAAW,EAAE,mBAAmB;aACjC,CAAC;QACJ,KAAK,WAAW,CAAC,cAAc;YAC7B,OAAO;gBACL,WAAW,EAAE,iBAAiB;aAC/B,CAAC;QACJ,KAAK,WAAW,CAAC,IAAI;YACnB,OAAO;gBACL,WAAW,EAAE,MAAM;aACpB,CAAC;QACJ,OAAO,CAAC,CAAC,CAAC;YACR,oFAAoF;YACpF,MAAM,eAAe,GAAU,GAAG,CAAC;YACnC,OAAO,eAAe,CAAC;QACzB,CAAC;IACH,CAAC;AACH,CAAC;AAED,MAAM,UAAU,SAAS;IACvB,0DAA0D;IAC1D,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,KAAK,QAAQ,EAAE,CAAC;QAC7C,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,IAAI,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,EAAE,CAAC;QACxC,OAAO,WAAW,CAAC,KAAK,CAAC;IAC3B,CAAC;IACD,IAAI,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,EAAE,CAAC;QAC/B,OAAO,WAAW,CAAC,MAAM,CAAC;IAC5B,CAAC;IACD,IAAI,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,EAAE,CAAC;QACnC,OAAO,WAAW,CAAC,MAAM,CAAC;IAC5B,CAAC;IACD,IAAI,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,CAAC;QAC9B,OAAO,WAAW,CAAC,UAAU,CAAC;IAChC,CAAC;IACD,IAAI,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,IAAI,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,EAAE,CAAC;QACvE,OAAO,WAAW,CAAC,UAAU,CAAC;IAChC,CAAC;IACD,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,KAAK,MAAM,EAAE,CAAC;QAC3C,OAAO,WAAW,CAAC,IAAI,CAAC;IAC1B,CAAC;IACD,IAAI,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,IAAI,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,EAAE,CAAC;QACzE,OAAO,WAAW,CAAC,cAAc,CAAC;IACpC,CAAC;IACD,OAAO,WAAW,CAAC,MAAM,CAAC;AAC5B,CAAC"}
+{"version":3,"file":"detect-ide.js","sourceRoot":"","sources":["../../../src/ide/detect-ide.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,MAAM,CAAC,MAAM,eAAe,GAAG;IAC7B,KAAK,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE;IAC9C,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,QAAQ,EAAE;IACjD,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,QAAQ,EAAE;IACjD,UAAU,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE,WAAW,EAAE,aAAa,EAAE;IAC9D,UAAU,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE,WAAW,EAAE,mBAAmB,EAAE;IACpE,cAAc,EAAE,EAAE,IAAI,EAAE,gBAAgB,EAAE,WAAW,EAAE,iBAAiB,EAAE;IAC1E,IAAI,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE;IAC3C,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,SAAS,EAAE;IAClD,UAAU,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE,WAAW,EAAE,KAAK,EAAE;IACtD,WAAW,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,WAAW,EAAE,aAAa,EAAE;CACxD,CAAC;AAOX,MAAM,UAAU,YAAY;IAC1B,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,IAAI,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC;AAChF,CAAC;AAED,MAAM,UAAU,gBAAgB;IAC9B,IAAI,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,EAAE,CAAC;QACzC,OAAO,eAAe,CAAC,WAAW,CAAC;IACrC,CAAC;IACD,IAAI,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,EAAE,CAAC;QACxC,OAAO,eAAe,CAAC,KAAK,CAAC;IAC/B,CAAC;IACD,IAAI,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,EAAE,CAAC;QAC/B,OAAO,eAAe,CAAC,MAAM,CAAC;IAChC,CAAC;IACD,IAAI,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,EAAE,CAAC;QACnC,OAAO,eAAe,CAAC,MAAM,CAAC;IAChC,CAAC;IACD,IAAI,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,CAAC;QAC9B,OAAO,eAAe,CAAC,UAAU,CAAC;IACpC,CAAC;IACD,IAAI,YAAY,EAAE,EAAE,CAAC;QACnB,OAAO,eAAe,CAAC,UAAU,CAAC;IACpC,CAAC;IACD,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,KAAK,MAAM,EAAE,CAAC;QAC3C,OAAO,eAAe,CAAC,IAAI,CAAC;IAC9B,CAAC;IACD,IAAI,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,EAAE,CAAC;QACjC,OAAO,eAAe,CAAC,cAAc,CAAC;IACxC,CAAC;IACD,OAAO,eAAe,CAAC,MAAM,CAAC;AAChC,CAAC;AAED,SAAS,YAAY,CACnB,GAAY,EACZ,cAGC;IAED,IAAI,GAAG,CAAC,IAAI,KAAK,eAAe,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;QAC7C,OAAO,GAAG,CAAC;IACb,CAAC;IACD,IACE,CAAC,cAAc,CAAC,OAAO;QACvB,cAAc,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,EACrD,CAAC;QACD,OAAO,eAAe,CAAC,MAAM,CAAC;IAChC,CAAC;IACD,OAAO,eAAe,CAAC,UAAU,CAAC;AACpC,CAAC;AAED,MAAM,UAAU,SAAS,CACvB,cAGC,EACD,eAAyD;IAEzD,IAAI,eAAe,EAAE,IAAI,IAAI,eAAe,CAAC,WAAW,EAAE,CAAC;QACzD,OAAO;YACL,IAAI,EAAE,eAAe,CAAC,IAAI;YAC1B,WAAW,EAAE,eAAe,CAAC,WAAW;SACzC,CAAC;IACJ,CAAC;IAED,0DAA0D;IAC1D,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,KAAK,QAAQ,EAAE,CAAC;QAC7C,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,GAAG,GAAG,gBAAgB,EAAE,CAAC;IAC/B,OAAO,YAAY,CAAC,GAAG,EAAE,cAAc,CAAC,CAAC;AAC3C,CAAC"}