@google/gemini-cli-core 0.19.0-nightly.20251123.dadd606c0 → 0.19.0-nightly.20251125.f6d97d448
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/google-gemini-cli-core-0.19.0-nightly.20251124.e177314a4.tgz +0 -0
- package/dist/src/availability/modelAvailabilityService.d.ts +1 -0
- package/dist/src/availability/modelAvailabilityService.js.map +1 -1
- package/dist/src/availability/modelPolicy.d.ts +42 -0
- package/dist/src/availability/modelPolicy.js +7 -0
- package/dist/src/availability/modelPolicy.js.map +1 -0
- package/dist/src/availability/policyCatalog.d.ts +20 -0
- package/dist/src/availability/policyCatalog.js +83 -0
- package/dist/src/availability/policyCatalog.js.map +1 -0
- package/dist/src/availability/policyCatalog.test.d.ts +6 -0
- package/dist/src/availability/policyCatalog.test.js +70 -0
- package/dist/src/availability/policyCatalog.test.js.map +1 -0
- package/dist/src/config/config.d.ts +6 -0
- package/dist/src/config/config.js +13 -0
- package/dist/src/config/config.js.map +1 -1
- package/dist/src/config/defaultModelConfigs.js +25 -0
- package/dist/src/config/defaultModelConfigs.js.map +1 -1
- package/dist/src/confirmation-bus/message-bus.d.ts +6 -0
- package/dist/src/confirmation-bus/message-bus.js +63 -2
- package/dist/src/confirmation-bus/message-bus.js.map +1 -1
- package/dist/src/confirmation-bus/types.d.ts +25 -2
- package/dist/src/confirmation-bus/types.js +3 -0
- package/dist/src/confirmation-bus/types.js.map +1 -1
- package/dist/src/core/client.js +75 -4
- package/dist/src/core/client.js.map +1 -1
- package/dist/src/core/client.test.js +56 -2
- package/dist/src/core/client.test.js.map +1 -1
- package/dist/src/core/clientHookTriggers.d.ts +36 -0
- package/dist/src/core/clientHookTriggers.js +76 -0
- package/dist/src/core/clientHookTriggers.js.map +1 -0
- package/dist/src/core/nonInteractiveToolExecutor.test.js +9 -1
- package/dist/src/core/nonInteractiveToolExecutor.test.js.map +1 -1
- package/dist/src/core/prompts.js +12 -1
- package/dist/src/core/prompts.js.map +1 -1
- package/dist/src/core/prompts.test.js +23 -0
- package/dist/src/core/prompts.test.js.map +1 -1
- package/dist/src/core/turn.d.ts +5 -0
- package/dist/src/core/turn.js +10 -0
- package/dist/src/core/turn.js.map +1 -1
- package/dist/src/generated/git-commit.d.ts +2 -2
- package/dist/src/generated/git-commit.js +2 -2
- package/dist/src/hooks/hookEventHandler.d.ts +109 -0
- package/dist/src/hooks/hookEventHandler.js +486 -0
- package/dist/src/hooks/hookEventHandler.js.map +1 -0
- package/dist/src/hooks/hookEventHandler.test.d.ts +6 -0
- package/dist/src/hooks/hookEventHandler.test.js +384 -0
- package/dist/src/hooks/hookEventHandler.test.js.map +1 -0
- package/dist/src/hooks/hookSystem.d.ts +48 -0
- package/dist/src/hooks/hookSystem.js +83 -0
- package/dist/src/hooks/hookSystem.js.map +1 -0
- package/dist/src/hooks/hookSystem.test.d.ts +6 -0
- package/dist/src/hooks/hookSystem.test.js +213 -0
- package/dist/src/hooks/hookSystem.test.js.map +1 -0
- package/dist/src/hooks/index.d.ts +15 -0
- package/dist/src/hooks/index.js +15 -0
- package/dist/src/hooks/index.js.map +1 -0
- package/dist/src/index.d.ts +1 -0
- package/dist/src/index.js +2 -0
- package/dist/src/index.js.map +1 -1
- package/dist/src/policy/policy-engine.d.ts +17 -1
- package/dist/src/policy/policy-engine.js +90 -1
- package/dist/src/policy/policy-engine.js.map +1 -1
- package/dist/src/policy/policy-engine.test.js +222 -0
- package/dist/src/policy/policy-engine.test.js.map +1 -1
- package/dist/src/policy/types.d.ts +51 -1
- package/dist/src/policy/types.js +21 -0
- package/dist/src/policy/types.js.map +1 -1
- package/dist/src/services/chatCompressionService.d.ts +1 -0
- package/dist/src/services/chatCompressionService.js +22 -6
- package/dist/src/services/chatCompressionService.js.map +1 -1
- package/dist/src/services/chatCompressionService.test.js +27 -32
- package/dist/src/services/chatCompressionService.test.js.map +1 -1
- package/dist/src/services/test-data/resolved-aliases.golden.json +20 -0
- package/dist/src/telemetry/loggers.d.ts +2 -1
- package/dist/src/telemetry/loggers.js +12 -1
- package/dist/src/telemetry/loggers.js.map +1 -1
- package/dist/src/telemetry/metrics.d.ts +22 -0
- package/dist/src/telemetry/metrics.js +32 -0
- package/dist/src/telemetry/metrics.js.map +1 -1
- package/dist/src/telemetry/metrics.test.js +64 -0
- package/dist/src/telemetry/metrics.test.js.map +1 -1
- package/dist/src/telemetry/sanitize.d.ts +25 -0
- package/dist/src/telemetry/sanitize.js +48 -0
- package/dist/src/telemetry/sanitize.js.map +1 -0
- package/dist/src/telemetry/sanitize.test.d.ts +6 -0
- package/dist/src/telemetry/sanitize.test.js +279 -0
- package/dist/src/telemetry/sanitize.test.js.map +1 -0
- package/dist/src/telemetry/types.d.ts +19 -0
- package/dist/src/telemetry/types.js +65 -0
- package/dist/src/telemetry/types.js.map +1 -1
- package/dist/src/test-utils/mock-message-bus.d.ts +60 -0
- package/dist/src/test-utils/mock-message-bus.js +131 -0
- package/dist/src/test-utils/mock-message-bus.js.map +1 -0
- package/dist/src/utils/package.d.ts +14 -0
- package/dist/src/utils/package.js +15 -2
- package/dist/src/utils/package.js.map +1 -1
- package/dist/tsconfig.tsbuildinfo +1 -1
- package/package.json +1 -1
- package/dist/google-gemini-cli-core-0.19.0-nightly.20251122.42c2e1b21.tgz +0 -0
|
@@ -0,0 +1,384 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @license
|
|
3
|
+
* Copyright 2025 Google LLC
|
|
4
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
5
|
+
*/
|
|
6
|
+
import { describe, it, expect, vi, beforeEach } from 'vitest';
|
|
7
|
+
import { HookEventHandler } from './hookEventHandler.js';
|
|
8
|
+
import { HookEventName, HookType } from './types.js';
|
|
9
|
+
import { NotificationType, SessionStartSource, } from './types.js';
|
|
10
|
+
// Mock debugLogger
|
|
11
|
+
const mockDebugLogger = vi.hoisted(() => ({
|
|
12
|
+
log: vi.fn(),
|
|
13
|
+
warn: vi.fn(),
|
|
14
|
+
error: vi.fn(),
|
|
15
|
+
debug: vi.fn(),
|
|
16
|
+
}));
|
|
17
|
+
vi.mock('../utils/debugLogger.js', () => ({
|
|
18
|
+
debugLogger: mockDebugLogger,
|
|
19
|
+
}));
|
|
20
|
+
describe('HookEventHandler', () => {
|
|
21
|
+
let hookEventHandler;
|
|
22
|
+
let mockConfig;
|
|
23
|
+
let mockLogger;
|
|
24
|
+
let mockHookPlanner;
|
|
25
|
+
let mockHookRunner;
|
|
26
|
+
let mockHookAggregator;
|
|
27
|
+
beforeEach(() => {
|
|
28
|
+
vi.resetAllMocks();
|
|
29
|
+
mockConfig = {
|
|
30
|
+
getSessionId: vi.fn().mockReturnValue('test-session'),
|
|
31
|
+
getWorkingDir: vi.fn().mockReturnValue('/test/project'),
|
|
32
|
+
};
|
|
33
|
+
mockLogger = {};
|
|
34
|
+
mockHookPlanner = {
|
|
35
|
+
createExecutionPlan: vi.fn(),
|
|
36
|
+
};
|
|
37
|
+
mockHookRunner = {
|
|
38
|
+
executeHooksParallel: vi.fn(),
|
|
39
|
+
executeHooksSequential: vi.fn(),
|
|
40
|
+
};
|
|
41
|
+
mockHookAggregator = {
|
|
42
|
+
aggregateResults: vi.fn(),
|
|
43
|
+
};
|
|
44
|
+
hookEventHandler = new HookEventHandler(mockConfig, mockLogger, mockHookPlanner, mockHookRunner, mockHookAggregator);
|
|
45
|
+
});
|
|
46
|
+
describe('fireBeforeToolEvent', () => {
|
|
47
|
+
it('should fire BeforeTool event with correct input', async () => {
|
|
48
|
+
const mockPlan = [
|
|
49
|
+
{
|
|
50
|
+
hookConfig: {
|
|
51
|
+
type: HookType.Command,
|
|
52
|
+
command: './test.sh',
|
|
53
|
+
},
|
|
54
|
+
eventName: HookEventName.BeforeTool,
|
|
55
|
+
},
|
|
56
|
+
];
|
|
57
|
+
const mockResults = [
|
|
58
|
+
{
|
|
59
|
+
success: true,
|
|
60
|
+
duration: 100,
|
|
61
|
+
hookConfig: {
|
|
62
|
+
type: HookType.Command,
|
|
63
|
+
command: './test.sh',
|
|
64
|
+
timeout: 30000,
|
|
65
|
+
},
|
|
66
|
+
eventName: HookEventName.BeforeTool,
|
|
67
|
+
},
|
|
68
|
+
];
|
|
69
|
+
const mockAggregated = {
|
|
70
|
+
success: true,
|
|
71
|
+
allOutputs: [],
|
|
72
|
+
errors: [],
|
|
73
|
+
totalDuration: 100,
|
|
74
|
+
};
|
|
75
|
+
vi.mocked(mockHookPlanner.createExecutionPlan).mockReturnValue({
|
|
76
|
+
eventName: HookEventName.BeforeTool,
|
|
77
|
+
hookConfigs: mockPlan.map((p) => p.hookConfig),
|
|
78
|
+
sequential: false,
|
|
79
|
+
});
|
|
80
|
+
vi.mocked(mockHookRunner.executeHooksParallel).mockResolvedValue(mockResults);
|
|
81
|
+
vi.mocked(mockHookAggregator.aggregateResults).mockReturnValue(mockAggregated);
|
|
82
|
+
const result = await hookEventHandler.fireBeforeToolEvent('EditTool', {
|
|
83
|
+
file: 'test.txt',
|
|
84
|
+
});
|
|
85
|
+
expect(mockHookPlanner.createExecutionPlan).toHaveBeenCalledWith(HookEventName.BeforeTool, { toolName: 'EditTool' });
|
|
86
|
+
expect(mockHookRunner.executeHooksParallel).toHaveBeenCalledWith([mockPlan[0].hookConfig], HookEventName.BeforeTool, expect.objectContaining({
|
|
87
|
+
session_id: 'test-session',
|
|
88
|
+
cwd: '/test/project',
|
|
89
|
+
hook_event_name: 'BeforeTool',
|
|
90
|
+
tool_name: 'EditTool',
|
|
91
|
+
tool_input: { file: 'test.txt' },
|
|
92
|
+
}));
|
|
93
|
+
expect(result).toBe(mockAggregated);
|
|
94
|
+
});
|
|
95
|
+
it('should return empty result when no hooks to execute', async () => {
|
|
96
|
+
vi.mocked(mockHookPlanner.createExecutionPlan).mockReturnValue(null);
|
|
97
|
+
const result = await hookEventHandler.fireBeforeToolEvent('EditTool', {});
|
|
98
|
+
expect(result.success).toBe(true);
|
|
99
|
+
expect(result.allOutputs).toHaveLength(0);
|
|
100
|
+
expect(result.errors).toHaveLength(0);
|
|
101
|
+
expect(result.totalDuration).toBe(0);
|
|
102
|
+
});
|
|
103
|
+
it('should handle execution errors gracefully', async () => {
|
|
104
|
+
vi.mocked(mockHookPlanner.createExecutionPlan).mockImplementation(() => {
|
|
105
|
+
throw new Error('Planning failed');
|
|
106
|
+
});
|
|
107
|
+
const result = await hookEventHandler.fireBeforeToolEvent('EditTool', {});
|
|
108
|
+
expect(result.success).toBe(false);
|
|
109
|
+
expect(result.errors).toHaveLength(1);
|
|
110
|
+
expect(result.errors[0].message).toBe('Planning failed');
|
|
111
|
+
expect(mockDebugLogger.error).toHaveBeenCalled();
|
|
112
|
+
});
|
|
113
|
+
});
|
|
114
|
+
describe('fireAfterToolEvent', () => {
|
|
115
|
+
it('should fire AfterTool event with tool response', async () => {
|
|
116
|
+
const mockPlan = [
|
|
117
|
+
{
|
|
118
|
+
hookConfig: {
|
|
119
|
+
type: HookType.Command,
|
|
120
|
+
command: './after.sh',
|
|
121
|
+
},
|
|
122
|
+
eventName: HookEventName.AfterTool,
|
|
123
|
+
},
|
|
124
|
+
];
|
|
125
|
+
const mockResults = [
|
|
126
|
+
{
|
|
127
|
+
success: true,
|
|
128
|
+
duration: 100,
|
|
129
|
+
hookConfig: {
|
|
130
|
+
type: HookType.Command,
|
|
131
|
+
command: './test.sh',
|
|
132
|
+
timeout: 30000,
|
|
133
|
+
},
|
|
134
|
+
eventName: HookEventName.BeforeTool,
|
|
135
|
+
},
|
|
136
|
+
];
|
|
137
|
+
const mockAggregated = {
|
|
138
|
+
success: true,
|
|
139
|
+
allOutputs: [],
|
|
140
|
+
errors: [],
|
|
141
|
+
totalDuration: 100,
|
|
142
|
+
};
|
|
143
|
+
vi.mocked(mockHookPlanner.createExecutionPlan).mockReturnValue({
|
|
144
|
+
eventName: HookEventName.BeforeTool,
|
|
145
|
+
hookConfigs: mockPlan.map((p) => p.hookConfig),
|
|
146
|
+
sequential: false,
|
|
147
|
+
});
|
|
148
|
+
vi.mocked(mockHookRunner.executeHooksParallel).mockResolvedValue(mockResults);
|
|
149
|
+
vi.mocked(mockHookAggregator.aggregateResults).mockReturnValue(mockAggregated);
|
|
150
|
+
const toolInput = { file: 'test.txt' };
|
|
151
|
+
const toolResponse = { success: true, content: 'File edited' };
|
|
152
|
+
const result = await hookEventHandler.fireAfterToolEvent('EditTool', toolInput, toolResponse);
|
|
153
|
+
expect(mockHookRunner.executeHooksParallel).toHaveBeenCalledWith([mockPlan[0].hookConfig], HookEventName.AfterTool, expect.objectContaining({
|
|
154
|
+
tool_name: 'EditTool',
|
|
155
|
+
tool_input: toolInput,
|
|
156
|
+
tool_response: toolResponse,
|
|
157
|
+
}));
|
|
158
|
+
expect(result).toBe(mockAggregated);
|
|
159
|
+
});
|
|
160
|
+
});
|
|
161
|
+
describe('fireBeforeAgentEvent', () => {
|
|
162
|
+
it('should fire BeforeAgent event with prompt', async () => {
|
|
163
|
+
const mockPlan = [
|
|
164
|
+
{
|
|
165
|
+
hookConfig: {
|
|
166
|
+
type: HookType.Command,
|
|
167
|
+
command: './before_agent.sh',
|
|
168
|
+
},
|
|
169
|
+
eventName: HookEventName.BeforeAgent,
|
|
170
|
+
},
|
|
171
|
+
];
|
|
172
|
+
const mockResults = [
|
|
173
|
+
{
|
|
174
|
+
success: true,
|
|
175
|
+
duration: 100,
|
|
176
|
+
hookConfig: {
|
|
177
|
+
type: HookType.Command,
|
|
178
|
+
command: './test.sh',
|
|
179
|
+
timeout: 30000,
|
|
180
|
+
},
|
|
181
|
+
eventName: HookEventName.BeforeTool,
|
|
182
|
+
},
|
|
183
|
+
];
|
|
184
|
+
const mockAggregated = {
|
|
185
|
+
success: true,
|
|
186
|
+
allOutputs: [],
|
|
187
|
+
errors: [],
|
|
188
|
+
totalDuration: 100,
|
|
189
|
+
};
|
|
190
|
+
vi.mocked(mockHookPlanner.createExecutionPlan).mockReturnValue({
|
|
191
|
+
eventName: HookEventName.BeforeTool,
|
|
192
|
+
hookConfigs: mockPlan.map((p) => p.hookConfig),
|
|
193
|
+
sequential: false,
|
|
194
|
+
});
|
|
195
|
+
vi.mocked(mockHookRunner.executeHooksParallel).mockResolvedValue(mockResults);
|
|
196
|
+
vi.mocked(mockHookAggregator.aggregateResults).mockReturnValue(mockAggregated);
|
|
197
|
+
const prompt = 'Please help me with this task';
|
|
198
|
+
const result = await hookEventHandler.fireBeforeAgentEvent(prompt);
|
|
199
|
+
expect(mockHookRunner.executeHooksParallel).toHaveBeenCalledWith([mockPlan[0].hookConfig], HookEventName.BeforeAgent, expect.objectContaining({
|
|
200
|
+
prompt,
|
|
201
|
+
}));
|
|
202
|
+
expect(result).toBe(mockAggregated);
|
|
203
|
+
});
|
|
204
|
+
});
|
|
205
|
+
describe('fireNotificationEvent', () => {
|
|
206
|
+
it('should fire Notification event', async () => {
|
|
207
|
+
const mockPlan = [
|
|
208
|
+
{
|
|
209
|
+
hookConfig: {
|
|
210
|
+
type: HookType.Command,
|
|
211
|
+
command: './notification-hook.sh',
|
|
212
|
+
},
|
|
213
|
+
eventName: HookEventName.Notification,
|
|
214
|
+
},
|
|
215
|
+
];
|
|
216
|
+
const mockResults = [
|
|
217
|
+
{
|
|
218
|
+
success: true,
|
|
219
|
+
duration: 50,
|
|
220
|
+
hookConfig: {
|
|
221
|
+
type: HookType.Command,
|
|
222
|
+
command: './notification-hook.sh',
|
|
223
|
+
timeout: 30000,
|
|
224
|
+
},
|
|
225
|
+
eventName: HookEventName.Notification,
|
|
226
|
+
},
|
|
227
|
+
];
|
|
228
|
+
const mockAggregated = {
|
|
229
|
+
success: true,
|
|
230
|
+
allOutputs: [],
|
|
231
|
+
errors: [],
|
|
232
|
+
totalDuration: 50,
|
|
233
|
+
};
|
|
234
|
+
vi.mocked(mockHookPlanner.createExecutionPlan).mockReturnValue({
|
|
235
|
+
eventName: HookEventName.Notification,
|
|
236
|
+
hookConfigs: mockPlan.map((p) => p.hookConfig),
|
|
237
|
+
sequential: false,
|
|
238
|
+
});
|
|
239
|
+
vi.mocked(mockHookRunner.executeHooksParallel).mockResolvedValue(mockResults);
|
|
240
|
+
vi.mocked(mockHookAggregator.aggregateResults).mockReturnValue(mockAggregated);
|
|
241
|
+
const message = 'Tool execution requires permission';
|
|
242
|
+
const result = await hookEventHandler.fireNotificationEvent(NotificationType.ToolPermission, message, { type: 'ToolPermission', title: 'Test Permission' });
|
|
243
|
+
expect(mockHookRunner.executeHooksParallel).toHaveBeenCalledWith([mockPlan[0].hookConfig], HookEventName.Notification, expect.objectContaining({
|
|
244
|
+
notification_type: 'ToolPermission',
|
|
245
|
+
details: { type: 'ToolPermission', title: 'Test Permission' },
|
|
246
|
+
}));
|
|
247
|
+
expect(result).toBe(mockAggregated);
|
|
248
|
+
});
|
|
249
|
+
});
|
|
250
|
+
describe('fireSessionStartEvent', () => {
|
|
251
|
+
it('should fire SessionStart event with source', async () => {
|
|
252
|
+
const mockPlan = [
|
|
253
|
+
{
|
|
254
|
+
hookConfig: {
|
|
255
|
+
type: HookType.Command,
|
|
256
|
+
command: './session_start.sh',
|
|
257
|
+
},
|
|
258
|
+
eventName: HookEventName.SessionStart,
|
|
259
|
+
},
|
|
260
|
+
];
|
|
261
|
+
const mockResults = [
|
|
262
|
+
{
|
|
263
|
+
success: true,
|
|
264
|
+
duration: 200,
|
|
265
|
+
hookConfig: {
|
|
266
|
+
type: HookType.Command,
|
|
267
|
+
command: './session_start.sh',
|
|
268
|
+
timeout: 30000,
|
|
269
|
+
},
|
|
270
|
+
eventName: HookEventName.SessionStart,
|
|
271
|
+
},
|
|
272
|
+
];
|
|
273
|
+
const mockAggregated = {
|
|
274
|
+
success: true,
|
|
275
|
+
allOutputs: [],
|
|
276
|
+
errors: [],
|
|
277
|
+
totalDuration: 200,
|
|
278
|
+
};
|
|
279
|
+
vi.mocked(mockHookPlanner.createExecutionPlan).mockReturnValue({
|
|
280
|
+
eventName: HookEventName.SessionStart,
|
|
281
|
+
hookConfigs: mockPlan.map((p) => p.hookConfig),
|
|
282
|
+
sequential: false,
|
|
283
|
+
});
|
|
284
|
+
vi.mocked(mockHookRunner.executeHooksParallel).mockResolvedValue(mockResults);
|
|
285
|
+
vi.mocked(mockHookAggregator.aggregateResults).mockReturnValue(mockAggregated);
|
|
286
|
+
const result = await hookEventHandler.fireSessionStartEvent(SessionStartSource.Startup);
|
|
287
|
+
expect(mockHookPlanner.createExecutionPlan).toHaveBeenCalledWith(HookEventName.SessionStart, { trigger: 'startup' });
|
|
288
|
+
expect(mockHookRunner.executeHooksParallel).toHaveBeenCalledWith([mockPlan[0].hookConfig], HookEventName.SessionStart, expect.objectContaining({
|
|
289
|
+
source: 'startup',
|
|
290
|
+
}));
|
|
291
|
+
expect(result).toBe(mockAggregated);
|
|
292
|
+
});
|
|
293
|
+
});
|
|
294
|
+
describe('fireBeforeModelEvent', () => {
|
|
295
|
+
it('should fire BeforeModel event with LLM request', async () => {
|
|
296
|
+
const mockPlan = [
|
|
297
|
+
{
|
|
298
|
+
hookConfig: {
|
|
299
|
+
type: HookType.Command,
|
|
300
|
+
command: './model-hook.sh',
|
|
301
|
+
},
|
|
302
|
+
eventName: HookEventName.BeforeModel,
|
|
303
|
+
},
|
|
304
|
+
];
|
|
305
|
+
const mockResults = [
|
|
306
|
+
{
|
|
307
|
+
success: true,
|
|
308
|
+
duration: 150,
|
|
309
|
+
hookConfig: {
|
|
310
|
+
type: HookType.Command,
|
|
311
|
+
command: './model-hook.sh',
|
|
312
|
+
timeout: 30000,
|
|
313
|
+
},
|
|
314
|
+
eventName: HookEventName.BeforeModel,
|
|
315
|
+
},
|
|
316
|
+
];
|
|
317
|
+
const mockAggregated = {
|
|
318
|
+
success: true,
|
|
319
|
+
allOutputs: [],
|
|
320
|
+
errors: [],
|
|
321
|
+
totalDuration: 150,
|
|
322
|
+
};
|
|
323
|
+
vi.mocked(mockHookPlanner.createExecutionPlan).mockReturnValue({
|
|
324
|
+
eventName: HookEventName.BeforeModel,
|
|
325
|
+
hookConfigs: mockPlan.map((p) => p.hookConfig),
|
|
326
|
+
sequential: false,
|
|
327
|
+
});
|
|
328
|
+
vi.mocked(mockHookRunner.executeHooksParallel).mockResolvedValue(mockResults);
|
|
329
|
+
vi.mocked(mockHookAggregator.aggregateResults).mockReturnValue(mockAggregated);
|
|
330
|
+
const llmRequest = {
|
|
331
|
+
model: 'gemini-pro',
|
|
332
|
+
config: { temperature: 0.7 },
|
|
333
|
+
contents: [{ role: 'user', parts: [{ text: 'Hello' }] }],
|
|
334
|
+
};
|
|
335
|
+
const result = await hookEventHandler.fireBeforeModelEvent(llmRequest);
|
|
336
|
+
expect(mockHookRunner.executeHooksParallel).toHaveBeenCalledWith([mockPlan[0].hookConfig], HookEventName.BeforeModel, expect.objectContaining({
|
|
337
|
+
llm_request: expect.objectContaining({
|
|
338
|
+
model: 'gemini-pro',
|
|
339
|
+
messages: expect.arrayContaining([
|
|
340
|
+
expect.objectContaining({
|
|
341
|
+
role: 'user',
|
|
342
|
+
content: 'Hello',
|
|
343
|
+
}),
|
|
344
|
+
]),
|
|
345
|
+
}),
|
|
346
|
+
}));
|
|
347
|
+
expect(result).toBe(mockAggregated);
|
|
348
|
+
});
|
|
349
|
+
});
|
|
350
|
+
describe('createBaseInput', () => {
|
|
351
|
+
it('should create base input with correct fields', async () => {
|
|
352
|
+
const mockPlan = [
|
|
353
|
+
{
|
|
354
|
+
hookConfig: {
|
|
355
|
+
type: HookType.Command,
|
|
356
|
+
command: './test.sh',
|
|
357
|
+
},
|
|
358
|
+
eventName: HookEventName.BeforeTool,
|
|
359
|
+
},
|
|
360
|
+
];
|
|
361
|
+
vi.mocked(mockHookPlanner.createExecutionPlan).mockReturnValue({
|
|
362
|
+
eventName: HookEventName.BeforeTool,
|
|
363
|
+
hookConfigs: mockPlan.map((p) => p.hookConfig),
|
|
364
|
+
sequential: false,
|
|
365
|
+
});
|
|
366
|
+
vi.mocked(mockHookRunner.executeHooksParallel).mockResolvedValue([]);
|
|
367
|
+
vi.mocked(mockHookAggregator.aggregateResults).mockReturnValue({
|
|
368
|
+
success: true,
|
|
369
|
+
allOutputs: [],
|
|
370
|
+
errors: [],
|
|
371
|
+
totalDuration: 0,
|
|
372
|
+
});
|
|
373
|
+
await hookEventHandler.fireBeforeToolEvent('TestTool', {});
|
|
374
|
+
expect(mockHookRunner.executeHooksParallel).toHaveBeenCalledWith(expect.any(Array), HookEventName.BeforeTool, expect.objectContaining({
|
|
375
|
+
session_id: 'test-session',
|
|
376
|
+
transcript_path: '',
|
|
377
|
+
cwd: '/test/project',
|
|
378
|
+
hook_event_name: 'BeforeTool',
|
|
379
|
+
timestamp: expect.any(String),
|
|
380
|
+
}));
|
|
381
|
+
});
|
|
382
|
+
});
|
|
383
|
+
});
|
|
384
|
+
//# sourceMappingURL=hookEventHandler.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"hookEventHandler.test.js","sourceRoot":"","sources":["../../../src/hooks/hookEventHandler.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAOzD,OAAO,EAAE,aAAa,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AACrD,OAAO,EACL,gBAAgB,EAChB,kBAAkB,GAEnB,MAAM,YAAY,CAAC;AAEpB,mBAAmB;AACnB,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,kBAAkB,EAAE,GAAG,EAAE;IAChC,IAAI,gBAAkC,CAAC;IACvC,IAAI,UAAkB,CAAC;IACvB,IAAI,UAAkB,CAAC;IACvB,IAAI,eAA4B,CAAC;IACjC,IAAI,cAA0B,CAAC;IAC/B,IAAI,kBAAkC,CAAC;IAEvC,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QAEnB,UAAU,GAAG;YACX,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,cAAc,CAAC;YACrD,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,eAAe,CAAC;SACnC,CAAC;QAEvB,UAAU,GAAG,EAAY,CAAC;QAE1B,eAAe,GAAG;YAChB,mBAAmB,EAAE,EAAE,CAAC,EAAE,EAAE;SACH,CAAC;QAE5B,cAAc,GAAG;YACf,oBAAoB,EAAE,EAAE,CAAC,EAAE,EAAE;YAC7B,sBAAsB,EAAE,EAAE,CAAC,EAAE,EAAE;SACP,CAAC;QAE3B,kBAAkB,GAAG;YACnB,gBAAgB,EAAE,EAAE,CAAC,EAAE,EAAE;SACG,CAAC;QAE/B,gBAAgB,GAAG,IAAI,gBAAgB,CACrC,UAAU,EACV,UAAU,EACV,eAAe,EACf,cAAc,EACd,kBAAkB,CACnB,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC/D,MAAM,QAAQ,GAAG;gBACf;oBACE,UAAU,EAAE;wBACV,IAAI,EAAE,QAAQ,CAAC,OAAO;wBACtB,OAAO,EAAE,WAAW;qBACI;oBAC1B,SAAS,EAAE,aAAa,CAAC,UAAU;iBACpC;aACF,CAAC;YACF,MAAM,WAAW,GAA0B;gBACzC;oBACE,OAAO,EAAE,IAAI;oBACb,QAAQ,EAAE,GAAG;oBACb,UAAU,EAAE;wBACV,IAAI,EAAE,QAAQ,CAAC,OAAO;wBACtB,OAAO,EAAE,WAAW;wBACpB,OAAO,EAAE,KAAK;qBACf;oBACD,SAAS,EAAE,aAAa,CAAC,UAAU;iBACpC;aACF,CAAC;YACF,MAAM,cAAc,GAAG;gBACrB,OAAO,EAAE,IAAI;gBACb,UAAU,EAAE,EAAE;gBACd,MAAM,EAAE,EAAE;gBACV,aAAa,EAAE,GAAG;aACnB,CAAC;YAEF,EAAE,CAAC,MAAM,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC,eAAe,CAAC;gBAC7D,SAAS,EAAE,aAAa,CAAC,UAAU;gBACnC,WAAW,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC;gBAC9C,UAAU,EAAE,KAAK;aAClB,CAAC,CAAC;YACH,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,oBAAoB,CAAC,CAAC,iBAAiB,CAC9D,WAAW,CACZ,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,kBAAkB,CAAC,gBAAgB,CAAC,CAAC,eAAe,CAC5D,cAAc,CACf,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,gBAAgB,CAAC,mBAAmB,CAAC,UAAU,EAAE;gBACpE,IAAI,EAAE,UAAU;aACjB,CAAC,CAAC;YAEH,MAAM,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC,oBAAoB,CAC9D,aAAa,CAAC,UAAU,EACxB,EAAE,QAAQ,EAAE,UAAU,EAAE,CACzB,CAAC;YAEF,MAAM,CAAC,cAAc,CAAC,oBAAoB,CAAC,CAAC,oBAAoB,CAC9D,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,EACxB,aAAa,CAAC,UAAU,EACxB,MAAM,CAAC,gBAAgB,CAAC;gBACtB,UAAU,EAAE,cAAc;gBAC1B,GAAG,EAAE,eAAe;gBACpB,eAAe,EAAE,YAAY;gBAC7B,SAAS,EAAE,UAAU;gBACrB,UAAU,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE;aACjC,CAAC,CACH,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,EAAE,CAAC,MAAM,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAErE,MAAM,MAAM,GAAG,MAAM,gBAAgB,CAAC,mBAAmB,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;YAE1E,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC1C,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACtC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,EAAE,CAAC,MAAM,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE;gBACrE,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;YACrC,CAAC,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,gBAAgB,CAAC,mBAAmB,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;YAE1E,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACnC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACtC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;YACzD,MAAM,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACnD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,QAAQ,GAAG;gBACf;oBACE,UAAU,EAAE;wBACV,IAAI,EAAE,QAAQ,CAAC,OAAO;wBACtB,OAAO,EAAE,YAAY;qBACG;oBAC1B,SAAS,EAAE,aAAa,CAAC,SAAS;iBACnC;aACF,CAAC;YACF,MAAM,WAAW,GAA0B;gBACzC;oBACE,OAAO,EAAE,IAAI;oBACb,QAAQ,EAAE,GAAG;oBACb,UAAU,EAAE;wBACV,IAAI,EAAE,QAAQ,CAAC,OAAO;wBACtB,OAAO,EAAE,WAAW;wBACpB,OAAO,EAAE,KAAK;qBACf;oBACD,SAAS,EAAE,aAAa,CAAC,UAAU;iBACpC;aACF,CAAC;YACF,MAAM,cAAc,GAAG;gBACrB,OAAO,EAAE,IAAI;gBACb,UAAU,EAAE,EAAE;gBACd,MAAM,EAAE,EAAE;gBACV,aAAa,EAAE,GAAG;aACnB,CAAC;YAEF,EAAE,CAAC,MAAM,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC,eAAe,CAAC;gBAC7D,SAAS,EAAE,aAAa,CAAC,UAAU;gBACnC,WAAW,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC;gBAC9C,UAAU,EAAE,KAAK;aAClB,CAAC,CAAC;YACH,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,oBAAoB,CAAC,CAAC,iBAAiB,CAC9D,WAAW,CACZ,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,kBAAkB,CAAC,gBAAgB,CAAC,CAAC,eAAe,CAC5D,cAAc,CACf,CAAC;YAEF,MAAM,SAAS,GAAG,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;YACvC,MAAM,YAAY,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC;YAE/D,MAAM,MAAM,GAAG,MAAM,gBAAgB,CAAC,kBAAkB,CACtD,UAAU,EACV,SAAS,EACT,YAAY,CACb,CAAC;YAEF,MAAM,CAAC,cAAc,CAAC,oBAAoB,CAAC,CAAC,oBAAoB,CAC9D,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,EACxB,aAAa,CAAC,SAAS,EACvB,MAAM,CAAC,gBAAgB,CAAC;gBACtB,SAAS,EAAE,UAAU;gBACrB,UAAU,EAAE,SAAS;gBACrB,aAAa,EAAE,YAAY;aAC5B,CAAC,CACH,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,QAAQ,GAAG;gBACf;oBACE,UAAU,EAAE;wBACV,IAAI,EAAE,QAAQ,CAAC,OAAO;wBACtB,OAAO,EAAE,mBAAmB;qBACJ;oBAC1B,SAAS,EAAE,aAAa,CAAC,WAAW;iBACrC;aACF,CAAC;YACF,MAAM,WAAW,GAA0B;gBACzC;oBACE,OAAO,EAAE,IAAI;oBACb,QAAQ,EAAE,GAAG;oBACb,UAAU,EAAE;wBACV,IAAI,EAAE,QAAQ,CAAC,OAAO;wBACtB,OAAO,EAAE,WAAW;wBACpB,OAAO,EAAE,KAAK;qBACf;oBACD,SAAS,EAAE,aAAa,CAAC,UAAU;iBACpC;aACF,CAAC;YACF,MAAM,cAAc,GAAG;gBACrB,OAAO,EAAE,IAAI;gBACb,UAAU,EAAE,EAAE;gBACd,MAAM,EAAE,EAAE;gBACV,aAAa,EAAE,GAAG;aACnB,CAAC;YAEF,EAAE,CAAC,MAAM,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC,eAAe,CAAC;gBAC7D,SAAS,EAAE,aAAa,CAAC,UAAU;gBACnC,WAAW,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC;gBAC9C,UAAU,EAAE,KAAK;aAClB,CAAC,CAAC;YACH,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,oBAAoB,CAAC,CAAC,iBAAiB,CAC9D,WAAW,CACZ,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,kBAAkB,CAAC,gBAAgB,CAAC,CAAC,eAAe,CAC5D,cAAc,CACf,CAAC;YAEF,MAAM,MAAM,GAAG,+BAA+B,CAAC;YAE/C,MAAM,MAAM,GAAG,MAAM,gBAAgB,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC;YAEnE,MAAM,CAAC,cAAc,CAAC,oBAAoB,CAAC,CAAC,oBAAoB,CAC9D,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,EACxB,aAAa,CAAC,WAAW,EACzB,MAAM,CAAC,gBAAgB,CAAC;gBACtB,MAAM;aACP,CAAC,CACH,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC9C,MAAM,QAAQ,GAAG;gBACf;oBACE,UAAU,EAAE;wBACV,IAAI,EAAE,QAAQ,CAAC,OAAO;wBACtB,OAAO,EAAE,wBAAwB;qBACpB;oBACf,SAAS,EAAE,aAAa,CAAC,YAAY;iBACtC;aACF,CAAC;YACF,MAAM,WAAW,GAA0B;gBACzC;oBACE,OAAO,EAAE,IAAI;oBACb,QAAQ,EAAE,EAAE;oBACZ,UAAU,EAAE;wBACV,IAAI,EAAE,QAAQ,CAAC,OAAO;wBACtB,OAAO,EAAE,wBAAwB;wBACjC,OAAO,EAAE,KAAK;qBACf;oBACD,SAAS,EAAE,aAAa,CAAC,YAAY;iBACtC;aACF,CAAC;YACF,MAAM,cAAc,GAAG;gBACrB,OAAO,EAAE,IAAI;gBACb,UAAU,EAAE,EAAE;gBACd,MAAM,EAAE,EAAE;gBACV,aAAa,EAAE,EAAE;aAClB,CAAC;YAEF,EAAE,CAAC,MAAM,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC,eAAe,CAAC;gBAC7D,SAAS,EAAE,aAAa,CAAC,YAAY;gBACrC,WAAW,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC;gBAC9C,UAAU,EAAE,KAAK;aAClB,CAAC,CAAC;YACH,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,oBAAoB,CAAC,CAAC,iBAAiB,CAC9D,WAAW,CACZ,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,kBAAkB,CAAC,gBAAgB,CAAC,CAAC,eAAe,CAC5D,cAAc,CACf,CAAC;YAEF,MAAM,OAAO,GAAG,oCAAoC,CAAC;YAErD,MAAM,MAAM,GAAG,MAAM,gBAAgB,CAAC,qBAAqB,CACzD,gBAAgB,CAAC,cAAc,EAC/B,OAAO,EACP,EAAE,IAAI,EAAE,gBAAgB,EAAE,KAAK,EAAE,iBAAiB,EAAE,CACrD,CAAC;YAEF,MAAM,CAAC,cAAc,CAAC,oBAAoB,CAAC,CAAC,oBAAoB,CAC9D,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,EACxB,aAAa,CAAC,YAAY,EAC1B,MAAM,CAAC,gBAAgB,CAAC;gBACtB,iBAAiB,EAAE,gBAAgB;gBACnC,OAAO,EAAE,EAAE,IAAI,EAAE,gBAAgB,EAAE,KAAK,EAAE,iBAAiB,EAAE;aAC9D,CAAC,CACH,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,QAAQ,GAAG;gBACf;oBACE,UAAU,EAAE;wBACV,IAAI,EAAE,QAAQ,CAAC,OAAO;wBACtB,OAAO,EAAE,oBAAoB;qBACL;oBAC1B,SAAS,EAAE,aAAa,CAAC,YAAY;iBACtC;aACF,CAAC;YACF,MAAM,WAAW,GAA0B;gBACzC;oBACE,OAAO,EAAE,IAAI;oBACb,QAAQ,EAAE,GAAG;oBACb,UAAU,EAAE;wBACV,IAAI,EAAE,QAAQ,CAAC,OAAO;wBACtB,OAAO,EAAE,oBAAoB;wBAC7B,OAAO,EAAE,KAAK;qBACf;oBACD,SAAS,EAAE,aAAa,CAAC,YAAY;iBACtC;aACF,CAAC;YACF,MAAM,cAAc,GAAG;gBACrB,OAAO,EAAE,IAAI;gBACb,UAAU,EAAE,EAAE;gBACd,MAAM,EAAE,EAAE;gBACV,aAAa,EAAE,GAAG;aACnB,CAAC;YAEF,EAAE,CAAC,MAAM,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC,eAAe,CAAC;gBAC7D,SAAS,EAAE,aAAa,CAAC,YAAY;gBACrC,WAAW,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC;gBAC9C,UAAU,EAAE,KAAK;aAClB,CAAC,CAAC;YACH,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,oBAAoB,CAAC,CAAC,iBAAiB,CAC9D,WAAW,CACZ,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,kBAAkB,CAAC,gBAAgB,CAAC,CAAC,eAAe,CAC5D,cAAc,CACf,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,gBAAgB,CAAC,qBAAqB,CACzD,kBAAkB,CAAC,OAAO,CAC3B,CAAC;YAEF,MAAM,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC,oBAAoB,CAC9D,aAAa,CAAC,YAAY,EAC1B,EAAE,OAAO,EAAE,SAAS,EAAE,CACvB,CAAC;YAEF,MAAM,CAAC,cAAc,CAAC,oBAAoB,CAAC,CAAC,oBAAoB,CAC9D,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,EACxB,aAAa,CAAC,YAAY,EAC1B,MAAM,CAAC,gBAAgB,CAAC;gBACtB,MAAM,EAAE,SAAS;aAClB,CAAC,CACH,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,QAAQ,GAAG;gBACf;oBACE,UAAU,EAAE;wBACV,IAAI,EAAE,QAAQ,CAAC,OAAO;wBACtB,OAAO,EAAE,iBAAiB;qBACb;oBACf,SAAS,EAAE,aAAa,CAAC,WAAW;iBACrC;aACF,CAAC;YACF,MAAM,WAAW,GAA0B;gBACzC;oBACE,OAAO,EAAE,IAAI;oBACb,QAAQ,EAAE,GAAG;oBACb,UAAU,EAAE;wBACV,IAAI,EAAE,QAAQ,CAAC,OAAO;wBACtB,OAAO,EAAE,iBAAiB;wBAC1B,OAAO,EAAE,KAAK;qBACf;oBACD,SAAS,EAAE,aAAa,CAAC,WAAW;iBACrC;aACF,CAAC;YACF,MAAM,cAAc,GAAG;gBACrB,OAAO,EAAE,IAAI;gBACb,UAAU,EAAE,EAAE;gBACd,MAAM,EAAE,EAAE;gBACV,aAAa,EAAE,GAAG;aACnB,CAAC;YAEF,EAAE,CAAC,MAAM,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC,eAAe,CAAC;gBAC7D,SAAS,EAAE,aAAa,CAAC,WAAW;gBACpC,WAAW,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC;gBAC9C,UAAU,EAAE,KAAK;aAClB,CAAC,CAAC;YACH,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,oBAAoB,CAAC,CAAC,iBAAiB,CAC9D,WAAW,CACZ,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,kBAAkB,CAAC,gBAAgB,CAAC,CAAC,eAAe,CAC5D,cAAc,CACf,CAAC;YAEF,MAAM,UAAU,GAAG;gBACjB,KAAK,EAAE,YAAY;gBACnB,MAAM,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE;gBAC5B,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;aACzD,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,gBAAgB,CAAC,oBAAoB,CAAC,UAAU,CAAC,CAAC;YAEvE,MAAM,CAAC,cAAc,CAAC,oBAAoB,CAAC,CAAC,oBAAoB,CAC9D,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,EACxB,aAAa,CAAC,WAAW,EACzB,MAAM,CAAC,gBAAgB,CAAC;gBACtB,WAAW,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBACnC,KAAK,EAAE,YAAY;oBACnB,QAAQ,EAAE,MAAM,CAAC,eAAe,CAAC;wBAC/B,MAAM,CAAC,gBAAgB,CAAC;4BACtB,IAAI,EAAE,MAAM;4BACZ,OAAO,EAAE,OAAO;yBACjB,CAAC;qBACH,CAAC;iBACH,CAAC;aACH,CAAC,CACH,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,MAAM,QAAQ,GAAG;gBACf;oBACE,UAAU,EAAE;wBACV,IAAI,EAAE,QAAQ,CAAC,OAAO;wBACtB,OAAO,EAAE,WAAW;qBACI;oBAC1B,SAAS,EAAE,aAAa,CAAC,UAAU;iBACpC;aACF,CAAC;YAEF,EAAE,CAAC,MAAM,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC,eAAe,CAAC;gBAC7D,SAAS,EAAE,aAAa,CAAC,UAAU;gBACnC,WAAW,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC;gBAC9C,UAAU,EAAE,KAAK;aAClB,CAAC,CAAC;YACH,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,oBAAoB,CAAC,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YACrE,EAAE,CAAC,MAAM,CAAC,kBAAkB,CAAC,gBAAgB,CAAC,CAAC,eAAe,CAAC;gBAC7D,OAAO,EAAE,IAAI;gBACb,UAAU,EAAE,EAAE;gBACd,MAAM,EAAE,EAAE;gBACV,aAAa,EAAE,CAAC;aACjB,CAAC,CAAC;YAEH,MAAM,gBAAgB,CAAC,mBAAmB,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;YAE3D,MAAM,CAAC,cAAc,CAAC,oBAAoB,CAAC,CAAC,oBAAoB,CAC9D,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EACjB,aAAa,CAAC,UAAU,EACxB,MAAM,CAAC,gBAAgB,CAAC;gBACtB,UAAU,EAAE,cAAc;gBAC1B,eAAe,EAAE,EAAE;gBACnB,GAAG,EAAE,eAAe;gBACpB,eAAe,EAAE,YAAY;gBAC7B,SAAS,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;aAC9B,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,48 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @license
|
|
3
|
+
* Copyright 2025 Google LLC
|
|
4
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
5
|
+
*/
|
|
6
|
+
import type { Config } from '../config/config.js';
|
|
7
|
+
import { HookRegistry } from './hookRegistry.js';
|
|
8
|
+
import { HookEventHandler } from './hookEventHandler.js';
|
|
9
|
+
import type { HookRegistryEntry } from './hookRegistry.js';
|
|
10
|
+
/**
|
|
11
|
+
* Main hook system that coordinates all hook-related functionality
|
|
12
|
+
*/
|
|
13
|
+
export declare class HookSystem {
|
|
14
|
+
private readonly hookRegistry;
|
|
15
|
+
private readonly hookRunner;
|
|
16
|
+
private readonly hookAggregator;
|
|
17
|
+
private readonly hookPlanner;
|
|
18
|
+
private readonly hookEventHandler;
|
|
19
|
+
private initialized;
|
|
20
|
+
constructor(config: Config);
|
|
21
|
+
/**
|
|
22
|
+
* Initialize the hook system
|
|
23
|
+
*/
|
|
24
|
+
initialize(): Promise<void>;
|
|
25
|
+
/**
|
|
26
|
+
* Get the hook event bus for firing events
|
|
27
|
+
*/
|
|
28
|
+
getEventHandler(): HookEventHandler;
|
|
29
|
+
/**
|
|
30
|
+
* Get hook registry for management operations
|
|
31
|
+
*/
|
|
32
|
+
getRegistry(): HookRegistry;
|
|
33
|
+
/**
|
|
34
|
+
* Enable or disable a hook
|
|
35
|
+
*/
|
|
36
|
+
setHookEnabled(hookName: string, enabled: boolean): void;
|
|
37
|
+
/**
|
|
38
|
+
* Get all registered hooks for display/management
|
|
39
|
+
*/
|
|
40
|
+
getAllHooks(): HookRegistryEntry[];
|
|
41
|
+
/**
|
|
42
|
+
* Get hook system status for debugging
|
|
43
|
+
*/
|
|
44
|
+
getStatus(): {
|
|
45
|
+
initialized: boolean;
|
|
46
|
+
totalHooks: number;
|
|
47
|
+
};
|
|
48
|
+
}
|
|
@@ -0,0 +1,83 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @license
|
|
3
|
+
* Copyright 2025 Google LLC
|
|
4
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
5
|
+
*/
|
|
6
|
+
import { HookRegistry } from './hookRegistry.js';
|
|
7
|
+
import { HookRunner } from './hookRunner.js';
|
|
8
|
+
import { HookAggregator } from './hookAggregator.js';
|
|
9
|
+
import { HookPlanner } from './hookPlanner.js';
|
|
10
|
+
import { HookEventHandler } from './hookEventHandler.js';
|
|
11
|
+
import { logs } from '@opentelemetry/api-logs';
|
|
12
|
+
import { SERVICE_NAME } from '../telemetry/constants.js';
|
|
13
|
+
import { debugLogger } from '../utils/debugLogger.js';
|
|
14
|
+
/**
|
|
15
|
+
* Main hook system that coordinates all hook-related functionality
|
|
16
|
+
*/
|
|
17
|
+
export class HookSystem {
|
|
18
|
+
hookRegistry;
|
|
19
|
+
hookRunner;
|
|
20
|
+
hookAggregator;
|
|
21
|
+
hookPlanner;
|
|
22
|
+
hookEventHandler;
|
|
23
|
+
initialized = false;
|
|
24
|
+
constructor(config) {
|
|
25
|
+
const logger = logs.getLogger(SERVICE_NAME);
|
|
26
|
+
const messageBus = config.getMessageBus();
|
|
27
|
+
// Initialize components
|
|
28
|
+
this.hookRegistry = new HookRegistry(config);
|
|
29
|
+
this.hookRunner = new HookRunner();
|
|
30
|
+
this.hookAggregator = new HookAggregator();
|
|
31
|
+
this.hookPlanner = new HookPlanner(this.hookRegistry);
|
|
32
|
+
this.hookEventHandler = new HookEventHandler(config, logger, this.hookPlanner, this.hookRunner, this.hookAggregator, messageBus);
|
|
33
|
+
}
|
|
34
|
+
/**
|
|
35
|
+
* Initialize the hook system
|
|
36
|
+
*/
|
|
37
|
+
async initialize() {
|
|
38
|
+
if (this.initialized) {
|
|
39
|
+
return;
|
|
40
|
+
}
|
|
41
|
+
await this.hookRegistry.initialize();
|
|
42
|
+
this.initialized = true;
|
|
43
|
+
debugLogger.debug('Hook system initialized successfully');
|
|
44
|
+
}
|
|
45
|
+
/**
|
|
46
|
+
* Get the hook event bus for firing events
|
|
47
|
+
*/
|
|
48
|
+
getEventHandler() {
|
|
49
|
+
if (!this.initialized) {
|
|
50
|
+
throw new Error('Hook system not initialized');
|
|
51
|
+
}
|
|
52
|
+
return this.hookEventHandler;
|
|
53
|
+
}
|
|
54
|
+
/**
|
|
55
|
+
* Get hook registry for management operations
|
|
56
|
+
*/
|
|
57
|
+
getRegistry() {
|
|
58
|
+
return this.hookRegistry;
|
|
59
|
+
}
|
|
60
|
+
/**
|
|
61
|
+
* Enable or disable a hook
|
|
62
|
+
*/
|
|
63
|
+
setHookEnabled(hookName, enabled) {
|
|
64
|
+
this.hookRegistry.setHookEnabled(hookName, enabled);
|
|
65
|
+
}
|
|
66
|
+
/**
|
|
67
|
+
* Get all registered hooks for display/management
|
|
68
|
+
*/
|
|
69
|
+
getAllHooks() {
|
|
70
|
+
return this.hookRegistry.getAllHooks();
|
|
71
|
+
}
|
|
72
|
+
/**
|
|
73
|
+
* Get hook system status for debugging
|
|
74
|
+
*/
|
|
75
|
+
getStatus() {
|
|
76
|
+
const allHooks = this.initialized ? this.hookRegistry.getAllHooks() : [];
|
|
77
|
+
return {
|
|
78
|
+
initialized: this.initialized,
|
|
79
|
+
totalHooks: allHooks.length,
|
|
80
|
+
};
|
|
81
|
+
}
|
|
82
|
+
}
|
|
83
|
+
//# sourceMappingURL=hookSystem.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"hookSystem.js","sourceRoot":"","sources":["../../../src/hooks/hookSystem.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjD,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AACrD,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAC/C,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAEzD,OAAO,EAAE,IAAI,EAAe,MAAM,yBAAyB,CAAC;AAC5D,OAAO,EAAE,YAAY,EAAE,MAAM,2BAA2B,CAAC;AACzD,OAAO,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AAEtD;;GAEG;AACH,MAAM,OAAO,UAAU;IACJ,YAAY,CAAe;IAC3B,UAAU,CAAa;IACvB,cAAc,CAAiB;IAC/B,WAAW,CAAc;IACzB,gBAAgB,CAAmB;IAC5C,WAAW,GAAG,KAAK,CAAC;IAE5B,YAAY,MAAc;QACxB,MAAM,MAAM,GAAW,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;QACpD,MAAM,UAAU,GAAG,MAAM,CAAC,aAAa,EAAE,CAAC;QAE1C,wBAAwB;QACxB,IAAI,CAAC,YAAY,GAAG,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC;QAC7C,IAAI,CAAC,UAAU,GAAG,IAAI,UAAU,EAAE,CAAC;QACnC,IAAI,CAAC,cAAc,GAAG,IAAI,cAAc,EAAE,CAAC;QAC3C,IAAI,CAAC,WAAW,GAAG,IAAI,WAAW,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACtD,IAAI,CAAC,gBAAgB,GAAG,IAAI,gBAAgB,CAC1C,MAAM,EACN,MAAM,EACN,IAAI,CAAC,WAAW,EAChB,IAAI,CAAC,UAAU,EACf,IAAI,CAAC,cAAc,EACnB,UAAU,CACX,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU;QACd,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,OAAO;QACT,CAAC;QAED,MAAM,IAAI,CAAC,YAAY,CAAC,UAAU,EAAE,CAAC;QACrC,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QACxB,WAAW,CAAC,KAAK,CAAC,sCAAsC,CAAC,CAAC;IAC5D,CAAC;IAED;;OAEG;IACH,eAAe;QACb,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;QACjD,CAAC;QACD,OAAO,IAAI,CAAC,gBAAgB,CAAC;IAC/B,CAAC;IAED;;OAEG;IACH,WAAW;QACT,OAAO,IAAI,CAAC,YAAY,CAAC;IAC3B,CAAC;IAED;;OAEG;IACH,cAAc,CAAC,QAAgB,EAAE,OAAgB;QAC/C,IAAI,CAAC,YAAY,CAAC,cAAc,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IACtD,CAAC;IAED;;OAEG;IACH,WAAW;QACT,OAAO,IAAI,CAAC,YAAY,CAAC,WAAW,EAAE,CAAC;IACzC,CAAC;IAED;;OAEG;IACH,SAAS;QAIP,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAEzE,OAAO;YACL,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,UAAU,EAAE,QAAQ,CAAC,MAAM;SAC5B,CAAC;IACJ,CAAC;CACF"}
|