@nowcrew/daemon 0.6.103 → 0.6.105

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (49) hide show
  1. package/dist/external-output.js +96 -69
  2. package/dist/local-executor.js +10 -22
  3. package/dist/runtime-answer-collector.js +184 -0
  4. package/package.json +1 -1
  5. package/dist/codex-home-launch.js +0 -7
  6. package/dist/codex-home-mode-controller.js +0 -68
  7. package/dist/daemon-migration-controller.js +0 -210
  8. package/dist/daemon-migration-wiring.js +0 -22
  9. package/dist/project-skills/capability.js +0 -109
  10. package/dist/project-skills/controller-convergence.js +0 -57
  11. package/dist/project-skills/projection-set-switch.js +0 -419
  12. package/dist/project-skills/projection-state-domain.js +0 -170
  13. package/dist/project-skills/projection-state-store.js +0 -842
  14. package/dist/project-skills/projection-state-transaction.js +0 -322
  15. package/dist/project-skills/projection-state.js +0 -3
  16. package/dist/project-skills/runtime-root-bootstrap.js +0 -47
  17. package/dist/project-skills/runtime-root-domain.js +0 -268
  18. package/dist/project-skills/runtime-root-gc.js +0 -293
  19. package/dist/project-skills/runtime-root-lease-artifact.js +0 -46
  20. package/dist/project-skills/runtime-root-leases.js +0 -487
  21. package/dist/project-skills/runtime-root-source-identity.js +0 -60
  22. package/dist/project-skills/runtime-root-startup.js +0 -49
  23. package/dist/project-skills/runtime-root-state-artifact-domain.js +0 -143
  24. package/dist/project-skills/runtime-root-state-index.js +0 -356
  25. package/dist/project-skills/runtime-root-store.js +0 -722
  26. package/dist/project-skills/serve-capability.js +0 -28
  27. package/dist/project-skills/serve-startup.js +0 -37
  28. package/dist/runtimes/codex-home-migration-cli.js +0 -26
  29. package/dist/runtimes/codex-home-migration.js +0 -112
  30. package/dist/runtimes/codex-home.js +0 -236
  31. package/dist/storage-control.js +0 -212
  32. package/dist/storage-copy.js +0 -628
  33. package/dist/storage-credential-guard.js +0 -21
  34. package/dist/storage-doctor.js +0 -228
  35. package/dist/storage-manifest.js +0 -1027
  36. package/dist/storage-migration-controller.js +0 -441
  37. package/dist/storage-migration-manifest.js +0 -105
  38. package/dist/storage-migration-policy.js +0 -100
  39. package/dist/storage-runtime-controller.js +0 -361
  40. package/dist/storage-runtime-readiness.js +0 -206
  41. package/dist/storage-safe-fs.js +0 -302
  42. package/dist/storage-status-delivery.js +0 -224
  43. package/dist/storage-status-replay.js +0 -140
  44. package/dist/storage-target-confirmation.js +0 -129
  45. package/dist/storage-worker-command.js +0 -37
  46. package/dist/storage-worker-executor.js +0 -431
  47. package/dist/storage-worker-handoff.js +0 -151
  48. package/dist/storage-worker-operation.js +0 -288
  49. package/dist/storage-worker-status.js +0 -259
@@ -1,28 +0,0 @@
1
- import { WebSocket } from "ws";
2
- import { daemonCapabilityBindings, } from "../machine-info.js";
3
- import { prepareProjectSkillConnectionCapabilities, } from "./capability.js";
4
- export function createProjectSkillConnectionCapabilityProvider(input) {
5
- return async () => daemonCapabilityBindings(input.platform, await prepareProjectSkillConnectionCapabilities(input));
6
- }
7
- export function createProjectSkillMachineHelloPublisher(input) {
8
- let latest = null;
9
- let latestSocket = null;
10
- const republish = () => {
11
- if (latest === null || latestSocket?.readyState !== WebSocket.OPEN)
12
- return;
13
- const hello = input.effective(latest, input.status());
14
- try {
15
- latestSocket.send(JSON.stringify(hello));
16
- input.onSent(hello);
17
- }
18
- catch { /* 非 OPEN,忽略 */ }
19
- };
20
- return Object.freeze({
21
- publish: (hello, socket) => {
22
- latest = hello;
23
- latestSocket = socket;
24
- republish();
25
- },
26
- republish,
27
- });
28
- }
@@ -1,37 +0,0 @@
1
- import { createExecutionJournal } from "../execution-journal.js";
2
- import { reconcileExecutionJournal } from "../execution-recovery.js";
3
- import { dslog, flushSlog } from "../slog.js";
4
- export async function initializeProjectSkillProtectedExecutionJournal(input) {
5
- return input.startup.initializeAfterLeaseProtection(input.config.agentsRoot, async (snapshot) => {
6
- input.protectedExecutionIds.clear();
7
- for (const executionId of snapshot)
8
- input.protectedExecutionIds.add(executionId);
9
- const journal = input.injectedJournal ?? createExecutionJournal(input.config.agentsRoot, {
10
- protectedExecutionIds: () => input.protectedExecutionIds,
11
- onOperationTiming: (event) => {
12
- if (event.operation !== "mark_runtime_ready"
13
- && event.queueWaitMs < 250
14
- && event.operationMs < 250)
15
- return;
16
- dslog("execution.journal_operation", "execution journal 操作完成", {
17
- level: event.outcome === "succeeded" ? "INFO" : "ERROR",
18
- execution_id: event.executionId,
19
- operation: event.operation,
20
- outcome: event.outcome,
21
- queue_wait_ms: event.queueWaitMs,
22
- operation_ms: event.operationMs,
23
- total_ms: event.queueWaitMs + event.operationMs,
24
- });
25
- },
26
- });
27
- await reconcileExecutionJournal(journal, {
28
- agentsRoot: input.config.agentsRoot,
29
- serverUrl: input.config.serverUrl,
30
- ...(input.profileName === undefined ? {} : { profileName: input.profileName }),
31
- log: dslog,
32
- flush: flushSlog,
33
- writeStderr: (line) => process.stderr.write(line),
34
- });
35
- return journal;
36
- });
37
- }
@@ -1,26 +0,0 @@
1
- #!/usr/bin/env node
2
- import { parseArgs } from "node:util";
3
- import { migrateCodexHome } from "./codex-home-migration.js";
4
- const { values } = parseArgs({
5
- args: process.argv.slice(2),
6
- strict: true,
7
- options: {
8
- "agent-home": { type: "string" },
9
- "source-home": { type: "string" },
10
- "backup-root": { type: "string" },
11
- yes: { type: "boolean", default: false },
12
- },
13
- });
14
- function required(value, name) {
15
- if (!value?.trim())
16
- throw new Error(`Missing --${name}`);
17
- return value;
18
- }
19
- if (!values.yes)
20
- throw new Error("Refusing Codex home migration without --yes");
21
- const result = await migrateCodexHome({
22
- agentHome: required(values["agent-home"], "agent-home"),
23
- sourceHome: required(values["source-home"], "source-home"),
24
- backupRoot: required(values["backup-root"], "backup-root"),
25
- });
26
- process.stdout.write(`${result.status}${result.backupDir ? ` backup=${result.backupDir}` : ""}\n`);
@@ -1,112 +0,0 @@
1
- import { execFile } from "node:child_process";
2
- import { promisify } from "node:util";
3
- import { lstat, mkdir, readlink, rename, rm, symlink, unlink, } from "node:fs/promises";
4
- import { join, resolve } from "node:path";
5
- import { randomUUID } from "node:crypto";
6
- import { atomicPrivateWrite } from "../atomic-private-write.js";
7
- import { defaultCodexHome, materializeDefaultCodexHome } from "./codex-home.js";
8
- const execFileAsync = promisify(execFile);
9
- const SQLITE_FILES = ["state_5.sqlite", "state_5.sqlite-wal", "state_5.sqlite-shm"];
10
- const MARKER_NAME = ".codex-home-migration.json";
11
- async function exists(path) {
12
- try {
13
- await lstat(path);
14
- return true;
15
- }
16
- catch (error) {
17
- if (error.code === "ENOENT")
18
- return false;
19
- throw error;
20
- }
21
- }
22
- async function assertNoOpenSqliteFiles(codexHome) {
23
- for (const name of SQLITE_FILES) {
24
- const path = join(codexHome, name);
25
- if (!(await exists(path)))
26
- continue;
27
- try {
28
- const result = await execFileAsync("lsof", ["-t", "--", path], { encoding: "utf8" });
29
- if (result.stdout.trim()) {
30
- throw new Error(`Codex SQLite file is still open: ${path}`);
31
- }
32
- }
33
- catch (error) {
34
- const code = error.code;
35
- if (code === 1)
36
- continue;
37
- if (code === "ENOENT")
38
- throw new Error("lsof is required to verify Codex home quiescence");
39
- throw error;
40
- }
41
- }
42
- }
43
- async function symlinkTarget(path) {
44
- try {
45
- const info = await lstat(path);
46
- if (!info.isSymbolicLink())
47
- return null;
48
- return await readlink(path);
49
- }
50
- catch (error) {
51
- if (error.code === "ENOENT")
52
- return null;
53
- throw error;
54
- }
55
- }
56
- export async function migrateCodexHome(input, dependencies = {}) {
57
- const agentHome = resolve(input.agentHome);
58
- const sourceHome = resolve(input.sourceHome);
59
- const backupRoot = resolve(input.backupRoot);
60
- const codexHome = defaultCodexHome(agentHome);
61
- const sessions = join(codexHome, "sessions");
62
- const index = join(codexHome, "session_index.jsonl");
63
- const expectedSessions = resolve(sourceHome, "sessions");
64
- const expectedIndex = resolve(sourceHome, "session_index.jsonl");
65
- const marker = join(codexHome, MARKER_NAME);
66
- const sessionLink = await symlinkTarget(sessions);
67
- const indexLink = await symlinkTarget(index);
68
- if (sessionLink === null && indexLink === null && await exists(marker)) {
69
- return { status: "already-migrated" };
70
- }
71
- if (sessionLink === null || indexLink === null) {
72
- throw new Error("Refusing migration: Codex home is not in the complete legacy symlink layout");
73
- }
74
- if (resolve(codexHome, sessionLink) !== expectedSessions || resolve(codexHome, indexLink) !== expectedIndex) {
75
- throw new Error("Refusing migration: legacy symlinks do not point to the configured source home");
76
- }
77
- await (dependencies.ensureQuiesced ?? assertNoOpenSqliteFiles)(codexHome);
78
- const stamp = (dependencies.now ?? (() => new Date()))().toISOString().replaceAll(/[^0-9]/g, "").slice(0, 14);
79
- const backupDir = join(backupRoot, `${stamp}-${dependencies.id?.() ?? randomUUID()}`);
80
- await mkdir(backupDir, { recursive: true, mode: 0o700 });
81
- await symlink(sessionLink, join(backupDir, "sessions"), "dir");
82
- await symlink(indexLink, join(backupDir, "session_index.jsonl"), "file");
83
- const moved = [];
84
- try {
85
- for (const name of SQLITE_FILES) {
86
- const current = join(codexHome, name);
87
- if (await exists(current)) {
88
- await rename(current, join(backupDir, name));
89
- moved.push(name);
90
- }
91
- }
92
- await materializeDefaultCodexHome(agentHome, sourceHome);
93
- await atomicPrivateWrite(marker, JSON.stringify({
94
- version: 1,
95
- migrated_at: new Date().toISOString(),
96
- source_home: sourceHome,
97
- backup_dir: backupDir,
98
- }) + "\n");
99
- return { status: "migrated", backupDir };
100
- }
101
- catch (error) {
102
- for (const name of moved.reverse()) {
103
- await rename(join(backupDir, name), join(codexHome, name)).catch(() => undefined);
104
- }
105
- await rm(sessions, { recursive: true, force: true });
106
- await unlink(index).catch(() => undefined);
107
- await symlink(sessionLink, sessions, "dir").catch(() => undefined);
108
- await symlink(indexLink, index, "file").catch(() => undefined);
109
- throw error;
110
- }
111
- }
112
- export const codexHomeMigrationMarkerName = MARKER_NAME;
@@ -1,236 +0,0 @@
1
- import { chmod, copyFile, lstat, mkdir, open, readdir, readlink, rename, rm, symlink, unlink, link, writeFile, } from "node:fs/promises";
2
- import { constants } from "node:fs";
3
- import { dirname, join, relative, resolve, sep } from "node:path";
4
- const PRIVATE_CODEX_HOME = ".codex";
5
- export function defaultCodexHome(homeDir) {
6
- return resolve(homeDir, PRIVATE_CODEX_HOME);
7
- }
8
- async function linkIfMissing(source, target) {
9
- try {
10
- await lstat(target);
11
- return;
12
- }
13
- catch (error) {
14
- if (error.code !== "ENOENT")
15
- throw error;
16
- }
17
- try {
18
- await lstat(source);
19
- }
20
- catch (error) {
21
- if (error.code === "ENOENT")
22
- return;
23
- throw error;
24
- }
25
- try {
26
- await symlink(source, target, "file");
27
- }
28
- catch (error) {
29
- if (error.code !== "EEXIST")
30
- throw error;
31
- }
32
- }
33
- function isBelow(root, candidate) {
34
- const child = relative(root, candidate);
35
- return child !== "" && child !== ".." && !child.startsWith(`..${sep}`) && !child.startsWith(sep);
36
- }
37
- async function isExactSymlink(target, expectedSource) {
38
- const linkTarget = await readlink(target);
39
- return resolve(dirname(target), linkTarget) === resolve(expectedSource);
40
- }
41
- async function ensurePrivateSessions(target, legacySource) {
42
- try {
43
- const current = await lstat(target);
44
- if (current.isSymbolicLink()) {
45
- if (legacySource === undefined || !(await isExactSymlink(target, legacySource))) {
46
- throw new Error(`Refusing to replace unexpected Codex sessions symlink: ${target}`);
47
- }
48
- const replacement = `${target}.private-${process.pid}-${Math.random().toString(36).slice(2)}`;
49
- const legacyBackup = `${target}.legacy-${process.pid}-${Math.random().toString(36).slice(2)}`;
50
- try {
51
- await mkdir(replacement, { mode: 0o700 });
52
- await rename(target, legacyBackup);
53
- try {
54
- await rename(replacement, target);
55
- }
56
- catch (error) {
57
- await rename(legacyBackup, target).catch(() => undefined);
58
- throw error;
59
- }
60
- }
61
- finally {
62
- await rm(replacement, { recursive: true, force: true }).catch(() => undefined);
63
- await rm(legacyBackup, { recursive: true, force: true }).catch(() => undefined);
64
- }
65
- }
66
- else if (!current.isDirectory()) {
67
- throw new Error(`Codex sessions target is not a directory: ${target}`);
68
- }
69
- }
70
- catch (error) {
71
- if (error.code !== "ENOENT")
72
- throw error;
73
- await mkdir(target, { mode: 0o700 });
74
- }
75
- await chmod(target, 0o700);
76
- }
77
- async function atomicCopyFile(source, target, mode) {
78
- const parent = dirname(target);
79
- await mkdir(parent, { recursive: true, mode: 0o700 });
80
- const temporary = `${target}.tmp-${process.pid}-${Math.random().toString(36).slice(2)}`;
81
- try {
82
- await copyFile(source, temporary, constants.COPYFILE_EXCL);
83
- await chmod(temporary, mode);
84
- try {
85
- await link(temporary, target);
86
- return true;
87
- }
88
- catch (error) {
89
- if (error.code !== "EEXIST")
90
- throw error;
91
- return false;
92
- }
93
- }
94
- finally {
95
- await unlink(temporary).catch(() => undefined);
96
- }
97
- }
98
- async function ensurePrivateIndex(target, source) {
99
- let current;
100
- try {
101
- current = await lstat(target);
102
- }
103
- catch (error) {
104
- if (error.code !== "ENOENT")
105
- throw error;
106
- }
107
- if (current?.isSymbolicLink()) {
108
- if (source === undefined || !(await isExactSymlink(target, source))) {
109
- throw new Error(`Refusing to replace unexpected Codex session index symlink: ${target}`);
110
- }
111
- const replacement = `${target}.private-${process.pid}-${Math.random().toString(36).slice(2)}`;
112
- const legacyBackup = `${target}.legacy-${process.pid}-${Math.random().toString(36).slice(2)}`;
113
- try {
114
- try {
115
- await lstat(source);
116
- await copyFile(source, replacement, constants.COPYFILE_EXCL);
117
- }
118
- catch (error) {
119
- if (error.code !== "ENOENT")
120
- throw error;
121
- await writeFile(replacement, "", { encoding: "utf8", flag: "wx" });
122
- }
123
- await chmod(replacement, 0o600);
124
- await rename(target, legacyBackup);
125
- try {
126
- await rename(replacement, target);
127
- }
128
- catch (error) {
129
- await rename(legacyBackup, target).catch(() => undefined);
130
- throw error;
131
- }
132
- }
133
- finally {
134
- await rm(replacement, { force: true }).catch(() => undefined);
135
- await rm(legacyBackup, { force: true }).catch(() => undefined);
136
- }
137
- current = undefined;
138
- }
139
- if (current !== undefined && !current.isFile()) {
140
- throw new Error(`Codex session index target is not a regular file: ${target}`);
141
- }
142
- if (current !== undefined) {
143
- await chmod(target, 0o600);
144
- return;
145
- }
146
- if (source !== undefined) {
147
- try {
148
- await lstat(source);
149
- if (await atomicCopyFile(source, target, 0o600))
150
- return;
151
- return;
152
- }
153
- catch (error) {
154
- if (error.code !== "ENOENT")
155
- throw error;
156
- }
157
- }
158
- const handle = await open(target, "wx", 0o600).catch((error) => {
159
- if (error.code === "EEXIST")
160
- return null;
161
- throw error;
162
- });
163
- await handle?.close();
164
- }
165
- async function findRollout(root, sessionId) {
166
- const entries = await readdir(root, { withFileTypes: true }).catch((error) => {
167
- if (error.code === "ENOENT")
168
- return [];
169
- throw error;
170
- });
171
- const matches = [];
172
- for (const entry of entries) {
173
- const path = join(root, entry.name);
174
- if (entry.isDirectory()) {
175
- const nested = await findRollout(path, sessionId);
176
- if (nested !== null)
177
- matches.push(nested);
178
- }
179
- else if (entry.isFile() && entry.name.includes(sessionId)) {
180
- matches.push(path);
181
- }
182
- }
183
- if (matches.length > 1)
184
- throw new Error(`Multiple Codex rollout files match session ${sessionId}`);
185
- return matches[0] ?? null;
186
- }
187
- async function hydrateResumeRollout(sourceSessions, privateSessions, sessionId) {
188
- if (sessionId.includes("/") || sessionId.includes("\\") || sessionId.includes("..")) {
189
- throw new Error("Invalid Codex resume session id");
190
- }
191
- const sourcePath = await findRollout(sourceSessions, sessionId);
192
- if (sourcePath === null)
193
- return;
194
- const relativePath = relative(sourceSessions, sourcePath);
195
- const destination = resolve(privateSessions, relativePath);
196
- if (!isBelow(resolve(privateSessions), destination)) {
197
- throw new Error("Codex resume rollout escaped the private sessions directory");
198
- }
199
- let existing;
200
- try {
201
- existing = await lstat(destination);
202
- }
203
- catch (error) {
204
- if (error.code !== "ENOENT")
205
- throw error;
206
- }
207
- if (existing !== undefined) {
208
- if (!existing.isFile() || existing.isSymbolicLink()) {
209
- throw new Error(`Refusing to replace existing Codex rollout: ${destination}`);
210
- }
211
- return;
212
- }
213
- const mode = (await lstat(sourcePath)).mode & 0o777;
214
- await atomicCopyFile(sourcePath, destination, mode);
215
- }
216
- /** Isolate Codex high-volume state per Agent while preserving auth and exact resume. */
217
- export async function materializeDefaultCodexHome(agentHome, sourceHome, resumeSessionId) {
218
- const codexHome = defaultCodexHome(agentHome);
219
- await mkdir(codexHome, { recursive: true, mode: 0o700 });
220
- await chmod(codexHome, 0o700);
221
- const source = sourceHome === undefined ? null : resolve(sourceHome);
222
- if (source !== null && source !== codexHome) {
223
- await Promise.all([
224
- linkIfMissing(join(source, "auth.json"), join(codexHome, "auth.json")),
225
- linkIfMissing(join(source, "config.toml"), join(codexHome, "config.toml")),
226
- ]);
227
- }
228
- const sourceSessions = source === null || source === codexHome ? undefined : join(source, "sessions");
229
- await ensurePrivateSessions(join(codexHome, "sessions"), sourceSessions);
230
- const sourceIndex = source === null || source === codexHome ? undefined : join(source, "session_index.jsonl");
231
- await ensurePrivateIndex(join(codexHome, "session_index.jsonl"), sourceIndex);
232
- if (sourceSessions !== undefined && resumeSessionId) {
233
- await hydrateResumeRollout(sourceSessions, join(codexHome, "sessions"), resumeSessionId);
234
- }
235
- return codexHome;
236
- }
@@ -1,212 +0,0 @@
1
- import { z } from "zod";
2
- export const STORAGE_MIGRATION_CAPABILITY = "storage_migration_v1";
3
- export const STORAGE_MIGRATION_STATUS_CAPABILITY = "storage_migration_status_v1";
4
- const IdentitySchema = z.object({
5
- migrationId: z.string().uuid(),
6
- environment: z.string().regex(/^[a-z0-9][a-z0-9_-]{0,31}$/u),
7
- workspaceId: z.string().uuid(),
8
- ownerEpoch: z.number().int().positive().safe().optional(),
9
- ownershipClaimFingerprint: z.string().regex(/^sha256:[0-9a-f]{64}$/u).optional(),
10
- });
11
- const DoctorRequestSchema = IdentitySchema.extend({ type: z.literal("daemon:storage-doctor") }).strict();
12
- const MigrateRequestSchema = IdentitySchema.extend({
13
- type: z.literal("daemon:storage-migrate"),
14
- manifestDigest: z.string().regex(/^[0-9a-f]{64}$/u),
15
- }).strict();
16
- const StatusSchema = z.enum([
17
- "preflight_running", "preflight_passed", "backing_up", "initial_sync", "quiescing", "final_sync", "verified",
18
- "service_switched", "hello_confirmed", "completed", "failed", "rollback_pending", "rolled_back",
19
- "rollback_failed",
20
- ]);
21
- const TOKEN = /(?:\bsk_(?:machine|user|agent)_[A-Za-z0-9_-]{4,}\b|\bbearer\s+[^\s]+|(?:password|secret|token)\s*[:=]\s*[^\s]+)/giu;
22
- const safeError = (error) => (error instanceof Error ? error.message : String(error))
23
- .replaceAll(TOKEN, "<redacted>").slice(0, 500) || "storage migration failed";
24
- const stableErrorCode = (candidate, fallback) => typeof candidate === "string" && /^[a-z0-9._-]{1,128}$/u.test(candidate) ? candidate : fallback;
25
- export function createStorageControlController(input) {
26
- const environment = z.string().regex(/^[a-z0-9][a-z0-9_-]{0,31}$/u).parse(input.environment);
27
- const workspaceId = z.string().uuid().parse(input.workspaceId);
28
- const profile = z.string().regex(/^[a-z0-9][a-z0-9_-]{0,47}$/u).parse(input.profile);
29
- const sourceStorageFingerprint = z.string().regex(/^sha256:[0-9a-f]{64}$/u).parse(input.sourceStorageFingerprint);
30
- const targetStorageFingerprint = z.string().regex(/^sha256:[0-9a-f]{64}$/u).parse(input.targetStorageFingerprint);
31
- const ownership = input.ownership === undefined ? undefined : Object.freeze({
32
- ownerEpoch: z.number().int().positive().safe().parse(input.ownership.ownerEpoch),
33
- ownershipClaimFingerprint: z.string().regex(/^sha256:[0-9a-f]{64}$/u)
34
- .parse(input.ownership.ownershipClaimFingerprint),
35
- });
36
- const preflights = new Map();
37
- const preflightRuns = new Map();
38
- const appliedMigrationIds = new Set();
39
- let runningMigrationId = null;
40
- const preflightTimeoutMs = z.number().int().positive().max(24 * 60 * 60 * 1000)
41
- .parse(input.preflightTimeoutMs ?? 30 * 60 * 1000);
42
- const frame = (identity, status, extra = {}) => ({
43
- type: "daemon:storage-migration-status",
44
- migrationId: identity.migrationId,
45
- status,
46
- environment,
47
- workspaceId,
48
- profile,
49
- ...(ownership === undefined ? {} : ownership),
50
- ...extra,
51
- });
52
- const failure = (identity, errorCode, error, reopenAdmission = false, extra = {}) => {
53
- reportStatus(frame(identity, "failed", {
54
- errorCode,
55
- ...extra,
56
- ...(error === undefined ? {} : { errorMessage: safeError(error) }),
57
- ...(reopenAdmission || error?.reopenAdmission === true
58
- ? { reopenAdmission: true }
59
- : {}),
60
- }));
61
- };
62
- const reportStatus = (statusFrame) => {
63
- try {
64
- input.sendStatus(statusFrame);
65
- }
66
- catch (error) {
67
- input.onStatusError?.(error, statusFrame);
68
- }
69
- };
70
- const matchesIdentity = (identity) => identity.environment === environment && identity.workspaceId === workspaceId
71
- && (ownership === undefined
72
- || (identity.ownerEpoch === ownership.ownerEpoch
73
- && identity.ownershipClaimFingerprint === ownership.ownershipClaimFingerprint));
74
- const preflightKey = (identity) => `${identity.migrationId}\0${identity.environment}\0${identity.workspaceId}`;
75
- const runPreflight = (identity, failureFallback = "doctor_failed") => {
76
- const key = preflightKey(identity);
77
- const existing = preflightRuns.get(key);
78
- if (existing !== undefined)
79
- return existing;
80
- const controller = new AbortController();
81
- let scannedEntries = 0;
82
- let timedOut = false;
83
- reportStatus(frame(identity, "preflight_running", {
84
- progress: { state: "running", scannedEntries: 0 },
85
- }));
86
- const doctorPromise = Promise.resolve().then(() => input.runDoctor(identity.migrationId, {
87
- signal: controller.signal,
88
- onProgress: (progress) => {
89
- if (timedOut)
90
- return;
91
- scannedEntries = progress.scannedEntries;
92
- reportStatus(frame(identity, "preflight_running", {
93
- progress: { state: "running", scannedEntries },
94
- }));
95
- },
96
- }));
97
- let timeoutHandle;
98
- const timeout = new Promise((_, reject) => {
99
- timeoutHandle = setTimeout(() => {
100
- timedOut = true;
101
- controller.abort();
102
- reject(Object.assign(new Error("storage doctor preflight timed out"), {
103
- code: "preflight_stale",
104
- scannedEntries,
105
- }));
106
- }, preflightTimeoutMs);
107
- timeoutHandle.unref?.();
108
- });
109
- void doctorPromise.catch(() => undefined);
110
- let outcome;
111
- outcome = Promise.race([doctorPromise, timeout]).then((doctor) => {
112
- if (doctor.manifest.migrationId !== identity.migrationId
113
- || doctor.environment !== environment || doctor.workspaceId !== workspaceId
114
- || doctor.profile !== profile) {
115
- failure(identity, "storage_identity_mismatch", undefined, true);
116
- return null;
117
- }
118
- if (!doctor.applyAllowed) {
119
- failure(identity, doctor.blockers[0]?.code ?? "doctor_blocked", undefined, true);
120
- return null;
121
- }
122
- preflights.set(identity.migrationId, doctor);
123
- reportStatus(frame(identity, "preflight_passed", {
124
- sourceStorageFingerprint,
125
- targetStorageFingerprint,
126
- manifestDigest: doctor.manifestDigest,
127
- }));
128
- return doctor;
129
- }).catch((error) => {
130
- const code = error.code;
131
- if (code === "preflight_stale") {
132
- failure(identity, "preflight_stale", undefined, true, {
133
- retryable: true,
134
- progress: { state: "stale", scannedEntries: Number(error.scannedEntries) || scannedEntries },
135
- });
136
- }
137
- else if (code === "preflight_aborted") {
138
- failure(identity, "preflight_aborted", error, true, {
139
- retryable: true,
140
- progress: { state: "stale", scannedEntries },
141
- });
142
- }
143
- else {
144
- failure(identity, stableErrorCode(code, failureFallback), error, true);
145
- }
146
- return null;
147
- }).finally(() => {
148
- if (timeoutHandle !== undefined)
149
- clearTimeout(timeoutHandle);
150
- if (preflightRuns.get(key) === outcome)
151
- preflightRuns.delete(key);
152
- });
153
- preflightRuns.set(key, outcome);
154
- return outcome;
155
- };
156
- const handle = async (raw) => {
157
- const type = typeof raw === "object" && raw !== null && "type" in raw ? raw.type : null;
158
- if (type !== "daemon:storage-doctor" && type !== "daemon:storage-migrate")
159
- return false;
160
- const schema = type === "daemon:storage-doctor" ? DoctorRequestSchema : MigrateRequestSchema;
161
- const parsed = schema.safeParse(raw);
162
- if (!parsed.success)
163
- return true;
164
- const identity = parsed.data;
165
- if (!matchesIdentity(identity)) {
166
- failure(identity, "storage_identity_mismatch", undefined, true);
167
- return true;
168
- }
169
- if (type === "daemon:storage-doctor") {
170
- await runPreflight(identity);
171
- return true;
172
- }
173
- if (runningMigrationId !== null) {
174
- if (runningMigrationId !== identity.migrationId)
175
- failure(identity, "migration_active");
176
- return true;
177
- }
178
- const migrate = MigrateRequestSchema.parse(identity);
179
- if (appliedMigrationIds.has(identity.migrationId))
180
- return true;
181
- let doctor = preflights.get(identity.migrationId);
182
- if (doctor === undefined) {
183
- const rebuilt = await runPreflight(identity, "preflight_rebuild_failed");
184
- if (rebuilt === null)
185
- return true;
186
- doctor = rebuilt;
187
- }
188
- if (!doctor.applyAllowed || doctor.manifestDigest !== migrate.manifestDigest
189
- || doctor.manifest.migrationId !== identity.migrationId
190
- || doctor.environment !== environment || doctor.workspaceId !== workspaceId || doctor.profile !== profile) {
191
- failure(identity, "preflight_digest_mismatch", undefined, true);
192
- return true;
193
- }
194
- preflights.delete(identity.migrationId);
195
- runningMigrationId = identity.migrationId;
196
- try {
197
- await input.runMigration(doctor, (status) => {
198
- if (StatusSchema.safeParse(status).success)
199
- reportStatus(frame(identity, status));
200
- });
201
- appliedMigrationIds.add(identity.migrationId);
202
- }
203
- catch (error) {
204
- failure(identity, stableErrorCode(error.code, "storage_migration_failed"), error);
205
- }
206
- finally {
207
- runningMigrationId = null;
208
- }
209
- return true;
210
- };
211
- return Object.freeze({ handle });
212
- }