@assemblyline-agents/runtime 0.1.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (331) hide show
  1. package/LICENSE +21 -0
  2. package/README.md +19 -0
  3. package/dist/agent-harness.d.ts +108 -0
  4. package/dist/agent-harness.d.ts.map +1 -0
  5. package/dist/agent-harness.js +2 -0
  6. package/dist/agent-harness.js.map +1 -0
  7. package/dist/attachments-legacy.d.ts +17 -0
  8. package/dist/attachments-legacy.d.ts.map +1 -0
  9. package/dist/attachments-legacy.js +273 -0
  10. package/dist/attachments-legacy.js.map +1 -0
  11. package/dist/attachments.d.ts +49 -0
  12. package/dist/attachments.d.ts.map +1 -0
  13. package/dist/attachments.js +832 -0
  14. package/dist/attachments.js.map +1 -0
  15. package/dist/builtin-tool-executors.d.ts +96 -0
  16. package/dist/builtin-tool-executors.d.ts.map +1 -0
  17. package/dist/builtin-tool-executors.js +501 -0
  18. package/dist/builtin-tool-executors.js.map +1 -0
  19. package/dist/builtin-tools.d.ts +30 -0
  20. package/dist/builtin-tools.d.ts.map +1 -0
  21. package/dist/builtin-tools.js +690 -0
  22. package/dist/builtin-tools.js.map +1 -0
  23. package/dist/capability-catalog.d.ts +27 -0
  24. package/dist/capability-catalog.d.ts.map +1 -0
  25. package/dist/capability-catalog.js +193 -0
  26. package/dist/capability-catalog.js.map +1 -0
  27. package/dist/channel-http-util.d.ts +34 -0
  28. package/dist/channel-http-util.d.ts.map +1 -0
  29. package/dist/channel-http-util.js +93 -0
  30. package/dist/channel-http-util.js.map +1 -0
  31. package/dist/channel-ingress.d.ts +56 -0
  32. package/dist/channel-ingress.d.ts.map +1 -0
  33. package/dist/channel-ingress.js +273 -0
  34. package/dist/channel-ingress.js.map +1 -0
  35. package/dist/channel.d.ts +146 -0
  36. package/dist/channel.d.ts.map +1 -0
  37. package/dist/channel.js +2 -0
  38. package/dist/channel.js.map +1 -0
  39. package/dist/checkpoints.d.ts +34 -0
  40. package/dist/checkpoints.d.ts.map +1 -0
  41. package/dist/checkpoints.js +86 -0
  42. package/dist/checkpoints.js.map +1 -0
  43. package/dist/code-mode.d.ts +20 -0
  44. package/dist/code-mode.d.ts.map +1 -0
  45. package/dist/code-mode.js +658 -0
  46. package/dist/code-mode.js.map +1 -0
  47. package/dist/computer-relay-protocol.d.ts +36 -0
  48. package/dist/computer-relay-protocol.d.ts.map +1 -0
  49. package/dist/computer-relay-protocol.js +127 -0
  50. package/dist/computer-relay-protocol.js.map +1 -0
  51. package/dist/concurrency.d.ts +68 -0
  52. package/dist/concurrency.d.ts.map +1 -0
  53. package/dist/concurrency.js +193 -0
  54. package/dist/concurrency.js.map +1 -0
  55. package/dist/connection-auth-util.d.ts +56 -0
  56. package/dist/connection-auth-util.d.ts.map +1 -0
  57. package/dist/connection-auth-util.js +154 -0
  58. package/dist/connection-auth-util.js.map +1 -0
  59. package/dist/connection-clients.d.ts +50 -0
  60. package/dist/connection-clients.d.ts.map +1 -0
  61. package/dist/connection-clients.js +402 -0
  62. package/dist/connection-clients.js.map +1 -0
  63. package/dist/connection-stores.d.ts +108 -0
  64. package/dist/connection-stores.d.ts.map +1 -0
  65. package/dist/connection-stores.js +127 -0
  66. package/dist/connection-stores.js.map +1 -0
  67. package/dist/connection-util.d.ts +27 -0
  68. package/dist/connection-util.d.ts.map +1 -0
  69. package/dist/connection-util.js +118 -0
  70. package/dist/connection-util.js.map +1 -0
  71. package/dist/connections.d.ts +152 -0
  72. package/dist/connections.d.ts.map +1 -0
  73. package/dist/connections.js +838 -0
  74. package/dist/connections.js.map +1 -0
  75. package/dist/content-capture.d.ts +32 -0
  76. package/dist/content-capture.d.ts.map +1 -0
  77. package/dist/content-capture.js +77 -0
  78. package/dist/content-capture.js.map +1 -0
  79. package/dist/context-policy-util.d.ts +45 -0
  80. package/dist/context-policy-util.d.ts.map +1 -0
  81. package/dist/context-policy-util.js +120 -0
  82. package/dist/context-policy-util.js.map +1 -0
  83. package/dist/conversation-context.d.ts +55 -0
  84. package/dist/conversation-context.d.ts.map +1 -0
  85. package/dist/conversation-context.js +179 -0
  86. package/dist/conversation-context.js.map +1 -0
  87. package/dist/conversation-transcript.d.ts +49 -0
  88. package/dist/conversation-transcript.d.ts.map +1 -0
  89. package/dist/conversation-transcript.js +77 -0
  90. package/dist/conversation-transcript.js.map +1 -0
  91. package/dist/conversation-turn-dispatch.d.ts +46 -0
  92. package/dist/conversation-turn-dispatch.d.ts.map +1 -0
  93. package/dist/conversation-turn-dispatch.js +201 -0
  94. package/dist/conversation-turn-dispatch.js.map +1 -0
  95. package/dist/delivery-queue.d.ts +89 -0
  96. package/dist/delivery-queue.d.ts.map +1 -0
  97. package/dist/delivery-queue.js +244 -0
  98. package/dist/delivery-queue.js.map +1 -0
  99. package/dist/delivery-send.d.ts +44 -0
  100. package/dist/delivery-send.d.ts.map +1 -0
  101. package/dist/delivery-send.js +494 -0
  102. package/dist/delivery-send.js.map +1 -0
  103. package/dist/env-config.d.ts +6 -0
  104. package/dist/env-config.d.ts.map +1 -0
  105. package/dist/env-config.js +27 -0
  106. package/dist/env-config.js.map +1 -0
  107. package/dist/eval-run-options.d.ts +37 -0
  108. package/dist/eval-run-options.d.ts.map +1 -0
  109. package/dist/eval-run-options.js +106 -0
  110. package/dist/eval-run-options.js.map +1 -0
  111. package/dist/harness-continuation.d.ts +54 -0
  112. package/dist/harness-continuation.d.ts.map +1 -0
  113. package/dist/harness-continuation.js +58 -0
  114. package/dist/harness-continuation.js.map +1 -0
  115. package/dist/harness-loop.d.ts +110 -0
  116. package/dist/harness-loop.d.ts.map +1 -0
  117. package/dist/harness-loop.js +575 -0
  118. package/dist/harness-loop.js.map +1 -0
  119. package/dist/hook-dispatch.d.ts +19 -0
  120. package/dist/hook-dispatch.d.ts.map +1 -0
  121. package/dist/hook-dispatch.js +37 -0
  122. package/dist/hook-dispatch.js.map +1 -0
  123. package/dist/http.d.ts +26 -0
  124. package/dist/http.d.ts.map +1 -0
  125. package/dist/http.js +114 -0
  126. package/dist/http.js.map +1 -0
  127. package/dist/in-memory-conversation-turn-store.d.ts +29 -0
  128. package/dist/in-memory-conversation-turn-store.d.ts.map +1 -0
  129. package/dist/in-memory-conversation-turn-store.js +166 -0
  130. package/dist/in-memory-conversation-turn-store.js.map +1 -0
  131. package/dist/in-memory-run-store.d.ts +39 -0
  132. package/dist/in-memory-run-store.d.ts.map +1 -0
  133. package/dist/in-memory-run-store.js +332 -0
  134. package/dist/in-memory-run-store.js.map +1 -0
  135. package/dist/in-memory-sandbox-session-store.d.ts +33 -0
  136. package/dist/in-memory-sandbox-session-store.d.ts.map +1 -0
  137. package/dist/in-memory-sandbox-session-store.js +186 -0
  138. package/dist/in-memory-sandbox-session-store.js.map +1 -0
  139. package/dist/in-memory-small-stores.d.ts +117 -0
  140. package/dist/in-memory-small-stores.d.ts.map +1 -0
  141. package/dist/in-memory-small-stores.js +340 -0
  142. package/dist/in-memory-small-stores.js.map +1 -0
  143. package/dist/in-memory-usage-store.d.ts +15 -0
  144. package/dist/in-memory-usage-store.d.ts.map +1 -0
  145. package/dist/in-memory-usage-store.js +49 -0
  146. package/dist/in-memory-usage-store.js.map +1 -0
  147. package/dist/index.d.ts +467 -0
  148. package/dist/index.d.ts.map +1 -0
  149. package/dist/index.js +2046 -0
  150. package/dist/index.js.map +1 -0
  151. package/dist/json-util.d.ts +19 -0
  152. package/dist/json-util.d.ts.map +1 -0
  153. package/dist/json-util.js +59 -0
  154. package/dist/json-util.js.map +1 -0
  155. package/dist/local-sandbox.d.ts +16 -0
  156. package/dist/local-sandbox.d.ts.map +1 -0
  157. package/dist/local-sandbox.js +327 -0
  158. package/dist/local-sandbox.js.map +1 -0
  159. package/dist/logging.d.ts +16 -0
  160. package/dist/logging.d.ts.map +1 -0
  161. package/dist/logging.js +41 -0
  162. package/dist/logging.js.map +1 -0
  163. package/dist/mcp-client.d.ts +27 -0
  164. package/dist/mcp-client.d.ts.map +1 -0
  165. package/dist/mcp-client.js +169 -0
  166. package/dist/mcp-client.js.map +1 -0
  167. package/dist/model-harnesses.d.ts +4 -0
  168. package/dist/model-harnesses.d.ts.map +1 -0
  169. package/dist/model-harnesses.js +6 -0
  170. package/dist/model-harnesses.js.map +1 -0
  171. package/dist/model-media.d.ts +36 -0
  172. package/dist/model-media.d.ts.map +1 -0
  173. package/dist/model-media.js +234 -0
  174. package/dist/model-media.js.map +1 -0
  175. package/dist/operator-controls.d.ts +31 -0
  176. package/dist/operator-controls.d.ts.map +1 -0
  177. package/dist/operator-controls.js +71 -0
  178. package/dist/operator-controls.js.map +1 -0
  179. package/dist/pi-agent-harness.d.ts +40 -0
  180. package/dist/pi-agent-harness.d.ts.map +1 -0
  181. package/dist/pi-agent-harness.js +222 -0
  182. package/dist/pi-agent-harness.js.map +1 -0
  183. package/dist/pi-harness.d.ts +42 -0
  184. package/dist/pi-harness.d.ts.map +1 -0
  185. package/dist/pi-harness.js +142 -0
  186. package/dist/pi-harness.js.map +1 -0
  187. package/dist/relay-mcp-client.d.ts +17 -0
  188. package/dist/relay-mcp-client.d.ts.map +1 -0
  189. package/dist/relay-mcp-client.js +119 -0
  190. package/dist/relay-mcp-client.js.map +1 -0
  191. package/dist/replay-resume.d.ts +107 -0
  192. package/dist/replay-resume.d.ts.map +1 -0
  193. package/dist/replay-resume.js +495 -0
  194. package/dist/replay-resume.js.map +1 -0
  195. package/dist/resource-projection-util.d.ts +58 -0
  196. package/dist/resource-projection-util.d.ts.map +1 -0
  197. package/dist/resource-projection-util.js +298 -0
  198. package/dist/resource-projection-util.js.map +1 -0
  199. package/dist/resource-projection.d.ts +45 -0
  200. package/dist/resource-projection.d.ts.map +1 -0
  201. package/dist/resource-projection.js +481 -0
  202. package/dist/resource-projection.js.map +1 -0
  203. package/dist/run-heartbeat.d.ts +34 -0
  204. package/dist/run-heartbeat.d.ts.map +1 -0
  205. package/dist/run-heartbeat.js +62 -0
  206. package/dist/run-heartbeat.js.map +1 -0
  207. package/dist/run-outcome.d.ts +47 -0
  208. package/dist/run-outcome.d.ts.map +1 -0
  209. package/dist/run-outcome.js +92 -0
  210. package/dist/run-outcome.js.map +1 -0
  211. package/dist/run-recovery.d.ts +117 -0
  212. package/dist/run-recovery.d.ts.map +1 -0
  213. package/dist/run-recovery.js +466 -0
  214. package/dist/run-recovery.js.map +1 -0
  215. package/dist/run-resolver.d.ts +35 -0
  216. package/dist/run-resolver.d.ts.map +1 -0
  217. package/dist/run-resolver.js +84 -0
  218. package/dist/run-resolver.js.map +1 -0
  219. package/dist/run-stream.d.ts +37 -0
  220. package/dist/run-stream.d.ts.map +1 -0
  221. package/dist/run-stream.js +78 -0
  222. package/dist/run-stream.js.map +1 -0
  223. package/dist/runs-query.d.ts +29 -0
  224. package/dist/runs-query.d.ts.map +1 -0
  225. package/dist/runs-query.js +142 -0
  226. package/dist/runs-query.js.map +1 -0
  227. package/dist/runtime-bundle.d.ts +3 -0
  228. package/dist/runtime-bundle.d.ts.map +1 -0
  229. package/dist/runtime-bundle.js +18 -0
  230. package/dist/runtime-bundle.js.map +1 -0
  231. package/dist/runtime-types.d.ts +397 -0
  232. package/dist/runtime-types.d.ts.map +1 -0
  233. package/dist/runtime-types.js +2 -0
  234. package/dist/runtime-types.js.map +1 -0
  235. package/dist/runtime-util.d.ts +45 -0
  236. package/dist/runtime-util.d.ts.map +1 -0
  237. package/dist/runtime-util.js +116 -0
  238. package/dist/runtime-util.js.map +1 -0
  239. package/dist/sandbox-cli-client.d.ts +20 -0
  240. package/dist/sandbox-cli-client.d.ts.map +1 -0
  241. package/dist/sandbox-cli-client.js +97 -0
  242. package/dist/sandbox-cli-client.js.map +1 -0
  243. package/dist/sandbox-fs.d.ts +56 -0
  244. package/dist/sandbox-fs.d.ts.map +1 -0
  245. package/dist/sandbox-fs.js +276 -0
  246. package/dist/sandbox-fs.js.map +1 -0
  247. package/dist/sandbox-session-key.d.ts +8 -0
  248. package/dist/sandbox-session-key.d.ts.map +1 -0
  249. package/dist/sandbox-session-key.js +32 -0
  250. package/dist/sandbox-session-key.js.map +1 -0
  251. package/dist/sandbox-snapshot-policy.d.ts +27 -0
  252. package/dist/sandbox-snapshot-policy.d.ts.map +1 -0
  253. package/dist/sandbox-snapshot-policy.js +94 -0
  254. package/dist/sandbox-snapshot-policy.js.map +1 -0
  255. package/dist/sandbox-sync-worker.d.ts +63 -0
  256. package/dist/sandbox-sync-worker.d.ts.map +1 -0
  257. package/dist/sandbox-sync-worker.js +448 -0
  258. package/dist/sandbox-sync-worker.js.map +1 -0
  259. package/dist/schedule-execution.d.ts +59 -0
  260. package/dist/schedule-execution.d.ts.map +1 -0
  261. package/dist/schedule-execution.js +951 -0
  262. package/dist/schedule-execution.js.map +1 -0
  263. package/dist/scheduler.d.ts +19 -0
  264. package/dist/scheduler.d.ts.map +1 -0
  265. package/dist/scheduler.js +38 -0
  266. package/dist/scheduler.js.map +1 -0
  267. package/dist/security.d.ts +20 -0
  268. package/dist/security.d.ts.map +1 -0
  269. package/dist/security.js +145 -0
  270. package/dist/security.js.map +1 -0
  271. package/dist/self-improvement.d.ts +245 -0
  272. package/dist/self-improvement.d.ts.map +1 -0
  273. package/dist/self-improvement.js +554 -0
  274. package/dist/self-improvement.js.map +1 -0
  275. package/dist/source-loader.d.ts +19 -0
  276. package/dist/source-loader.d.ts.map +1 -0
  277. package/dist/source-loader.js +160 -0
  278. package/dist/source-loader.js.map +1 -0
  279. package/dist/state-store-helpers.d.ts +28 -0
  280. package/dist/state-store-helpers.d.ts.map +1 -0
  281. package/dist/state-store-helpers.js +164 -0
  282. package/dist/state-store-helpers.js.map +1 -0
  283. package/dist/state-stores.d.ts +25 -0
  284. package/dist/state-stores.d.ts.map +1 -0
  285. package/dist/state-stores.js +157 -0
  286. package/dist/state-stores.js.map +1 -0
  287. package/dist/state.d.ts +736 -0
  288. package/dist/state.d.ts.map +1 -0
  289. package/dist/state.js +46 -0
  290. package/dist/state.js.map +1 -0
  291. package/dist/storage.d.ts +188 -0
  292. package/dist/storage.d.ts.map +1 -0
  293. package/dist/storage.js +752 -0
  294. package/dist/storage.js.map +1 -0
  295. package/dist/subagent-execution.d.ts +80 -0
  296. package/dist/subagent-execution.d.ts.map +1 -0
  297. package/dist/subagent-execution.js +135 -0
  298. package/dist/subagent-execution.js.map +1 -0
  299. package/dist/telemetry.d.ts +49 -0
  300. package/dist/telemetry.d.ts.map +1 -0
  301. package/dist/telemetry.js +60 -0
  302. package/dist/telemetry.js.map +1 -0
  303. package/dist/tool-apis.d.ts +22 -0
  304. package/dist/tool-apis.d.ts.map +1 -0
  305. package/dist/tool-apis.js +344 -0
  306. package/dist/tool-apis.js.map +1 -0
  307. package/dist/tool-execution.d.ts +143 -0
  308. package/dist/tool-execution.d.ts.map +1 -0
  309. package/dist/tool-execution.js +573 -0
  310. package/dist/tool-execution.js.map +1 -0
  311. package/dist/trusted-attachments.d.ts +20 -0
  312. package/dist/trusted-attachments.d.ts.map +1 -0
  313. package/dist/trusted-attachments.js +276 -0
  314. package/dist/trusted-attachments.js.map +1 -0
  315. package/dist/usage-accounting.d.ts +37 -0
  316. package/dist/usage-accounting.d.ts.map +1 -0
  317. package/dist/usage-accounting.js +181 -0
  318. package/dist/usage-accounting.js.map +1 -0
  319. package/dist/usage-budget.d.ts +29 -0
  320. package/dist/usage-budget.d.ts.map +1 -0
  321. package/dist/usage-budget.js +145 -0
  322. package/dist/usage-budget.js.map +1 -0
  323. package/dist/usage-reconciliation.d.ts +76 -0
  324. package/dist/usage-reconciliation.d.ts.map +1 -0
  325. package/dist/usage-reconciliation.js +463 -0
  326. package/dist/usage-reconciliation.js.map +1 -0
  327. package/dist/usage-records.d.ts +24 -0
  328. package/dist/usage-records.d.ts.map +1 -0
  329. package/dist/usage-records.js +445 -0
  330. package/dist/usage-records.js.map +1 -0
  331. package/package.json +51 -0
@@ -0,0 +1,332 @@
1
+ import { randomUUID } from "node:crypto";
2
+ import { applyDeliveryPatch, createDeliveryRecord, deliveryQueueRetryDelayMs, isDeliveryDue } from "./delivery-queue.js";
3
+ import { ACTIVE_RUN_STATUSES, TERMINAL_RUN_STATUSES, checkpointApproxBytes, checkpointNowMs, emptyCheckpointRetentionReasons, noPersist, positiveLimit, terminalCheckpointTtlMs, timestamp } from "./state-store-helpers.js";
4
+ export class InMemoryRunStore {
5
+ persist;
6
+ runs = new Map();
7
+ events = [];
8
+ toolCalls = new Map();
9
+ checkpoints = [];
10
+ deliveries = new Map();
11
+ idempotency = new Set();
12
+ constructor(persist = noPersist) {
13
+ this.persist = persist;
14
+ }
15
+ async createRun(input) {
16
+ const now = timestamp();
17
+ const run = {
18
+ ...input,
19
+ id: randomUUID(),
20
+ createdAt: now,
21
+ updatedAt: now
22
+ };
23
+ this.runs.set(run.id, run);
24
+ await this.persist();
25
+ return run;
26
+ }
27
+ async updateRun(id, patch) {
28
+ const current = this.runs.get(id);
29
+ if (!current)
30
+ throw new Error(`Run not found: ${id}`);
31
+ const updated = { ...current, ...patch, updatedAt: timestamp() };
32
+ this.runs.set(id, updated);
33
+ await this.persist();
34
+ return updated;
35
+ }
36
+ async touchRun(id) {
37
+ const current = this.runs.get(id);
38
+ if (!current || (current.status !== "created" && current.status !== "running"))
39
+ return;
40
+ this.runs.set(id, { ...current, updatedAt: timestamp() });
41
+ await this.persist();
42
+ }
43
+ async requestRunControl(id, intent) {
44
+ const current = this.runs.get(id);
45
+ if (!current)
46
+ throw new Error(`Run not found: ${id}`);
47
+ let updated;
48
+ if (intent === "cancel" && current.status !== "running") {
49
+ if (["completed", "failed", "cancelled"].includes(current.status))
50
+ return current;
51
+ const { controlIntent: _ignored, ...rest } = current;
52
+ updated = { ...rest, status: "cancelled", updatedAt: timestamp() };
53
+ }
54
+ else if (current.status === "running") {
55
+ updated = {
56
+ ...current,
57
+ controlIntent: current.controlIntent === "cancel" || intent === "cancel" ? "cancel" : "suspend",
58
+ updatedAt: timestamp()
59
+ };
60
+ }
61
+ else {
62
+ return current;
63
+ }
64
+ this.runs.set(id, updated);
65
+ await this.persist();
66
+ return updated;
67
+ }
68
+ async settleRunControl(id) {
69
+ const current = this.runs.get(id);
70
+ if (!current)
71
+ throw new Error(`Run not found: ${id}`);
72
+ if (current.status !== "running" || !current.controlIntent)
73
+ return current;
74
+ const status = current.controlIntent === "cancel" ? "cancelled" : "suspended";
75
+ const { controlIntent: _ignored, ...rest } = current;
76
+ const updated = { ...rest, status, updatedAt: timestamp() };
77
+ this.runs.set(id, updated);
78
+ await this.persist();
79
+ return updated;
80
+ }
81
+ async getRun(id) {
82
+ return this.runs.get(id);
83
+ }
84
+ async listRuns() {
85
+ return [...this.runs.values()].sort((a, b) => a.createdAt.localeCompare(b.createdAt));
86
+ }
87
+ async appendEvent(runId, type, data = {}) {
88
+ const event = {
89
+ id: randomUUID(),
90
+ runId,
91
+ sequence: this.events.filter((candidate) => candidate.runId === runId).length + 1,
92
+ type,
93
+ timestamp: timestamp(),
94
+ data
95
+ };
96
+ this.events.push(event);
97
+ await this.persist();
98
+ return event;
99
+ }
100
+ async listEvents(runId) {
101
+ return this.events.filter((event) => event.runId === runId).sort((a, b) => a.sequence - b.sequence);
102
+ }
103
+ async createToolCall(input) {
104
+ const now = timestamp();
105
+ const record = {
106
+ ...input,
107
+ id: randomUUID(),
108
+ createdAt: now,
109
+ updatedAt: now
110
+ };
111
+ this.toolCalls.set(record.id, record);
112
+ await this.persist();
113
+ return record;
114
+ }
115
+ async updateToolCall(id, patch) {
116
+ const current = this.toolCalls.get(id);
117
+ if (!current)
118
+ throw new Error(`Tool call not found: ${id}`);
119
+ const updated = { ...current, ...patch, updatedAt: timestamp() };
120
+ this.toolCalls.set(id, updated);
121
+ await this.persist();
122
+ return updated;
123
+ }
124
+ async listToolCalls(runId) {
125
+ return [...this.toolCalls.values()].filter((toolCall) => toolCall.runId === runId);
126
+ }
127
+ async createCheckpoint(input) {
128
+ const checkpoint = {
129
+ ...input,
130
+ id: randomUUID(),
131
+ createdAt: timestamp()
132
+ };
133
+ this.checkpoints.push(checkpoint);
134
+ await this.persist();
135
+ return checkpoint;
136
+ }
137
+ async listCheckpoints(runId) {
138
+ return this.checkpoints.filter((checkpoint) => checkpoint.runId === runId);
139
+ }
140
+ async compactCheckpoints(options = {}) {
141
+ const maxActive = options.maxActivePerRunName === undefined ? 0 : positiveLimit(options.maxActivePerRunName, 1);
142
+ const nowMs = checkpointNowMs(options.now);
143
+ const activeNames = options.activeNames ? new Set(options.activeNames) : undefined;
144
+ const activeCandidates = new Set();
145
+ if (maxActive > 0) {
146
+ const groups = new Map();
147
+ for (const checkpoint of this.checkpoints) {
148
+ if (options.runId && checkpoint.runId !== options.runId)
149
+ continue;
150
+ if (activeNames && !activeNames.has(checkpoint.name))
151
+ continue;
152
+ const run = this.runs.get(checkpoint.runId);
153
+ if (!run || !ACTIVE_RUN_STATUSES.has(run.status))
154
+ continue;
155
+ const key = `${checkpoint.runId}\0${checkpoint.name}`;
156
+ const group = groups.get(key) ?? [];
157
+ group.push(checkpoint);
158
+ groups.set(key, group);
159
+ }
160
+ for (const group of groups.values()) {
161
+ group.sort((a, b) => b.createdAt.localeCompare(a.createdAt) || b.id.localeCompare(a.id));
162
+ for (const checkpoint of group.slice(maxActive))
163
+ activeCandidates.add(checkpoint.id);
164
+ }
165
+ }
166
+ const terminalCandidates = new Set();
167
+ for (const checkpoint of this.checkpoints) {
168
+ if (options.runId && checkpoint.runId !== options.runId)
169
+ continue;
170
+ const run = this.runs.get(checkpoint.runId);
171
+ if (!run || !TERMINAL_RUN_STATUSES.has(run.status))
172
+ continue;
173
+ const ttlMs = terminalCheckpointTtlMs(run, options);
174
+ if (ttlMs === undefined)
175
+ continue;
176
+ if (Date.parse(run.updatedAt) <= nowMs - ttlMs)
177
+ terminalCandidates.add(checkpoint.id);
178
+ }
179
+ const reasons = new Map();
180
+ for (const id of activeCandidates)
181
+ reasons.set(id, "active_compaction");
182
+ for (const id of terminalCandidates)
183
+ reasons.set(id, "terminal_ttl");
184
+ const byReason = emptyCheckpointRetentionReasons();
185
+ let candidateRows = 0;
186
+ let candidateBytes = 0;
187
+ for (const checkpoint of this.checkpoints) {
188
+ const reason = reasons.get(checkpoint.id);
189
+ if (!reason)
190
+ continue;
191
+ const bytes = checkpointApproxBytes(checkpoint);
192
+ candidateRows += 1;
193
+ candidateBytes += bytes;
194
+ byReason[reason].rows += 1;
195
+ byReason[reason].bytes += bytes;
196
+ }
197
+ if (!options.dryRun && reasons.size > 0) {
198
+ this.checkpoints = this.checkpoints.filter((checkpoint) => !reasons.has(checkpoint.id));
199
+ await this.persist();
200
+ }
201
+ return {
202
+ dryRun: options.dryRun === true,
203
+ candidateRows,
204
+ candidateBytes,
205
+ prunedRows: options.dryRun === true ? 0 : candidateRows,
206
+ prunedBytes: options.dryRun === true ? 0 : candidateBytes,
207
+ byReason
208
+ };
209
+ }
210
+ async createDelivery(input) {
211
+ const delivery = createDeliveryRecord(input);
212
+ this.deliveries.set(delivery.id, delivery);
213
+ await this.persist();
214
+ return delivery;
215
+ }
216
+ async updateDelivery(id, patch) {
217
+ const current = this.deliveries.get(id);
218
+ if (!current)
219
+ throw new Error(`Delivery not found: ${id}`);
220
+ const updated = applyDeliveryPatch(current, patch);
221
+ this.deliveries.set(id, updated);
222
+ await this.persist();
223
+ return updated;
224
+ }
225
+ async listDeliveries(runId) {
226
+ return [...this.deliveries.values()].filter((delivery) => delivery.runId === runId);
227
+ }
228
+ async leaseDueDeliveries(options = {}) {
229
+ const now = options.now ?? new Date();
230
+ const limit = positiveLimit(options.limit, 10);
231
+ const leaseForMs = positiveLimit(options.leaseForMs, 60_000);
232
+ const leaseToken = randomUUID();
233
+ const leaseExpiresAt = new Date(now.getTime() + leaseForMs).toISOString();
234
+ const leased = [];
235
+ const due = [...this.deliveries.values()]
236
+ .filter((delivery) => isDeliveryDue(delivery, now))
237
+ .sort((a, b) => a.nextAttemptAt.localeCompare(b.nextAttemptAt) || a.createdAt.localeCompare(b.createdAt))
238
+ .slice(0, limit);
239
+ for (const delivery of due) {
240
+ const updated = {
241
+ ...delivery,
242
+ status: "sending",
243
+ leaseToken,
244
+ leaseExpiresAt,
245
+ attemptCount: delivery.attemptCount + 1,
246
+ updatedAt: timestamp()
247
+ };
248
+ this.deliveries.set(updated.id, updated);
249
+ leased.push(updated);
250
+ }
251
+ if (leased.length > 0)
252
+ await this.persist();
253
+ return leased;
254
+ }
255
+ async completeDeliverySend(id, leaseToken, patch = {}) {
256
+ const current = this.deliveries.get(id);
257
+ if (!current)
258
+ throw new Error(`Delivery not found: ${id}`);
259
+ if (current.status === "sent")
260
+ return current;
261
+ if (current.leaseToken !== leaseToken)
262
+ throw new Error(`Delivery lease mismatch: ${id}`);
263
+ const now = timestamp();
264
+ const updated = {
265
+ ...current,
266
+ status: "sent",
267
+ payload: patch.payload ?? current.payload,
268
+ sentAt: now,
269
+ updatedAt: now
270
+ };
271
+ delete updated.leaseToken;
272
+ delete updated.leaseExpiresAt;
273
+ delete updated.lastError;
274
+ this.deliveries.set(id, updated);
275
+ await this.persist();
276
+ return updated;
277
+ }
278
+ async failDeliverySend(id, leaseToken, failure) {
279
+ const current = this.deliveries.get(id);
280
+ if (!current)
281
+ throw new Error(`Delivery not found: ${id}`);
282
+ if (current.leaseToken !== leaseToken)
283
+ throw new Error(`Delivery lease mismatch: ${id}`);
284
+ const now = failure.now ?? new Date();
285
+ const permanent = failure.permanent === true || current.attemptCount >= current.maxAttempts;
286
+ const updated = {
287
+ ...current,
288
+ status: permanent ? "failed" : "pending",
289
+ nextAttemptAt: permanent
290
+ ? current.nextAttemptAt
291
+ : new Date(now.getTime() + positiveLimit(failure.retryDelayMs, deliveryQueueRetryDelayMs(current.attemptCount))).toISOString(),
292
+ lastError: failure.error,
293
+ updatedAt: timestamp()
294
+ };
295
+ if (permanent)
296
+ updated.failedAt = now.toISOString();
297
+ delete updated.leaseToken;
298
+ delete updated.leaseExpiresAt;
299
+ this.deliveries.set(id, updated);
300
+ await this.persist();
301
+ return updated;
302
+ }
303
+ async recoverExpiredDeliveries(now = new Date()) {
304
+ const recovered = [];
305
+ for (const delivery of this.deliveries.values()) {
306
+ if (delivery.status !== "sending" || !delivery.leaseExpiresAt || Date.parse(delivery.leaseExpiresAt) > now.getTime())
307
+ continue;
308
+ const updated = {
309
+ ...delivery,
310
+ status: "pending",
311
+ nextAttemptAt: now.toISOString(),
312
+ updatedAt: timestamp()
313
+ };
314
+ delete updated.leaseToken;
315
+ delete updated.leaseExpiresAt;
316
+ this.deliveries.set(updated.id, updated);
317
+ recovered.push(updated);
318
+ }
319
+ if (recovered.length > 0)
320
+ await this.persist();
321
+ return recovered;
322
+ }
323
+ async reserveIdempotencyKey(scope, key, _ownerId) {
324
+ const composite = `${scope}:${key}`;
325
+ if (this.idempotency.has(composite))
326
+ return false;
327
+ this.idempotency.add(composite);
328
+ await this.persist();
329
+ return true;
330
+ }
331
+ }
332
+ //# sourceMappingURL=in-memory-run-store.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"in-memory-run-store.js","sourceRoot":"","sources":["../src/in-memory-run-store.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAEzC,OAAO,EACL,kBAAkB,EAClB,oBAAoB,EACpB,yBAAyB,EACzB,aAAa,EAKd,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EACL,mBAAmB,EACnB,qBAAqB,EACrB,qBAAqB,EACrB,eAAe,EACf,+BAA+B,EAC/B,SAAS,EACT,aAAa,EACb,uBAAuB,EACvB,SAAS,EACV,MAAM,0BAA0B,CAAC;AAelC,MAAM,OAAO,gBAAgB;IAQE;IAP7B,IAAI,GAAG,IAAI,GAAG,EAAqB,CAAC;IACpC,MAAM,GAAe,EAAE,CAAC;IACxB,SAAS,GAAG,IAAI,GAAG,EAA0B,CAAC;IAC9C,WAAW,GAAuB,EAAE,CAAC;IACrC,UAAU,GAAG,IAAI,GAAG,EAA8B,CAAC;IACnD,WAAW,GAAG,IAAI,GAAG,EAAU,CAAC;IAEhC,YAA6B,UAA+B,SAAS;QAAxC,YAAO,GAAP,OAAO,CAAiC;IAAG,CAAC;IAEzE,KAAK,CAAC,SAAS,CAAC,KAAwD;QACtE,MAAM,GAAG,GAAG,SAAS,EAAE,CAAC;QACxB,MAAM,GAAG,GAAc;YACrB,GAAG,KAAK;YACR,EAAE,EAAE,UAAU,EAAE;YAChB,SAAS,EAAE,GAAG;YACd,SAAS,EAAE,GAAG;SACf,CAAC;QACF,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;QAC3B,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACrB,OAAO,GAAG,CAAC;IACb,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,EAAU,EAAE,KAAmD;QAC7E,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAClC,IAAI,CAAC,OAAO;YAAE,MAAM,IAAI,KAAK,CAAC,kBAAkB,EAAE,EAAE,CAAC,CAAC;QACtD,MAAM,OAAO,GAAc,EAAE,GAAG,OAAO,EAAE,GAAG,KAAK,EAAE,SAAS,EAAE,SAAS,EAAE,EAAE,CAAC;QAC5E,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QAC3B,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACrB,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,EAAU;QACvB,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAClC,IAAI,CAAC,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,KAAK,SAAS,IAAI,OAAO,CAAC,MAAM,KAAK,SAAS,CAAC;YAAE,OAAO;QACvF,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,GAAG,OAAO,EAAE,SAAS,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC;QAC1D,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,EAAU,EAAE,MAAwB;QAC1D,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAClC,IAAI,CAAC,OAAO;YAAE,MAAM,IAAI,KAAK,CAAC,kBAAkB,EAAE,EAAE,CAAC,CAAC;QACtD,IAAI,OAAkB,CAAC;QACvB,IAAI,MAAM,KAAK,QAAQ,IAAI,OAAO,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;YACxD,IAAI,CAAC,WAAW,EAAE,QAAQ,EAAE,WAAW,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC;gBAAE,OAAO,OAAO,CAAC;YAClF,MAAM,EAAE,aAAa,EAAE,QAAQ,EAAE,GAAG,IAAI,EAAE,GAAG,OAAO,CAAC;YACrD,OAAO,GAAG,EAAE,GAAG,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,SAAS,EAAE,SAAS,EAAE,EAAE,CAAC;QACrE,CAAC;aAAM,IAAI,OAAO,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;YACxC,OAAO,GAAG;gBACR,GAAG,OAAO;gBACV,aAAa,EAAE,OAAO,CAAC,aAAa,KAAK,QAAQ,IAAI,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS;gBAC/F,SAAS,EAAE,SAAS,EAAE;aACvB,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,OAAO,OAAO,CAAC;QACjB,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QAC3B,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACrB,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,EAAU;QAC/B,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAClC,IAAI,CAAC,OAAO;YAAE,MAAM,IAAI,KAAK,CAAC,kBAAkB,EAAE,EAAE,CAAC,CAAC;QACtD,IAAI,OAAO,CAAC,MAAM,KAAK,SAAS,IAAI,CAAC,OAAO,CAAC,aAAa;YAAE,OAAO,OAAO,CAAC;QAC3E,MAAM,MAAM,GAAG,OAAO,CAAC,aAAa,KAAK,QAAQ,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,WAAW,CAAC;QAC9E,MAAM,EAAE,aAAa,EAAE,QAAQ,EAAE,GAAG,IAAI,EAAE,GAAG,OAAO,CAAC;QACrD,MAAM,OAAO,GAAc,EAAE,GAAG,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,EAAE,CAAC;QACvE,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QAC3B,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACrB,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,EAAU;QACrB,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,OAAO,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;IACxF,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,KAAa,EAAE,IAAkB,EAAE,OAAmB,EAAE;QACxE,MAAM,KAAK,GAAa;YACtB,EAAE,EAAE,UAAU,EAAE;YAChB,KAAK;YACL,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,KAAK,KAAK,KAAK,CAAC,CAAC,MAAM,GAAG,CAAC;YACjF,IAAI;YACJ,SAAS,EAAE,SAAS,EAAE;YACtB,IAAI;SACL,CAAC;QACF,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACxB,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACrB,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,KAAa;QAC5B,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,KAAK,KAAK,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,GAAG,CAAC,CAAC,QAAQ,CAAC,CAAC;IACtG,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,KAA6D;QAChF,MAAM,GAAG,GAAG,SAAS,EAAE,CAAC;QACxB,MAAM,MAAM,GAAmB;YAC7B,GAAG,KAAK;YACR,EAAE,EAAE,UAAU,EAAE;YAChB,SAAS,EAAE,GAAG;YACd,SAAS,EAAE,GAAG;SACf,CAAC;QACF,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;QACtC,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACrB,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,EAAU,EAAE,KAAwD;QACvF,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACvC,IAAI,CAAC,OAAO;YAAE,MAAM,IAAI,KAAK,CAAC,wBAAwB,EAAE,EAAE,CAAC,CAAC;QAC5D,MAAM,OAAO,GAAmB,EAAE,GAAG,OAAO,EAAE,GAAG,KAAK,EAAE,SAAS,EAAE,SAAS,EAAE,EAAE,CAAC;QACjF,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QAChC,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACrB,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,KAAa;QAC/B,OAAO,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,KAAK,KAAK,KAAK,CAAC,CAAC;IACrF,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,KAAiD;QACtE,MAAM,UAAU,GAAqB;YACnC,GAAG,KAAK;YACR,EAAE,EAAE,UAAU,EAAE;YAChB,SAAS,EAAE,SAAS,EAAE;SACvB,CAAC;QACF,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAClC,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACrB,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,KAAa;QACjC,OAAO,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,KAAK,KAAK,KAAK,CAAC,CAAC;IAC7E,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,UAAsC,EAAE;QAC/D,MAAM,SAAS,GAAG,OAAO,CAAC,mBAAmB,KAAK,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,OAAO,CAAC,mBAAmB,EAAE,CAAC,CAAC,CAAC;QAChH,MAAM,KAAK,GAAG,eAAe,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAC3C,MAAM,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QACnF,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAU,CAAC;QAC3C,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;YAClB,MAAM,MAAM,GAAG,IAAI,GAAG,EAA8B,CAAC;YACrD,KAAK,MAAM,UAAU,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;gBAC1C,IAAI,OAAO,CAAC,KAAK,IAAI,UAAU,CAAC,KAAK,KAAK,OAAO,CAAC,KAAK;oBAAE,SAAS;gBAClE,IAAI,WAAW,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC;oBAAE,SAAS;gBAC/D,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;gBAC5C,IAAI,CAAC,GAAG,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC;oBAAE,SAAS;gBAC3D,MAAM,GAAG,GAAG,GAAG,UAAU,CAAC,KAAK,KAAK,UAAU,CAAC,IAAI,EAAE,CAAC;gBACtD,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;gBACpC,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;gBACvB,MAAM,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;YACzB,CAAC;YACD,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC;gBACpC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;gBACzF,KAAK,MAAM,UAAU,IAAI,KAAK,CAAC,KAAK,CAAC,SAAS,CAAC;oBAAE,gBAAgB,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;YACvF,CAAC;QACH,CAAC;QAED,MAAM,kBAAkB,GAAG,IAAI,GAAG,EAAU,CAAC;QAC7C,KAAK,MAAM,UAAU,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YAC1C,IAAI,OAAO,CAAC,KAAK,IAAI,UAAU,CAAC,KAAK,KAAK,OAAO,CAAC,KAAK;gBAAE,SAAS;YAClE,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YAC5C,IAAI,CAAC,GAAG,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC;gBAAE,SAAS;YAC7D,MAAM,KAAK,GAAG,uBAAuB,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;YACpD,IAAI,KAAK,KAAK,SAAS;gBAAE,SAAS;YAClC,IAAI,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,KAAK,GAAG,KAAK;gBAAE,kBAAkB,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;QACxF,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,GAAG,EAAqC,CAAC;QAC7D,KAAK,MAAM,EAAE,IAAI,gBAAgB;YAAE,OAAO,CAAC,GAAG,CAAC,EAAE,EAAE,mBAAmB,CAAC,CAAC;QACxE,KAAK,MAAM,EAAE,IAAI,kBAAkB;YAAE,OAAO,CAAC,GAAG,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;QAErE,MAAM,QAAQ,GAA0C,+BAA+B,EAAE,CAAC;QAC1F,IAAI,aAAa,GAAG,CAAC,CAAC;QACtB,IAAI,cAAc,GAAG,CAAC,CAAC;QACvB,KAAK,MAAM,UAAU,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YAC1C,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;YAC1C,IAAI,CAAC,MAAM;gBAAE,SAAS;YACtB,MAAM,KAAK,GAAG,qBAAqB,CAAC,UAAU,CAAC,CAAC;YAChD,aAAa,IAAI,CAAC,CAAC;YACnB,cAAc,IAAI,KAAK,CAAC;YACxB,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC;YAC3B,QAAQ,CAAC,MAAM,CAAC,CAAC,KAAK,IAAI,KAAK,CAAC;QAClC,CAAC;QAED,IAAI,CAAC,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YACxC,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC;YACxF,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACvB,CAAC;QAED,OAAO;YACL,MAAM,EAAE,OAAO,CAAC,MAAM,KAAK,IAAI;YAC/B,aAAa;YACb,cAAc;YACd,UAAU,EAAE,OAAO,CAAC,MAAM,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa;YACvD,WAAW,EAAE,OAAO,CAAC,MAAM,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc;YACzD,QAAQ;SACT,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,KAA8B;QACjD,MAAM,QAAQ,GAAG,oBAAoB,CAAC,KAAK,CAAC,CAAC;QAC7C,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,EAAE,QAAQ,CAAC,CAAC;QAC3C,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACrB,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,EAAU,EAAE,KAA8B;QAC7D,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACxC,IAAI,CAAC,OAAO;YAAE,MAAM,IAAI,KAAK,CAAC,uBAAuB,EAAE,EAAE,CAAC,CAAC;QAC3D,MAAM,OAAO,GAAG,kBAAkB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QACnD,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QACjC,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACrB,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,KAAa;QAChC,OAAO,CAAC,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,KAAK,KAAK,KAAK,CAAC,CAAC;IACtF,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,UAAgC,EAAE;QACzD,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,IAAI,IAAI,IAAI,EAAE,CAAC;QACtC,MAAM,KAAK,GAAG,aAAa,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAC/C,MAAM,UAAU,GAAG,aAAa,CAAC,OAAO,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;QAC7D,MAAM,UAAU,GAAG,UAAU,EAAE,CAAC;QAChC,MAAM,cAAc,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG,UAAU,CAAC,CAAC,WAAW,EAAE,CAAC;QAC1E,MAAM,MAAM,GAAyB,EAAE,CAAC;QACxC,MAAM,GAAG,GAAG,CAAC,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC;aACtC,MAAM,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,aAAa,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;aAClD,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,aAAa,CAAC,aAAa,CAAC,CAAC,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;aACxG,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;QACnB,KAAK,MAAM,QAAQ,IAAI,GAAG,EAAE,CAAC;YAC3B,MAAM,OAAO,GAAuB;gBAClC,GAAG,QAAQ;gBACX,MAAM,EAAE,SAAS;gBACjB,UAAU;gBACV,cAAc;gBACd,YAAY,EAAE,QAAQ,CAAC,YAAY,GAAG,CAAC;gBACvC,SAAS,EAAE,SAAS,EAAE;aACvB,CAAC;YACF,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YACzC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACvB,CAAC;QACD,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC;YAAE,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QAC5C,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,oBAAoB,CAAC,EAAU,EAAE,UAAkB,EAAE,QAAkC,EAAE;QAC7F,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACxC,IAAI,CAAC,OAAO;YAAE,MAAM,IAAI,KAAK,CAAC,uBAAuB,EAAE,EAAE,CAAC,CAAC;QAC3D,IAAI,OAAO,CAAC,MAAM,KAAK,MAAM;YAAE,OAAO,OAAO,CAAC;QAC9C,IAAI,OAAO,CAAC,UAAU,KAAK,UAAU;YAAE,MAAM,IAAI,KAAK,CAAC,4BAA4B,EAAE,EAAE,CAAC,CAAC;QACzF,MAAM,GAAG,GAAG,SAAS,EAAE,CAAC;QACxB,MAAM,OAAO,GAAuB;YAClC,GAAG,OAAO;YACV,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,KAAK,CAAC,OAAO,IAAI,OAAO,CAAC,OAAO;YACzC,MAAM,EAAE,GAAG;YACX,SAAS,EAAE,GAAG;SACf,CAAC;QACF,OAAO,OAAO,CAAC,UAAU,CAAC;QAC1B,OAAO,OAAO,CAAC,cAAc,CAAC;QAC9B,OAAO,OAAO,CAAC,SAAS,CAAC;QACzB,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QACjC,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACrB,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,EAAU,EAAE,UAAkB,EAAE,OAA6B;QAClF,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACxC,IAAI,CAAC,OAAO;YAAE,MAAM,IAAI,KAAK,CAAC,uBAAuB,EAAE,EAAE,CAAC,CAAC;QAC3D,IAAI,OAAO,CAAC,UAAU,KAAK,UAAU;YAAE,MAAM,IAAI,KAAK,CAAC,4BAA4B,EAAE,EAAE,CAAC,CAAC;QACzF,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,IAAI,IAAI,IAAI,EAAE,CAAC;QACtC,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,KAAK,IAAI,IAAI,OAAO,CAAC,YAAY,IAAI,OAAO,CAAC,WAAW,CAAC;QAC5F,MAAM,OAAO,GAAuB;YAClC,GAAG,OAAO;YACV,MAAM,EAAE,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS;YACxC,aAAa,EAAE,SAAS;gBACtB,CAAC,CAAC,OAAO,CAAC,aAAa;gBACvB,CAAC,CAAC,IAAI,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG,aAAa,CAAC,OAAO,CAAC,YAAY,EAAE,yBAAyB,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE;YAChI,SAAS,EAAE,OAAO,CAAC,KAAK;YACxB,SAAS,EAAE,SAAS,EAAE;SACvB,CAAC;QACF,IAAI,SAAS;YAAE,OAAO,CAAC,QAAQ,GAAG,GAAG,CAAC,WAAW,EAAE,CAAC;QACpD,OAAO,OAAO,CAAC,UAAU,CAAC;QAC1B,OAAO,OAAO,CAAC,cAAc,CAAC;QAC9B,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QACjC,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACrB,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,wBAAwB,CAAC,MAAY,IAAI,IAAI,EAAE;QACnD,MAAM,SAAS,GAAyB,EAAE,CAAC;QAC3C,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,EAAE,CAAC;YAChD,IAAI,QAAQ,CAAC,MAAM,KAAK,SAAS,IAAI,CAAC,QAAQ,CAAC,cAAc,IAAI,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,cAAc,CAAC,GAAG,GAAG,CAAC,OAAO,EAAE;gBAAE,SAAS;YAC/H,MAAM,OAAO,GAAuB;gBAClC,GAAG,QAAQ;gBACX,MAAM,EAAE,SAAS;gBACjB,aAAa,EAAE,GAAG,CAAC,WAAW,EAAE;gBAChC,SAAS,EAAE,SAAS,EAAE;aACvB,CAAC;YACF,OAAO,OAAO,CAAC,UAAU,CAAC;YAC1B,OAAO,OAAO,CAAC,cAAc,CAAC;YAC9B,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YACzC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC1B,CAAC;QACD,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC;YAAE,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QAC/C,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,KAAK,CAAC,qBAAqB,CAAC,KAAa,EAAE,GAAW,EAAE,QAAiB;QACvE,MAAM,SAAS,GAAG,GAAG,KAAK,IAAI,GAAG,EAAE,CAAC;QACpC,IAAI,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,SAAS,CAAC;YAAE,OAAO,KAAK,CAAC;QAClD,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAChC,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACrB,OAAO,IAAI,CAAC;IACd,CAAC;CACF"}
@@ -0,0 +1,33 @@
1
+ import type { JsonObject } from "@assemblyline-agents/core";
2
+ import type { SandboxSessionInput, SandboxSessionPatch, SandboxSessionRecord, SandboxSessionStore, SandboxSnapshotRecord, SandboxSyncFailureInput, SandboxSyncJobInput, SandboxSyncJobPatch, SandboxSyncJobRecord, SandboxSyncLeaseOptions } from "./state.js";
3
+ export declare class InMemorySandboxSessionStore implements SandboxSessionStore {
4
+ private readonly persist;
5
+ sandboxSnapshots: SandboxSnapshotRecord[];
6
+ sandboxSessions: Map<string, SandboxSessionRecord>;
7
+ sandboxSyncJobs: Map<string, SandboxSyncJobRecord>;
8
+ constructor(persist?: () => Promise<void>);
9
+ recordSandboxSnapshot(input: {
10
+ runId: string;
11
+ sandboxId: string;
12
+ snapshotId: string;
13
+ metadata?: JsonObject;
14
+ }): Promise<SandboxSnapshotRecord>;
15
+ upsertSandboxSession(input: SandboxSessionInput): Promise<SandboxSessionRecord>;
16
+ getSandboxSession(sessionKey: string): Promise<SandboxSessionRecord | undefined>;
17
+ listSandboxSessions(options?: {
18
+ dirty?: boolean;
19
+ }): Promise<SandboxSessionRecord[]>;
20
+ updateSandboxSession(sessionKey: string, patch: SandboxSessionPatch): Promise<SandboxSessionRecord>;
21
+ createSandboxSyncJob(input: SandboxSyncJobInput): Promise<SandboxSyncJobRecord>;
22
+ enqueueSandboxSyncJob(input: SandboxSyncJobInput): Promise<SandboxSyncJobRecord>;
23
+ updateSandboxSyncJob(id: string, patch: SandboxSyncJobPatch): Promise<SandboxSyncJobRecord>;
24
+ listSandboxSyncJobs(runId?: string): Promise<SandboxSyncJobRecord[]>;
25
+ leaseSandboxSyncJobs(options?: SandboxSyncLeaseOptions): Promise<SandboxSyncJobRecord[]>;
26
+ completeSandboxSyncJob(id: string, leaseToken: string, patch?: {
27
+ manifest?: JsonObject;
28
+ metadata?: JsonObject;
29
+ }): Promise<SandboxSyncJobRecord>;
30
+ failSandboxSyncJob(id: string, leaseToken: string, failure: SandboxSyncFailureInput): Promise<SandboxSyncJobRecord>;
31
+ recoverExpiredSandboxSyncJobs(now?: Date): Promise<SandboxSyncJobRecord[]>;
32
+ }
33
+ //# sourceMappingURL=in-memory-sandbox-session-store.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"in-memory-sandbox-session-store.d.ts","sourceRoot":"","sources":["../src/in-memory-sandbox-session-store.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,2BAA2B,CAAC;AAU5D,OAAO,KAAK,EACV,mBAAmB,EACnB,mBAAmB,EACnB,oBAAoB,EACpB,mBAAmB,EACnB,qBAAqB,EACrB,uBAAuB,EACvB,mBAAmB,EACnB,mBAAmB,EACnB,oBAAoB,EACpB,uBAAuB,EACxB,MAAM,YAAY,CAAC;AAEpB,qBAAa,2BAA4B,YAAW,mBAAmB;IAKzD,OAAO,CAAC,QAAQ,CAAC,OAAO;IAJpC,gBAAgB,EAAE,qBAAqB,EAAE,CAAM;IAC/C,eAAe,oCAA2C;IAC1D,eAAe,oCAA2C;gBAE7B,OAAO,GAAE,MAAM,OAAO,CAAC,IAAI,CAAa;IAE/D,qBAAqB,CAAC,KAAK,EAAE;QACjC,KAAK,EAAE,MAAM,CAAC;QACd,SAAS,EAAE,MAAM,CAAC;QAClB,UAAU,EAAE,MAAM,CAAC;QACnB,QAAQ,CAAC,EAAE,UAAU,CAAC;KACvB,GAAG,OAAO,CAAC,qBAAqB,CAAC;IAc5B,oBAAoB,CAAC,KAAK,EAAE,mBAAmB,GAAG,OAAO,CAAC,oBAAoB,CAAC;IAe/E,iBAAiB,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,oBAAoB,GAAG,SAAS,CAAC;IAIhF,mBAAmB,CAAC,OAAO,GAAE;QAAE,KAAK,CAAC,EAAE,OAAO,CAAA;KAAO,GAAG,OAAO,CAAC,oBAAoB,EAAE,CAAC;IAMvF,oBAAoB,CAAC,UAAU,EAAE,MAAM,EAAE,KAAK,EAAE,mBAAmB,GAAG,OAAO,CAAC,oBAAoB,CAAC;IAanG,oBAAoB,CAAC,KAAK,EAAE,mBAAmB,GAAG,OAAO,CAAC,oBAAoB,CAAC;IAI/E,qBAAqB,CAAC,KAAK,EAAE,mBAAmB,GAAG,OAAO,CAAC,oBAAoB,CAAC;IAShF,oBAAoB,CACxB,EAAE,EAAE,MAAM,EACV,KAAK,EAAE,mBAAmB,GACzB,OAAO,CAAC,oBAAoB,CAAC;IAS1B,mBAAmB,CAAC,KAAK,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,oBAAoB,EAAE,CAAC;IAMpE,oBAAoB,CAAC,OAAO,GAAE,uBAA4B,GAAG,OAAO,CAAC,oBAAoB,EAAE,CAAC;IA4B5F,sBAAsB,CAC1B,EAAE,EAAE,MAAM,EACV,UAAU,EAAE,MAAM,EAClB,KAAK,GAAE;QAAE,QAAQ,CAAC,EAAE,UAAU,CAAC;QAAC,QAAQ,CAAC,EAAE,UAAU,CAAA;KAAO,GAC3D,OAAO,CAAC,oBAAoB,CAAC;IAsB1B,kBAAkB,CACtB,EAAE,EAAE,MAAM,EACV,UAAU,EAAE,MAAM,EAClB,OAAO,EAAE,uBAAuB,GAC/B,OAAO,CAAC,oBAAoB,CAAC;IAwB1B,6BAA6B,CAAC,GAAG,GAAE,IAAiB,GAAG,OAAO,CAAC,oBAAoB,EAAE,CAAC;CAkB7F"}
@@ -0,0 +1,186 @@
1
+ import { randomUUID } from "node:crypto";
2
+ import { applySandboxSyncJobPatch, createSandboxSyncJobRecord, isSandboxSyncJobDue, noPersist, positiveLimit, sandboxSyncRetryDelayMs, timestamp } from "./state-store-helpers.js";
3
+ export class InMemorySandboxSessionStore {
4
+ persist;
5
+ sandboxSnapshots = [];
6
+ sandboxSessions = new Map();
7
+ sandboxSyncJobs = new Map();
8
+ constructor(persist = noPersist) {
9
+ this.persist = persist;
10
+ }
11
+ async recordSandboxSnapshot(input) {
12
+ const record = {
13
+ id: randomUUID(),
14
+ runId: input.runId,
15
+ sandboxId: input.sandboxId,
16
+ snapshotId: input.snapshotId,
17
+ metadata: input.metadata ?? {},
18
+ createdAt: timestamp()
19
+ };
20
+ this.sandboxSnapshots.push(record);
21
+ await this.persist();
22
+ return record;
23
+ }
24
+ async upsertSandboxSession(input) {
25
+ const current = this.sandboxSessions.get(input.sessionKey);
26
+ const now = timestamp();
27
+ const record = {
28
+ ...(current ?? {}),
29
+ ...input,
30
+ id: current?.id ?? input.id ?? randomUUID(),
31
+ createdAt: current?.createdAt ?? input.createdAt ?? now,
32
+ updatedAt: input.updatedAt ?? now
33
+ };
34
+ this.sandboxSessions.set(record.sessionKey, record);
35
+ await this.persist();
36
+ return record;
37
+ }
38
+ async getSandboxSession(sessionKey) {
39
+ return this.sandboxSessions.get(sessionKey);
40
+ }
41
+ async listSandboxSessions(options = {}) {
42
+ return [...this.sandboxSessions.values()]
43
+ .filter((session) => options.dirty === undefined || session.dirty === options.dirty)
44
+ .sort((a, b) => a.updatedAt.localeCompare(b.updatedAt));
45
+ }
46
+ async updateSandboxSession(sessionKey, patch) {
47
+ const current = this.sandboxSessions.get(sessionKey);
48
+ if (!current)
49
+ throw new Error(`Sandbox session not found: ${sessionKey}`);
50
+ const updated = { ...current, updatedAt: timestamp() };
51
+ for (const [key, value] of Object.entries(patch)) {
52
+ if (value === undefined)
53
+ delete updated[key];
54
+ else
55
+ updated[key] = value;
56
+ }
57
+ this.sandboxSessions.set(sessionKey, updated);
58
+ await this.persist();
59
+ return updated;
60
+ }
61
+ async createSandboxSyncJob(input) {
62
+ return this.enqueueSandboxSyncJob(input);
63
+ }
64
+ async enqueueSandboxSyncJob(input) {
65
+ const existing = [...this.sandboxSyncJobs.values()].find((job) => job.idempotencyKey === input.idempotencyKey);
66
+ if (existing)
67
+ return existing;
68
+ const record = createSandboxSyncJobRecord(input);
69
+ this.sandboxSyncJobs.set(record.id, record);
70
+ await this.persist();
71
+ return record;
72
+ }
73
+ async updateSandboxSyncJob(id, patch) {
74
+ const current = this.sandboxSyncJobs.get(id);
75
+ if (!current)
76
+ throw new Error(`Sandbox sync job not found: ${id}`);
77
+ const updated = applySandboxSyncJobPatch(current, patch);
78
+ this.sandboxSyncJobs.set(id, updated);
79
+ await this.persist();
80
+ return updated;
81
+ }
82
+ async listSandboxSyncJobs(runId) {
83
+ const jobs = [...this.sandboxSyncJobs.values()];
84
+ return (runId ? jobs.filter((job) => job.runId === runId) : jobs)
85
+ .sort((a, b) => a.createdAt.localeCompare(b.createdAt));
86
+ }
87
+ async leaseSandboxSyncJobs(options = {}) {
88
+ const now = options.now ?? new Date();
89
+ const limit = positiveLimit(options.limit, 10);
90
+ const leaseForMs = positiveLimit(options.leaseForMs, 5 * 60_000);
91
+ const leaseToken = randomUUID();
92
+ const leaseExpiresAt = new Date(now.getTime() + leaseForMs).toISOString();
93
+ const leased = [];
94
+ const due = [...this.sandboxSyncJobs.values()]
95
+ .filter((job) => isSandboxSyncJobDue(job, now))
96
+ .sort((a, b) => a.nextAttemptAt.localeCompare(b.nextAttemptAt) || a.createdAt.localeCompare(b.createdAt))
97
+ .slice(0, limit);
98
+ for (const job of due) {
99
+ const updated = {
100
+ ...job,
101
+ status: "running",
102
+ leaseToken,
103
+ leaseExpiresAt,
104
+ attemptCount: job.attemptCount + 1,
105
+ updatedAt: timestamp()
106
+ };
107
+ delete updated.error;
108
+ this.sandboxSyncJobs.set(updated.id, updated);
109
+ leased.push(updated);
110
+ }
111
+ if (leased.length > 0)
112
+ await this.persist();
113
+ return leased;
114
+ }
115
+ async completeSandboxSyncJob(id, leaseToken, patch = {}) {
116
+ const current = this.sandboxSyncJobs.get(id);
117
+ if (!current)
118
+ throw new Error(`Sandbox sync job not found: ${id}`);
119
+ if (current.status === "completed")
120
+ return current;
121
+ if (current.leaseToken !== leaseToken)
122
+ throw new Error(`Sandbox sync job lease mismatch: ${id}`);
123
+ const now = timestamp();
124
+ const updated = {
125
+ ...current,
126
+ status: "completed",
127
+ manifest: patch.manifest ?? current.manifest,
128
+ metadata: patch.metadata ?? current.metadata,
129
+ completedAt: now,
130
+ updatedAt: now
131
+ };
132
+ delete updated.leaseToken;
133
+ delete updated.leaseExpiresAt;
134
+ delete updated.error;
135
+ this.sandboxSyncJobs.set(id, updated);
136
+ await this.persist();
137
+ return updated;
138
+ }
139
+ async failSandboxSyncJob(id, leaseToken, failure) {
140
+ const current = this.sandboxSyncJobs.get(id);
141
+ if (!current)
142
+ throw new Error(`Sandbox sync job not found: ${id}`);
143
+ if (current.leaseToken !== leaseToken)
144
+ throw new Error(`Sandbox sync job lease mismatch: ${id}`);
145
+ const now = failure.now ?? new Date();
146
+ const permanent = current.attemptCount >= current.maxAttempts;
147
+ const updated = {
148
+ ...current,
149
+ status: permanent ? "blocked_requires_operator" : "pending",
150
+ nextAttemptAt: permanent
151
+ ? current.nextAttemptAt
152
+ : new Date(now.getTime() + positiveLimit(failure.retryDelayMs, sandboxSyncRetryDelayMs(current.attemptCount))).toISOString(),
153
+ lastError: failure.error,
154
+ error: failure.error,
155
+ updatedAt: timestamp()
156
+ };
157
+ if (permanent)
158
+ updated.failedAt = now.toISOString();
159
+ delete updated.leaseToken;
160
+ delete updated.leaseExpiresAt;
161
+ this.sandboxSyncJobs.set(id, updated);
162
+ await this.persist();
163
+ return updated;
164
+ }
165
+ async recoverExpiredSandboxSyncJobs(now = new Date()) {
166
+ const recovered = [];
167
+ for (const job of this.sandboxSyncJobs.values()) {
168
+ if (job.status !== "running" || !job.leaseExpiresAt || Date.parse(job.leaseExpiresAt) > now.getTime())
169
+ continue;
170
+ const updated = {
171
+ ...job,
172
+ status: "pending",
173
+ nextAttemptAt: now.toISOString(),
174
+ updatedAt: timestamp()
175
+ };
176
+ delete updated.leaseToken;
177
+ delete updated.leaseExpiresAt;
178
+ this.sandboxSyncJobs.set(updated.id, updated);
179
+ recovered.push(updated);
180
+ }
181
+ if (recovered.length > 0)
182
+ await this.persist();
183
+ return recovered;
184
+ }
185
+ }
186
+ //# sourceMappingURL=in-memory-sandbox-session-store.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"in-memory-sandbox-session-store.js","sourceRoot":"","sources":["../src/in-memory-sandbox-session-store.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAEzC,OAAO,EACL,wBAAwB,EACxB,0BAA0B,EAC1B,mBAAmB,EACnB,SAAS,EACT,aAAa,EACb,uBAAuB,EACvB,SAAS,EACV,MAAM,0BAA0B,CAAC;AAclC,MAAM,OAAO,2BAA2B;IAKT;IAJ7B,gBAAgB,GAA4B,EAAE,CAAC;IAC/C,eAAe,GAAG,IAAI,GAAG,EAAgC,CAAC;IAC1D,eAAe,GAAG,IAAI,GAAG,EAAgC,CAAC;IAE1D,YAA6B,UAA+B,SAAS;QAAxC,YAAO,GAAP,OAAO,CAAiC;IAAG,CAAC;IAEzE,KAAK,CAAC,qBAAqB,CAAC,KAK3B;QACC,MAAM,MAAM,GAA0B;YACpC,EAAE,EAAE,UAAU,EAAE;YAChB,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,UAAU,EAAE,KAAK,CAAC,UAAU;YAC5B,QAAQ,EAAE,KAAK,CAAC,QAAQ,IAAI,EAAE;YAC9B,SAAS,EAAE,SAAS,EAAE;SACvB,CAAC;QACF,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACnC,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACrB,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,oBAAoB,CAAC,KAA0B;QACnD,MAAM,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;QAC3D,MAAM,GAAG,GAAG,SAAS,EAAE,CAAC;QACxB,MAAM,MAAM,GAAyB;YACnC,GAAG,CAAC,OAAO,IAAI,EAAE,CAAC;YAClB,GAAG,KAAK;YACR,EAAE,EAAE,OAAO,EAAE,EAAE,IAAI,KAAK,CAAC,EAAE,IAAI,UAAU,EAAE;YAC3C,SAAS,EAAE,OAAO,EAAE,SAAS,IAAI,KAAK,CAAC,SAAS,IAAI,GAAG;YACvD,SAAS,EAAE,KAAK,CAAC,SAAS,IAAI,GAAG;SAClC,CAAC;QACF,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;QACpD,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACrB,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,UAAkB;QACxC,OAAO,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IAC9C,CAAC;IAED,KAAK,CAAC,mBAAmB,CAAC,UAA+B,EAAE;QACzD,OAAO,CAAC,GAAG,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,CAAC;aACtC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,KAAK,SAAS,IAAI,OAAO,CAAC,KAAK,KAAK,OAAO,CAAC,KAAK,CAAC;aACnF,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;IAC5D,CAAC;IAED,KAAK,CAAC,oBAAoB,CAAC,UAAkB,EAAE,KAA0B;QACvE,MAAM,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QACrD,IAAI,CAAC,OAAO;YAAE,MAAM,IAAI,KAAK,CAAC,8BAA8B,UAAU,EAAE,CAAC,CAAC;QAC1E,MAAM,OAAO,GAAyB,EAAE,GAAG,OAAO,EAAE,SAAS,EAAE,SAAS,EAAE,EAAE,CAAC;QAC7E,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAuF,EAAE,CAAC;YACvI,IAAI,KAAK,KAAK,SAAS;gBAAE,OAAQ,OAAyC,CAAC,GAAiC,CAAC,CAAC;;gBACxG,OAA8C,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;QACpE,CAAC;QACD,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QAC9C,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACrB,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,oBAAoB,CAAC,KAA0B;QACnD,OAAO,IAAI,CAAC,qBAAqB,CAAC,KAAK,CAAC,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,qBAAqB,CAAC,KAA0B;QACpD,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,cAAc,KAAK,KAAK,CAAC,cAAc,CAAC,CAAC;QAC/G,IAAI,QAAQ;YAAE,OAAO,QAAQ,CAAC;QAC9B,MAAM,MAAM,GAAG,0BAA0B,CAAC,KAAK,CAAC,CAAC;QACjD,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;QAC5C,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACrB,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,oBAAoB,CACxB,EAAU,EACV,KAA0B;QAE1B,MAAM,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAC7C,IAAI,CAAC,OAAO;YAAE,MAAM,IAAI,KAAK,CAAC,+BAA+B,EAAE,EAAE,CAAC,CAAC;QACnE,MAAM,OAAO,GAAG,wBAAwB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QACzD,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QACtC,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACrB,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,mBAAmB,CAAC,KAAc;QACtC,MAAM,IAAI,GAAG,CAAC,GAAG,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,CAAC,CAAC;QAChD,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,KAAK,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;aAC9D,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;IAC5D,CAAC;IAED,KAAK,CAAC,oBAAoB,CAAC,UAAmC,EAAE;QAC9D,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,IAAI,IAAI,IAAI,EAAE,CAAC;QACtC,MAAM,KAAK,GAAG,aAAa,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAC/C,MAAM,UAAU,GAAG,aAAa,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC,GAAG,MAAM,CAAC,CAAC;QACjE,MAAM,UAAU,GAAG,UAAU,EAAE,CAAC;QAChC,MAAM,cAAc,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG,UAAU,CAAC,CAAC,WAAW,EAAE,CAAC;QAC1E,MAAM,MAAM,GAA2B,EAAE,CAAC;QAC1C,MAAM,GAAG,GAAG,CAAC,GAAG,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,CAAC;aAC3C,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,mBAAmB,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;aAC9C,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,aAAa,CAAC,aAAa,CAAC,CAAC,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;aACxG,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;QACnB,KAAK,MAAM,GAAG,IAAI,GAAG,EAAE,CAAC;YACtB,MAAM,OAAO,GAAyB;gBACpC,GAAG,GAAG;gBACN,MAAM,EAAE,SAAS;gBACjB,UAAU;gBACV,cAAc;gBACd,YAAY,EAAE,GAAG,CAAC,YAAY,GAAG,CAAC;gBAClC,SAAS,EAAE,SAAS,EAAE;aACvB,CAAC;YACF,OAAO,OAAO,CAAC,KAAK,CAAC;YACrB,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YAC9C,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACvB,CAAC;QACD,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC;YAAE,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QAC5C,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,sBAAsB,CAC1B,EAAU,EACV,UAAkB,EAClB,QAA0D,EAAE;QAE5D,MAAM,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAC7C,IAAI,CAAC,OAAO;YAAE,MAAM,IAAI,KAAK,CAAC,+BAA+B,EAAE,EAAE,CAAC,CAAC;QACnE,IAAI,OAAO,CAAC,MAAM,KAAK,WAAW;YAAE,OAAO,OAAO,CAAC;QACnD,IAAI,OAAO,CAAC,UAAU,KAAK,UAAU;YAAE,MAAM,IAAI,KAAK,CAAC,oCAAoC,EAAE,EAAE,CAAC,CAAC;QACjG,MAAM,GAAG,GAAG,SAAS,EAAE,CAAC;QACxB,MAAM,OAAO,GAAyB;YACpC,GAAG,OAAO;YACV,MAAM,EAAE,WAAW;YACnB,QAAQ,EAAE,KAAK,CAAC,QAAQ,IAAI,OAAO,CAAC,QAAQ;YAC5C,QAAQ,EAAE,KAAK,CAAC,QAAQ,IAAI,OAAO,CAAC,QAAQ;YAC5C,WAAW,EAAE,GAAG;YAChB,SAAS,EAAE,GAAG;SACf,CAAC;QACF,OAAO,OAAO,CAAC,UAAU,CAAC;QAC1B,OAAO,OAAO,CAAC,cAAc,CAAC;QAC9B,OAAO,OAAO,CAAC,KAAK,CAAC;QACrB,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QACtC,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACrB,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,kBAAkB,CACtB,EAAU,EACV,UAAkB,EAClB,OAAgC;QAEhC,MAAM,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAC7C,IAAI,CAAC,OAAO;YAAE,MAAM,IAAI,KAAK,CAAC,+BAA+B,EAAE,EAAE,CAAC,CAAC;QACnE,IAAI,OAAO,CAAC,UAAU,KAAK,UAAU;YAAE,MAAM,IAAI,KAAK,CAAC,oCAAoC,EAAE,EAAE,CAAC,CAAC;QACjG,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,IAAI,IAAI,IAAI,EAAE,CAAC;QACtC,MAAM,SAAS,GAAG,OAAO,CAAC,YAAY,IAAI,OAAO,CAAC,WAAW,CAAC;QAC9D,MAAM,OAAO,GAAyB;YACpC,GAAG,OAAO;YACV,MAAM,EAAE,SAAS,CAAC,CAAC,CAAC,2BAA2B,CAAC,CAAC,CAAC,SAAS;YAC3D,aAAa,EAAE,SAAS;gBACtB,CAAC,CAAC,OAAO,CAAC,aAAa;gBACvB,CAAC,CAAC,IAAI,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG,aAAa,CAAC,OAAO,CAAC,YAAY,EAAE,uBAAuB,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE;YAC9H,SAAS,EAAE,OAAO,CAAC,KAAK;YACxB,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,SAAS,EAAE,SAAS,EAAE;SACvB,CAAC;QACF,IAAI,SAAS;YAAE,OAAO,CAAC,QAAQ,GAAG,GAAG,CAAC,WAAW,EAAE,CAAC;QACpD,OAAO,OAAO,CAAC,UAAU,CAAC;QAC1B,OAAO,OAAO,CAAC,cAAc,CAAC;QAC9B,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QACtC,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACrB,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,6BAA6B,CAAC,MAAY,IAAI,IAAI,EAAE;QACxD,MAAM,SAAS,GAA2B,EAAE,CAAC;QAC7C,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,EAAE,CAAC;YAChD,IAAI,GAAG,CAAC,MAAM,KAAK,SAAS,IAAI,CAAC,GAAG,CAAC,cAAc,IAAI,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,cAAc,CAAC,GAAG,GAAG,CAAC,OAAO,EAAE;gBAAE,SAAS;YAChH,MAAM,OAAO,GAAyB;gBACpC,GAAG,GAAG;gBACN,MAAM,EAAE,SAAS;gBACjB,aAAa,EAAE,GAAG,CAAC,WAAW,EAAE;gBAChC,SAAS,EAAE,SAAS,EAAE;aACvB,CAAC;YACF,OAAO,OAAO,CAAC,UAAU,CAAC;YAC1B,OAAO,OAAO,CAAC,cAAc,CAAC;YAC9B,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YAC9C,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC1B,CAAC;QACD,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC;YAAE,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QAC/C,OAAO,SAAS,CAAC;IACnB,CAAC;CACF"}