@matthias-hausberger/beige 0.0.1
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/LICENSE.md +8 -0
- package/README.md +183 -0
- package/dist/channels/registry.d.ts +14 -0
- package/dist/channels/registry.d.ts.map +1 -0
- package/dist/channels/registry.js +14 -0
- package/dist/channels/registry.js.map +1 -0
- package/dist/channels/telegram.d.ts +92 -0
- package/dist/channels/telegram.d.ts.map +1 -0
- package/dist/channels/telegram.js +469 -0
- package/dist/channels/telegram.js.map +1 -0
- package/dist/channels/tui.d.ts +8 -0
- package/dist/channels/tui.d.ts.map +1 -0
- package/dist/channels/tui.js +574 -0
- package/dist/channels/tui.js.map +1 -0
- package/dist/cli.d.ts +23 -0
- package/dist/cli.d.ts.map +1 -0
- package/dist/cli.js +571 -0
- package/dist/cli.js.map +1 -0
- package/dist/config/loader.d.ts +6 -0
- package/dist/config/loader.d.ts.map +1 -0
- package/dist/config/loader.js +103 -0
- package/dist/config/loader.js.map +1 -0
- package/dist/config/loader.spec.d.ts +2 -0
- package/dist/config/loader.spec.d.ts.map +1 -0
- package/dist/config/loader.spec.js +195 -0
- package/dist/config/loader.spec.js.map +1 -0
- package/dist/config/schema.d.ts +107 -0
- package/dist/config/schema.d.ts.map +1 -0
- package/dist/config/schema.js +42 -0
- package/dist/config/schema.js.map +1 -0
- package/dist/config/schema.spec.d.ts +2 -0
- package/dist/config/schema.spec.d.ts.map +1 -0
- package/dist/config/schema.spec.js +180 -0
- package/dist/config/schema.spec.js.map +1 -0
- package/dist/gateway/agent-manager.d.ts +138 -0
- package/dist/gateway/agent-manager.d.ts.map +1 -0
- package/dist/gateway/agent-manager.js +532 -0
- package/dist/gateway/agent-manager.js.map +1 -0
- package/dist/gateway/api.d.ts +43 -0
- package/dist/gateway/api.d.ts.map +1 -0
- package/dist/gateway/api.js +256 -0
- package/dist/gateway/api.js.map +1 -0
- package/dist/gateway/api.spec.d.ts +2 -0
- package/dist/gateway/api.spec.d.ts.map +1 -0
- package/dist/gateway/api.spec.js +256 -0
- package/dist/gateway/api.spec.js.map +1 -0
- package/dist/gateway/audit.d.ts +38 -0
- package/dist/gateway/audit.d.ts.map +1 -0
- package/dist/gateway/audit.js +82 -0
- package/dist/gateway/audit.js.map +1 -0
- package/dist/gateway/audit.spec.d.ts +2 -0
- package/dist/gateway/audit.spec.d.ts.map +1 -0
- package/dist/gateway/audit.spec.js +212 -0
- package/dist/gateway/audit.spec.js.map +1 -0
- package/dist/gateway/gateway.d.ts +27 -0
- package/dist/gateway/gateway.d.ts.map +1 -0
- package/dist/gateway/gateway.js +158 -0
- package/dist/gateway/gateway.js.map +1 -0
- package/dist/gateway/policy.d.ts +27 -0
- package/dist/gateway/policy.d.ts.map +1 -0
- package/dist/gateway/policy.js +40 -0
- package/dist/gateway/policy.js.map +1 -0
- package/dist/gateway/policy.spec.d.ts +2 -0
- package/dist/gateway/policy.spec.d.ts.map +1 -0
- package/dist/gateway/policy.spec.js +121 -0
- package/dist/gateway/policy.spec.js.map +1 -0
- package/dist/gateway/provider-health.d.ts +83 -0
- package/dist/gateway/provider-health.d.ts.map +1 -0
- package/dist/gateway/provider-health.js +219 -0
- package/dist/gateway/provider-health.js.map +1 -0
- package/dist/gateway/provider-health.spec.d.ts +2 -0
- package/dist/gateway/provider-health.spec.d.ts.map +1 -0
- package/dist/gateway/provider-health.spec.js +278 -0
- package/dist/gateway/provider-health.spec.js.map +1 -0
- package/dist/gateway/session-settings.d.ts +62 -0
- package/dist/gateway/session-settings.d.ts.map +1 -0
- package/dist/gateway/session-settings.js +91 -0
- package/dist/gateway/session-settings.js.map +1 -0
- package/dist/gateway/session-settings.spec.d.ts +2 -0
- package/dist/gateway/session-settings.spec.d.ts.map +1 -0
- package/dist/gateway/session-settings.spec.js +141 -0
- package/dist/gateway/session-settings.spec.js.map +1 -0
- package/dist/gateway/sessions.d.ts +68 -0
- package/dist/gateway/sessions.d.ts.map +1 -0
- package/dist/gateway/sessions.js +177 -0
- package/dist/gateway/sessions.js.map +1 -0
- package/dist/gateway/sessions.spec.d.ts +2 -0
- package/dist/gateway/sessions.spec.d.ts.map +1 -0
- package/dist/gateway/sessions.spec.js +190 -0
- package/dist/gateway/sessions.spec.js.map +1 -0
- package/dist/index.d.ts +11 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +10 -0
- package/dist/index.js.map +1 -0
- package/dist/install.d.ts +39 -0
- package/dist/install.d.ts.map +1 -0
- package/dist/install.js +144 -0
- package/dist/install.js.map +1 -0
- package/dist/sandbox/manager.d.ts +63 -0
- package/dist/sandbox/manager.d.ts.map +1 -0
- package/dist/sandbox/manager.js +294 -0
- package/dist/sandbox/manager.js.map +1 -0
- package/dist/skills/index.d.ts +2 -0
- package/dist/skills/index.d.ts.map +1 -0
- package/dist/skills/index.js +2 -0
- package/dist/skills/index.js.map +1 -0
- package/dist/skills/registry.d.ts +11 -0
- package/dist/skills/registry.d.ts.map +1 -0
- package/dist/skills/registry.js +86 -0
- package/dist/skills/registry.js.map +1 -0
- package/dist/skills/registry.spec.d.ts +2 -0
- package/dist/skills/registry.spec.d.ts.map +1 -0
- package/dist/skills/registry.spec.js +220 -0
- package/dist/skills/registry.spec.js.map +1 -0
- package/dist/socket/protocol.d.ts +21 -0
- package/dist/socket/protocol.d.ts.map +1 -0
- package/dist/socket/protocol.js +7 -0
- package/dist/socket/protocol.js.map +1 -0
- package/dist/socket/protocol.spec.d.ts +2 -0
- package/dist/socket/protocol.spec.d.ts.map +1 -0
- package/dist/socket/protocol.spec.js +135 -0
- package/dist/socket/protocol.spec.js.map +1 -0
- package/dist/socket/server.d.ts +21 -0
- package/dist/socket/server.d.ts.map +1 -0
- package/dist/socket/server.js +133 -0
- package/dist/socket/server.js.map +1 -0
- package/dist/socket/server.spec.d.ts +2 -0
- package/dist/socket/server.spec.d.ts.map +1 -0
- package/dist/socket/server.spec.js +333 -0
- package/dist/socket/server.spec.js.map +1 -0
- package/dist/test/fixtures.d.ts +47 -0
- package/dist/test/fixtures.d.ts.map +1 -0
- package/dist/test/fixtures.js +144 -0
- package/dist/test/fixtures.js.map +1 -0
- package/dist/toolkit/index.d.ts +4 -0
- package/dist/toolkit/index.d.ts.map +1 -0
- package/dist/toolkit/index.js +4 -0
- package/dist/toolkit/index.js.map +1 -0
- package/dist/toolkit/installer.d.ts +26 -0
- package/dist/toolkit/installer.d.ts.map +1 -0
- package/dist/toolkit/installer.js +247 -0
- package/dist/toolkit/installer.js.map +1 -0
- package/dist/toolkit/registry.d.ts +19 -0
- package/dist/toolkit/registry.d.ts.map +1 -0
- package/dist/toolkit/registry.js +119 -0
- package/dist/toolkit/registry.js.map +1 -0
- package/dist/toolkit/registry.spec.d.ts +2 -0
- package/dist/toolkit/registry.spec.d.ts.map +1 -0
- package/dist/toolkit/registry.spec.js +194 -0
- package/dist/toolkit/registry.spec.js.map +1 -0
- package/dist/toolkit/schema.d.ts +61 -0
- package/dist/toolkit/schema.d.ts.map +1 -0
- package/dist/toolkit/schema.js +116 -0
- package/dist/toolkit/schema.js.map +1 -0
- package/dist/toolkit/schema.spec.d.ts +2 -0
- package/dist/toolkit/schema.spec.d.ts.map +1 -0
- package/dist/toolkit/schema.spec.js +202 -0
- package/dist/toolkit/schema.spec.js.map +1 -0
- package/dist/tools/core.d.ts +10 -0
- package/dist/tools/core.d.ts.map +1 -0
- package/dist/tools/core.js +246 -0
- package/dist/tools/core.js.map +1 -0
- package/dist/tools/core.spec.d.ts +2 -0
- package/dist/tools/core.spec.d.ts.map +1 -0
- package/dist/tools/core.spec.js +315 -0
- package/dist/tools/core.spec.js.map +1 -0
- package/dist/tools/registry.d.ts +15 -0
- package/dist/tools/registry.d.ts.map +1 -0
- package/dist/tools/registry.js +62 -0
- package/dist/tools/registry.js.map +1 -0
- package/dist/tools/registry.spec.d.ts +2 -0
- package/dist/tools/registry.spec.d.ts.map +1 -0
- package/dist/tools/registry.spec.js +228 -0
- package/dist/tools/registry.spec.js.map +1 -0
- package/dist/tools/runner.d.ts +25 -0
- package/dist/tools/runner.d.ts.map +1 -0
- package/dist/tools/runner.js +35 -0
- package/dist/tools/runner.js.map +1 -0
- package/dist/tools/runner.spec.d.ts +2 -0
- package/dist/tools/runner.spec.d.ts.map +1 -0
- package/dist/tools/runner.spec.js +129 -0
- package/dist/tools/runner.spec.js.map +1 -0
- package/dist/types/session.d.ts +8 -0
- package/dist/types/session.d.ts.map +1 -0
- package/dist/types/session.js +23 -0
- package/dist/types/session.js.map +1 -0
- package/package.json +76 -0
- package/tools/README.md +1 -0
- package/tools/kv/README.md +150 -0
- package/tools/kv/index.ts +149 -0
- package/tools/kv/tool.json +23 -0
- package/tools/message/README.md +53 -0
- package/tools/message/index.ts +183 -0
- package/tools/message/tool.json +11 -0
|
@@ -0,0 +1,82 @@
|
|
|
1
|
+
import { appendFileSync, mkdirSync } from "fs";
|
|
2
|
+
import { dirname } from "path";
|
|
3
|
+
/** Format a Date as "YYYY-MM-DD HH:mm:ss" in local time. */
|
|
4
|
+
function formatTimestamp(date) {
|
|
5
|
+
const pad = (n, w = 2) => String(n).padStart(w, "0");
|
|
6
|
+
return (`${date.getFullYear()}-${pad(date.getMonth() + 1)}-${pad(date.getDate())} ` +
|
|
7
|
+
`${pad(date.getHours())}:${pad(date.getMinutes())}:${pad(date.getSeconds())}`);
|
|
8
|
+
}
|
|
9
|
+
export class AuditLogger {
|
|
10
|
+
logPath;
|
|
11
|
+
constructor(logPath) {
|
|
12
|
+
this.logPath = logPath;
|
|
13
|
+
mkdirSync(dirname(logPath), { recursive: true });
|
|
14
|
+
}
|
|
15
|
+
log(entry) {
|
|
16
|
+
const line = JSON.stringify(entry) + "\n";
|
|
17
|
+
appendFileSync(this.logPath, line);
|
|
18
|
+
const ts = formatTimestamp(new Date(entry.ts));
|
|
19
|
+
const emoji = entry.decision === "allowed" ? "✓" : "✗";
|
|
20
|
+
const typeLabel = entry.type === "core_tool" ? "CORE" : "TOOL";
|
|
21
|
+
const argsStr = entry.args.join(" ");
|
|
22
|
+
const agentTool = `${entry.agent}/${entry.tool}`;
|
|
23
|
+
if (entry.phase === "started") {
|
|
24
|
+
// "→ started" line — no duration yet
|
|
25
|
+
console.log(`[AUDIT] [${ts}] ${emoji} ${typeLabel} ${agentTool} ${argsStr} → ${entry.decision}, started`);
|
|
26
|
+
}
|
|
27
|
+
else {
|
|
28
|
+
// "→ finished" line — include duration (and error if present)
|
|
29
|
+
const durationStr = entry.durationMs !== undefined ? ` (${entry.durationMs}ms)` : "";
|
|
30
|
+
const errorStr = entry.error ? `, error: ${entry.error}` : "";
|
|
31
|
+
console.log(`[AUDIT] [${ts}] ${emoji} ${typeLabel} ${agentTool} ${argsStr} → finished${durationStr}${errorStr}`);
|
|
32
|
+
}
|
|
33
|
+
}
|
|
34
|
+
/**
|
|
35
|
+
* Create a timed audit entry. Call .finish() when done.
|
|
36
|
+
* For "denied" decisions, logs immediately (no started/finished split).
|
|
37
|
+
* For "allowed" decisions, logs a "started" line immediately, then a "finished"
|
|
38
|
+
* line when .finish() is called.
|
|
39
|
+
*/
|
|
40
|
+
start(agent, type, tool, args, decision, target) {
|
|
41
|
+
const entry = {
|
|
42
|
+
ts: new Date().toISOString(),
|
|
43
|
+
agent,
|
|
44
|
+
phase: decision === "denied" ? "finished" : "started",
|
|
45
|
+
type,
|
|
46
|
+
tool,
|
|
47
|
+
args,
|
|
48
|
+
decision,
|
|
49
|
+
target,
|
|
50
|
+
};
|
|
51
|
+
// Always log on start — denied tools are fully logged here; allowed tools
|
|
52
|
+
// get a "started" line now and a "finished" line after execution.
|
|
53
|
+
this.log(entry);
|
|
54
|
+
return new AuditTimer(this, entry);
|
|
55
|
+
}
|
|
56
|
+
}
|
|
57
|
+
export class AuditTimer {
|
|
58
|
+
logger;
|
|
59
|
+
entry;
|
|
60
|
+
startTime;
|
|
61
|
+
constructor(logger, entry) {
|
|
62
|
+
this.logger = logger;
|
|
63
|
+
this.entry = entry;
|
|
64
|
+
this.startTime = Date.now();
|
|
65
|
+
}
|
|
66
|
+
finish(result) {
|
|
67
|
+
// Only write a "finished" entry for allowed tools (denied was already fully logged).
|
|
68
|
+
if (this.entry.decision === "denied")
|
|
69
|
+
return;
|
|
70
|
+
const finishedEntry = {
|
|
71
|
+
...this.entry,
|
|
72
|
+
ts: new Date().toISOString(),
|
|
73
|
+
phase: "finished",
|
|
74
|
+
durationMs: Date.now() - this.startTime,
|
|
75
|
+
exitCode: result.exitCode,
|
|
76
|
+
outputBytes: result.outputBytes,
|
|
77
|
+
error: result.error,
|
|
78
|
+
};
|
|
79
|
+
this.logger.log(finishedEntry);
|
|
80
|
+
}
|
|
81
|
+
}
|
|
82
|
+
//# sourceMappingURL=audit.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"audit.js","sourceRoot":"","sources":["../../src/gateway/audit.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAC/C,OAAO,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AAiB/B,4DAA4D;AAC5D,SAAS,eAAe,CAAC,IAAU;IACjC,MAAM,GAAG,GAAG,CAAC,CAAS,EAAE,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;IAC7D,OAAO,CACL,GAAG,IAAI,CAAC,WAAW,EAAE,IAAI,GAAG,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,GAAG;QAC3E,GAAG,GAAG,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC,EAAE,CAC9E,CAAC;AACJ,CAAC;AAED,MAAM,OAAO,WAAW;IACd,OAAO,CAAS;IAExB,YAAY,OAAe;QACzB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,SAAS,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACnD,CAAC;IAED,GAAG,CAAC,KAAiB;QACnB,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC;QAC1C,cAAc,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QAEnC,MAAM,EAAE,GAAG,eAAe,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC;QAC/C,MAAM,KAAK,GAAG,KAAK,CAAC,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC;QACvD,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC;QAC/D,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACrC,MAAM,SAAS,GAAG,GAAG,KAAK,CAAC,KAAK,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;QAEjD,IAAI,KAAK,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;YAC9B,qCAAqC;YACrC,OAAO,CAAC,GAAG,CACT,YAAY,EAAE,KAAK,KAAK,IAAI,SAAS,IAAI,SAAS,IAAI,OAAO,MAAM,KAAK,CAAC,QAAQ,WAAW,CAC7F,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,8DAA8D;YAC9D,MAAM,WAAW,GAAG,KAAK,CAAC,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,KAAK,KAAK,CAAC,UAAU,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;YACrF,MAAM,QAAQ,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,YAAY,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC9D,OAAO,CAAC,GAAG,CACT,YAAY,EAAE,KAAK,KAAK,IAAI,SAAS,IAAI,SAAS,IAAI,OAAO,cAAc,WAAW,GAAG,QAAQ,EAAE,CACpG,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACH,KAAK,CACH,KAAa,EACb,IAA0B,EAC1B,IAAY,EACZ,IAAc,EACd,QAA8B,EAC9B,MAA8B;QAE9B,MAAM,KAAK,GAAe;YACxB,EAAE,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YAC5B,KAAK;YACL,KAAK,EAAE,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS;YACrD,IAAI;YACJ,IAAI;YACJ,IAAI;YACJ,QAAQ;YACR,MAAM;SACP,CAAC;QAEF,0EAA0E;QAC1E,kEAAkE;QAClE,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAEhB,OAAO,IAAI,UAAU,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;IACrC,CAAC;CACF;AAED,MAAM,OAAO,UAAU;IAIX;IACA;IAJF,SAAS,CAAS;IAE1B,YACU,MAAmB,EACnB,KAAiB;QADjB,WAAM,GAAN,MAAM,CAAa;QACnB,UAAK,GAAL,KAAK,CAAY;QAEzB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC9B,CAAC;IAED,MAAM,CAAC,MAAmE;QACxE,qFAAqF;QACrF,IAAI,IAAI,CAAC,KAAK,CAAC,QAAQ,KAAK,QAAQ;YAAE,OAAO;QAE7C,MAAM,aAAa,GAAe;YAChC,GAAG,IAAI,CAAC,KAAK;YACb,EAAE,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YAC5B,KAAK,EAAE,UAAU;YACjB,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,SAAS;YACvC,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,WAAW,EAAE,MAAM,CAAC,WAAW;YAC/B,KAAK,EAAE,MAAM,CAAC,KAAK;SACpB,CAAC;QACF,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;IACjC,CAAC;CACF"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"audit.spec.d.ts","sourceRoot":"","sources":["../../src/gateway/audit.spec.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,212 @@
|
|
|
1
|
+
import { describe, it, expect, beforeEach, afterEach } from "vitest";
|
|
2
|
+
import { AuditLogger, AuditTimer } from "./audit.js";
|
|
3
|
+
import { readFileSync, mkdirSync, rmSync, existsSync } from "fs";
|
|
4
|
+
import { join } from "path";
|
|
5
|
+
import { tmpdir } from "os";
|
|
6
|
+
describe("AuditLogger", () => {
|
|
7
|
+
let tempDir;
|
|
8
|
+
let logPath;
|
|
9
|
+
let logger;
|
|
10
|
+
beforeEach(() => {
|
|
11
|
+
tempDir = join(tmpdir(), `beige-audit-test-${Date.now()}`);
|
|
12
|
+
mkdirSync(tempDir, { recursive: true });
|
|
13
|
+
logPath = join(tempDir, "audit.jsonl");
|
|
14
|
+
logger = new AuditLogger(logPath);
|
|
15
|
+
});
|
|
16
|
+
afterEach(() => {
|
|
17
|
+
rmSync(tempDir, { recursive: true, force: true });
|
|
18
|
+
});
|
|
19
|
+
describe("log", () => {
|
|
20
|
+
it("writes JSONL entry to file", () => {
|
|
21
|
+
logger.log({
|
|
22
|
+
ts: "2026-03-08T00:00:00.000Z",
|
|
23
|
+
agent: "assistant",
|
|
24
|
+
phase: "finished",
|
|
25
|
+
type: "core_tool",
|
|
26
|
+
tool: "exec",
|
|
27
|
+
args: ["ls"],
|
|
28
|
+
decision: "allowed",
|
|
29
|
+
exitCode: 0,
|
|
30
|
+
outputBytes: 100,
|
|
31
|
+
});
|
|
32
|
+
const content = readFileSync(logPath, "utf-8");
|
|
33
|
+
const lines = content.trim().split("\n");
|
|
34
|
+
expect(lines.length).toBe(1);
|
|
35
|
+
expect(JSON.parse(lines[0]).tool).toBe("exec");
|
|
36
|
+
});
|
|
37
|
+
it("appends multiple entries", () => {
|
|
38
|
+
logger.log({
|
|
39
|
+
ts: "2026-03-08T00:00:00.000Z",
|
|
40
|
+
agent: "assistant",
|
|
41
|
+
phase: "finished",
|
|
42
|
+
type: "core_tool",
|
|
43
|
+
tool: "read",
|
|
44
|
+
args: ["/test.txt"],
|
|
45
|
+
decision: "allowed",
|
|
46
|
+
});
|
|
47
|
+
logger.log({
|
|
48
|
+
ts: "2026-03-08T00:00:01.000Z",
|
|
49
|
+
agent: "assistant",
|
|
50
|
+
phase: "finished",
|
|
51
|
+
type: "core_tool",
|
|
52
|
+
tool: "write",
|
|
53
|
+
args: ["/test.txt"],
|
|
54
|
+
decision: "allowed",
|
|
55
|
+
});
|
|
56
|
+
const content = readFileSync(logPath, "utf-8");
|
|
57
|
+
const lines = content.trim().split("\n");
|
|
58
|
+
expect(lines.length).toBe(2);
|
|
59
|
+
});
|
|
60
|
+
it("includes all required fields", () => {
|
|
61
|
+
logger.log({
|
|
62
|
+
ts: "2026-03-08T00:00:00.000Z",
|
|
63
|
+
agent: "assistant",
|
|
64
|
+
phase: "finished",
|
|
65
|
+
type: "core_tool",
|
|
66
|
+
tool: "exec",
|
|
67
|
+
args: ["ls", "-la"],
|
|
68
|
+
decision: "allowed",
|
|
69
|
+
durationMs: 123,
|
|
70
|
+
exitCode: 0,
|
|
71
|
+
outputBytes: 500,
|
|
72
|
+
});
|
|
73
|
+
const content = readFileSync(logPath, "utf-8");
|
|
74
|
+
const entry = JSON.parse(content);
|
|
75
|
+
expect(entry.ts).toBe("2026-03-08T00:00:00.000Z");
|
|
76
|
+
expect(entry.agent).toBe("assistant");
|
|
77
|
+
expect(entry.phase).toBe("finished");
|
|
78
|
+
expect(entry.type).toBe("core_tool");
|
|
79
|
+
expect(entry.tool).toBe("exec");
|
|
80
|
+
expect(entry.args).toEqual(["ls", "-la"]);
|
|
81
|
+
expect(entry.decision).toBe("allowed");
|
|
82
|
+
expect(entry.durationMs).toBe(123);
|
|
83
|
+
expect(entry.exitCode).toBe(0);
|
|
84
|
+
expect(entry.outputBytes).toBe(500);
|
|
85
|
+
});
|
|
86
|
+
it("logs denied decisions", () => {
|
|
87
|
+
logger.log({
|
|
88
|
+
ts: "2026-03-08T00:00:00.000Z",
|
|
89
|
+
agent: "assistant",
|
|
90
|
+
phase: "finished",
|
|
91
|
+
type: "tool",
|
|
92
|
+
tool: "kv",
|
|
93
|
+
args: ["set", "key", "value"],
|
|
94
|
+
decision: "denied",
|
|
95
|
+
error: "Permission denied",
|
|
96
|
+
});
|
|
97
|
+
const content = readFileSync(logPath, "utf-8");
|
|
98
|
+
const entry = JSON.parse(content);
|
|
99
|
+
expect(entry.decision).toBe("denied");
|
|
100
|
+
expect(entry.error).toBe("Permission denied");
|
|
101
|
+
});
|
|
102
|
+
it("logs with target field for tool calls", () => {
|
|
103
|
+
logger.log({
|
|
104
|
+
ts: "2026-03-08T00:00:00.000Z",
|
|
105
|
+
agent: "assistant",
|
|
106
|
+
phase: "finished",
|
|
107
|
+
type: "tool",
|
|
108
|
+
tool: "kv",
|
|
109
|
+
args: ["get", "key"],
|
|
110
|
+
decision: "allowed",
|
|
111
|
+
target: "gateway",
|
|
112
|
+
});
|
|
113
|
+
const content = readFileSync(logPath, "utf-8");
|
|
114
|
+
const entry = JSON.parse(content);
|
|
115
|
+
expect(entry.target).toBe("gateway");
|
|
116
|
+
});
|
|
117
|
+
});
|
|
118
|
+
describe("start/finish pattern", () => {
|
|
119
|
+
it("creates a timer for allowed decisions", () => {
|
|
120
|
+
const timer = logger.start("assistant", "core_tool", "exec", ["ls"], "allowed");
|
|
121
|
+
expect(timer).toBeInstanceOf(AuditTimer);
|
|
122
|
+
});
|
|
123
|
+
it("logs immediately for denied decisions", () => {
|
|
124
|
+
logger.start("assistant", "tool", "kv", ["set", "key"], "denied");
|
|
125
|
+
const content = readFileSync(logPath, "utf-8");
|
|
126
|
+
const lines = content.trim().split("\n");
|
|
127
|
+
// Denied decisions log immediately as "finished"
|
|
128
|
+
expect(lines.length).toBe(1);
|
|
129
|
+
const entry = JSON.parse(lines[0]);
|
|
130
|
+
expect(entry.phase).toBe("finished");
|
|
131
|
+
expect(entry.decision).toBe("denied");
|
|
132
|
+
});
|
|
133
|
+
it("logs 'started' phase for allowed decisions", () => {
|
|
134
|
+
logger.start("assistant", "core_tool", "exec", ["ls"], "allowed");
|
|
135
|
+
const content = readFileSync(logPath, "utf-8");
|
|
136
|
+
const lines = content.trim().split("\n");
|
|
137
|
+
expect(lines.length).toBe(1);
|
|
138
|
+
const entry = JSON.parse(lines[0]);
|
|
139
|
+
expect(entry.phase).toBe("started");
|
|
140
|
+
});
|
|
141
|
+
it("logs 'finished' phase when timer.finish() is called", () => {
|
|
142
|
+
const timer = logger.start("assistant", "core_tool", "exec", ["ls"], "allowed");
|
|
143
|
+
timer.finish({ exitCode: 0, outputBytes: 100 });
|
|
144
|
+
const content = readFileSync(logPath, "utf-8");
|
|
145
|
+
const lines = content.trim().split("\n");
|
|
146
|
+
expect(lines.length).toBe(2);
|
|
147
|
+
const startedEntry = JSON.parse(lines[0]);
|
|
148
|
+
const finishedEntry = JSON.parse(lines[1]);
|
|
149
|
+
expect(startedEntry.phase).toBe("started");
|
|
150
|
+
expect(finishedEntry.phase).toBe("finished");
|
|
151
|
+
});
|
|
152
|
+
it("tracks duration between start and finish", async () => {
|
|
153
|
+
const timer = logger.start("assistant", "core_tool", "exec", ["ls"], "allowed");
|
|
154
|
+
await new Promise((resolve) => setTimeout(resolve, 50));
|
|
155
|
+
timer.finish({ exitCode: 0 });
|
|
156
|
+
const content = readFileSync(logPath, "utf-8");
|
|
157
|
+
const lines = content.trim().split("\n");
|
|
158
|
+
const finishedEntry = JSON.parse(lines[1]);
|
|
159
|
+
expect(finishedEntry.durationMs).toBeGreaterThanOrEqual(40);
|
|
160
|
+
});
|
|
161
|
+
it("finish() does nothing for denied decisions", () => {
|
|
162
|
+
const timer = logger.start("assistant", "tool", "kv", ["get"], "denied");
|
|
163
|
+
timer.finish({ exitCode: 1, error: "Denied" });
|
|
164
|
+
const content = readFileSync(logPath, "utf-8");
|
|
165
|
+
const lines = content.trim().split("\n");
|
|
166
|
+
// Only one line (the immediate "finished" log)
|
|
167
|
+
expect(lines.length).toBe(1);
|
|
168
|
+
});
|
|
169
|
+
it("includes error in finish result", () => {
|
|
170
|
+
const timer = logger.start("assistant", "core_tool", "exec", ["fail"], "allowed");
|
|
171
|
+
timer.finish({ exitCode: 1, error: "Command failed" });
|
|
172
|
+
const content = readFileSync(logPath, "utf-8");
|
|
173
|
+
const lines = content.trim().split("\n");
|
|
174
|
+
const finishedEntry = JSON.parse(lines[1]);
|
|
175
|
+
expect(finishedEntry.error).toBe("Command failed");
|
|
176
|
+
});
|
|
177
|
+
});
|
|
178
|
+
describe("concurrent writes", () => {
|
|
179
|
+
it("handles multiple concurrent log calls", async () => {
|
|
180
|
+
const promises = Array.from({ length: 10 }, (_, i) => Promise.resolve(logger.log({
|
|
181
|
+
ts: new Date().toISOString(),
|
|
182
|
+
agent: "assistant",
|
|
183
|
+
phase: "finished",
|
|
184
|
+
type: "core_tool",
|
|
185
|
+
tool: "exec",
|
|
186
|
+
args: [`cmd${i}`],
|
|
187
|
+
decision: "allowed",
|
|
188
|
+
})));
|
|
189
|
+
await Promise.all(promises);
|
|
190
|
+
const content = readFileSync(logPath, "utf-8");
|
|
191
|
+
const lines = content.trim().split("\n");
|
|
192
|
+
expect(lines.length).toBe(10);
|
|
193
|
+
});
|
|
194
|
+
});
|
|
195
|
+
describe("directory creation", () => {
|
|
196
|
+
it("creates parent directories if they don't exist", () => {
|
|
197
|
+
const nestedPath = join(tempDir, "deeply", "nested", "audit.jsonl");
|
|
198
|
+
const nestedLogger = new AuditLogger(nestedPath);
|
|
199
|
+
nestedLogger.log({
|
|
200
|
+
ts: "2026-03-08T00:00:00.000Z",
|
|
201
|
+
agent: "assistant",
|
|
202
|
+
phase: "finished",
|
|
203
|
+
type: "core_tool",
|
|
204
|
+
tool: "exec",
|
|
205
|
+
args: [],
|
|
206
|
+
decision: "allowed",
|
|
207
|
+
});
|
|
208
|
+
expect(existsSync(nestedPath)).toBe(true);
|
|
209
|
+
});
|
|
210
|
+
});
|
|
211
|
+
});
|
|
212
|
+
//# sourceMappingURL=audit.spec.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"audit.spec.js","sourceRoot":"","sources":["../../src/gateway/audit.spec.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACrE,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AACrD,OAAO,EAAiB,YAAY,EAAE,SAAS,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,IAAI,CAAC;AAChF,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAC5B,OAAO,EAAE,MAAM,EAAE,MAAM,IAAI,CAAC;AAE5B,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,IAAI,OAAe,CAAC;IACpB,IAAI,OAAe,CAAC;IACpB,IAAI,MAAmB,CAAC;IAExB,UAAU,CAAC,GAAG,EAAE;QACd,OAAO,GAAG,IAAI,CAAC,MAAM,EAAE,EAAE,oBAAoB,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QAC3D,SAAS,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACxC,OAAO,GAAG,IAAI,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;QACvC,MAAM,GAAG,IAAI,WAAW,CAAC,OAAO,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,MAAM,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,KAAK,EAAE,GAAG,EAAE;QACnB,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;YACpC,MAAM,CAAC,GAAG,CAAC;gBACT,EAAE,EAAE,0BAA0B;gBAC9B,KAAK,EAAE,WAAW;gBAClB,KAAK,EAAE,UAAU;gBACjB,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,CAAC,IAAI,CAAC;gBACZ,QAAQ,EAAE,SAAS;gBACnB,QAAQ,EAAE,CAAC;gBACX,WAAW,EAAE,GAAG;aACjB,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,YAAY,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAC/C,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAEzC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC7B,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;YAClC,MAAM,CAAC,GAAG,CAAC;gBACT,EAAE,EAAE,0BAA0B;gBAC9B,KAAK,EAAE,WAAW;gBAClB,KAAK,EAAE,UAAU;gBACjB,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,CAAC,WAAW,CAAC;gBACnB,QAAQ,EAAE,SAAS;aACpB,CAAC,CAAC;YAEH,MAAM,CAAC,GAAG,CAAC;gBACT,EAAE,EAAE,0BAA0B;gBAC9B,KAAK,EAAE,WAAW;gBAClB,KAAK,EAAE,UAAU;gBACjB,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,CAAC,WAAW,CAAC;gBACnB,QAAQ,EAAE,SAAS;aACpB,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,YAAY,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAC/C,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAEzC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;YACtC,MAAM,CAAC,GAAG,CAAC;gBACT,EAAE,EAAE,0BAA0B;gBAC9B,KAAK,EAAE,WAAW;gBAClB,KAAK,EAAE,UAAU;gBACjB,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,CAAC,IAAI,EAAE,KAAK,CAAC;gBACnB,QAAQ,EAAE,SAAS;gBACnB,UAAU,EAAE,GAAG;gBACf,QAAQ,EAAE,CAAC;gBACX,WAAW,EAAE,GAAG;aACjB,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,YAAY,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAC/C,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAElC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;YAClD,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACtC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACrC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACrC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAChC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC;YAC1C,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACvC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACnC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uBAAuB,EAAE,GAAG,EAAE;YAC/B,MAAM,CAAC,GAAG,CAAC;gBACT,EAAE,EAAE,0BAA0B;gBAC9B,KAAK,EAAE,WAAW;gBAClB,KAAK,EAAE,UAAU;gBACjB,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,IAAI;gBACV,IAAI,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,OAAO,CAAC;gBAC7B,QAAQ,EAAE,QAAQ;gBAClB,KAAK,EAAE,mBAAmB;aAC3B,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,YAAY,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAC/C,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAElC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACtC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,MAAM,CAAC,GAAG,CAAC;gBACT,EAAE,EAAE,0BAA0B;gBAC9B,KAAK,EAAE,WAAW;gBAClB,KAAK,EAAE,UAAU;gBACjB,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,IAAI;gBACV,IAAI,EAAE,CAAC,KAAK,EAAE,KAAK,CAAC;gBACpB,QAAQ,EAAE,SAAS;gBACnB,MAAM,EAAE,SAAS;aAClB,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,YAAY,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAC/C,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAElC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CACxB,WAAW,EACX,WAAW,EACX,MAAM,EACN,CAAC,IAAI,CAAC,EACN,SAAS,CACV,CAAC;YAEF,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,MAAM,CAAC,KAAK,CACV,WAAW,EACX,MAAM,EACN,IAAI,EACJ,CAAC,KAAK,EAAE,KAAK,CAAC,EACd,QAAQ,CACT,CAAC;YAEF,MAAM,OAAO,GAAG,YAAY,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAC/C,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAEzC,iDAAiD;YACjD,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC7B,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;YACnC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACrC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;YACpD,MAAM,CAAC,KAAK,CACV,WAAW,EACX,WAAW,EACX,MAAM,EACN,CAAC,IAAI,CAAC,EACN,SAAS,CACV,CAAC;YAEF,MAAM,OAAO,GAAG,YAAY,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAC/C,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAEzC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC7B,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;YACnC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CACxB,WAAW,EACX,WAAW,EACX,MAAM,EACN,CAAC,IAAI,CAAC,EACN,SAAS,CACV,CAAC;YAEF,KAAK,CAAC,MAAM,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC;YAEhD,MAAM,OAAO,GAAG,YAAY,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAC/C,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAEzC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAE7B,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;YAC1C,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;YAE3C,MAAM,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC3C,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CACxB,WAAW,EACX,WAAW,EACX,MAAM,EACN,CAAC,IAAI,CAAC,EACN,SAAS,CACV,CAAC;YAEF,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;YAExD,KAAK,CAAC,MAAM,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC;YAE9B,MAAM,OAAO,GAAG,YAAY,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAC/C,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACzC,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;YAE3C,MAAM,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC,sBAAsB,CAAC,EAAE,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;YACpD,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CACxB,WAAW,EACX,MAAM,EACN,IAAI,EACJ,CAAC,KAAK,CAAC,EACP,QAAQ,CACT,CAAC;YAEF,KAAK,CAAC,MAAM,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;YAE/C,MAAM,OAAO,GAAG,YAAY,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAC/C,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAEzC,+CAA+C;YAC/C,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CACxB,WAAW,EACX,WAAW,EACX,MAAM,EACN,CAAC,MAAM,CAAC,EACR,SAAS,CACV,CAAC;YAEF,KAAK,CAAC,MAAM,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,KAAK,EAAE,gBAAgB,EAAE,CAAC,CAAC;YAEvD,MAAM,OAAO,GAAG,YAAY,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAC/C,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACzC,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;YAE3C,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CACnD,OAAO,CAAC,OAAO,CACb,MAAM,CAAC,GAAG,CAAC;gBACT,EAAE,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;gBAC5B,KAAK,EAAE,WAAW;gBAClB,KAAK,EAAE,UAAU;gBACjB,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC;gBACjB,QAAQ,EAAE,SAAS;aACpB,CAAC,CACH,CACF,CAAC;YAEF,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAE5B,MAAM,OAAO,GAAG,YAAY,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAC/C,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAEzC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,aAAa,CAAC,CAAC;YACpE,MAAM,YAAY,GAAG,IAAI,WAAW,CAAC,UAAU,CAAC,CAAC;YAEjD,YAAY,CAAC,GAAG,CAAC;gBACf,EAAE,EAAE,0BAA0B;gBAC9B,KAAK,EAAE,WAAW;gBAClB,KAAK,EAAE,UAAU;gBACjB,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,EAAE;gBACR,QAAQ,EAAE,SAAS;aACpB,CAAC,CAAC;YAEH,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,27 @@
|
|
|
1
|
+
import type { BeigeConfig } from "../config/schema.js";
|
|
2
|
+
export declare class Gateway {
|
|
3
|
+
private config;
|
|
4
|
+
private configPath;
|
|
5
|
+
private audit;
|
|
6
|
+
private policy;
|
|
7
|
+
private toolRunner;
|
|
8
|
+
private sessionStore;
|
|
9
|
+
private settingsStore;
|
|
10
|
+
private sandboxManager;
|
|
11
|
+
private agentManager;
|
|
12
|
+
private api;
|
|
13
|
+
private socketServers;
|
|
14
|
+
private telegramChannel?;
|
|
15
|
+
private loadedTools;
|
|
16
|
+
private loadedSkills;
|
|
17
|
+
private channelRegistry;
|
|
18
|
+
private restarting;
|
|
19
|
+
constructor(config: BeigeConfig, configPath: string);
|
|
20
|
+
start(): Promise<void>;
|
|
21
|
+
stop(): Promise<void>;
|
|
22
|
+
restart(): Promise<void>;
|
|
23
|
+
private teardown;
|
|
24
|
+
private startAgentInfra;
|
|
25
|
+
private setupAuth;
|
|
26
|
+
}
|
|
27
|
+
//# sourceMappingURL=gateway.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"gateway.d.ts","sourceRoot":"","sources":["../../src/gateway/gateway.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAgBvD,qBAAa,OAAO;IAClB,OAAO,CAAC,MAAM,CAAc;IAC5B,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,KAAK,CAAc;IAC3B,OAAO,CAAC,MAAM,CAAe;IAC7B,OAAO,CAAC,UAAU,CAAa;IAC/B,OAAO,CAAC,YAAY,CAAoB;IACxC,OAAO,CAAC,aAAa,CAAuB;IAC5C,OAAO,CAAC,cAAc,CAAkB;IACxC,OAAO,CAAC,YAAY,CAAgB;IACpC,OAAO,CAAC,GAAG,CAAc;IACzB,OAAO,CAAC,aAAa,CAAwC;IAC7D,OAAO,CAAC,eAAe,CAAC,CAAkB;IAC1C,OAAO,CAAC,WAAW,CAA2B;IAC9C,OAAO,CAAC,YAAY,CAA4B;IAChD,OAAO,CAAC,eAAe,CAAmB;IAC1C,OAAO,CAAC,UAAU,CAAS;gBAEf,MAAM,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM;IAY7C,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IA0EtB,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAMrB,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;YAmChB,QAAQ;YAsBR,eAAe;IAoB7B,OAAO,CAAC,SAAS;CAWlB"}
|
|
@@ -0,0 +1,158 @@
|
|
|
1
|
+
import { resolve } from "path";
|
|
2
|
+
import { homedir } from "os";
|
|
3
|
+
import { AuthStorage, ModelRegistry } from "@mariozechner/pi-coding-agent";
|
|
4
|
+
import { loadConfig } from "../config/loader.js";
|
|
5
|
+
import { AuditLogger } from "./audit.js";
|
|
6
|
+
import { PolicyEngine } from "./policy.js";
|
|
7
|
+
import { AgentManager } from "./agent-manager.js";
|
|
8
|
+
import { BeigeSessionStore } from "./sessions.js";
|
|
9
|
+
import { SessionSettingsStore } from "./session-settings.js";
|
|
10
|
+
import { GatewayAPI } from "./api.js";
|
|
11
|
+
import { SandboxManager } from "../sandbox/manager.js";
|
|
12
|
+
import { AgentSocketServer } from "../socket/server.js";
|
|
13
|
+
import { ToolRunner } from "../tools/runner.js";
|
|
14
|
+
import { loadTools } from "../tools/registry.js";
|
|
15
|
+
import { loadSkills } from "../skills/registry.js";
|
|
16
|
+
import { TelegramChannel } from "../channels/telegram.js";
|
|
17
|
+
import { ChannelRegistry } from "../channels/registry.js";
|
|
18
|
+
export class Gateway {
|
|
19
|
+
config;
|
|
20
|
+
configPath;
|
|
21
|
+
audit;
|
|
22
|
+
policy;
|
|
23
|
+
toolRunner;
|
|
24
|
+
sessionStore;
|
|
25
|
+
settingsStore;
|
|
26
|
+
sandboxManager;
|
|
27
|
+
agentManager;
|
|
28
|
+
api;
|
|
29
|
+
socketServers = new Map();
|
|
30
|
+
telegramChannel;
|
|
31
|
+
loadedTools;
|
|
32
|
+
loadedSkills;
|
|
33
|
+
channelRegistry;
|
|
34
|
+
restarting = false;
|
|
35
|
+
constructor(config, configPath) {
|
|
36
|
+
this.config = config;
|
|
37
|
+
this.configPath = resolve(configPath);
|
|
38
|
+
this.audit = new AuditLogger(resolve(homedir(), ".beige", "logs", "audit.jsonl"));
|
|
39
|
+
this.policy = new PolicyEngine(config);
|
|
40
|
+
this.toolRunner = new ToolRunner();
|
|
41
|
+
this.sessionStore = new BeigeSessionStore();
|
|
42
|
+
this.settingsStore = new SessionSettingsStore();
|
|
43
|
+
}
|
|
44
|
+
async start() {
|
|
45
|
+
console.log("[GATEWAY] Starting Beige gateway...");
|
|
46
|
+
// 1. Create channel registry
|
|
47
|
+
this.channelRegistry = new ChannelRegistry();
|
|
48
|
+
// 2. Load tool packages and register handlers
|
|
49
|
+
this.loadedTools = await loadTools(this.config, this.toolRunner, this.channelRegistry);
|
|
50
|
+
console.log(`[GATEWAY] Loaded ${this.loadedTools.size} tool(s)`);
|
|
51
|
+
// 3. Load skill packages
|
|
52
|
+
this.loadedSkills = await loadSkills(this.config);
|
|
53
|
+
console.log(`[GATEWAY] Loaded ${this.loadedSkills.size} skill(s)`);
|
|
54
|
+
// 4. Set up auth and model registry for pi SDK
|
|
55
|
+
const authStorage = this.setupAuth();
|
|
56
|
+
const modelRegistry = new ModelRegistry(authStorage);
|
|
57
|
+
// 5. Create sandbox manager
|
|
58
|
+
this.sandboxManager = new SandboxManager(this.config, this.loadedTools, this.loadedSkills);
|
|
59
|
+
// 6. Create agent manager
|
|
60
|
+
this.agentManager = new AgentManager(this.config, this.sandboxManager, this.audit, this.loadedTools, this.loadedSkills, authStorage, modelRegistry, this.sessionStore);
|
|
61
|
+
// 7. Build beige-sandbox image if any agent needs it (no-op otherwise)
|
|
62
|
+
await this.sandboxManager.ensureSandboxImage();
|
|
63
|
+
// 8. Start sandboxes and socket servers for each agent
|
|
64
|
+
for (const agentName of Object.keys(this.config.agents)) {
|
|
65
|
+
await this.startAgentInfra(agentName);
|
|
66
|
+
}
|
|
67
|
+
// 9. Start HTTP API (for TUI and other external channels)
|
|
68
|
+
const host = this.config.gateway?.host ?? "127.0.0.1";
|
|
69
|
+
const port = this.config.gateway?.port ?? 7433;
|
|
70
|
+
this.api = new GatewayAPI({
|
|
71
|
+
config: this.config,
|
|
72
|
+
gateway: this,
|
|
73
|
+
agentManager: this.agentManager,
|
|
74
|
+
sessionStore: this.sessionStore,
|
|
75
|
+
sandbox: this.sandboxManager,
|
|
76
|
+
audit: this.audit,
|
|
77
|
+
host,
|
|
78
|
+
port,
|
|
79
|
+
});
|
|
80
|
+
await this.api.start();
|
|
81
|
+
// 10. Start Telegram channel (non-blocking)
|
|
82
|
+
if (this.config.channels?.telegram?.enabled) {
|
|
83
|
+
this.telegramChannel = new TelegramChannel(this.config.channels.telegram, this.agentManager, this.sessionStore, this.settingsStore, this.channelRegistry);
|
|
84
|
+
this.telegramChannel.start().catch((err) => {
|
|
85
|
+
console.error("[GATEWAY] Telegram bot error:", err);
|
|
86
|
+
});
|
|
87
|
+
}
|
|
88
|
+
console.log("[GATEWAY] Beige gateway started ✓");
|
|
89
|
+
}
|
|
90
|
+
async stop() {
|
|
91
|
+
console.log("[GATEWAY] Shutting down...");
|
|
92
|
+
await this.teardown({ drain: true });
|
|
93
|
+
console.log("[GATEWAY] Shutdown complete");
|
|
94
|
+
}
|
|
95
|
+
async restart() {
|
|
96
|
+
if (this.restarting) {
|
|
97
|
+
console.log("[GATEWAY] Restart already in progress — ignoring duplicate request");
|
|
98
|
+
return;
|
|
99
|
+
}
|
|
100
|
+
this.restarting = true;
|
|
101
|
+
try {
|
|
102
|
+
console.log("[GATEWAY] ── Restart requested ──────────────────────────────");
|
|
103
|
+
console.log("[GATEWAY] Phase 1/3: Draining in-flight calls...");
|
|
104
|
+
await this.agentManager?.drainAll();
|
|
105
|
+
console.log("[GATEWAY] Phase 2/3: Tearing down infrastructure...");
|
|
106
|
+
await this.teardown({ drain: false });
|
|
107
|
+
console.log(`[GATEWAY] Phase 3/3: Reloading config from ${this.configPath}...`);
|
|
108
|
+
try {
|
|
109
|
+
this.config = loadConfig(this.configPath);
|
|
110
|
+
}
|
|
111
|
+
catch (err) {
|
|
112
|
+
console.error("[GATEWAY] Failed to reload config — aborting restart:", err);
|
|
113
|
+
console.error("[GATEWAY] Gateway is now stopped. Fix the config and run 'beige gateway start'.");
|
|
114
|
+
return;
|
|
115
|
+
}
|
|
116
|
+
this.policy = new PolicyEngine(this.config);
|
|
117
|
+
this.toolRunner = new ToolRunner();
|
|
118
|
+
await this.start();
|
|
119
|
+
console.log("[GATEWAY] ── Restart complete ───────────────────────────────");
|
|
120
|
+
}
|
|
121
|
+
finally {
|
|
122
|
+
this.restarting = false;
|
|
123
|
+
}
|
|
124
|
+
}
|
|
125
|
+
async teardown(opts) {
|
|
126
|
+
if (opts.drain) {
|
|
127
|
+
await this.agentManager?.drainAll();
|
|
128
|
+
}
|
|
129
|
+
if (this.telegramChannel) {
|
|
130
|
+
await this.telegramChannel.stop();
|
|
131
|
+
this.telegramChannel = undefined;
|
|
132
|
+
}
|
|
133
|
+
await this.api?.stop();
|
|
134
|
+
for (const [, server] of this.socketServers) {
|
|
135
|
+
await server.stop();
|
|
136
|
+
}
|
|
137
|
+
this.socketServers.clear();
|
|
138
|
+
await this.agentManager?.shutdown();
|
|
139
|
+
await this.sandboxManager?.shutdown();
|
|
140
|
+
}
|
|
141
|
+
async startAgentInfra(agentName) {
|
|
142
|
+
const socketPath = resolve(homedir(), ".beige", "sockets", `${agentName}.sock`);
|
|
143
|
+
const socketServer = new AgentSocketServer(agentName, socketPath, this.audit, this.policy, this.toolRunner);
|
|
144
|
+
await socketServer.start();
|
|
145
|
+
this.socketServers.set(agentName, socketServer);
|
|
146
|
+
await this.sandboxManager.createSandbox(agentName);
|
|
147
|
+
}
|
|
148
|
+
setupAuth() {
|
|
149
|
+
const authStorage = AuthStorage.create();
|
|
150
|
+
for (const [providerName, providerConfig] of Object.entries(this.config.llm.providers)) {
|
|
151
|
+
if (providerConfig.apiKey) {
|
|
152
|
+
authStorage.setRuntimeApiKey(providerName, providerConfig.apiKey);
|
|
153
|
+
}
|
|
154
|
+
}
|
|
155
|
+
return authStorage;
|
|
156
|
+
}
|
|
157
|
+
}
|
|
158
|
+
//# sourceMappingURL=gateway.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"gateway.js","sourceRoot":"","sources":["../../src/gateway/gateway.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AAC/B,OAAO,EAAE,OAAO,EAAE,MAAM,IAAI,CAAC;AAC7B,OAAO,EAAE,WAAW,EAAE,aAAa,EAAE,MAAM,+BAA+B,CAAC;AAE3E,OAAO,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AACjD,OAAO,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AACzC,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,iBAAiB,EAAE,MAAM,eAAe,CAAC;AAClD,OAAO,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AAC7D,OAAO,EAAE,UAAU,EAAE,MAAM,UAAU,CAAC;AACtC,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AACvD,OAAO,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AACxD,OAAO,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAChD,OAAO,EAAE,SAAS,EAAmB,MAAM,sBAAsB,CAAC;AAClE,OAAO,EAAE,UAAU,EAAoB,MAAM,uBAAuB,CAAC;AACrE,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC1D,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAE1D,MAAM,OAAO,OAAO;IACV,MAAM,CAAc;IACpB,UAAU,CAAS;IACnB,KAAK,CAAc;IACnB,MAAM,CAAe;IACrB,UAAU,CAAa;IACvB,YAAY,CAAoB;IAChC,aAAa,CAAuB;IACpC,cAAc,CAAkB;IAChC,YAAY,CAAgB;IAC5B,GAAG,CAAc;IACjB,aAAa,GAAG,IAAI,GAAG,EAA6B,CAAC;IACrD,eAAe,CAAmB;IAClC,WAAW,CAA2B;IACtC,YAAY,CAA4B;IACxC,eAAe,CAAmB;IAClC,UAAU,GAAG,KAAK,CAAC;IAE3B,YAAY,MAAmB,EAAE,UAAkB;QACjD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;QACtC,IAAI,CAAC,KAAK,GAAG,IAAI,WAAW,CAC1B,OAAO,CAAC,OAAO,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,aAAa,CAAC,CACpD,CAAC;QACF,IAAI,CAAC,MAAM,GAAG,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC;QACvC,IAAI,CAAC,UAAU,GAAG,IAAI,UAAU,EAAE,CAAC;QACnC,IAAI,CAAC,YAAY,GAAG,IAAI,iBAAiB,EAAE,CAAC;QAC5C,IAAI,CAAC,aAAa,GAAG,IAAI,oBAAoB,EAAE,CAAC;IAClD,CAAC;IAED,KAAK,CAAC,KAAK;QACT,OAAO,CAAC,GAAG,CAAC,qCAAqC,CAAC,CAAC;QAEnD,6BAA6B;QAC7B,IAAI,CAAC,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;QAE7C,8CAA8C;QAC9C,IAAI,CAAC,WAAW,GAAG,MAAM,SAAS,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,eAAe,CAAC,CAAC;QACvF,OAAO,CAAC,GAAG,CAAC,oBAAoB,IAAI,CAAC,WAAW,CAAC,IAAI,UAAU,CAAC,CAAC;QAEjE,yBAAyB;QACzB,IAAI,CAAC,YAAY,GAAG,MAAM,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAClD,OAAO,CAAC,GAAG,CAAC,oBAAoB,IAAI,CAAC,YAAY,CAAC,IAAI,WAAW,CAAC,CAAC;QAEnE,+CAA+C;QAC/C,MAAM,WAAW,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;QACrC,MAAM,aAAa,GAAG,IAAI,aAAa,CAAC,WAAW,CAAC,CAAC;QAErD,4BAA4B;QAC5B,IAAI,CAAC,cAAc,GAAG,IAAI,cAAc,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;QAE3F,0BAA0B;QAC1B,IAAI,CAAC,YAAY,GAAG,IAAI,YAAY,CAClC,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,cAAc,EACnB,IAAI,CAAC,KAAK,EACV,IAAI,CAAC,WAAW,EAChB,IAAI,CAAC,YAAY,EACjB,WAAW,EACX,aAAa,EACb,IAAI,CAAC,YAAY,CAClB,CAAC;QAEF,uEAAuE;QACvE,MAAM,IAAI,CAAC,cAAc,CAAC,kBAAkB,EAAE,CAAC;QAE/C,uDAAuD;QACvD,KAAK,MAAM,SAAS,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;YACxD,MAAM,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;QACxC,CAAC;QAED,0DAA0D;QAC1D,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,IAAI,IAAI,WAAW,CAAC;QACtD,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,IAAI,IAAI,IAAI,CAAC;QAE/C,IAAI,CAAC,GAAG,GAAG,IAAI,UAAU,CAAC;YACxB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,OAAO,EAAE,IAAI;YACb,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,OAAO,EAAE,IAAI,CAAC,cAAc;YAC5B,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,IAAI;YACJ,IAAI;SACL,CAAC,CAAC;QACH,MAAM,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC;QAEvB,4CAA4C;QAC5C,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC;YAC5C,IAAI,CAAC,eAAe,GAAG,IAAI,eAAe,CACxC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAC7B,IAAI,CAAC,YAAY,EACjB,IAAI,CAAC,YAAY,EACjB,IAAI,CAAC,aAAa,EAClB,IAAI,CAAC,eAAe,CACrB,CAAC;YACF,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;gBACzC,OAAO,CAAC,KAAK,CAAC,+BAA+B,EAAE,GAAG,CAAC,CAAC;YACtD,CAAC,CAAC,CAAC;QACL,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,mCAAmC,CAAC,CAAC;IACnD,CAAC;IAED,KAAK,CAAC,IAAI;QACR,OAAO,CAAC,GAAG,CAAC,4BAA4B,CAAC,CAAC;QAC1C,MAAM,IAAI,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QACrC,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC,CAAC;IAC7C,CAAC;IAED,KAAK,CAAC,OAAO;QACX,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,OAAO,CAAC,GAAG,CAAC,oEAAoE,CAAC,CAAC;YAClF,OAAO;QACT,CAAC;QACD,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;QAEvB,IAAI,CAAC;YACH,OAAO,CAAC,GAAG,CAAC,+DAA+D,CAAC,CAAC;YAE7E,OAAO,CAAC,GAAG,CAAC,kDAAkD,CAAC,CAAC;YAChE,MAAM,IAAI,CAAC,YAAY,EAAE,QAAQ,EAAE,CAAC;YAEpC,OAAO,CAAC,GAAG,CAAC,qDAAqD,CAAC,CAAC;YACnE,MAAM,IAAI,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;YAEtC,OAAO,CAAC,GAAG,CAAC,8CAA8C,IAAI,CAAC,UAAU,KAAK,CAAC,CAAC;YAChF,IAAI,CAAC;gBACH,IAAI,CAAC,MAAM,GAAG,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAC5C,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,OAAO,CAAC,KAAK,CAAC,uDAAuD,EAAE,GAAG,CAAC,CAAC;gBAC5E,OAAO,CAAC,KAAK,CAAC,iFAAiF,CAAC,CAAC;gBACjG,OAAO;YACT,CAAC;YAED,IAAI,CAAC,MAAM,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC5C,IAAI,CAAC,UAAU,GAAG,IAAI,UAAU,EAAE,CAAC;YAEnC,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;YACnB,OAAO,CAAC,GAAG,CAAC,+DAA+D,CAAC,CAAC;QAC/E,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;QAC1B,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,QAAQ,CAAC,IAAwB;QAC7C,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,CAAC,YAAY,EAAE,QAAQ,EAAE,CAAC;QACtC,CAAC;QAED,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YACzB,MAAM,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC;YAClC,IAAI,CAAC,eAAe,GAAG,SAAS,CAAC;QACnC,CAAC;QAED,MAAM,IAAI,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;QAEvB,KAAK,MAAM,CAAC,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YAC5C,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;QACtB,CAAC;QACD,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,CAAC;QAE3B,MAAM,IAAI,CAAC,YAAY,EAAE,QAAQ,EAAE,CAAC;QAEpC,MAAM,IAAI,CAAC,cAAc,EAAE,QAAQ,EAAE,CAAC;IACxC,CAAC;IAEO,KAAK,CAAC,eAAe,CAAC,SAAiB;QAC7C,MAAM,UAAU,GAAG,OAAO,CACxB,OAAO,EAAE,EACT,QAAQ,EACR,SAAS,EACT,GAAG,SAAS,OAAO,CACpB,CAAC;QACF,MAAM,YAAY,GAAG,IAAI,iBAAiB,CACxC,SAAS,EACT,UAAU,EACV,IAAI,CAAC,KAAK,EACV,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,UAAU,CAChB,CAAC;QACF,MAAM,YAAY,CAAC,KAAK,EAAE,CAAC;QAC3B,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC;QAEhD,MAAM,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;IACrD,CAAC;IAEO,SAAS;QACf,MAAM,WAAW,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC;QACzC,KAAK,MAAM,CAAC,YAAY,EAAE,cAAc,CAAC,IAAI,MAAM,CAAC,OAAO,CACzD,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,SAAS,CAC1B,EAAE,CAAC;YACF,IAAI,cAAc,CAAC,MAAM,EAAE,CAAC;gBAC1B,WAAW,CAAC,gBAAgB,CAAC,YAAY,EAAE,cAAc,CAAC,MAAM,CAAC,CAAC;YACpE,CAAC;QACH,CAAC;QACD,OAAO,WAAW,CAAC;IACrB,CAAC;CACF"}
|
|
@@ -0,0 +1,27 @@
|
|
|
1
|
+
import type { BeigeConfig } from "../config/schema.js";
|
|
2
|
+
/**
|
|
3
|
+
* Policy engine. Deny by default.
|
|
4
|
+
* Checks whether an agent is allowed to use a specific tool.
|
|
5
|
+
*/
|
|
6
|
+
export declare class PolicyEngine {
|
|
7
|
+
private config;
|
|
8
|
+
constructor(config: BeigeConfig);
|
|
9
|
+
/**
|
|
10
|
+
* Check if an agent is allowed to use a core tool (read, write, patch, exec).
|
|
11
|
+
* Core tools are always allowed for any configured agent.
|
|
12
|
+
*/
|
|
13
|
+
isCoreTool(tool: string): boolean;
|
|
14
|
+
/**
|
|
15
|
+
* Check if an agent is allowed to execute a tool.
|
|
16
|
+
*/
|
|
17
|
+
isToolAllowed(agentName: string, toolName: string): boolean;
|
|
18
|
+
/**
|
|
19
|
+
* Check if an agent exists in the config.
|
|
20
|
+
*/
|
|
21
|
+
isAgentValid(agentName: string): boolean;
|
|
22
|
+
/**
|
|
23
|
+
* Get the target for a tool (gateway or sandbox).
|
|
24
|
+
*/
|
|
25
|
+
getToolTarget(toolName: string): "gateway" | "sandbox" | undefined;
|
|
26
|
+
}
|
|
27
|
+
//# sourceMappingURL=policy.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"policy.d.ts","sourceRoot":"","sources":["../../src/gateway/policy.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAEvD;;;GAGG;AACH,qBAAa,YAAY;IACX,OAAO,CAAC,MAAM;gBAAN,MAAM,EAAE,WAAW;IAEvC;;;OAGG;IACH,UAAU,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO;IAIjC;;OAEG;IACH,aAAa,CAAC,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,OAAO;IAM3D;;OAEG;IACH,YAAY,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO;IAIxC;;OAEG;IACH,aAAa,CAAC,QAAQ,EAAE,MAAM,GAAG,SAAS,GAAG,SAAS,GAAG,SAAS;CAInE"}
|
|
@@ -0,0 +1,40 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Policy engine. Deny by default.
|
|
3
|
+
* Checks whether an agent is allowed to use a specific tool.
|
|
4
|
+
*/
|
|
5
|
+
export class PolicyEngine {
|
|
6
|
+
config;
|
|
7
|
+
constructor(config) {
|
|
8
|
+
this.config = config;
|
|
9
|
+
}
|
|
10
|
+
/**
|
|
11
|
+
* Check if an agent is allowed to use a core tool (read, write, patch, exec).
|
|
12
|
+
* Core tools are always allowed for any configured agent.
|
|
13
|
+
*/
|
|
14
|
+
isCoreTool(tool) {
|
|
15
|
+
return ["read", "write", "patch", "exec"].includes(tool);
|
|
16
|
+
}
|
|
17
|
+
/**
|
|
18
|
+
* Check if an agent is allowed to execute a tool.
|
|
19
|
+
*/
|
|
20
|
+
isToolAllowed(agentName, toolName) {
|
|
21
|
+
const agent = this.config.agents[agentName];
|
|
22
|
+
if (!agent)
|
|
23
|
+
return false;
|
|
24
|
+
return agent.tools.includes(toolName);
|
|
25
|
+
}
|
|
26
|
+
/**
|
|
27
|
+
* Check if an agent exists in the config.
|
|
28
|
+
*/
|
|
29
|
+
isAgentValid(agentName) {
|
|
30
|
+
return agentName in this.config.agents;
|
|
31
|
+
}
|
|
32
|
+
/**
|
|
33
|
+
* Get the target for a tool (gateway or sandbox).
|
|
34
|
+
*/
|
|
35
|
+
getToolTarget(toolName) {
|
|
36
|
+
const tool = this.config.tools[toolName];
|
|
37
|
+
return tool?.target;
|
|
38
|
+
}
|
|
39
|
+
}
|
|
40
|
+
//# sourceMappingURL=policy.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"policy.js","sourceRoot":"","sources":["../../src/gateway/policy.ts"],"names":[],"mappings":"AAEA;;;GAGG;AACH,MAAM,OAAO,YAAY;IACH;IAApB,YAAoB,MAAmB;QAAnB,WAAM,GAAN,MAAM,CAAa;IAAG,CAAC;IAE3C;;;OAGG;IACH,UAAU,CAAC,IAAY;QACrB,OAAO,CAAC,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;IAC3D,CAAC;IAED;;OAEG;IACH,aAAa,CAAC,SAAiB,EAAE,QAAgB;QAC/C,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAC5C,IAAI,CAAC,KAAK;YAAE,OAAO,KAAK,CAAC;QACzB,OAAO,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;IACxC,CAAC;IAED;;OAEG;IACH,YAAY,CAAC,SAAiB;QAC5B,OAAO,SAAS,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC;IACzC,CAAC;IAED;;OAEG;IACH,aAAa,CAAC,QAAgB;QAC5B,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QACzC,OAAO,IAAI,EAAE,MAAM,CAAC;IACtB,CAAC;CACF"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"policy.spec.d.ts","sourceRoot":"","sources":["../../src/gateway/policy.spec.ts"],"names":[],"mappings":""}
|