@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.
- package/assertion-protocol/source-location-forwarding.js +14 -2
- package/assertion-protocol/source-location-forwarding.js.map +1 -1
- package/assertion-protocol/source-location.js +1 -0
- package/assertion-protocol/source-location.js.map +1 -1
- package/attachments/attachment-context.js +31 -0
- package/attachments/attachment-context.js.map +1 -0
- package/engine/async-leak-diagnostics.js +9 -9
- package/engine/async-leak-diagnostics.js.map +1 -1
- package/engine/case-attempt-execution.js +191 -0
- package/engine/case-attempt-execution.js.map +1 -0
- package/engine/case-execution.d.ts +5 -3
- package/engine/case-execution.d.ts.map +1 -1
- package/engine/case-execution.js +25 -24
- package/engine/case-execution.js.map +1 -1
- package/engine/coverage-artifact.d.ts +1 -0
- package/engine/coverage-artifact.d.ts.map +1 -1
- package/engine/execution-async-leak-policy.js +3 -10
- package/engine/execution-async-leak-policy.js.map +1 -1
- package/engine/execution-global-error-observer.d.ts +2 -1
- package/engine/execution-global-error-observer.d.ts.map +1 -1
- package/engine/execution-global-error-observer.js +19 -15
- package/engine/execution-global-error-observer.js.map +1 -1
- package/engine/execution-options.d.ts +4 -1
- package/engine/execution-options.d.ts.map +1 -1
- package/engine/execution-options.js +7 -2
- package/engine/execution-options.js.map +1 -1
- package/engine/execution-resource-budget-tracking.js.map +1 -1
- package/engine/execution-resource-tracked-cases.js.map +1 -1
- package/engine/execution-result.js.map +1 -1
- package/engine/execution-supervision-state.js.map +1 -1
- package/engine/execution-supervision.d.ts.map +1 -1
- package/engine/execution-supervision.js +50 -33
- package/engine/execution-supervision.js.map +1 -1
- package/engine/execution-test-plan-cases.js +6 -126
- package/engine/execution-test-plan-cases.js.map +1 -1
- package/engine/execution-test-plan-resource-usage.js.map +1 -1
- package/engine/execution.js +3 -3
- package/engine/execution.js.map +1 -1
- package/engine/identity.d.ts +3 -0
- package/engine/identity.d.ts.map +1 -1
- package/engine/reporter-dispatcher.js +1 -0
- package/engine/reporter-dispatcher.js.map +1 -1
- package/engine/reporter-event-queue.js.map +1 -1
- package/engine/reporter.d.ts +2 -1
- package/engine/reporter.d.ts.map +1 -1
- package/engine/reporter.js.map +1 -1
- package/engine/resource-usage.d.ts +22 -0
- package/engine/resource-usage.d.ts.map +1 -0
- package/engine/retry-policy.d.ts +4 -0
- package/engine/retry-policy.d.ts.map +1 -0
- package/engine/retry-policy.js +22 -0
- package/engine/retry-policy.js.map +1 -0
- package/engine/run-artifact.d.ts +45 -0
- package/engine/run-artifact.d.ts.map +1 -0
- package/engine/run-result.d.ts +22 -53
- package/engine/run-result.d.ts.map +1 -1
- package/engine/run-result.js +14 -0
- package/engine/run-result.js.map +1 -1
- package/engine/runner-error-reporting.js.map +1 -1
- package/engine/runtime-attachment.d.ts +62 -0
- package/engine/runtime-attachment.d.ts.map +1 -0
- package/engine/runtime-attachment.js +7 -0
- package/engine/runtime-attachment.js.map +1 -0
- package/engine/test-attempt-history.js +24 -0
- package/engine/test-attempt-history.js.map +1 -0
- package/package.json +25 -6
- package/packages/resources/attachment-context.entry-point.js +2 -0
- package/packages/resources/attachment-context.entry-point.js.map +1 -0
- package/packages/run/config.entry-point.d.ts +3 -1
- package/packages/run/config.entry-point.d.ts.map +1 -1
- package/packages/run/run.entry-point.d.ts +3 -1
- package/packages/run/run.entry-point.d.ts.map +1 -1
- package/packages/run/transcript-store.entry-point.d.ts +3 -0
- package/packages/run/transcript-store.entry-point.d.ts.map +1 -0
- package/packages/run/transcript-store.entry-point.js +2 -0
- package/packages/run/transcript-store.entry-point.js.map +1 -0
- package/packages/simulation/transcript.entry-point.js +2 -0
- package/packages/simulation/transcript.entry-point.js.map +1 -0
- package/readme.md +69 -2
- package/reporters/human-reporter-rendering.js +25 -2
- package/reporters/human-reporter-rendering.js.map +1 -1
- package/reporters/line-reporter.js +6 -1
- package/reporters/line-reporter.js.map +1 -1
- package/resources/resource-definition-brand.js +1 -1
- package/resources/resource-definition-brand.js.map +1 -1
- package/resources/resource-definition-types.d.ts.map +1 -1
- package/resources/resource-scenario-binding.js +2 -0
- package/resources/resource-scenario-binding.js.map +1 -1
- package/resources/resource-session.d.ts.map +1 -1
- package/resources/resources.d.ts.map +1 -1
- package/resources/resources.js +3 -6
- package/resources/resources.js.map +1 -1
- package/resources/runtime-definition.d.ts.map +1 -1
- package/resources/runtime-definition.js +1 -1
- package/resources/runtime-definition.js.map +1 -1
- package/resources/runtime-matrix-definition.d.ts.map +1 -1
- package/resources/runtime-matrix-definition.js +2 -2
- package/resources/runtime-matrix-definition.js.map +1 -1
- package/run/attachment-connection.js +108 -0
- package/run/attachment-connection.js.map +1 -0
- package/run/attachment-coordinator-context.js +9 -0
- package/run/attachment-coordinator-context.js.map +1 -0
- package/run/attachment-coordinator.js +77 -0
- package/run/attachment-coordinator.js.map +1 -0
- package/run/attachment-eligibility.js +21 -0
- package/run/attachment-eligibility.js.map +1 -0
- package/run/attachment-execution.js +255 -0
- package/run/attachment-execution.js.map +1 -0
- package/run/attachment-failure.js +63 -0
- package/run/attachment-failure.js.map +1 -0
- package/run/attachment-frames.js +22 -0
- package/run/attachment-frames.js.map +1 -0
- package/run/attachment-json.js +118 -0
- package/run/attachment-json.js.map +1 -0
- package/run/attachment-protocol.js +5 -0
- package/run/attachment-protocol.js.map +1 -0
- package/run/attachment-record.js +60 -0
- package/run/attachment-record.js.map +1 -0
- package/run/attachment-reporter.js +39 -0
- package/run/attachment-reporter.js.map +1 -0
- package/run/attachment-results.js +115 -0
- package/run/attachment-results.js.map +1 -0
- package/run/attachment-retention.js +153 -0
- package/run/attachment-retention.js.map +1 -0
- package/run/attachment-run.js +36 -0
- package/run/attachment-run.js.map +1 -0
- package/run/attachment-server.js +136 -0
- package/run/attachment-server.js.map +1 -0
- package/run/attachment-storage-session.js +21 -0
- package/run/attachment-storage-session.js.map +1 -0
- package/run/attachment-store.js +330 -0
- package/run/attachment-store.js.map +1 -0
- package/run/attachment-wire-schema.js +123 -0
- package/run/attachment-wire-schema.js.map +1 -0
- package/run/attachment-worker-context.js +49 -0
- package/run/attachment-worker-context.js.map +1 -0
- package/run/attachment-writer.js +117 -0
- package/run/attachment-writer.js.map +1 -0
- package/run/capability-policy.js +10 -5
- package/run/capability-policy.js.map +1 -1
- package/run/collected-run-plan.js.map +1 -1
- package/run/coverage-config-schema.d.ts +4 -4
- package/run/coverage-loaded-sources.js +112 -0
- package/run/coverage-loaded-sources.js.map +1 -0
- package/run/coverage-native-data.js +18 -0
- package/run/coverage-native-data.js.map +1 -0
- package/run/coverage-paths.js +45 -4
- package/run/coverage-paths.js.map +1 -1
- package/run/coverage-reporting.js +104 -8
- package/run/coverage-reporting.js.map +1 -1
- package/run/coverage-run-record.js +50 -0
- package/run/coverage-run-record.js.map +1 -0
- package/run/coverage-runtime-source.js +37 -0
- package/run/coverage-runtime-source.js.map +1 -0
- package/run/coverage-session.js +344 -6
- package/run/coverage-session.js.map +1 -1
- package/run/coverage-source-map.js +125 -0
- package/run/coverage-source-map.js.map +1 -0
- package/run/coverage-source-selection.js +120 -0
- package/run/coverage-source-selection.js.map +1 -0
- package/run/duration-history.d.ts.map +1 -1
- package/run/duration-history.js +1 -0
- package/run/duration-history.js.map +1 -1
- package/run/node-run-execution.js +25 -0
- package/run/node-run-execution.js.map +1 -0
- package/run/node-runtime-state-store.js +38 -0
- package/run/node-runtime-state-store.js.map +1 -0
- package/run/node-worker-pool-factory.js +0 -31
- package/run/node-worker-pool-factory.js.map +1 -1
- package/run/path-containment.js +6 -1
- package/run/path-containment.js.map +1 -1
- package/run/permission-denial-runtime-policy.js +21 -13
- package/run/permission-denial-runtime-policy.js.map +1 -1
- package/run/recorded-coverage-run.js +45 -0
- package/run/recorded-coverage-run.js.map +1 -0
- package/run/recorded-coverage-session.js +50 -0
- package/run/recorded-coverage-session.js.map +1 -0
- package/run/resource-lifecycle-projection.js +6 -2
- package/run/resource-lifecycle-projection.js.map +1 -1
- package/run/resource-lifecycle-session-acquirer.js +13 -6
- package/run/resource-lifecycle-session-acquirer.js.map +1 -1
- package/run/resource-lifecycle-startup-budget.js +1 -0
- package/run/resource-lifecycle-startup-budget.js.map +1 -1
- package/run/resource-lifecycle-state.d.ts +4 -2
- package/run/resource-lifecycle-state.d.ts.map +1 -1
- package/run/resource-lifecycle-state.js +19 -3
- package/run/resource-lifecycle-state.js.map +1 -1
- package/run/resource-lifecycle.js +57 -13
- package/run/resource-lifecycle.js.map +1 -1
- package/run/resource-usage.js.map +1 -1
- package/run/retry-artifact-retention.js +37 -0
- package/run/retry-artifact-retention.js.map +1 -0
- package/run/run-config-schema.d.ts +48 -4
- package/run/run-config-schema.d.ts.map +1 -1
- package/run/run-config-schema.js +18 -0
- package/run/run-config-schema.js.map +1 -1
- package/run/run-config.d.ts +9 -6
- package/run/run-config.d.ts.map +1 -1
- package/run/run-config.js +6 -1
- package/run/run-config.js.map +1 -1
- package/run/run-coverage.js +5 -2
- package/run/run-coverage.js.map +1 -1
- package/run/run-engine-primitives.d.ts +13 -1
- package/run/run-engine-primitives.d.ts.map +1 -1
- package/run/run-errors.d.ts.map +1 -1
- package/run/run-errors.js +1 -0
- package/run/run-errors.js.map +1 -1
- package/run/run-execution-config.d.ts +5 -0
- package/run/run-execution-config.d.ts.map +1 -1
- package/run/run-facts.js +3 -0
- package/run/run-facts.js.map +1 -1
- package/run/run-isolated-command.js +3 -0
- package/run/run-isolated-command.js.map +1 -1
- package/run/run-local-command.js +12 -83
- package/run/run-local-command.js.map +1 -1
- package/run/run-local-coverage.js +72 -0
- package/run/run-local-coverage.js.map +1 -0
- package/run/run-local-execution.js +3 -1
- package/run/run-local-execution.js.map +1 -1
- package/run/run-local-resolution.js +6 -3
- package/run/run-local-resolution.js.map +1 -1
- package/run/run-orchestrator-dependencies.d.ts +1 -1
- package/run/run-orchestrator-dependencies.d.ts.map +1 -1
- package/run/run-orchestrator.d.ts.map +1 -1
- package/run/run-orchestrator.js +6 -18
- package/run/run-orchestrator.js.map +1 -1
- package/run/run-record-error.js +26 -0
- package/run/run-record-error.js.map +1 -0
- package/run/run-record-input.js +34 -0
- package/run/run-record-input.js.map +1 -0
- package/run/run-record-outcomes.d.ts +49 -0
- package/run/run-record-outcomes.d.ts.map +1 -0
- package/run/run-record-result.js +58 -0
- package/run/run-record-result.js.map +1 -0
- package/run/run-record-storage.js +51 -0
- package/run/run-record-storage.js.map +1 -0
- package/run/run-record-types.d.ts +73 -0
- package/run/run-record-types.d.ts.map +1 -0
- package/run/run-record-versions.js +51 -0
- package/run/run-record-versions.js.map +1 -0
- package/run/run-record.js +48 -0
- package/run/run-record.js.map +1 -0
- package/run/run-result-reporting.js +16 -0
- package/run/run-result-reporting.js.map +1 -0
- package/run/run-supervised-finalization.js +8 -0
- package/run/run-supervised-finalization.js.map +1 -0
- package/run/run-supervised-process.js +34 -33
- package/run/run-supervised-process.js.map +1 -1
- package/run/run-support.js +23 -9
- package/run/run-support.js.map +1 -1
- package/run/run-types.d.ts +7 -5
- package/run/run-types.d.ts.map +1 -1
- package/run/run-types.js.map +1 -1
- package/run/run-validation.js +9 -0
- package/run/run-validation.js.map +1 -1
- package/run/runtime-attachment-boundary.js +11 -0
- package/run/runtime-attachment-boundary.js.map +1 -0
- package/run/supervised-child-test-plan.js +2 -0
- package/run/supervised-child-test-plan.js.map +1 -1
- package/run/supervised-child.js +12 -9
- package/run/supervised-child.js.map +1 -1
- package/run/supervised-output-capture.js +66 -0
- package/run/supervised-output-capture.js.map +1 -0
- package/run/supervised-protocol.js.map +1 -1
- package/run/supervised-run-collection.js +3 -1
- package/run/supervised-run-collection.js.map +1 -1
- package/run/supervised-run-resource-policy.js +12 -1
- package/run/supervised-run-resource-policy.js.map +1 -1
- package/run/supervised-run-runtime.js +25 -30
- package/run/supervised-run-runtime.js.map +1 -1
- package/run/supervised-run-state.js +64 -81
- package/run/supervised-run-state.js.map +1 -1
- package/run/supervised-run.js +14 -4
- package/run/supervised-run.js.map +1 -1
- package/run/worker-pool-collection.js +1 -0
- package/run/worker-pool-collection.js.map +1 -1
- package/run/worker-pool-command.js +2 -0
- package/run/worker-pool-command.js.map +1 -1
- package/run/worker-pool-execution.js +12 -3
- package/run/worker-pool-execution.js.map +1 -1
- package/run/worker-pool-hedged-arbitration.js +11 -2
- package/run/worker-pool-hedged-arbitration.js.map +1 -1
- package/run/worker-pool-host-process.js.map +1 -1
- package/run/worker-pool-host-protocol.js.map +1 -1
- package/run/worker-pool-protocol.js.map +1 -1
- package/run/worker-pool-resource-lifecycle-execution.js +1 -0
- package/run/worker-pool-resource-lifecycle-execution.js.map +1 -1
- package/run/worker-pool-resource-tracking.js +2 -2
- package/run/worker-pool-resource-tracking.js.map +1 -1
- package/run/worker-pool-results.js +7 -9
- package/run/worker-pool-results.js.map +1 -1
- package/run/worker-pool-run.js +10 -3
- package/run/worker-pool-run.js.map +1 -1
- package/run/worker-pool-runtime.js +3 -7
- package/run/worker-pool-runtime.js.map +1 -1
- package/run/worker-pool-task-events.js +12 -4
- package/run/worker-pool-task-events.js.map +1 -1
- package/run/worker-pool-worker-plan.js +1 -0
- package/run/worker-pool-worker-plan.js.map +1 -1
- package/run/worker-pool-worker-resources.js +1 -0
- package/run/worker-pool-worker-resources.js.map +1 -1
- package/run/worker-pool-worker.js +15 -4
- package/run/worker-pool-worker.js.map +1 -1
- package/sbom.cdx.json +65 -14
- package/transcript/http-transcript.js +3 -0
- package/transcript/http-transcript.js.map +1 -0
- package/transcript/transcript-store.d.ts +25 -0
- package/transcript/transcript-store.d.ts.map +1 -0
- package/transcript/transcript-store.js +73 -0
- 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
|