@bitkyc08/opencodex 2.10.0 → 2.10.1
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/AGENTS_INSTALL.md +77 -0
- package/README.md +4 -10
- package/bin/ocx.mjs +71 -18
- package/gui/dist/assets/index-Cd6_PBKn.css +1 -0
- package/gui/dist/assets/index-ChZQsmBY.js +70 -0
- package/gui/dist/index.html +2 -2
- package/gui/dist/provider-icons/alibaba-color.svg +1 -1
- package/gui/dist/provider-icons/antigravity-color.svg +1 -1
- package/gui/dist/provider-icons/claude-color.svg +1 -1
- package/gui/dist/provider-icons/cline-color.svg +16 -0
- package/gui/dist/provider-icons/cloudflare-ai-gateway-color.svg +1 -1
- package/gui/dist/provider-icons/copilot-color.svg +1 -1
- package/gui/dist/provider-icons/cursor-color.svg +1 -1
- package/gui/dist/provider-icons/deepseek-color.svg +1 -1
- package/gui/dist/provider-icons/firepass-color.svg +1 -1
- package/gui/dist/provider-icons/fireworks-color.svg +1 -1
- package/gui/dist/provider-icons/gemini-color.svg +1 -1
- package/gui/dist/provider-icons/github-copilot-color.svg +1 -1
- package/gui/dist/provider-icons/gitlab-duo-color.svg +1 -1
- package/gui/dist/provider-icons/grok.svg +1 -1
- package/gui/dist/provider-icons/groq-color.svg +1 -1
- package/gui/dist/provider-icons/huggingface-color.svg +1 -1
- package/gui/dist/provider-icons/kimi-color.svg +1 -1
- package/gui/dist/provider-icons/kiro-color.svg +2 -2
- package/gui/dist/provider-icons/lm-studio-color.svg +1 -1
- package/gui/dist/provider-icons/mistral-color.svg +1 -1
- package/gui/dist/provider-icons/moonshot-color.svg +1 -1
- package/gui/dist/provider-icons/nvidia-color.svg +1 -1
- package/gui/dist/provider-icons/ollama-color.svg +1 -1
- package/gui/dist/provider-icons/openai.svg +1 -1
- package/gui/dist/provider-icons/opencode.svg +2 -1
- package/gui/dist/provider-icons/openrouter-color.svg +1 -1
- package/gui/dist/provider-icons/pi.svg +2 -2
- package/gui/dist/provider-icons/qianfan-color.svg +1 -1
- package/gui/dist/provider-icons/qwen-portal-color.svg +1 -1
- package/gui/dist/provider-icons/vercel-ai-gateway-color.svg +1 -1
- package/gui/dist/provider-icons/vllm-color.svg +1 -1
- package/gui/dist/provider-icons/xiaomi-color.svg +1 -1
- package/package.json +8 -4
- package/src/adapters/anthropic.ts +208 -14
- package/src/adapters/base.ts +16 -5
- package/src/adapters/cursor/effort-map.ts +3 -2
- package/src/adapters/cursor/framing.ts +39 -0
- package/src/adapters/cursor/live-transport.ts +105 -95
- package/src/adapters/cursor/native-exec.ts +32 -6
- package/src/adapters/cursor/protobuf-request.ts +20 -15
- package/src/adapters/cursor/request-builder.ts +21 -7
- package/src/adapters/cursor/types.ts +7 -0
- package/src/adapters/google-antigravity-replay.ts +237 -21
- package/src/adapters/google-truncation.ts +11 -0
- package/src/adapters/google.ts +50 -9
- package/src/adapters/identity.ts +39 -6
- package/src/adapters/kiro-errors.ts +11 -0
- package/src/adapters/kiro-events.ts +19 -1
- package/src/adapters/kiro-thinking.ts +10 -2
- package/src/adapters/kiro-tools.ts +10 -1
- package/src/adapters/kiro.ts +37 -11
- package/src/adapters/openai-chat.ts +284 -83
- package/src/adapters/openai-responses.ts +182 -24
- package/src/bridge.ts +177 -7
- package/src/chat/outbound.ts +78 -23
- package/src/claude/agents-inject.ts +27 -5
- package/src/claude/inbound.ts +11 -1
- package/src/claude/model-info.ts +13 -10
- package/src/claude/outbound.ts +17 -0
- package/src/cli/account-api.ts +24 -0
- package/src/cli/account-auth.ts +31 -6
- package/src/cli/account-main.ts +317 -0
- package/src/cli/account.ts +5 -0
- package/src/cli/claude.ts +2 -1
- package/src/cli/doctor.ts +93 -22
- package/src/cli/export-command.ts +26 -12
- package/src/cli/help.ts +8 -6
- package/src/cli/index.ts +56 -22
- package/src/cli/integrations.ts +84 -1
- package/src/cli/observe.ts +54 -1
- package/src/cli/opencode.ts +2 -1
- package/src/cli/provider-runtime.ts +18 -1
- package/src/cli/route-policy.ts +92 -0
- package/src/cli/runtime-api.ts +6 -3
- package/src/cli/star-prompt.ts +71 -15
- package/src/cli/status.ts +1 -1
- package/src/cli/v2.ts +36 -9
- package/src/clients/config-export.ts +687 -10
- package/src/codex/account-lifecycle.ts +30 -5
- package/src/codex/account-usability.ts +22 -2
- package/src/codex/admission.ts +255 -0
- package/src/codex/auth-api.ts +427 -140
- package/src/codex/auth-context.ts +155 -30
- package/src/codex/autostart-health.ts +8 -1
- package/src/codex/catalog/account-models.ts +62 -0
- package/src/codex/catalog/aggregation.ts +14 -1
- package/src/codex/catalog/bundled.ts +282 -32
- package/src/codex/catalog/filesystem-evidence.ts +302 -0
- package/src/codex/catalog/metadata.ts +51 -6
- package/src/codex/catalog/parsing.ts +6 -3
- package/src/codex/catalog/provider-fetch.ts +576 -41
- package/src/codex/catalog/sync.ts +505 -66
- package/src/codex/catalog-admission.ts +197 -0
- package/src/codex/catalog-write-serialization.ts +241 -0
- package/src/codex/catalog.ts +2 -1
- package/src/codex/codex-write-lock.ts +372 -0
- package/src/codex/convergence-types.ts +593 -0
- package/src/codex/convergence.ts +441 -0
- package/src/codex/desired-state.ts +177 -0
- package/src/codex/features.ts +52 -8
- package/src/codex/generation.ts +202 -0
- package/src/codex/history-job.ts +257 -0
- package/src/codex/history-lock.ts +241 -0
- package/src/codex/history-migration-guardian.ts +18 -5
- package/src/codex/history-provider.ts +9 -2
- package/src/codex/history-transition.ts +105 -0
- package/src/codex/history-worker.ts +176 -0
- package/src/codex/inject-coordination.ts +245 -0
- package/src/codex/inject.ts +605 -124
- package/src/codex/integration-record.ts +266 -0
- package/src/codex/internal/catalog-writer.ts +203 -0
- package/src/codex/internal/history-writer.ts +80 -0
- package/src/codex/journal.ts +10 -1
- package/src/codex/main-account-cache.ts +24 -0
- package/src/codex/management-convergence.ts +114 -0
- package/src/codex/native-main-admission.ts +47 -0
- package/src/codex/native-main-auth-temp.ts +187 -0
- package/src/codex/native-main-claim.ts +167 -0
- package/src/codex/native-main-lock-file.ts +158 -0
- package/src/codex/native-main-owner.ts +315 -0
- package/src/codex/native-profile-api.ts +247 -0
- package/src/codex/native-profile-manager.ts +1512 -0
- package/src/codex/native-profile-processes.ts +121 -0
- package/src/codex/native-profile-recovery.ts +99 -0
- package/src/codex/native-profile-stage-store.ts +387 -0
- package/src/codex/native-profile-startup.ts +340 -0
- package/src/codex/native-profile-store.ts +855 -0
- package/src/codex/native-profile-types.ts +120 -0
- package/src/codex/native-residue.ts +557 -0
- package/src/codex/project-config-warnings.ts +18 -4
- package/src/codex/prompt-journal.ts +311 -0
- package/src/codex/prompt-layers.ts +967 -0
- package/src/codex/prompt-lock.ts +143 -0
- package/src/codex/quota-rejection.ts +224 -0
- package/src/codex/quota.ts +86 -3
- package/src/codex/routing.ts +299 -62
- package/src/codex/runtime.ts +159 -38
- package/src/codex/shim.ts +39 -13
- package/src/codex/subagent-model-fallback.ts +73 -12
- package/src/codex/transition-state.ts +604 -0
- package/src/codex/upstream-host-health.ts +70 -0
- package/src/codex/user-identity.ts +266 -0
- package/src/codex/write-coordination.ts +114 -0
- package/src/config.ts +562 -26
- package/src/generated/jawcode-model-metadata.ts +2 -2
- package/src/grok/inject.ts +15 -4
- package/src/grok/inspect.ts +45 -0
- package/src/images/loop.ts +113 -20
- package/src/integrations/config-io.ts +151 -0
- package/src/integrations/journal.ts +315 -0
- package/src/integrations/merge.ts +135 -0
- package/src/integrations/native/ownership-preflight.ts +165 -0
- package/src/integrations/ownership.ts +111 -0
- package/src/integrations/registry.ts +101 -0
- package/src/integrations/serialize.ts +235 -0
- package/src/integrations/state.ts +290 -0
- package/src/integrations/store.ts +103 -0
- package/src/integrations/writer.ts +492 -0
- package/src/lib/bounded-body.ts +46 -8
- package/src/lib/bun-runtime.ts +110 -1
- package/src/lib/bun-stream-caps.ts +2 -1
- package/src/lib/redact.ts +407 -2
- package/src/lib/shadow-call.ts +24 -0
- package/src/lib/translator-budget.ts +10 -0
- package/src/lib/upstream-reachability.ts +91 -0
- package/src/lib/upstream-retry.ts +154 -2
- package/src/lib/windows-secret-acl.ts +212 -11
- package/src/lib/winsw.ts +9 -3
- package/src/oauth/index.ts +61 -3
- package/src/oauth/key-providers.ts +4 -0
- package/src/oauth/kiro.ts +50 -6
- package/src/oauth/store.ts +31 -6
- package/src/oauth/token-guardian.ts +9 -3
- package/src/providers/codex-capacity.ts +288 -0
- package/src/providers/derive.ts +33 -1
- package/src/providers/free-directory.ts +3 -1
- package/src/providers/key-failover.ts +71 -3
- package/src/providers/openai-sidecar.ts +64 -4
- package/src/providers/openai-virtual-models.ts +1 -0
- package/src/providers/quota.ts +334 -26
- package/src/providers/registry.ts +284 -16
- package/src/providers/xai-transport.ts +11 -4
- package/src/responses/compaction.ts +8 -1
- package/src/responses/hosted-tool-policy.ts +9 -0
- package/src/responses/parser.ts +44 -2
- package/src/responses/reasoning-envelope.ts +9 -1
- package/src/responses/reasoning-replay-cache.ts +105 -0
- package/src/responses/spill-store.ts +45 -8
- package/src/responses/state.ts +161 -17
- package/src/router.ts +302 -16
- package/src/routing/analytics.ts +377 -0
- package/src/routing/capability.ts +204 -0
- package/src/routing/cost.ts +77 -0
- package/src/routing/evaluator.ts +444 -0
- package/src/routing/health.ts +401 -0
- package/src/routing/history/cursor.ts +43 -0
- package/src/routing/history/indexer.ts +590 -0
- package/src/routing/history/schema.ts +72 -0
- package/src/routing/profile.ts +423 -0
- package/src/routing/quota.ts +124 -0
- package/src/routing/request-evidence.ts +45 -0
- package/src/routing/trace.ts +686 -0
- package/src/server/auth-cors.ts +46 -6
- package/src/server/chat-completions.ts +28 -13
- package/src/server/claude-messages.ts +23 -15
- package/src/server/gui-static.ts +39 -10
- package/src/server/images.ts +10 -1
- package/src/server/index.ts +238 -52
- package/src/server/lifecycle.ts +228 -9
- package/src/server/live.ts +14 -3
- package/src/server/management/agent-settings-routes.ts +64 -14
- package/src/server/management/combo-routes.ts +5 -5
- package/src/server/management/config-routes.ts +1 -1
- package/src/server/management/context.ts +42 -2
- package/src/server/management/integration-routes.ts +538 -0
- package/src/server/management/logs-usage-routes.ts +1 -1
- package/src/server/management/model-routes.ts +32 -113
- package/src/server/management/model-rows.ts +117 -0
- package/src/server/management/native-integration-routes.ts +587 -0
- package/src/server/management/oauth-account-routes.ts +1 -1
- package/src/server/management/provider-routes.ts +218 -117
- package/src/server/management/request-history-routes.ts +191 -0
- package/src/server/management/routing-analytics-routes.ts +74 -0
- package/src/server/management/routing-profile-routes.ts +177 -0
- package/src/server/management/shared.ts +2 -2
- package/src/server/management/sidebar-routes.ts +47 -31
- package/src/server/management/sync-response.ts +69 -0
- package/src/server/management/system-restart.ts +276 -24
- package/src/server/management/system-routes.ts +4 -0
- package/src/server/management-api.ts +84 -9
- package/src/server/management-auth.ts +43 -5
- package/src/server/relay-eager.ts +82 -42
- package/src/server/relay.ts +120 -6
- package/src/server/request-log.ts +26 -6
- package/src/server/responses/collaboration.ts +63 -8
- package/src/server/responses/compact.ts +272 -41
- package/src/server/responses/core.ts +730 -132
- package/src/server/responses/fetch-helpers.ts +15 -1
- package/src/server/responses-item-id-repair.ts +32 -3
- package/src/server/responses-json-events.ts +52 -0
- package/src/server/responses-snapshot-repair.ts +621 -0
- package/src/server/search.ts +51 -6
- package/src/server/sse-payload-rewrite.ts +89 -12
- package/src/server/startup-health-cache.ts +7 -1
- package/src/server/ws-bridge.ts +11 -17
- package/src/service-manager-probe.ts +297 -0
- package/src/service.ts +222 -32
- package/src/tray/windows-tray.ps1 +9 -0
- package/src/tray/windows.ts +15 -7
- package/src/types.ts +194 -14
- package/src/update/index.ts +13 -13
- package/src/update/job.ts +24 -21
- package/src/update/notify.ts +7 -3
- package/src/usage/cost.ts +0 -0
- package/src/usage/expected-prices.ts +129 -10
- package/src/usage/log.ts +50 -15
- package/src/usage/summary.ts +4 -4
- package/src/vision/index.ts +6 -1
- package/src/web-search/loop.ts +161 -34
- package/gui/dist/assets/index-OY43ubAq.css +0 -1
- package/gui/dist/assets/index-YwNnKZcL.js +0 -67
- package/gui/dist/provider-icons/antigravity.svg +0 -1
- package/gui/dist/provider-icons/claude.svg +0 -1
- package/gui/dist/provider-icons/copilot.svg +0 -1
- package/gui/dist/provider-icons/cursor.svg +0 -2
- package/gui/dist/provider-icons/gemini.svg +0 -1
- package/gui/dist/provider-icons/grok-color.svg +0 -1
- package/gui/dist/provider-icons/kiro.svg +0 -14
|
@@ -0,0 +1,492 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Apply, disable and restore an opencodex provider block in a client's config.
|
|
3
|
+
*
|
|
4
|
+
* Everything here exists to keep one promise: a toggle can always be undone.
|
|
5
|
+
* That means every mutation snapshots first, writes atomically, and journals
|
|
6
|
+
* what it did — and that a mutation refuses rather than guesses whenever the
|
|
7
|
+
* file is not in a state we can reason about. A disable that deletes work the
|
|
8
|
+
* user did after us would be worse than never shipping the feature.
|
|
9
|
+
*
|
|
10
|
+
* Design of record: devlog/_plan/260802_client_toggle_api/030 and 031.
|
|
11
|
+
*/
|
|
12
|
+
import { dirname } from "node:path";
|
|
13
|
+
import { EXPORT_CLIENTS, type ExportModel } from "../clients/config-export";
|
|
14
|
+
import { isLoopbackHostname } from "../codex/inject";
|
|
15
|
+
import type { OcxConfig } from "../types";
|
|
16
|
+
import { PARSE_FAILED, defaultIntegrationIO, loadTarget, parseConfig, type IntegrationIO } from "./config-io";
|
|
17
|
+
import { fingerprint, canonicalContribution, fragmentPathsOf, type OwnershipRecord } from "./ownership";
|
|
18
|
+
import { createdContainerPaths, mergeContribution, removeFragments } from "./merge";
|
|
19
|
+
import { INTEGRATION_CLIENTS, isLoopbackOnly, type IntegrationClientId } from "./registry";
|
|
20
|
+
import { classifyIntegration, exportContextOf } from "./state";
|
|
21
|
+
import type { IntegrationState } from "./state";
|
|
22
|
+
import { serializeDocument, UnserializableValueError } from "./serialize";
|
|
23
|
+
import { ClientPathError } from "../clients/config-export";
|
|
24
|
+
import { matchesOperationResult, newOpId, type JournalEntry } from "./journal";
|
|
25
|
+
import { createIntegrationStateStore, type IntegrationStateStore } from "./store";
|
|
26
|
+
|
|
27
|
+
export type RefusalReason =
|
|
28
|
+
| "not_installed"
|
|
29
|
+
| "conflict"
|
|
30
|
+
| "unsafe"
|
|
31
|
+
| "non_loopback"
|
|
32
|
+
| "drift_requires_confirm"
|
|
33
|
+
| "snapshot_expired"
|
|
34
|
+
| "write_failed";
|
|
35
|
+
|
|
36
|
+
export interface WriteOk {
|
|
37
|
+
ok: true;
|
|
38
|
+
changed: boolean;
|
|
39
|
+
state: IntegrationState;
|
|
40
|
+
clientId: IntegrationClientId;
|
|
41
|
+
opId?: string;
|
|
42
|
+
message: string;
|
|
43
|
+
}
|
|
44
|
+
|
|
45
|
+
export interface WriteRefused {
|
|
46
|
+
ok: false;
|
|
47
|
+
reason: RefusalReason;
|
|
48
|
+
state: IntegrationState;
|
|
49
|
+
clientId: IntegrationClientId;
|
|
50
|
+
message: string;
|
|
51
|
+
/** Absolute path of a recoverable snapshot, when one exists. */
|
|
52
|
+
snapshotPath?: string;
|
|
53
|
+
/** True when compensation itself failed and the file is intermediate. */
|
|
54
|
+
residual?: boolean;
|
|
55
|
+
}
|
|
56
|
+
|
|
57
|
+
export type WriteOutcome = WriteOk | WriteRefused;
|
|
58
|
+
|
|
59
|
+
export interface IntegrationWriteInput {
|
|
60
|
+
clientId: IntegrationClientId;
|
|
61
|
+
models: readonly ExportModel[];
|
|
62
|
+
config: OcxConfig;
|
|
63
|
+
port: number;
|
|
64
|
+
env?: NodeJS.ProcessEnv;
|
|
65
|
+
home?: string;
|
|
66
|
+
store?: IntegrationStateStore;
|
|
67
|
+
io?: IntegrationIO;
|
|
68
|
+
}
|
|
69
|
+
|
|
70
|
+
export interface IntegrationRestoreInput extends IntegrationWriteInput {
|
|
71
|
+
opId: string;
|
|
72
|
+
confirmDrift?: boolean;
|
|
73
|
+
}
|
|
74
|
+
|
|
75
|
+
function refuse(
|
|
76
|
+
clientId: IntegrationClientId,
|
|
77
|
+
reason: RefusalReason,
|
|
78
|
+
state: IntegrationState,
|
|
79
|
+
message: string,
|
|
80
|
+
snapshotPath?: string,
|
|
81
|
+
): WriteRefused {
|
|
82
|
+
return { ok: false, reason, state, clientId, message, ...(snapshotPath ? { snapshotPath } : {}) };
|
|
83
|
+
}
|
|
84
|
+
|
|
85
|
+
function messageOf(error: unknown): string {
|
|
86
|
+
return error instanceof Error ? error.message : String(error);
|
|
87
|
+
}
|
|
88
|
+
|
|
89
|
+
interface CommitArgs {
|
|
90
|
+
io: IntegrationIO;
|
|
91
|
+
store: IntegrationStateStore;
|
|
92
|
+
clientId: IntegrationClientId;
|
|
93
|
+
configPath: string;
|
|
94
|
+
before: string | null;
|
|
95
|
+
nextText: string | null;
|
|
96
|
+
record: OwnershipRecord | null;
|
|
97
|
+
priorRecord: OwnershipRecord | null;
|
|
98
|
+
entry: JournalEntry;
|
|
99
|
+
state: IntegrationState;
|
|
100
|
+
snapshotPath?: string;
|
|
101
|
+
}
|
|
102
|
+
|
|
103
|
+
/**
|
|
104
|
+
* Client file, then ownership record, then journal row.
|
|
105
|
+
*
|
|
106
|
+
* The row is last on purpose: a record without a row is a thin history, while
|
|
107
|
+
* a row without a record would advertise an operation the classifier cannot
|
|
108
|
+
* corroborate. If either bookkeeping step fails we put the file back AND
|
|
109
|
+
* restore the ownership the operation replaced.
|
|
110
|
+
*/
|
|
111
|
+
function commit(args: CommitArgs): WriteOutcome {
|
|
112
|
+
const { io, clientId, configPath } = args;
|
|
113
|
+
try {
|
|
114
|
+
if (args.nextText === null) io.removeFile(configPath);
|
|
115
|
+
else {
|
|
116
|
+
io.mkdirp(dirname(configPath));
|
|
117
|
+
io.writeText(configPath, args.nextText);
|
|
118
|
+
}
|
|
119
|
+
} catch (error) {
|
|
120
|
+
return refuse(clientId, "write_failed", args.state, messageOf(error), args.snapshotPath);
|
|
121
|
+
}
|
|
122
|
+
try {
|
|
123
|
+
if (args.record) io.putRecord(args.record);
|
|
124
|
+
else io.dropRecord(clientId);
|
|
125
|
+
} catch (error) {
|
|
126
|
+
return compensate(args, error, "could not record ownership");
|
|
127
|
+
}
|
|
128
|
+
try {
|
|
129
|
+
io.appendJournal(args.entry);
|
|
130
|
+
} catch (error) {
|
|
131
|
+
return compensate(args, error, "could not append the journal row");
|
|
132
|
+
}
|
|
133
|
+
return {
|
|
134
|
+
ok: true,
|
|
135
|
+
changed: true,
|
|
136
|
+
state: args.state,
|
|
137
|
+
clientId,
|
|
138
|
+
opId: args.entry.opId,
|
|
139
|
+
message: "ok",
|
|
140
|
+
};
|
|
141
|
+
}
|
|
142
|
+
|
|
143
|
+
function compensate(args: CommitArgs, cause: unknown, what: string): WriteRefused {
|
|
144
|
+
const { io, clientId, configPath } = args;
|
|
145
|
+
try {
|
|
146
|
+
if (args.before === null) io.removeFile(configPath);
|
|
147
|
+
else io.writeText(configPath, args.before);
|
|
148
|
+
if (args.priorRecord) io.putRecord(args.priorRecord);
|
|
149
|
+
else io.dropRecord(clientId);
|
|
150
|
+
} catch {
|
|
151
|
+
// Say so. A false "rolled back" is worse than the original error, because
|
|
152
|
+
// the user would stop looking for the file we left half-written.
|
|
153
|
+
return {
|
|
154
|
+
ok: false,
|
|
155
|
+
reason: "write_failed",
|
|
156
|
+
state: args.state,
|
|
157
|
+
clientId,
|
|
158
|
+
residual: true,
|
|
159
|
+
...(args.snapshotPath ? { snapshotPath: args.snapshotPath } : {}),
|
|
160
|
+
message: `${what}, and the change could not be rolled back. The file or its ownership record is in an intermediate state; the backup is at ${args.snapshotPath ?? "(none)"}.`,
|
|
161
|
+
};
|
|
162
|
+
}
|
|
163
|
+
return refuse(clientId, "write_failed", args.state, `${what}; the change was rolled back. Cause: ${messageOf(cause)}`, args.snapshotPath);
|
|
164
|
+
}
|
|
165
|
+
|
|
166
|
+
function snapshotAbsPath(store: IntegrationStateStore, entry: JournalEntry): string | undefined {
|
|
167
|
+
const snapshot = store.readSnapshot(entry);
|
|
168
|
+
return snapshot.kind === "stored" ? snapshot.path : undefined;
|
|
169
|
+
}
|
|
170
|
+
|
|
171
|
+
/** Shared preflight: detect, gate, read, parse and classify. */
|
|
172
|
+
function preflight(input: IntegrationWriteInput) {
|
|
173
|
+
const store = input.store ?? createIntegrationStateStore();
|
|
174
|
+
const io = input.io ?? defaultIntegrationIO(store);
|
|
175
|
+
const clientId = input.clientId;
|
|
176
|
+
const spec = INTEGRATION_CLIENTS[clientId];
|
|
177
|
+
const exportSpec = EXPORT_CLIENTS[clientId];
|
|
178
|
+
/*
|
|
179
|
+
* Resolution itself can refuse: a relative OPENCLAW_* selector is rejected
|
|
180
|
+
* because we cannot know the gateway's working directory. That is a refusal
|
|
181
|
+
* about the user's configuration, not an internal fault, so it must not
|
|
182
|
+
* escape as an exception — the collection route would answer 500 for the
|
|
183
|
+
* whole Integrations page because one client is misconfigured.
|
|
184
|
+
*/
|
|
185
|
+
let configPath: string;
|
|
186
|
+
try {
|
|
187
|
+
configPath = spec.configPath(input.env, input.home);
|
|
188
|
+
} catch (error) {
|
|
189
|
+
if (!(error instanceof ClientPathError)) throw error;
|
|
190
|
+
return { failed: refuse(clientId, "unsafe", "unsafe", error.message) } as const;
|
|
191
|
+
}
|
|
192
|
+
store.retryPendingPrunes();
|
|
193
|
+
|
|
194
|
+
const target = loadTarget(io, configPath);
|
|
195
|
+
if (!target.ok) {
|
|
196
|
+
return {
|
|
197
|
+
failed: refuse(clientId, "unsafe", "unsafe",
|
|
198
|
+
target.why === "read-failed"
|
|
199
|
+
? `${configPath} exists but could not be read`
|
|
200
|
+
: `${configPath} is not a regular file`),
|
|
201
|
+
} as const;
|
|
202
|
+
}
|
|
203
|
+
const before = target.before;
|
|
204
|
+
const parsed = parseConfig(before, exportSpec.format);
|
|
205
|
+
if (parsed === PARSE_FAILED) {
|
|
206
|
+
return { failed: refuse(clientId, "unsafe", "unsafe", `${configPath} could not be parsed`) } as const;
|
|
207
|
+
}
|
|
208
|
+
const contribution = exportSpec.buildContribution(exportContextOf(input));
|
|
209
|
+
// A record proves ownership of the file it was written FOR. Matching only by
|
|
210
|
+
// client id let a record for one home authorize a write to another whose
|
|
211
|
+
// bytes happened to hash the same — which deleted a config we never touched.
|
|
212
|
+
const stored = store.readRecords()[clientId] ?? null;
|
|
213
|
+
const record = stored && stored.clientId === clientId && stored.configPath === configPath
|
|
214
|
+
? stored
|
|
215
|
+
: null;
|
|
216
|
+
// `configPath`/`clientId` are load-bearing, not decoration: a record proves
|
|
217
|
+
// ownership of ONE file, and the writer mutates whatever path resolves NOW.
|
|
218
|
+
// Without them a record written for another home directory would grant
|
|
219
|
+
// ownership here and disable would delete fragments it never wrote.
|
|
220
|
+
const classified = classifyIntegration({
|
|
221
|
+
fileText: before, fileIsRegular: true, parsed, record, contribution, configPath, clientId,
|
|
222
|
+
});
|
|
223
|
+
return { failed: undefined, store, io, clientId, spec, exportSpec, configPath, before, parsed, contribution, record, classified } as const;
|
|
224
|
+
}
|
|
225
|
+
|
|
226
|
+
export function applyIntegration(input: IntegrationWriteInput): WriteOutcome {
|
|
227
|
+
const pre = preflight(input);
|
|
228
|
+
if (pre.failed) return pre.failed;
|
|
229
|
+
const { store, io, clientId, spec, exportSpec, configPath, before, parsed, contribution, record, classified } = pre;
|
|
230
|
+
|
|
231
|
+
if (io.statKind(spec.detectDir(input.env, input.home)) !== "dir") {
|
|
232
|
+
return refuse(clientId, "not_installed", "absent", `${clientId} is not installed`);
|
|
233
|
+
}
|
|
234
|
+
if (isLoopbackOnly(clientId) && !isLoopbackHostname(input.config.hostname)) {
|
|
235
|
+
return refuse(clientId, "non_loopback", classified.state,
|
|
236
|
+
`${clientId} has nowhere to put the admission header a non-loopback bind requires, so a generated config would be rejected — and writing one by hand would not help either. Give it loopback access instead, through a tunnel or a local forwarder.`);
|
|
237
|
+
}
|
|
238
|
+
if (classified.state === "conflict") {
|
|
239
|
+
return refuse(clientId, "conflict", "conflict",
|
|
240
|
+
classified.reason === "foreign-edit"
|
|
241
|
+
? `${configPath} changed after opencodex wrote it`
|
|
242
|
+
: `${configPath} already contains an opencodex block we did not write`);
|
|
243
|
+
}
|
|
244
|
+
/*
|
|
245
|
+
* `unsafe` from the classifier means the document is not one we may write
|
|
246
|
+
* through — today that is a container on our fragment path holding a
|
|
247
|
+
* non-object value the merge would replace with `{}`. Unreadable and
|
|
248
|
+
* unparseable files are caught earlier in preflight; this branch exists
|
|
249
|
+
* because the classifier can also refuse a file it CAN read.
|
|
250
|
+
*/
|
|
251
|
+
if (classified.state === "unsafe") {
|
|
252
|
+
return refuse(clientId, "unsafe", "unsafe",
|
|
253
|
+
classified.reason === "blocked-container"
|
|
254
|
+
? `${configPath} holds a value where opencodex would have to write a section, so applying would replace it`
|
|
255
|
+
: `${configPath} cannot be changed safely`);
|
|
256
|
+
}
|
|
257
|
+
if (classified.state === "current") {
|
|
258
|
+
return { ok: true, changed: false, state: "current", clientId, message: "already applied" };
|
|
259
|
+
}
|
|
260
|
+
|
|
261
|
+
// A stale refresh drops what the PREVIOUS record owned before merging: a
|
|
262
|
+
// model that left the catalog would otherwise stay behind as an orphan the
|
|
263
|
+
// new record no longer covers, and disable could never remove it.
|
|
264
|
+
/*
|
|
265
|
+
* The previous record's `createdContainers` has to travel with this removal.
|
|
266
|
+
* Without it the refresh leaves our own empty scaffolding behind in `base`,
|
|
267
|
+
* `createdContainerPaths` then sees the container already present and
|
|
268
|
+
* concludes the user owns it, and the replacement record forgets we made it
|
|
269
|
+
* — so a later disable strands it forever.
|
|
270
|
+
*/
|
|
271
|
+
const base = classified.state === "stale" && record
|
|
272
|
+
? removeFragments(parsed, record.fragmentPaths, new Set(record.createdContainers ?? [])).doc
|
|
273
|
+
: parsed;
|
|
274
|
+
// Computed against the document as it stands BEFORE the merge: afterwards
|
|
275
|
+
// every container exists and "did we create this?" is unanswerable.
|
|
276
|
+
const created = createdContainerPaths(base, contribution);
|
|
277
|
+
/*
|
|
278
|
+
* A document can hold a value its own format cannot round-trip through our
|
|
279
|
+
* renderers. That used to throw straight out of the writer and reach the
|
|
280
|
+
* user as a 500 with no path and no advice; it is a refusal like any other,
|
|
281
|
+
* and the file is untouched because this happens before any write.
|
|
282
|
+
*/
|
|
283
|
+
let text: string;
|
|
284
|
+
try {
|
|
285
|
+
text = serializeDocument(mergeContribution(base, contribution), exportSpec.format);
|
|
286
|
+
} catch (error) {
|
|
287
|
+
if (!(error instanceof UnserializableValueError)) throw error;
|
|
288
|
+
return refuse(clientId, "unsafe", "unsafe",
|
|
289
|
+
`${configPath} contains something opencodex cannot rewrite safely (${error.message}), so it was left alone`);
|
|
290
|
+
}
|
|
291
|
+
|
|
292
|
+
// Compare-before-commit: someone may have written between classify and now.
|
|
293
|
+
const recheck = io.readText(configPath);
|
|
294
|
+
const rechecked = recheck.kind === "text" ? recheck.text : recheck.kind === "missing" ? null : undefined;
|
|
295
|
+
if (rechecked === undefined || rechecked !== before) {
|
|
296
|
+
return refuse(clientId, "conflict", "conflict", `${configPath} changed while applying`);
|
|
297
|
+
}
|
|
298
|
+
|
|
299
|
+
const opId = newOpId();
|
|
300
|
+
const snapshot = store.captureSnapshot(clientId, opId, before);
|
|
301
|
+
const at = new Date(io.now()).toISOString();
|
|
302
|
+
const entry: JournalEntry = {
|
|
303
|
+
opId, clientId, kind: classified.state === "stale" ? "refresh" : "apply", at, configPath,
|
|
304
|
+
snapshot, resultFingerprint: fingerprint(text), resultAbsent: false, priorRecord: record,
|
|
305
|
+
};
|
|
306
|
+
return commit({
|
|
307
|
+
io, store, clientId, configPath, before, nextText: text, state: "current",
|
|
308
|
+
priorRecord: record,
|
|
309
|
+
record: {
|
|
310
|
+
clientId, configPath, fileFingerprint: fingerprint(text),
|
|
311
|
+
blockFingerprint: fingerprint(canonicalContribution(contribution)),
|
|
312
|
+
fragmentPaths: fragmentPathsOf(contribution), createdContainers: created,
|
|
313
|
+
appliedAt: at, opId,
|
|
314
|
+
},
|
|
315
|
+
entry,
|
|
316
|
+
snapshotPath: snapshotAbsPath(store, entry),
|
|
317
|
+
});
|
|
318
|
+
}
|
|
319
|
+
|
|
320
|
+
export function disableIntegration(input: IntegrationWriteInput): WriteOutcome {
|
|
321
|
+
const pre = preflight(input);
|
|
322
|
+
if (pre.failed) return pre.failed;
|
|
323
|
+
const { store, io, clientId, exportSpec, configPath, before, parsed, record, classified } = pre;
|
|
324
|
+
|
|
325
|
+
if (classified.state === "absent") {
|
|
326
|
+
return { ok: true, changed: false, state: "absent", clientId, message: "not applied" };
|
|
327
|
+
}
|
|
328
|
+
if (classified.state === "conflict") {
|
|
329
|
+
return refuse(clientId, "conflict", "conflict",
|
|
330
|
+
classified.reason === "foreign-edit"
|
|
331
|
+
? `${configPath} changed after opencodex wrote it; disabling would discard that edit`
|
|
332
|
+
: `${configPath} contains an opencodex block we did not write`);
|
|
333
|
+
}
|
|
334
|
+
/*
|
|
335
|
+
* `unsafe` reaches here the same way it reaches apply, and the code below
|
|
336
|
+
* dereferences `record` on the assumption that anything past this point is
|
|
337
|
+
* `current` or `stale`. A blocked container has no record, so disable threw
|
|
338
|
+
* a TypeError and the route answered 500 — the GUI locks the switch, but the
|
|
339
|
+
* CLI and direct API callers do not.
|
|
340
|
+
*/
|
|
341
|
+
if (classified.state === "unsafe") {
|
|
342
|
+
return refuse(clientId, "unsafe", "unsafe",
|
|
343
|
+
classified.reason === "blocked-container"
|
|
344
|
+
? `${configPath} holds a value where opencodex would have to read a section, so nothing can be removed safely`
|
|
345
|
+
: `${configPath} cannot be changed safely`);
|
|
346
|
+
}
|
|
347
|
+
|
|
348
|
+
// current | stale only: the file fingerprint still matches our record, so the
|
|
349
|
+
// recorded paths are exactly what we put there.
|
|
350
|
+
const { doc, removed } = removeFragments(
|
|
351
|
+
parsed,
|
|
352
|
+
record!.fragmentPaths,
|
|
353
|
+
new Set(record!.createdContainers ?? []),
|
|
354
|
+
);
|
|
355
|
+
if (!removed) {
|
|
356
|
+
return { ok: true, changed: false, state: "absent", clientId, message: "nothing to remove" };
|
|
357
|
+
}
|
|
358
|
+
let text: string;
|
|
359
|
+
try {
|
|
360
|
+
text = serializeDocument(doc, exportSpec.format);
|
|
361
|
+
} catch (error) {
|
|
362
|
+
if (!(error instanceof UnserializableValueError)) throw error;
|
|
363
|
+
return refuse(clientId, "unsafe", "unsafe",
|
|
364
|
+
`${configPath} contains something opencodex cannot rewrite safely (${error.message}), so nothing was removed`);
|
|
365
|
+
}
|
|
366
|
+
|
|
367
|
+
const recheck = io.readText(configPath);
|
|
368
|
+
const rechecked = recheck.kind === "text" ? recheck.text : recheck.kind === "missing" ? null : undefined;
|
|
369
|
+
if (rechecked === undefined || rechecked !== before) {
|
|
370
|
+
return refuse(clientId, "conflict", "conflict", `${configPath} changed while disabling`);
|
|
371
|
+
}
|
|
372
|
+
|
|
373
|
+
const opId = newOpId();
|
|
374
|
+
const snapshot = store.captureSnapshot(clientId, opId, before);
|
|
375
|
+
const at = new Date(io.now()).toISOString();
|
|
376
|
+
const entry: JournalEntry = {
|
|
377
|
+
opId, clientId, kind: "disable", at, configPath, snapshot,
|
|
378
|
+
resultFingerprint: fingerprint(text), resultAbsent: false, priorRecord: record,
|
|
379
|
+
};
|
|
380
|
+
return commit({
|
|
381
|
+
io, store, clientId, configPath, before, nextText: text, state: "absent",
|
|
382
|
+
record: null, priorRecord: record, entry,
|
|
383
|
+
snapshotPath: snapshotAbsPath(store, entry),
|
|
384
|
+
});
|
|
385
|
+
}
|
|
386
|
+
|
|
387
|
+
export function restoreIntegration(input: IntegrationRestoreInput): WriteOutcome {
|
|
388
|
+
const store = input.store ?? createIntegrationStateStore();
|
|
389
|
+
const io = input.io ?? defaultIntegrationIO(store);
|
|
390
|
+
const entry = store.findOperation(input.opId);
|
|
391
|
+
if (!entry) throw new Error(`unknown operation ${input.opId}`);
|
|
392
|
+
if (entry.clientId !== input.clientId) throw new Error("restore input names a different client than the operation");
|
|
393
|
+
|
|
394
|
+
const clientId = entry.clientId;
|
|
395
|
+
const resolvedPath = INTEGRATION_CLIENTS[clientId].configPath(input.env, input.home);
|
|
396
|
+
// Restore acts on the path the operation was journaled against. Resolving a
|
|
397
|
+
// different path here would let an operation recorded for one home delete a
|
|
398
|
+
// file in another.
|
|
399
|
+
const configPath = entry.configPath;
|
|
400
|
+
if (resolvedPath !== configPath) {
|
|
401
|
+
return refuse(clientId, "conflict", "conflict",
|
|
402
|
+
`that operation was recorded for ${configPath}, but this client now resolves to ${resolvedPath}`);
|
|
403
|
+
}
|
|
404
|
+
const snapshot = store.readSnapshot(entry);
|
|
405
|
+
if (snapshot.kind === "expired") {
|
|
406
|
+
return refuse(clientId, "snapshot_expired", "absent", "that backup has expired");
|
|
407
|
+
}
|
|
408
|
+
const backupHint = snapshot.kind === "stored" ? snapshot.path : undefined;
|
|
409
|
+
|
|
410
|
+
const target = loadTarget(io, configPath);
|
|
411
|
+
if (!target.ok) {
|
|
412
|
+
return refuse(clientId, "unsafe", "unsafe",
|
|
413
|
+
target.why === "read-failed"
|
|
414
|
+
? `${configPath} exists but could not be read; the backup is at ${backupHint ?? "(none)"}`
|
|
415
|
+
: `${configPath} is not a regular file; the backup is at ${backupHint ?? "(none)"}`,
|
|
416
|
+
backupHint);
|
|
417
|
+
}
|
|
418
|
+
const current = target.before;
|
|
419
|
+
|
|
420
|
+
/*
|
|
421
|
+
* Drift: the file changed after the operation we are undoing.
|
|
422
|
+
*
|
|
423
|
+
* Through the shared matcher, because `fingerprint(current ?? "")` treated a
|
|
424
|
+
* MISSING file as an empty one — so restoring an operation whose result was
|
|
425
|
+
* absence, with the file still absent, read as drift and demanded a
|
|
426
|
+
* confirmation for edits nobody had made. The journal meanwhile offered the
|
|
427
|
+
* same row as Undo.
|
|
428
|
+
*/
|
|
429
|
+
if (!matchesOperationResult(entry, current) && !input.confirmDrift) {
|
|
430
|
+
return refuse(clientId, "drift_requires_confirm", "conflict",
|
|
431
|
+
"this file changed after that operation; confirm to replace it (the current version is backed up first)");
|
|
432
|
+
}
|
|
433
|
+
|
|
434
|
+
// Restore is itself journaled and itself undoable: snapshot the CURRENT file
|
|
435
|
+
// first, so a confirmed drift-restore never destroys the newer edits.
|
|
436
|
+
const opId = newOpId();
|
|
437
|
+
const preSnapshot = store.captureSnapshot(clientId, opId, current);
|
|
438
|
+
const restoredText = snapshot.kind === "none" ? null : snapshot.text;
|
|
439
|
+
|
|
440
|
+
// Provenance is RESTORED, never re-derived: `priorRecord` described these
|
|
441
|
+
// exact bytes when the snapshot was taken. Re-deriving it from the file would
|
|
442
|
+
// mean guessing which entries are ours, and a wrong guess deletes a user's.
|
|
443
|
+
const restoredRecord = entry.priorRecord;
|
|
444
|
+
const fresh = EXPORT_CLIENTS[clientId].buildContribution(exportContextOf(input));
|
|
445
|
+
/*
|
|
446
|
+
* Does the restored record actually describe the restored bytes?
|
|
447
|
+
*
|
|
448
|
+
* It usually does — `priorRecord` was written for exactly this snapshot. But
|
|
449
|
+
* a CONFIRMED drift-restore snapshots the user's edited file first, and
|
|
450
|
+
* undoing that restore puts those edited bytes back while carrying a record
|
|
451
|
+
* that describes what opencodex had written. Overwriting the record's
|
|
452
|
+
* fingerprint with the restored bytes then laundered a foreign edit into
|
|
453
|
+
* owned content: the state read `current`, and a later disable deleted
|
|
454
|
+
* fields the user had added by hand.
|
|
455
|
+
*/
|
|
456
|
+
const restoredFingerprint = restoredText === null ? "" : fingerprint(restoredText);
|
|
457
|
+
const recordDescribesBytes = restoredRecord !== null
|
|
458
|
+
&& restoredRecord.fileFingerprint === restoredFingerprint;
|
|
459
|
+
const state: IntegrationState = restoredRecord === null
|
|
460
|
+
? (restoredText === null ? "absent" : "conflict")
|
|
461
|
+
: !recordDescribesBytes
|
|
462
|
+
? "conflict"
|
|
463
|
+
: restoredRecord.blockFingerprint === fingerprint(canonicalContribution(fresh))
|
|
464
|
+
? "current"
|
|
465
|
+
: "stale";
|
|
466
|
+
|
|
467
|
+
const at = new Date(io.now()).toISOString();
|
|
468
|
+
const priorRecord = store.readRecords()[clientId] ?? null;
|
|
469
|
+
const restoreEntry: JournalEntry = {
|
|
470
|
+
opId, clientId, kind: "restore", at, configPath, snapshot: preSnapshot,
|
|
471
|
+
resultFingerprint: restoredText === null ? "" : fingerprint(restoredText),
|
|
472
|
+
resultAbsent: restoredText === null,
|
|
473
|
+
priorRecord,
|
|
474
|
+
};
|
|
475
|
+
return commit({
|
|
476
|
+
io, store, clientId, configPath, before: current, nextText: restoredText, state,
|
|
477
|
+
priorRecord,
|
|
478
|
+
/*
|
|
479
|
+
* Keep the record's ORIGINAL `fileFingerprint`. Restoring bytes it does not
|
|
480
|
+
* describe leaves the classifier reading `conflict`, which is the honest
|
|
481
|
+
* answer — we are no longer sure the block on disk is ours, and refusing
|
|
482
|
+
* is what stops a disable from deleting the user's edit.
|
|
483
|
+
*/
|
|
484
|
+
record: restoredRecord === null ? null : {
|
|
485
|
+
...restoredRecord,
|
|
486
|
+
appliedAt: at,
|
|
487
|
+
opId,
|
|
488
|
+
},
|
|
489
|
+
entry: restoreEntry,
|
|
490
|
+
snapshotPath: snapshotAbsPath(store, restoreEntry),
|
|
491
|
+
});
|
|
492
|
+
}
|
package/src/lib/bounded-body.ts
CHANGED
|
@@ -7,6 +7,18 @@ export const BOUNDED_BODY_TIMEOUT_MS = 5_000;
|
|
|
7
7
|
export interface BoundedBodyOptions {
|
|
8
8
|
/** Abort the read with this signal. Its reason is rethrown by identity. */
|
|
9
9
|
signal?: AbortSignal;
|
|
10
|
+
/**
|
|
11
|
+
* Reject the returned promise with TypeError on malformed or truncated UTF-8
|
|
12
|
+
* instead of replacing invalid bytes, including during timeout-path flushes.
|
|
13
|
+
* Reader cancellation and lock release still run. Defaults to false.
|
|
14
|
+
*/
|
|
15
|
+
fatalUtf8?: boolean;
|
|
16
|
+
/**
|
|
17
|
+
* Byte ceiling for retained body data. Defaults to BOUNDED_BODY_MAX_BYTES (64 KiB),
|
|
18
|
+
* which suits error bodies; callers materializing whole success payloads (e.g. a
|
|
19
|
+
* non-streaming upstream JSON completion) pass a larger explicit budget.
|
|
20
|
+
*/
|
|
21
|
+
maxBytes?: number;
|
|
10
22
|
/** Total wall-clock deadline. Exposed for focused tests. */
|
|
11
23
|
totalTimeoutMs?: number;
|
|
12
24
|
/** Deadline between non-empty raw chunks. Exposed for focused tests. */
|
|
@@ -33,6 +45,19 @@ export interface BoundedBodyResult {
|
|
|
33
45
|
const TOTAL_TIMEOUT = Symbol("bounded body total timeout");
|
|
34
46
|
const INACTIVITY_TIMEOUT = Symbol("bounded body inactivity timeout");
|
|
35
47
|
|
|
48
|
+
/**
|
|
49
|
+
* Test-only instrumentation: how many times the retained buffer was reallocated
|
|
50
|
+
* during the most recent read. The accumulator grows geometrically, so this is
|
|
51
|
+
* logarithmic in the body size and independent of how many chunks the peer sends.
|
|
52
|
+
* The per-chunk array it replaced retained one object per chunk instead, which a
|
|
53
|
+
* fragmenting peer can inflate far past the payload ceiling — a property no
|
|
54
|
+
* correctness assertion can see, which is why it is observable here.
|
|
55
|
+
*/
|
|
56
|
+
let bufferGrowthsForTests = 0;
|
|
57
|
+
export function boundedBodyBufferGrowthsForTests(): number {
|
|
58
|
+
return bufferGrowthsForTests;
|
|
59
|
+
}
|
|
60
|
+
|
|
36
61
|
function timeoutPromise(ms: number, value: symbol): { promise: Promise<symbol>; clear: () => void } {
|
|
37
62
|
let timer: ReturnType<typeof setTimeout> | undefined;
|
|
38
63
|
const promise = new Promise<symbol>((resolve) => {
|
|
@@ -56,8 +81,8 @@ function cancelWithoutWaiting(reader: ReadableStreamDefaultReader<Uint8Array>, r
|
|
|
56
81
|
}
|
|
57
82
|
}
|
|
58
83
|
|
|
59
|
-
function decodeUtf8(chunks: readonly Uint8Array[]): string {
|
|
60
|
-
const decoder = new TextDecoder();
|
|
84
|
+
function decodeUtf8(chunks: readonly Uint8Array[], fatal: boolean): string {
|
|
85
|
+
const decoder = new TextDecoder("utf-8", { fatal });
|
|
61
86
|
let text = "";
|
|
62
87
|
for (const chunk of chunks) text += decoder.decode(chunk, { stream: true });
|
|
63
88
|
// Flush an incomplete trailing UTF-8 sequence deterministically.
|
|
@@ -93,8 +118,13 @@ export async function readBoundedResponseBody(
|
|
|
93
118
|
}
|
|
94
119
|
|
|
95
120
|
const reader = body.getReader();
|
|
96
|
-
const
|
|
121
|
+
const maxBytes = options.maxBytes ?? BOUNDED_BODY_MAX_BYTES;
|
|
122
|
+
// Geometrically growing single buffer: per-chunk arrays would retain one object per
|
|
123
|
+
// transport chunk, which a hostile peer could inflate into metadata amplification far
|
|
124
|
+
// beyond the payload ceiling on large budgets.
|
|
125
|
+
let retained = new Uint8Array(Math.min(maxBytes, 64 * 1024));
|
|
97
126
|
let retainedBytes = 0;
|
|
127
|
+
bufferGrowthsForTests = 0;
|
|
98
128
|
let mustCancel = false;
|
|
99
129
|
let cancelReason: unknown;
|
|
100
130
|
const total = timeoutPromise(options.totalTimeoutMs ?? BOUNDED_BODY_TIMEOUT_MS, TOTAL_TIMEOUT);
|
|
@@ -134,7 +164,7 @@ export async function readBoundedResponseBody(
|
|
|
134
164
|
"TimeoutError",
|
|
135
165
|
);
|
|
136
166
|
return {
|
|
137
|
-
text: decodeUtf8(
|
|
167
|
+
text: decodeUtf8([retained.subarray(0, retainedBytes)], options.fatalUtf8 === true),
|
|
138
168
|
truncated: true,
|
|
139
169
|
timedOut: true,
|
|
140
170
|
totalTimedOut: outcome === TOTAL_TIMEOUT,
|
|
@@ -147,7 +177,7 @@ export async function readBoundedResponseBody(
|
|
|
147
177
|
const { value, done } = outcome as ReadableStreamReadResult<Uint8Array>;
|
|
148
178
|
if (done) {
|
|
149
179
|
return {
|
|
150
|
-
text: decodeUtf8(
|
|
180
|
+
text: decodeUtf8([retained.subarray(0, retainedBytes)], options.fatalUtf8 === true),
|
|
151
181
|
truncated: false,
|
|
152
182
|
timedOut: false,
|
|
153
183
|
totalTimedOut: false,
|
|
@@ -165,10 +195,10 @@ export async function readBoundedResponseBody(
|
|
|
165
195
|
INACTIVITY_TIMEOUT,
|
|
166
196
|
);
|
|
167
197
|
|
|
168
|
-
if (value.byteLength >
|
|
198
|
+
if (value.byteLength > maxBytes - retainedBytes) {
|
|
169
199
|
mustCancel = true;
|
|
170
200
|
cancelReason = new DOMException("Error body size limit reached", "QuotaExceededError");
|
|
171
|
-
|
|
201
|
+
retained = new Uint8Array(0);
|
|
172
202
|
retainedBytes = 0;
|
|
173
203
|
return {
|
|
174
204
|
text: "",
|
|
@@ -181,7 +211,15 @@ export async function readBoundedResponseBody(
|
|
|
181
211
|
};
|
|
182
212
|
}
|
|
183
213
|
|
|
184
|
-
|
|
214
|
+
if (retainedBytes + value.byteLength > retained.length) {
|
|
215
|
+
const grown = new Uint8Array(
|
|
216
|
+
Math.min(maxBytes, Math.max(retained.length * 2, retainedBytes + value.byteLength)),
|
|
217
|
+
);
|
|
218
|
+
grown.set(retained.subarray(0, retainedBytes));
|
|
219
|
+
retained = grown;
|
|
220
|
+
bufferGrowthsForTests += 1;
|
|
221
|
+
}
|
|
222
|
+
retained.set(value, retainedBytes);
|
|
185
223
|
retainedBytes += value.byteLength;
|
|
186
224
|
}
|
|
187
225
|
} catch (error) {
|