@office-ai/aioncli-core 0.8.1 → 0.18.5

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

@@ -0,0 +1,341 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import * as fs from 'node:fs';
+import { HookRegistry, ConfigSource, HookRegistryNotInitializedError, } from './hookRegistry.js';
+import { HookEventName, HookType } from './types.js';
+// Mock fs
+vi.mock('fs', () => ({
+    existsSync: vi.fn(),
+    readFileSync: vi.fn(),
+}));
+// Mock debugLogger using vi.hoisted
+const mockDebugLogger = vi.hoisted(() => ({
+    log: vi.fn(),
+    warn: vi.fn(),
+    error: vi.fn(),
+    debug: vi.fn(),
+}));
+vi.mock('../utils/debugLogger.js', () => ({
+    debugLogger: mockDebugLogger,
+}));
+describe('HookRegistry', () => {
+    let hookRegistry;
+    let mockConfig;
+    let mockStorage;
+    beforeEach(() => {
+        vi.resetAllMocks();
+        mockStorage = {
+            getGeminiDir: vi.fn().mockReturnValue('/project/.gemini'),
+        };
+        mockConfig = {
+            storage: mockStorage,
+            getExtensions: vi.fn().mockReturnValue([]),
+            getHooks: vi.fn().mockReturnValue({}),
+        };
+        hookRegistry = new HookRegistry(mockConfig);
+    });
+    afterEach(() => {
+        vi.restoreAllMocks();
+    });
+    describe('initialize', () => {
+        it('should initialize successfully with no hooks', async () => {
+            vi.mocked(fs.existsSync).mockReturnValue(false);
+            await hookRegistry.initialize();
+            expect(hookRegistry.getAllHooks()).toHaveLength(0);
+            expect(mockDebugLogger.log).toHaveBeenCalledWith('Hook registry initialized with 0 hook entries');
+        });
+        it('should load hooks from project configuration', async () => {
+            const mockHooksConfig = {
+                BeforeTool: [
+                    {
+                        matcher: 'EditTool',
+                        hooks: [
+                            {
+                                type: 'command',
+                                command: './hooks/check_style.sh',
+                                timeout: 60,
+                            },
+                        ],
+                    },
+                ],
+            };
+            // Update mock to return the hooks configuration
+            vi.mocked(mockConfig.getHooks).mockReturnValue(mockHooksConfig);
+            await hookRegistry.initialize();
+            const hooks = hookRegistry.getAllHooks();
+            expect(hooks).toHaveLength(1);
+            expect(hooks[0].eventName).toBe(HookEventName.BeforeTool);
+            expect(hooks[0].config.type).toBe(HookType.Command);
+            expect(hooks[0].config.command).toBe('./hooks/check_style.sh');
+            expect(hooks[0].matcher).toBe('EditTool');
+            expect(hooks[0].source).toBe(ConfigSource.Project);
+        });
+        it('should load plugin hooks', async () => {
+            const mockHooksConfig = {
+                AfterTool: [
+                    {
+                        hooks: [
+                            {
+                                type: 'command',
+                                command: './hooks/after-tool.sh',
+                                timeout: 30,
+                            },
+                        ],
+                    },
+                ],
+            };
+            // Update mock to return the hooks configuration
+            vi.mocked(mockConfig.getHooks).mockReturnValue(mockHooksConfig);
+            await hookRegistry.initialize();
+            const hooks = hookRegistry.getAllHooks();
+            expect(hooks).toHaveLength(1);
+            expect(hooks[0].eventName).toBe(HookEventName.AfterTool);
+            expect(hooks[0].config.type).toBe(HookType.Command);
+            expect(hooks[0].config.command).toBe('./hooks/after-tool.sh');
+        });
+        it('should handle invalid configuration gracefully', async () => {
+            const invalidHooksConfig = {
+                BeforeTool: [
+                    {
+                        hooks: [
+                            {
+                                type: 'invalid-type', // Invalid hook type
+                                command: './hooks/test.sh',
+                            },
+                        ],
+                    },
+                ],
+            };
+            // Update mock to return invalid configuration
+            vi.mocked(mockConfig.getHooks).mockReturnValue(invalidHooksConfig);
+            await hookRegistry.initialize();
+            expect(hookRegistry.getAllHooks()).toHaveLength(0);
+            expect(mockDebugLogger.warn).toHaveBeenCalled();
+        });
+        it('should validate hook configurations', async () => {
+            const mockHooksConfig = {
+                BeforeTool: [
+                    {
+                        hooks: [
+                            {
+                                type: 'invalid',
+                                command: './hooks/test.sh',
+                            },
+                            {
+                                type: 'command',
+                                // Missing command field
+                            },
+                        ],
+                    },
+                ],
+            };
+            // Update mock to return invalid configuration
+            vi.mocked(mockConfig.getHooks).mockReturnValue(mockHooksConfig);
+            await hookRegistry.initialize();
+            expect(hookRegistry.getAllHooks()).toHaveLength(0);
+            expect(mockDebugLogger.warn).toHaveBeenCalled(); // At least some warnings should be logged
+        });
+    });
+    describe('getHooksForEvent', () => {
+        beforeEach(async () => {
+            const mockHooksConfig = {
+                BeforeTool: [
+                    {
+                        matcher: 'EditTool',
+                        hooks: [
+                            {
+                                type: 'command',
+                                command: './hooks/edit_check.sh',
+                            },
+                        ],
+                    },
+                    {
+                        hooks: [
+                            {
+                                type: 'command',
+                                command: './hooks/general_check.sh',
+                            },
+                        ],
+                    },
+                ],
+                AfterTool: [
+                    {
+                        hooks: [
+                            {
+                                type: 'command',
+                                command: './hooks/after-tool.sh',
+                            },
+                        ],
+                    },
+                ],
+            };
+            // Update mock to return the hooks configuration
+            vi.mocked(mockConfig.getHooks).mockReturnValue(mockHooksConfig);
+            await hookRegistry.initialize();
+        });
+        it('should return hooks for specific event', () => {
+            const beforeToolHooks = hookRegistry.getHooksForEvent(HookEventName.BeforeTool);
+            expect(beforeToolHooks).toHaveLength(2);
+            const afterToolHooks = hookRegistry.getHooksForEvent(HookEventName.AfterTool);
+            expect(afterToolHooks).toHaveLength(1);
+        });
+        it('should return empty array for events with no hooks', () => {
+            const notificationHooks = hookRegistry.getHooksForEvent(HookEventName.Notification);
+            expect(notificationHooks).toHaveLength(0);
+        });
+        it('should throw error if not initialized', () => {
+            const uninitializedRegistry = new HookRegistry(mockConfig);
+            expect(() => {
+                uninitializedRegistry.getHooksForEvent(HookEventName.BeforeTool);
+            }).toThrow(HookRegistryNotInitializedError);
+        });
+    });
+    describe('setHookEnabled', () => {
+        beforeEach(async () => {
+            const mockHooksConfig = {
+                BeforeTool: [
+                    {
+                        hooks: [
+                            {
+                                type: 'command',
+                                command: './hooks/test.sh',
+                            },
+                        ],
+                    },
+                ],
+            };
+            // Update mock to return the hooks configuration
+            vi.mocked(mockConfig.getHooks).mockReturnValue(mockHooksConfig);
+            await hookRegistry.initialize();
+        });
+        it('should enable and disable hooks', () => {
+            const hookName = './hooks/test.sh';
+            // Initially enabled
+            let hooks = hookRegistry.getHooksForEvent(HookEventName.BeforeTool);
+            expect(hooks).toHaveLength(1);
+            // Disable
+            hookRegistry.setHookEnabled(hookName, false);
+            hooks = hookRegistry.getHooksForEvent(HookEventName.BeforeTool);
+            expect(hooks).toHaveLength(0);
+            // Re-enable
+            hookRegistry.setHookEnabled(hookName, true);
+            hooks = hookRegistry.getHooksForEvent(HookEventName.BeforeTool);
+            expect(hooks).toHaveLength(1);
+        });
+        it('should warn when hook not found', () => {
+            hookRegistry.setHookEnabled('non-existent-hook', false);
+            expect(mockDebugLogger.warn).toHaveBeenCalledWith('No hooks found matching "non-existent-hook"');
+        });
+    });
+    describe('malformed configuration handling', () => {
+        it('should handle non-array definitions gracefully', async () => {
+            const malformedConfig = {
+                BeforeTool: 'not-an-array', // Should be an array of HookDefinition
+            };
+            vi.mocked(mockConfig.getHooks).mockReturnValue(malformedConfig);
+            await hookRegistry.initialize();
+            expect(hookRegistry.getAllHooks()).toHaveLength(0);
+            expect(mockDebugLogger.warn).toHaveBeenCalledWith(expect.stringContaining('is not an array'));
+        });
+        it('should handle object instead of array for definitions', async () => {
+            const malformedConfig = {
+                AfterTool: { hooks: [] }, // Should be an array, not a single object
+            };
+            vi.mocked(mockConfig.getHooks).mockReturnValue(malformedConfig);
+            await hookRegistry.initialize();
+            expect(hookRegistry.getAllHooks()).toHaveLength(0);
+            expect(mockDebugLogger.warn).toHaveBeenCalledWith(expect.stringContaining('is not an array'));
+        });
+        it('should handle null definition gracefully', async () => {
+            const malformedConfig = {
+                BeforeTool: [null], // Invalid: null definition
+            };
+            vi.mocked(mockConfig.getHooks).mockReturnValue(malformedConfig);
+            await hookRegistry.initialize();
+            expect(hookRegistry.getAllHooks()).toHaveLength(0);
+            expect(mockDebugLogger.warn).toHaveBeenCalledWith(expect.stringContaining('Discarding invalid hook definition'), null);
+        });
+        it('should handle definition without hooks array', async () => {
+            const malformedConfig = {
+                BeforeTool: [
+                    {
+                        matcher: 'EditTool',
+                        // Missing hooks array
+                    },
+                ],
+            };
+            vi.mocked(mockConfig.getHooks).mockReturnValue(malformedConfig);
+            await hookRegistry.initialize();
+            expect(hookRegistry.getAllHooks()).toHaveLength(0);
+            expect(mockDebugLogger.warn).toHaveBeenCalledWith(expect.stringContaining('Discarding invalid hook definition'), expect.objectContaining({ matcher: 'EditTool' }));
+        });
+        it('should handle non-array hooks property', async () => {
+            const malformedConfig = {
+                BeforeTool: [
+                    {
+                        matcher: 'EditTool',
+                        hooks: 'not-an-array', // Should be an array
+                    },
+                ],
+            };
+            vi.mocked(mockConfig.getHooks).mockReturnValue(malformedConfig);
+            await hookRegistry.initialize();
+            expect(hookRegistry.getAllHooks()).toHaveLength(0);
+            expect(mockDebugLogger.warn).toHaveBeenCalledWith(expect.stringContaining('Discarding invalid hook definition'), expect.objectContaining({ hooks: 'not-an-array', matcher: 'EditTool' }));
+        });
+        it('should handle non-object hookConfig in hooks array', async () => {
+            const malformedConfig = {
+                BeforeTool: [
+                    {
+                        hooks: [
+                            'not-an-object', // Should be an object
+                            42, // Should be an object
+                            null, // Should be an object
+                        ],
+                    },
+                ],
+            };
+            vi.mocked(mockConfig.getHooks).mockReturnValue(malformedConfig);
+            await hookRegistry.initialize();
+            expect(hookRegistry.getAllHooks()).toHaveLength(0);
+            expect(mockDebugLogger.warn).toHaveBeenCalledTimes(3); // One warning for each invalid hookConfig
+        });
+        it('should handle mixed valid and invalid hook configurations', async () => {
+            const mixedConfig = {
+                BeforeTool: [
+                    {
+                        hooks: [
+                            {
+                                type: 'command',
+                                command: './valid-hook.sh',
+                            },
+                            'invalid-string',
+                            {
+                                type: 'invalid-type',
+                                command: './invalid-type.sh',
+                            },
+                        ],
+                    },
+                ],
+            };
+            vi.mocked(mockConfig.getHooks).mockReturnValue(mixedConfig);
+            await hookRegistry.initialize();
+            // Should only load the valid hook
+            const hooks = hookRegistry.getAllHooks();
+            expect(hooks).toHaveLength(1);
+            expect(hooks[0].config.command).toBe('./valid-hook.sh');
+            // Verify the warnings for invalid configurations
+            // 1st warning: non-object hookConfig ('invalid-string')
+            expect(mockDebugLogger.warn).toHaveBeenCalledWith(expect.stringContaining('Discarding invalid hook configuration'), 'invalid-string');
+            // 2nd warning: validateHookConfig logs invalid type
+            expect(mockDebugLogger.warn).toHaveBeenCalledWith(expect.stringContaining('Invalid hook BeforeTool from project type'));
+            // 3rd warning: processHookDefinition logs the failed hookConfig
+            expect(mockDebugLogger.warn).toHaveBeenCalledWith(expect.stringContaining('Discarding invalid hook configuration'), expect.objectContaining({ type: 'invalid-type' }));
+        });
+    });
+});
+//# sourceMappingURL=hookRegistry.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"hookRegistry.test.js","sourceRoot":"","sources":["../../../src/hooks/hookRegistry.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,EACL,YAAY,EACZ,YAAY,EACZ,+BAA+B,GAChC,MAAM,mBAAmB,CAAC;AAE3B,OAAO,EAAE,aAAa,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AAIrD,UAAU;AACV,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,GAAG,EAAE,CAAC,CAAC;IACnB,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE;IACnB,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;CACtB,CAAC,CAAC,CAAC;AAEJ,oCAAoC;AACpC,MAAM,eAAe,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC;IACxC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE;IACZ,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;IACb,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;IACd,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;CACf,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,yBAAyB,EAAE,GAAG,EAAE,CAAC,CAAC;IACxC,WAAW,EAAE,eAAe;CAC7B,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,IAAI,YAA0B,CAAC;IAC/B,IAAI,UAAkB,CAAC;IACvB,IAAI,WAAoB,CAAC;IAEzB,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QAEnB,WAAW,GAAG;YACZ,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,kBAAkB,CAAC;SACpC,CAAC;QAExB,UAAU,GAAG;YACX,OAAO,EAAE,WAAW;YACpB,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC;YAC1C,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC;SACjB,CAAC;QAEvB,YAAY,GAAG,IAAI,YAAY,CAAC,UAAU,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,eAAe,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;YAEhD,MAAM,YAAY,CAAC,UAAU,EAAE,CAAC;YAEhC,MAAM,CAAC,YAAY,CAAC,WAAW,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACnD,MAAM,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,oBAAoB,CAC9C,+CAA+C,CAChD,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,MAAM,eAAe,GAAG;gBACtB,UAAU,EAAE;oBACV;wBACE,OAAO,EAAE,UAAU;wBACnB,KAAK,EAAE;4BACL;gCACE,IAAI,EAAE,SAAS;gCACf,OAAO,EAAE,wBAAwB;gCACjC,OAAO,EAAE,EAAE;6BACZ;yBACF;qBACF;iBACF;aACF,CAAC;YAEF,gDAAgD;YAChD,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,eAAe,CAC5C,eAEC,CACF,CAAC;YAEF,MAAM,YAAY,CAAC,UAAU,EAAE,CAAC;YAEhC,MAAM,KAAK,GAAG,YAAY,CAAC,WAAW,EAAE,CAAC;YACzC,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;YAC1D,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;YACpD,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;YAC/D,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAC1C,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;YACxC,MAAM,eAAe,GAAG;gBACtB,SAAS,EAAE;oBACT;wBACE,KAAK,EAAE;4BACL;gCACE,IAAI,EAAE,SAAS;gCACf,OAAO,EAAE,uBAAuB;gCAChC,OAAO,EAAE,EAAE;6BACZ;yBACF;qBACF;iBACF;aACF,CAAC;YAEF,gDAAgD;YAChD,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,eAAe,CAC5C,eAEC,CACF,CAAC;YAEF,MAAM,YAAY,CAAC,UAAU,EAAE,CAAC;YAEhC,MAAM,KAAK,GAAG,YAAY,CAAC,WAAW,EAAE,CAAC;YACzC,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;YACzD,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;YACpD,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,kBAAkB,GAAG;gBACzB,UAAU,EAAE;oBACV;wBACE,KAAK,EAAE;4BACL;gCACE,IAAI,EAAE,cAAc,EAAE,oBAAoB;gCAC1C,OAAO,EAAE,iBAAiB;6BAC3B;yBACF;qBACF;iBACF;aACF,CAAC;YAEF,8CAA8C;YAC9C,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,eAAe,CAC5C,kBAEC,CACF,CAAC;YAEF,MAAM,YAAY,CAAC,UAAU,EAAE,CAAC;YAEhC,MAAM,CAAC,YAAY,CAAC,WAAW,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACnD,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;YACnD,MAAM,eAAe,GAAG;gBACtB,UAAU,EAAE;oBACV;wBACE,KAAK,EAAE;4BACL;gCACE,IAAI,EAAE,SAAS;gCACf,OAAO,EAAE,iBAAiB;6BAC3B;4BACD;gCACE,IAAI,EAAE,SAAS;gCACf,wBAAwB;6BACzB;yBACF;qBACF;iBACF;aACF,CAAC;YAEF,8CAA8C;YAC9C,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,eAAe,CAC5C,eAEC,CACF,CAAC;YAEF,MAAM,YAAY,CAAC,UAAU,EAAE,CAAC;YAEhC,MAAM,CAAC,YAAY,CAAC,WAAW,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACnD,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,gBAAgB,EAAE,CAAC,CAAC,0CAA0C;QAC7F,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,UAAU,CAAC,KAAK,IAAI,EAAE;YACpB,MAAM,eAAe,GAAG;gBACtB,UAAU,EAAE;oBACV;wBACE,OAAO,EAAE,UAAU;wBACnB,KAAK,EAAE;4BACL;gCACE,IAAI,EAAE,SAAS;gCACf,OAAO,EAAE,uBAAuB;6BACjC;yBACF;qBACF;oBACD;wBACE,KAAK,EAAE;4BACL;gCACE,IAAI,EAAE,SAAS;gCACf,OAAO,EAAE,0BAA0B;6BACpC;yBACF;qBACF;iBACF;gBACD,SAAS,EAAE;oBACT;wBACE,KAAK,EAAE;4BACL;gCACE,IAAI,EAAE,SAAS;gCACf,OAAO,EAAE,uBAAuB;6BACjC;yBACF;qBACF;iBACF;aACF,CAAC;YAEF,gDAAgD;YAChD,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,eAAe,CAC5C,eAEC,CACF,CAAC;YAEF,MAAM,YAAY,CAAC,UAAU,EAAE,CAAC;QAClC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,MAAM,eAAe,GAAG,YAAY,CAAC,gBAAgB,CACnD,aAAa,CAAC,UAAU,CACzB,CAAC;YACF,MAAM,CAAC,eAAe,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAExC,MAAM,cAAc,GAAG,YAAY,CAAC,gBAAgB,CAClD,aAAa,CAAC,SAAS,CACxB,CAAC;YACF,MAAM,CAAC,cAAc,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;YAC5D,MAAM,iBAAiB,GAAG,YAAY,CAAC,gBAAgB,CACrD,aAAa,CAAC,YAAY,CAC3B,CAAC;YACF,MAAM,CAAC,iBAAiB,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,MAAM,qBAAqB,GAAG,IAAI,YAAY,CAAC,UAAU,CAAC,CAAC;YAE3D,MAAM,CAAC,GAAG,EAAE;gBACV,qBAAqB,CAAC,gBAAgB,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;YACnE,CAAC,CAAC,CAAC,OAAO,CAAC,+BAA+B,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,UAAU,CAAC,KAAK,IAAI,EAAE;YACpB,MAAM,eAAe,GAAG;gBACtB,UAAU,EAAE;oBACV;wBACE,KAAK,EAAE;4BACL;gCACE,IAAI,EAAE,SAAS;gCACf,OAAO,EAAE,iBAAiB;6BAC3B;yBACF;qBACF;iBACF;aACF,CAAC;YAEF,gDAAgD;YAChD,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,eAAe,CAC5C,eAEC,CACF,CAAC;YAEF,MAAM,YAAY,CAAC,UAAU,EAAE,CAAC;QAClC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,QAAQ,GAAG,iBAAiB,CAAC;YAEnC,oBAAoB;YACpB,IAAI,KAAK,GAAG,YAAY,CAAC,gBAAgB,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;YACpE,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAE9B,UAAU;YACV,YAAY,CAAC,cAAc,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;YAC7C,KAAK,GAAG,YAAY,CAAC,gBAAgB,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;YAChE,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAE9B,YAAY;YACZ,YAAY,CAAC,cAAc,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;YAC5C,KAAK,GAAG,YAAY,CAAC,gBAAgB,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;YAChE,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,YAAY,CAAC,cAAc,CAAC,mBAAmB,EAAE,KAAK,CAAC,CAAC;YACxD,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAC/C,6CAA6C,CAC9C,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAChD,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,eAAe,GAAG;gBACtB,UAAU,EAAE,cAAc,EAAE,uCAAuC;aACpE,CAAC;YAEF,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,eAAe,CAC5C,eAEC,CACF,CAAC;YAEF,MAAM,YAAY,CAAC,UAAU,EAAE,CAAC;YAEhC,MAAM,CAAC,YAAY,CAAC,WAAW,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACnD,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAC/C,MAAM,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,CAC3C,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,eAAe,GAAG;gBACtB,SAAS,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,0CAA0C;aACrE,CAAC;YAEF,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,eAAe,CAC5C,eAEC,CACF,CAAC;YAEF,MAAM,YAAY,CAAC,UAAU,EAAE,CAAC;YAEhC,MAAM,CAAC,YAAY,CAAC,WAAW,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACnD,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAC/C,MAAM,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,CAC3C,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,MAAM,eAAe,GAAG;gBACtB,UAAU,EAAE,CAAC,IAAI,CAAC,EAAE,2BAA2B;aAChD,CAAC;YAEF,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,eAAe,CAC5C,eAEC,CACF,CAAC;YAEF,MAAM,YAAY,CAAC,UAAU,EAAE,CAAC;YAEhC,MAAM,CAAC,YAAY,CAAC,WAAW,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACnD,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAC/C,MAAM,CAAC,gBAAgB,CAAC,oCAAoC,CAAC,EAC7D,IAAI,CACL,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,MAAM,eAAe,GAAG;gBACtB,UAAU,EAAE;oBACV;wBACE,OAAO,EAAE,UAAU;wBACnB,sBAAsB;qBACvB;iBACF;aACF,CAAC;YAEF,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,eAAe,CAC5C,eAEC,CACF,CAAC;YAEF,MAAM,YAAY,CAAC,UAAU,EAAE,CAAC;YAEhC,MAAM,CAAC,YAAY,CAAC,WAAW,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACnD,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAC/C,MAAM,CAAC,gBAAgB,CAAC,oCAAoC,CAAC,EAC7D,MAAM,CAAC,gBAAgB,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CACjD,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACtD,MAAM,eAAe,GAAG;gBACtB,UAAU,EAAE;oBACV;wBACE,OAAO,EAAE,UAAU;wBACnB,KAAK,EAAE,cAAc,EAAE,qBAAqB;qBAC7C;iBACF;aACF,CAAC;YAEF,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,eAAe,CAC5C,eAEC,CACF,CAAC;YAEF,MAAM,YAAY,CAAC,UAAU,EAAE,CAAC;YAEhC,MAAM,CAAC,YAAY,CAAC,WAAW,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACnD,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAC/C,MAAM,CAAC,gBAAgB,CAAC,oCAAoC,CAAC,EAC7D,MAAM,CAAC,gBAAgB,CAAC,EAAE,KAAK,EAAE,cAAc,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CACxE,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;YAClE,MAAM,eAAe,GAAG;gBACtB,UAAU,EAAE;oBACV;wBACE,KAAK,EAAE;4BACL,eAAe,EAAE,sBAAsB;4BACvC,EAAE,EAAE,sBAAsB;4BAC1B,IAAI,EAAE,sBAAsB;yBAC7B;qBACF;iBACF;aACF,CAAC;YAEF,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,eAAe,CAC5C,eAEC,CACF,CAAC;YAEF,MAAM,YAAY,CAAC,UAAU,EAAE,CAAC;YAEhC,MAAM,CAAC,YAAY,CAAC,WAAW,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACnD,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,CAAC,0CAA0C;QACnG,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACzE,MAAM,WAAW,GAAG;gBAClB,UAAU,EAAE;oBACV;wBACE,KAAK,EAAE;4BACL;gCACE,IAAI,EAAE,SAAS;gCACf,OAAO,EAAE,iBAAiB;6BAC3B;4BACD,gBAAgB;4BAChB;gCACE,IAAI,EAAE,cAAc;gCACpB,OAAO,EAAE,mBAAmB;6BAC7B;yBACF;qBACF;iBACF;aACF,CAAC;YAEF,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,eAAe,CAC5C,WAEC,CACF,CAAC;YAEF,MAAM,YAAY,CAAC,UAAU,EAAE,CAAC;YAEhC,kCAAkC;YAClC,MAAM,KAAK,GAAG,YAAY,CAAC,WAAW,EAAE,CAAC;YACzC,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;YAExD,iDAAiD;YACjD,wDAAwD;YACxD,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAC/C,MAAM,CAAC,gBAAgB,CAAC,uCAAuC,CAAC,EAChE,gBAAgB,CACjB,CAAC;YACF,oDAAoD;YACpD,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAC/C,MAAM,CAAC,gBAAgB,CAAC,2CAA2C,CAAC,CACrE,CAAC;YACF,gEAAgE;YAChE,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAC/C,MAAM,CAAC,gBAAgB,CAAC,uCAAuC,CAAC,EAChE,MAAM,CAAC,gBAAgB,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC,CAClD,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/hooks/hookRunner.d.ts

@@ -0,0 +1,42 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { HookConfig } from './types.js';
+import { HookEventName } from './types.js';
+import type { HookInput, HookExecutionResult } from './types.js';
+/**
+ * Hook runner that executes command hooks
+ */
+export declare class HookRunner {
+    constructor();
+    /**
+     * Execute a single hook
+     */
+    executeHook(hookConfig: HookConfig, eventName: HookEventName, input: HookInput): Promise<HookExecutionResult>;
+    /**
+     * Execute multiple hooks in parallel
+     */
+    executeHooksParallel(hookConfigs: HookConfig[], eventName: HookEventName, input: HookInput): Promise<HookExecutionResult[]>;
+    /**
+     * Execute multiple hooks sequentially
+     */
+    executeHooksSequential(hookConfigs: HookConfig[], eventName: HookEventName, input: HookInput): Promise<HookExecutionResult[]>;
+    /**
+     * Apply hook output to modify input for the next hook in sequential execution
+     */
+    private applyHookOutputToInput;
+    /**
+     * Execute a command hook
+     */
+    private executeCommandHook;
+    /**
+     * Expand command with environment variables and input context
+     */
+    private expandCommand;
+    /**
+     * Convert plain text output to structured HookOutput
+     */
+    private convertPlainTextToHookOutput;
+}

package/dist/src/hooks/hookRunner.js

@@ -0,0 +1,272 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { spawn } from 'node:child_process';
+import { HookEventName } from './types.js';
+import { debugLogger } from '../utils/debugLogger.js';
+/**
+ * Default timeout for hook execution (60 seconds)
+ */
+const DEFAULT_HOOK_TIMEOUT = 60000;
+/**
+ * Exit code constants for hook execution
+ */
+const EXIT_CODE_SUCCESS = 0;
+const EXIT_CODE_BLOCKING_ERROR = 2;
+const EXIT_CODE_NON_BLOCKING_ERROR = 1;
+/**
+ * Hook runner that executes command hooks
+ */
+export class HookRunner {
+    constructor() { }
+    /**
+     * Execute a single hook
+     */
+    async executeHook(hookConfig, eventName, input) {
+        const startTime = Date.now();
+        try {
+            return await this.executeCommandHook(hookConfig, eventName, input, startTime);
+        }
+        catch (error) {
+            const duration = Date.now() - startTime;
+            const hookSource = hookConfig.command || 'unknown';
+            const errorMessage = `Hook execution failed for event '${eventName}' (source: ${hookSource}): ${error}`;
+            debugLogger.warn(`Hook execution error (non-fatal): ${errorMessage}`);
+            return {
+                hookConfig,
+                eventName,
+                success: false,
+                error: error instanceof Error ? error : new Error(errorMessage),
+                duration,
+            };
+        }
+    }
+    /**
+     * Execute multiple hooks in parallel
+     */
+    async executeHooksParallel(hookConfigs, eventName, input) {
+        const promises = hookConfigs.map((config) => this.executeHook(config, eventName, input));
+        return await Promise.all(promises);
+    }
+    /**
+     * Execute multiple hooks sequentially
+     */
+    async executeHooksSequential(hookConfigs, eventName, input) {
+        const results = [];
+        let currentInput = input;
+        for (const config of hookConfigs) {
+            const result = await this.executeHook(config, eventName, currentInput);
+            results.push(result);
+            // If the hook succeeded and has output, use it to modify the input for the next hook
+            if (result.success && result.output) {
+                currentInput = this.applyHookOutputToInput(currentInput, result.output, eventName);
+            }
+        }
+        return results;
+    }
+    /**
+     * Apply hook output to modify input for the next hook in sequential execution
+     */
+    applyHookOutputToInput(originalInput, hookOutput, eventName) {
+        // Create a copy of the original input
+        const modifiedInput = { ...originalInput };
+        // Apply modifications based on hook output and event type
+        if (hookOutput.hookSpecificOutput) {
+            switch (eventName) {
+                case HookEventName.BeforeAgent:
+                    if ('additionalContext' in hookOutput.hookSpecificOutput) {
+                        // For BeforeAgent, we could modify the prompt with additional context
+                        const additionalContext = hookOutput.hookSpecificOutput['additionalContext'];
+                        if (typeof additionalContext === 'string' &&
+                            'prompt' in modifiedInput) {
+                            modifiedInput.prompt +=
+                                '\n\n' + additionalContext;
+                        }
+                    }
+                    break;
+                case HookEventName.BeforeModel:
+                    if ('llm_request' in hookOutput.hookSpecificOutput) {
+                        // For BeforeModel, we update the LLM request
+                        const hookBeforeModelOutput = hookOutput;
+                        if (hookBeforeModelOutput.hookSpecificOutput?.llm_request &&
+                            'llm_request' in modifiedInput) {
+                            // Merge the partial request with the existing request
+                            const currentRequest = modifiedInput
+                                .llm_request;
+                            const partialRequest = hookBeforeModelOutput.hookSpecificOutput.llm_request;
+                            modifiedInput.llm_request = {
+                                ...currentRequest,
+                                ...partialRequest,
+                            };
+                        }
+                    }
+                    break;
+                default:
+                    // For other events, no special input modification is needed
+                    break;
+            }
+        }
+        return modifiedInput;
+    }
+    /**
+     * Execute a command hook
+     */
+    async executeCommandHook(hookConfig, eventName, input, startTime) {
+        const timeout = hookConfig.timeout ?? DEFAULT_HOOK_TIMEOUT;
+        return new Promise((resolve) => {
+            if (!hookConfig.command) {
+                const errorMessage = 'Command hook missing command';
+                debugLogger.warn(`Hook configuration error (non-fatal): ${errorMessage}`);
+                resolve({
+                    hookConfig,
+                    eventName,
+                    success: false,
+                    error: new Error(errorMessage),
+                    duration: Date.now() - startTime,
+                });
+                return;
+            }
+            let stdout = '';
+            let stderr = '';
+            let timedOut = false;
+            const command = this.expandCommand(hookConfig.command, input);
+            // Set up environment variables
+            const env = {
+                ...process.env,
+                GEMINI_PROJECT_DIR: input.cwd,
+                CLAUDE_PROJECT_DIR: input.cwd, // For compatibility
+            };
+            const child = spawn(command, {
+                env,
+                cwd: input.cwd,
+                stdio: ['pipe', 'pipe', 'pipe'],
+                shell: true,
+            });
+            // Set up timeout
+            const timeoutHandle = setTimeout(() => {
+                timedOut = true;
+                child.kill('SIGTERM');
+                // Force kill after 5 seconds
+                setTimeout(() => {
+                    if (!child.killed) {
+                        child.kill('SIGKILL');
+                    }
+                }, 5000);
+            }, timeout);
+            // Send input to stdin
+            if (child.stdin) {
+                child.stdin.write(JSON.stringify(input));
+                child.stdin.end();
+            }
+            // Collect stdout
+            child.stdout?.on('data', (data) => {
+                stdout += data.toString();
+            });
+            // Collect stderr
+            child.stderr?.on('data', (data) => {
+                stderr += data.toString();
+            });
+            // Handle process exit
+            child.on('close', (exitCode) => {
+                clearTimeout(timeoutHandle);
+                const duration = Date.now() - startTime;
+                if (timedOut) {
+                    resolve({
+                        hookConfig,
+                        eventName,
+                        success: false,
+                        error: new Error(`Hook timed out after ${timeout}ms`),
+                        stdout,
+                        stderr,
+                        duration,
+                    });
+                    return;
+                }
+                // Parse output
+                let output;
+                if (exitCode === EXIT_CODE_SUCCESS && stdout.trim()) {
+                    try {
+                        let parsed = JSON.parse(stdout.trim());
+                        if (typeof parsed === 'string') {
+                            // If the output is a string, parse it in case
+                            // it's double-encoded JSON string.
+                            parsed = JSON.parse(parsed);
+                        }
+                        if (parsed) {
+                            output = parsed;
+                        }
+                    }
+                    catch {
+                        // Not JSON, convert plain text to structured output
+                        output = this.convertPlainTextToHookOutput(stdout.trim(), exitCode);
+                    }
+                }
+                else if (exitCode !== EXIT_CODE_SUCCESS && stderr.trim()) {
+                    // Convert error output to structured format
+                    output = this.convertPlainTextToHookOutput(stderr.trim(), exitCode || EXIT_CODE_NON_BLOCKING_ERROR);
+                }
+                resolve({
+                    hookConfig,
+                    eventName,
+                    success: exitCode === EXIT_CODE_SUCCESS,
+                    output,
+                    stdout,
+                    stderr,
+                    exitCode: exitCode || EXIT_CODE_SUCCESS,
+                    duration,
+                });
+            });
+            // Handle process errors
+            child.on('error', (error) => {
+                clearTimeout(timeoutHandle);
+                const duration = Date.now() - startTime;
+                resolve({
+                    hookConfig,
+                    eventName,
+                    success: false,
+                    error,
+                    stdout,
+                    stderr,
+                    duration,
+                });
+            });
+        });
+    }
+    /**
+     * Expand command with environment variables and input context
+     */
+    expandCommand(command, input) {
+        return command
+            .replace(/\$GEMINI_PROJECT_DIR/g, input.cwd)
+            .replace(/\$CLAUDE_PROJECT_DIR/g, input.cwd); // For compatibility
+    }
+    /**
+     * Convert plain text output to structured HookOutput
+     */
+    convertPlainTextToHookOutput(text, exitCode) {
+        if (exitCode === EXIT_CODE_SUCCESS) {
+            // Success - treat as system message or additional context
+            return {
+                decision: 'allow',
+                systemMessage: text,
+            };
+        }
+        else if (exitCode === EXIT_CODE_BLOCKING_ERROR) {
+            // Blocking error
+            return {
+                decision: 'deny',
+                reason: text,
+            };
+        }
+        else {
+            // Non-blocking error (EXIT_CODE_NON_BLOCKING_ERROR or any other code)
+            return {
+                decision: 'allow',
+                systemMessage: `Warning: ${text}`,
+            };
+        }
+    }
+}
+//# sourceMappingURL=hookRunner.js.map