@botlearn-course/daemon 0.0.19 → 0.0.20-beta.2

package/dist/workspace-restore.js

@@ -0,0 +1,347 @@
+import { createHash, randomUUID } from "node:crypto";
+import { chmodSync, closeSync, constants, existsSync, fstatSync, fsyncSync, lstatSync, mkdirSync, openSync, readFileSync, readSync, renameSync, rmSync, statfsSync, unlinkSync, writeFileSync, } from "node:fs";
+import path from "node:path";
+import { validateWorkspaceEntrySet, } from "./workspace-entry-set.js";
+import { WORKSPACE_MATERIALIZATION_SCHEMA, writeWorkspaceMaterializationMarker, } from "./workspace-materialization.js";
+const UUID = /^[0-9a-f]{8}-[0-9a-f]{4}-[1-8][0-9a-f]{3}-[89ab][0-9a-f]{3}-[0-9a-f]{12}$/i;
+const SHA256 = /^[0-9a-f]{64}$/;
+const RESTORE_JOURNAL_SCHEMA = "agent-workspace-restore-journal/1";
+const COPY_CHUNK_BYTES = 64 * 1024;
+export class WorkspaceRestoreError extends Error {
+    code;
+    retryable;
+    constructor(code, retryable = false) {
+        super(code);
+        this.code = code;
+        this.retryable = retryable;
+        this.name = "WorkspaceRestoreError";
+    }
+}
+function fail(code) {
+    throw new WorkspaceRestoreError(code, code === "workspace_restore_unavailable" ||
+        code === "workspace_restore_disk_pressure" ||
+        code === "workspace_restore_inode_pressure");
+}
+function assertUuid(value, label) {
+    if (!UUID.test(value))
+        fail(`workspace_restore_${label}_invalid`);
+}
+function hasExactKeys(value, expected) {
+    const actual = Object.keys(value).sort();
+    const sortedExpected = [...expected].sort();
+    return actual.length === sortedExpected.length &&
+        actual.every((key, index) => key === sortedExpected[index]);
+}
+function containsPath(parent, candidate) {
+    const relative = path.relative(parent, candidate);
+    return relative === "" || (!relative.startsWith(".." + path.sep) && relative !== "..");
+}
+function validJournalSwapName(value, prefix) {
+    return typeof value === "string" && value.startsWith(prefix) &&
+        UUID.test(value.slice(prefix.length));
+}
+function assertControlPath(workspaceDirectory, controlDirectory) {
+    if (containsPath(path.resolve(workspaceDirectory), path.resolve(controlDirectory))) {
+        fail("workspace_restore_control_path_invalid");
+    }
+}
+function assertDescriptor(descriptor) {
+    assertUuid(descriptor.sandboxId, "sandbox_id");
+    assertUuid(descriptor.runtimeSessionId, "session_id");
+    if (!Number.isSafeInteger(descriptor.sandboxGeneration) ||
+        descriptor.sandboxGeneration < 1 ||
+        !Number.isSafeInteger(descriptor.revision) ||
+        descriptor.revision < 0 ||
+        !Number.isSafeInteger(descriptor.expectedFileCount) ||
+        descriptor.expectedFileCount < 0 ||
+        !Number.isSafeInteger(descriptor.expectedEntryCount) ||
+        descriptor.expectedEntryCount < 0 ||
+        !Number.isSafeInteger(descriptor.expectedTotalBytes) ||
+        descriptor.expectedTotalBytes < 0 ||
+        (descriptor.continuityState !== "healthy" && descriptor.continuityState !== "degraded") ||
+        (descriptor.continuityState === "degraded" && !descriptor.continuityErrorCode) ||
+        (descriptor.continuityState === "healthy" && descriptor.continuityErrorCode !== null))
+        fail("workspace_restore_descriptor_invalid");
+    const revisionZero = descriptor.revision === 0;
+    if (revisionZero !== (descriptor.snapshotId === null) ||
+        revisionZero !== (descriptor.contentSha256 === null) ||
+        (!revisionZero &&
+            (!UUID.test(descriptor.snapshotId) || !SHA256.test(descriptor.contentSha256))) ||
+        (revisionZero &&
+            (descriptor.expectedFileCount !== 0 || descriptor.expectedEntryCount !== 0 ||
+                descriptor.expectedTotalBytes !== 0)))
+        fail("workspace_restore_descriptor_invalid");
+}
+export function workspaceRestoreJournalPath(controlDirectory, runtimeSessionId) {
+    assertUuid(runtimeSessionId, "session_id");
+    return path.join(controlDirectory, "workspace-restore-journals", `${runtimeSessionId}.json`);
+}
+function writeJournal(file, journal) {
+    const directory = path.dirname(file);
+    mkdirSync(directory, { recursive: true, mode: 0o700 });
+    chmodSync(directory, 0o700);
+    const staging = `${file}.tmp-${process.pid}-${randomUUID()}`;
+    try {
+        writeFileSync(staging, JSON.stringify(journal), { encoding: "utf8", mode: 0o600 });
+        const handle = openSync(staging, constants.O_RDONLY | constants.O_NOFOLLOW);
+        try {
+            fsyncSync(handle);
+        }
+        finally {
+            closeSync(handle);
+        }
+        renameSync(staging, file);
+        fsyncDirectory(directory);
+    }
+    catch (error) {
+        try {
+            unlinkSync(staging);
+        }
+        catch {
+            // The atomic staging file may not have been created.
+        }
+        throw error;
+    }
+}
+function parseJournal(raw, descriptor) {
+    if (!raw || typeof raw !== "object" || Array.isArray(raw)) {
+        fail("workspace_restore_journal_invalid");
+    }
+    const value = raw;
+    if (!hasExactKeys(value, [
+        "schemaVersion",
+        "sandboxId",
+        "sandboxGeneration",
+        "runtimeSessionId",
+        "stagingName",
+        "backupName",
+        "hadWorkspace",
+        "phase",
+        "marker",
+    ]))
+        fail("workspace_restore_journal_invalid");
+    const namePrefix = `.botlearn-restore-${descriptor.runtimeSessionId}-`;
+    const backupPrefix = `.botlearn-backup-${descriptor.runtimeSessionId}-`;
+    if (value.schemaVersion !== RESTORE_JOURNAL_SCHEMA ||
+        value.sandboxId !== descriptor.sandboxId ||
+        value.sandboxGeneration !== descriptor.sandboxGeneration ||
+        value.runtimeSessionId !== descriptor.runtimeSessionId ||
+        !validJournalSwapName(value.stagingName, namePrefix) ||
+        !validJournalSwapName(value.backupName, backupPrefix) ||
+        typeof value.hadWorkspace !== "boolean" ||
+        !["prepared", "old_moved", "new_moved"].includes(String(value.phase)))
+        fail("workspace_restore_journal_invalid");
+    const marker = value.marker;
+    if (!marker || marker.schemaVersion !== WORKSPACE_MATERIALIZATION_SCHEMA ||
+        marker.sandboxId !== descriptor.sandboxId ||
+        marker.sandboxGeneration !== descriptor.sandboxGeneration ||
+        marker.runtimeSessionId !== descriptor.runtimeSessionId ||
+        marker.workspaceContinuityState !== descriptor.continuityState ||
+        marker.snapshotId !== descriptor.snapshotId || marker.revision !== descriptor.revision ||
+        marker.contentSha256 !== descriptor.contentSha256)
+        fail("workspace_restore_journal_invalid");
+    return value;
+}
+function fsyncDirectory(directory) {
+    const handle = openSync(directory, constants.O_RDONLY | constants.O_DIRECTORY | constants.O_NOFOLLOW);
+    try {
+        fsyncSync(handle);
+    }
+    finally {
+        closeSync(handle);
+    }
+}
+function verifyDownloadedFile(file, expected) {
+    const handle = openSync(file, constants.O_RDONLY | constants.O_NOFOLLOW);
+    try {
+        const stats = fstatSync(handle, { bigint: true });
+        if (!stats.isFile() || stats.nlink !== 1n || stats.size !== BigInt(expected.expectedSize)) {
+            fail("workspace_restore_integrity_failed");
+        }
+        const hash = createHash("sha256");
+        const buffer = Buffer.allocUnsafe(COPY_CHUNK_BYTES);
+        let total = 0;
+        while (true) {
+            const bytesRead = readSync(handle, buffer, 0, buffer.length, null);
+            if (bytesRead === 0)
+                break;
+            total += bytesRead;
+            hash.update(buffer.subarray(0, bytesRead));
+        }
+        if (total !== expected.expectedSize || hash.digest("hex") !== expected.expectedSha256) {
+            fail("workspace_restore_integrity_failed");
+        }
+        fsyncSync(handle);
+    }
+    finally {
+        closeSync(handle);
+    }
+}
+function markerFromDescriptor(descriptor) {
+    return {
+        schemaVersion: WORKSPACE_MATERIALIZATION_SCHEMA,
+        sandboxId: descriptor.sandboxId,
+        sandboxGeneration: descriptor.sandboxGeneration,
+        runtimeSessionId: descriptor.runtimeSessionId,
+        workspaceContinuityState: descriptor.continuityState,
+        snapshotId: descriptor.snapshotId,
+        revision: descriptor.revision,
+        contentSha256: descriptor.contentSha256,
+        writtenAt: new Date().toISOString(),
+    };
+}
+export function recoverWorkspaceRestoreSwap(options) {
+    assertDescriptor(options.descriptor);
+    assertControlPath(options.workspaceDirectory, options.controlDirectory);
+    const journalFile = workspaceRestoreJournalPath(options.controlDirectory, options.descriptor.runtimeSessionId);
+    if (!existsSync(journalFile))
+        return false;
+    const journal = parseJournal(JSON.parse(readFileSync(journalFile, "utf8")), options.descriptor);
+    const parent = path.dirname(options.workspaceDirectory);
+    const staging = path.join(parent, journal.stagingName);
+    const backup = path.join(parent, journal.backupName);
+    const workspaceExists = existsSync(options.workspaceDirectory);
+    const stagingExists = existsSync(staging);
+    const backupExists = existsSync(backup);
+    if (journal.phase === "new_moved" ||
+        (journal.phase === "old_moved" && workspaceExists && !stagingExists)) {
+        if (!workspaceExists)
+            fail("workspace_restore_journal_invalid");
+        writeWorkspaceMaterializationMarker(journal.marker);
+        if (backupExists)
+            rmSync(backup, { recursive: true, force: true });
+        unlinkSync(journalFile);
+        fsyncDirectory(path.dirname(journalFile));
+        fsyncDirectory(parent);
+        return true;
+    }
+    if (!stagingExists)
+        fail("workspace_restore_journal_invalid");
+    if (!workspaceExists && backupExists)
+        renameSync(backup, options.workspaceDirectory);
+    else if (!workspaceExists && journal.hadWorkspace)
+        fail("workspace_restore_journal_invalid");
+    else if (workspaceExists && backupExists)
+        fail("workspace_restore_journal_invalid");
+    rmSync(staging, { recursive: true, force: true });
+    unlinkSync(journalFile);
+    fsyncDirectory(path.dirname(journalFile));
+    fsyncDirectory(parent);
+    return true;
+}
+export async function restoreWorkspaceSnapshot(options) {
+    const descriptor = options.descriptor;
+    assertDescriptor(descriptor);
+    assertControlPath(options.workspaceDirectory, options.controlDirectory);
+    assertWorkspaceRestoreCapacity(options);
+    recoverWorkspaceRestoreSwap({
+        workspaceDirectory: options.workspaceDirectory,
+        controlDirectory: options.controlDirectory,
+        descriptor,
+    });
+    const canonical = validateWorkspaceEntrySet(options.entrySet, options.limits);
+    if (canonical.entries.length !== descriptor.expectedEntryCount ||
+        canonical.fileCount !== descriptor.expectedFileCount ||
+        canonical.totalBytes !== descriptor.expectedTotalBytes ||
+        (descriptor.revision > 0 && canonical.contentSha256 !== descriptor.contentSha256))
+        fail("workspace_restore_integrity_failed");
+    const parent = path.dirname(options.workspaceDirectory);
+    const nonce = randomUUID();
+    const stagingName = `.botlearn-restore-${descriptor.runtimeSessionId}-${nonce}`;
+    const backupName = `.botlearn-backup-${descriptor.runtimeSessionId}-${nonce}`;
+    const staging = path.join(parent, stagingName);
+    const backup = path.join(parent, backupName);
+    mkdirSync(staging, { mode: 0o700 });
+    chmodSync(staging, 0o700);
+    let journalWritten = false;
+    try {
+        for (const entry of canonical.entries) {
+            const destination = path.join(staging, ...entry.path.split("/"));
+            if (entry.type === "directory") {
+                mkdirSync(destination, { recursive: true, mode: 0o700 });
+                chmodSync(destination, 0o700);
+                continue;
+            }
+            mkdirSync(path.dirname(destination), { recursive: true, mode: 0o700 });
+            const part = `${destination}.part`;
+            await options.downloadFile(entry, part);
+            verifyDownloadedFile(part, {
+                expectedSize: entry.sizeBytes,
+                expectedSha256: entry.sha256,
+            });
+            chmodSync(part, entry.mode === "0700" ? 0o700 : 0o600);
+            renameSync(part, destination);
+        }
+        const directories = new Set([staging]);
+        for (const entry of canonical.entries) {
+            let directory = entry.type === "directory"
+                ? path.join(staging, ...entry.path.split("/"))
+                : path.dirname(path.join(staging, ...entry.path.split("/")));
+            while (directory.startsWith(staging)) {
+                directories.add(directory);
+                if (directory === staging)
+                    break;
+                directory = path.dirname(directory);
+            }
+        }
+        for (const directory of [...directories].sort((a, b) => b.length - a.length)) {
+            fsyncDirectory(directory);
+        }
+        const journalFile = workspaceRestoreJournalPath(options.controlDirectory, descriptor.runtimeSessionId);
+        const journal = {
+            schemaVersion: RESTORE_JOURNAL_SCHEMA,
+            sandboxId: descriptor.sandboxId,
+            sandboxGeneration: descriptor.sandboxGeneration,
+            runtimeSessionId: descriptor.runtimeSessionId,
+            stagingName,
+            backupName,
+            hadWorkspace: existsSync(options.workspaceDirectory),
+            phase: "prepared",
+            marker: markerFromDescriptor(descriptor),
+        };
+        writeJournal(journalFile, journal);
+        journalWritten = true;
+        if (journal.hadWorkspace)
+            renameSync(options.workspaceDirectory, backup);
+        journal.phase = "old_moved";
+        writeJournal(journalFile, journal);
+        renameSync(staging, options.workspaceDirectory);
+        fsyncDirectory(parent);
+        journal.phase = "new_moved";
+        writeJournal(journalFile, journal);
+        writeWorkspaceMaterializationMarker(journal.marker);
+        if (journal.hadWorkspace)
+            rmSync(backup, { recursive: true, force: true });
+        unlinkSync(journalFile);
+        fsyncDirectory(path.dirname(journalFile));
+        fsyncDirectory(parent);
+    }
+    catch (error) {
+        if (!journalWritten)
+            rmSync(staging, { recursive: true, force: true });
+        throw error;
+    }
+}
+export function assertWorkspaceRestoreCapacity(options) {
+    const { descriptor } = options;
+    assertDescriptor(descriptor);
+    if (!Number.isSafeInteger(options.requiredFreeBytes) ||
+        options.requiredFreeBytes < descriptor.expectedTotalBytes)
+        fail("workspace_restore_reservation_invalid");
+    const parent = path.dirname(options.workspaceDirectory);
+    mkdirSync(parent, { recursive: true, mode: 0o700 });
+    if (existsSync(options.workspaceDirectory)) {
+        const current = lstatSync(options.workspaceDirectory, { bigint: true });
+        if (!current.isDirectory() || current.isSymbolicLink()) {
+            fail("workspace_restore_existing_workspace_invalid");
+        }
+    }
+    const filesystem = statfsSync(parent, { bigint: true });
+    if (filesystem.bavail * filesystem.bsize < BigInt(options.requiredFreeBytes)) {
+        fail("workspace_restore_disk_pressure");
+    }
+    const entryCopies = options.entryCopies ?? 1;
+    if (filesystem.ffree < BigInt(descriptor.expectedEntryCount * entryCopies + 16)) {
+        fail("workspace_restore_inode_pressure");
+    }
+}

package/dist/workspace-snapshot-control.d.ts

@@ -0,0 +1,29 @@
+import type { WorkspaceQuiescenceProof } from "./runtime-quiescence.js";
+interface WorkspaceCheckpointScope {
+    checkpointId: string;
+    baseRevision: number;
+    sandboxId: string;
+    sandboxGeneration: number;
+    connectionEpoch: number;
+    runtimeSessionId: string;
+    agentRunId?: string;
+    workerAttempt?: number;
+    activationId?: string;
+}
+export declare class WorkspaceSnapshotControlError extends Error {
+    readonly code: string;
+    readonly retryable: boolean;
+    constructor(code: string, retryable: boolean);
+}
+export declare function checkpointWorkspace(options: {
+    wsUrl: string;
+    reconnectToken: string;
+    workspaceDirectory: string;
+    scope: WorkspaceCheckpointScope;
+    quiescenceProof: WorkspaceQuiescenceProof;
+}): Promise<{
+    revision: number;
+    snapshotId: string | null;
+    contentSha256: string;
+}>;
+export {};

package/dist/workspace-snapshot-control.js

@@ -0,0 +1,169 @@
+import { createReadStream, rmSync } from "node:fs";
+import path from "node:path";
+import { ensureDaemonHome } from "./auth-store.js";
+import { writeWorkspaceMaterializationMarker } from "./workspace-materialization.js";
+import { WORKSPACE_SNAPSHOT_POLICY_V1 } from "./workspace-snapshot-policy.js";
+import { stageWorkspaceSnapshot } from "./workspace-snapshot-staging.js";
+export class WorkspaceSnapshotControlError extends Error {
+    code;
+    retryable;
+    constructor(code, retryable) {
+        super(code);
+        this.code = code;
+        this.retryable = retryable;
+        this.name = "WorkspaceSnapshotControlError";
+    }
+}
+function controlBase(wsUrl) {
+    const url = new URL(wsUrl);
+    url.protocol = url.protocol === "wss:" ? "https:" : "http:";
+    url.pathname = "/internal/v1/workspace-checkpoints";
+    url.search = "";
+    url.hash = "";
+    return url.toString().replace(/\/$/, "");
+}
+async function requestJson(url, token, body, signal) {
+    const response = await fetch(url, {
+        method: "POST",
+        headers: {
+            Authorization: `Bearer ${token}`,
+            "Content-Type": "application/json",
+        },
+        body: JSON.stringify(body),
+        signal,
+    });
+    if (!response.ok) {
+        let code = `workspace_snapshot_control_${response.status}`;
+        let explicitRetryable;
+        try {
+            const payload = await response.json();
+            const detail = payload.detail;
+            if (detail && typeof detail === "object" && !Array.isArray(detail)) {
+                const stable = detail.code;
+                if (typeof stable === "string" && /^workspace_[a-z0-9_]+$/u.test(stable))
+                    code = stable;
+                const retryable = detail.retryable;
+                if (typeof retryable === "boolean")
+                    explicitRetryable = retryable;
+            }
+        }
+        catch {
+            // Content-free HTTP fallback remains stable and retry classification stays status-based.
+        }
+        throw new WorkspaceSnapshotControlError(code, explicitRetryable ?? (response.status >= 500 || response.status === 408 || response.status === 429 ||
+            response.status === 409));
+    }
+    return await response.json();
+}
+function fence(scope) {
+    return {
+        sandbox_id: scope.sandboxId,
+        sandbox_generation: scope.sandboxGeneration,
+        connection_epoch: scope.connectionEpoch,
+        runtime_session_id: scope.runtimeSessionId,
+        ...(scope.agentRunId ? { agent_run_id: scope.agentRunId } : {}),
+        ...(scope.workerAttempt ? { worker_attempt: scope.workerAttempt } : {}),
+        ...(scope.activationId ? { activation_id: scope.activationId } : {}),
+    };
+}
+export async function checkpointWorkspace(options) {
+    const { scope } = options;
+    const signal = AbortSignal.timeout(WORKSPACE_SNAPSHOT_POLICY_V1.checkpointAttemptDeadlineMs);
+    const checkpointUrl = `${controlBase(options.wsUrl)}/${scope.checkpointId}`;
+    const controlDirectory = path.join(ensureDaemonHome(), "agent-service-sandboxes", scope.sandboxId, "workspace-control");
+    const staged = stageWorkspaceSnapshot({
+        workspaceDirectory: options.workspaceDirectory,
+        controlDirectory,
+        checkpointId: scope.checkpointId,
+        limits: WORKSPACE_SNAPSHOT_POLICY_V1.limits,
+        maxStabilityRetries: WORKSPACE_SNAPSHOT_POLICY_V1.maxStabilityRetries,
+        quiescenceProof: options.quiescenceProof,
+    });
+    try {
+        const entrySet = {
+            schemaVersion: "agent-workspace-entry-set/1",
+            entries: staged.entrySet.entries,
+        };
+        const proposal = await requestJson(`${checkpointUrl}/proposal`, options.reconnectToken, {
+            fence: fence(scope),
+            policy_id: WORKSPACE_SNAPSHOT_POLICY_V1.policyId,
+            base_revision: scope.baseRevision,
+            entry_count: staged.entrySet.entries.length,
+            file_count: staged.entrySet.fileCount,
+            total_bytes: staged.entrySet.totalBytes,
+            content_sha256: staged.entrySet.contentSha256,
+            entry_set: entrySet,
+        }, signal);
+        if (proposal.unchanged) {
+            if (proposal.revision === 0 && proposal.snapshot_id === null) {
+                writeWorkspaceMaterializationMarker({
+                    schemaVersion: "agent-workspace-materialization/1",
+                    sandboxId: scope.sandboxId,
+                    sandboxGeneration: scope.sandboxGeneration,
+                    runtimeSessionId: scope.runtimeSessionId,
+                    workspaceContinuityState: "healthy",
+                    snapshotId: null,
+                    revision: 0,
+                    contentSha256: null,
+                    writtenAt: new Date().toISOString(),
+                });
+            }
+            return {
+                revision: proposal.revision,
+                snapshotId: proposal.snapshot_id,
+                contentSha256: proposal.content_sha256,
+            };
+        }
+        for (let offset = 0; offset < proposal.missing_blob_ids.length; offset += 25) {
+            const requestedBlobIds = proposal.missing_blob_ids.slice(offset, offset + 25);
+            const grants = await requestJson(`${checkpointUrl}/upload-grants`, options.reconnectToken, { fence: fence(scope), blob_ids: requestedBlobIds }, signal);
+            if (grants.length !== requestedBlobIds.length) {
+                throw new Error("workspace_snapshot_upload_grant_count_mismatch");
+            }
+            for (const item of grants) {
+                const source = staged.files.find((file) => file.sha256 === item.sha256 && file.sizeBytes === item.size_bytes);
+                if (!source)
+                    throw new Error("workspace_snapshot_upload_source_missing");
+                const response = await fetch(item.grant.url, {
+                    method: item.grant.method,
+                    headers: item.grant.headers,
+                    body: createReadStream(source.stagingPath),
+                    duplex: "half",
+                    signal,
+                });
+                if (!response.ok)
+                    throw new Error(`workspace_snapshot_upload_${response.status}`);
+            }
+        }
+        let committed = await requestJson(`${checkpointUrl}/complete`, options.reconnectToken, { fence: fence(scope), policy_id: WORKSPACE_SNAPSHOT_POLICY_V1.policyId }, signal);
+        while (committed.status === "pending") {
+            await new Promise((resolve) => setTimeout(resolve, 500));
+            committed = await requestJson(`${checkpointUrl}/status`, options.reconnectToken, { fence: fence(scope) }, signal);
+        }
+        if (committed.status !== "committed" ||
+            committed.snapshot_id === null ||
+            committed.revision === null ||
+            committed.content_sha256 === null) {
+            throw new WorkspaceSnapshotControlError(committed.error_code || "workspace_snapshot_verification_failed", false);
+        }
+        writeWorkspaceMaterializationMarker({
+            schemaVersion: "agent-workspace-materialization/1",
+            sandboxId: scope.sandboxId,
+            sandboxGeneration: scope.sandboxGeneration,
+            runtimeSessionId: scope.runtimeSessionId,
+            workspaceContinuityState: "healthy",
+            snapshotId: committed.snapshot_id,
+            revision: committed.revision,
+            contentSha256: committed.content_sha256,
+            writtenAt: new Date().toISOString(),
+        });
+        return {
+            revision: committed.revision,
+            snapshotId: committed.snapshot_id,
+            contentSha256: committed.content_sha256,
+        };
+    }
+    finally {
+        rmSync(staged.stagingDirectory, { recursive: true, force: true });
+    }
+}

package/dist/workspace-snapshot-policy.d.ts

@@ -0,0 +1,24 @@
+/**
+ * Protocol limits frozen with Course Service's WorkspaceSnapshotPolicyV1.
+ * Deployment may enable the capability, but must never override these values.
+ */
+export declare const WORKSPACE_SNAPSHOT_POLICY_V1: {
+    readonly policyId: "WorkspaceSnapshotPolicyV1";
+    readonly limits: {
+        maxPathSegmentBytes: number;
+        maxPathBytes: number;
+        maxDepth: number;
+        maxFileBytes: number;
+        maxFileCount: number;
+        maxTotalBytes: number;
+        maxEntrySetBytes: number;
+    };
+    readonly maxStabilityRetries: 2;
+    readonly checkpointAttemptDeadlineMs: 30000;
+    readonly requiredRestoreFreeBytes: number;
+    readonly sessionRetainedHardLimitBytes: number;
+    readonly checkpointReservationBytes: number;
+    readonly restorePageEntries: 25;
+};
+/** Wire representation validated exactly by Agent Service during the v0.3 handshake. */
+export declare function workspaceSnapshotPolicyCapability(): Record<string, unknown>;

package/dist/workspace-snapshot-policy.js

@@ -0,0 +1,45 @@
+import { RUNTIME_WORKSPACE_INODE_QUOTA, RUNTIME_WORKSPACE_QUOTA_BYTES, } from "./workspace-quota.js";
+/**
+ * Protocol limits frozen with Course Service's WorkspaceSnapshotPolicyV1.
+ * Deployment may enable the capability, but must never override these values.
+ */
+export const WORKSPACE_SNAPSHOT_POLICY_V1 = {
+    policyId: "WorkspaceSnapshotPolicyV1",
+    limits: {
+        maxPathSegmentBytes: 255,
+        maxPathBytes: 1024,
+        maxDepth: 32,
+        maxFileBytes: 50 * 1024 * 1024,
+        maxFileCount: 2_000,
+        maxTotalBytes: 512 * 1024 * 1024,
+        maxEntrySetBytes: 1024 * 1024,
+    },
+    maxStabilityRetries: 2,
+    checkpointAttemptDeadlineMs: 30_000,
+    requiredRestoreFreeBytes: 1024 * 1024,
+    sessionRetainedHardLimitBytes: 2 * 1024 * 1024 * 1024,
+    checkpointReservationBytes: 512 * 1024 * 1024,
+    restorePageEntries: 25,
+};
+/** Wire representation validated exactly by Agent Service during the v0.3 handshake. */
+export function workspaceSnapshotPolicyCapability() {
+    return {
+        policy_id: WORKSPACE_SNAPSHOT_POLICY_V1.policyId,
+        max_path_segment_bytes: WORKSPACE_SNAPSHOT_POLICY_V1.limits.maxPathSegmentBytes,
+        max_path_bytes: WORKSPACE_SNAPSHOT_POLICY_V1.limits.maxPathBytes,
+        max_depth: WORKSPACE_SNAPSHOT_POLICY_V1.limits.maxDepth,
+        max_file_bytes: WORKSPACE_SNAPSHOT_POLICY_V1.limits.maxFileBytes,
+        max_file_count: WORKSPACE_SNAPSHOT_POLICY_V1.limits.maxFileCount,
+        max_total_bytes: WORKSPACE_SNAPSHOT_POLICY_V1.limits.maxTotalBytes,
+        max_entry_set_bytes: WORKSPACE_SNAPSHOT_POLICY_V1.limits.maxEntrySetBytes,
+        required_restore_free_bytes: WORKSPACE_SNAPSHOT_POLICY_V1.requiredRestoreFreeBytes,
+        session_retained_hard_limit_bytes: WORKSPACE_SNAPSHOT_POLICY_V1.sessionRetainedHardLimitBytes,
+        checkpoint_reservation_bytes: WORKSPACE_SNAPSHOT_POLICY_V1.checkpointReservationBytes,
+        restore_page_entries: WORKSPACE_SNAPSHOT_POLICY_V1.restorePageEntries,
+        max_stability_retries: WORKSPACE_SNAPSHOT_POLICY_V1.maxStabilityRetries,
+        checkpoint_attempt_deadline_ms: WORKSPACE_SNAPSHOT_POLICY_V1.checkpointAttemptDeadlineMs,
+        prepare_deadline_seconds: 15 * 60,
+        runtime_filesystem_quota_bytes: RUNTIME_WORKSPACE_QUOTA_BYTES,
+        runtime_inode_quota: RUNTIME_WORKSPACE_INODE_QUOTA,
+    };
+}

package/dist/workspace-snapshot-staging.d.ts

@@ -0,0 +1,27 @@
+import { type WorkspaceQuiescenceProof } from "./runtime-quiescence.js";
+import { type CanonicalWorkspaceEntrySet, type WorkspaceEntrySetLimits } from "./workspace-entry-set.js";
+export declare class WorkspaceSnapshotStagingError extends Error {
+    readonly code: string;
+    readonly retryable: boolean;
+    constructor(code: string, retryable?: boolean);
+}
+export interface StagedWorkspaceFile {
+    path: string;
+    stagingPath: string;
+    sizeBytes: number;
+    sha256: string;
+}
+export interface StagedWorkspaceSnapshot {
+    stagingDirectory: string;
+    entrySet: CanonicalWorkspaceEntrySet;
+    files: StagedWorkspaceFile[];
+}
+export interface WorkspaceSnapshotStagingOptions {
+    workspaceDirectory: string;
+    controlDirectory: string;
+    checkpointId: string;
+    limits: WorkspaceEntrySetLimits;
+    maxStabilityRetries: number;
+    quiescenceProof: WorkspaceQuiescenceProof;
+}
+export declare function stageWorkspaceSnapshot(options: WorkspaceSnapshotStagingOptions): StagedWorkspaceSnapshot;