@aiden-ade/sandbox-agent 0.1.17 → 0.1.19

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.
Files changed (66) hide show
  1. package/dist/__tests__/cli-executable.test.d.ts +2 -0
  2. package/dist/__tests__/cli-executable.test.d.ts.map +1 -0
  3. package/dist/__tests__/cli-executable.test.js +83 -0
  4. package/dist/__tests__/cli-executable.test.js.map +1 -0
  5. package/dist/__tests__/cli-help.test.d.ts +2 -0
  6. package/dist/__tests__/cli-help.test.d.ts.map +1 -0
  7. package/dist/__tests__/cli-help.test.js +33 -0
  8. package/dist/__tests__/cli-help.test.js.map +1 -0
  9. package/dist/__tests__/daemon-socket.integration.test.d.ts +2 -0
  10. package/dist/__tests__/daemon-socket.integration.test.d.ts.map +1 -0
  11. package/dist/__tests__/daemon-socket.integration.test.js +152 -0
  12. package/dist/__tests__/daemon-socket.integration.test.js.map +1 -0
  13. package/dist/__tests__/daemon.test.d.ts +2 -0
  14. package/dist/__tests__/daemon.test.d.ts.map +1 -0
  15. package/dist/__tests__/daemon.test.js +978 -0
  16. package/dist/__tests__/daemon.test.js.map +1 -0
  17. package/dist/__tests__/generated-images.test.d.ts +2 -0
  18. package/dist/__tests__/generated-images.test.d.ts.map +1 -0
  19. package/dist/__tests__/generated-images.test.js +38 -0
  20. package/dist/__tests__/generated-images.test.js.map +1 -0
  21. package/dist/aiden-system-prompt.d.ts +6 -0
  22. package/dist/aiden-system-prompt.d.ts.map +1 -0
  23. package/dist/aiden-system-prompt.js +6 -0
  24. package/dist/aiden-system-prompt.js.map +1 -0
  25. package/dist/cli-executable.d.ts +65 -0
  26. package/dist/cli-executable.d.ts.map +1 -0
  27. package/dist/cli-executable.js +298 -0
  28. package/dist/cli-executable.js.map +1 -0
  29. package/dist/cli-help.d.ts +7 -0
  30. package/dist/cli-help.d.ts.map +1 -0
  31. package/dist/cli-help.js +45 -0
  32. package/dist/cli-help.js.map +1 -0
  33. package/dist/core-agent.d.ts +5 -1
  34. package/dist/core-agent.d.ts.map +1 -1
  35. package/dist/core-agent.js +101 -12
  36. package/dist/core-agent.js.map +1 -1
  37. package/dist/daemon.d.ts +46 -0
  38. package/dist/daemon.d.ts.map +1 -0
  39. package/dist/daemon.js +836 -0
  40. package/dist/daemon.js.map +1 -0
  41. package/dist/generated-images.d.ts +3 -0
  42. package/dist/generated-images.d.ts.map +1 -0
  43. package/dist/generated-images.js +123 -0
  44. package/dist/generated-images.js.map +1 -0
  45. package/dist/index.cjs +65 -143
  46. package/dist/index.js +94 -42
  47. package/dist/index.js.map +1 -1
  48. package/dist/sandbox.d.ts.map +1 -1
  49. package/dist/sandbox.js +18 -3
  50. package/dist/sandbox.js.map +1 -1
  51. package/dist/updater.d.ts.map +1 -1
  52. package/dist/updater.js +0 -4
  53. package/dist/updater.js.map +1 -1
  54. package/dist/version.d.ts +1 -1
  55. package/dist/version.d.ts.map +1 -1
  56. package/dist/version.js +1 -1
  57. package/dist/version.js.map +1 -1
  58. package/dist/web-presenter.d.ts +6 -2
  59. package/dist/web-presenter.d.ts.map +1 -1
  60. package/dist/web-presenter.js +20 -1
  61. package/dist/web-presenter.js.map +1 -1
  62. package/dist/ws-client.d.ts +35 -0
  63. package/dist/ws-client.d.ts.map +1 -1
  64. package/dist/ws-client.js +1 -0
  65. package/dist/ws-client.js.map +1 -1
  66. package/package.json +1 -1
@@ -0,0 +1,2 @@
1
+ export {};
2
+ //# sourceMappingURL=cli-executable.test.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"cli-executable.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/cli-executable.test.ts"],"names":[],"mappings":""}
@@ -0,0 +1,83 @@
1
+ import { chmodSync, mkdirSync, mkdtempSync, rmSync, writeFileSync } from "node:fs";
2
+ import { tmpdir } from "node:os";
3
+ import { join } from "node:path";
4
+ import { afterEach, describe, expect, it } from "vitest";
5
+ import { augmentCliPath, clearLoginShellEnvironmentCache, getDaemonCliEnvironment, resolveBackendRuntimeCommand, resolveCliExecutable, runCliVersionProbe, } from "../cli-executable.js";
6
+ function seedUnixExecutable(dir, name) {
7
+ const path = join(dir, name);
8
+ writeFileSync(path, "#!/bin/sh\n");
9
+ chmodSync(path, 0o755);
10
+ return path;
11
+ }
12
+ describe("cli-executable", () => {
13
+ let tempDir;
14
+ afterEach(() => {
15
+ clearLoginShellEnvironmentCache();
16
+ if (tempDir)
17
+ rmSync(tempDir, { recursive: true, force: true });
18
+ });
19
+ it("resolves CLIs from ~/.local/bin when PATH is sparse", () => {
20
+ tempDir = mkdtempSync(join(tmpdir(), "aiden-cli-exec-"));
21
+ const home = join(tempDir, "home");
22
+ const binDir = join(home, ".local", "bin");
23
+ mkdirSync(binDir, { recursive: true });
24
+ const codexPath = seedUnixExecutable(binDir, "codex");
25
+ const resolved = resolveCliExecutable("codex", {
26
+ HOME: home,
27
+ PATH: "/usr/bin:/bin:/usr/sbin:/sbin",
28
+ });
29
+ expect(resolved).toBe(codexPath);
30
+ });
31
+ it("getDaemonCliEnvironment includes user bin dirs from login shell or augmentation", () => {
32
+ const env = getDaemonCliEnvironment();
33
+ expect(env.PATH).toContain(".local/bin");
34
+ expect(env.HOME).toBeTruthy();
35
+ });
36
+ it("augments PATH with Homebrew and user bin dirs on macOS/linux", () => {
37
+ const env = augmentCliPath({
38
+ HOME: "/tmp/home",
39
+ PATH: "/usr/bin:/bin",
40
+ });
41
+ expect(env.PATH).toContain("/opt/homebrew/bin");
42
+ expect(env.PATH).toContain("/tmp/home/.local/bin");
43
+ });
44
+ it("probes Windows cmd shims through ComSpec", () => {
45
+ if (process.platform !== "win32")
46
+ return;
47
+ const tempDir = mkdtempSync(join(tmpdir(), "aiden-cli-cmd-"));
48
+ const shim = join(tempDir, "codex.cmd");
49
+ writeFileSync(shim, '@echo codex-cli 0.0.0-test\r\n');
50
+ const result = runCliVersionProbe(shim, {
51
+ ComSpec: process.env.ComSpec ?? "cmd.exe",
52
+ SystemRoot: process.env.SystemRoot ?? "C:\\Windows",
53
+ });
54
+ expect(result.status).toBe(0);
55
+ rmSync(tempDir, { recursive: true, force: true });
56
+ });
57
+ it("maps codex_app_server using explicit AIDEN_CODEX_PATH override", () => {
58
+ tempDir = mkdtempSync(join(tmpdir(), "aiden-cli-exec-"));
59
+ const binDir = join(tempDir, "bin");
60
+ mkdirSync(binDir, { recursive: true });
61
+ const codexPath = seedUnixExecutable(binDir, "codex");
62
+ const resolved = resolveBackendRuntimeCommand("codex_app_server", {
63
+ HOME: tempDir,
64
+ PATH: "/usr/bin:/bin",
65
+ AIDEN_CODEX_PATH: codexPath,
66
+ });
67
+ expect(resolved).toBe(codexPath);
68
+ });
69
+ it("maps kimi_cli to the kimi-cli executable when the legacy kimi shim is present", () => {
70
+ tempDir = mkdtempSync(join(tmpdir(), "aiden-kimi-cli-exec-"));
71
+ const home = join(tempDir, "home");
72
+ const binDir = join(home, ".local", "bin");
73
+ mkdirSync(binDir, { recursive: true });
74
+ seedUnixExecutable(binDir, "kimi");
75
+ const kimiCliPath = seedUnixExecutable(binDir, "kimi-cli");
76
+ const resolved = resolveBackendRuntimeCommand("kimi_cli", {
77
+ HOME: home,
78
+ PATH: "/usr/bin:/bin",
79
+ });
80
+ expect(resolved).toBe(kimiCliPath);
81
+ });
82
+ });
83
+ //# sourceMappingURL=cli-executable.test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"cli-executable.test.js","sourceRoot":"","sources":["../../src/__tests__/cli-executable.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,EAAE,aAAa,EAAE,MAAM,SAAS,CAAC;AACnF,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AACjC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AACzD,OAAO,EACL,cAAc,EACd,+BAA+B,EAC/B,uBAAuB,EACvB,4BAA4B,EAC5B,oBAAoB,EACpB,kBAAkB,GACnB,MAAM,sBAAsB,CAAC;AAE9B,SAAS,kBAAkB,CAAC,GAAW,EAAE,IAAY;IACnD,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;IAC7B,aAAa,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC;IACnC,SAAS,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;IACvB,OAAO,IAAI,CAAC;AACd,CAAC;AAED,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,IAAI,OAAe,CAAC;IAEpB,SAAS,CAAC,GAAG,EAAE;QACb,+BAA+B,EAAE,CAAC;QAClC,IAAI,OAAO;YAAE,MAAM,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACjE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,OAAO,GAAG,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,iBAAiB,CAAC,CAAC,CAAC;QACzD,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QACnC,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,EAAE,QAAQ,EAAE,KAAK,CAAC,CAAC;QAC3C,SAAS,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACvC,MAAM,SAAS,GAAG,kBAAkB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAEtD,MAAM,QAAQ,GAAG,oBAAoB,CAAC,OAAO,EAAE;YAC7C,IAAI,EAAE,IAAI;YACV,IAAI,EAAE,+BAA+B;SACtC,CAAC,CAAC;QAEH,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iFAAiF,EAAE,GAAG,EAAE;QACzF,MAAM,GAAG,GAAG,uBAAuB,EAAE,CAAC;QACtC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;QACzC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,UAAU,EAAE,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;QACtE,MAAM,GAAG,GAAG,cAAc,CAAC;YACzB,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,eAAe;SACtB,CAAC,CAAC;QACH,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;QAChD,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,sBAAsB,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,IAAI,OAAO,CAAC,QAAQ,KAAK,OAAO;YAAE,OAAO;QACzC,MAAM,OAAO,GAAG,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,gBAAgB,CAAC,CAAC,CAAC;QAC9D,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;QACxC,aAAa,CAAC,IAAI,EAAE,gCAAgC,CAAC,CAAC;QACtD,MAAM,MAAM,GAAG,kBAAkB,CAAC,IAAI,EAAE;YACtC,OAAO,EAAE,OAAO,CAAC,GAAG,CAAC,OAAO,IAAI,SAAS;YACzC,UAAU,EAAE,OAAO,CAAC,GAAG,CAAC,UAAU,IAAI,aAAa;SACpD,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gEAAgE,EAAE,GAAG,EAAE;QACxE,OAAO,GAAG,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,iBAAiB,CAAC,CAAC,CAAC;QACzD,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QACpC,SAAS,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACvC,MAAM,SAAS,GAAG,kBAAkB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAEtD,MAAM,QAAQ,GAAG,4BAA4B,CAAC,kBAAkB,EAAE;YAChE,IAAI,EAAE,OAAO;YACb,IAAI,EAAE,eAAe;YACrB,gBAAgB,EAAE,SAAS;SAC5B,CAAC,CAAC;QAEH,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+EAA+E,EAAE,GAAG,EAAE;QACvF,OAAO,GAAG,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,sBAAsB,CAAC,CAAC,CAAC;QAC9D,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QACnC,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,EAAE,QAAQ,EAAE,KAAK,CAAC,CAAC;QAC3C,SAAS,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACvC,kBAAkB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACnC,MAAM,WAAW,GAAG,kBAAkB,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;QAE3D,MAAM,QAAQ,GAAG,4BAA4B,CAAC,UAAU,EAAE;YACxD,IAAI,EAAE,IAAI;YACV,IAAI,EAAE,eAAe;SACtB,CAAC,CAAC;QAEH,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
@@ -0,0 +1,2 @@
1
+ export {};
2
+ //# sourceMappingURL=cli-help.test.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"cli-help.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/cli-help.test.ts"],"names":[],"mappings":""}
@@ -0,0 +1,33 @@
1
+ import { describe, expect, it } from "vitest";
2
+ import { HELP_TEXT, hasRunSessionEnv, resolveCommand } from "../cli-help.js";
3
+ describe("aiden-agent CLI help", () => {
4
+ it("shows help for bare human invocation without session env", () => {
5
+ expect(resolveCommand([], {})).toEqual({ command: "help", commandArgs: [] });
6
+ });
7
+ it("preserves bare sandbox invocation when session env is present", () => {
8
+ const env = {
9
+ AIDEN_WS_URL: "ws://ws.test",
10
+ AIDEN_SESSION_ID: "session-1",
11
+ AIDEN_TASK_ID: "task-1",
12
+ AIDEN_SESSION_TOKEN: "token",
13
+ AIDEN_TASK: "do work",
14
+ };
15
+ expect(hasRunSessionEnv(env)).toBe(true);
16
+ expect(resolveCommand([], env)).toEqual({ command: "run-session", commandArgs: [] });
17
+ });
18
+ it("supports explicit help aliases", () => {
19
+ expect(resolveCommand(["help"], {})).toEqual({ command: "help", commandArgs: [] });
20
+ expect(resolveCommand(["--help"], {})).toEqual({ command: "help", commandArgs: [] });
21
+ expect(resolveCommand(["-h"], {})).toEqual({ command: "help", commandArgs: [] });
22
+ expect(HELP_TEXT).toContain("aiden-agent login");
23
+ expect(HELP_TEXT).toContain("aiden-agent setup --setup-token <token>");
24
+ expect(HELP_TEXT).toContain("aiden-agent login --profile local");
25
+ });
26
+ it("passes through normal commands and arguments", () => {
27
+ expect(resolveCommand(["daemon", "--ws-url", "ws://ws.test"], {})).toEqual({
28
+ command: "daemon",
29
+ commandArgs: ["--ws-url", "ws://ws.test"],
30
+ });
31
+ });
32
+ });
33
+ //# sourceMappingURL=cli-help.test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"cli-help.test.js","sourceRoot":"","sources":["../../src/__tests__/cli-help.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,SAAS,EAAE,gBAAgB,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AAE7E,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;IACpC,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;QAClE,MAAM,CAAC,cAAc,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC,CAAC;IAC/E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;QACvE,MAAM,GAAG,GAAG;YACV,YAAY,EAAE,cAAc;YAC5B,gBAAgB,EAAE,WAAW;YAC7B,aAAa,EAAE,QAAQ;YACvB,mBAAmB,EAAE,OAAO;YAC5B,UAAU,EAAE,SAAS;SACtB,CAAC;QAEF,MAAM,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzC,MAAM,CAAC,cAAc,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,aAAa,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC,CAAC;IACvF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;QACxC,MAAM,CAAC,cAAc,CAAC,CAAC,MAAM,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC,CAAC;QACnF,MAAM,CAAC,cAAc,CAAC,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC,CAAC;QACrF,MAAM,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC,CAAC;QACjF,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;QACjD,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,yCAAyC,CAAC,CAAC;QACvE,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,mCAAmC,CAAC,CAAC;IACnE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,CAAC,cAAc,CAAC,CAAC,QAAQ,EAAE,UAAU,EAAE,cAAc,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC;YACzE,OAAO,EAAE,QAAQ;YACjB,WAAW,EAAE,CAAC,UAAU,EAAE,cAAc,CAAC;SAC1C,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
@@ -0,0 +1,2 @@
1
+ export {};
2
+ //# sourceMappingURL=daemon-socket.integration.test.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"daemon-socket.integration.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/daemon-socket.integration.test.ts"],"names":[],"mappings":""}
@@ -0,0 +1,152 @@
1
+ import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
2
+ import { chmodSync, mkdtempSync, mkdirSync, writeFileSync } from "node:fs";
3
+ import { tmpdir } from "node:os";
4
+ import { join } from "node:path";
5
+ import { createServer } from "node:http";
6
+ import { Server as SocketServer } from "socket.io";
7
+ import { once } from "node:events";
8
+ const mocks = vi.hoisted(() => ({
9
+ coreAgentConstructor: vi.fn(),
10
+ agentRun: vi.fn(),
11
+ agentKill: vi.fn(),
12
+ collectLocalAgentProviderLimits: vi.fn(),
13
+ }));
14
+ vi.mock("../core-agent.js", () => ({
15
+ CoreAgent: mocks.coreAgentConstructor,
16
+ }));
17
+ vi.mock("@aiden/shared/node/provider-limits", () => ({
18
+ collectLocalAgentProviderLimits: mocks.collectLocalAgentProviderLimits,
19
+ }));
20
+ function waitForEvent(socket, event) {
21
+ return new Promise((resolve) => {
22
+ socket.once(event, (payload) => resolve(payload));
23
+ });
24
+ }
25
+ function seedCodexBinary(baseDir) {
26
+ const binDir = join(baseDir, "bin");
27
+ mkdirSync(binDir, { recursive: true });
28
+ const codexPath = join(binDir, "codex");
29
+ writeFileSync(codexPath, "#!/bin/sh\n");
30
+ chmodSync(codexPath, 0o755);
31
+ return codexPath;
32
+ }
33
+ describe("aiden-agent daemon real websocket loop", () => {
34
+ let io;
35
+ let httpServer;
36
+ let tempDir;
37
+ let previousCodexPath;
38
+ beforeEach(() => {
39
+ vi.resetModules();
40
+ tempDir = mkdtempSync(join(tmpdir(), "aiden-agent-socket-test-"));
41
+ previousCodexPath = process.env.AIDEN_CODEX_PATH;
42
+ process.env.AIDEN_CODEX_PATH = seedCodexBinary(tempDir);
43
+ mocks.collectLocalAgentProviderLimits.mockResolvedValue({});
44
+ });
45
+ afterEach(async () => {
46
+ vi.clearAllMocks();
47
+ await new Promise((resolve) => {
48
+ if (!io) {
49
+ resolve();
50
+ return;
51
+ }
52
+ io.close(() => resolve());
53
+ });
54
+ await new Promise((resolve) => {
55
+ if (!httpServer?.listening) {
56
+ resolve();
57
+ return;
58
+ }
59
+ httpServer.close(() => resolve());
60
+ });
61
+ io = undefined;
62
+ httpServer = undefined;
63
+ if (previousCodexPath === undefined) {
64
+ delete process.env.AIDEN_CODEX_PATH;
65
+ }
66
+ else {
67
+ process.env.AIDEN_CODEX_PATH = previousCodexPath;
68
+ }
69
+ });
70
+ it("connects with runtime auth, sends hello, accepts execution, and emits agent events over Socket.IO", async () => {
71
+ mocks.agentRun.mockImplementation(async function fakeRun(config) {
72
+ const presenter = mocks.coreAgentConstructor.mock.calls[0]?.[0];
73
+ presenter.onStart(config);
74
+ presenter.onAssistantText("runtime response");
75
+ presenter.onComplete({ success: true, durationMs: 1, totalCostUsd: 0 });
76
+ });
77
+ mocks.coreAgentConstructor.mockImplementation(function MockCoreAgent() {
78
+ return {
79
+ run: mocks.agentRun,
80
+ kill: mocks.agentKill,
81
+ sendToolResponse: vi.fn(),
82
+ sendUserMessage: vi.fn(),
83
+ };
84
+ });
85
+ httpServer = createServer();
86
+ io = new SocketServer(httpServer, { cors: { origin: "*" } });
87
+ httpServer.listen(0, "127.0.0.1");
88
+ await once(httpServer, "listening");
89
+ const address = httpServer.address();
90
+ if (!address || typeof address === "string")
91
+ throw new Error("Socket.IO server did not bind to a TCP port");
92
+ const wsUrl = `ws://127.0.0.1:${address.port}`;
93
+ const connection = new Promise((resolve) => {
94
+ io.once("connection", (socket) => {
95
+ socket.once("runtime.hello", (hello) => {
96
+ resolve({ socket, hello });
97
+ });
98
+ });
99
+ });
100
+ const { startDaemon } = await import("../daemon.js");
101
+ const controller = await startDaemon([
102
+ "--runtime-id", "runtime-real-loop",
103
+ "--token", "aiden_runtime_real_loop",
104
+ "--ws-url", wsUrl,
105
+ "--local-api-port", "0",
106
+ ]);
107
+ try {
108
+ const { socket, hello } = await connection;
109
+ expect(socket.handshake.query).toMatchObject({
110
+ type: "runtime",
111
+ runtimeId: "runtime-real-loop",
112
+ hostKind: "daemon",
113
+ });
114
+ expect(socket.handshake.auth).toEqual({ token: "aiden_runtime_real_loop" });
115
+ expect(hello.capabilities.supportsFilesystem).toBe(true);
116
+ const accepted = waitForEvent(socket, "agent.accepted");
117
+ const agentEvents = [];
118
+ socket.on("agent.event", (payload) => agentEvents.push(payload));
119
+ socket.emit("agent.execute", {
120
+ conversationId: "conversation-real-loop",
121
+ runId: "run-real-loop",
122
+ content: "Run through the real websocket",
123
+ backendKind: "codex",
124
+ teamId: "team-real-loop",
125
+ });
126
+ await expect(accepted).resolves.toEqual({ runId: "run-real-loop" });
127
+ await vi.waitFor(() => {
128
+ expect(agentEvents).toEqual(expect.arrayContaining([
129
+ expect.objectContaining({
130
+ conversationId: "conversation-real-loop",
131
+ runId: "run-real-loop",
132
+ event: expect.objectContaining({ type: "text", text: "runtime response" }),
133
+ }),
134
+ expect.objectContaining({
135
+ conversationId: "conversation-real-loop",
136
+ runId: "run-real-loop",
137
+ event: expect.objectContaining({ type: "query_complete" }),
138
+ }),
139
+ ]));
140
+ });
141
+ expect(mocks.agentRun).toHaveBeenCalledWith(expect.objectContaining({
142
+ task: "Run through the real websocket",
143
+ conversationId: "conversation-real-loop",
144
+ teamId: "team-real-loop",
145
+ }));
146
+ }
147
+ finally {
148
+ await controller.stop();
149
+ }
150
+ }, 20_000);
151
+ });
152
+ //# sourceMappingURL=daemon-socket.integration.test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"daemon-socket.integration.test.js","sourceRoot":"","sources":["../../src/__tests__/daemon-socket.integration.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,aAAa,EAAE,MAAM,SAAS,CAAC;AAC3E,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AACjC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,YAAY,EAA6B,MAAM,WAAW,CAAC;AACpE,OAAO,EAAE,MAAM,IAAI,YAAY,EAA+B,MAAM,WAAW,CAAC;AAChF,OAAO,EAAE,IAAI,EAAE,MAAM,aAAa,CAAC;AAEnC,MAAM,KAAK,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC;IAC9B,oBAAoB,EAAE,EAAE,CAAC,EAAE,EAAE;IAC7B,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;IACjB,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE;IAClB,+BAA+B,EAAE,EAAE,CAAC,EAAE,EAAE;CACzC,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,kBAAkB,EAAE,GAAG,EAAE,CAAC,CAAC;IACjC,SAAS,EAAE,KAAK,CAAC,oBAAoB;CACtC,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,oCAAoC,EAAE,GAAG,EAAE,CAAC,CAAC;IACnD,+BAA+B,EAAE,KAAK,CAAC,+BAA+B;CACvE,CAAC,CAAC,CAAC;AAEJ,SAAS,YAAY,CAAI,MAAoB,EAAE,KAAa;IAC1D,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAC7B,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,OAAU,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,eAAe,CAAC,OAAe;IACtC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;IACpC,SAAS,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACvC,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACxC,aAAa,CAAC,SAAS,EAAE,aAAa,CAAC,CAAC;IACxC,SAAS,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;IAC5B,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,QAAQ,CAAC,wCAAwC,EAAE,GAAG,EAAE;IACtD,IAAI,EAA4B,CAAC;IACjC,IAAI,UAAkC,CAAC;IACvC,IAAI,OAAe,CAAC;IACpB,IAAI,iBAAqC,CAAC;IAE1C,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,YAAY,EAAE,CAAC;QAClB,OAAO,GAAG,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,0BAA0B,CAAC,CAAC,CAAC;QAClE,iBAAiB,GAAG,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC;QACjD,OAAO,CAAC,GAAG,CAAC,gBAAgB,GAAG,eAAe,CAAC,OAAO,CAAC,CAAC;QACxD,KAAK,CAAC,+BAA+B,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE;YAClC,IAAI,CAAC,EAAE,EAAE,CAAC;gBACR,OAAO,EAAE,CAAC;gBACV,OAAO;YACT,CAAC;YACD,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC;QAC5B,CAAC,CAAC,CAAC;QACH,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE;YAClC,IAAI,CAAC,UAAU,EAAE,SAAS,EAAE,CAAC;gBAC3B,OAAO,EAAE,CAAC;gBACV,OAAO;YACT,CAAC;YACD,UAAU,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;QACH,EAAE,GAAG,SAAS,CAAC;QACf,UAAU,GAAG,SAAS,CAAC;QACvB,IAAI,iBAAiB,KAAK,SAAS,EAAE,CAAC;YACpC,OAAO,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC;QACtC,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,GAAG,CAAC,gBAAgB,GAAG,iBAAiB,CAAC;QACnD,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CACA,mGAAmG,EACnG,KAAK,IAAI,EAAE;QACX,KAAK,CAAC,QAAQ,CAAC,kBAAkB,CAAC,KAAK,UAAU,OAAO,CAAgB,MAAwB;YAC9F,MAAM,SAAS,GAAG,KAAK,CAAC,oBAAoB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YAChE,SAAS,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAC1B,SAAS,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC;YAC9C,SAAS,CAAC,UAAU,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC,CAAC;QAC1E,CAAC,CAAC,CAAC;QACH,KAAK,CAAC,oBAAoB,CAAC,kBAAkB,CAAC,SAAS,aAAa;YAClE,OAAO;gBACL,GAAG,EAAE,KAAK,CAAC,QAAQ;gBACnB,IAAI,EAAE,KAAK,CAAC,SAAS;gBACrB,gBAAgB,EAAE,EAAE,CAAC,EAAE,EAAE;gBACzB,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;aACzB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,UAAU,GAAG,YAAY,EAAE,CAAC;QAC5B,EAAE,GAAG,IAAI,YAAY,CAAC,UAAU,EAAE,EAAE,IAAI,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;QAC7D,UAAU,CAAC,MAAM,CAAC,CAAC,EAAE,WAAW,CAAC,CAAC;QAClC,MAAM,IAAI,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC;QACpC,MAAM,OAAO,GAAG,UAAU,CAAC,OAAO,EAAE,CAAC;QACrC,IAAI,CAAC,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ;YAAE,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC;QAC5G,MAAM,KAAK,GAAG,kBAAkB,OAAO,CAAC,IAAI,EAAE,CAAC;QAE/C,MAAM,UAAU,GAAG,IAAI,OAAO,CAG3B,CAAC,OAAO,EAAE,EAAE;YACb,EAAG,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC,MAAM,EAAE,EAAE;gBAChC,MAAM,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC,KAAK,EAAE,EAAE;oBACrC,OAAO,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;gBAC7B,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,EAAE,WAAW,EAAE,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC,CAAC;QACrD,MAAM,UAAU,GAAG,MAAM,WAAW,CAAC;YACnC,cAAc,EAAE,mBAAmB;YACnC,SAAS,EAAE,yBAAyB;YACpC,UAAU,EAAE,KAAK;YACjB,kBAAkB,EAAE,GAAG;SACxB,CAAC,CAAC;QAEH,IAAI,CAAC;YACH,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,UAAU,CAAC;YAC3C,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,aAAa,CAAC;gBAC3C,IAAI,EAAE,SAAS;gBACf,SAAS,EAAE,mBAAmB;gBAC9B,QAAQ,EAAE,QAAQ;aACnB,CAAC,CAAC;YACH,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,yBAAyB,EAAE,CAAC,CAAC;YAC5E,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAEzD,MAAM,QAAQ,GAAG,YAAY,CAAoB,MAAM,EAAE,gBAAgB,CAAC,CAAC;YAC3E,MAAM,WAAW,GAAc,EAAE,CAAC;YAClC,MAAM,CAAC,EAAE,CAAC,aAAa,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;YACjE,MAAM,CAAC,IAAI,CAAC,eAAe,EAAE;gBAC3B,cAAc,EAAE,wBAAwB;gBACxC,KAAK,EAAE,eAAe;gBACtB,OAAO,EAAE,gCAAgC;gBACzC,WAAW,EAAE,OAAO;gBACpB,MAAM,EAAE,gBAAgB;aACzB,CAAC,CAAC;YAEH,MAAM,MAAM,CAAC,QAAQ,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,eAAe,EAAE,CAAC,CAAC;YACpE,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;gBACpB,MAAM,CAAC,WAAW,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,eAAe,CAAC;oBACjD,MAAM,CAAC,gBAAgB,CAAC;wBACtB,cAAc,EAAE,wBAAwB;wBACxC,KAAK,EAAE,eAAe;wBACtB,KAAK,EAAE,MAAM,CAAC,gBAAgB,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,kBAAkB,EAAE,CAAC;qBAC3E,CAAC;oBACF,MAAM,CAAC,gBAAgB,CAAC;wBACtB,cAAc,EAAE,wBAAwB;wBACxC,KAAK,EAAE,eAAe;wBACtB,KAAK,EAAE,MAAM,CAAC,gBAAgB,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,CAAC;qBAC3D,CAAC;iBACH,CAAC,CAAC,CAAC;YACN,CAAC,CAAC,CAAC;YACH,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,gBAAgB,CAAC;gBAClE,IAAI,EAAE,gCAAgC;gBACtC,cAAc,EAAE,wBAAwB;gBACxC,MAAM,EAAE,gBAAgB;aACzB,CAAC,CAAC,CAAC;QACN,CAAC;gBAAS,CAAC;YACT,MAAM,UAAU,CAAC,IAAI,EAAE,CAAC;QAC1B,CAAC;IACH,CAAC,EACC,MAAM,CACP,CAAC;AACJ,CAAC,CAAC,CAAC"}
@@ -0,0 +1,2 @@
1
+ export {};
2
+ //# sourceMappingURL=daemon.test.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"daemon.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/daemon.test.ts"],"names":[],"mappings":""}