@overkill-dev/run 0.0.63 → 0.0.65

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 (310) hide show
  1. package/assertion-protocol/source-location-forwarding.js +14 -2
  2. package/assertion-protocol/source-location-forwarding.js.map +1 -1
  3. package/assertion-protocol/source-location.js +1 -0
  4. package/assertion-protocol/source-location.js.map +1 -1
  5. package/attachments/attachment-context.js +31 -0
  6. package/attachments/attachment-context.js.map +1 -0
  7. package/engine/async-leak-diagnostics.js +9 -9
  8. package/engine/async-leak-diagnostics.js.map +1 -1
  9. package/engine/case-attempt-execution.js +191 -0
  10. package/engine/case-attempt-execution.js.map +1 -0
  11. package/engine/case-execution.d.ts +5 -3
  12. package/engine/case-execution.d.ts.map +1 -1
  13. package/engine/case-execution.js +25 -24
  14. package/engine/case-execution.js.map +1 -1
  15. package/engine/coverage-artifact.d.ts +1 -0
  16. package/engine/coverage-artifact.d.ts.map +1 -1
  17. package/engine/execution-async-leak-policy.js +3 -10
  18. package/engine/execution-async-leak-policy.js.map +1 -1
  19. package/engine/execution-global-error-observer.d.ts +2 -1
  20. package/engine/execution-global-error-observer.d.ts.map +1 -1
  21. package/engine/execution-global-error-observer.js +19 -15
  22. package/engine/execution-global-error-observer.js.map +1 -1
  23. package/engine/execution-options.d.ts +4 -1
  24. package/engine/execution-options.d.ts.map +1 -1
  25. package/engine/execution-options.js +7 -2
  26. package/engine/execution-options.js.map +1 -1
  27. package/engine/execution-resource-budget-tracking.js.map +1 -1
  28. package/engine/execution-resource-tracked-cases.js.map +1 -1
  29. package/engine/execution-result.js.map +1 -1
  30. package/engine/execution-supervision-state.js.map +1 -1
  31. package/engine/execution-supervision.d.ts.map +1 -1
  32. package/engine/execution-supervision.js +50 -33
  33. package/engine/execution-supervision.js.map +1 -1
  34. package/engine/execution-test-plan-cases.js +6 -126
  35. package/engine/execution-test-plan-cases.js.map +1 -1
  36. package/engine/execution-test-plan-resource-usage.js.map +1 -1
  37. package/engine/execution.js +3 -3
  38. package/engine/execution.js.map +1 -1
  39. package/engine/identity.d.ts +3 -0
  40. package/engine/identity.d.ts.map +1 -1
  41. package/engine/reporter-dispatcher.js +1 -0
  42. package/engine/reporter-dispatcher.js.map +1 -1
  43. package/engine/reporter-event-queue.js.map +1 -1
  44. package/engine/reporter.d.ts +2 -1
  45. package/engine/reporter.d.ts.map +1 -1
  46. package/engine/reporter.js.map +1 -1
  47. package/engine/resource-usage.d.ts +22 -0
  48. package/engine/resource-usage.d.ts.map +1 -0
  49. package/engine/retry-policy.d.ts +4 -0
  50. package/engine/retry-policy.d.ts.map +1 -0
  51. package/engine/retry-policy.js +22 -0
  52. package/engine/retry-policy.js.map +1 -0
  53. package/engine/run-artifact.d.ts +45 -0
  54. package/engine/run-artifact.d.ts.map +1 -0
  55. package/engine/run-result.d.ts +22 -53
  56. package/engine/run-result.d.ts.map +1 -1
  57. package/engine/run-result.js +14 -0
  58. package/engine/run-result.js.map +1 -1
  59. package/engine/runner-error-reporting.js.map +1 -1
  60. package/engine/runtime-attachment.d.ts +62 -0
  61. package/engine/runtime-attachment.d.ts.map +1 -0
  62. package/engine/runtime-attachment.js +7 -0
  63. package/engine/runtime-attachment.js.map +1 -0
  64. package/engine/test-attempt-history.js +24 -0
  65. package/engine/test-attempt-history.js.map +1 -0
  66. package/package.json +25 -6
  67. package/packages/resources/attachment-context.entry-point.js +2 -0
  68. package/packages/resources/attachment-context.entry-point.js.map +1 -0
  69. package/packages/run/config.entry-point.d.ts +3 -1
  70. package/packages/run/config.entry-point.d.ts.map +1 -1
  71. package/packages/run/run.entry-point.d.ts +3 -1
  72. package/packages/run/run.entry-point.d.ts.map +1 -1
  73. package/packages/run/transcript-store.entry-point.d.ts +3 -0
  74. package/packages/run/transcript-store.entry-point.d.ts.map +1 -0
  75. package/packages/run/transcript-store.entry-point.js +2 -0
  76. package/packages/run/transcript-store.entry-point.js.map +1 -0
  77. package/packages/simulation/transcript.entry-point.js +2 -0
  78. package/packages/simulation/transcript.entry-point.js.map +1 -0
  79. package/readme.md +69 -2
  80. package/reporters/human-reporter-rendering.js +25 -2
  81. package/reporters/human-reporter-rendering.js.map +1 -1
  82. package/reporters/line-reporter.js +6 -1
  83. package/reporters/line-reporter.js.map +1 -1
  84. package/resources/resource-definition-brand.js +1 -1
  85. package/resources/resource-definition-brand.js.map +1 -1
  86. package/resources/resource-definition-types.d.ts.map +1 -1
  87. package/resources/resource-scenario-binding.js +2 -0
  88. package/resources/resource-scenario-binding.js.map +1 -1
  89. package/resources/resource-session.d.ts.map +1 -1
  90. package/resources/resources.d.ts.map +1 -1
  91. package/resources/resources.js +3 -6
  92. package/resources/resources.js.map +1 -1
  93. package/resources/runtime-definition.d.ts.map +1 -1
  94. package/resources/runtime-definition.js +1 -1
  95. package/resources/runtime-definition.js.map +1 -1
  96. package/resources/runtime-matrix-definition.d.ts.map +1 -1
  97. package/resources/runtime-matrix-definition.js +2 -2
  98. package/resources/runtime-matrix-definition.js.map +1 -1
  99. package/run/attachment-connection.js +108 -0
  100. package/run/attachment-connection.js.map +1 -0
  101. package/run/attachment-coordinator-context.js +9 -0
  102. package/run/attachment-coordinator-context.js.map +1 -0
  103. package/run/attachment-coordinator.js +77 -0
  104. package/run/attachment-coordinator.js.map +1 -0
  105. package/run/attachment-eligibility.js +21 -0
  106. package/run/attachment-eligibility.js.map +1 -0
  107. package/run/attachment-execution.js +255 -0
  108. package/run/attachment-execution.js.map +1 -0
  109. package/run/attachment-failure.js +63 -0
  110. package/run/attachment-failure.js.map +1 -0
  111. package/run/attachment-frames.js +22 -0
  112. package/run/attachment-frames.js.map +1 -0
  113. package/run/attachment-json.js +118 -0
  114. package/run/attachment-json.js.map +1 -0
  115. package/run/attachment-protocol.js +5 -0
  116. package/run/attachment-protocol.js.map +1 -0
  117. package/run/attachment-record.js +60 -0
  118. package/run/attachment-record.js.map +1 -0
  119. package/run/attachment-reporter.js +39 -0
  120. package/run/attachment-reporter.js.map +1 -0
  121. package/run/attachment-results.js +115 -0
  122. package/run/attachment-results.js.map +1 -0
  123. package/run/attachment-retention.js +153 -0
  124. package/run/attachment-retention.js.map +1 -0
  125. package/run/attachment-run.js +36 -0
  126. package/run/attachment-run.js.map +1 -0
  127. package/run/attachment-server.js +136 -0
  128. package/run/attachment-server.js.map +1 -0
  129. package/run/attachment-storage-session.js +21 -0
  130. package/run/attachment-storage-session.js.map +1 -0
  131. package/run/attachment-store.js +330 -0
  132. package/run/attachment-store.js.map +1 -0
  133. package/run/attachment-wire-schema.js +123 -0
  134. package/run/attachment-wire-schema.js.map +1 -0
  135. package/run/attachment-worker-context.js +49 -0
  136. package/run/attachment-worker-context.js.map +1 -0
  137. package/run/attachment-writer.js +117 -0
  138. package/run/attachment-writer.js.map +1 -0
  139. package/run/capability-policy.js +10 -5
  140. package/run/capability-policy.js.map +1 -1
  141. package/run/collected-run-plan.js.map +1 -1
  142. package/run/coverage-config-schema.d.ts +4 -4
  143. package/run/coverage-loaded-sources.js +112 -0
  144. package/run/coverage-loaded-sources.js.map +1 -0
  145. package/run/coverage-native-data.js +18 -0
  146. package/run/coverage-native-data.js.map +1 -0
  147. package/run/coverage-paths.js +45 -4
  148. package/run/coverage-paths.js.map +1 -1
  149. package/run/coverage-reporting.js +104 -8
  150. package/run/coverage-reporting.js.map +1 -1
  151. package/run/coverage-run-record.js +50 -0
  152. package/run/coverage-run-record.js.map +1 -0
  153. package/run/coverage-runtime-source.js +37 -0
  154. package/run/coverage-runtime-source.js.map +1 -0
  155. package/run/coverage-session.js +344 -6
  156. package/run/coverage-session.js.map +1 -1
  157. package/run/coverage-source-map.js +125 -0
  158. package/run/coverage-source-map.js.map +1 -0
  159. package/run/coverage-source-selection.js +120 -0
  160. package/run/coverage-source-selection.js.map +1 -0
  161. package/run/duration-history.d.ts.map +1 -1
  162. package/run/duration-history.js +1 -0
  163. package/run/duration-history.js.map +1 -1
  164. package/run/node-run-execution.js +25 -0
  165. package/run/node-run-execution.js.map +1 -0
  166. package/run/node-runtime-state-store.js +38 -0
  167. package/run/node-runtime-state-store.js.map +1 -0
  168. package/run/node-worker-pool-factory.js +0 -31
  169. package/run/node-worker-pool-factory.js.map +1 -1
  170. package/run/path-containment.js +6 -1
  171. package/run/path-containment.js.map +1 -1
  172. package/run/permission-denial-runtime-policy.js +21 -13
  173. package/run/permission-denial-runtime-policy.js.map +1 -1
  174. package/run/recorded-coverage-run.js +45 -0
  175. package/run/recorded-coverage-run.js.map +1 -0
  176. package/run/recorded-coverage-session.js +50 -0
  177. package/run/recorded-coverage-session.js.map +1 -0
  178. package/run/resource-lifecycle-projection.js +6 -2
  179. package/run/resource-lifecycle-projection.js.map +1 -1
  180. package/run/resource-lifecycle-session-acquirer.js +13 -6
  181. package/run/resource-lifecycle-session-acquirer.js.map +1 -1
  182. package/run/resource-lifecycle-startup-budget.js +1 -0
  183. package/run/resource-lifecycle-startup-budget.js.map +1 -1
  184. package/run/resource-lifecycle-state.d.ts +4 -2
  185. package/run/resource-lifecycle-state.d.ts.map +1 -1
  186. package/run/resource-lifecycle-state.js +19 -3
  187. package/run/resource-lifecycle-state.js.map +1 -1
  188. package/run/resource-lifecycle.js +57 -13
  189. package/run/resource-lifecycle.js.map +1 -1
  190. package/run/resource-usage.js.map +1 -1
  191. package/run/retry-artifact-retention.js +37 -0
  192. package/run/retry-artifact-retention.js.map +1 -0
  193. package/run/run-config-schema.d.ts +48 -4
  194. package/run/run-config-schema.d.ts.map +1 -1
  195. package/run/run-config-schema.js +18 -0
  196. package/run/run-config-schema.js.map +1 -1
  197. package/run/run-config.d.ts +9 -6
  198. package/run/run-config.d.ts.map +1 -1
  199. package/run/run-config.js +6 -1
  200. package/run/run-config.js.map +1 -1
  201. package/run/run-coverage.js +5 -2
  202. package/run/run-coverage.js.map +1 -1
  203. package/run/run-engine-primitives.d.ts +13 -1
  204. package/run/run-engine-primitives.d.ts.map +1 -1
  205. package/run/run-errors.d.ts.map +1 -1
  206. package/run/run-errors.js +1 -0
  207. package/run/run-errors.js.map +1 -1
  208. package/run/run-execution-config.d.ts +5 -0
  209. package/run/run-execution-config.d.ts.map +1 -1
  210. package/run/run-facts.js +3 -0
  211. package/run/run-facts.js.map +1 -1
  212. package/run/run-isolated-command.js +3 -0
  213. package/run/run-isolated-command.js.map +1 -1
  214. package/run/run-local-command.js +12 -83
  215. package/run/run-local-command.js.map +1 -1
  216. package/run/run-local-coverage.js +72 -0
  217. package/run/run-local-coverage.js.map +1 -0
  218. package/run/run-local-execution.js +3 -1
  219. package/run/run-local-execution.js.map +1 -1
  220. package/run/run-local-resolution.js +6 -3
  221. package/run/run-local-resolution.js.map +1 -1
  222. package/run/run-orchestrator-dependencies.d.ts +1 -1
  223. package/run/run-orchestrator-dependencies.d.ts.map +1 -1
  224. package/run/run-orchestrator.d.ts.map +1 -1
  225. package/run/run-orchestrator.js +6 -18
  226. package/run/run-orchestrator.js.map +1 -1
  227. package/run/run-record-error.js +26 -0
  228. package/run/run-record-error.js.map +1 -0
  229. package/run/run-record-input.js +34 -0
  230. package/run/run-record-input.js.map +1 -0
  231. package/run/run-record-outcomes.d.ts +49 -0
  232. package/run/run-record-outcomes.d.ts.map +1 -0
  233. package/run/run-record-result.js +58 -0
  234. package/run/run-record-result.js.map +1 -0
  235. package/run/run-record-storage.js +51 -0
  236. package/run/run-record-storage.js.map +1 -0
  237. package/run/run-record-types.d.ts +73 -0
  238. package/run/run-record-types.d.ts.map +1 -0
  239. package/run/run-record-versions.js +51 -0
  240. package/run/run-record-versions.js.map +1 -0
  241. package/run/run-record.js +48 -0
  242. package/run/run-record.js.map +1 -0
  243. package/run/run-result-reporting.js +16 -0
  244. package/run/run-result-reporting.js.map +1 -0
  245. package/run/run-supervised-finalization.js +8 -0
  246. package/run/run-supervised-finalization.js.map +1 -0
  247. package/run/run-supervised-process.js +34 -33
  248. package/run/run-supervised-process.js.map +1 -1
  249. package/run/run-support.js +23 -9
  250. package/run/run-support.js.map +1 -1
  251. package/run/run-types.d.ts +7 -5
  252. package/run/run-types.d.ts.map +1 -1
  253. package/run/run-types.js.map +1 -1
  254. package/run/run-validation.js +9 -0
  255. package/run/run-validation.js.map +1 -1
  256. package/run/runtime-attachment-boundary.js +11 -0
  257. package/run/runtime-attachment-boundary.js.map +1 -0
  258. package/run/supervised-child-test-plan.js +2 -0
  259. package/run/supervised-child-test-plan.js.map +1 -1
  260. package/run/supervised-child.js +12 -9
  261. package/run/supervised-child.js.map +1 -1
  262. package/run/supervised-output-capture.js +66 -0
  263. package/run/supervised-output-capture.js.map +1 -0
  264. package/run/supervised-protocol.js.map +1 -1
  265. package/run/supervised-run-collection.js +3 -1
  266. package/run/supervised-run-collection.js.map +1 -1
  267. package/run/supervised-run-resource-policy.js +12 -1
  268. package/run/supervised-run-resource-policy.js.map +1 -1
  269. package/run/supervised-run-runtime.js +25 -30
  270. package/run/supervised-run-runtime.js.map +1 -1
  271. package/run/supervised-run-state.js +64 -81
  272. package/run/supervised-run-state.js.map +1 -1
  273. package/run/supervised-run.js +14 -4
  274. package/run/supervised-run.js.map +1 -1
  275. package/run/worker-pool-collection.js +1 -0
  276. package/run/worker-pool-collection.js.map +1 -1
  277. package/run/worker-pool-command.js +2 -0
  278. package/run/worker-pool-command.js.map +1 -1
  279. package/run/worker-pool-execution.js +12 -3
  280. package/run/worker-pool-execution.js.map +1 -1
  281. package/run/worker-pool-hedged-arbitration.js +11 -2
  282. package/run/worker-pool-hedged-arbitration.js.map +1 -1
  283. package/run/worker-pool-host-process.js.map +1 -1
  284. package/run/worker-pool-host-protocol.js.map +1 -1
  285. package/run/worker-pool-protocol.js.map +1 -1
  286. package/run/worker-pool-resource-lifecycle-execution.js +1 -0
  287. package/run/worker-pool-resource-lifecycle-execution.js.map +1 -1
  288. package/run/worker-pool-resource-tracking.js +2 -2
  289. package/run/worker-pool-resource-tracking.js.map +1 -1
  290. package/run/worker-pool-results.js +7 -9
  291. package/run/worker-pool-results.js.map +1 -1
  292. package/run/worker-pool-run.js +10 -3
  293. package/run/worker-pool-run.js.map +1 -1
  294. package/run/worker-pool-runtime.js +3 -7
  295. package/run/worker-pool-runtime.js.map +1 -1
  296. package/run/worker-pool-task-events.js +12 -4
  297. package/run/worker-pool-task-events.js.map +1 -1
  298. package/run/worker-pool-worker-plan.js +1 -0
  299. package/run/worker-pool-worker-plan.js.map +1 -1
  300. package/run/worker-pool-worker-resources.js +1 -0
  301. package/run/worker-pool-worker-resources.js.map +1 -1
  302. package/run/worker-pool-worker.js +15 -4
  303. package/run/worker-pool-worker.js.map +1 -1
  304. package/sbom.cdx.json +65 -14
  305. package/transcript/http-transcript.js +3 -0
  306. package/transcript/http-transcript.js.map +1 -0
  307. package/transcript/transcript-store.d.ts +25 -0
  308. package/transcript/transcript-store.d.ts.map +1 -0
  309. package/transcript/transcript-store.js +73 -0
  310. package/transcript/transcript-store.js.map +1 -0
@@ -0,0 +1,330 @@
1
+ import { createHash } from 'node:crypto';
2
+ import { rm } from 'node:fs/promises';
3
+ import path from 'node:path';
4
+ import { workIdentityKey } from "../engine/identity.js";
5
+ import { createStoredRunValue } from "./supervised-run-state.js";
6
+ import { attachmentMetadataBytes } from "./attachment-protocol.js";
7
+ import { createAttachmentRetention } from "./attachment-retention.js";
8
+ function createAttachmentScopeBudget() {
9
+ const bytes = createStoredRunValue(0);
10
+ const count = createStoredRunValue(0);
11
+ function reserve(limits) {
12
+ if (count.read() >= limits.maxScopeAttachments) {
13
+ throw new Error('Attachment count limit exceeded.');
14
+ }
15
+ count.write(count.read() + 1);
16
+ }
17
+ function available(limits) {
18
+ return limits.maxScopeBytes - bytes.read();
19
+ }
20
+ function retained(retainedBytes) {
21
+ bytes.write(bytes.read() + retainedBytes);
22
+ }
23
+ return { reserve, available, retained };
24
+ }
25
+ function storedAttachmentArtifact(state) {
26
+ const { initial, retention } = state;
27
+ return { ...state.initial, payload: { ...initial.payload, content: retention.content() } };
28
+ }
29
+ async function storedAttachmentClose(state, reason) {
30
+ const { closed, retention } = state;
31
+ if (closed.read()) {
32
+ return;
33
+ }
34
+ closed.write(true);
35
+ await retention.finish(reason === 'complete' ? { kind: 'complete' } : { kind: 'incomplete', reason });
36
+ }
37
+ async function storedAttachmentInterrupt(state, reason) {
38
+ const { closed, retention, failed } = state;
39
+ if (closed.read()) {
40
+ return null;
41
+ }
42
+ closed.write(true);
43
+ await retention.finish({ kind: 'incomplete', reason });
44
+ return failed.read() ? null : state.owner;
45
+ }
46
+ function storedAttachmentAssertOpen(state) {
47
+ const { closed, failed } = state;
48
+ if (closed.read() || failed.read()) {
49
+ throw new Error('Attachment writer is closed.');
50
+ }
51
+ }
52
+ function storedAttachmentOmit(state) {
53
+ const { retention } = state;
54
+ storedAttachmentAssertOpen(state);
55
+ retention.omit();
56
+ }
57
+ async function storedAttachmentRetain(state, bytes, limits) {
58
+ const { retention, budget, failed } = state;
59
+ try {
60
+ return await retention.retain(bytes, budget.available(limits));
61
+ }
62
+ catch (error) {
63
+ failed.write(true);
64
+ await storedAttachmentClose(state, 'write-error');
65
+ throw error;
66
+ }
67
+ }
68
+ async function storedAttachmentWrite(state, data, limits) {
69
+ const { budget, initial, failed, retention } = state;
70
+ storedAttachmentAssertOpen(state);
71
+ const bytes = Buffer.from(data, 'base64');
72
+ const retained = await storedAttachmentRetain(state, bytes, limits);
73
+ budget.retained(retained);
74
+ if (retained < bytes.length && initial.payload.content.kind === 'file') {
75
+ failed.write(true);
76
+ await retention.finish({ kind: 'incomplete', reason: 'byte-limit' });
77
+ return { kind: 'error', message: 'Binary attachment byte limit exceeded.', reason: 'byte-limit' };
78
+ }
79
+ return { kind: 'written' };
80
+ }
81
+ function createStoredAttachment(channel, operation, content) {
82
+ const { initial, retention, budget } = content;
83
+ const branch = operation.branch;
84
+ const owner = operation.owner;
85
+ const closed = createStoredRunValue(false);
86
+ const failed = createStoredRunValue(false);
87
+ const state = {
88
+ channel,
89
+ operation,
90
+ initial,
91
+ retention,
92
+ budget,
93
+ branch,
94
+ owner,
95
+ closed,
96
+ failed
97
+ };
98
+ return {
99
+ channel: state.channel,
100
+ branch: state.branch,
101
+ owner: state.owner,
102
+ artifact: storedAttachmentArtifact.bind(null, state),
103
+ close: storedAttachmentClose.bind(null, state),
104
+ interrupt: storedAttachmentInterrupt.bind(null, state),
105
+ omit: storedAttachmentOmit.bind(null, state),
106
+ write: storedAttachmentWrite.bind(null, state)
107
+ };
108
+ }
109
+ function ownerKey(owner) {
110
+ return owner.kind === 'run' ? 'run' : workIdentityKey(owner.work);
111
+ }
112
+ function attachmentId(owner, sequence) {
113
+ if (owner.kind === 'run') {
114
+ return { attempt: null, runtimes: [], scope: { kind: 'run' }, sequence, subtype: 'attachment', workload: null };
115
+ }
116
+ return {
117
+ attempt: owner.attempt,
118
+ runtimes: owner.work.runtimes,
119
+ scope: { activeCases: [owner.work.case], case: owner.work.case, confidence: 'active-case', kind: 'case' },
120
+ sequence,
121
+ subtype: 'attachment',
122
+ workload: owner.work.workload
123
+ };
124
+ }
125
+ const firstPrintableCodePoint = 32;
126
+ const deleteCodePoint = 127;
127
+ function isControlCharacter(character) {
128
+ const code = character.codePointAt(0) ?? 0;
129
+ return code < firstPrintableCodePoint || code === deleteCodePoint;
130
+ }
131
+ function validateMetadata(operation) {
132
+ for (const value of [operation.metadata.name, operation.metadata.mediaType]) {
133
+ if (value.length === 0 || Buffer.byteLength(value) > attachmentMetadataBytes ||
134
+ Array.from(value).some(isControlCharacter)) {
135
+ throw new TypeError('Attachment names and media types require 1 to 256 bytes without control characters.');
136
+ }
137
+ }
138
+ }
139
+ function runAttachmentStoreArtifacts(state) {
140
+ const { writers } = state;
141
+ return Array.from(writers.values(), function retainedArtifact(writer) {
142
+ return writer.artifact();
143
+ });
144
+ }
145
+ function runAttachmentStoreRegisterBranch(state, branch, retain) {
146
+ const { branches } = state;
147
+ branches.set(branch, retain);
148
+ }
149
+ function runAttachmentStoreBranchArtifacts(state, branch) {
150
+ const { writers } = state;
151
+ return Array
152
+ .from(writers.values())
153
+ .filter(function inBranch(writer) {
154
+ return writer.branch === branch;
155
+ })
156
+ .map(function retainedArtifact(writer) {
157
+ return writer.artifact();
158
+ });
159
+ }
160
+ function runAttachmentStoreRetainsBranch(state, branch) {
161
+ return branch === null || state.branches.get(branch)?.() === true;
162
+ }
163
+ function runAttachmentStoreRetains(state, writer) {
164
+ return runAttachmentStoreRetainsBranch(state, writer.branch);
165
+ }
166
+ function runAttachmentStoreSelectedArtifacts(state) {
167
+ const { writers } = state;
168
+ return Array
169
+ .from(writers.values())
170
+ .filter(function (writer) {
171
+ return runAttachmentStoreRetains(state, writer);
172
+ })
173
+ .map(function retainedArtifact(writer) {
174
+ return writer.artifact();
175
+ });
176
+ }
177
+ function runAttachmentStoreValidateOwner(state, operation) {
178
+ const { branches, planned } = state;
179
+ validateMetadata(operation);
180
+ if (operation.branch !== null && !branches.has(operation.branch)) {
181
+ throw new TypeError('Unknown attachment execution branch.');
182
+ }
183
+ if (operation.owner.kind === 'case' && !planned.has(workIdentityKey(operation.owner.work))) {
184
+ throw new TypeError('Attachment owner is outside the selected work.');
185
+ }
186
+ }
187
+ function runAttachmentStoreScopeBudget(state, channel, operation) {
188
+ const { budgets, options } = state;
189
+ const key = operation.owner.kind === 'run'
190
+ ? 'run'
191
+ : `${operation.branch ?? channel}:${ownerKey(operation.owner)}`;
192
+ const budget = budgets.get(key) ?? createAttachmentScopeBudget();
193
+ budgets.set(key, budget);
194
+ budget.reserve(options.limits);
195
+ return budget;
196
+ }
197
+ function runAttachmentStoreInitialArtifact(state, operation, writer, retention) {
198
+ return {
199
+ id: attachmentId(operation.owner, writer),
200
+ source: 'instrumented',
201
+ payload: {
202
+ ...operation.metadata,
203
+ capture: 'opt-in',
204
+ capturedAtMicroseconds: state.options.captureTime(),
205
+ content: retention.content(),
206
+ kind: 'runtime-attachment',
207
+ producer: operation.producer
208
+ }
209
+ };
210
+ }
211
+ async function createRetainedAttachment(state, creation) {
212
+ const { channel, operation, writer, filePath, limit, budget } = creation;
213
+ const retention = await createAttachmentRetention(operation.contentKind, state.options.projectRoot, filePath, limit);
214
+ try {
215
+ return createStoredAttachment(channel, operation, {
216
+ initial: runAttachmentStoreInitialArtifact(state, operation, writer, retention),
217
+ retention,
218
+ budget
219
+ });
220
+ }
221
+ catch (error) {
222
+ await retention.finish({ kind: 'incomplete', reason: 'write-error' });
223
+ await rm(filePath, { force: true });
224
+ throw error;
225
+ }
226
+ }
227
+ async function runAttachmentStoreCreateWriter(state, channel, operation) {
228
+ const { sequence, options, writers } = state;
229
+ const budget = runAttachmentStoreScopeBudget(state, channel, operation);
230
+ const writer = sequence.read();
231
+ sequence.write(sequence.read() + 1);
232
+ const folder = createHash('sha256').update(ownerKey(operation.owner)).digest('hex');
233
+ const filePath = path.join(options.directory, folder, `${writer}.attachment.bin`);
234
+ const limit = operation.contentKind === 'binary'
235
+ ? options.limits.maxArtifactBytes
236
+ : options.limits.maxInlineBytes;
237
+ writers.set(writer, await createRetainedAttachment(state, { channel, operation, writer, filePath, limit, budget }));
238
+ return writer;
239
+ }
240
+ async function runAttachmentStoreStart(state, channel, operation) {
241
+ const { options } = state;
242
+ runAttachmentStoreValidateOwner(state, operation);
243
+ const writer = await runAttachmentStoreCreateWriter(state, channel, operation);
244
+ if (operation.contentKind === 'binary') {
245
+ await options.checkpoint(runAttachmentStoreArtifacts(state));
246
+ }
247
+ return { kind: 'opened', writer };
248
+ }
249
+ async function processAttachmentWriterOperation(state, writer, operation) {
250
+ if (operation.kind === 'close') {
251
+ await writer.close(operation.reason);
252
+ await state.options.checkpoint(runAttachmentStoreArtifacts(state));
253
+ return { kind: 'closed', artifact: writer.artifact() };
254
+ }
255
+ if (operation.kind === 'omit') {
256
+ writer.omit();
257
+ return { kind: 'written' };
258
+ }
259
+ return await writer.write(operation.data, state.options.limits);
260
+ }
261
+ async function runAttachmentStoreProcessOperation(state, channel, operation) {
262
+ if (operation.kind === 'open') {
263
+ return await runAttachmentStoreStart(state, channel, operation);
264
+ }
265
+ const writer = state.writers.get(operation.writer);
266
+ if (writer?.channel !== channel) {
267
+ throw new TypeError('Unknown attachment writer.');
268
+ }
269
+ return await processAttachmentWriterOperation(state, writer, operation);
270
+ }
271
+ async function runAttachmentStoreExchange(state, channel, operation) {
272
+ try {
273
+ return await runAttachmentStoreProcessOperation(state, channel, operation);
274
+ }
275
+ catch (error) {
276
+ return {
277
+ kind: 'error',
278
+ message: error instanceof Error ? error.message : 'Attachment operation failed.',
279
+ reason: 'operation-error'
280
+ };
281
+ }
282
+ }
283
+ async function runAttachmentStoreFinish(state, channel) {
284
+ const { writers } = state;
285
+ const owners = [];
286
+ const selected = Array.from(writers.values()).filter(function inChannel(writer) {
287
+ return channel === null || writer.channel === channel;
288
+ });
289
+ for (const writer of selected) {
290
+ const owner = await writer.interrupt(channel === null ? 'unclosed' : 'interrupted');
291
+ if (owner !== null && runAttachmentStoreRetains(state, writer)) {
292
+ owners.push(owner);
293
+ }
294
+ }
295
+ return owners;
296
+ }
297
+ async function runAttachmentStorePrune(state, retainedArtifacts) {
298
+ const { options, writers } = state;
299
+ const retained = new Set(retainedArtifacts.map(function artifactSequence(artifact) {
300
+ return artifact.id.sequence;
301
+ }));
302
+ const discarded = runAttachmentStoreArtifacts(state).filter(function discardedArtifact(artifact) {
303
+ return !retained.has(artifact.id.sequence);
304
+ });
305
+ for (const artifact of discarded) {
306
+ if (artifact.payload.content.kind === 'file') {
307
+ await rm(path.resolve(options.projectRoot, artifact.payload.content.path), { force: true });
308
+ }
309
+ writers.delete(artifact.id.sequence);
310
+ }
311
+ }
312
+ export function createAttachmentStore(options) {
313
+ const writers = new Map();
314
+ const branches = new Map();
315
+ const budgets = new Map();
316
+ const planned = new Set(options.work.map(workIdentityKey));
317
+ const sequence = createStoredRunValue(0);
318
+ const state = { options, writers, branches, budgets, planned, sequence };
319
+ return {
320
+ artifacts: runAttachmentStoreArtifacts.bind(null, state),
321
+ retainsBranch: runAttachmentStoreRetainsBranch.bind(null, state),
322
+ registerBranch: runAttachmentStoreRegisterBranch.bind(null, state),
323
+ branchArtifacts: runAttachmentStoreBranchArtifacts.bind(null, state),
324
+ selectedArtifacts: runAttachmentStoreSelectedArtifacts.bind(null, state),
325
+ exchange: runAttachmentStoreExchange.bind(null, state),
326
+ finish: runAttachmentStoreFinish.bind(null, state),
327
+ prune: runAttachmentStorePrune.bind(null, state)
328
+ };
329
+ }
330
+ //# sourceMappingURL=attachment-store.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"attachment-store.js","sourceRoot":"","sources":["../../../../source/run/attachment-store.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,EAAE,EAAE,MAAM,kBAAkB,CAAC;AACtC,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,eAAe,EAAe,MAAM,uBAAuB,CAAC;AAErE,OAAO,EAAE,oBAAoB,EAAuB,MAAM,2BAA2B,CAAC;AACtF,OAAO,EACH,uBAAuB,EAK1B,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,yBAAyB,EAA4B,MAAM,2BAA2B,CAAC;AAQhG,SAAS,2BAA2B;IAChC,MAAM,KAAK,GAAG,oBAAoB,CAAS,CAAC,CAAC,CAAC;IAC9C,MAAM,KAAK,GAAG,oBAAoB,CAAS,CAAC,CAAC,CAAC;IAC9C,SAAS,OAAO,CAAC,MAAwB;QACrC,IAAI,KAAK,CAAC,IAAI,EAAE,IAAI,MAAM,CAAC,mBAAmB,EAAE,CAAC;YAC7C,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;QACxD,CAAC;QACD,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC,CAAC;IAClC,CAAC;IACD,SAAS,SAAS,CAAC,MAAwB;QACvC,OAAO,MAAM,CAAC,aAAa,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;IAC/C,CAAC;IACD,SAAS,QAAQ,CAAC,aAAqB;QACnC,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE,GAAG,aAAa,CAAC,CAAC;IAC9C,CAAC;IAED,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,CAAC;AAC5C,CAAC;AAsBD,SAAS,wBAAwB,CAAC,KAA4B;IAC1D,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,GAAG,KAAK,CAAC;IAErC,OAAO,EAAE,GAAG,KAAK,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,GAAG,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE,SAAS,CAAC,OAAO,EAAE,EAAE,EAAE,CAAC;AAC/F,CAAC;AACD,KAAK,UAAU,qBAAqB,CAAC,KAA4B,EAAE,MAA6B;IAC5F,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,KAAK,CAAC;IAEpC,IAAI,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;QAChB,OAAO;IACX,CAAC;IACD,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IACnB,MAAM,SAAS,CAAC,MAAM,CAAC,MAAM,KAAK,UAAU,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,MAAM,EAAE,CAAC,CAAC;AAC1G,CAAC;AACD,KAAK,UAAU,yBAAyB,CACpC,KAA4B,EAC5B,MAAkC;IAElC,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,GAAG,KAAK,CAAC;IAE5C,IAAI,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;QAChB,OAAO,IAAI,CAAC;IAChB,CAAC;IACD,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IACnB,MAAM,SAAS,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,MAAM,EAAE,CAAC,CAAC;IACvD,OAAO,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC;AAC9C,CAAC;AACD,SAAS,0BAA0B,CAAC,KAA4B;IAC5D,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,KAAK,CAAC;IAEjC,IAAI,MAAM,CAAC,IAAI,EAAE,IAAI,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;QACjC,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC;IACpD,CAAC;AACL,CAAC;AACD,SAAS,oBAAoB,CAAC,KAA4B;IACtD,MAAM,EAAE,SAAS,EAAE,GAAG,KAAK,CAAC;IAE5B,0BAA0B,CAAC,KAAK,CAAC,CAAC;IAClC,SAAS,CAAC,IAAI,EAAE,CAAC;AACrB,CAAC;AACD,KAAK,UAAU,sBAAsB,CACjC,KAA4B,EAC5B,KAAiB,EACjB,MAAwB;IAExB,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,KAAK,CAAC;IAE5C,IAAI,CAAC;QACD,OAAO,MAAM,SAAS,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC;IACnE,CAAC;IAAC,OAAO,KAAc,EAAE,CAAC;QACtB,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACnB,MAAM,qBAAqB,CAAC,KAAK,EAAE,aAAa,CAAC,CAAC;QAClD,MAAM,KAAK,CAAC;IAChB,CAAC;AACL,CAAC;AACD,KAAK,UAAU,qBAAqB,CAChC,KAA4B,EAC5B,IAAY,EACZ,MAAwB;IAExB,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,KAAK,CAAC;IAErD,0BAA0B,CAAC,KAAK,CAAC,CAAC;IAClC,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IAC1C,MAAM,QAAQ,GAAG,MAAM,sBAAsB,CAAC,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC;IACpE,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;IAC1B,IAAI,QAAQ,GAAG,KAAK,CAAC,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;QACrE,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACnB,MAAM,SAAS,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,MAAM,EAAE,YAAY,EAAE,CAAC,CAAC;QACrE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,wCAAwC,EAAE,MAAM,EAAE,YAAY,EAAE,CAAC;IACtG,CAAC;IACD,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;AAC/B,CAAC;AAMD,SAAS,sBAAsB,CAC3B,OAAe,EACf,SAAwB,EACxB,OAAgC;IAEhC,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;IAC/C,MAAM,MAAM,GAAkB,SAAS,CAAC,MAAM,CAAC;IAC/C,MAAM,KAAK,GAAoB,SAAS,CAAC,KAAK,CAAC;IAC/C,MAAM,MAAM,GAAG,oBAAoB,CAAU,KAAK,CAAC,CAAC;IACpD,MAAM,MAAM,GAAG,oBAAoB,CAAU,KAAK,CAAC,CAAC;IACpD,MAAM,KAAK,GAA0B;QACjC,OAAO;QACP,SAAS;QACT,OAAO;QACP,SAAS;QACT,MAAM;QACN,MAAM;QACN,KAAK;QACL,MAAM;QACN,MAAM;KACT,CAAC;IACF,OAAO;QACH,OAAO,EAAE,KAAK,CAAC,OAAO;QACtB,MAAM,EAAE,KAAK,CAAC,MAAM;QACpB,KAAK,EAAE,KAAK,CAAC,KAAK;QAClB,QAAQ,EAAE,wBAAwB,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC;QACpD,KAAK,EAAE,qBAAqB,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC;QAC9C,SAAS,EAAE,yBAAyB,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC;QACtD,IAAI,EAAE,oBAAoB,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC;QAC5C,KAAK,EAAE,qBAAqB,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC;KACjD,CAAC;AACN,CAAC;AAmBD,SAAS,QAAQ,CAAC,KAAsB;IACpC,OAAO,KAAK,CAAC,IAAI,KAAK,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;AACtE,CAAC;AACD,SAAS,YAAY,CAAC,KAAsB,EAAE,QAAgB;IAC1D,IAAI,KAAK,CAAC,IAAI,KAAK,KAAK,EAAE,CAAC;QACvB,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,YAAY,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;IACpH,CAAC;IACD,OAAO;QACH,OAAO,EAAE,KAAK,CAAC,OAAO;QACtB,QAAQ,EAAE,KAAK,CAAC,IAAI,CAAC,QAAQ;QAC7B,KAAK,EAAE,EAAE,WAAW,EAAE,CAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAE,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,UAAU,EAAE,aAAa,EAAE,IAAI,EAAE,MAAM,EAAE;QAC3G,QAAQ;QACR,OAAO,EAAE,YAAY;QACrB,QAAQ,EAAE,KAAK,CAAC,IAAI,CAAC,QAAQ;KAChC,CAAC;AACN,CAAC;AACD,MAAM,uBAAuB,GAAG,EAAE,CAAC;AACnC,MAAM,eAAe,GAAG,GAAG,CAAC;AAC5B,SAAS,kBAAkB,CAAC,SAAiB;IACzC,MAAM,IAAI,GAAG,SAAS,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IAC3C,OAAO,IAAI,GAAG,uBAAuB,IAAI,IAAI,KAAK,eAAe,CAAC;AACtE,CAAC;AACD,SAAS,gBAAgB,CAAC,SAAwB;IAC9C,KAAK,MAAM,KAAK,IAAI,CAAE,SAAS,CAAC,QAAQ,CAAC,IAAI,EAAE,SAAS,CAAC,QAAQ,CAAC,SAAS,CAAE,EAAE,CAAC;QAC5E,IACI,KAAK,CAAC,MAAM,KAAK,CAAC,IAAI,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,GAAG,uBAAuB;YACxE,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,EAC5C,CAAC;YACC,MAAM,IAAI,SAAS,CAAC,qFAAqF,CAAC,CAAC;QAC/G,CAAC;IACL,CAAC;AACL,CAAC;AASD,SAAS,2BAA2B,CAAC,KAA8B;IAC/D,MAAM,EAAE,OAAO,EAAE,GAAG,KAAK,CAAC;IAE1B,OAAO,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,SAAS,gBAAgB,CAAC,MAAM;QAChE,OAAO,MAAM,CAAC,QAAQ,EAAE,CAAC;IAC7B,CAAC,CAAC,CAAC;AACP,CAAC;AACD,SAAS,gCAAgC,CAAC,KAA8B,EAAE,MAAc,EAAE,MAAqB;IAC3G,MAAM,EAAE,QAAQ,EAAE,GAAG,KAAK,CAAC;IAE3B,QAAQ,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;AACjC,CAAC;AACD,SAAS,iCAAiC,CACtC,KAA8B,EAC9B,MAAqB;IAErB,MAAM,EAAE,OAAO,EAAE,GAAG,KAAK,CAAC;IAE1B,OAAO,KAAK;SACP,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;SACtB,MAAM,CAAC,SAAS,QAAQ,CAAC,MAAM;QAC5B,OAAO,MAAM,CAAC,MAAM,KAAK,MAAM,CAAC;IACpC,CAAC,CAAC;SACD,GAAG,CAAC,SAAS,gBAAgB,CAAC,MAAM;QACjC,OAAO,MAAM,CAAC,QAAQ,EAAE,CAAC;IAC7B,CAAC,CAAC,CAAC;AACX,CAAC;AACD,SAAS,+BAA+B,CAAC,KAA8B,EAAE,MAAqB;IAC1F,OAAO,MAAM,KAAK,IAAI,IAAI,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,EAAE,KAAK,IAAI,CAAC;AACtE,CAAC;AACD,SAAS,yBAAyB,CAAC,KAA8B,EAAE,MAAwB;IACvF,OAAO,+BAA+B,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;AACjE,CAAC;AACD,SAAS,mCAAmC,CAAC,KAA8B;IACvE,MAAM,EAAE,OAAO,EAAE,GAAG,KAAK,CAAC;IAE1B,OAAO,KAAK;SACP,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;SACtB,MAAM,CAAC,UAAU,MAAM;QACpB,OAAO,yBAAyB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IACpD,CAAC,CAAC;SACD,GAAG,CAAC,SAAS,gBAAgB,CAAC,MAAM;QACjC,OAAO,MAAM,CAAC,QAAQ,EAAE,CAAC;IAC7B,CAAC,CAAC,CAAC;AACX,CAAC;AACD,SAAS,+BAA+B,CAAC,KAA8B,EAAE,SAAwB;IAC7F,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,KAAK,CAAC;IAEpC,gBAAgB,CAAC,SAAS,CAAC,CAAC;IAC5B,IAAI,SAAS,CAAC,MAAM,KAAK,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC;QAC/D,MAAM,IAAI,SAAS,CAAC,sCAAsC,CAAC,CAAC;IAChE,CAAC;IACD,IAAI,SAAS,CAAC,KAAK,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC;QACzF,MAAM,IAAI,SAAS,CAAC,gDAAgD,CAAC,CAAC;IAC1E,CAAC;AACL,CAAC;AACD,SAAS,6BAA6B,CAClC,KAA8B,EAC9B,OAAe,EACf,SAAwB;IAExB,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,KAAK,CAAC;IAEnC,MAAM,GAAG,GAAG,SAAS,CAAC,KAAK,CAAC,IAAI,KAAK,KAAK;QACtC,CAAC,CAAC,KAAK;QACP,CAAC,CAAC,GAAG,SAAS,CAAC,MAAM,IAAI,OAAO,IAAI,QAAQ,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC;IACpE,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,2BAA2B,EAAE,CAAC;IACjE,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;IACzB,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IAC/B,OAAO,MAAM,CAAC;AAClB,CAAC;AACD,SAAS,iCAAiC,CACtC,KAA8B,EAC9B,SAAwB,EACxB,MAAc,EACd,SAA8B;IAE9B,OAAO;QACH,EAAE,EAAE,YAAY,CAAC,SAAS,CAAC,KAAK,EAAE,MAAM,CAAC;QACzC,MAAM,EAAE,cAAc;QACtB,OAAO,EAAE;YACL,GAAG,SAAS,CAAC,QAAQ;YACrB,OAAO,EAAE,QAAQ;YACjB,sBAAsB,EAAE,KAAK,CAAC,OAAO,CAAC,WAAW,EAAE;YACnD,OAAO,EAAE,SAAS,CAAC,OAAO,EAAE;YAC5B,IAAI,EAAE,oBAAoB;YAC1B,QAAQ,EAAE,SAAS,CAAC,QAAQ;SAC/B;KACJ,CAAC;AACN,CAAC;AASD,KAAK,UAAU,wBAAwB,CACnC,KAA8B,EAC9B,QAAkC;IAElC,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,QAAQ,CAAC;IACzE,MAAM,SAAS,GAAG,MAAM,yBAAyB,CAC7C,SAAS,CAAC,WAAW,EACrB,KAAK,CAAC,OAAO,CAAC,WAAW,EACzB,QAAQ,EACR,KAAK,CACR,CAAC;IACF,IAAI,CAAC;QACD,OAAO,sBAAsB,CAAC,OAAO,EAAE,SAAS,EAAE;YAC9C,OAAO,EAAE,iCAAiC,CAAC,KAAK,EAAE,SAAS,EAAE,MAAM,EAAE,SAAS,CAAC;YAC/E,SAAS;YACT,MAAM;SACT,CAAC,CAAC;IACP,CAAC;IAAC,OAAO,KAAc,EAAE,CAAC;QACtB,MAAM,SAAS,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC,CAAC;QACtE,MAAM,EAAE,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QACpC,MAAM,KAAK,CAAC;IAChB,CAAC;AACL,CAAC;AACD,KAAK,UAAU,8BAA8B,CACzC,KAA8B,EAC9B,OAAe,EACf,SAAwB;IAExB,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,KAAK,CAAC;IAE7C,MAAM,MAAM,GAAG,6BAA6B,CAAC,KAAK,EAAE,OAAO,EAAE,SAAS,CAAC,CAAC;IACxE,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC;IAC/B,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC,CAAC;IACpC,MAAM,MAAM,GAAG,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IACpF,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,MAAM,EAAE,GAAG,MAAM,iBAAiB,CAAC,CAAC;IAClF,MAAM,KAAK,GAAG,SAAS,CAAC,WAAW,KAAK,QAAQ;QAC5C,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,gBAAgB;QACjC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,cAAc,CAAC;IACpC,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,wBAAwB,CAAC,KAAK,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;IACpH,OAAO,MAAM,CAAC;AAClB,CAAC;AACD,KAAK,UAAU,uBAAuB,CAClC,KAA8B,EAC9B,OAAe,EACf,SAAwB;IAExB,MAAM,EAAE,OAAO,EAAE,GAAG,KAAK,CAAC;IAE1B,+BAA+B,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;IAClD,MAAM,MAAM,GAAG,MAAM,8BAA8B,CAAC,KAAK,EAAE,OAAO,EAAE,SAAS,CAAC,CAAC;IAC/E,IAAI,SAAS,CAAC,WAAW,KAAK,QAAQ,EAAE,CAAC;QACrC,MAAM,OAAO,CAAC,UAAU,CAAC,2BAA2B,CAAC,KAAK,CAAC,CAAC,CAAC;IACjE,CAAC;IACD,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC;AACtC,CAAC;AAED,KAAK,UAAU,gCAAgC,CAC3C,KAA8B,EAC9B,MAAwB,EACxB,SAAoC;IAEpC,IAAI,SAAS,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;QAC7B,MAAM,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QACrC,MAAM,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,2BAA2B,CAAC,KAAK,CAAC,CAAC,CAAC;QACnE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,CAAC,QAAQ,EAAE,EAAE,CAAC;IAC3D,CAAC;IACD,IAAI,SAAS,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;QAC5B,MAAM,CAAC,IAAI,EAAE,CAAC;QACd,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;IAC/B,CAAC;IACD,OAAO,MAAM,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,EAAE,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;AACpE,CAAC;AACD,KAAK,UAAU,kCAAkC,CAC7C,KAA8B,EAC9B,OAAe,EACf,SAA8B;IAE9B,IAAI,SAAS,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;QAC5B,OAAO,MAAM,uBAAuB,CAAC,KAAK,EAAE,OAAO,EAAE,SAAS,CAAC,CAAC;IACpE,CAAC;IACD,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IACnD,IAAI,MAAM,EAAE,OAAO,KAAK,OAAO,EAAE,CAAC;QAC9B,MAAM,IAAI,SAAS,CAAC,4BAA4B,CAAC,CAAC;IACtD,CAAC;IACD,OAAO,MAAM,gCAAgC,CAAC,KAAK,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;AAC5E,CAAC;AACD,KAAK,UAAU,0BAA0B,CACrC,KAA8B,EAC9B,OAAe,EACf,SAA8B;IAE9B,IAAI,CAAC;QACD,OAAO,MAAM,kCAAkC,CAAC,KAAK,EAAE,OAAO,EAAE,SAAS,CAAC,CAAC;IAC/E,CAAC;IAAC,OAAO,KAAc,EAAE,CAAC;QACtB,OAAO;YACH,IAAI,EAAE,OAAO;YACb,OAAO,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,8BAA8B;YAChF,MAAM,EAAE,iBAAiB;SAC5B,CAAC;IACN,CAAC;AACL,CAAC;AACD,KAAK,UAAU,wBAAwB,CACnC,KAA8B,EAC9B,OAAsB;IAEtB,MAAM,EAAE,OAAO,EAAE,GAAG,KAAK,CAAC;IAE1B,MAAM,MAAM,GAAsB,EAAE,CAAC;IACrC,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,SAAS,SAAS,CAAC,MAAM;QAC1E,OAAO,OAAO,KAAK,IAAI,IAAI,MAAM,CAAC,OAAO,KAAK,OAAO,CAAC;IAC1D,CAAC,CAAC,CAAC;IACH,KAAK,MAAM,MAAM,IAAI,QAAQ,EAAE,CAAC;QAC5B,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,OAAO,KAAK,IAAI,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC;QACpF,IAAI,KAAK,KAAK,IAAI,IAAI,yBAAyB,CAAC,KAAK,EAAE,MAAM,CAAC,EAAE,CAAC;YAC7D,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACvB,CAAC;IACL,CAAC;IACD,OAAO,MAAM,CAAC;AAClB,CAAC;AACD,KAAK,UAAU,uBAAuB,CAClC,KAA8B,EAC9B,iBAAuD;IAEvD,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,KAAK,CAAC;IAEnC,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAC,iBAAiB,CAAC,GAAG,CAAC,SAAS,gBAAgB,CAAC,QAAQ;QAC7E,OAAO,QAAQ,CAAC,EAAE,CAAC,QAAQ,CAAC;IAChC,CAAC,CAAC,CAAC,CAAC;IACJ,MAAM,SAAS,GAAG,2BAA2B,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,SAAS,iBAAiB,CAAC,QAAQ;QAC3F,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IACH,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;QAC/B,IAAI,QAAQ,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC3C,MAAM,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,WAAW,EAAE,QAAQ,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QAChG,CAAC;QACD,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC;IACzC,CAAC;AACL,CAAC;AACD,MAAM,UAAU,qBAAqB,CAAC,OAA+B;IACjE,MAAM,OAAO,GAAG,IAAI,GAAG,EAA4B,CAAC;IACpD,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAyB,CAAC;IAClD,MAAM,OAAO,GAAG,IAAI,GAAG,EAAiC,CAAC;IACzD,MAAM,OAAO,GAAwB,IAAI,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC,CAAC;IAChF,MAAM,QAAQ,GAAG,oBAAoB,CAAS,CAAC,CAAC,CAAC;IACjD,MAAM,KAAK,GAA4B,EAAE,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC;IAClG,OAAO;QACH,SAAS,EAAE,2BAA2B,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC;QACxD,aAAa,EAAE,+BAA+B,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC;QAChE,cAAc,EAAE,gCAAgC,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC;QAClE,eAAe,EAAE,iCAAiC,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC;QACpE,iBAAiB,EAAE,mCAAmC,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC;QACxE,QAAQ,EAAE,0BAA0B,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC;QACtD,MAAM,EAAE,wBAAwB,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC;QAClD,KAAK,EAAE,uBAAuB,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC;KACnD,CAAC;AACN,CAAC"}
@@ -0,0 +1,123 @@
1
+ import { z } from 'zod/v4';
2
+ import { attachmentChunkBytes, attachmentMetadataBytes } from "./attachment-protocol.js";
3
+ const caseSchema = z.object({
4
+ file: z.string().nullable(),
5
+ params: z.string().nullable(),
6
+ suite: z.array(z.string()),
7
+ title: z.string()
8
+ });
9
+ const runtimeSchema = z.object({
10
+ dimensions: z.record(z.string(), z.string()),
11
+ name: z.string(),
12
+ scenarios: z.record(z.string(), z.string()),
13
+ variantId: z.string().nullable()
14
+ });
15
+ const workloadSchema = z.object({ name: z.string(), params: z.record(z.string(), z.string()) });
16
+ const attemptSchema = z.object({ index: z.number().int().nonnegative() });
17
+ const workSchema = z.object({
18
+ case: caseSchema,
19
+ runtimes: z.array(runtimeSchema),
20
+ workload: workloadSchema.nullable()
21
+ });
22
+ const ownerSchema = z.union([
23
+ z.object({ kind: z.literal('run') }),
24
+ z.object({ attempt: attemptSchema, kind: z.literal('case'), work: workSchema })
25
+ ]);
26
+ const producerSchema = z.union([
27
+ z.object({ kind: z.literal('case') }),
28
+ z.object({ kind: z.literal('resource'), name: z.string().max(attachmentMetadataBytes) })
29
+ ]);
30
+ const writerSchema = z.number().int().nonnegative();
31
+ const metadataSchema = z.object({
32
+ mediaType: z.string().max(attachmentMetadataBytes),
33
+ name: z.string().max(attachmentMetadataBytes)
34
+ });
35
+ const base64Quantum = 3;
36
+ const base64Chars = 4;
37
+ const maxEncodedChunk = Math.ceil(attachmentChunkBytes / base64Quantum) * base64Chars;
38
+ export const attachmentRequestSchema = z.object({
39
+ request: writerSchema,
40
+ token: z.string(),
41
+ operation: z.discriminatedUnion('kind', [
42
+ z.object({
43
+ branch: z.string().nullable(),
44
+ contentKind: z.enum(['binary', 'json', 'text']),
45
+ kind: z.literal('open'),
46
+ metadata: metadataSchema,
47
+ owner: ownerSchema,
48
+ producer: producerSchema
49
+ }),
50
+ z.object({
51
+ data: z.string().max(maxEncodedChunk).regex(/^[A-Za-z0-9+/]*={0,2}$/u),
52
+ kind: z.literal('write'),
53
+ writer: writerSchema
54
+ }),
55
+ z.object({ kind: z.literal('omit'), writer: writerSchema }),
56
+ z.object({
57
+ kind: z.literal('close'),
58
+ reason: z.enum(['complete', 'unclosed', 'write-error']),
59
+ writer: writerSchema
60
+ })
61
+ ])
62
+ });
63
+ const completionSchema = z.union([
64
+ z.object({ kind: z.literal('complete') }),
65
+ z.object({
66
+ kind: z.literal('incomplete'),
67
+ reason: z.enum(['byte-limit', 'interrupted', 'unclosed', 'write-error'])
68
+ })
69
+ ]);
70
+ const inlineCompletionSchema = z.union([
71
+ completionSchema,
72
+ z.object({ kind: z.literal('truncated'), reason: z.literal('byte-limit') })
73
+ ]);
74
+ const contentSchema = z.discriminatedUnion('kind', [
75
+ z.object({
76
+ byteLength: writerSchema,
77
+ completion: inlineCompletionSchema,
78
+ kind: z.literal('text'),
79
+ text: z.string()
80
+ }),
81
+ z.object({ byteLength: writerSchema, kind: z.literal('json'), value: z.json() }),
82
+ z.object({ byteLength: writerSchema, completion: completionSchema, kind: z.literal('file'), path: z.string() }),
83
+ z.object({ kind: z.literal('omitted'), limit: writerSchema, reason: z.enum(['byte-limit', 'count-limit']) })
84
+ ]);
85
+ const artifactBaseSchema = z.object({
86
+ runtimes: z.array(runtimeSchema),
87
+ sequence: writerSchema,
88
+ subtype: z.literal('attachment'),
89
+ workload: workloadSchema.nullable()
90
+ });
91
+ const artifactIdSchema = z.union([
92
+ artifactBaseSchema.extend({ attempt: z.null(), scope: z.object({ kind: z.literal('run') }) }),
93
+ artifactBaseSchema.extend({
94
+ attempt: attemptSchema,
95
+ scope: z.object({
96
+ activeCases: z.array(caseSchema),
97
+ case: caseSchema,
98
+ confidence: z.enum(['active-case', 'concurrent-active']),
99
+ kind: z.literal('case')
100
+ })
101
+ })
102
+ ]);
103
+ const artifactSchema = z.object({
104
+ id: artifactIdSchema,
105
+ payload: metadataSchema.extend({
106
+ capture: z.literal('opt-in'),
107
+ capturedAtMicroseconds: z.number(),
108
+ content: contentSchema,
109
+ kind: z.literal('runtime-attachment'),
110
+ producer: producerSchema
111
+ }),
112
+ source: z.literal('instrumented')
113
+ });
114
+ export const attachmentReplySchema = z.object({
115
+ request: writerSchema,
116
+ result: z.discriminatedUnion('kind', [
117
+ z.object({ kind: z.literal('opened'), writer: writerSchema }),
118
+ z.object({ kind: z.literal('written') }),
119
+ z.object({ artifact: artifactSchema, kind: z.literal('closed') }),
120
+ z.object({ kind: z.literal('error'), message: z.string(), reason: z.string() })
121
+ ])
122
+ });
123
+ //# sourceMappingURL=attachment-wire-schema.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"attachment-wire-schema.js","sourceRoot":"","sources":["../../../../source/run/attachment-wire-schema.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,QAAQ,CAAC;AAC3B,OAAO,EACH,oBAAoB,EACpB,uBAAuB,EAG1B,MAAM,0BAA0B,CAAC;AAElC,MAAM,UAAU,GAAG,CAAC,CAAC,MAAM,CAAC;IACxB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC3B,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC7B,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IAC1B,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;CACpB,CAAC,CAAC;AACH,MAAM,aAAa,GAAG,CAAC,CAAC,MAAM,CAAC;IAC3B,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC;IAC5C,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;IAChB,SAAS,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC;IAC3C,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;CACnC,CAAC,CAAC;AACH,MAAM,cAAc,GAAG,CAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC;AAChG,MAAM,aAAa,GAAG,CAAC,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;AAC1E,MAAM,UAAU,GAAG,CAAC,CAAC,MAAM,CAAC;IACxB,IAAI,EAAE,UAAU;IAChB,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC,aAAa,CAAC;IAChC,QAAQ,EAAE,cAAc,CAAC,QAAQ,EAAE;CACtC,CAAC,CAAC;AACH,MAAM,WAAW,GAAG,CAAC,CAAC,KAAK,CAAC;IACxB,CAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;IACpC,CAAC,CAAC,MAAM,CAAC,EAAE,OAAO,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;CAClF,CAAC,CAAC;AACH,MAAM,cAAc,GAAG,CAAC,CAAC,KAAK,CAAC;IAC3B,CAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;IACrC,CAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,uBAAuB,CAAC,EAAE,CAAC;CAC3F,CAAC,CAAC;AACH,MAAM,YAAY,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,WAAW,EAAE,CAAC;AACpD,MAAM,cAAc,GAAG,CAAC,CAAC,MAAM,CAAC;IAC5B,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,uBAAuB,CAAC;IAClD,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,uBAAuB,CAAC;CAChD,CAAC,CAAC;AACH,MAAM,aAAa,GAAG,CAAC,CAAC;AACxB,MAAM,WAAW,GAAG,CAAC,CAAC;AACtB,MAAM,eAAe,GAAG,IAAI,CAAC,IAAI,CAAC,oBAAoB,GAAG,aAAa,CAAC,GAAG,WAAW,CAAC;AACtF,MAAM,CAAC,MAAM,uBAAuB,GAAiC,CAAC,CAAC,MAAM,CAAC;IAC1E,OAAO,EAAE,YAAY;IACrB,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;IACjB,SAAS,EAAE,CAAC,CAAC,kBAAkB,CAAC,MAAM,EAAE;QACpC,CAAC,CAAC,MAAM,CAAC;YACL,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;YAC7B,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,CAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,CAAE,CAAC;YACjD,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC;YACvB,QAAQ,EAAE,cAAc;YACxB,KAAK,EAAE,WAAW;YAClB,QAAQ,EAAE,cAAc;SAC3B,CAAC;QACF,CAAC,CAAC,MAAM,CAAC;YACL,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC,KAAK,CAAC,yBAAyB,CAAC;YACtE,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;YACxB,MAAM,EAAE,YAAY;SACvB,CAAC;QACF,CAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,YAAY,EAAE,CAAC;QAC3D,CAAC,CAAC,MAAM,CAAC;YACL,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;YACxB,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAE,UAAU,EAAE,UAAU,EAAE,aAAa,CAAE,CAAC;YACzD,MAAM,EAAE,YAAY;SACvB,CAAC;KACL,CAAC;CACL,CAAC,CAAC;AACH,MAAM,gBAAgB,GAAG,CAAC,CAAC,KAAK,CAAC;IAC7B,CAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;IACzC,CAAC,CAAC,MAAM,CAAC;QACL,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC;QAC7B,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAE,YAAY,EAAE,aAAa,EAAE,UAAU,EAAE,aAAa,CAAE,CAAC;KAC7E,CAAC;CACL,CAAC,CAAC;AACH,MAAM,sBAAsB,GAAG,CAAC,CAAC,KAAK,CAAC;IACnC,gBAAgB;IAChB,CAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE,CAAC;CAC9E,CAAC,CAAC;AACH,MAAM,aAAa,GAAG,CAAC,CAAC,kBAAkB,CAAC,MAAM,EAAE;IAC/C,CAAC,CAAC,MAAM,CAAC;QACL,UAAU,EAAE,YAAY;QACxB,UAAU,EAAE,sBAAsB;QAClC,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC;QACvB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;KACnB,CAAC;IACF,CAAC,CAAC,MAAM,CAAC,EAAE,UAAU,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC;IAChF,CAAC,CAAC,MAAM,CAAC,EAAE,UAAU,EAAE,YAAY,EAAE,UAAU,EAAE,gBAAgB,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;IAC/G,CAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAE,YAAY,EAAE,aAAa,CAAE,CAAC,EAAE,CAAC;CACjH,CAAC,CAAC;AACH,MAAM,kBAAkB,GAAG,CAAC,CAAC,MAAM,CAAC;IAChC,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC,aAAa,CAAC;IAChC,QAAQ,EAAE,YAAY;IACtB,OAAO,EAAE,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC;IAChC,QAAQ,EAAE,cAAc,CAAC,QAAQ,EAAE;CACtC,CAAC,CAAC;AACH,MAAM,gBAAgB,GAAG,CAAC,CAAC,KAAK,CAAC;IAC7B,kBAAkB,CAAC,MAAM,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,EAAE,CAAC;IAC7F,kBAAkB,CAAC,MAAM,CAAC;QACtB,OAAO,EAAE,aAAa;QACtB,KAAK,EAAE,CAAC,CAAC,MAAM,CAAC;YACZ,WAAW,EAAE,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC;YAChC,IAAI,EAAE,UAAU;YAChB,UAAU,EAAE,CAAC,CAAC,IAAI,CAAC,CAAE,aAAa,EAAE,mBAAmB,CAAE,CAAC;YAC1D,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC;SAC1B,CAAC;KACL,CAAC;CACL,CAAC,CAAC;AACH,MAAM,cAAc,GAAG,CAAC,CAAC,MAAM,CAAC;IAC5B,EAAE,EAAE,gBAAgB;IACpB,OAAO,EAAE,cAAc,CAAC,MAAM,CAAC;QAC3B,OAAO,EAAE,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC;QAC5B,sBAAsB,EAAE,CAAC,CAAC,MAAM,EAAE;QAClC,OAAO,EAAE,aAAa;QACtB,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,oBAAoB,CAAC;QACrC,QAAQ,EAAE,cAAc;KAC3B,CAAC;IACF,MAAM,EAAE,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC;CACpC,CAAC,CAAC;AACH,MAAM,CAAC,MAAM,qBAAqB,GAA+B,CAAC,CAAC,MAAM,CAAC;IACtE,OAAO,EAAE,YAAY;IACrB,MAAM,EAAE,CAAC,CAAC,kBAAkB,CAAC,MAAM,EAAE;QACjC,CAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,YAAY,EAAE,CAAC;QAC7D,CAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC;QACxC,CAAC,CAAC,MAAM,CAAC,EAAE,QAAQ,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;QACjE,CAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;KAClF,CAAC;CACL,CAAC,CAAC"}
@@ -0,0 +1,49 @@
1
+ import { runWithAttachmentExecution } from "./resource-lifecycle-state.js";
2
+ const workerSessions = new Map();
3
+ async function createWorkerAttachmentSession(endpoint) {
4
+ const { createAttachmentConnection } = await import("./attachment-connection.js");
5
+ const { createAttachmentExecution } = await import("./attachment-execution.js");
6
+ const connection = createAttachmentConnection(endpoint);
7
+ return { connection, execution: createAttachmentExecution(connection.exchange, endpoint.limits.maxInlineBytes) };
8
+ }
9
+ async function observeWorkerAttachmentSession(key, pending) {
10
+ try {
11
+ const session = await pending;
12
+ session.connection.observeDisconnect(function forgetDisconnectedAttachmentSession() {
13
+ if (workerSessions.get(key) === pending) {
14
+ workerSessions.delete(key);
15
+ }
16
+ });
17
+ return session;
18
+ }
19
+ catch (error) {
20
+ if (workerSessions.get(key) === pending) {
21
+ workerSessions.delete(key);
22
+ }
23
+ throw error;
24
+ }
25
+ }
26
+ async function workerAttachmentSession(endpoint, lifecycle) {
27
+ const key = lifecycle ?? endpoint?.token ?? '';
28
+ const existing = workerSessions.get(key);
29
+ if (existing !== undefined) {
30
+ return existing;
31
+ }
32
+ if (endpoint === null) {
33
+ return null;
34
+ }
35
+ const pending = createWorkerAttachmentSession(endpoint);
36
+ workerSessions.set(key, pending);
37
+ return await observeWorkerAttachmentSession(key, pending);
38
+ }
39
+ export async function runWithWorkerAttachments(endpoint, lifecycle, run) {
40
+ const session = await workerAttachmentSession(endpoint, lifecycle);
41
+ if (session === null) {
42
+ return await run();
43
+ }
44
+ await session.connection.ready;
45
+ return await session.execution.runBranch(endpoint?.branch ?? null, async function executeAttachmentBranch() {
46
+ return await runWithAttachmentExecution(session.execution, run);
47
+ });
48
+ }
49
+ //# sourceMappingURL=attachment-worker-context.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"attachment-worker-context.js","sourceRoot":"","sources":["../../../../source/run/attachment-worker-context.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,0BAA0B,EAAE,MAAM,+BAA+B,CAAC;AAI3E,MAAM,cAAc,GAAG,IAAI,GAAG,EAA4C,CAAC;AAC3E,KAAK,UAAU,6BAA6B,CAAC,QAA4B;IACrE,MAAM,EAAE,0BAA0B,EAAE,GAAG,MAAM,MAAM,CAAC,4BAA4B,CAAC,CAAC;IAClF,MAAM,EAAE,yBAAyB,EAAE,GAAG,MAAM,MAAM,CAAC,2BAA2B,CAAC,CAAC;IAChF,MAAM,UAAU,GAAG,0BAA0B,CAAC,QAAQ,CAAC,CAAC;IACxD,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,yBAAyB,CAAC,UAAU,CAAC,QAAQ,EAAE,QAAQ,CAAC,MAAM,CAAC,cAAc,CAAC,EAAE,CAAC;AACrH,CAAC;AACD,KAAK,UAAU,8BAA8B,CACzC,GAAW,EACX,OAAyC;IAEzC,IAAI,CAAC;QACD,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC;QAC9B,OAAO,CAAC,UAAU,CAAC,iBAAiB,CAAC,SAAS,mCAAmC;YAC7E,IAAI,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,OAAO,EAAE,CAAC;gBACtC,cAAc,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAC/B,CAAC;QACL,CAAC,CAAC,CAAC;QACH,OAAO,OAAO,CAAC;IACnB,CAAC;IAAC,OAAO,KAAc,EAAE,CAAC;QACtB,IAAI,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,OAAO,EAAE,CAAC;YACtC,cAAc,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC/B,CAAC;QACD,MAAM,KAAK,CAAC;IAChB,CAAC;AACL,CAAC;AACD,KAAK,UAAU,uBAAuB,CAClC,QAAmC,EACnC,SAAwB;IAExB,MAAM,GAAG,GAAG,SAAS,IAAI,QAAQ,EAAE,KAAK,IAAI,EAAE,CAAC;IAC/C,MAAM,QAAQ,GAAG,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IACzC,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;QACzB,OAAO,QAAQ,CAAC;IACpB,CAAC;IACD,IAAI,QAAQ,KAAK,IAAI,EAAE,CAAC;QACpB,OAAO,IAAI,CAAC;IAChB,CAAC;IACD,MAAM,OAAO,GAAG,6BAA6B,CAAC,QAAQ,CAAC,CAAC;IACxD,cAAc,CAAC,GAAG,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;IACjC,OAAO,MAAM,8BAA8B,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;AAC9D,CAAC;AACD,MAAM,CAAC,KAAK,UAAU,wBAAwB,CAC1C,QAAmC,EACnC,SAAwB,EACxB,GAAyB;IAEzB,MAAM,OAAO,GAAG,MAAM,uBAAuB,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;IACnE,IAAI,OAAO,KAAK,IAAI,EAAE,CAAC;QACnB,OAAO,MAAM,GAAG,EAAE,CAAC;IACvB,CAAC;IACD,MAAM,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC;IAC/B,OAAO,MAAM,OAAO,CAAC,SAAS,CAAC,SAAS,CAAC,QAAQ,EAAE,MAAM,IAAI,IAAI,EAAE,KAAK,UAAU,uBAAuB;QACrG,OAAO,MAAM,0BAA0B,CAAC,OAAO,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;AACP,CAAC"}
@@ -0,0 +1,117 @@
1
+ import { createReadStream } from 'node:fs';
2
+ import { createStoredRunValue } from "./supervised-run-state.js";
3
+ import { attachmentChunkBytes } from "./attachment-protocol.js";
4
+ function assertWritableAttachmentStream(state) {
5
+ const { operations, closing, pending, failure } = state;
6
+ operations.verify();
7
+ if (closing.read() !== null || pending.read() !== null) {
8
+ throw operations.fail('Attachment writes require an open writer and an awaited previous write.', 'writer-state');
9
+ }
10
+ const previousFailure = failure.read();
11
+ if (previousFailure !== null) {
12
+ throw previousFailure;
13
+ }
14
+ }
15
+ function attachmentStreamValidateWrite(state, chunk) {
16
+ assertWritableAttachmentStream(state);
17
+ if (state.kind === 'binary' && typeof chunk === 'string') {
18
+ throw state.operations.fail('Binary attachment writers require bytes.', 'invalid-chunk');
19
+ }
20
+ return typeof chunk === 'string' ? Buffer.from(chunk) : chunk;
21
+ }
22
+ async function attachmentStreamWriteChunks(state, bytes) {
23
+ const { operations } = state;
24
+ for (let offset = 0; offset < bytes.byteLength; offset += attachmentChunkBytes) {
25
+ await operations.send({
26
+ data: Buffer.from(bytes.subarray(offset, offset + attachmentChunkBytes)).toString('base64'),
27
+ kind: 'write',
28
+ writer: state.writer
29
+ });
30
+ }
31
+ }
32
+ async function attachmentStreamWrite(state, chunk) {
33
+ const { pending, failure, operations } = state;
34
+ pending.write(attachmentStreamWriteChunks(state, attachmentStreamValidateWrite(state, chunk)));
35
+ try {
36
+ await pending.read();
37
+ }
38
+ catch (error) {
39
+ const writeError = error instanceof Error ? error : operations.fail('Attachment write failed.', 'write-error');
40
+ failure.write(writeError);
41
+ throw writeError;
42
+ }
43
+ finally {
44
+ pending.write(null);
45
+ }
46
+ }
47
+ async function attachmentStreamOmit(state) {
48
+ const { operations } = state;
49
+ await operations.send({ kind: 'omit', writer: state.writer });
50
+ }
51
+ async function attachmentStreamSettleWrite(state) {
52
+ const { pending } = state;
53
+ try {
54
+ await pending.read();
55
+ }
56
+ catch {
57
+ }
58
+ }
59
+ async function attachmentStreamCloseOnce(state, reason) {
60
+ const { operations, failure } = state;
61
+ await attachmentStreamSettleWrite(state);
62
+ const completion = failure.read() !== null && reason === 'complete' ? 'write-error' : reason;
63
+ const response = await operations.send({ kind: 'close', reason: completion, writer: state.writer });
64
+ operations.release();
65
+ if (response.kind !== 'closed') {
66
+ throw operations.fail('Attachment close returned an invalid reply.', 'protocol');
67
+ }
68
+ if (reason === 'unclosed' && failure.read() === null) {
69
+ operations.fail('Attachment writer was not closed.', 'unclosed');
70
+ }
71
+ return response.artifact;
72
+ }
73
+ async function attachmentStreamFinish(state, reason) {
74
+ const { closing } = state;
75
+ closing.write(closing.read() ?? attachmentStreamCloseOnce(state, reason));
76
+ const result = closing.read();
77
+ if (result === null) {
78
+ throw new Error('Attachment writer did not start closing.');
79
+ }
80
+ return await result;
81
+ }
82
+ async function attachmentStreamClose(state) {
83
+ const { closing, operations, failure } = state;
84
+ if (closing.read() === null) {
85
+ operations.verify();
86
+ }
87
+ const artifact = await attachmentStreamFinish(state, 'complete');
88
+ const previousFailure = failure.read();
89
+ if (previousFailure !== null) {
90
+ throw previousFailure;
91
+ }
92
+ return artifact;
93
+ }
94
+ export function createAttachmentStream(operations, writer, kind) {
95
+ const owner = operations.owner;
96
+ const pending = createStoredRunValue(null);
97
+ const closing = createStoredRunValue(null);
98
+ const failure = createStoredRunValue(null);
99
+ const state = { operations, writer, kind, owner, pending, closing, failure };
100
+ return {
101
+ owner: state.owner,
102
+ write: attachmentStreamWrite.bind(null, state),
103
+ omit: attachmentStreamOmit.bind(null, state),
104
+ finish: attachmentStreamFinish.bind(null, state),
105
+ close: attachmentStreamClose.bind(null, state)
106
+ };
107
+ }
108
+ export async function copyAttachmentFile(writer, filePath) {
109
+ const chunks = createReadStream(filePath, { highWaterMark: attachmentChunkBytes });
110
+ for await (const chunk of chunks) {
111
+ if (!(chunk instanceof Uint8Array)) {
112
+ throw new TypeError('Attachment file reader did not return bytes.');
113
+ }
114
+ await writer.write(chunk);
115
+ }
116
+ }
117
+ //# sourceMappingURL=attachment-writer.js.map