@masons/agent-network 0.5.28 → 0.5.30

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 (72) hide show
  1. package/dist/broker/broker-daemon.d.ts +34 -0
  2. package/dist/broker/broker-daemon.d.ts.map +1 -0
  3. package/dist/broker/broker-daemon.js +471 -0
  4. package/dist/broker/closed-endpoint-lookup.d.ts +25 -0
  5. package/dist/broker/closed-endpoint-lookup.d.ts.map +1 -0
  6. package/dist/broker/closed-endpoint-lookup.js +59 -0
  7. package/dist/broker/connector-ws.d.ts +47 -0
  8. package/dist/broker/connector-ws.d.ts.map +1 -0
  9. package/dist/broker/connector-ws.js +60 -0
  10. package/dist/broker/control-event-dispatcher.d.ts +20 -0
  11. package/dist/broker/control-event-dispatcher.d.ts.map +1 -0
  12. package/dist/broker/control-event-dispatcher.js +45 -0
  13. package/dist/broker/control-event-types.d.ts +30 -0
  14. package/dist/broker/control-event-types.d.ts.map +1 -0
  15. package/dist/broker/control-event-types.js +1 -0
  16. package/dist/broker/correlation-ring.d.ts +10 -0
  17. package/dist/broker/correlation-ring.d.ts.map +1 -0
  18. package/dist/broker/correlation-ring.js +32 -0
  19. package/dist/broker/discovery-file.d.ts +12 -0
  20. package/dist/broker/discovery-file.d.ts.map +1 -0
  21. package/dist/broker/discovery-file.js +77 -0
  22. package/dist/broker/endpoint-registry.d.ts +53 -0
  23. package/dist/broker/endpoint-registry.d.ts.map +1 -0
  24. package/dist/broker/endpoint-registry.js +83 -0
  25. package/dist/broker/endpoint-state-machine.d.ts +40 -0
  26. package/dist/broker/endpoint-state-machine.d.ts.map +1 -0
  27. package/dist/broker/endpoint-state-machine.js +92 -0
  28. package/dist/broker/entry.d.ts +5 -0
  29. package/dist/broker/entry.d.ts.map +1 -0
  30. package/dist/broker/entry.js +123 -0
  31. package/dist/broker/grace-timer.d.ts +9 -0
  32. package/dist/broker/grace-timer.d.ts.map +1 -0
  33. package/dist/broker/grace-timer.js +34 -0
  34. package/dist/broker/ipc-server.d.ts +72 -0
  35. package/dist/broker/ipc-server.d.ts.map +1 -0
  36. package/dist/broker/ipc-server.js +243 -0
  37. package/dist/broker/logger.d.ts +10 -0
  38. package/dist/broker/logger.d.ts.map +1 -0
  39. package/dist/broker/logger.js +34 -0
  40. package/dist/broker/paths.d.ts +11 -0
  41. package/dist/broker/paths.d.ts.map +1 -0
  42. package/dist/broker/paths.js +30 -0
  43. package/dist/broker/plugin-liveness.d.ts +2 -0
  44. package/dist/broker/plugin-liveness.d.ts.map +1 -0
  45. package/dist/broker/plugin-liveness.js +15 -0
  46. package/dist/broker/reconnecting-buffer.d.ts +23 -0
  47. package/dist/broker/reconnecting-buffer.d.ts.map +1 -0
  48. package/dist/broker/reconnecting-buffer.js +107 -0
  49. package/dist/broker/routing-table.d.ts +22 -0
  50. package/dist/broker/routing-table.d.ts.map +1 -0
  51. package/dist/broker/routing-table.js +35 -0
  52. package/dist/broker/runtime-endpoint-port.d.ts +8 -0
  53. package/dist/broker/runtime-endpoint-port.d.ts.map +1 -0
  54. package/dist/broker/runtime-endpoint-port.js +1 -0
  55. package/dist/broker/undispatched-inbox.d.ts +26 -0
  56. package/dist/broker/undispatched-inbox.d.ts.map +1 -0
  57. package/dist/broker/undispatched-inbox.js +53 -0
  58. package/dist/broker/version-handshake.d.ts +30 -0
  59. package/dist/broker/version-handshake.d.ts.map +1 -0
  60. package/dist/broker/version-handshake.js +47 -0
  61. package/dist/broker-client/broker-client.d.ts +64 -0
  62. package/dist/broker-client/broker-client.d.ts.map +1 -0
  63. package/dist/broker-client/broker-client.js +164 -0
  64. package/dist/broker-client/lazy-spawn.d.ts +18 -0
  65. package/dist/broker-client/lazy-spawn.d.ts.map +1 -0
  66. package/dist/broker-client/lazy-spawn.js +61 -0
  67. package/dist/runtime-endpoint-client.d.ts +9 -2
  68. package/dist/runtime-endpoint-client.d.ts.map +1 -1
  69. package/dist/runtime-endpoint-client.js +15 -2
  70. package/dist/version.d.ts +1 -1
  71. package/dist/version.js +1 -1
  72. package/package.json +11 -2
@@ -0,0 +1,107 @@
1
+ import { closeSync, fdatasyncSync, mkdirSync, openSync, readFileSync, unlinkSync, writeSync, } from "node:fs";
2
+ import { join } from "node:path";
3
+ class FileBackedBuffer {
4
+ path;
5
+ cached = null;
6
+ constructor(path) {
7
+ this.path = path;
8
+ }
9
+ append(msg) {
10
+ const line = `${JSON.stringify(msg)}\n`;
11
+ try {
12
+ mkdirSync(dirname(this.path), { recursive: true, mode: 0o700 });
13
+ }
14
+ catch {
15
+ }
16
+ const fd = openSync(this.path, "a", 0o600);
17
+ try {
18
+ writeSync(fd, line);
19
+ fdatasyncSync(fd);
20
+ }
21
+ finally {
22
+ closeSync(fd);
23
+ }
24
+ this.cached = null;
25
+ }
26
+ read() {
27
+ if (this.cached !== null)
28
+ return [...this.cached];
29
+ let raw;
30
+ try {
31
+ raw = readFileSync(this.path, "utf8");
32
+ }
33
+ catch (err) {
34
+ if (isNodeErr(err, "ENOENT")) {
35
+ this.cached = [];
36
+ return [];
37
+ }
38
+ throw err;
39
+ }
40
+ const lines = raw.split("\n").filter((l) => l.length > 0);
41
+ const parsed = [];
42
+ for (const line of lines) {
43
+ try {
44
+ parsed.push(JSON.parse(line));
45
+ }
46
+ catch {
47
+ }
48
+ }
49
+ this.cached = parsed;
50
+ return [...parsed];
51
+ }
52
+ drain() {
53
+ const out = this.read();
54
+ this.delete();
55
+ return out;
56
+ }
57
+ delete() {
58
+ try {
59
+ unlinkSync(this.path);
60
+ }
61
+ catch (err) {
62
+ if (isNodeErr(err, "ENOENT")) {
63
+ }
64
+ else {
65
+ throw err;
66
+ }
67
+ }
68
+ this.cached = [];
69
+ }
70
+ size() {
71
+ if (this.cached !== null)
72
+ return this.cached.length;
73
+ return this.read().length;
74
+ }
75
+ }
76
+ export function createReconnectingBufferManager(buffersDir) {
77
+ const cache = new Map();
78
+ return {
79
+ forEndpoint(endpoint_id) {
80
+ let buf = cache.get(endpoint_id);
81
+ if (!buf) {
82
+ buf = new FileBackedBuffer(join(buffersDir, `${sanitizeId(endpoint_id)}.ndjson`));
83
+ cache.set(endpoint_id, buf);
84
+ }
85
+ return buf;
86
+ },
87
+ cleanup(endpoint_id) {
88
+ const buf = cache.get(endpoint_id);
89
+ if (buf)
90
+ buf.delete();
91
+ cache.delete(endpoint_id);
92
+ },
93
+ };
94
+ }
95
+ function sanitizeId(id) {
96
+ return id.replace(/[^a-zA-Z0-9._-]/g, "_");
97
+ }
98
+ function dirname(p) {
99
+ const idx = p.lastIndexOf("/");
100
+ return idx >= 0 ? p.slice(0, idx) : ".";
101
+ }
102
+ function isNodeErr(err, code) {
103
+ return (typeof err === "object" &&
104
+ err !== null &&
105
+ "code" in err &&
106
+ err.code === code);
107
+ }
@@ -0,0 +1,22 @@
1
+ import type { BrokerEndpointEntry, EndpointRegistry } from "./endpoint-registry.js";
2
+ export interface InboundMessageMeta {
3
+ target_endpoint_id?: string;
4
+ correlation_id?: string;
5
+ }
6
+ export type RouteResult = {
7
+ kind: "endpoint";
8
+ entry: BrokerEndpointEntry;
9
+ via: "target_endpoint_id" | "correlation_id";
10
+ } | {
11
+ kind: "no_match";
12
+ };
13
+ export interface RoutingTableOptions {
14
+ onCollision?: (correlation_id: string, count: number) => void;
15
+ }
16
+ export declare class RoutingTable {
17
+ private readonly registry;
18
+ private readonly opts;
19
+ constructor(registry: EndpointRegistry, opts?: RoutingTableOptions);
20
+ route(meta: InboundMessageMeta): RouteResult;
21
+ }
22
+ //# sourceMappingURL=routing-table.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"routing-table.d.ts","sourceRoot":"","sources":["../../src/broker/routing-table.ts"],"names":[],"mappings":"AAqBA,OAAO,KAAK,EACV,mBAAmB,EACnB,gBAAgB,EACjB,MAAM,wBAAwB,CAAC;AAEhC,MAAM,WAAW,kBAAkB;IACjC,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,cAAc,CAAC,EAAE,MAAM,CAAC;CACzB;AAED,MAAM,MAAM,WAAW,GACnB;IACE,IAAI,EAAE,UAAU,CAAC;IACjB,KAAK,EAAE,mBAAmB,CAAC;IAC3B,GAAG,EAAE,oBAAoB,GAAG,gBAAgB,CAAC;CAC9C,GACD;IAAE,IAAI,EAAE,UAAU,CAAA;CAAE,CAAC;AAEzB,MAAM,WAAW,mBAAmB;IAElC,WAAW,CAAC,EAAE,CAAC,cAAc,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;CAC/D;AAED,qBAAa,YAAY;IAErB,OAAO,CAAC,QAAQ,CAAC,QAAQ;IACzB,OAAO,CAAC,QAAQ,CAAC,IAAI;gBADJ,QAAQ,EAAE,gBAAgB,EAC1B,IAAI,GAAE,mBAAwB;IASjD,KAAK,CAAC,IAAI,EAAE,kBAAkB,GAAG,WAAW;CAmC7C"}
@@ -0,0 +1,35 @@
1
+ export class RoutingTable {
2
+ registry;
3
+ opts;
4
+ constructor(registry, opts = {}) {
5
+ this.registry = registry;
6
+ this.opts = opts;
7
+ }
8
+ route(meta) {
9
+ if (meta.target_endpoint_id) {
10
+ const entry = this.registry.get(meta.target_endpoint_id);
11
+ if (entry) {
12
+ return { kind: "endpoint", entry, via: "target_endpoint_id" };
13
+ }
14
+ }
15
+ if (meta.correlation_id) {
16
+ let matched;
17
+ let collisionCount = 0;
18
+ for (const entry of this.registry.list()) {
19
+ if (entry.correlation_ring.has(meta.correlation_id)) {
20
+ collisionCount++;
21
+ if (!matched) {
22
+ matched = entry;
23
+ }
24
+ }
25
+ }
26
+ if (collisionCount > 1 && this.opts.onCollision) {
27
+ this.opts.onCollision(meta.correlation_id, collisionCount);
28
+ }
29
+ if (matched) {
30
+ return { kind: "endpoint", entry: matched, via: "correlation_id" };
31
+ }
32
+ }
33
+ return { kind: "no_match" };
34
+ }
35
+ }
@@ -0,0 +1,8 @@
1
+ import type { RegisterRuntimeEndpointParams, RegisterRuntimeEndpointResponse, TransitionRuntimeEndpointStateParams } from "../runtime-endpoint-client.js";
2
+ export interface RuntimeEndpointPort {
3
+ register(params: RegisterRuntimeEndpointParams): Promise<RegisterRuntimeEndpointResponse>;
4
+ heartbeat(endpointId: string): Promise<void>;
5
+ unregister(endpointId: string, asNodeId?: string): Promise<void>;
6
+ transitionState(endpointId: string, params: TransitionRuntimeEndpointStateParams): Promise<void>;
7
+ }
8
+ //# sourceMappingURL=runtime-endpoint-port.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"runtime-endpoint-port.d.ts","sourceRoot":"","sources":["../../src/broker/runtime-endpoint-port.ts"],"names":[],"mappings":"AAoBA,OAAO,KAAK,EACV,6BAA6B,EAC7B,+BAA+B,EAC/B,oCAAoC,EACrC,MAAM,+BAA+B,CAAC;AAEvC,MAAM,WAAW,mBAAmB;IAClC,QAAQ,CACN,MAAM,EAAE,6BAA6B,GACpC,OAAO,CAAC,+BAA+B,CAAC,CAAC;IAC5C,SAAS,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAO7C,UAAU,CAAC,UAAU,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAQjE,eAAe,CACb,UAAU,EAAE,MAAM,EAClB,MAAM,EAAE,oCAAoC,GAC3C,OAAO,CAAC,IAAI,CAAC,CAAC;CAClB"}
@@ -0,0 +1 @@
1
+ export {};
@@ -0,0 +1,26 @@
1
+ export declare const DEFAULT_INBOX_CAPACITY = 256;
2
+ export interface UndispatchedMessage {
3
+ id: string;
4
+ arrived_at: string;
5
+ sender_address: string;
6
+ content: string;
7
+ content_type: string;
8
+ metadata: Record<string, unknown>;
9
+ reason: "unaddressed" | "closed_session_continuation";
10
+ original_target_endpoint_id?: string;
11
+ original_correlation_id?: string;
12
+ }
13
+ export interface UndispatchedInbox {
14
+ tryAdd(msg: UndispatchedMessage): boolean;
15
+ list(): UndispatchedMessage[];
16
+ get(id: string): UndispatchedMessage | undefined;
17
+ take(id: string): UndispatchedMessage | undefined;
18
+ size(): number;
19
+ capacity(): number;
20
+ isUnderPressure(): boolean;
21
+ }
22
+ export interface UndispatchedInboxOptions {
23
+ capacity?: number;
24
+ }
25
+ export declare function createUndispatchedInbox(opts?: UndispatchedInboxOptions): UndispatchedInbox;
26
+ //# sourceMappingURL=undispatched-inbox.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"undispatched-inbox.d.ts","sourceRoot":"","sources":["../../src/broker/undispatched-inbox.ts"],"names":[],"mappings":"AA2BA,eAAO,MAAM,sBAAsB,MAAM,CAAC;AAE1C,MAAM,WAAW,mBAAmB;IAClC,EAAE,EAAE,MAAM,CAAC;IACX,UAAU,EAAE,MAAM,CAAC;IACnB,cAAc,EAAE,MAAM,CAAC;IACvB,OAAO,EAAE,MAAM,CAAC;IAChB,YAAY,EAAE,MAAM,CAAC;IACrB,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAMlC,MAAM,EAAE,aAAa,GAAG,6BAA6B,CAAC;IAEtD,2BAA2B,CAAC,EAAE,MAAM,CAAC;IAErC,uBAAuB,CAAC,EAAE,MAAM,CAAC;CAClC;AAED,MAAM,WAAW,iBAAiB;IAMhC,MAAM,CAAC,GAAG,EAAE,mBAAmB,GAAG,OAAO,CAAC;IAE1C,IAAI,IAAI,mBAAmB,EAAE,CAAC;IAE9B,GAAG,CAAC,EAAE,EAAE,MAAM,GAAG,mBAAmB,GAAG,SAAS,CAAC;IAKjD,IAAI,CAAC,EAAE,EAAE,MAAM,GAAG,mBAAmB,GAAG,SAAS,CAAC;IAElD,IAAI,IAAI,MAAM,CAAC;IAEf,QAAQ,IAAI,MAAM,CAAC;IAEnB,eAAe,IAAI,OAAO,CAAC;CAC5B;AAED,MAAM,WAAW,wBAAwB;IACvC,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED,wBAAgB,uBAAuB,CACrC,IAAI,GAAE,wBAA6B,GAClC,iBAAiB,CAkDnB"}
@@ -0,0 +1,53 @@
1
+ export const DEFAULT_INBOX_CAPACITY = 256;
2
+ export function createUndispatchedInbox(opts = {}) {
3
+ const cap = opts.capacity ?? DEFAULT_INBOX_CAPACITY;
4
+ if (cap <= 0)
5
+ throw new Error(`capacity must be positive: ${cap}`);
6
+ const byId = new Map();
7
+ const order = [];
8
+ return {
9
+ tryAdd(msg) {
10
+ if (byId.has(msg.id)) {
11
+ byId.set(msg.id, msg);
12
+ return true;
13
+ }
14
+ if (byId.size >= cap) {
15
+ return false;
16
+ }
17
+ byId.set(msg.id, msg);
18
+ order.push(msg.id);
19
+ return true;
20
+ },
21
+ list() {
22
+ const out = [];
23
+ for (const id of order) {
24
+ const m = byId.get(id);
25
+ if (m)
26
+ out.push(m);
27
+ }
28
+ return out;
29
+ },
30
+ get(id) {
31
+ return byId.get(id);
32
+ },
33
+ take(id) {
34
+ const m = byId.get(id);
35
+ if (!m)
36
+ return undefined;
37
+ byId.delete(id);
38
+ const idx = order.indexOf(id);
39
+ if (idx >= 0)
40
+ order.splice(idx, 1);
41
+ return m;
42
+ },
43
+ size() {
44
+ return byId.size;
45
+ },
46
+ capacity() {
47
+ return cap;
48
+ },
49
+ isUnderPressure() {
50
+ return byId.size >= Math.floor(cap * 0.9);
51
+ },
52
+ };
53
+ }
@@ -0,0 +1,30 @@
1
+ export declare const IPC_PROTOCOL_VERSION = "1.0";
2
+ export interface IpcCapabilities {
3
+ endpoint_metadata_v1?: boolean;
4
+ remote_spawn_v1?: boolean;
5
+ undispatched_inbox_v1?: boolean;
6
+ }
7
+ export declare const SERVER_CAPABILITIES: Readonly<IpcCapabilities>;
8
+ export interface InitializeRequestBody {
9
+ client_protocol_version: string;
10
+ client_kind: string;
11
+ client_version: string;
12
+ capabilities?: IpcCapabilities;
13
+ }
14
+ export interface InitializeResponseBody {
15
+ server_protocol_version: string;
16
+ server_capabilities: IpcCapabilities;
17
+ session_id: string;
18
+ }
19
+ export interface InitializeErrorBody {
20
+ error: "protocol_version_mismatch" | "bad_request";
21
+ message: string;
22
+ }
23
+ export declare function handleInitialize(req: InitializeRequestBody): {
24
+ status: 200;
25
+ body: InitializeResponseBody;
26
+ } | {
27
+ status: 400;
28
+ body: InitializeErrorBody;
29
+ };
30
+ //# sourceMappingURL=version-handshake.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"version-handshake.d.ts","sourceRoot":"","sources":["../../src/broker/version-handshake.ts"],"names":[],"mappings":"AAoBA,eAAO,MAAM,oBAAoB,QAAQ,CAAC;AAM1C,MAAM,WAAW,eAAe;IAE9B,oBAAoB,CAAC,EAAE,OAAO,CAAC;IAE/B,eAAe,CAAC,EAAE,OAAO,CAAC;IAG1B,qBAAqB,CAAC,EAAE,OAAO,CAAC;CACjC;AAGD,eAAO,MAAM,mBAAmB,EAAE,QAAQ,CAAC,eAAe,CAIxD,CAAC;AAEH,MAAM,WAAW,qBAAqB;IAEpC,uBAAuB,EAAE,MAAM,CAAC;IAEhC,WAAW,EAAE,MAAM,CAAC;IACpB,cAAc,EAAE,MAAM,CAAC;IACvB,YAAY,CAAC,EAAE,eAAe,CAAC;CAChC;AAED,MAAM,WAAW,sBAAsB;IACrC,uBAAuB,EAAE,MAAM,CAAC;IAChC,mBAAmB,EAAE,eAAe,CAAC;IAErC,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,mBAAmB;IAClC,KAAK,EAAE,2BAA2B,GAAG,aAAa,CAAC;IACnD,OAAO,EAAE,MAAM,CAAC;CACjB;AAMD,wBAAgB,gBAAgB,CAC9B,GAAG,EAAE,qBAAqB,GAExB;IAAE,MAAM,EAAE,GAAG,CAAC;IAAC,IAAI,EAAE,sBAAsB,CAAA;CAAE,GAC7C;IAAE,MAAM,EAAE,GAAG,CAAC;IAAC,IAAI,EAAE,mBAAmB,CAAA;CAAE,CA0C7C"}
@@ -0,0 +1,47 @@
1
+ import { randomUUID } from "node:crypto";
2
+ export const IPC_PROTOCOL_VERSION = "1.0";
3
+ export const SERVER_CAPABILITIES = Object.freeze({
4
+ endpoint_metadata_v1: true,
5
+ remote_spawn_v1: false,
6
+ undispatched_inbox_v1: false,
7
+ });
8
+ export function handleInitialize(req) {
9
+ if (typeof req.client_protocol_version !== "string") {
10
+ return {
11
+ status: 400,
12
+ body: {
13
+ error: "bad_request",
14
+ message: "client_protocol_version is required and must be a string",
15
+ },
16
+ };
17
+ }
18
+ if (typeof req.client_kind !== "string" || !req.client_kind) {
19
+ return {
20
+ status: 400,
21
+ body: {
22
+ error: "bad_request",
23
+ message: "client_kind is required and must be a non-empty string",
24
+ },
25
+ };
26
+ }
27
+ const clientMajor = req.client_protocol_version.split(".")[0];
28
+ const serverMajor = IPC_PROTOCOL_VERSION.split(".")[0];
29
+ if (clientMajor !== serverMajor) {
30
+ return {
31
+ status: 400,
32
+ body: {
33
+ error: "protocol_version_mismatch",
34
+ message: `broker IPC protocol major ${serverMajor} does not match client ` +
35
+ `protocol major ${clientMajor}`,
36
+ },
37
+ };
38
+ }
39
+ return {
40
+ status: 200,
41
+ body: {
42
+ server_protocol_version: IPC_PROTOCOL_VERSION,
43
+ server_capabilities: { ...SERVER_CAPABILITIES },
44
+ session_id: randomUUID(),
45
+ },
46
+ };
47
+ }
@@ -0,0 +1,64 @@
1
+ import { EventEmitter } from "node:events";
2
+ import { type InitializeResponseBody } from "../broker/version-handshake.js";
3
+ export interface BrokerClientDiscoverOptions {
4
+ accountId: string;
5
+ runtimeToken: string;
6
+ userDataDir?: string;
7
+ env?: Record<string, string>;
8
+ timeoutMs?: number;
9
+ brokerEntry?: string;
10
+ nodeBinary?: string;
11
+ }
12
+ export interface BrokerClientConnectOptions {
13
+ pluginPid: number;
14
+ clientKind: string;
15
+ clientVersion: string;
16
+ }
17
+ export interface RegisterEndpointArgs {
18
+ agentId: string;
19
+ kind: string;
20
+ sessionName?: string;
21
+ workspace?: string;
22
+ trackingRef?: string;
23
+ taskHint?: string;
24
+ }
25
+ export interface RegisterEndpointOutcome {
26
+ endpoint_id: string;
27
+ }
28
+ export interface BrokerInboundMessage {
29
+ from: string;
30
+ content: string;
31
+ contentType: string;
32
+ metadata?: Record<string, unknown>;
33
+ }
34
+ type Events = {
35
+ message_received: (msg: BrokerInboundMessage) => void;
36
+ disconnected: () => void;
37
+ error: (err: Error) => void;
38
+ };
39
+ export declare class BrokerClient extends EventEmitter {
40
+ private readonly handle;
41
+ private ws;
42
+ private pluginPid;
43
+ private _sessionId;
44
+ private constructor();
45
+ static discover(opts: BrokerClientDiscoverOptions): Promise<BrokerClient>;
46
+ get spawnedNew(): boolean;
47
+ get ipcUrl(): string;
48
+ get sessionId(): string | null;
49
+ connect(opts: BrokerClientConnectOptions): Promise<InitializeResponseBody>;
50
+ registerEndpoint(args: RegisterEndpointArgs): Promise<RegisterEndpointOutcome>;
51
+ send(endpointId: string, to: string, content: string, contentType?: string, metadata?: Record<string, unknown>): Promise<{
52
+ messageId: string;
53
+ status: string;
54
+ }>;
55
+ close(endpointId?: string): Promise<void>;
56
+ on<K extends keyof Events>(event: K, listener: Events[K]): this;
57
+ on(event: string | symbol, listener: (...args: unknown[]) => void): this;
58
+ emit<K extends keyof Events>(event: K, ...args: Parameters<Events[K]>): boolean;
59
+ emit(event: string | symbol, ...args: unknown[]): boolean;
60
+ private httpJson;
61
+ private handleFrame;
62
+ }
63
+ export {};
64
+ //# sourceMappingURL=broker-client.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"broker-client.d.ts","sourceRoot":"","sources":["../../src/broker-client/broker-client.ts"],"names":[],"mappings":"AAiBA,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAG3C,OAAO,EACL,KAAK,sBAAsB,EAE5B,MAAM,gCAAgC,CAAC;AAGxC,MAAM,WAAW,2BAA2B;IAC1C,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,EAAE,MAAM,CAAC;IAErB,WAAW,CAAC,EAAE,MAAM,CAAC;IAErB,GAAG,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC7B,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,0BAA0B;IACzC,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;IACnB,aAAa,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,oBAAoB;IACnC,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,uBAAuB;IACtC,WAAW,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,oBAAoB;IACnC,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,WAAW,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC;AAED,KAAK,MAAM,GAAG;IACZ,gBAAgB,EAAE,CAAC,GAAG,EAAE,oBAAoB,KAAK,IAAI,CAAC;IACtD,YAAY,EAAE,MAAM,IAAI,CAAC;IACzB,KAAK,EAAE,CAAC,GAAG,EAAE,KAAK,KAAK,IAAI,CAAC;CAC7B,CAAC;AAEF,qBAAa,YAAa,SAAQ,YAAY;IAC5C,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAkB;IACzC,OAAO,CAAC,EAAE,CAA0B;IACpC,OAAO,CAAC,SAAS,CAAuB;IACxC,OAAO,CAAC,UAAU,CAAuB;IAEzC,OAAO;WAUM,QAAQ,CACnB,IAAI,EAAE,2BAA2B,GAChC,OAAO,CAAC,YAAY,CAAC;IAqBxB,IAAI,UAAU,IAAI,OAAO,CAExB;IAGD,IAAI,MAAM,IAAI,MAAM,CAEnB;IAGD,IAAI,SAAS,IAAI,MAAM,GAAG,IAAI,CAE7B;IAGK,OAAO,CACX,IAAI,EAAE,0BAA0B,GAC/B,OAAO,CAAC,sBAAsB,CAAC;IAqD5B,gBAAgB,CACpB,IAAI,EAAE,oBAAoB,GACzB,OAAO,CAAC,uBAAuB,CAAC;IAsB7B,IAAI,CACR,UAAU,EAAE,MAAM,EAClB,EAAE,EAAE,MAAM,EACV,OAAO,EAAE,MAAM,EACf,WAAW,SAAS,EACpB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GACjC,OAAO,CAAC;QAAE,SAAS,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,CAAC;IAW3C,KAAK,CAAC,UAAU,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAsBtC,EAAE,CAAC,CAAC,SAAS,MAAM,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAC,CAAC,GAAG,IAAI;IAC/D,EAAE,CACT,KAAK,EAAE,MAAM,GAAG,MAAM,EACtB,QAAQ,EAAE,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,IAAI,GACrC,IAAI;IAQE,IAAI,CAAC,CAAC,SAAS,MAAM,MAAM,EAClC,KAAK,EAAE,CAAC,EACR,GAAG,IAAI,EAAE,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,GAC7B,OAAO;IACD,IAAI,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,OAAO;YASpD,QAAQ;IAsBtB,OAAO,CAAC,WAAW;CAoBpB"}
@@ -0,0 +1,164 @@
1
+ import { EventEmitter } from "node:events";
2
+ import WebSocket from "ws";
3
+ import { resolveBrokerPaths } from "../broker/paths.js";
4
+ import { IPC_PROTOCOL_VERSION, } from "../broker/version-handshake.js";
5
+ import { discoverBroker } from "./lazy-spawn.js";
6
+ export class BrokerClient extends EventEmitter {
7
+ handle;
8
+ ws = null;
9
+ pluginPid = null;
10
+ _sessionId = null;
11
+ constructor(handle) {
12
+ super();
13
+ this.handle = handle;
14
+ }
15
+ static async discover(opts) {
16
+ const paths = resolveBrokerPaths(opts.accountId, opts.runtimeToken, {
17
+ userDataDir: opts.userDataDir,
18
+ });
19
+ const handle = await discoverBroker({
20
+ discoveryFile: paths.discoveryFile,
21
+ timeoutMs: opts.timeoutMs,
22
+ brokerEntry: opts.brokerEntry,
23
+ nodeBinary: opts.nodeBinary,
24
+ env: {
25
+ MASONS_BROKER_ACCOUNT_ID: opts.accountId,
26
+ ...(opts.userDataDir
27
+ ? { MASONS_BROKER_USER_DATA: opts.userDataDir }
28
+ : {}),
29
+ ...(opts.env ?? {}),
30
+ },
31
+ });
32
+ return new BrokerClient(handle);
33
+ }
34
+ get spawnedNew() {
35
+ return this.handle.spawned;
36
+ }
37
+ get ipcUrl() {
38
+ return this.handle.record.ipcUrl;
39
+ }
40
+ get sessionId() {
41
+ return this._sessionId;
42
+ }
43
+ async connect(opts) {
44
+ if (this.ws) {
45
+ throw new Error("BrokerClient already connected");
46
+ }
47
+ this.pluginPid = opts.pluginPid;
48
+ const initRes = await this.httpJson("POST", "/v1/initialize", {
49
+ client_protocol_version: IPC_PROTOCOL_VERSION,
50
+ client_kind: opts.clientKind,
51
+ client_version: opts.clientVersion,
52
+ capabilities: { endpoint_metadata_v1: true },
53
+ });
54
+ this._sessionId = initRes.session_id;
55
+ const wsUrl = `${this.handle.record.ipcUrl.replace(/^http/, "ws")}/v1/stream`;
56
+ const ws = new WebSocket(wsUrl, {
57
+ headers: {
58
+ Authorization: `Bearer ${this.handle.record.bearerToken}`,
59
+ "x-plugin-pid": String(opts.pluginPid),
60
+ },
61
+ });
62
+ this.ws = ws;
63
+ await new Promise((resolve, reject) => {
64
+ const onOpen = () => {
65
+ ws.off("error", onError);
66
+ resolve();
67
+ };
68
+ const onError = (err) => {
69
+ ws.off("open", onOpen);
70
+ reject(err);
71
+ };
72
+ ws.once("open", onOpen);
73
+ ws.once("error", onError);
74
+ });
75
+ ws.on("message", (data) => this.handleFrame(data.toString()));
76
+ ws.on("close", () => {
77
+ this.ws = null;
78
+ this.emit("disconnected");
79
+ });
80
+ ws.on("error", (err) => this.emit("error", err));
81
+ return initRes;
82
+ }
83
+ async registerEndpoint(args) {
84
+ if (this.pluginPid === null) {
85
+ throw new Error("BrokerClient.connect must be called before registerEndpoint");
86
+ }
87
+ return this.httpJson("POST", "/v1/endpoint/register", {
88
+ agent_id: args.agentId,
89
+ plugin_pid: this.pluginPid,
90
+ kind: args.kind,
91
+ session_name: args.sessionName,
92
+ workspace: args.workspace,
93
+ tracking_ref: args.trackingRef,
94
+ task_hint: args.taskHint,
95
+ });
96
+ }
97
+ async send(endpointId, to, content, contentType = "text", metadata) {
98
+ return this.httpJson("POST", "/v1/send", {
99
+ endpoint_id: endpointId,
100
+ to,
101
+ content,
102
+ contentType,
103
+ metadata,
104
+ });
105
+ }
106
+ async close(endpointId) {
107
+ if (endpointId) {
108
+ try {
109
+ await this.httpJson("DELETE", `/v1/endpoint/${encodeURIComponent(endpointId)}`);
110
+ }
111
+ catch {
112
+ }
113
+ }
114
+ if (this.ws &&
115
+ (this.ws.readyState === WebSocket.OPEN ||
116
+ this.ws.readyState === WebSocket.CONNECTING)) {
117
+ this.ws.close(1000, "client_close");
118
+ }
119
+ this.ws = null;
120
+ }
121
+ on(event, listener) {
122
+ return super.on(event, listener);
123
+ }
124
+ emit(event, ...args) {
125
+ return super.emit(event, ...args);
126
+ }
127
+ async httpJson(method, path, body) {
128
+ const res = await fetch(`${this.handle.record.ipcUrl}${path}`, {
129
+ method,
130
+ headers: {
131
+ Authorization: `Bearer ${this.handle.record.bearerToken}`,
132
+ ...(body ? { "Content-Type": "application/json" } : {}),
133
+ },
134
+ body: body ? JSON.stringify(body) : undefined,
135
+ });
136
+ if (!res.ok) {
137
+ const errText = await res.text().catch(() => "");
138
+ throw new Error(`broker ${method} ${path} failed: ${res.status} ${errText}`);
139
+ }
140
+ return (await res.json());
141
+ }
142
+ handleFrame(raw) {
143
+ let parsed;
144
+ try {
145
+ parsed = JSON.parse(raw);
146
+ }
147
+ catch {
148
+ this.emit("error", new Error("BrokerClient received non-JSON frame"));
149
+ return;
150
+ }
151
+ if (!parsed || typeof parsed !== "object" || !("event" in parsed))
152
+ return;
153
+ const event = parsed.event;
154
+ if (event === "message_received") {
155
+ const msg = parsed;
156
+ this.emit("message_received", {
157
+ from: msg.from,
158
+ content: msg.content,
159
+ contentType: msg.contentType,
160
+ metadata: msg.metadata,
161
+ });
162
+ }
163
+ }
164
+ }
@@ -0,0 +1,18 @@
1
+ import { type DiscoveryRecord } from "../broker/discovery-file.js";
2
+ export interface LazySpawnOptions {
3
+ brokerEntry?: string;
4
+ env?: Record<string, string>;
5
+ nodeBinary?: string;
6
+ }
7
+ export interface DiscoverOptions extends LazySpawnOptions {
8
+ discoveryFile: string;
9
+ timeoutMs?: number;
10
+ pollIntervalMs?: number;
11
+ forceSpawn?: boolean;
12
+ }
13
+ export interface DiscoveryHandle {
14
+ record: DiscoveryRecord;
15
+ spawned: boolean;
16
+ }
17
+ export declare function discoverBroker(opts: DiscoverOptions): Promise<DiscoveryHandle>;
18
+ //# sourceMappingURL=lazy-spawn.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"lazy-spawn.d.ts","sourceRoot":"","sources":["../../src/broker-client/lazy-spawn.ts"],"names":[],"mappings":"AA0CA,OAAO,EACL,KAAK,eAAe,EAErB,MAAM,6BAA6B,CAAC;AAErC,MAAM,WAAW,gBAAgB;IAE/B,WAAW,CAAC,EAAE,MAAM,CAAC;IAErB,GAAG,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAE7B,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAMD,MAAM,WAAW,eAAgB,SAAQ,gBAAgB;IAEvD,aAAa,EAAE,MAAM,CAAC;IAEtB,SAAS,CAAC,EAAE,MAAM,CAAC;IAEnB,cAAc,CAAC,EAAE,MAAM,CAAC;IAExB,UAAU,CAAC,EAAE,OAAO,CAAC;CACtB;AAED,MAAM,WAAW,eAAe;IAC9B,MAAM,EAAE,eAAe,CAAC;IAExB,OAAO,EAAE,OAAO,CAAC;CAClB;AAMD,wBAAsB,cAAc,CAClC,IAAI,EAAE,eAAe,GACpB,OAAO,CAAC,eAAe,CAAC,CA8B1B"}
@@ -0,0 +1,61 @@
1
+ import { spawn } from "node:child_process";
2
+ import { fileURLToPath } from "node:url";
3
+ import { readDiscoveryFile, } from "../broker/discovery-file.js";
4
+ const DEFAULT_POLL_INTERVAL_MS = 100;
5
+ const DEFAULT_DISCOVER_TIMEOUT_MS = 5_000;
6
+ const DEFAULT_HEALTH_TIMEOUT_MS = 2_000;
7
+ export async function discoverBroker(opts) {
8
+ const deadline = Date.now() + (opts.timeoutMs ?? DEFAULT_DISCOVER_TIMEOUT_MS);
9
+ const poll = opts.pollIntervalMs ?? DEFAULT_POLL_INTERVAL_MS;
10
+ if (!opts.forceSpawn) {
11
+ const existing = readDiscoveryFile(opts.discoveryFile);
12
+ if (existing &&
13
+ (await probeHealth(existing.ipcUrl, existing.bearerToken))) {
14
+ return { record: existing, spawned: false };
15
+ }
16
+ }
17
+ spawnDaemon(opts);
18
+ while (Date.now() < deadline) {
19
+ const rec = readDiscoveryFile(opts.discoveryFile);
20
+ if (rec && (await probeHealth(rec.ipcUrl, rec.bearerToken))) {
21
+ return { record: rec, spawned: true };
22
+ }
23
+ await sleep(poll);
24
+ }
25
+ throw new Error(`broker did not become healthy within ${opts.timeoutMs ?? DEFAULT_DISCOVER_TIMEOUT_MS}ms; ` +
26
+ `check ${opts.discoveryFile} and the daemon log`);
27
+ }
28
+ function spawnDaemon(opts) {
29
+ const entry = opts.brokerEntry ?? resolveDefaultBrokerEntry();
30
+ const child = spawn(opts.nodeBinary ?? process.execPath, [entry], {
31
+ detached: true,
32
+ stdio: "ignore",
33
+ windowsHide: true,
34
+ env: { ...process.env, ...(opts.env ?? {}) },
35
+ });
36
+ child.unref();
37
+ }
38
+ function resolveDefaultBrokerEntry() {
39
+ const self = fileURLToPath(import.meta.url);
40
+ return self.replace(/broker-client\/lazy-spawn\.(c?js|ts)$/, "broker/entry.$1");
41
+ }
42
+ async function probeHealth(ipcUrl, bearerToken) {
43
+ const ctrl = new AbortController();
44
+ const timer = setTimeout(() => ctrl.abort(), DEFAULT_HEALTH_TIMEOUT_MS);
45
+ try {
46
+ const res = await fetch(`${ipcUrl}/health`, {
47
+ headers: { Authorization: `Bearer ${bearerToken}` },
48
+ signal: ctrl.signal,
49
+ });
50
+ return res.ok;
51
+ }
52
+ catch {
53
+ return false;
54
+ }
55
+ finally {
56
+ clearTimeout(timer);
57
+ }
58
+ }
59
+ function sleep(ms) {
60
+ return new Promise((resolve) => setTimeout(resolve, ms));
61
+ }