@polka-codes/cli 0.10.23 → 0.10.24
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/index.js +1 -1
- package/package.json +3 -3
- package/dist/ApiProviderConfig.js +0 -57
- package/dist/ApiProviderConfig.js.map +0 -1
- package/dist/ApiProviderConfig.test.js +0 -278
- package/dist/ApiProviderConfig.test.js.map +0 -1
- package/dist/agent/advanced-discovery.js +0 -604
- package/dist/agent/advanced-discovery.js.map +0 -1
- package/dist/agent/config.js +0 -139
- package/dist/agent/config.js.map +0 -1
- package/dist/agent/config.test.js +0 -140
- package/dist/agent/config.test.js.map +0 -1
- package/dist/agent/constants.js +0 -172
- package/dist/agent/constants.js.map +0 -1
- package/dist/agent/constants.test.js +0 -132
- package/dist/agent/constants.test.js.map +0 -1
- package/dist/agent/debug-logger.js +0 -281
- package/dist/agent/debug-logger.js.map +0 -1
- package/dist/agent/debug-logger.test.js +0 -294
- package/dist/agent/debug-logger.test.js.map +0 -1
- package/dist/agent/error-handling.js +0 -114
- package/dist/agent/error-handling.js.map +0 -1
- package/dist/agent/error-handling.test.js +0 -191
- package/dist/agent/error-handling.test.js.map +0 -1
- package/dist/agent/errors.js +0 -301
- package/dist/agent/errors.js.map +0 -1
- package/dist/agent/executor.js +0 -206
- package/dist/agent/executor.js.map +0 -1
- package/dist/agent/executor.test.js +0 -228
- package/dist/agent/executor.test.js.map +0 -1
- package/dist/agent/goal-decomposer.js +0 -281
- package/dist/agent/goal-decomposer.js.map +0 -1
- package/dist/agent/goal-decomposer.test.js +0 -225
- package/dist/agent/goal-decomposer.test.js.map +0 -1
- package/dist/agent/health-monitor.js +0 -89
- package/dist/agent/health-monitor.js.map +0 -1
- package/dist/agent/improvement-loop.js +0 -177
- package/dist/agent/improvement-loop.js.map +0 -1
- package/dist/agent/index.js +0 -38
- package/dist/agent/index.js.map +0 -1
- package/dist/agent/metrics.js +0 -158
- package/dist/agent/metrics.js.map +0 -1
- package/dist/agent/metrics.test.js +0 -311
- package/dist/agent/metrics.test.js.map +0 -1
- package/dist/agent/orchestrator.js +0 -438
- package/dist/agent/orchestrator.js.map +0 -1
- package/dist/agent/planner.js +0 -199
- package/dist/agent/planner.js.map +0 -1
- package/dist/agent/planner.test.js +0 -135
- package/dist/agent/planner.test.js.map +0 -1
- package/dist/agent/progress.js +0 -298
- package/dist/agent/progress.js.map +0 -1
- package/dist/agent/progress.test.js +0 -255
- package/dist/agent/progress.test.js.map +0 -1
- package/dist/agent/resource-monitor.js +0 -114
- package/dist/agent/resource-monitor.js.map +0 -1
- package/dist/agent/safety/approval.js +0 -178
- package/dist/agent/safety/approval.js.map +0 -1
- package/dist/agent/safety/approval.test.js +0 -142
- package/dist/agent/safety/approval.test.js.map +0 -1
- package/dist/agent/safety/checks.js +0 -152
- package/dist/agent/safety/checks.js.map +0 -1
- package/dist/agent/safety/checks.test.js +0 -89
- package/dist/agent/safety/checks.test.js.map +0 -1
- package/dist/agent/safety/interrupt.js +0 -92
- package/dist/agent/safety/interrupt.js.map +0 -1
- package/dist/agent/safety/interrupt.test.js +0 -53
- package/dist/agent/safety/interrupt.test.js.map +0 -1
- package/dist/agent/session.js +0 -117
- package/dist/agent/session.js.map +0 -1
- package/dist/agent/session.test.js +0 -213
- package/dist/agent/session.test.js.map +0 -1
- package/dist/agent/state-manager.js +0 -287
- package/dist/agent/state-manager.js.map +0 -1
- package/dist/agent/task-discovery.js +0 -433
- package/dist/agent/task-discovery.js.map +0 -1
- package/dist/agent/task-discovery.test.js +0 -40
- package/dist/agent/task-discovery.test.js.map +0 -1
- package/dist/agent/task-history.js +0 -128
- package/dist/agent/task-history.js.map +0 -1
- package/dist/agent/task-prioritizer.js +0 -286
- package/dist/agent/task-prioritizer.js.map +0 -1
- package/dist/agent/test-fixtures.js +0 -112
- package/dist/agent/test-fixtures.js.map +0 -1
- package/dist/agent/types.js +0 -15
- package/dist/agent/types.js.map +0 -1
- package/dist/agent/workflow-adapter.js +0 -219
- package/dist/agent/workflow-adapter.js.map +0 -1
- package/dist/agent/workflow-adapter.test.js +0 -11
- package/dist/agent/workflow-adapter.test.js.map +0 -1
- package/dist/agent/working-dir-discovery.js +0 -54
- package/dist/agent/working-dir-discovery.js.map +0 -1
- package/dist/agent/working-space.js +0 -484
- package/dist/agent/working-space.js.map +0 -1
- package/dist/agent/working-space.test.js +0 -436
- package/dist/agent/working-space.test.js.map +0 -1
- package/dist/api.js +0 -279
- package/dist/api.js.map +0 -1
- package/dist/builtin-commands.js +0 -20
- package/dist/builtin-commands.js.map +0 -1
- package/dist/commandSummarizer.js +0 -96
- package/dist/commandSummarizer.js.map +0 -1
- package/dist/commands/__tests__/command-detection.integration.test.js +0 -238
- package/dist/commands/__tests__/command-detection.integration.test.js.map +0 -1
- package/dist/commands/__tests__/script-generator.integration.test.js +0 -271
- package/dist/commands/__tests__/script-generator.integration.test.js.map +0 -1
- package/dist/commands/agent.js +0 -218
- package/dist/commands/agent.js.map +0 -1
- package/dist/commands/code.js +0 -102
- package/dist/commands/code.js.map +0 -1
- package/dist/commands/command.constants.js +0 -13
- package/dist/commands/command.constants.js.map +0 -1
- package/dist/commands/commit.js +0 -17
- package/dist/commands/commit.js.map +0 -1
- package/dist/commands/fix.js +0 -17
- package/dist/commands/fix.js.map +0 -1
- package/dist/commands/init.js +0 -501
- package/dist/commands/init.js.map +0 -1
- package/dist/commands/mcp-server.js +0 -122
- package/dist/commands/mcp-server.js.map +0 -1
- package/dist/commands/memory.js +0 -410
- package/dist/commands/memory.js.map +0 -1
- package/dist/commands/memory.test.js +0 -453
- package/dist/commands/memory.test.js.map +0 -1
- package/dist/commands/meta.js +0 -142
- package/dist/commands/meta.js.map +0 -1
- package/dist/commands/plan.js +0 -39
- package/dist/commands/plan.js.map +0 -1
- package/dist/commands/pr.js +0 -14
- package/dist/commands/pr.js.map +0 -1
- package/dist/commands/review-json.test.js +0 -33
- package/dist/commands/review-json.test.js.map +0 -1
- package/dist/commands/review.js +0 -137
- package/dist/commands/review.js.map +0 -1
- package/dist/commands/review.usage.test.js +0 -41
- package/dist/commands/review.usage.test.js.map +0 -1
- package/dist/commands/run.js +0 -107
- package/dist/commands/run.js.map +0 -1
- package/dist/commands/skills.js +0 -175
- package/dist/commands/skills.js.map +0 -1
- package/dist/commands/task.js +0 -34
- package/dist/commands/task.js.map +0 -1
- package/dist/commands/workflow.js +0 -117
- package/dist/commands/workflow.js.map +0 -1
- package/dist/configPrompt.js +0 -39
- package/dist/configPrompt.js.map +0 -1
- package/dist/env.js +0 -22
- package/dist/env.js.map +0 -1
- package/dist/errors.js +0 -132
- package/dist/errors.js.map +0 -1
- package/dist/errors.test.js +0 -21
- package/dist/errors.test.js.map +0 -1
- package/dist/file-attachments.js +0 -77
- package/dist/file-attachments.js.map +0 -1
- package/dist/file-attachments.test.js +0 -214
- package/dist/file-attachments.test.js.map +0 -1
- package/dist/getModel.js +0 -229
- package/dist/getModel.js.map +0 -1
- package/dist/getModel.test.js +0 -67
- package/dist/getModel.test.js.map +0 -1
- package/dist/getProviderOptions.js +0 -58
- package/dist/getProviderOptions.js.map +0 -1
- package/dist/git-operations.js +0 -363
- package/dist/git-operations.js.map +0 -1
- package/dist/git-operations.test.js +0 -81
- package/dist/git-operations.test.js.map +0 -1
- package/dist/index.js.map +0 -1
- package/dist/logger.js +0 -41
- package/dist/logger.js.map +0 -1
- package/dist/mcp/client.js +0 -11
- package/dist/mcp/client.js.map +0 -1
- package/dist/mcp/error-scenarios.test.js +0 -206
- package/dist/mcp/error-scenarios.test.js.map +0 -1
- package/dist/mcp/errors.js +0 -63
- package/dist/mcp/errors.js.map +0 -1
- package/dist/mcp/index.js +0 -9
- package/dist/mcp/index.js.map +0 -1
- package/dist/mcp/manager.js +0 -231
- package/dist/mcp/manager.js.map +0 -1
- package/dist/mcp/manager.test.js +0 -40
- package/dist/mcp/manager.test.js.map +0 -1
- package/dist/mcp/sdk-client.js +0 -166
- package/dist/mcp/sdk-client.js.map +0 -1
- package/dist/mcp/shared-types.js +0 -3
- package/dist/mcp/shared-types.js.map +0 -1
- package/dist/mcp/tools-integration.test.js +0 -198
- package/dist/mcp/tools-integration.test.js.map +0 -1
- package/dist/mcp/tools.js +0 -60
- package/dist/mcp/tools.js.map +0 -1
- package/dist/mcp/transport.js +0 -257
- package/dist/mcp/transport.js.map +0 -1
- package/dist/mcp/types.js +0 -3
- package/dist/mcp/types.js.map +0 -1
- package/dist/mcp-server/index.js +0 -5
- package/dist/mcp-server/index.js.map +0 -1
- package/dist/mcp-server/plan-formatting.test.js +0 -74
- package/dist/mcp-server/plan-formatting.test.js.map +0 -1
- package/dist/mcp-server/sdk-server.js +0 -102
- package/dist/mcp-server/sdk-server.js.map +0 -1
- package/dist/mcp-server/sdk-server.test.js +0 -363
- package/dist/mcp-server/sdk-server.test.js.map +0 -1
- package/dist/mcp-server/tools.js +0 -785
- package/dist/mcp-server/tools.js.map +0 -1
- package/dist/mcp-server/types.js +0 -3
- package/dist/mcp-server/types.js.map +0 -1
- package/dist/options.js +0 -85
- package/dist/options.js.map +0 -1
- package/dist/options.test.js +0 -177
- package/dist/options.test.js.map +0 -1
- package/dist/prices.js +0 -61
- package/dist/prices.js.map +0 -1
- package/dist/prices.test.js +0 -148
- package/dist/prices.test.js.map +0 -1
- package/dist/runWorkflow.js +0 -313
- package/dist/runWorkflow.js.map +0 -1
- package/dist/script/__tests__/execution.integration.test.js +0 -274
- package/dist/script/__tests__/execution.integration.test.js.map +0 -1
- package/dist/script/__tests__/runner.test.js +0 -176
- package/dist/script/__tests__/runner.test.js.map +0 -1
- package/dist/script/__tests__/validator.test.js +0 -180
- package/dist/script/__tests__/validator.test.js.map +0 -1
- package/dist/script/executor.js +0 -127
- package/dist/script/executor.js.map +0 -1
- package/dist/script/index.js +0 -4
- package/dist/script/index.js.map +0 -1
- package/dist/script/runner.js +0 -258
- package/dist/script/runner.js.map +0 -1
- package/dist/skillIntegration.js +0 -46
- package/dist/skillIntegration.js.map +0 -1
- package/dist/skillIntegration.test.js +0 -128
- package/dist/skillIntegration.test.js.map +0 -1
- package/dist/test/utils.js +0 -200
- package/dist/test/utils.js.map +0 -1
- package/dist/test/workflow-fixtures.js +0 -120
- package/dist/test/workflow-fixtures.js.map +0 -1
- package/dist/tool-implementations.js +0 -521
- package/dist/tool-implementations.js.map +0 -1
- package/dist/tool-implementations.skill-tools.test.js +0 -106
- package/dist/tool-implementations.skill-tools.test.js.map +0 -1
- package/dist/tools/getTodoItem.js +0 -33
- package/dist/tools/getTodoItem.js.map +0 -1
- package/dist/tools/gitDiff.js +0 -108
- package/dist/tools/gitDiff.js.map +0 -1
- package/dist/tools/index.js +0 -8
- package/dist/tools/index.js.map +0 -1
- package/dist/tools/listMemoryTopics.js +0 -24
- package/dist/tools/listMemoryTopics.js.map +0 -1
- package/dist/tools/listTodoItems.js +0 -35
- package/dist/tools/listTodoItems.js.map +0 -1
- package/dist/tools/listTodoItems.test.js +0 -89
- package/dist/tools/listTodoItems.test.js.map +0 -1
- package/dist/tools/readMemory.js +0 -33
- package/dist/tools/readMemory.js.map +0 -1
- package/dist/tools/updateMemory.js +0 -62
- package/dist/tools/updateMemory.js.map +0 -1
- package/dist/tools/updateMemory.test.js +0 -109
- package/dist/tools/updateMemory.test.js.map +0 -1
- package/dist/tools/updateTodoItem.js +0 -31
- package/dist/tools/updateTodoItem.js.map +0 -1
- package/dist/tools/utils/diffLineNumbers.js +0 -178
- package/dist/tools/utils/diffLineNumbers.js.map +0 -1
- package/dist/utils/cacheControl.js +0 -59
- package/dist/utils/cacheControl.js.map +0 -1
- package/dist/utils/cacheControl.test.js +0 -128
- package/dist/utils/cacheControl.test.js.map +0 -1
- package/dist/utils/command.js +0 -50
- package/dist/utils/command.js.map +0 -1
- package/dist/utils/shell.js +0 -56
- package/dist/utils/shell.js.map +0 -1
- package/dist/utils/userInput.js +0 -47
- package/dist/utils/userInput.js.map +0 -1
- package/dist/workflow-tools.js +0 -21
- package/dist/workflow-tools.js.map +0 -1
- package/dist/workflows/agent-builder.js +0 -90
- package/dist/workflows/agent-builder.js.map +0 -1
- package/dist/workflows/agent-builder.test.js +0 -115
- package/dist/workflows/agent-builder.test.js.map +0 -1
- package/dist/workflows/code.workflow.js +0 -145
- package/dist/workflows/code.workflow.js.map +0 -1
- package/dist/workflows/commit.workflow.js +0 -111
- package/dist/workflows/commit.workflow.js.map +0 -1
- package/dist/workflows/commit.workflow.test.js +0 -141
- package/dist/workflows/commit.workflow.test.js.map +0 -1
- package/dist/workflows/fix.workflow.js +0 -172
- package/dist/workflows/fix.workflow.js.map +0 -1
- package/dist/workflows/fix.workflow.test.js +0 -137
- package/dist/workflows/fix.workflow.test.js.map +0 -1
- package/dist/workflows/git-file-tools.js +0 -408
- package/dist/workflows/git-file-tools.js.map +0 -1
- package/dist/workflows/index.js +0 -12
- package/dist/workflows/index.js.map +0 -1
- package/dist/workflows/init-interactive.workflow.js +0 -198
- package/dist/workflows/init-interactive.workflow.js.map +0 -1
- package/dist/workflows/init.workflow.js +0 -41
- package/dist/workflows/init.workflow.js.map +0 -1
- package/dist/workflows/meta.workflow.js +0 -107
- package/dist/workflows/meta.workflow.js.map +0 -1
- package/dist/workflows/plan.workflow.js +0 -275
- package/dist/workflows/plan.workflow.js.map +0 -1
- package/dist/workflows/plan.workflow.test.js +0 -419
- package/dist/workflows/plan.workflow.test.js.map +0 -1
- package/dist/workflows/pr.workflow.js +0 -54
- package/dist/workflows/pr.workflow.js.map +0 -1
- package/dist/workflows/pr.workflow.test.js +0 -98
- package/dist/workflows/pr.workflow.test.js.map +0 -1
- package/dist/workflows/prompts/coder.js +0 -85
- package/dist/workflows/prompts/coder.js.map +0 -1
- package/dist/workflows/prompts/commit.js +0 -16
- package/dist/workflows/prompts/commit.js.map +0 -1
- package/dist/workflows/prompts/fix.js +0 -44
- package/dist/workflows/prompts/fix.js.map +0 -1
- package/dist/workflows/prompts/index.js +0 -10
- package/dist/workflows/prompts/index.js.map +0 -1
- package/dist/workflows/prompts/init.js +0 -48
- package/dist/workflows/prompts/init.js.map +0 -1
- package/dist/workflows/prompts/meta.js +0 -17
- package/dist/workflows/prompts/meta.js.map +0 -1
- package/dist/workflows/prompts/plan.js +0 -212
- package/dist/workflows/prompts/plan.js.map +0 -1
- package/dist/workflows/prompts/pr.js +0 -15
- package/dist/workflows/prompts/pr.js.map +0 -1
- package/dist/workflows/prompts/review.js +0 -145
- package/dist/workflows/prompts/review.js.map +0 -1
- package/dist/workflows/prompts/shared.js +0 -93
- package/dist/workflows/prompts/shared.js.map +0 -1
- package/dist/workflows/review.workflow.js +0 -357
- package/dist/workflows/review.workflow.js.map +0 -1
- package/dist/workflows/task.workflow.js +0 -47
- package/dist/workflows/task.workflow.js.map +0 -1
- package/dist/workflows/testing/helper.js +0 -41
- package/dist/workflows/testing/helper.js.map +0 -1
- package/dist/workflows/workflow.utils.js +0 -351
- package/dist/workflows/workflow.utils.js.map +0 -1
- package/dist/workflows/workflow.utils.test.js +0 -45
- package/dist/workflows/workflow.utils.test.js.map +0 -1
|
@@ -1,142 +0,0 @@
|
|
|
1
|
-
import { afterEach, beforeEach, describe, expect, it, mock } from 'bun:test';
|
|
2
|
-
import { Priority } from '../types';
|
|
3
|
-
import { ApprovalManager } from './approval';
|
|
4
|
-
// Mock Logger
|
|
5
|
-
const infoMock = mock(() => { });
|
|
6
|
-
const warnMock = mock(() => { });
|
|
7
|
-
const errorMock = mock(() => { });
|
|
8
|
-
const debugMock = mock(() => { });
|
|
9
|
-
const mockLogger = {
|
|
10
|
-
info: infoMock,
|
|
11
|
-
warn: warnMock,
|
|
12
|
-
error: errorMock,
|
|
13
|
-
debug: debugMock,
|
|
14
|
-
};
|
|
15
|
-
// Mock readline
|
|
16
|
-
let mockAnswer = 'yes';
|
|
17
|
-
const mockQuestion = mock((_query, cb) => {
|
|
18
|
-
cb(mockAnswer);
|
|
19
|
-
});
|
|
20
|
-
mock.module('node:readline', () => {
|
|
21
|
-
return {
|
|
22
|
-
createInterface: () => ({
|
|
23
|
-
question: mockQuestion,
|
|
24
|
-
close: () => { },
|
|
25
|
-
}),
|
|
26
|
-
};
|
|
27
|
-
});
|
|
28
|
-
describe('ApprovalManager', () => {
|
|
29
|
-
let manager;
|
|
30
|
-
// Helper to create tasks
|
|
31
|
-
const createTask = (overrides = {}) => ({
|
|
32
|
-
id: 'test-task',
|
|
33
|
-
type: 'feature',
|
|
34
|
-
title: 'Test Task',
|
|
35
|
-
description: 'Description',
|
|
36
|
-
priority: Priority.MEDIUM,
|
|
37
|
-
complexity: 'medium',
|
|
38
|
-
dependencies: [],
|
|
39
|
-
estimatedTime: 10,
|
|
40
|
-
status: 'pending',
|
|
41
|
-
files: [],
|
|
42
|
-
workflow: 'code',
|
|
43
|
-
workflowInput: {},
|
|
44
|
-
retryCount: 0,
|
|
45
|
-
createdAt: Date.now(),
|
|
46
|
-
...overrides,
|
|
47
|
-
});
|
|
48
|
-
beforeEach(() => {
|
|
49
|
-
// Reset mocks
|
|
50
|
-
infoMock.mockClear();
|
|
51
|
-
warnMock.mockClear();
|
|
52
|
-
mockQuestion.mockClear();
|
|
53
|
-
// Default manager setup
|
|
54
|
-
manager = new ApprovalManager(mockLogger, 'destructive', // approvalLevel
|
|
55
|
-
true, // autoApproveSafeTasks
|
|
56
|
-
30, // maxAutoApprovalCost
|
|
57
|
-
['delete', 'force-push']);
|
|
58
|
-
});
|
|
59
|
-
describe('requiresApproval', () => {
|
|
60
|
-
it('should return false when approval level is none', () => {
|
|
61
|
-
manager = new ApprovalManager(mockLogger, 'none', false, 0, []);
|
|
62
|
-
const task = createTask({ type: 'delete' }); // Even destructive
|
|
63
|
-
expect(manager.requiresApproval(task)).toBe(false);
|
|
64
|
-
});
|
|
65
|
-
it('should return true when approval level is all', () => {
|
|
66
|
-
manager = new ApprovalManager(mockLogger, 'all', true, 100, []);
|
|
67
|
-
const task = createTask({ type: 'feature' });
|
|
68
|
-
expect(manager.requiresApproval(task)).toBe(true);
|
|
69
|
-
});
|
|
70
|
-
it('should return true for destructive tasks', () => {
|
|
71
|
-
const task = createTask({ type: 'delete' });
|
|
72
|
-
expect(manager.requiresApproval(task)).toBe(true);
|
|
73
|
-
});
|
|
74
|
-
it('should return true for tasks affecting many files', () => {
|
|
75
|
-
const files = Array(11).fill('file.ts');
|
|
76
|
-
const task = createTask({ files });
|
|
77
|
-
expect(manager.requiresApproval(task)).toBe(true);
|
|
78
|
-
});
|
|
79
|
-
it('should return true for tasks with destructive keywords in description', () => {
|
|
80
|
-
const task = createTask({ description: 'This will delete everything' });
|
|
81
|
-
expect(manager.requiresApproval(task)).toBe(true);
|
|
82
|
-
});
|
|
83
|
-
it('should return true for commits when level is commits', () => {
|
|
84
|
-
manager = new ApprovalManager(mockLogger, 'commits', true, 30, []);
|
|
85
|
-
const task = createTask({ type: 'commit' });
|
|
86
|
-
expect(manager.requiresApproval(task)).toBe(true);
|
|
87
|
-
});
|
|
88
|
-
it('should auto-approve safe tasks within limits', () => {
|
|
89
|
-
const task = createTask({
|
|
90
|
-
estimatedTime: 20,
|
|
91
|
-
priority: Priority.TRIVIAL,
|
|
92
|
-
});
|
|
93
|
-
expect(manager.requiresApproval(task)).toBe(false);
|
|
94
|
-
});
|
|
95
|
-
it('should require approval if task exceeds auto-approval cost', () => {
|
|
96
|
-
const task = createTask({
|
|
97
|
-
estimatedTime: 40,
|
|
98
|
-
priority: Priority.TRIVIAL,
|
|
99
|
-
});
|
|
100
|
-
expect(manager.requiresApproval(task)).toBe(true);
|
|
101
|
-
});
|
|
102
|
-
it('should require approval if task is not trivial priority', () => {
|
|
103
|
-
const task = createTask({
|
|
104
|
-
estimatedTime: 20,
|
|
105
|
-
priority: Priority.HIGH,
|
|
106
|
-
});
|
|
107
|
-
expect(manager.requiresApproval(task)).toBe(true);
|
|
108
|
-
});
|
|
109
|
-
});
|
|
110
|
-
describe('requestApproval', () => {
|
|
111
|
-
// We need to handle process.stdin.isTTY
|
|
112
|
-
const originalIsTTY = process.stdin.isTTY;
|
|
113
|
-
beforeEach(() => {
|
|
114
|
-
Object.defineProperty(process.stdin, 'isTTY', { value: true, configurable: true });
|
|
115
|
-
});
|
|
116
|
-
afterEach(() => {
|
|
117
|
-
Object.defineProperty(process.stdin, 'isTTY', { value: originalIsTTY, configurable: true });
|
|
118
|
-
});
|
|
119
|
-
it('should return approved when user answers yes', async () => {
|
|
120
|
-
mockAnswer = 'yes';
|
|
121
|
-
const task = createTask();
|
|
122
|
-
const result = await manager.requestApproval(task);
|
|
123
|
-
expect(result.approved).toBe(true);
|
|
124
|
-
expect(mockQuestion).toHaveBeenCalled();
|
|
125
|
-
});
|
|
126
|
-
it('should return rejected when user answers no', async () => {
|
|
127
|
-
mockAnswer = 'no';
|
|
128
|
-
const task = createTask();
|
|
129
|
-
const result = await manager.requestApproval(task);
|
|
130
|
-
expect(result.approved).toBe(false);
|
|
131
|
-
expect(result.reason).toBe('Rejected by user');
|
|
132
|
-
});
|
|
133
|
-
it('should auto-reject if not TTY', async () => {
|
|
134
|
-
Object.defineProperty(process.stdin, 'isTTY', { value: false, configurable: true });
|
|
135
|
-
const task = createTask();
|
|
136
|
-
const result = await manager.requestApproval(task);
|
|
137
|
-
expect(result.approved).toBe(false);
|
|
138
|
-
expect(result.reason).toContain('Non-interactive');
|
|
139
|
-
});
|
|
140
|
-
});
|
|
141
|
-
});
|
|
142
|
-
//# sourceMappingURL=approval.test.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"approval.test.js","sourceRoot":"","sources":["../../../src/agent/safety/approval.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,UAAU,CAAA;AAE5E,OAAO,EAAE,QAAQ,EAAa,MAAM,UAAU,CAAA;AAC9C,OAAO,EAAE,eAAe,EAAE,MAAM,YAAY,CAAA;AAE5C,cAAc;AACd,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAA;AAC/B,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAA;AAC/B,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAA;AAChC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAA;AAEhC,MAAM,UAAU,GAAG;IACjB,IAAI,EAAE,QAAQ;IACd,IAAI,EAAE,QAAQ;IACd,KAAK,EAAE,SAAS;IAChB,KAAK,EAAE,SAAS;CACI,CAAA;AAEtB,gBAAgB;AAChB,IAAI,UAAU,GAAG,KAAK,CAAA;AACtB,MAAM,YAAY,GAAG,IAAI,CAAC,CAAC,MAAc,EAAE,EAA4B,EAAE,EAAE;IACzE,EAAE,CAAC,UAAU,CAAC,CAAA;AAChB,CAAC,CAAC,CAAA;AAEF,IAAI,CAAC,MAAM,CAAC,eAAe,EAAE,GAAG,EAAE;IAChC,OAAO;QACL,eAAe,EAAE,GAAG,EAAE,CAAC,CAAC;YACtB,QAAQ,EAAE,YAAY;YACtB,KAAK,EAAE,GAAG,EAAE,GAAE,CAAC;SAChB,CAAC;KACH,CAAA;AACH,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,IAAI,OAAwB,CAAA;IAE5B,yBAAyB;IACzB,MAAM,UAAU,GAAG,CAAC,YAA2B,EAAE,EAAQ,EAAE,CAAC,CAAC;QAC3D,EAAE,EAAE,WAAW;QACf,IAAI,EAAE,SAAS;QACf,KAAK,EAAE,WAAW;QAClB,WAAW,EAAE,aAAa;QAC1B,QAAQ,EAAE,QAAQ,CAAC,MAAM;QACzB,UAAU,EAAE,QAAQ;QACpB,YAAY,EAAE,EAAE;QAChB,aAAa,EAAE,EAAE;QACjB,MAAM,EAAE,SAAS;QACjB,KAAK,EAAE,EAAE;QACT,QAAQ,EAAE,MAAM;QAChB,aAAa,EAAE,EAAE;QACjB,UAAU,EAAE,CAAC;QACb,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;QACrB,GAAG,SAAS;KACb,CAAC,CAAA;IAEF,UAAU,CAAC,GAAG,EAAE;QACd,cAAc;QACd,QAAQ,CAAC,SAAS,EAAE,CAAA;QACpB,QAAQ,CAAC,SAAS,EAAE,CAAA;QACpB,YAAY,CAAC,SAAS,EAAE,CAAA;QAExB,wBAAwB;QACxB,OAAO,GAAG,IAAI,eAAe,CAC3B,UAAU,EACV,aAAa,EAAE,gBAAgB;QAC/B,IAAI,EAAE,uBAAuB;QAC7B,EAAE,EAAE,sBAAsB;QAC1B,CAAC,QAAQ,EAAE,YAAY,CAAC,CACzB,CAAA;IACH,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,OAAO,GAAG,IAAI,eAAe,CAAC,UAAU,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC,CAAA;YAC/D,MAAM,IAAI,GAAG,UAAU,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAA,CAAC,mBAAmB;YAC/D,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QACpD,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,OAAO,GAAG,IAAI,eAAe,CAAC,UAAU,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE,CAAC,CAAA;YAC/D,MAAM,IAAI,GAAG,UAAU,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAA;YAC5C,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACnD,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,IAAI,GAAG,UAAU,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAA;YAC3C,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACnD,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;YAC3D,MAAM,KAAK,GAAG,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;YACvC,MAAM,IAAI,GAAG,UAAU,CAAC,EAAE,KAAK,EAAE,CAAC,CAAA;YAClC,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACnD,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,uEAAuE,EAAE,GAAG,EAAE;YAC/E,MAAM,IAAI,GAAG,UAAU,CAAC,EAAE,WAAW,EAAE,6BAA6B,EAAE,CAAC,CAAA;YACvE,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACnD,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;YAC9D,OAAO,GAAG,IAAI,eAAe,CAAC,UAAU,EAAE,SAAS,EAAE,IAAI,EAAE,EAAE,EAAE,EAAE,CAAC,CAAA;YAClE,MAAM,IAAI,GAAG,UAAU,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAA;YAC3C,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACnD,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,MAAM,IAAI,GAAG,UAAU,CAAC;gBACtB,aAAa,EAAE,EAAE;gBACjB,QAAQ,EAAE,QAAQ,CAAC,OAAO;aAC3B,CAAC,CAAA;YACF,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QACpD,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;YACpE,MAAM,IAAI,GAAG,UAAU,CAAC;gBACtB,aAAa,EAAE,EAAE;gBACjB,QAAQ,EAAE,QAAQ,CAAC,OAAO;aAC3B,CAAC,CAAA;YACF,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACnD,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;YACjE,MAAM,IAAI,GAAG,UAAU,CAAC;gBACtB,aAAa,EAAE,EAAE;gBACjB,QAAQ,EAAE,QAAQ,CAAC,IAAI;aACxB,CAAC,CAAA;YACF,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACnD,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,wCAAwC;QACxC,MAAM,aAAa,GAAG,OAAO,CAAC,KAAK,CAAC,KAAK,CAAA;QAEzC,UAAU,CAAC,GAAG,EAAE;YACd,MAAM,CAAC,cAAc,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC,CAAA;QACpF,CAAC,CAAC,CAAA;QAEF,SAAS,CAAC,GAAG,EAAE;YACb,MAAM,CAAC,cAAc,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,aAAa,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC,CAAA;QAC7F,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,UAAU,GAAG,KAAK,CAAA;YAClB,MAAM,IAAI,GAAG,UAAU,EAAE,CAAA;YACzB,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,eAAe,CAAC,IAAI,CAAC,CAAA;YAClD,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;YAClC,MAAM,CAAC,YAAY,CAAC,CAAC,gBAAgB,EAAE,CAAA;QACzC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,UAAU,GAAG,IAAI,CAAA;YACjB,MAAM,IAAI,GAAG,UAAU,EAAE,CAAA;YACzB,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,eAAe,CAAC,IAAI,CAAC,CAAA;YAClD,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;YACnC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAA;QAChD,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;YAC7C,MAAM,CAAC,cAAc,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC,CAAA;YACnF,MAAM,IAAI,GAAG,UAAU,EAAE,CAAA;YACzB,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,eAAe,CAAC,IAAI,CAAC,CAAA;YAClD,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;YACnC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAA;QACpD,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}
|
|
@@ -1,152 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* Performs safety checks before task execution
|
|
3
|
-
*/
|
|
4
|
-
export class SafetyChecker {
|
|
5
|
-
#tools;
|
|
6
|
-
constructor(_logger, tools) {
|
|
7
|
-
this.#tools = tools;
|
|
8
|
-
}
|
|
9
|
-
/**
|
|
10
|
-
* Perform pre-execution safety checks
|
|
11
|
-
*/
|
|
12
|
-
async preExecutionCheck(task) {
|
|
13
|
-
const checks = [];
|
|
14
|
-
// Check 1: Repository has uncommitted changes
|
|
15
|
-
checks.push(await this.checkUncommittedChanges(task));
|
|
16
|
-
// Check 2: Task affects critical files
|
|
17
|
-
checks.push(await this.checkCriticalFiles(task));
|
|
18
|
-
// Check 3: Working branch is clean
|
|
19
|
-
checks.push(await this.checkWorkingBranch(task));
|
|
20
|
-
const failed = checks.filter((c) => !c.passed && c.action === 'block');
|
|
21
|
-
const warnings = checks.filter((c) => !c.passed && c.action === 'warn');
|
|
22
|
-
const _passed = checks.filter((c) => c.passed);
|
|
23
|
-
return {
|
|
24
|
-
safe: failed.length === 0,
|
|
25
|
-
checks,
|
|
26
|
-
failed,
|
|
27
|
-
warnings,
|
|
28
|
-
};
|
|
29
|
-
}
|
|
30
|
-
/**
|
|
31
|
-
* Check multiple tasks
|
|
32
|
-
*/
|
|
33
|
-
async checkTasks(tasks) {
|
|
34
|
-
const allChecks = [];
|
|
35
|
-
const allFailed = [];
|
|
36
|
-
const allWarnings = [];
|
|
37
|
-
for (const task of tasks) {
|
|
38
|
-
const result = await this.preExecutionCheck(task);
|
|
39
|
-
allChecks.push(...result.checks);
|
|
40
|
-
allFailed.push(...result.failed);
|
|
41
|
-
allWarnings.push(...result.warnings);
|
|
42
|
-
}
|
|
43
|
-
return {
|
|
44
|
-
safe: allFailed.length === 0,
|
|
45
|
-
checks: allChecks,
|
|
46
|
-
failed: allFailed,
|
|
47
|
-
warnings: allWarnings,
|
|
48
|
-
};
|
|
49
|
-
}
|
|
50
|
-
/**
|
|
51
|
-
* Check for uncommitted changes
|
|
52
|
-
*/
|
|
53
|
-
async checkUncommittedChanges(task) {
|
|
54
|
-
try {
|
|
55
|
-
const result = await this.#tools.executeCommand({
|
|
56
|
-
command: 'git status --porcelain',
|
|
57
|
-
shell: true,
|
|
58
|
-
});
|
|
59
|
-
const hasChanges = result.stdout.trim().length > 0;
|
|
60
|
-
if (hasChanges && task.type === 'commit') {
|
|
61
|
-
return {
|
|
62
|
-
name: 'uncommitted-changes',
|
|
63
|
-
passed: false,
|
|
64
|
-
message: 'Repository has uncommitted changes before commit task',
|
|
65
|
-
action: 'warn',
|
|
66
|
-
};
|
|
67
|
-
}
|
|
68
|
-
return {
|
|
69
|
-
name: 'uncommitted-changes',
|
|
70
|
-
passed: true,
|
|
71
|
-
message: hasChanges ? 'Repository has uncommitted changes' : 'Repository is clean',
|
|
72
|
-
action: 'ignore',
|
|
73
|
-
};
|
|
74
|
-
}
|
|
75
|
-
catch {
|
|
76
|
-
return {
|
|
77
|
-
name: 'uncommitted-changes',
|
|
78
|
-
passed: true,
|
|
79
|
-
message: 'Could not check git status',
|
|
80
|
-
action: 'ignore',
|
|
81
|
-
};
|
|
82
|
-
}
|
|
83
|
-
}
|
|
84
|
-
/**
|
|
85
|
-
* Check for critical files
|
|
86
|
-
*/
|
|
87
|
-
async checkCriticalFiles(task) {
|
|
88
|
-
const criticalPatterns = [
|
|
89
|
-
'package.json',
|
|
90
|
-
'tsconfig.json',
|
|
91
|
-
'.env',
|
|
92
|
-
'.gitignore',
|
|
93
|
-
'yarn.lock',
|
|
94
|
-
'package-lock.json',
|
|
95
|
-
'bun.lock',
|
|
96
|
-
'bun.lockb',
|
|
97
|
-
];
|
|
98
|
-
const affectedCritical = task.files.filter((file) => criticalPatterns.some((pattern) => file.includes(pattern)));
|
|
99
|
-
if (affectedCritical.length > 0) {
|
|
100
|
-
return {
|
|
101
|
-
name: 'critical-files',
|
|
102
|
-
passed: false,
|
|
103
|
-
message: `Task affects critical files: ${affectedCritical.join(', ')}`,
|
|
104
|
-
action: 'warn',
|
|
105
|
-
error: `Critical files: ${affectedCritical.join(', ')}`,
|
|
106
|
-
};
|
|
107
|
-
}
|
|
108
|
-
return {
|
|
109
|
-
name: 'critical-files',
|
|
110
|
-
passed: true,
|
|
111
|
-
message: 'No critical files affected',
|
|
112
|
-
action: 'ignore',
|
|
113
|
-
};
|
|
114
|
-
}
|
|
115
|
-
/**
|
|
116
|
-
* Check working branch
|
|
117
|
-
*/
|
|
118
|
-
async checkWorkingBranch(task) {
|
|
119
|
-
try {
|
|
120
|
-
const result = await this.#tools.executeCommand({
|
|
121
|
-
command: 'git branch --show-current',
|
|
122
|
-
shell: true,
|
|
123
|
-
});
|
|
124
|
-
const currentBranch = result.stdout.trim();
|
|
125
|
-
// Check if we're on main/master
|
|
126
|
-
if ((currentBranch === 'main' || currentBranch === 'master') && task.type === 'commit') {
|
|
127
|
-
return {
|
|
128
|
-
name: 'working-branch',
|
|
129
|
-
passed: false,
|
|
130
|
-
message: `Committing directly to ${currentBranch} branch`,
|
|
131
|
-
action: 'warn',
|
|
132
|
-
error: `Consider creating a feature branch instead of committing to ${currentBranch}`,
|
|
133
|
-
};
|
|
134
|
-
}
|
|
135
|
-
return {
|
|
136
|
-
name: 'working-branch',
|
|
137
|
-
passed: true,
|
|
138
|
-
message: `Working on branch: ${currentBranch}`,
|
|
139
|
-
action: 'ignore',
|
|
140
|
-
};
|
|
141
|
-
}
|
|
142
|
-
catch {
|
|
143
|
-
return {
|
|
144
|
-
name: 'working-branch',
|
|
145
|
-
passed: true,
|
|
146
|
-
message: 'Could not determine current branch',
|
|
147
|
-
action: 'ignore',
|
|
148
|
-
};
|
|
149
|
-
}
|
|
150
|
-
}
|
|
151
|
-
}
|
|
152
|
-
//# sourceMappingURL=checks.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"checks.js","sourceRoot":"","sources":["../../../src/agent/safety/checks.ts"],"names":[],"mappings":"AAGA;;GAEG;AACH,MAAM,OAAO,aAAa;IACxB,MAAM,CAAK;IAEX,YAAY,OAAe,EAAE,KAAU;QACrC,IAAI,CAAC,MAAM,GAAG,KAAK,CAAA;IACrB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,iBAAiB,CAAC,IAAU;QAChC,MAAM,MAAM,GAAU,EAAE,CAAA;QAExB,8CAA8C;QAC9C,MAAM,CAAC,IAAI,CAAC,MAAM,IAAI,CAAC,uBAAuB,CAAC,IAAI,CAAC,CAAC,CAAA;QAErD,uCAAuC;QACvC,MAAM,CAAC,IAAI,CAAC,MAAM,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC,CAAA;QAEhD,mCAAmC;QACnC,MAAM,CAAC,IAAI,CAAC,MAAM,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC,CAAA;QAEhD,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,CAAC,MAAM,KAAK,OAAO,CAAC,CAAA;QACtE,MAAM,QAAQ,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,CAAC,MAAM,KAAK,MAAM,CAAC,CAAA;QACvE,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAA;QAE9C,OAAO;YACL,IAAI,EAAE,MAAM,CAAC,MAAM,KAAK,CAAC;YACzB,MAAM;YACN,MAAM;YACN,QAAQ;SACT,CAAA;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU,CAAC,KAAa;QAC5B,MAAM,SAAS,GAAU,EAAE,CAAA;QAC3B,MAAM,SAAS,GAAU,EAAE,CAAA;QAC3B,MAAM,WAAW,GAAU,EAAE,CAAA;QAE7B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAA;YACjD,SAAS,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,CAAA;YAChC,SAAS,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,CAAA;YAChC,WAAW,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAA;QACtC,CAAC;QAED,OAAO;YACL,IAAI,EAAE,SAAS,CAAC,MAAM,KAAK,CAAC;YAC5B,MAAM,EAAE,SAAS;YACjB,MAAM,EAAE,SAAS;YACjB,QAAQ,EAAE,WAAW;SACtB,CAAA;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,uBAAuB,CAAC,IAAU;QAC9C,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC;gBAC9C,OAAO,EAAE,wBAAwB;gBACjC,KAAK,EAAE,IAAI;aACZ,CAAC,CAAA;YAEF,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,CAAA;YAElD,IAAI,UAAU,IAAI,IAAI,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACzC,OAAO;oBACL,IAAI,EAAE,qBAAqB;oBAC3B,MAAM,EAAE,KAAK;oBACb,OAAO,EAAE,uDAAuD;oBAChE,MAAM,EAAE,MAAM;iBACf,CAAA;YACH,CAAC;YAED,OAAO;gBACL,IAAI,EAAE,qBAAqB;gBAC3B,MAAM,EAAE,IAAI;gBACZ,OAAO,EAAE,UAAU,CAAC,CAAC,CAAC,oCAAoC,CAAC,CAAC,CAAC,qBAAqB;gBAClF,MAAM,EAAE,QAAQ;aACjB,CAAA;QACH,CAAC;QAAC,MAAM,CAAC;YACP,OAAO;gBACL,IAAI,EAAE,qBAAqB;gBAC3B,MAAM,EAAE,IAAI;gBACZ,OAAO,EAAE,4BAA4B;gBACrC,MAAM,EAAE,QAAQ;aACjB,CAAA;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,kBAAkB,CAAC,IAAU;QACzC,MAAM,gBAAgB,GAAG;YACvB,cAAc;YACd,eAAe;YACf,MAAM;YACN,YAAY;YACZ,WAAW;YACX,mBAAmB;YACnB,UAAU;YACV,WAAW;SACZ,CAAA;QAED,MAAM,gBAAgB,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAA;QAEhH,IAAI,gBAAgB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAChC,OAAO;gBACL,IAAI,EAAE,gBAAgB;gBACtB,MAAM,EAAE,KAAK;gBACb,OAAO,EAAE,gCAAgC,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;gBACtE,MAAM,EAAE,MAAM;gBACd,KAAK,EAAE,mBAAmB,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;aACxD,CAAA;QACH,CAAC;QAED,OAAO;YACL,IAAI,EAAE,gBAAgB;YACtB,MAAM,EAAE,IAAI;YACZ,OAAO,EAAE,4BAA4B;YACrC,MAAM,EAAE,QAAQ;SACjB,CAAA;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,kBAAkB,CAAC,IAAU;QACzC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC;gBAC9C,OAAO,EAAE,2BAA2B;gBACpC,KAAK,EAAE,IAAI;aACZ,CAAC,CAAA;YAEF,MAAM,aAAa,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAA;YAE1C,gCAAgC;YAChC,IAAI,CAAC,aAAa,KAAK,MAAM,IAAI,aAAa,KAAK,QAAQ,CAAC,IAAI,IAAI,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACvF,OAAO;oBACL,IAAI,EAAE,gBAAgB;oBACtB,MAAM,EAAE,KAAK;oBACb,OAAO,EAAE,0BAA0B,aAAa,SAAS;oBACzD,MAAM,EAAE,MAAM;oBACd,KAAK,EAAE,+DAA+D,aAAa,EAAE;iBACtF,CAAA;YACH,CAAC;YAED,OAAO;gBACL,IAAI,EAAE,gBAAgB;gBACtB,MAAM,EAAE,IAAI;gBACZ,OAAO,EAAE,sBAAsB,aAAa,EAAE;gBAC9C,MAAM,EAAE,QAAQ;aACjB,CAAA;QACH,CAAC;QAAC,MAAM,CAAC;YACP,OAAO;gBACL,IAAI,EAAE,gBAAgB;gBACtB,MAAM,EAAE,IAAI;gBACZ,OAAO,EAAE,oCAAoC;gBAC7C,MAAM,EAAE,QAAQ;aACjB,CAAA;QACH,CAAC;IACH,CAAC;CACF"}
|
|
@@ -1,89 +0,0 @@
|
|
|
1
|
-
import { beforeEach, describe, expect, it, mock } from 'bun:test';
|
|
2
|
-
import { Priority } from '../types';
|
|
3
|
-
import { SafetyChecker } from './checks';
|
|
4
|
-
// Mock Logger
|
|
5
|
-
const mockLogger = {
|
|
6
|
-
info: mock(() => { }),
|
|
7
|
-
warn: mock(() => { }),
|
|
8
|
-
error: mock(() => { }),
|
|
9
|
-
};
|
|
10
|
-
// Mock Tools
|
|
11
|
-
const mockExecuteCommand = mock(async (_args) => ({ stdout: '', stderr: '', exitCode: 0 }));
|
|
12
|
-
const mockTools = {
|
|
13
|
-
executeCommand: mockExecuteCommand,
|
|
14
|
-
};
|
|
15
|
-
describe('SafetyChecker', () => {
|
|
16
|
-
let checker;
|
|
17
|
-
const createTask = (overrides = {}) => ({
|
|
18
|
-
id: 'test-task',
|
|
19
|
-
type: 'feature',
|
|
20
|
-
title: 'Test Task',
|
|
21
|
-
description: 'Description',
|
|
22
|
-
priority: Priority.MEDIUM,
|
|
23
|
-
complexity: 'medium',
|
|
24
|
-
dependencies: [],
|
|
25
|
-
estimatedTime: 10,
|
|
26
|
-
status: 'pending',
|
|
27
|
-
files: [],
|
|
28
|
-
workflow: 'code',
|
|
29
|
-
workflowInput: {},
|
|
30
|
-
retryCount: 0,
|
|
31
|
-
createdAt: Date.now(),
|
|
32
|
-
...overrides,
|
|
33
|
-
});
|
|
34
|
-
beforeEach(() => {
|
|
35
|
-
mockExecuteCommand.mockClear();
|
|
36
|
-
checker = new SafetyChecker(mockLogger, mockTools);
|
|
37
|
-
});
|
|
38
|
-
describe('checkTasks', () => {
|
|
39
|
-
it('should pass when all checks pass', async () => {
|
|
40
|
-
mockExecuteCommand.mockResolvedValue({ stdout: '', stderr: '', exitCode: 0 }); // Clean git status
|
|
41
|
-
const task = createTask();
|
|
42
|
-
const result = await checker.checkTasks([task]);
|
|
43
|
-
expect(result.safe).toBe(true);
|
|
44
|
-
expect(result.failed).toHaveLength(0);
|
|
45
|
-
});
|
|
46
|
-
it('should warn on uncommitted changes for commit tasks', async () => {
|
|
47
|
-
mockExecuteCommand.mockImplementation(async ({ command }) => {
|
|
48
|
-
if (command.includes('git status')) {
|
|
49
|
-
return { stdout: 'M file.ts', stderr: '', exitCode: 0 };
|
|
50
|
-
}
|
|
51
|
-
return { stdout: 'feature-branch', stderr: '', exitCode: 0 };
|
|
52
|
-
});
|
|
53
|
-
const task = createTask({ type: 'commit' });
|
|
54
|
-
const result = await checker.checkTasks([task]);
|
|
55
|
-
expect(result.safe).toBe(true); // It's a warning, not a block
|
|
56
|
-
expect(result.warnings).toHaveLength(1);
|
|
57
|
-
expect(result.warnings[0].name).toBe('uncommitted-changes');
|
|
58
|
-
});
|
|
59
|
-
it('should warn when affecting critical files', async () => {
|
|
60
|
-
mockExecuteCommand.mockResolvedValue({ stdout: '', stderr: '', exitCode: 0 });
|
|
61
|
-
const task = createTask({ files: ['package.json'] });
|
|
62
|
-
const result = await checker.checkTasks([task]);
|
|
63
|
-
expect(result.safe).toBe(true); // Warning
|
|
64
|
-
expect(result.warnings).toHaveLength(1);
|
|
65
|
-
expect(result.warnings[0].name).toBe('critical-files');
|
|
66
|
-
});
|
|
67
|
-
it('should warn when committing to main branch', async () => {
|
|
68
|
-
mockExecuteCommand.mockImplementation(async ({ command }) => {
|
|
69
|
-
if (command.includes('git branch')) {
|
|
70
|
-
return { stdout: 'main', stderr: '', exitCode: 0 };
|
|
71
|
-
}
|
|
72
|
-
return { stdout: '', stderr: '', exitCode: 0 };
|
|
73
|
-
});
|
|
74
|
-
const task = createTask({ type: 'commit' });
|
|
75
|
-
const result = await checker.checkTasks([task]);
|
|
76
|
-
expect(result.safe).toBe(true); // Warning
|
|
77
|
-
expect(result.warnings).toHaveLength(1);
|
|
78
|
-
expect(result.warnings[0].name).toBe('working-branch');
|
|
79
|
-
});
|
|
80
|
-
it('should handle multiple tasks', async () => {
|
|
81
|
-
mockExecuteCommand.mockResolvedValue({ stdout: '', stderr: '', exitCode: 0 });
|
|
82
|
-
const tasks = [createTask(), createTask()];
|
|
83
|
-
const result = await checker.checkTasks(tasks);
|
|
84
|
-
expect(result.safe).toBe(true);
|
|
85
|
-
expect(result.checks.length).toBeGreaterThan(0);
|
|
86
|
-
});
|
|
87
|
-
});
|
|
88
|
-
});
|
|
89
|
-
//# sourceMappingURL=checks.test.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"checks.test.js","sourceRoot":"","sources":["../../../src/agent/safety/checks.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,UAAU,CAAA;AAEjE,OAAO,EAAE,QAAQ,EAAa,MAAM,UAAU,CAAA;AAC9C,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAA;AAExC,cAAc;AACd,MAAM,UAAU,GAAG;IACjB,IAAI,EAAE,IAAI,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC;IACpB,IAAI,EAAE,IAAI,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC;IACpB,KAAK,EAAE,IAAI,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC;CACD,CAAA;AAEtB,aAAa;AACb,MAAM,kBAAkB,GAAG,IAAI,CAAC,KAAK,EAAE,KAA0B,EAAE,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC,CAAA;AAChH,MAAM,SAAS,GAAG;IAChB,cAAc,EAAE,kBAAkB;CACnC,CAAA;AAED,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,IAAI,OAAsB,CAAA;IAE1B,MAAM,UAAU,GAAG,CAAC,YAA2B,EAAE,EAAQ,EAAE,CAAC,CAAC;QAC3D,EAAE,EAAE,WAAW;QACf,IAAI,EAAE,SAAS;QACf,KAAK,EAAE,WAAW;QAClB,WAAW,EAAE,aAAa;QAC1B,QAAQ,EAAE,QAAQ,CAAC,MAAM;QACzB,UAAU,EAAE,QAAQ;QACpB,YAAY,EAAE,EAAE;QAChB,aAAa,EAAE,EAAE;QACjB,MAAM,EAAE,SAAS;QACjB,KAAK,EAAE,EAAE;QACT,QAAQ,EAAE,MAAM;QAChB,aAAa,EAAE,EAAE;QACjB,UAAU,EAAE,CAAC;QACb,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;QACrB,GAAG,SAAS;KACb,CAAC,CAAA;IAEF,UAAU,CAAC,GAAG,EAAE;QACd,kBAAkB,CAAC,SAAS,EAAE,CAAA;QAC9B,OAAO,GAAG,IAAI,aAAa,CAAC,UAAU,EAAE,SAAS,CAAC,CAAA;IACpD,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;YAChD,kBAAkB,CAAC,iBAAiB,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAA,CAAC,mBAAmB;YAEjG,MAAM,IAAI,GAAG,UAAU,EAAE,CAAA;YACzB,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,CAAA;YAE/C,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;YAC9B,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;QACvC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,kBAAkB,CAAC,kBAAkB,CAAC,KAAK,EAAE,EAAE,OAAO,EAAuB,EAAE,EAAE;gBAC/E,IAAI,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;oBACnC,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAA;gBACzD,CAAC;gBACD,OAAO,EAAE,MAAM,EAAE,gBAAgB,EAAE,MAAM,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAA;YAC9D,CAAC,CAAC,CAAA;YAEF,MAAM,IAAI,GAAG,UAAU,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAA;YAC3C,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,CAAA;YAE/C,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA,CAAC,8BAA8B;YAC7D,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YACvC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAA;QAC7D,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,kBAAkB,CAAC,iBAAiB,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAA;YAE7E,MAAM,IAAI,GAAG,UAAU,CAAC,EAAE,KAAK,EAAE,CAAC,cAAc,CAAC,EAAE,CAAC,CAAA;YACpD,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,CAAA;YAE/C,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA,CAAC,UAAU;YACzC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YACvC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAA;QACxD,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,kBAAkB,CAAC,kBAAkB,CAAC,KAAK,EAAE,EAAE,OAAO,EAAuB,EAAE,EAAE;gBAC/E,IAAI,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;oBACnC,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAA;gBACpD,CAAC;gBACD,OAAO,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAA;YAChD,CAAC,CAAC,CAAA;YAEF,MAAM,IAAI,GAAG,UAAU,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAA;YAC3C,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,CAAA;YAE/C,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA,CAAC,UAAU;YACzC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YACvC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAA;QACxD,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;YAC5C,kBAAkB,CAAC,iBAAiB,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAA;YAC7E,MAAM,KAAK,GAAG,CAAC,UAAU,EAAE,EAAE,UAAU,EAAE,CAAC,CAAA;YAC1C,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,CAAA;YAC9C,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;YAC9B,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAA;QACjD,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}
|
|
@@ -1,92 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* Handles graceful shutdown and interruption
|
|
3
|
-
*/
|
|
4
|
-
export class InterruptHandler {
|
|
5
|
-
#interrupted = false;
|
|
6
|
-
#interruptReason = '';
|
|
7
|
-
#logger;
|
|
8
|
-
#agent;
|
|
9
|
-
constructor(logger, agent) {
|
|
10
|
-
this.#logger = logger;
|
|
11
|
-
this.#agent = agent;
|
|
12
|
-
this.setupInterruptHandlers();
|
|
13
|
-
}
|
|
14
|
-
/**
|
|
15
|
-
* Set up interrupt handlers
|
|
16
|
-
*/
|
|
17
|
-
setupInterruptHandlers() {
|
|
18
|
-
// Handle SIGINT (Ctrl+C)
|
|
19
|
-
process.on('SIGINT', () => {
|
|
20
|
-
this.handleInterrupt('SIGINT');
|
|
21
|
-
});
|
|
22
|
-
// Handle SIGTERM
|
|
23
|
-
process.on('SIGTERM', () => {
|
|
24
|
-
this.handleInterrupt('SIGTERM');
|
|
25
|
-
});
|
|
26
|
-
// Handle uncaught exceptions
|
|
27
|
-
process.on('uncaughtException', (error) => {
|
|
28
|
-
this.#logger.error('UncaughtException', error);
|
|
29
|
-
this.#interrupted = true;
|
|
30
|
-
this.#interruptReason = `Uncaught exception: ${error.message}`;
|
|
31
|
-
});
|
|
32
|
-
// Handle unhandled promise rejections
|
|
33
|
-
process.on('unhandledRejection', (reason) => {
|
|
34
|
-
this.#logger.error('UnhandledRejection', reason);
|
|
35
|
-
this.#interrupted = true;
|
|
36
|
-
this.#interruptReason = `Unhandled rejection: ${reason}`;
|
|
37
|
-
});
|
|
38
|
-
}
|
|
39
|
-
/**
|
|
40
|
-
* Handle interrupt signal
|
|
41
|
-
*/
|
|
42
|
-
async handleInterrupt(signal) {
|
|
43
|
-
if (this.#interrupted) {
|
|
44
|
-
this.#logger.warn(`[Interrupt] Already interrupted, forcing exit`);
|
|
45
|
-
process.exit(1);
|
|
46
|
-
}
|
|
47
|
-
this.#logger.info(`\n[Interrupt] Received ${signal}, shutting down gracefully...`);
|
|
48
|
-
this.#interrupted = true;
|
|
49
|
-
this.#interruptReason = `Received ${signal}`;
|
|
50
|
-
try {
|
|
51
|
-
if (this.#agent && typeof this.#agent.stop === 'function') {
|
|
52
|
-
await this.#agent.stop();
|
|
53
|
-
}
|
|
54
|
-
if (this.#agent && typeof this.#agent.cleanup === 'function') {
|
|
55
|
-
await this.#agent.cleanup();
|
|
56
|
-
}
|
|
57
|
-
this.#logger.info('[Interrupt] Shutdown complete');
|
|
58
|
-
process.exit(0);
|
|
59
|
-
}
|
|
60
|
-
catch (error) {
|
|
61
|
-
this.#logger.error('[Interrupt] Error during shutdown', error);
|
|
62
|
-
process.exit(1);
|
|
63
|
-
}
|
|
64
|
-
}
|
|
65
|
-
/**
|
|
66
|
-
* Check if should stop
|
|
67
|
-
*/
|
|
68
|
-
shouldStop() {
|
|
69
|
-
return this.#interrupted;
|
|
70
|
-
}
|
|
71
|
-
/**
|
|
72
|
-
* Get interrupt reason
|
|
73
|
-
*/
|
|
74
|
-
getReason() {
|
|
75
|
-
return this.#interruptReason;
|
|
76
|
-
}
|
|
77
|
-
/**
|
|
78
|
-
* Manually trigger interrupt
|
|
79
|
-
*/
|
|
80
|
-
interrupt(reason = 'Manual interrupt') {
|
|
81
|
-
this.#interrupted = true;
|
|
82
|
-
this.#interruptReason = reason;
|
|
83
|
-
}
|
|
84
|
-
/**
|
|
85
|
-
* Reset interrupt state
|
|
86
|
-
*/
|
|
87
|
-
reset() {
|
|
88
|
-
this.#interrupted = false;
|
|
89
|
-
this.#interruptReason = '';
|
|
90
|
-
}
|
|
91
|
-
}
|
|
92
|
-
//# sourceMappingURL=interrupt.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"interrupt.js","sourceRoot":"","sources":["../../../src/agent/safety/interrupt.ts"],"names":[],"mappings":"AAEA;;GAEG;AACH,MAAM,OAAO,gBAAgB;IAC3B,YAAY,GAAY,KAAK,CAAA;IAC7B,gBAAgB,GAAW,EAAE,CAAA;IAC7B,OAAO,CAAQ;IACf,MAAM,CAAK;IAEX,YACE,MAAc,EACd,KAAU;QAEV,IAAI,CAAC,OAAO,GAAG,MAAM,CAAA;QACrB,IAAI,CAAC,MAAM,GAAG,KAAK,CAAA;QACnB,IAAI,CAAC,sBAAsB,EAAE,CAAA;IAC/B,CAAC;IAED;;OAEG;IACK,sBAAsB;QAC5B,yBAAyB;QACzB,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,GAAG,EAAE;YACxB,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAA;QAChC,CAAC,CAAC,CAAA;QAEF,iBAAiB;QACjB,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE;YACzB,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,CAAA;QACjC,CAAC,CAAC,CAAA;QAEF,6BAA6B;QAC7B,OAAO,CAAC,EAAE,CAAC,mBAAmB,EAAE,CAAC,KAAK,EAAE,EAAE;YACxC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,mBAAmB,EAAE,KAAK,CAAC,CAAA;YAC9C,IAAI,CAAC,YAAY,GAAG,IAAI,CAAA;YACxB,IAAI,CAAC,gBAAgB,GAAG,uBAAuB,KAAK,CAAC,OAAO,EAAE,CAAA;QAChE,CAAC,CAAC,CAAA;QAEF,sCAAsC;QACtC,OAAO,CAAC,EAAE,CAAC,oBAAoB,EAAE,CAAC,MAAM,EAAE,EAAE;YAC1C,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,oBAAoB,EAAE,MAAe,CAAC,CAAA;YACzD,IAAI,CAAC,YAAY,GAAG,IAAI,CAAA;YACxB,IAAI,CAAC,gBAAgB,GAAG,wBAAwB,MAAM,EAAE,CAAA;QAC1D,CAAC,CAAC,CAAA;IACJ,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,eAAe,CAAC,MAAc;QAC1C,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtB,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,+CAA+C,CAAC,CAAA;YAClE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QACjB,CAAC;QAED,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,0BAA0B,MAAM,+BAA+B,CAAC,CAAA;QAClF,IAAI,CAAC,YAAY,GAAG,IAAI,CAAA;QACxB,IAAI,CAAC,gBAAgB,GAAG,YAAY,MAAM,EAAE,CAAA;QAE5C,IAAI,CAAC;YACH,IAAI,IAAI,CAAC,MAAM,IAAI,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gBAC1D,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAA;YAC1B,CAAC;YAED,IAAI,IAAI,CAAC,MAAM,IAAI,OAAO,IAAI,CAAC,MAAM,CAAC,OAAO,KAAK,UAAU,EAAE,CAAC;gBAC7D,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAA;YAC7B,CAAC;YAED,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,+BAA+B,CAAC,CAAA;YAClD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QACjB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,mCAAmC,EAAE,KAAc,CAAC,CAAA;YACvE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QACjB,CAAC;IACH,CAAC;IAED;;OAEG;IACH,UAAU;QACR,OAAO,IAAI,CAAC,YAAY,CAAA;IAC1B,CAAC;IAED;;OAEG;IACH,SAAS;QACP,OAAO,IAAI,CAAC,gBAAgB,CAAA;IAC9B,CAAC;IAED;;OAEG;IACH,SAAS,CAAC,SAAiB,kBAAkB;QAC3C,IAAI,CAAC,YAAY,GAAG,IAAI,CAAA;QACxB,IAAI,CAAC,gBAAgB,GAAG,MAAM,CAAA;IAChC,CAAC;IAED;;OAEG;IACH,KAAK;QACH,IAAI,CAAC,YAAY,GAAG,KAAK,CAAA;QACzB,IAAI,CAAC,gBAAgB,GAAG,EAAE,CAAA;IAC5B,CAAC;CACF"}
|
|
@@ -1,53 +0,0 @@
|
|
|
1
|
-
import { beforeEach, describe, expect, it, mock } from 'bun:test';
|
|
2
|
-
import { InterruptHandler } from './interrupt';
|
|
3
|
-
// Mock Logger
|
|
4
|
-
const mockLogger = {
|
|
5
|
-
info: mock(() => { }),
|
|
6
|
-
warn: mock(() => { }),
|
|
7
|
-
error: mock(() => { }),
|
|
8
|
-
};
|
|
9
|
-
// Mock Agent
|
|
10
|
-
const mockAgent = {
|
|
11
|
-
stop: mock(async () => { }),
|
|
12
|
-
cleanup: mock(async () => { }),
|
|
13
|
-
};
|
|
14
|
-
describe('InterruptHandler', () => {
|
|
15
|
-
let handler;
|
|
16
|
-
beforeEach(() => {
|
|
17
|
-
mockAgent.stop.mockClear();
|
|
18
|
-
mockAgent.cleanup.mockClear();
|
|
19
|
-
handler = new InterruptHandler(mockLogger, mockAgent);
|
|
20
|
-
});
|
|
21
|
-
it('should start not interrupted', () => {
|
|
22
|
-
expect(handler.shouldStop()).toBe(false);
|
|
23
|
-
expect(handler.getReason()).toBe('');
|
|
24
|
-
});
|
|
25
|
-
it('should set interrupted state when interrupt is called', () => {
|
|
26
|
-
handler.interrupt('Manual stop');
|
|
27
|
-
expect(handler.shouldStop()).toBe(true);
|
|
28
|
-
expect(handler.getReason()).toBe('Manual stop');
|
|
29
|
-
});
|
|
30
|
-
it('should reset state', () => {
|
|
31
|
-
handler.interrupt('Stop');
|
|
32
|
-
handler.reset();
|
|
33
|
-
expect(handler.shouldStop()).toBe(false);
|
|
34
|
-
expect(handler.getReason()).toBe('');
|
|
35
|
-
});
|
|
36
|
-
it('should call agent stop and cleanup on interrupt', async () => {
|
|
37
|
-
// Mock process.exit to prevent actual exit
|
|
38
|
-
const originalExit = process.exit;
|
|
39
|
-
const mockExit = mock(() => { });
|
|
40
|
-
process.exit = mockExit;
|
|
41
|
-
try {
|
|
42
|
-
// We access the private method via any cast to simulate the event handler
|
|
43
|
-
await handler.handleInterrupt('SIGINT');
|
|
44
|
-
expect(mockAgent.stop).toHaveBeenCalled();
|
|
45
|
-
expect(mockAgent.cleanup).toHaveBeenCalled();
|
|
46
|
-
expect(mockExit).toHaveBeenCalledWith(0);
|
|
47
|
-
}
|
|
48
|
-
finally {
|
|
49
|
-
process.exit = originalExit;
|
|
50
|
-
}
|
|
51
|
-
});
|
|
52
|
-
});
|
|
53
|
-
//# sourceMappingURL=interrupt.test.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"interrupt.test.js","sourceRoot":"","sources":["../../../src/agent/safety/interrupt.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,UAAU,CAAA;AAEjE,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAA;AAE9C,cAAc;AACd,MAAM,UAAU,GAAG;IACjB,IAAI,EAAE,IAAI,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC;IACpB,IAAI,EAAE,IAAI,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC;IACpB,KAAK,EAAE,IAAI,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC;CACD,CAAA;AAEtB,aAAa;AACb,MAAM,SAAS,GAAG;IAChB,IAAI,EAAE,IAAI,CAAC,KAAK,IAAI,EAAE,GAAE,CAAC,CAAC;IAC1B,OAAO,EAAE,IAAI,CAAC,KAAK,IAAI,EAAE,GAAE,CAAC,CAAC;CAC9B,CAAA;AAED,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,IAAI,OAAyB,CAAA;IAE7B,UAAU,CAAC,GAAG,EAAE;QACd,SAAS,CAAC,IAAI,CAAC,SAAS,EAAE,CAAA;QAC1B,SAAS,CAAC,OAAO,CAAC,SAAS,EAAE,CAAA;QAC7B,OAAO,GAAG,IAAI,gBAAgB,CAAC,UAAU,EAAE,SAAS,CAAC,CAAA;IACvD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;QACtC,MAAM,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QACxC,MAAM,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;IACtC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;QAC/D,OAAO,CAAC,SAAS,CAAC,aAAa,CAAC,CAAA;QAChC,MAAM,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACvC,MAAM,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAA;IACjD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAC5B,OAAO,CAAC,SAAS,CAAC,MAAM,CAAC,CAAA;QACzB,OAAO,CAAC,KAAK,EAAE,CAAA;QACf,MAAM,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QACxC,MAAM,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;IACtC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAC/D,2CAA2C;QAC3C,MAAM,YAAY,GAAG,OAAO,CAAC,IAAI,CAAA;QACjC,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAE,CAAC,CAAmC,CAAA;QACjE,OAAO,CAAC,IAAI,GAAG,QAAQ,CAAA;QAEvB,IAAI,CAAC;YACH,0EAA0E;YAC1E,MAAO,OAAe,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAA;YAEhD,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,gBAAgB,EAAE,CAAA;YACzC,MAAM,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,gBAAgB,EAAE,CAAA;YAC5C,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC,CAAA;QAC1C,CAAC;gBAAS,CAAC;YACT,OAAO,CAAC,IAAI,GAAG,YAAY,CAAA;QAC7B,CAAC;IACH,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}
|