@researchcomputer/agents-sdk 0.1.0
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/README.md +60 -0
- package/dist/agents/messages.d.ts +29 -0
- package/dist/agents/messages.d.ts.map +1 -0
- package/dist/agents/messages.js +56 -0
- package/dist/agents/messages.js.map +1 -0
- package/dist/agents/messages.test.d.ts +2 -0
- package/dist/agents/messages.test.d.ts.map +1 -0
- package/dist/agents/messages.test.js +80 -0
- package/dist/agents/messages.test.js.map +1 -0
- package/dist/agents/subagent.d.ts +16 -0
- package/dist/agents/subagent.d.ts.map +1 -0
- package/dist/agents/subagent.js +46 -0
- package/dist/agents/subagent.js.map +1 -0
- package/dist/agents/swarm.d.ts +22 -0
- package/dist/agents/swarm.d.ts.map +1 -0
- package/dist/agents/swarm.js +140 -0
- package/dist/agents/swarm.js.map +1 -0
- package/dist/agents/tools.d.ts +6 -0
- package/dist/agents/tools.d.ts.map +1 -0
- package/dist/agents/tools.js +71 -0
- package/dist/agents/tools.js.map +1 -0
- package/dist/agents/tools.test.d.ts +2 -0
- package/dist/agents/tools.test.d.ts.map +1 -0
- package/dist/agents/tools.test.js +70 -0
- package/dist/agents/tools.test.js.map +1 -0
- package/dist/bridge/config.d.ts +21 -0
- package/dist/bridge/config.d.ts.map +1 -0
- package/dist/bridge/config.js +35 -0
- package/dist/bridge/config.js.map +1 -0
- package/dist/bridge/config.test.d.ts +2 -0
- package/dist/bridge/config.test.d.ts.map +1 -0
- package/dist/bridge/config.test.js +79 -0
- package/dist/bridge/config.test.js.map +1 -0
- package/dist/bridge/server.d.ts +3 -0
- package/dist/bridge/server.d.ts.map +1 -0
- package/dist/bridge/server.js +316 -0
- package/dist/bridge/server.js.map +1 -0
- package/dist/context/compression.d.ts +27 -0
- package/dist/context/compression.d.ts.map +1 -0
- package/dist/context/compression.js +117 -0
- package/dist/context/compression.js.map +1 -0
- package/dist/context/compression.test.d.ts +2 -0
- package/dist/context/compression.test.d.ts.map +1 -0
- package/dist/context/compression.test.js +170 -0
- package/dist/context/compression.test.js.map +1 -0
- package/dist/context/converter.d.ts +11 -0
- package/dist/context/converter.d.ts.map +1 -0
- package/dist/context/converter.js +40 -0
- package/dist/context/converter.js.map +1 -0
- package/dist/context/converter.test.d.ts +2 -0
- package/dist/context/converter.test.d.ts.map +1 -0
- package/dist/context/converter.test.js +110 -0
- package/dist/context/converter.test.js.map +1 -0
- package/dist/context/run-context.d.ts +9 -0
- package/dist/context/run-context.d.ts.map +1 -0
- package/dist/context/run-context.js +14 -0
- package/dist/context/run-context.js.map +1 -0
- package/dist/context/run-context.test.d.ts +2 -0
- package/dist/context/run-context.test.d.ts.map +1 -0
- package/dist/context/run-context.test.js +34 -0
- package/dist/context/run-context.test.js.map +1 -0
- package/dist/context/system-prompt.d.ts +24 -0
- package/dist/context/system-prompt.d.ts.map +1 -0
- package/dist/context/system-prompt.js +69 -0
- package/dist/context/system-prompt.js.map +1 -0
- package/dist/context/system-prompt.test.d.ts +2 -0
- package/dist/context/system-prompt.test.d.ts.map +1 -0
- package/dist/context/system-prompt.test.js +134 -0
- package/dist/context/system-prompt.test.js.map +1 -0
- package/dist/errors.d.ts +24 -0
- package/dist/errors.d.ts.map +1 -0
- package/dist/errors.js +47 -0
- package/dist/errors.js.map +1 -0
- package/dist/errors.test.d.ts +2 -0
- package/dist/errors.test.d.ts.map +1 -0
- package/dist/errors.test.js +68 -0
- package/dist/errors.test.js.map +1 -0
- package/dist/factory.d.ts +58 -0
- package/dist/factory.d.ts.map +1 -0
- package/dist/factory.js +282 -0
- package/dist/factory.js.map +1 -0
- package/dist/factory.test.d.ts +2 -0
- package/dist/factory.test.d.ts.map +1 -0
- package/dist/factory.test.js +63 -0
- package/dist/factory.test.js.map +1 -0
- package/dist/fork.test.d.ts +2 -0
- package/dist/fork.test.d.ts.map +1 -0
- package/dist/fork.test.js +456 -0
- package/dist/fork.test.js.map +1 -0
- package/dist/index.d.ts +29 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +32 -0
- package/dist/index.js.map +1 -0
- package/dist/mcp/client.d.ts +3 -0
- package/dist/mcp/client.d.ts.map +1 -0
- package/dist/mcp/client.js +75 -0
- package/dist/mcp/client.js.map +1 -0
- package/dist/mcp/schema-convert.d.ts +3 -0
- package/dist/mcp/schema-convert.d.ts.map +1 -0
- package/dist/mcp/schema-convert.js +73 -0
- package/dist/mcp/schema-convert.js.map +1 -0
- package/dist/mcp/schema-convert.test.d.ts +2 -0
- package/dist/mcp/schema-convert.test.d.ts.map +1 -0
- package/dist/mcp/schema-convert.test.js +95 -0
- package/dist/mcp/schema-convert.test.js.map +1 -0
- package/dist/mcp/tools.d.ts +14 -0
- package/dist/mcp/tools.d.ts.map +1 -0
- package/dist/mcp/tools.js +23 -0
- package/dist/mcp/tools.js.map +1 -0
- package/dist/mcp/tools.test.d.ts +2 -0
- package/dist/mcp/tools.test.d.ts.map +1 -0
- package/dist/mcp/tools.test.js +45 -0
- package/dist/mcp/tools.test.js.map +1 -0
- package/dist/memory/memory.d.ts +3 -0
- package/dist/memory/memory.d.ts.map +1 -0
- package/dist/memory/memory.js +130 -0
- package/dist/memory/memory.js.map +1 -0
- package/dist/memory/memory.test.d.ts +2 -0
- package/dist/memory/memory.test.d.ts.map +1 -0
- package/dist/memory/memory.test.js +94 -0
- package/dist/memory/memory.test.js.map +1 -0
- package/dist/middleware/hooks.d.ts +18 -0
- package/dist/middleware/hooks.d.ts.map +1 -0
- package/dist/middleware/hooks.js +71 -0
- package/dist/middleware/hooks.js.map +1 -0
- package/dist/middleware/hooks.test.d.ts +2 -0
- package/dist/middleware/hooks.test.d.ts.map +1 -0
- package/dist/middleware/hooks.test.js +172 -0
- package/dist/middleware/hooks.test.js.map +1 -0
- package/dist/middleware/index.d.ts +7 -0
- package/dist/middleware/index.d.ts.map +1 -0
- package/dist/middleware/index.js +5 -0
- package/dist/middleware/index.js.map +1 -0
- package/dist/middleware/permission-middleware.d.ts +11 -0
- package/dist/middleware/permission-middleware.d.ts.map +1 -0
- package/dist/middleware/permission-middleware.js +56 -0
- package/dist/middleware/permission-middleware.js.map +1 -0
- package/dist/middleware/permission-middleware.test.d.ts +2 -0
- package/dist/middleware/permission-middleware.test.d.ts.map +1 -0
- package/dist/middleware/permission-middleware.test.js +170 -0
- package/dist/middleware/permission-middleware.test.js.map +1 -0
- package/dist/middleware/permissions.d.ts +14 -0
- package/dist/middleware/permissions.d.ts.map +1 -0
- package/dist/middleware/permissions.js +118 -0
- package/dist/middleware/permissions.js.map +1 -0
- package/dist/middleware/permissions.test.d.ts +2 -0
- package/dist/middleware/permissions.test.d.ts.map +1 -0
- package/dist/middleware/permissions.test.js +187 -0
- package/dist/middleware/permissions.test.js.map +1 -0
- package/dist/middleware/pipeline.d.ts +13 -0
- package/dist/middleware/pipeline.d.ts.map +1 -0
- package/dist/middleware/pipeline.js +28 -0
- package/dist/middleware/pipeline.js.map +1 -0
- package/dist/middleware/pipeline.test.d.ts +2 -0
- package/dist/middleware/pipeline.test.d.ts.map +1 -0
- package/dist/middleware/pipeline.test.js +189 -0
- package/dist/middleware/pipeline.test.js.map +1 -0
- package/dist/observability/cost-tracker.d.ts +3 -0
- package/dist/observability/cost-tracker.d.ts.map +1 -0
- package/dist/observability/cost-tracker.js +26 -0
- package/dist/observability/cost-tracker.js.map +1 -0
- package/dist/observability/cost-tracker.test.d.ts +2 -0
- package/dist/observability/cost-tracker.test.d.ts.map +1 -0
- package/dist/observability/cost-tracker.test.js +49 -0
- package/dist/observability/cost-tracker.test.js.map +1 -0
- package/dist/observability/trace.d.ts +2 -0
- package/dist/observability/trace.d.ts.map +1 -0
- package/dist/observability/trace.js +5 -0
- package/dist/observability/trace.js.map +1 -0
- package/dist/session/session.d.ts +3 -0
- package/dist/session/session.d.ts.map +1 -0
- package/dist/session/session.js +62 -0
- package/dist/session/session.js.map +1 -0
- package/dist/session/session.test.d.ts +2 -0
- package/dist/session/session.test.d.ts.map +1 -0
- package/dist/session/session.test.js +84 -0
- package/dist/session/session.test.js.map +1 -0
- package/dist/skills.d.ts +14 -0
- package/dist/skills.d.ts.map +1 -0
- package/dist/skills.js +31 -0
- package/dist/skills.js.map +1 -0
- package/dist/skills.test.d.ts +2 -0
- package/dist/skills.test.d.ts.map +1 -0
- package/dist/skills.test.js +61 -0
- package/dist/skills.test.js.map +1 -0
- package/dist/tools/ask-user.d.ts +10 -0
- package/dist/tools/ask-user.d.ts.map +1 -0
- package/dist/tools/ask-user.js +27 -0
- package/dist/tools/ask-user.js.map +1 -0
- package/dist/tools/bash.d.ts +11 -0
- package/dist/tools/bash.d.ts.map +1 -0
- package/dist/tools/bash.js +54 -0
- package/dist/tools/bash.js.map +1 -0
- package/dist/tools/bash.test.d.ts +2 -0
- package/dist/tools/bash.test.d.ts.map +1 -0
- package/dist/tools/bash.test.js +63 -0
- package/dist/tools/bash.test.js.map +1 -0
- package/dist/tools/edit.d.ts +10 -0
- package/dist/tools/edit.d.ts.map +1 -0
- package/dist/tools/edit.js +65 -0
- package/dist/tools/edit.js.map +1 -0
- package/dist/tools/edit.test.d.ts +2 -0
- package/dist/tools/edit.test.d.ts.map +1 -0
- package/dist/tools/edit.test.js +60 -0
- package/dist/tools/edit.test.js.map +1 -0
- package/dist/tools/glob.d.ts +8 -0
- package/dist/tools/glob.d.ts.map +1 -0
- package/dist/tools/glob.js +43 -0
- package/dist/tools/glob.js.map +1 -0
- package/dist/tools/glob.test.d.ts +2 -0
- package/dist/tools/glob.test.d.ts.map +1 -0
- package/dist/tools/glob.test.js +54 -0
- package/dist/tools/glob.test.js.map +1 -0
- package/dist/tools/grep.d.ts +9 -0
- package/dist/tools/grep.d.ts.map +1 -0
- package/dist/tools/grep.js +74 -0
- package/dist/tools/grep.js.map +1 -0
- package/dist/tools/grep.test.d.ts +2 -0
- package/dist/tools/grep.test.d.ts.map +1 -0
- package/dist/tools/grep.test.js +48 -0
- package/dist/tools/grep.test.js.map +1 -0
- package/dist/tools/index.d.ts +25 -0
- package/dist/tools/index.d.ts.map +1 -0
- package/dist/tools/index.js +40 -0
- package/dist/tools/index.js.map +1 -0
- package/dist/tools/index.test.d.ts +2 -0
- package/dist/tools/index.test.d.ts.map +1 -0
- package/dist/tools/index.test.js +102 -0
- package/dist/tools/index.test.js.map +1 -0
- package/dist/tools/notebook-edit.d.ts +9 -0
- package/dist/tools/notebook-edit.d.ts.map +1 -0
- package/dist/tools/notebook-edit.js +55 -0
- package/dist/tools/notebook-edit.js.map +1 -0
- package/dist/tools/read.d.ts +9 -0
- package/dist/tools/read.d.ts.map +1 -0
- package/dist/tools/read.js +48 -0
- package/dist/tools/read.js.map +1 -0
- package/dist/tools/read.test.d.ts +2 -0
- package/dist/tools/read.test.d.ts.map +1 -0
- package/dist/tools/read.test.js +68 -0
- package/dist/tools/read.test.js.map +1 -0
- package/dist/tools/util.d.ts +19 -0
- package/dist/tools/util.d.ts.map +1 -0
- package/dist/tools/util.js +44 -0
- package/dist/tools/util.js.map +1 -0
- package/dist/tools/util.test.d.ts +2 -0
- package/dist/tools/util.test.d.ts.map +1 -0
- package/dist/tools/util.test.js +75 -0
- package/dist/tools/util.test.js.map +1 -0
- package/dist/tools/web-fetch.d.ts +7 -0
- package/dist/tools/web-fetch.d.ts.map +1 -0
- package/dist/tools/web-fetch.js +35 -0
- package/dist/tools/web-fetch.js.map +1 -0
- package/dist/tools/web-search.d.ts +7 -0
- package/dist/tools/web-search.d.ts.map +1 -0
- package/dist/tools/web-search.js +20 -0
- package/dist/tools/web-search.js.map +1 -0
- package/dist/tools/write.d.ts +8 -0
- package/dist/tools/write.d.ts.map +1 -0
- package/dist/tools/write.js +33 -0
- package/dist/tools/write.js.map +1 -0
- package/dist/tools/write.test.d.ts +2 -0
- package/dist/tools/write.test.d.ts.map +1 -0
- package/dist/tools/write.test.js +49 -0
- package/dist/tools/write.test.js.map +1 -0
- package/dist/types.d.ts +257 -0
- package/dist/types.d.ts.map +1 -0
- package/dist/types.js +2 -0
- package/dist/types.js.map +1 -0
- package/dist/types.test.d.ts +2 -0
- package/dist/types.test.d.ts.map +1 -0
- package/dist/types.test.js +69 -0
- package/dist/types.test.js.map +1 -0
- package/package.json +51 -0
|
@@ -0,0 +1,71 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Check if a hook matches the given event and tool name.
|
|
3
|
+
*/
|
|
4
|
+
function hookMatches(hook, event, toolName) {
|
|
5
|
+
if (hook.event !== event)
|
|
6
|
+
return false;
|
|
7
|
+
if (hook.matcher && toolName && hook.matcher !== toolName)
|
|
8
|
+
return false;
|
|
9
|
+
return true;
|
|
10
|
+
}
|
|
11
|
+
/**
|
|
12
|
+
* Run all matching PreToolUse hooks, chaining updated args through each hook.
|
|
13
|
+
* Returns the final (possibly modified) args.
|
|
14
|
+
*/
|
|
15
|
+
export async function runPreToolUseHooks(hooks, toolName, args, runContext) {
|
|
16
|
+
let currentArgs = args;
|
|
17
|
+
for (const hook of hooks) {
|
|
18
|
+
if (!hookMatches(hook, 'PreToolUse', toolName))
|
|
19
|
+
continue;
|
|
20
|
+
const context = {
|
|
21
|
+
event: 'PreToolUse',
|
|
22
|
+
runContext,
|
|
23
|
+
toolName,
|
|
24
|
+
toolArgs: currentArgs,
|
|
25
|
+
};
|
|
26
|
+
const result = await hook.handler(context);
|
|
27
|
+
if (result?.updatedArgs !== undefined) {
|
|
28
|
+
currentArgs = result.updatedArgs;
|
|
29
|
+
}
|
|
30
|
+
}
|
|
31
|
+
return currentArgs;
|
|
32
|
+
}
|
|
33
|
+
/**
|
|
34
|
+
* Run all matching PostToolUse hooks, chaining updated results through each hook.
|
|
35
|
+
* Returns the final (possibly modified) tool result.
|
|
36
|
+
*/
|
|
37
|
+
export async function runPostToolUseHooks(hooks, toolName, toolResult, runContext) {
|
|
38
|
+
let currentResult = toolResult;
|
|
39
|
+
for (const hook of hooks) {
|
|
40
|
+
if (!hookMatches(hook, 'PostToolUse', toolName))
|
|
41
|
+
continue;
|
|
42
|
+
const context = {
|
|
43
|
+
event: 'PostToolUse',
|
|
44
|
+
runContext,
|
|
45
|
+
toolName,
|
|
46
|
+
toolResult: currentResult,
|
|
47
|
+
};
|
|
48
|
+
const result = await hook.handler(context);
|
|
49
|
+
if (result?.updatedResult !== undefined) {
|
|
50
|
+
currentResult = result.updatedResult;
|
|
51
|
+
}
|
|
52
|
+
}
|
|
53
|
+
return currentResult;
|
|
54
|
+
}
|
|
55
|
+
/**
|
|
56
|
+
* Run all matching lifecycle hooks (SessionStart, SessionEnd, Stop, etc.).
|
|
57
|
+
* Lifecycle hooks ignore the matcher field.
|
|
58
|
+
*/
|
|
59
|
+
export async function runLifecycleHooks(hooks, event, runContext, extra) {
|
|
60
|
+
for (const hook of hooks) {
|
|
61
|
+
if (hook.event !== event)
|
|
62
|
+
continue;
|
|
63
|
+
const context = {
|
|
64
|
+
event,
|
|
65
|
+
runContext,
|
|
66
|
+
...extra,
|
|
67
|
+
};
|
|
68
|
+
await hook.handler(context);
|
|
69
|
+
}
|
|
70
|
+
}
|
|
71
|
+
//# sourceMappingURL=hooks.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"hooks.js","sourceRoot":"","sources":["../../src/middleware/hooks.ts"],"names":[],"mappings":"AAGA;;GAEG;AACH,SAAS,WAAW,CAAC,IAAiB,EAAE,KAAgB,EAAE,QAAiB;IACzE,IAAI,IAAI,CAAC,KAAK,KAAK,KAAK;QAAE,OAAO,KAAK,CAAC;IACvC,IAAI,IAAI,CAAC,OAAO,IAAI,QAAQ,IAAI,IAAI,CAAC,OAAO,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC;IACxE,OAAO,IAAI,CAAC;AACd,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACtC,KAAoB,EACpB,QAAgB,EAChB,IAAa,EACb,UAAsB;IAEtB,IAAI,WAAW,GAAG,IAAI,CAAC;IAEvB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,YAAY,EAAE,QAAQ,CAAC;YAAE,SAAS;QAEzD,MAAM,OAAO,GAAgB;YAC3B,KAAK,EAAE,YAAY;YACnB,UAAU;YACV,QAAQ;YACR,QAAQ,EAAE,WAAW;SACtB,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAC3C,IAAI,MAAM,EAAE,WAAW,KAAK,SAAS,EAAE,CAAC;YACtC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;QACnC,CAAC;IACH,CAAC;IAED,OAAO,WAAW,CAAC;AACrB,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,mBAAmB,CACvC,KAAoB,EACpB,QAAgB,EAChB,UAAgC,EAChC,UAAsB;IAEtB,IAAI,aAAa,GAAG,UAAU,CAAC;IAE/B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,aAAa,EAAE,QAAQ,CAAC;YAAE,SAAS;QAE1D,MAAM,OAAO,GAAgB;YAC3B,KAAK,EAAE,aAAa;YACpB,UAAU;YACV,QAAQ;YACR,UAAU,EAAE,aAAa;SAC1B,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAC3C,IAAI,MAAM,EAAE,aAAa,KAAK,SAAS,EAAE,CAAC;YACxC,aAAa,GAAG,MAAM,CAAC,aAAa,CAAC;QACvC,CAAC;IACH,CAAC;IAED,OAAO,aAAa,CAAC;AACvB,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACrC,KAAoB,EACpB,KAAgB,EAChB,UAAsB,EACtB,KAA4D;IAE5D,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,IAAI,IAAI,CAAC,KAAK,KAAK,KAAK;YAAE,SAAS;QAEnC,MAAM,OAAO,GAAgB;YAC3B,KAAK;YACL,UAAU;YACV,GAAG,KAAK;SACT,CAAC;QAEF,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IAC9B,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"hooks.test.d.ts","sourceRoot":"","sources":["../../src/middleware/hooks.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,172 @@
|
|
|
1
|
+
import { describe, it, expect, vi } from 'vitest';
|
|
2
|
+
import { runPreToolUseHooks, runPostToolUseHooks, runLifecycleHooks } from './hooks.js';
|
|
3
|
+
import { createRunContext } from '../context/run-context.js';
|
|
4
|
+
function makeRunContext() {
|
|
5
|
+
return createRunContext({ cwd: '/tmp' });
|
|
6
|
+
}
|
|
7
|
+
describe('runPreToolUseHooks', () => {
|
|
8
|
+
it('runs matching PreToolUse hooks and returns updated args', async () => {
|
|
9
|
+
const hooks = [
|
|
10
|
+
{
|
|
11
|
+
event: 'PreToolUse',
|
|
12
|
+
handler: async (ctx) => ({ updatedArgs: { ...ctx.toolArgs, extra: true } }),
|
|
13
|
+
},
|
|
14
|
+
];
|
|
15
|
+
const result = await runPreToolUseHooks(hooks, 'ReadFile', { path: '/tmp' }, makeRunContext());
|
|
16
|
+
expect(result).toEqual({ path: '/tmp', extra: true });
|
|
17
|
+
});
|
|
18
|
+
it('chains args through multiple hooks', async () => {
|
|
19
|
+
const hooks = [
|
|
20
|
+
{
|
|
21
|
+
event: 'PreToolUse',
|
|
22
|
+
handler: async (ctx) => ({ updatedArgs: { ...ctx.toolArgs, first: true } }),
|
|
23
|
+
},
|
|
24
|
+
{
|
|
25
|
+
event: 'PreToolUse',
|
|
26
|
+
handler: async (ctx) => ({ updatedArgs: { ...ctx.toolArgs, second: true } }),
|
|
27
|
+
},
|
|
28
|
+
];
|
|
29
|
+
const result = await runPreToolUseHooks(hooks, 'ReadFile', { path: '/tmp' }, makeRunContext());
|
|
30
|
+
expect(result).toEqual({ path: '/tmp', first: true, second: true });
|
|
31
|
+
});
|
|
32
|
+
it('returns original args if no hooks match', async () => {
|
|
33
|
+
const hooks = [
|
|
34
|
+
{ event: 'PostToolUse', handler: async () => ({}) },
|
|
35
|
+
];
|
|
36
|
+
const args = { path: '/tmp' };
|
|
37
|
+
const result = await runPreToolUseHooks(hooks, 'ReadFile', args, makeRunContext());
|
|
38
|
+
expect(result).toBe(args);
|
|
39
|
+
});
|
|
40
|
+
it('skips hooks whose matcher does not match toolName', async () => {
|
|
41
|
+
const handler = vi.fn().mockResolvedValue({ updatedArgs: { changed: true } });
|
|
42
|
+
const hooks = [
|
|
43
|
+
{ event: 'PreToolUse', matcher: 'WriteFile', handler },
|
|
44
|
+
];
|
|
45
|
+
const result = await runPreToolUseHooks(hooks, 'ReadFile', { path: '/tmp' }, makeRunContext());
|
|
46
|
+
expect(handler).not.toHaveBeenCalled();
|
|
47
|
+
expect(result).toEqual({ path: '/tmp' });
|
|
48
|
+
});
|
|
49
|
+
it('runs hooks when matcher matches toolName', async () => {
|
|
50
|
+
const handler = vi.fn().mockResolvedValue({ updatedArgs: { changed: true } });
|
|
51
|
+
const hooks = [
|
|
52
|
+
{ event: 'PreToolUse', matcher: 'ReadFile', handler },
|
|
53
|
+
];
|
|
54
|
+
const result = await runPreToolUseHooks(hooks, 'ReadFile', { path: '/tmp' }, makeRunContext());
|
|
55
|
+
expect(handler).toHaveBeenCalled();
|
|
56
|
+
expect(result).toEqual({ changed: true });
|
|
57
|
+
});
|
|
58
|
+
it('preserves args if hook returns void', async () => {
|
|
59
|
+
const hooks = [
|
|
60
|
+
{ event: 'PreToolUse', handler: async () => undefined },
|
|
61
|
+
];
|
|
62
|
+
const args = { path: '/tmp' };
|
|
63
|
+
const result = await runPreToolUseHooks(hooks, 'ReadFile', args, makeRunContext());
|
|
64
|
+
expect(result).toBe(args);
|
|
65
|
+
});
|
|
66
|
+
it('preserves args if hook returns empty object', async () => {
|
|
67
|
+
const hooks = [
|
|
68
|
+
{ event: 'PreToolUse', handler: async () => ({}) },
|
|
69
|
+
];
|
|
70
|
+
const args = { path: '/tmp' };
|
|
71
|
+
const result = await runPreToolUseHooks(hooks, 'ReadFile', args, makeRunContext());
|
|
72
|
+
expect(result).toBe(args);
|
|
73
|
+
});
|
|
74
|
+
});
|
|
75
|
+
describe('runPostToolUseHooks', () => {
|
|
76
|
+
it('runs matching PostToolUse hooks and returns updated result', async () => {
|
|
77
|
+
const hooks = [
|
|
78
|
+
{
|
|
79
|
+
event: 'PostToolUse',
|
|
80
|
+
handler: async () => ({
|
|
81
|
+
updatedResult: { content: [{ type: 'text', text: 'modified' }], details: {} },
|
|
82
|
+
}),
|
|
83
|
+
},
|
|
84
|
+
];
|
|
85
|
+
const original = { content: [{ type: 'text', text: 'original' }], details: {} };
|
|
86
|
+
const result = await runPostToolUseHooks(hooks, 'ReadFile', original, makeRunContext());
|
|
87
|
+
expect(result.content[0]).toEqual({ type: 'text', text: 'modified' });
|
|
88
|
+
});
|
|
89
|
+
it('chains results through multiple hooks', async () => {
|
|
90
|
+
const hooks = [
|
|
91
|
+
{
|
|
92
|
+
event: 'PostToolUse',
|
|
93
|
+
handler: async (ctx) => ({
|
|
94
|
+
updatedResult: { content: [{ type: 'text', text: 'step1' }], details: ctx.toolResult?.details },
|
|
95
|
+
}),
|
|
96
|
+
},
|
|
97
|
+
{
|
|
98
|
+
event: 'PostToolUse',
|
|
99
|
+
handler: async (ctx) => ({
|
|
100
|
+
updatedResult: { content: [{ type: 'text', text: 'step2' }], details: ctx.toolResult?.details },
|
|
101
|
+
}),
|
|
102
|
+
},
|
|
103
|
+
];
|
|
104
|
+
const original = { content: [{ type: 'text', text: 'original' }], details: {} };
|
|
105
|
+
const result = await runPostToolUseHooks(hooks, 'ReadFile', original, makeRunContext());
|
|
106
|
+
expect(result.content[0]).toEqual({ type: 'text', text: 'step2' });
|
|
107
|
+
});
|
|
108
|
+
it('returns original result if no hooks match', async () => {
|
|
109
|
+
const hooks = [
|
|
110
|
+
{ event: 'PreToolUse', handler: async () => ({}) },
|
|
111
|
+
];
|
|
112
|
+
const original = { content: [{ type: 'text', text: 'original' }], details: {} };
|
|
113
|
+
const result = await runPostToolUseHooks(hooks, 'ReadFile', original, makeRunContext());
|
|
114
|
+
expect(result).toBe(original);
|
|
115
|
+
});
|
|
116
|
+
it('skips hooks with non-matching matcher', async () => {
|
|
117
|
+
const handler = vi.fn().mockResolvedValue({
|
|
118
|
+
updatedResult: { content: [{ type: 'text', text: 'changed' }], details: {} },
|
|
119
|
+
});
|
|
120
|
+
const hooks = [
|
|
121
|
+
{ event: 'PostToolUse', matcher: 'WriteFile', handler },
|
|
122
|
+
];
|
|
123
|
+
const original = { content: [{ type: 'text', text: 'original' }], details: {} };
|
|
124
|
+
const result = await runPostToolUseHooks(hooks, 'ReadFile', original, makeRunContext());
|
|
125
|
+
expect(handler).not.toHaveBeenCalled();
|
|
126
|
+
expect(result).toBe(original);
|
|
127
|
+
});
|
|
128
|
+
});
|
|
129
|
+
describe('runLifecycleHooks', () => {
|
|
130
|
+
it('runs all matching lifecycle hooks', async () => {
|
|
131
|
+
const handler1 = vi.fn().mockResolvedValue(undefined);
|
|
132
|
+
const handler2 = vi.fn().mockResolvedValue(undefined);
|
|
133
|
+
const hooks = [
|
|
134
|
+
{ event: 'SessionStart', handler: handler1 },
|
|
135
|
+
{ event: 'SessionStart', handler: handler2 },
|
|
136
|
+
{ event: 'SessionEnd', handler: vi.fn() },
|
|
137
|
+
];
|
|
138
|
+
await runLifecycleHooks(hooks, 'SessionStart', makeRunContext());
|
|
139
|
+
expect(handler1).toHaveBeenCalled();
|
|
140
|
+
expect(handler2).toHaveBeenCalled();
|
|
141
|
+
});
|
|
142
|
+
it('passes extra context to hook', async () => {
|
|
143
|
+
const handler = vi.fn().mockResolvedValue(undefined);
|
|
144
|
+
const hooks = [
|
|
145
|
+
{ event: 'SessionStart', handler },
|
|
146
|
+
];
|
|
147
|
+
const extra = { agentName: 'test' };
|
|
148
|
+
await runLifecycleHooks(hooks, 'SessionStart', makeRunContext(), extra);
|
|
149
|
+
expect(handler).toHaveBeenCalledWith(expect.objectContaining({
|
|
150
|
+
event: 'SessionStart',
|
|
151
|
+
agentName: 'test',
|
|
152
|
+
}));
|
|
153
|
+
});
|
|
154
|
+
it('does nothing when no hooks match', async () => {
|
|
155
|
+
const handler = vi.fn();
|
|
156
|
+
const hooks = [
|
|
157
|
+
{ event: 'SessionEnd', handler },
|
|
158
|
+
];
|
|
159
|
+
await runLifecycleHooks(hooks, 'SessionStart', makeRunContext());
|
|
160
|
+
expect(handler).not.toHaveBeenCalled();
|
|
161
|
+
});
|
|
162
|
+
it('ignores matcher for lifecycle hooks', async () => {
|
|
163
|
+
const handler = vi.fn().mockResolvedValue(undefined);
|
|
164
|
+
const hooks = [
|
|
165
|
+
{ event: 'SessionStart', matcher: 'SomeTool', handler },
|
|
166
|
+
];
|
|
167
|
+
// Lifecycle hooks should still run even with a matcher set (matcher is for tool hooks)
|
|
168
|
+
await runLifecycleHooks(hooks, 'SessionStart', makeRunContext());
|
|
169
|
+
expect(handler).toHaveBeenCalled();
|
|
170
|
+
});
|
|
171
|
+
});
|
|
172
|
+
//# sourceMappingURL=hooks.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"hooks.test.js","sourceRoot":"","sources":["../../src/middleware/hooks.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAClD,OAAO,EAAE,kBAAkB,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAExF,OAAO,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAC;AAE7D,SAAS,cAAc;IACrB,OAAO,gBAAgB,CAAC,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC,CAAC;AAC3C,CAAC;AAED,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACvE,MAAM,KAAK,GAAkB;YAC3B;gBACE,KAAK,EAAE,YAAY;gBACnB,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC,EAAE,WAAW,EAAE,EAAE,GAAI,GAAG,CAAC,QAAmB,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC;aACxF;SACF,CAAC;QACF,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,KAAK,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,cAAc,EAAE,CAAC,CAAC;QAC/F,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;QAClD,MAAM,KAAK,GAAkB;YAC3B;gBACE,KAAK,EAAE,YAAY;gBACnB,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC,EAAE,WAAW,EAAE,EAAE,GAAI,GAAG,CAAC,QAAmB,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC;aACxF;YACD;gBACE,KAAK,EAAE,YAAY;gBACnB,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC,EAAE,WAAW,EAAE,EAAE,GAAI,GAAG,CAAC,QAAmB,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,CAAC;aACzF;SACF,CAAC;QACF,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,KAAK,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,cAAc,EAAE,CAAC,CAAC;QAC/F,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;IACtE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,MAAM,KAAK,GAAkB;YAC3B,EAAE,KAAK,EAAE,aAAa,EAAE,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE;SACpD,CAAC;QACF,MAAM,IAAI,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;QAC9B,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,KAAK,EAAE,UAAU,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC,CAAC;QACnF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC5B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;QACjE,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,WAAW,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC;QAC9E,MAAM,KAAK,GAAkB;YAC3B,EAAE,KAAK,EAAE,YAAY,EAAE,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE;SACvD,CAAC;QACF,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,KAAK,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,cAAc,EAAE,CAAC,CAAC;QAC/F,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACvC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACxD,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,WAAW,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC;QAC9E,MAAM,KAAK,GAAkB;YAC3B,EAAE,KAAK,EAAE,YAAY,EAAE,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE;SACtD,CAAC;QACF,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,KAAK,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,cAAc,EAAE,CAAC,CAAC;QAC/F,MAAM,CAAC,OAAO,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;QACnD,MAAM,KAAK,GAAkB;YAC3B,EAAE,KAAK,EAAE,YAAY,EAAE,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC,SAAS,EAAE;SACxD,CAAC;QACF,MAAM,IAAI,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;QAC9B,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,KAAK,EAAE,UAAU,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC,CAAC;QACnF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC5B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC3D,MAAM,KAAK,GAAkB;YAC3B,EAAE,KAAK,EAAE,YAAY,EAAE,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE;SACnD,CAAC;QACF,MAAM,IAAI,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;QAC9B,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,KAAK,EAAE,UAAU,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC,CAAC;QACnF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC5B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;QAC1E,MAAM,KAAK,GAAkB;YAC3B;gBACE,KAAK,EAAE,aAAa;gBACpB,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;oBACpB,aAAa,EAAE,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE;iBACvF,CAAC;aACH;SACF,CAAC;QACF,MAAM,QAAQ,GAAG,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;QACzF,MAAM,MAAM,GAAG,MAAM,mBAAmB,CAAC,KAAK,EAAE,UAAU,EAAE,QAAQ,EAAE,cAAc,EAAE,CAAC,CAAC;QACxF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACrD,MAAM,KAAK,GAAkB;YAC3B;gBACE,KAAK,EAAE,aAAa;gBACpB,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;oBACvB,aAAa,EAAE,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,GAAG,CAAC,UAAU,EAAE,OAAO,EAAE;iBACzG,CAAC;aACH;YACD;gBACE,KAAK,EAAE,aAAa;gBACpB,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;oBACvB,aAAa,EAAE,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,GAAG,CAAC,UAAU,EAAE,OAAO,EAAE;iBACzG,CAAC;aACH;SACF,CAAC;QACF,MAAM,QAAQ,GAAG,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;QACzF,MAAM,MAAM,GAAG,MAAM,mBAAmB,CAAC,KAAK,EAAE,UAAU,EAAE,QAAQ,EAAE,cAAc,EAAE,CAAC,CAAC;QACxF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;QACzD,MAAM,KAAK,GAAkB;YAC3B,EAAE,KAAK,EAAE,YAAY,EAAE,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE;SACnD,CAAC;QACF,MAAM,QAAQ,GAAG,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;QACzF,MAAM,MAAM,GAAG,MAAM,mBAAmB,CAAC,KAAK,EAAE,UAAU,EAAE,QAAQ,EAAE,cAAc,EAAE,CAAC,CAAC;QACxF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACrD,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;YACxC,aAAa,EAAE,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE;SACtF,CAAC,CAAC;QACH,MAAM,KAAK,GAAkB;YAC3B,EAAE,KAAK,EAAE,aAAa,EAAE,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE;SACxD,CAAC;QACF,MAAM,QAAQ,GAAG,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;QACzF,MAAM,MAAM,GAAG,MAAM,mBAAmB,CAAC,KAAK,EAAE,UAAU,EAAE,QAAQ,EAAE,cAAc,EAAE,CAAC,CAAC;QACxF,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACvC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;QACjD,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QACtD,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QACtD,MAAM,KAAK,GAAkB;YAC3B,EAAE,KAAK,EAAE,cAAc,EAAE,OAAO,EAAE,QAAQ,EAAE;YAC5C,EAAE,KAAK,EAAE,cAAc,EAAE,OAAO,EAAE,QAAQ,EAAE;YAC5C,EAAE,KAAK,EAAE,YAAY,EAAE,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE;SAC1C,CAAC;QACF,MAAM,iBAAiB,CAAC,KAAK,EAAE,cAAc,EAAE,cAAc,EAAE,CAAC,CAAC;QACjE,MAAM,CAAC,QAAQ,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACpC,MAAM,CAAC,QAAQ,CAAC,CAAC,gBAAgB,EAAE,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;QAC5C,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QACrD,MAAM,KAAK,GAAkB;YAC3B,EAAE,KAAK,EAAE,cAAc,EAAE,OAAO,EAAE;SACnC,CAAC;QACF,MAAM,KAAK,GAAG,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC;QACpC,MAAM,iBAAiB,CAAC,KAAK,EAAE,cAAc,EAAE,cAAc,EAAE,EAAE,KAAK,CAAC,CAAC;QACxE,MAAM,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,gBAAgB,CAAC;YAC3D,KAAK,EAAE,cAAc;YACrB,SAAS,EAAE,MAAM;SAClB,CAAC,CAAC,CAAC;IACN,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;QAChD,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACxB,MAAM,KAAK,GAAkB;YAC3B,EAAE,KAAK,EAAE,YAAY,EAAE,OAAO,EAAE;SACjC,CAAC;QACF,MAAM,iBAAiB,CAAC,KAAK,EAAE,cAAc,EAAE,cAAc,EAAE,CAAC,CAAC;QACjE,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;QACnD,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QACrD,MAAM,KAAK,GAAkB;YAC3B,EAAE,KAAK,EAAE,cAAc,EAAE,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE;SACxD,CAAC;QACF,uFAAuF;QACvF,MAAM,iBAAiB,CAAC,KAAK,EAAE,cAAc,EAAE,cAAc,EAAE,CAAC,CAAC;QACjE,MAAM,CAAC,OAAO,CAAC,CAAC,gBAAgB,EAAE,CAAC;IACrC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,7 @@
|
|
|
1
|
+
export { matchRule, findMatchingRule, evaluatePermission } from './permissions.js';
|
|
2
|
+
export { createPermissionMiddleware } from './permission-middleware.js';
|
|
3
|
+
export type { PermissionMiddlewareConfig } from './permission-middleware.js';
|
|
4
|
+
export { runPreToolUseHooks, runPostToolUseHooks, runLifecycleHooks } from './hooks.js';
|
|
5
|
+
export { composePipeline } from './pipeline.js';
|
|
6
|
+
export type { PipelineConfig, Pipeline } from './pipeline.js';
|
|
7
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/middleware/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,gBAAgB,EAAE,kBAAkB,EAAE,MAAM,kBAAkB,CAAC;AACnF,OAAO,EAAE,0BAA0B,EAAE,MAAM,4BAA4B,CAAC;AACxE,YAAY,EAAE,0BAA0B,EAAE,MAAM,4BAA4B,CAAC;AAC7E,OAAO,EAAE,kBAAkB,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AACxF,OAAO,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAChD,YAAY,EAAE,cAAc,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC"}
|
|
@@ -0,0 +1,5 @@
|
|
|
1
|
+
export { matchRule, findMatchingRule, evaluatePermission } from './permissions.js';
|
|
2
|
+
export { createPermissionMiddleware } from './permission-middleware.js';
|
|
3
|
+
export { runPreToolUseHooks, runPostToolUseHooks, runLifecycleHooks } from './hooks.js';
|
|
4
|
+
export { composePipeline } from './pipeline.js';
|
|
5
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/middleware/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,gBAAgB,EAAE,kBAAkB,EAAE,MAAM,kBAAkB,CAAC;AACnF,OAAO,EAAE,0BAA0B,EAAE,MAAM,4BAA4B,CAAC;AAExE,OAAO,EAAE,kBAAkB,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AACxF,OAAO,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC"}
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
import type { BeforeToolCallContext, BeforeToolCallResult } from '@mariozechner/pi-agent-core';
|
|
2
|
+
import type { SdkTool, PermissionMode, PermissionRule, RunContext } from '../types.js';
|
|
3
|
+
export interface PermissionMiddlewareConfig {
|
|
4
|
+
mode: PermissionMode;
|
|
5
|
+
rules: PermissionRule[];
|
|
6
|
+
tools: SdkTool<any, any>[];
|
|
7
|
+
runContext: RunContext;
|
|
8
|
+
onAsk?: (toolName: string, args: unknown) => Promise<boolean>;
|
|
9
|
+
}
|
|
10
|
+
export declare function createPermissionMiddleware(config: PermissionMiddlewareConfig): (context: BeforeToolCallContext, signal?: AbortSignal) => Promise<BeforeToolCallResult | undefined>;
|
|
11
|
+
//# sourceMappingURL=permission-middleware.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"permission-middleware.d.ts","sourceRoot":"","sources":["../../src/middleware/permission-middleware.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,qBAAqB,EAAE,oBAAoB,EAAE,MAAM,6BAA6B,CAAC;AAC/F,OAAO,KAAK,EAAE,OAAO,EAAE,cAAc,EAAE,cAAc,EAAE,UAAU,EAAkC,MAAM,aAAa,CAAC;AAGvH,MAAM,WAAW,0BAA0B;IACzC,IAAI,EAAE,cAAc,CAAC;IACrB,KAAK,EAAE,cAAc,EAAE,CAAC;IACxB,KAAK,EAAE,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,CAAC;IAC3B,UAAU,EAAE,UAAU,CAAC;IACvB,KAAK,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC;CAC/D;AAED,wBAAgB,0BAA0B,CACxC,MAAM,EAAE,0BAA0B,GACjC,CAAC,OAAO,EAAE,qBAAqB,EAAE,MAAM,CAAC,EAAE,WAAW,KAAK,OAAO,CAAC,oBAAoB,GAAG,SAAS,CAAC,CAkDrG"}
|
|
@@ -0,0 +1,56 @@
|
|
|
1
|
+
import { evaluatePermission } from './permissions.js';
|
|
2
|
+
export function createPermissionMiddleware(config) {
|
|
3
|
+
const { mode, rules, tools, runContext, onAsk } = config;
|
|
4
|
+
return async (context, _signal) => {
|
|
5
|
+
const toolName = context.toolCall.name;
|
|
6
|
+
const args = context.args;
|
|
7
|
+
// Look up tool to get capabilities
|
|
8
|
+
const tool = tools.find(t => t.name === toolName);
|
|
9
|
+
const capabilities = tool?.capabilities ?? [];
|
|
10
|
+
// Check tool-specific permissionCheck first
|
|
11
|
+
if (tool?.permissionCheck) {
|
|
12
|
+
const checkResult = tool.permissionCheck(args, rules);
|
|
13
|
+
if (checkResult.behavior === 'deny') {
|
|
14
|
+
logDecision(runContext, toolName, args, 'deny', undefined, toolName);
|
|
15
|
+
return { block: true, reason: checkResult.reason };
|
|
16
|
+
}
|
|
17
|
+
if (checkResult.behavior === 'allow') {
|
|
18
|
+
logDecision(runContext, toolName, args, 'allow', undefined, toolName);
|
|
19
|
+
return undefined;
|
|
20
|
+
}
|
|
21
|
+
// 'ask' falls through to evaluatePermission
|
|
22
|
+
}
|
|
23
|
+
// Fall back to evaluatePermission
|
|
24
|
+
const result = evaluatePermission(mode, rules, toolName, args, capabilities);
|
|
25
|
+
if (result.behavior === 'allow') {
|
|
26
|
+
logDecision(runContext, toolName, args, 'allow', undefined, toolName);
|
|
27
|
+
return undefined;
|
|
28
|
+
}
|
|
29
|
+
if (result.behavior === 'deny') {
|
|
30
|
+
logDecision(runContext, toolName, args, 'deny', undefined, toolName);
|
|
31
|
+
return { block: true, reason: result.reason };
|
|
32
|
+
}
|
|
33
|
+
// 'ask' behavior
|
|
34
|
+
if (onAsk) {
|
|
35
|
+
const allowed = await onAsk(toolName, args);
|
|
36
|
+
if (allowed) {
|
|
37
|
+
logDecision(runContext, toolName, args, 'allow', undefined, toolName);
|
|
38
|
+
return undefined;
|
|
39
|
+
}
|
|
40
|
+
}
|
|
41
|
+
logDecision(runContext, toolName, args, 'deny', undefined, toolName);
|
|
42
|
+
return { block: true, reason: 'Permission denied by user' };
|
|
43
|
+
};
|
|
44
|
+
}
|
|
45
|
+
function logDecision(runContext, toolName, args, behavior, matchedRule, normalizedTarget) {
|
|
46
|
+
const decision = {
|
|
47
|
+
toolName,
|
|
48
|
+
args,
|
|
49
|
+
behavior,
|
|
50
|
+
matchedRule,
|
|
51
|
+
normalizedTarget,
|
|
52
|
+
timestamp: Date.now(),
|
|
53
|
+
};
|
|
54
|
+
runContext.permissionDecisions.push(decision);
|
|
55
|
+
}
|
|
56
|
+
//# sourceMappingURL=permission-middleware.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"permission-middleware.js","sourceRoot":"","sources":["../../src/middleware/permission-middleware.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,kBAAkB,EAAE,MAAM,kBAAkB,CAAC;AAUtD,MAAM,UAAU,0BAA0B,CACxC,MAAkC;IAElC,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,MAAM,CAAC;IAEzD,OAAO,KAAK,EAAE,OAA8B,EAAE,OAAqB,EAA6C,EAAE;QAChH,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC;QACvC,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;QAE1B,mCAAmC;QACnC,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;QAClD,MAAM,YAAY,GAAiB,IAAI,EAAE,YAAY,IAAI,EAAE,CAAC;QAE5D,4CAA4C;QAC5C,IAAI,IAAI,EAAE,eAAe,EAAE,CAAC;YAC1B,MAAM,WAAW,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YACtD,IAAI,WAAW,CAAC,QAAQ,KAAK,MAAM,EAAE,CAAC;gBACpC,WAAW,CAAC,UAAU,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;gBACrE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,CAAC,MAAM,EAAE,CAAC;YACrD,CAAC;YACD,IAAI,WAAW,CAAC,QAAQ,KAAK,OAAO,EAAE,CAAC;gBACrC,WAAW,CAAC,UAAU,EAAE,QAAQ,EAAE,IAAI,EAAE,OAAO,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;gBACtE,OAAO,SAAS,CAAC;YACnB,CAAC;YACD,4CAA4C;QAC9C,CAAC;QAED,kCAAkC;QAClC,MAAM,MAAM,GAAG,kBAAkB,CAAC,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE,IAAI,EAAE,YAAY,CAAC,CAAC;QAE7E,IAAI,MAAM,CAAC,QAAQ,KAAK,OAAO,EAAE,CAAC;YAChC,WAAW,CAAC,UAAU,EAAE,QAAQ,EAAE,IAAI,EAAE,OAAO,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;YACtE,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,IAAI,MAAM,CAAC,QAAQ,KAAK,MAAM,EAAE,CAAC;YAC/B,WAAW,CAAC,UAAU,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;YACrE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC;QAChD,CAAC;QAED,iBAAiB;QACjB,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;YAC5C,IAAI,OAAO,EAAE,CAAC;gBACZ,WAAW,CAAC,UAAU,EAAE,QAAQ,EAAE,IAAI,EAAE,OAAO,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;gBACtE,OAAO,SAAS,CAAC;YACnB,CAAC;QACH,CAAC;QAED,WAAW,CAAC,UAAU,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;QACrE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,2BAA2B,EAAE,CAAC;IAC9D,CAAC,CAAC;AACJ,CAAC;AAED,SAAS,WAAW,CAClB,UAAsB,EACtB,QAAgB,EAChB,IAAa,EACb,QAA0B,EAC1B,WAAuC,EACvC,gBAAwB;IAExB,MAAM,QAAQ,GAAuB;QACnC,QAAQ;QACR,IAAI;QACJ,QAAQ;QACR,WAAW;QACX,gBAAgB;QAChB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;KACtB,CAAC;IACF,UAAU,CAAC,mBAAmB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;AAChD,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"permission-middleware.test.d.ts","sourceRoot":"","sources":["../../src/middleware/permission-middleware.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,170 @@
|
|
|
1
|
+
import { describe, it, expect, vi } from 'vitest';
|
|
2
|
+
import { Type } from '@sinclair/typebox';
|
|
3
|
+
import { createPermissionMiddleware } from './permission-middleware.js';
|
|
4
|
+
import { createRunContext } from '../context/run-context.js';
|
|
5
|
+
function makeTool(name, capabilities = [], permissionCheck) {
|
|
6
|
+
return {
|
|
7
|
+
name,
|
|
8
|
+
label: name,
|
|
9
|
+
description: `Tool ${name}`,
|
|
10
|
+
parameters: Type.Object({}),
|
|
11
|
+
capabilities,
|
|
12
|
+
permissionCheck,
|
|
13
|
+
execute: async () => ({ content: [{ type: 'text', text: 'ok' }], details: {} }),
|
|
14
|
+
};
|
|
15
|
+
}
|
|
16
|
+
function makeContext(toolName, args = {}) {
|
|
17
|
+
return {
|
|
18
|
+
assistantMessage: {
|
|
19
|
+
role: 'assistant',
|
|
20
|
+
content: [],
|
|
21
|
+
api: 'openai-completions',
|
|
22
|
+
provider: 'openai',
|
|
23
|
+
model: 'gpt-4',
|
|
24
|
+
usage: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, totalTokens: 0, cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 } },
|
|
25
|
+
stopReason: 'stop',
|
|
26
|
+
timestamp: Date.now(),
|
|
27
|
+
},
|
|
28
|
+
toolCall: { type: 'toolCall', id: 'tc-1', name: toolName, arguments: args },
|
|
29
|
+
args,
|
|
30
|
+
context: { systemPrompt: '', messages: [], tools: [] },
|
|
31
|
+
};
|
|
32
|
+
}
|
|
33
|
+
function makeRunContext() {
|
|
34
|
+
return createRunContext({ cwd: '/tmp' });
|
|
35
|
+
}
|
|
36
|
+
describe('createPermissionMiddleware', () => {
|
|
37
|
+
it('returns undefined (allow) for allowAll mode', async () => {
|
|
38
|
+
const middleware = createPermissionMiddleware({
|
|
39
|
+
mode: 'allowAll',
|
|
40
|
+
rules: [],
|
|
41
|
+
tools: [makeTool('ReadFile', ['fs:read'])],
|
|
42
|
+
runContext: makeRunContext(),
|
|
43
|
+
});
|
|
44
|
+
const result = await middleware(makeContext('ReadFile'));
|
|
45
|
+
expect(result).toBeUndefined();
|
|
46
|
+
});
|
|
47
|
+
it('returns block for denied tool', async () => {
|
|
48
|
+
const rules = [
|
|
49
|
+
{ target: { type: 'tool', name: 'WriteFile' }, behavior: 'deny', source: 'user' },
|
|
50
|
+
];
|
|
51
|
+
const middleware = createPermissionMiddleware({
|
|
52
|
+
mode: 'default',
|
|
53
|
+
rules,
|
|
54
|
+
tools: [makeTool('WriteFile', ['fs:write'])],
|
|
55
|
+
runContext: makeRunContext(),
|
|
56
|
+
});
|
|
57
|
+
const result = await middleware(makeContext('WriteFile'));
|
|
58
|
+
expect(result).toEqual({ block: true, reason: expect.any(String) });
|
|
59
|
+
});
|
|
60
|
+
it('uses tool-specific permissionCheck if defined', async () => {
|
|
61
|
+
const permissionCheck = vi.fn().mockReturnValue({ behavior: 'deny', reason: 'custom deny' });
|
|
62
|
+
const tool = makeTool('CustomTool', ['fs:write'], permissionCheck);
|
|
63
|
+
const middleware = createPermissionMiddleware({
|
|
64
|
+
mode: 'default',
|
|
65
|
+
rules: [],
|
|
66
|
+
tools: [tool],
|
|
67
|
+
runContext: makeRunContext(),
|
|
68
|
+
});
|
|
69
|
+
const result = await middleware(makeContext('CustomTool'));
|
|
70
|
+
expect(permissionCheck).toHaveBeenCalled();
|
|
71
|
+
expect(result).toEqual({ block: true, reason: 'custom deny' });
|
|
72
|
+
});
|
|
73
|
+
it('falls back to evaluatePermission when permissionCheck returns allow', async () => {
|
|
74
|
+
const permissionCheck = vi.fn().mockReturnValue({ behavior: 'allow' });
|
|
75
|
+
const tool = makeTool('CustomTool', ['fs:write'], permissionCheck);
|
|
76
|
+
const middleware = createPermissionMiddleware({
|
|
77
|
+
mode: 'default',
|
|
78
|
+
rules: [],
|
|
79
|
+
tools: [tool],
|
|
80
|
+
runContext: makeRunContext(),
|
|
81
|
+
});
|
|
82
|
+
// permissionCheck allows, but evaluatePermission should ask for mutation cap
|
|
83
|
+
const result = await middleware(makeContext('CustomTool'));
|
|
84
|
+
// Since permissionCheck returned allow, we trust it
|
|
85
|
+
expect(result).toBeUndefined();
|
|
86
|
+
});
|
|
87
|
+
it('calls onAsk callback for ask behavior and allows if true', async () => {
|
|
88
|
+
const onAsk = vi.fn().mockResolvedValue(true);
|
|
89
|
+
const middleware = createPermissionMiddleware({
|
|
90
|
+
mode: 'default',
|
|
91
|
+
rules: [],
|
|
92
|
+
tools: [makeTool('WriteFile', ['fs:write'])],
|
|
93
|
+
runContext: makeRunContext(),
|
|
94
|
+
onAsk,
|
|
95
|
+
});
|
|
96
|
+
const result = await middleware(makeContext('WriteFile'));
|
|
97
|
+
expect(onAsk).toHaveBeenCalledWith('WriteFile', {});
|
|
98
|
+
expect(result).toBeUndefined();
|
|
99
|
+
});
|
|
100
|
+
it('calls onAsk callback for ask behavior and blocks if false', async () => {
|
|
101
|
+
const onAsk = vi.fn().mockResolvedValue(false);
|
|
102
|
+
const middleware = createPermissionMiddleware({
|
|
103
|
+
mode: 'default',
|
|
104
|
+
rules: [],
|
|
105
|
+
tools: [makeTool('WriteFile', ['fs:write'])],
|
|
106
|
+
runContext: makeRunContext(),
|
|
107
|
+
onAsk,
|
|
108
|
+
});
|
|
109
|
+
const result = await middleware(makeContext('WriteFile'));
|
|
110
|
+
expect(result).toEqual({ block: true, reason: expect.any(String) });
|
|
111
|
+
});
|
|
112
|
+
it('denies when ask behavior and no onAsk callback', async () => {
|
|
113
|
+
const middleware = createPermissionMiddleware({
|
|
114
|
+
mode: 'default',
|
|
115
|
+
rules: [],
|
|
116
|
+
tools: [makeTool('WriteFile', ['fs:write'])],
|
|
117
|
+
runContext: makeRunContext(),
|
|
118
|
+
});
|
|
119
|
+
const result = await middleware(makeContext('WriteFile'));
|
|
120
|
+
expect(result).toEqual({ block: true, reason: expect.any(String) });
|
|
121
|
+
});
|
|
122
|
+
it('logs permission decision to runContext', async () => {
|
|
123
|
+
const runContext = makeRunContext();
|
|
124
|
+
const middleware = createPermissionMiddleware({
|
|
125
|
+
mode: 'allowAll',
|
|
126
|
+
rules: [],
|
|
127
|
+
tools: [makeTool('ReadFile', ['fs:read'])],
|
|
128
|
+
runContext,
|
|
129
|
+
});
|
|
130
|
+
await middleware(makeContext('ReadFile'));
|
|
131
|
+
expect(runContext.permissionDecisions).toHaveLength(1);
|
|
132
|
+
const decision = runContext.permissionDecisions[0];
|
|
133
|
+
expect(decision.toolName).toBe('ReadFile');
|
|
134
|
+
expect(decision.behavior).toBe('allow');
|
|
135
|
+
expect(decision.timestamp).toBeGreaterThan(0);
|
|
136
|
+
});
|
|
137
|
+
it('handles unknown tool (not in tools list) with empty capabilities', async () => {
|
|
138
|
+
const middleware = createPermissionMiddleware({
|
|
139
|
+
mode: 'default',
|
|
140
|
+
rules: [],
|
|
141
|
+
tools: [],
|
|
142
|
+
runContext: makeRunContext(),
|
|
143
|
+
});
|
|
144
|
+
// Unknown tool in default mode with no caps → allow (no mutation capabilities)
|
|
145
|
+
const result = await middleware(makeContext('UnknownTool'));
|
|
146
|
+
expect(result).toBeUndefined();
|
|
147
|
+
});
|
|
148
|
+
it('allows read-only tools in default mode without rules', async () => {
|
|
149
|
+
const middleware = createPermissionMiddleware({
|
|
150
|
+
mode: 'default',
|
|
151
|
+
rules: [],
|
|
152
|
+
tools: [makeTool('ReadFile', ['fs:read'])],
|
|
153
|
+
runContext: makeRunContext(),
|
|
154
|
+
});
|
|
155
|
+
const result = await middleware(makeContext('ReadFile'));
|
|
156
|
+
expect(result).toBeUndefined();
|
|
157
|
+
});
|
|
158
|
+
it('passes signal to middleware', async () => {
|
|
159
|
+
const controller = new AbortController();
|
|
160
|
+
const middleware = createPermissionMiddleware({
|
|
161
|
+
mode: 'allowAll',
|
|
162
|
+
rules: [],
|
|
163
|
+
tools: [makeTool('ReadFile', ['fs:read'])],
|
|
164
|
+
runContext: makeRunContext(),
|
|
165
|
+
});
|
|
166
|
+
const result = await middleware(makeContext('ReadFile'), controller.signal);
|
|
167
|
+
expect(result).toBeUndefined();
|
|
168
|
+
});
|
|
169
|
+
});
|
|
170
|
+
//# sourceMappingURL=permission-middleware.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"permission-middleware.test.js","sourceRoot":"","sources":["../../src/middleware/permission-middleware.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAClD,OAAO,EAAE,IAAI,EAAE,MAAM,mBAAmB,CAAC;AACzC,OAAO,EAAE,0BAA0B,EAAE,MAAM,4BAA4B,CAAC;AAGxE,OAAO,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAC;AAE7D,SAAS,QAAQ,CAAC,IAAY,EAAE,eAAwC,EAAE,EAAE,eAA4C;IACtH,OAAO;QACL,IAAI;QACJ,KAAK,EAAE,IAAI;QACX,WAAW,EAAE,QAAQ,IAAI,EAAE;QAC3B,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;QAC3B,YAAY;QACZ,eAAe;QACf,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;KACzF,CAAC;AACJ,CAAC;AAED,SAAS,WAAW,CAAC,QAAgB,EAAE,OAA4B,EAAE;IACnE,OAAO;QACL,gBAAgB,EAAE;YAChB,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,EAAE;YACX,GAAG,EAAE,oBAAoB;YACzB,QAAQ,EAAE,QAAQ;YAClB,KAAK,EAAE,OAAO;YACd,KAAK,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,IAAI,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE;YACjJ,UAAU,EAAE,MAAM;YAClB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB;QACD,QAAQ,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,SAAS,EAAE,IAAI,EAAE;QAC3E,IAAI;QACJ,OAAO,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE;KACvD,CAAC;AACJ,CAAC;AAED,SAAS,cAAc;IACrB,OAAO,gBAAgB,CAAC,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC,CAAC;AAC3C,CAAC;AAED,QAAQ,CAAC,4BAA4B,EAAE,GAAG,EAAE;IAC1C,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC3D,MAAM,UAAU,GAAG,0BAA0B,CAAC;YAC5C,IAAI,EAAE,UAAU;YAChB,KAAK,EAAE,EAAE;YACT,KAAK,EAAE,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;YAC1C,UAAU,EAAE,cAAc,EAAE;SAC7B,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC,CAAC;QACzD,MAAM,CAAC,MAAM,CAAC,CAAC,aAAa,EAAE,CAAC;IACjC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;QAC7C,MAAM,KAAK,GAAqB;YAC9B,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE;SAClF,CAAC;QACF,MAAM,UAAU,GAAG,0BAA0B,CAAC;YAC5C,IAAI,EAAE,SAAS;YACf,KAAK;YACL,KAAK,EAAE,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC;YAC5C,UAAU,EAAE,cAAc,EAAE;SAC7B,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,CAAC;QAC1D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IACtE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC7D,MAAM,eAAe,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC,CAAC;QAC7F,MAAM,IAAI,GAAG,QAAQ,CAAC,YAAY,EAAE,CAAC,UAAU,CAAC,EAAE,eAAe,CAAC,CAAC;QACnE,MAAM,UAAU,GAAG,0BAA0B,CAAC;YAC5C,IAAI,EAAE,SAAS;YACf,KAAK,EAAE,EAAE;YACT,KAAK,EAAE,CAAC,IAAI,CAAC;YACb,UAAU,EAAE,cAAc,EAAE;SAC7B,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC,CAAC;QAC3D,MAAM,CAAC,eAAe,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAC3C,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC,CAAC;IACjE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;QACnF,MAAM,eAAe,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC;QACvE,MAAM,IAAI,GAAG,QAAQ,CAAC,YAAY,EAAE,CAAC,UAAU,CAAC,EAAE,eAAe,CAAC,CAAC;QACnE,MAAM,UAAU,GAAG,0BAA0B,CAAC;YAC5C,IAAI,EAAE,SAAS;YACf,KAAK,EAAE,EAAE;YACT,KAAK,EAAE,CAAC,IAAI,CAAC;YACb,UAAU,EAAE,cAAc,EAAE;SAC7B,CAAC,CAAC;QACH,6EAA6E;QAC7E,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC,CAAC;QAC3D,oDAAoD;QACpD,MAAM,CAAC,MAAM,CAAC,CAAC,aAAa,EAAE,CAAC;IACjC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;QACxE,MAAM,KAAK,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAC9C,MAAM,UAAU,GAAG,0BAA0B,CAAC;YAC5C,IAAI,EAAE,SAAS;YACf,KAAK,EAAE,EAAE;YACT,KAAK,EAAE,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC;YAC5C,UAAU,EAAE,cAAc,EAAE;YAC5B,KAAK;SACN,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,CAAC;QAC1D,MAAM,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;QACpD,MAAM,CAAC,MAAM,CAAC,CAAC,aAAa,EAAE,CAAC;IACjC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QACzE,MAAM,KAAK,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;QAC/C,MAAM,UAAU,GAAG,0BAA0B,CAAC;YAC5C,IAAI,EAAE,SAAS;YACf,KAAK,EAAE,EAAE;YACT,KAAK,EAAE,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC;YAC5C,UAAU,EAAE,cAAc,EAAE;YAC5B,KAAK;SACN,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,CAAC;QAC1D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IACtE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QAC9D,MAAM,UAAU,GAAG,0BAA0B,CAAC;YAC5C,IAAI,EAAE,SAAS;YACf,KAAK,EAAE,EAAE;YACT,KAAK,EAAE,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC;YAC5C,UAAU,EAAE,cAAc,EAAE;SAC7B,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,CAAC;QAC1D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IACtE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;QACtD,MAAM,UAAU,GAAG,cAAc,EAAE,CAAC;QACpC,MAAM,UAAU,GAAG,0BAA0B,CAAC;YAC5C,IAAI,EAAE,UAAU;YAChB,KAAK,EAAE,EAAE;YACT,KAAK,EAAE,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;YAC1C,UAAU;SACX,CAAC,CAAC;QACH,MAAM,UAAU,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC,CAAC;QAC1C,MAAM,CAAC,UAAU,CAAC,mBAAmB,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACvD,MAAM,QAAQ,GAAG,UAAU,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC;QACnD,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC3C,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACxC,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;QAChF,MAAM,UAAU,GAAG,0BAA0B,CAAC;YAC5C,IAAI,EAAE,SAAS;YACf,KAAK,EAAE,EAAE;YACT,KAAK,EAAE,EAAE;YACT,UAAU,EAAE,cAAc,EAAE;SAC7B,CAAC,CAAC;QACH,+EAA+E;QAC/E,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC,CAAC;QAC5D,MAAM,CAAC,MAAM,CAAC,CAAC,aAAa,EAAE,CAAC;IACjC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;QACpE,MAAM,UAAU,GAAG,0BAA0B,CAAC;YAC5C,IAAI,EAAE,SAAS;YACf,KAAK,EAAE,EAAE;YACT,KAAK,EAAE,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;YAC1C,UAAU,EAAE,cAAc,EAAE;SAC7B,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC,CAAC;QACzD,MAAM,CAAC,MAAM,CAAC,CAAC,aAAa,EAAE,CAAC;IACjC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;QAC3C,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;QACzC,MAAM,UAAU,GAAG,0BAA0B,CAAC;YAC5C,IAAI,EAAE,UAAU;YAChB,KAAK,EAAE,EAAE;YACT,KAAK,EAAE,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;YAC1C,UAAU,EAAE,cAAc,EAAE;SAC7B,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,WAAW,CAAC,UAAU,CAAC,EAAE,UAAU,CAAC,MAAM,CAAC,CAAC;QAC5E,MAAM,CAAC,MAAM,CAAC,CAAC,aAAa,EAAE,CAAC;IACjC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
import type { PermissionRule, PermissionResult, PermissionMode, Capability } from '../types.js';
|
|
2
|
+
/**
|
|
3
|
+
* Check if a rule matches a tool invocation.
|
|
4
|
+
*/
|
|
5
|
+
export declare function matchRule(rule: PermissionRule, toolName: string, args: unknown, capabilities: Capability[]): boolean;
|
|
6
|
+
/**
|
|
7
|
+
* Find the best matching rule by combined score (specificity * 10 + source_priority).
|
|
8
|
+
*/
|
|
9
|
+
export declare function findMatchingRule(rules: PermissionRule[], toolName: string, args: unknown, capabilities: Capability[]): PermissionRule | null;
|
|
10
|
+
/**
|
|
11
|
+
* Evaluate permission result based on mode, rules, and capabilities.
|
|
12
|
+
*/
|
|
13
|
+
export declare function evaluatePermission(mode: PermissionMode, rules: PermissionRule[], toolName: string, args: unknown, capabilities: Capability[]): PermissionResult;
|
|
14
|
+
//# sourceMappingURL=permissions.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"permissions.d.ts","sourceRoot":"","sources":["../../src/middleware/permissions.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,gBAAgB,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AA8ChG;;GAEG;AACH,wBAAgB,SAAS,CACvB,IAAI,EAAE,cAAc,EACpB,QAAQ,EAAE,MAAM,EAChB,IAAI,EAAE,OAAO,EACb,YAAY,EAAE,UAAU,EAAE,GACzB,OAAO,CAwBT;AAED;;GAEG;AACH,wBAAgB,gBAAgB,CAC9B,KAAK,EAAE,cAAc,EAAE,EACvB,QAAQ,EAAE,MAAM,EAChB,IAAI,EAAE,OAAO,EACb,YAAY,EAAE,UAAU,EAAE,GACzB,cAAc,GAAG,IAAI,CAcvB;AAED;;GAEG;AACH,wBAAgB,kBAAkB,CAChC,IAAI,EAAE,cAAc,EACpB,KAAK,EAAE,cAAc,EAAE,EACvB,QAAQ,EAAE,MAAM,EAChB,IAAI,EAAE,OAAO,EACb,YAAY,EAAE,UAAU,EAAE,GACzB,gBAAgB,CA0BlB"}
|