@superblocksteam/workspace-sync 0.0.0-stage → 2.0.178-next.0

package/dist/snapshot/take-snapshot.js

@@ -0,0 +1,244 @@
+import * as fsp from "node:fs/promises";
+import * as path from "node:path";
+import { hashDirectoryContents, hashFileContents, } from "@superblocksteam/shared";
+import { unreachable } from "@superblocksteam/util";
+import { listLocalDirectory } from "../list-dir.js";
+/**
+ * Takes a snapshot of a local directory. The current implementation stores the snapshot in memory.
+ * This is not scalable for large directories and is subject to change.
+ *
+ * This function now captures the DRAFT state (current filesystem state).
+ * For committed state, use snapshotLocalDirectoryCommittedState.
+ *
+ * @param localDirectoryPath - The path to the local directory to snapshot.
+ * @returns A promise that resolves to a snapshot of the local directory.
+ */
+export async function snapshotLocalDirectory(localDirectoryPath, rootDirectoryPath = localDirectoryPath) {
+    const directoryContents = [];
+    const localDirListing = listLocalDirectory(localDirectoryPath, rootDirectoryPath);
+    for await (const localDirEntry of localDirListing) {
+        let entry;
+        switch (localDirEntry.type) {
+            case "-": {
+                const hash = await hashFileContents(localDirEntry.contents);
+                entry = {
+                    type: "-",
+                    name: localDirEntry.name,
+                    executable: localDirEntry.executable,
+                    hash,
+                    contents: localDirEntry.contents,
+                };
+                break;
+            }
+            case "d": {
+                const subdirectoryContents = await snapshotLocalDirectory(localDirEntry.localPath, rootDirectoryPath);
+                entry = {
+                    type: "d",
+                    name: localDirEntry.name,
+                    hash: subdirectoryContents.hash,
+                    contents: subdirectoryContents.contents,
+                };
+                break;
+            }
+            case "l": {
+                entry = {
+                    type: "l",
+                    target: localDirEntry.target,
+                    name: localDirEntry.name,
+                };
+                break;
+            }
+            default:
+                unreachable(localDirEntry);
+        }
+        directoryContents.push(entry);
+    }
+    const directoryHash = await hashDirectoryContents(directoryContents);
+    return {
+        hash: directoryHash,
+        contents: directoryContents,
+    };
+}
+/**
+ * Takes a snapshot of a local directory showing the COMMITTED state.
+ * This function:
+ * - Uses .sbswap.edit content from the drafts directory instead of current files (for edited files)
+ * - Excludes files that have .sbswap.add markers in the drafts directory (draft additions)
+ * - Excludes .sbswap.edit and .sbswap.add marker files themselves
+ *
+ * @param localDirectoryPath - The path to the local directory to snapshot.
+ * @param appRootPath - The app root path (defaults to localDirectoryPath for top-level calls).
+ * @param editMarkerMap - Map of edit markers (for internal recursive use).
+ * @param addMarkerSet - Set of add markers (for internal recursive use).
+ * @returns A promise that resolves to a snapshot of the committed state.
+ */
+export async function snapshotLocalDirectoryCommittedState(localDirectoryPath, appRootPath, editMarkerMap, addMarkerSet) {
+    const directoryContents = [];
+    // Use the provided appRootPath, or default to localDirectoryPath for top-level calls
+    const rootPath = appRootPath || localDirectoryPath;
+    const draftsDir = path.join(rootPath, ".superblocks", "drafts");
+    // Step 1: Collect all physical entries in the directory
+    const allPhysicalEntries = [];
+    const localDirListing = listLocalDirectory(localDirectoryPath, rootPath);
+    for await (const entry of localDirListing) {
+        allPhysicalEntries.push(entry);
+    }
+    // Step 2: Create maps for marker files from the drafts directory (only for top-level calls)
+    let finalEditMarkerMap;
+    let finalAddMarkerSet;
+    if (!editMarkerMap || !addMarkerSet) {
+        // This is the top-level call, scan the drafts directory
+        finalEditMarkerMap = new Map();
+        finalAddMarkerSet = new Set();
+        try {
+            await fsp.access(draftsDir);
+            await scanDraftsDirectory(draftsDir, rootPath, finalEditMarkerMap, finalAddMarkerSet);
+        }
+        catch {
+            // Drafts directory doesn't exist, no drafts to consider
+        }
+    }
+    else {
+        // Use the provided maps from recursive calls
+        finalEditMarkerMap = editMarkerMap;
+        finalAddMarkerSet = addMarkerSet;
+    }
+    // Step 3: Process entries for committed state
+    for (const localDirEntry of allPhysicalEntries) {
+        const relativePath = path.relative(rootPath, localDirEntry.localPath);
+        // Skip files that are draft additions
+        if (finalAddMarkerSet.has(relativePath)) {
+            continue;
+        }
+        let entry;
+        switch (localDirEntry.type) {
+            case "-": {
+                // Regular file
+                let fileContentsToUse;
+                let executableToUse;
+                if (finalEditMarkerMap.has(relativePath)) {
+                    // Use the original content from .sbswap.edit file in drafts directory
+                    fileContentsToUse = finalEditMarkerMap.get(relativePath);
+                    executableToUse = localDirEntry.executable; // Use current executable flag
+                }
+                else {
+                    // No edit marker, use current file content (this is committed content)
+                    fileContentsToUse = localDirEntry.contents;
+                    executableToUse = localDirEntry.executable;
+                }
+                const hash = await hashFileContents(fileContentsToUse);
+                entry = {
+                    type: "-",
+                    name: localDirEntry.name,
+                    executable: executableToUse,
+                    hash,
+                    contents: fileContentsToUse,
+                };
+                break;
+            }
+            case "d": {
+                // Directory
+                // Recursive call uses the same committed-state snapshotting, passing the app root and marker maps
+                const subdirectorySnapshot = await snapshotLocalDirectoryCommittedState(localDirEntry.localPath, rootPath, // Pass the app root path to recursive calls
+                finalEditMarkerMap, // Pass the edit marker map
+                finalAddMarkerSet);
+                entry = {
+                    type: "d",
+                    name: localDirEntry.name,
+                    hash: subdirectorySnapshot.hash,
+                    contents: subdirectorySnapshot.contents,
+                };
+                break;
+            }
+            case "l": {
+                // Symlink
+                entry = {
+                    type: "l",
+                    target: localDirEntry.target,
+                    name: localDirEntry.name,
+                };
+                break;
+            }
+            default:
+                unreachable(localDirEntry);
+        }
+        directoryContents.push(entry);
+    }
+    const directoryHash = await hashDirectoryContents(directoryContents);
+    return {
+        hash: directoryHash,
+        contents: directoryContents,
+    };
+}
+/**
+ * Recursively scans the drafts directory to build maps of edit markers and add markers.
+ *
+ * @param currentDraftsPath - The current path being scanned in the drafts directory.
+ * @param appRootPath - The app root directory path.
+ * @param editMarkerMap - Map to store edit marker file contents.
+ * @param addMarkerSet - Set to store paths of files that are draft additions.
+ */
+async function scanDraftsDirectory(currentDraftsPath, appRootPath, editMarkerMap, addMarkerSet) {
+    try {
+        const entries = await fsp.readdir(currentDraftsPath, {
+            withFileTypes: true,
+        });
+        const draftsDir = path.join(appRootPath, ".superblocks", "drafts");
+        for (const entry of entries) {
+            const fullPath = path.join(currentDraftsPath, entry.name);
+            if (entry.isFile()) {
+                if (entry.name.endsWith(".sbswap.edit")) {
+                    // This is the original content for an edited file
+                    const markerRelativePath = path.relative(draftsDir, fullPath);
+                    const originalRelativePath = markerRelativePath.slice(0, -".sbswap.edit".length);
+                    const originalContent = await fsp.readFile(fullPath);
+                    editMarkerMap.set(originalRelativePath, originalContent);
+                }
+                else if (entry.name.endsWith(".sbswap.add")) {
+                    // This marks a file as a draft addition - exclude the corresponding file
+                    const markerRelativePath = path.relative(draftsDir, fullPath);
+                    const addedFileRelativePath = markerRelativePath.slice(0, -".sbswap.add".length);
+                    addMarkerSet.add(addedFileRelativePath);
+                }
+            }
+            if (entry.isDirectory()) {
+                await scanDraftsDirectory(fullPath, appRootPath, editMarkerMap, addMarkerSet);
+            }
+        }
+    }
+    catch {
+        // Ignore errors reading directories
+    }
+}
+/**
+ * Takes a snapshot of a local directory, prioritizing .sbswap files for content.
+ * .sbswap files themselves are not included in the snapshot output.
+ *
+ * @param localDirectoryPath - The path to the local directory to snapshot.
+ * @returns A promise that resolves to a snapshot of the local directory, reflecting draft states.
+ */
+export async function snapshotLocalDirectoryWithDrafts(localDirectoryPath) {
+    return snapshotLocalDirectory(localDirectoryPath);
+}
+/**
+ * Flattens a snapshot into relative path -> content hash. A symlink maps to its
+ * target, so retargeting it counts as a change. Directories are left out; their
+ * files carry the change.
+ */
+export function snapshotPathHashes(snapshot) {
+    const hashes = new Map();
+    const walk = (entries, prefix) => {
+        for (const entry of entries) {
+            const entryPath = prefix + entry.name;
+            if (entry.type === "d") {
+                walk(entry.contents, `${entryPath}/`);
+            }
+            else {
+                hashes.set(entryPath, entry.type === "l" ? `symlink:${entry.target}` : entry.hash);
+            }
+        }
+    };
+    walk(snapshot.contents, "");
+    return hashes;
+}
+//# sourceMappingURL=take-snapshot.js.map

package/dist/snapshot/take-snapshot.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"take-snapshot.js","sourceRoot":"","sources":["../../src/snapshot/take-snapshot.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,GAAG,MAAM,kBAAkB,CAAC;AACxC,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAElC,OAAO,EACL,qBAAqB,EACrB,gBAAgB,GACjB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAEpD,OAAO,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AAIpD;;;;;;;;;GASG;AACH,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAC1C,kBAA0B,EAC1B,iBAAiB,GAAG,kBAAkB;IAEtC,MAAM,iBAAiB,GAAuB,EAAE,CAAC;IACjD,MAAM,eAAe,GAAG,kBAAkB,CACxC,kBAAkB,EAClB,iBAAiB,CAClB,CAAC;IACF,IAAI,KAAK,EAAE,MAAM,aAAa,IAAI,eAAe,EAAE,CAAC;QAClD,IAAI,KAAuB,CAAC;QAC5B,QAAQ,aAAa,CAAC,IAAI,EAAE,CAAC;YAC3B,KAAK,GAAG,EAAE,CAAC;gBACT,MAAM,IAAI,GAAG,MAAM,gBAAgB,CACjC,aAAa,CAAC,QAAwB,CACvC,CAAC;gBACF,KAAK,GAAG;oBACN,IAAI,EAAE,GAAG;oBACT,IAAI,EAAE,aAAa,CAAC,IAAI;oBACxB,UAAU,EAAE,aAAa,CAAC,UAAU;oBACpC,IAAI;oBACJ,QAAQ,EAAE,aAAa,CAAC,QAAQ;iBACjC,CAAC;gBACF,MAAM;YACR,CAAC;YACD,KAAK,GAAG,EAAE,CAAC;gBACT,MAAM,oBAAoB,GAAG,MAAM,sBAAsB,CACvD,aAAa,CAAC,SAAS,EACvB,iBAAiB,CAClB,CAAC;gBACF,KAAK,GAAG;oBACN,IAAI,EAAE,GAAG;oBACT,IAAI,EAAE,aAAa,CAAC,IAAI;oBACxB,IAAI,EAAE,oBAAoB,CAAC,IAAI;oBAC/B,QAAQ,EAAE,oBAAoB,CAAC,QAAQ;iBACxC,CAAC;gBACF,MAAM;YACR,CAAC;YACD,KAAK,GAAG,EAAE,CAAC;gBACT,KAAK,GAAG;oBACN,IAAI,EAAE,GAAG;oBACT,MAAM,EAAE,aAAa,CAAC,MAAM;oBAC5B,IAAI,EAAE,aAAa,CAAC,IAAI;iBACzB,CAAC;gBACF,MAAM;YACR,CAAC;YACD;gBACE,WAAW,CAAC,aAAa,CAAC,CAAC;QAC/B,CAAC;QACD,iBAAiB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAChC,CAAC;IACD,MAAM,aAAa,GAAG,MAAM,qBAAqB,CAAC,iBAAiB,CAAC,CAAC;IACrE,OAAO;QACL,IAAI,EAAE,aAAa;QACnB,QAAQ,EAAE,iBAAiB;KAC5B,CAAC;AACJ,CAAC;AAED;;;;;;;;;;;;GAYG;AACH,MAAM,CAAC,KAAK,UAAU,oCAAoC,CACxD,kBAA0B,EAC1B,WAAoB,EACpB,aAAmC,EACnC,YAA0B;IAE1B,MAAM,iBAAiB,GAAuB,EAAE,CAAC;IACjD,qFAAqF;IACrF,MAAM,QAAQ,GAAG,WAAW,IAAI,kBAAkB,CAAC;IACnD,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,cAAc,EAAE,QAAQ,CAAC,CAAC;IAEhE,wDAAwD;IACxD,MAAM,kBAAkB,GAAoB,EAAE,CAAC;IAC/C,MAAM,eAAe,GAAG,kBAAkB,CAAC,kBAAkB,EAAE,QAAQ,CAAC,CAAC;IACzE,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,eAAe,EAAE,CAAC;QAC1C,kBAAkB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACjC,CAAC;IAED,4FAA4F;IAC5F,IAAI,kBAAuC,CAAC;IAC5C,IAAI,iBAA8B,CAAC;IAEnC,IAAI,CAAC,aAAa,IAAI,CAAC,YAAY,EAAE,CAAC;QACpC,wDAAwD;QACxD,kBAAkB,GAAG,IAAI,GAAG,EAAkB,CAAC;QAC/C,iBAAiB,GAAG,IAAI,GAAG,EAAU,CAAC;QAEtC,IAAI,CAAC;YACH,MAAM,GAAG,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;YAC5B,MAAM,mBAAmB,CACvB,SAAS,EACT,QAAQ,EACR,kBAAkB,EAClB,iBAAiB,CAClB,CAAC;QACJ,CAAC;QAAC,MAAM,CAAC;YACP,wDAAwD;QAC1D,CAAC;IACH,CAAC;SAAM,CAAC;QACN,6CAA6C;QAC7C,kBAAkB,GAAG,aAAa,CAAC;QACnC,iBAAiB,GAAG,YAAY,CAAC;IACnC,CAAC;IAED,8CAA8C;IAC9C,KAAK,MAAM,aAAa,IAAI,kBAAkB,EAAE,CAAC;QAC/C,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,aAAa,CAAC,SAAS,CAAC,CAAC;QAEtE,sCAAsC;QACtC,IAAI,iBAAiB,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,CAAC;YACxC,SAAS;QACX,CAAC;QAED,IAAI,KAAuB,CAAC;QAC5B,QAAQ,aAAa,CAAC,IAAI,EAAE,CAAC;YAC3B,KAAK,GAAG,EAAE,CAAC;gBACT,eAAe;gBACf,IAAI,iBAAyB,CAAC;gBAC9B,IAAI,eAAwB,CAAC;gBAE7B,IAAI,kBAAkB,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,CAAC;oBACzC,sEAAsE;oBACtE,iBAAiB,GAAG,kBAAkB,CAAC,GAAG,CAAC,YAAY,CAAE,CAAC;oBAC1D,eAAe,GAAG,aAAa,CAAC,UAAU,CAAC,CAAC,8BAA8B;gBAC5E,CAAC;qBAAM,CAAC;oBACN,uEAAuE;oBACvE,iBAAiB,GAAG,aAAa,CAAC,QAAQ,CAAC;oBAC3C,eAAe,GAAG,aAAa,CAAC,UAAU,CAAC;gBAC7C,CAAC;gBAED,MAAM,IAAI,GAAG,MAAM,gBAAgB,CAAC,iBAAiC,CAAC,CAAC;gBACvE,KAAK,GAAG;oBACN,IAAI,EAAE,GAAG;oBACT,IAAI,EAAE,aAAa,CAAC,IAAI;oBACxB,UAAU,EAAE,eAAe;oBAC3B,IAAI;oBACJ,QAAQ,EAAE,iBAAiB;iBAC5B,CAAC;gBACF,MAAM;YACR,CAAC;YACD,KAAK,GAAG,EAAE,CAAC;gBACT,YAAY;gBACZ,kGAAkG;gBAClG,MAAM,oBAAoB,GAAG,MAAM,oCAAoC,CACrE,aAAa,CAAC,SAAS,EACvB,QAAQ,EAAE,4CAA4C;gBACtD,kBAAkB,EAAE,2BAA2B;gBAC/C,iBAAiB,CAClB,CAAC;gBACF,KAAK,GAAG;oBACN,IAAI,EAAE,GAAG;oBACT,IAAI,EAAE,aAAa,CAAC,IAAI;oBACxB,IAAI,EAAE,oBAAoB,CAAC,IAAI;oBAC/B,QAAQ,EAAE,oBAAoB,CAAC,QAAQ;iBACxC,CAAC;gBACF,MAAM;YACR,CAAC;YACD,KAAK,GAAG,EAAE,CAAC;gBACT,UAAU;gBACV,KAAK,GAAG;oBACN,IAAI,EAAE,GAAG;oBACT,MAAM,EAAE,aAAa,CAAC,MAAM;oBAC5B,IAAI,EAAE,aAAa,CAAC,IAAI;iBACzB,CAAC;gBACF,MAAM;YACR,CAAC;YACD;gBACE,WAAW,CAAC,aAAa,CAAC,CAAC;QAC/B,CAAC;QACD,iBAAiB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAChC,CAAC;IAED,MAAM,aAAa,GAAG,MAAM,qBAAqB,CAAC,iBAAiB,CAAC,CAAC;IACrE,OAAO;QACL,IAAI,EAAE,aAAa;QACnB,QAAQ,EAAE,iBAAiB;KAC5B,CAAC;AACJ,CAAC;AAED;;;;;;;GAOG;AACH,KAAK,UAAU,mBAAmB,CAChC,iBAAyB,EACzB,WAAmB,EACnB,aAAkC,EAClC,YAAyB;IAEzB,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,MAAM,GAAG,CAAC,OAAO,CAAC,iBAAiB,EAAE;YACnD,aAAa,EAAE,IAAI;SACpB,CAAC,CAAC;QACH,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,cAAc,EAAE,QAAQ,CAAC,CAAC;QAEnE,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;YAC5B,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;YAE1D,IAAI,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC;gBACnB,IAAI,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;oBACxC,kDAAkD;oBAClD,MAAM,kBAAkB,GAAG,IAAI,CAAC,QAAQ,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;oBAC9D,MAAM,oBAAoB,GAAG,kBAAkB,CAAC,KAAK,CACnD,CAAC,EACD,CAAC,cAAc,CAAC,MAAM,CACvB,CAAC;oBACF,MAAM,eAAe,GAAG,MAAM,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;oBACrD,aAAa,CAAC,GAAG,CAAC,oBAAoB,EAAE,eAAe,CAAC,CAAC;gBAC3D,CAAC;qBAAM,IAAI,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC;oBAC9C,yEAAyE;oBACzE,MAAM,kBAAkB,GAAG,IAAI,CAAC,QAAQ,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;oBAC9D,MAAM,qBAAqB,GAAG,kBAAkB,CAAC,KAAK,CACpD,CAAC,EACD,CAAC,aAAa,CAAC,MAAM,CACtB,CAAC;oBACF,YAAY,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC;gBAC1C,CAAC;YACH,CAAC;YAED,IAAI,KAAK,CAAC,WAAW,EAAE,EAAE,CAAC;gBACxB,MAAM,mBAAmB,CACvB,QAAQ,EACR,WAAW,EACX,aAAa,EACb,YAAY,CACb,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,oCAAoC;IACtC,CAAC;AACH,CAAC;AAED;;;;;;GAMG;AACH,MAAM,CAAC,KAAK,UAAU,gCAAgC,CACpD,kBAA0B;IAE1B,OAAO,sBAAsB,CAAC,kBAAkB,CAAC,CAAC;AACpD,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,kBAAkB,CAChC,QAA2B;IAE3B,MAAM,MAAM,GAAG,IAAI,GAAG,EAAkB,CAAC;IACzC,MAAM,IAAI,GAAG,CAAC,OAA2B,EAAE,MAAc,EAAE,EAAE;QAC3D,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;YAC5B,MAAM,SAAS,GAAG,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC;YACtC,IAAI,KAAK,CAAC,IAAI,KAAK,GAAG,EAAE,CAAC;gBACvB,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,GAAG,SAAS,GAAG,CAAC,CAAC;YACxC,CAAC;iBAAM,CAAC;gBACN,MAAM,CAAC,GAAG,CACR,SAAS,EACT,KAAK,CAAC,IAAI,KAAK,GAAG,CAAC,CAAC,CAAC,WAAW,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAC5D,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC,CAAC;IACF,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;IAC5B,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/snapshot/types.d.ts

@@ -0,0 +1,25 @@
+import type { DirectoryEntryDir, DirectoryEntryRegFile, DirectoryEntrySymlink } from "@superblocksteam/shared";
+interface SnapshotDirEntryRegFile extends DirectoryEntryRegFile {
+    contents: Buffer;
+}
+interface SnapshotDirEntryDir extends DirectoryEntryDir {
+    contents: SnapshotDirEntry[];
+}
+/** This is like `DirectoryEntry` but includes a complete snapshot of the entire directory tree, including all subdirectories and files. */
+export type SnapshotDirEntry = SnapshotDirEntryRegFile | SnapshotDirEntryDir | DirectoryEntrySymlink;
+export interface DirectorySnapshot {
+    hash: string;
+    contents: SnapshotDirEntry[];
+}
+/** This is like `DirectoryEntry` but includes the recursive contents of all subdirectories in the tree. */
+export interface DirectoryEntryDirWithContents extends DirectoryEntryDir {
+    contents: DirectoryEntryTree[];
+}
+export type DirectoryEntryTree = DirectoryEntryRegFile | DirectoryEntryDirWithContents | DirectoryEntrySymlink;
+/** The state of a directory, i.e. its hash and recursively the hashes of all the hashes of all the files and directories within it. */
+export interface DirectoryState {
+    hash: string;
+    contents: DirectoryEntryTree[];
+}
+export {};
+//# sourceMappingURL=types.d.ts.map

package/dist/snapshot/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/snapshot/types.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EACV,iBAAiB,EACjB,qBAAqB,EACrB,qBAAqB,EACtB,MAAM,yBAAyB,CAAC;AAEjC,UAAU,uBAAwB,SAAQ,qBAAqB;IAC7D,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED,UAAU,mBAAoB,SAAQ,iBAAiB;IACrD,QAAQ,EAAE,gBAAgB,EAAE,CAAC;CAC9B;AAED,2IAA2I;AAC3I,MAAM,MAAM,gBAAgB,GACxB,uBAAuB,GACvB,mBAAmB,GAGnB,qBAAqB,CAAC;AAE1B,MAAM,WAAW,iBAAiB;IAChC,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,gBAAgB,EAAE,CAAC;CAC9B;AAED,2GAA2G;AAC3G,MAAM,WAAW,6BAA8B,SAAQ,iBAAiB;IACtE,QAAQ,EAAE,kBAAkB,EAAE,CAAC;CAChC;AAED,MAAM,MAAM,kBAAkB,GAC1B,qBAAqB,GACrB,6BAA6B,GAC7B,qBAAqB,CAAC;AAE1B,uIAAuI;AACvI,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,kBAAkB,EAAE,CAAC;CAChC"}

package/dist/snapshot/types.js

@@ -0,0 +1,3 @@
+// `type` is a single character following the conventions of the 'ls' command
+export {};
+//# sourceMappingURL=types.js.map

package/dist/snapshot/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/snapshot/types.ts"],"names":[],"mappings":"AAAA,6EAA6E"}

package/dist/snapshot/upload-snapshot.d.ts

@@ -0,0 +1,33 @@
+import type { PutManyRequestPayload } from "@superblocksteam/shared";
+import type { HashCache } from "../hash-cache.js";
+import type { StdISocketRPCClient } from "../rpc-client.js";
+import type { DirectorySnapshot } from "./types.js";
+export type PutManyItems = PutManyRequestPayload["items"];
+/** The two DBFS writes a snapshot upload needs. */
+export interface SnapshotUploadTarget {
+    /**
+     * Checks directory listings before anything is written. Optional: a server
+     * without the method is uploaded to without validation.
+     */
+    validateMany?: (items: PutManyItems) => Promise<void>;
+    putMany: (items: PutManyItems) => Promise<void>;
+}
+/**
+ * The putMany items that bring the server up to `directorySnapshot`, children
+ * before the listings that reference them. Hashes the cache already knows are
+ * left out.
+ */
+export declare function collectSnapshotUploadItems(directorySnapshot: DirectorySnapshot, hashCache: HashCache): PutManyItems;
+/**
+ * Validates every directory listing, then writes every item, in batches of 64.
+ * Returns the number of items written.
+ */
+export declare function uploadSnapshotItems(target: SnapshotUploadTarget, itemsToUpload: PutManyItems): Promise<number>;
+export declare function uploadDirectorySnapshot(rpcClient: StdISocketRPCClient, directorySnapshot: DirectorySnapshot, hashCache: HashCache): Promise<void>;
+/**
+ * {@link SnapshotUploadTarget} over the socket RPC client. Validation is
+ * skipped when the client lacks the method, and once the server reports it
+ * as unsupported.
+ */
+export declare function socketSnapshotUploadTarget(rpcClient: StdISocketRPCClient): SnapshotUploadTarget;
+//# sourceMappingURL=upload-snapshot.d.ts.map

package/dist/snapshot/upload-snapshot.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"upload-snapshot.d.ts","sourceRoot":"","sources":["../../src/snapshot/upload-snapshot.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAEV,qBAAqB,EACtB,MAAM,yBAAyB,CAAC;AAEjC,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAClD,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,kBAAkB,CAAC;AAC5D,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAIpD,MAAM,MAAM,YAAY,GAAG,qBAAqB,CAAC,OAAO,CAAC,CAAC;AAE1D,mDAAmD;AACnD,MAAM,WAAW,oBAAoB;IACnC;;;OAGG;IACH,YAAY,CAAC,EAAE,CAAC,KAAK,EAAE,YAAY,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACtD,OAAO,EAAE,CAAC,KAAK,EAAE,YAAY,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;CACjD;AAWD;;;;GAIG;AACH,wBAAgB,0BAA0B,CACxC,iBAAiB,EAAE,iBAAiB,EACpC,SAAS,EAAE,SAAS,GACnB,YAAY,CAwDd;AAED;;;GAGG;AACH,wBAAsB,mBAAmB,CACvC,MAAM,EAAE,oBAAoB,EAC5B,aAAa,EAAE,YAAY,GAC1B,OAAO,CAAC,MAAM,CAAC,CAajB;AAED,wBAAsB,uBAAuB,CAC3C,SAAS,EAAE,mBAAmB,EAC9B,iBAAiB,EAAE,iBAAiB,EACpC,SAAS,EAAE,SAAS,GACnB,OAAO,CAAC,IAAI,CAAC,CAKf;AAED;;;;GAIG;AACH,wBAAgB,0BAA0B,CACxC,SAAS,EAAE,mBAAmB,GAC7B,oBAAoB,CAiBtB"}

package/dist/snapshot/upload-snapshot.js

@@ -0,0 +1,113 @@
+import { unwrapResponseDto } from "@superblocksteam/shared";
+const BATCH_SIZE = 64;
+function isUnsupportedMethod(error) {
+    return (typeof error === "object" &&
+        error !== null &&
+        "code" in error &&
+        error.code === 2);
+}
+/**
+ * The putMany items that bring the server up to `directorySnapshot`, children
+ * before the listings that reference them. Hashes the cache already knows are
+ * left out.
+ */
+export function collectSnapshotUploadItems(directorySnapshot, hashCache) {
+    const itemsToUpload = [];
+    const processRecursively = (directorySnapshot) => {
+        if (hashCache.check_directory(directorySnapshot.hash)) {
+            // The directory already exists on the server, we don't need to upload anything.
+            return;
+        }
+        const directoryContents = [];
+        for (const snapshotDirEntry of directorySnapshot.contents) {
+            const name = snapshotDirEntry.name;
+            let entry;
+            switch (snapshotDirEntry.type) {
+                case "-": {
+                    if (!hashCache.check_file_content(snapshotDirEntry.hash)) {
+                        // We are not sure if the file exists on the server, so we need to upload it
+                        itemsToUpload.push({
+                            type: "fileContents",
+                            contents: snapshotDirEntry.contents.toString("base64"),
+                        });
+                    }
+                    entry = {
+                        type: "-",
+                        hash: snapshotDirEntry.hash,
+                        executable: snapshotDirEntry.executable,
+                        name,
+                    };
+                    break;
+                }
+                case "d": {
+                    if (!hashCache.check_directory(snapshotDirEntry.hash)) {
+                        processRecursively({
+                            hash: snapshotDirEntry.hash,
+                            contents: snapshotDirEntry.contents,
+                        });
+                    }
+                    entry = { type: "d", name, hash: snapshotDirEntry.hash };
+                    break;
+                }
+                case "l": {
+                    entry = { type: "l", target: snapshotDirEntry.target, name };
+                    break;
+                }
+                default:
+                    throw new Error(`Unknown entry type: ${snapshotDirEntry}`);
+            }
+            directoryContents.push(entry);
+        }
+        itemsToUpload.push({
+            type: "directoryContents",
+            contents: directoryContents,
+        });
+    };
+    processRecursively(directorySnapshot);
+    return itemsToUpload;
+}
+/**
+ * Validates every directory listing, then writes every item, in batches of 64.
+ * Returns the number of items written.
+ */
+export async function uploadSnapshotItems(target, itemsToUpload) {
+    const itemsToValidate = itemsToUpload.filter((item) => item.type === "directoryContents");
+    if (target.validateMany) {
+        for (let idx = 0; idx < itemsToValidate.length; idx += BATCH_SIZE) {
+            await target.validateMany(itemsToValidate.slice(idx, idx + BATCH_SIZE));
+        }
+    }
+    for (let idx = 0; idx < itemsToUpload.length; idx += BATCH_SIZE) {
+        await target.putMany(itemsToUpload.slice(idx, idx + BATCH_SIZE));
+    }
+    return itemsToUpload.length;
+}
+export async function uploadDirectorySnapshot(rpcClient, directorySnapshot, hashCache) {
+    await uploadSnapshotItems(socketSnapshotUploadTarget(rpcClient), collectSnapshotUploadItems(directorySnapshot, hashCache));
+}
+/**
+ * {@link SnapshotUploadTarget} over the socket RPC client. Validation is
+ * skipped when the client lacks the method, and once the server reports it
+ * as unsupported.
+ */
+export function socketSnapshotUploadTarget(rpcClient) {
+    let validationSupported = typeof rpcClient.call.v1.dbfs.validateMany === "function";
+    return {
+        validateMany: async (items) => {
+            if (!validationSupported)
+                return;
+            try {
+                await unwrapResponseDto(rpcClient.call.v1.dbfs.validateMany({ items }));
+            }
+            catch (error) {
+                if (!isUnsupportedMethod(error))
+                    throw error;
+                validationSupported = false;
+            }
+        },
+        putMany: async (items) => {
+            await unwrapResponseDto(rpcClient.call.v1.dbfs.putMany({ items }));
+        },
+    };
+}
+//# sourceMappingURL=upload-snapshot.js.map

package/dist/snapshot/upload-snapshot.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"upload-snapshot.js","sourceRoot":"","sources":["../../src/snapshot/upload-snapshot.ts"],"names":[],"mappings":"AAAA,OAAO,EAAoB,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AAU9E,MAAM,UAAU,GAAG,EAAE,CAAC;AActB,SAAS,mBAAmB,CAAC,KAAc;IACzC,OAAO,CACL,OAAO,KAAK,KAAK,QAAQ;QACzB,KAAK,KAAK,IAAI;QACd,MAAM,IAAI,KAAK;QACd,KAAqB,CAAC,IAAI,KAAK,CAAC,CAClC,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,0BAA0B,CACxC,iBAAoC,EACpC,SAAoB;IAEpB,MAAM,aAAa,GAAiB,EAAE,CAAC;IAEvC,MAAM,kBAAkB,GAAG,CAAC,iBAAoC,EAAE,EAAE;QAClE,IAAI,SAAS,CAAC,eAAe,CAAC,iBAAiB,CAAC,IAAI,CAAC,EAAE,CAAC;YACtD,gFAAgF;YAChF,OAAO;QACT,CAAC;QACD,MAAM,iBAAiB,GAAqB,EAAE,CAAC;QAC/C,KAAK,MAAM,gBAAgB,IAAI,iBAAiB,CAAC,QAAQ,EAAE,CAAC;YAC1D,MAAM,IAAI,GAAG,gBAAgB,CAAC,IAAI,CAAC;YACnC,IAAI,KAAqB,CAAC;YAC1B,QAAQ,gBAAgB,CAAC,IAAI,EAAE,CAAC;gBAC9B,KAAK,GAAG,EAAE,CAAC;oBACT,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,gBAAgB,CAAC,IAAI,CAAC,EAAE,CAAC;wBACzD,4EAA4E;wBAC5E,aAAa,CAAC,IAAI,CAAC;4BACjB,IAAI,EAAE,cAAc;4BACpB,QAAQ,EAAE,gBAAgB,CAAC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC;yBACvD,CAAC,CAAC;oBACL,CAAC;oBACD,KAAK,GAAG;wBACN,IAAI,EAAE,GAAG;wBACT,IAAI,EAAE,gBAAgB,CAAC,IAAI;wBAC3B,UAAU,EAAE,gBAAgB,CAAC,UAAU;wBACvC,IAAI;qBACL,CAAC;oBACF,MAAM;gBACR,CAAC;gBACD,KAAK,GAAG,EAAE,CAAC;oBACT,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC,gBAAgB,CAAC,IAAI,CAAC,EAAE,CAAC;wBACtD,kBAAkB,CAAC;4BACjB,IAAI,EAAE,gBAAgB,CAAC,IAAI;4BAC3B,QAAQ,EAAE,gBAAgB,CAAC,QAAQ;yBACpC,CAAC,CAAC;oBACL,CAAC;oBACD,KAAK,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,IAAI,EAAE,gBAAgB,CAAC,IAAI,EAAE,CAAC;oBACzD,MAAM;gBACR,CAAC;gBACD,KAAK,GAAG,EAAE,CAAC;oBACT,KAAK,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,MAAM,EAAE,gBAAgB,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC;oBAC7D,MAAM;gBACR,CAAC;gBACD;oBACE,MAAM,IAAI,KAAK,CAAC,uBAAuB,gBAAgB,EAAE,CAAC,CAAC;YAC/D,CAAC;YACD,iBAAiB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC;QACD,aAAa,CAAC,IAAI,CAAC;YACjB,IAAI,EAAE,mBAAmB;YACzB,QAAQ,EAAE,iBAAiB;SAC5B,CAAC,CAAC;IACL,CAAC,CAAC;IAEF,kBAAkB,CAAC,iBAAiB,CAAC,CAAC;IACtC,OAAO,aAAa,CAAC;AACvB,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,mBAAmB,CACvC,MAA4B,EAC5B,aAA2B;IAE3B,MAAM,eAAe,GAAG,aAAa,CAAC,MAAM,CAC1C,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,mBAAmB,CAC5C,CAAC;IACF,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;QACxB,KAAK,IAAI,GAAG,GAAG,CAAC,EAAE,GAAG,GAAG,eAAe,CAAC,MAAM,EAAE,GAAG,IAAI,UAAU,EAAE,CAAC;YAClE,MAAM,MAAM,CAAC,YAAY,CAAC,eAAe,CAAC,KAAK,CAAC,GAAG,EAAE,GAAG,GAAG,UAAU,CAAC,CAAC,CAAC;QAC1E,CAAC;IACH,CAAC;IACD,KAAK,IAAI,GAAG,GAAG,CAAC,EAAE,GAAG,GAAG,aAAa,CAAC,MAAM,EAAE,GAAG,IAAI,UAAU,EAAE,CAAC;QAChE,MAAM,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,KAAK,CAAC,GAAG,EAAE,GAAG,GAAG,UAAU,CAAC,CAAC,CAAC;IACnE,CAAC;IACD,OAAO,aAAa,CAAC,MAAM,CAAC;AAC9B,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,uBAAuB,CAC3C,SAA8B,EAC9B,iBAAoC,EACpC,SAAoB;IAEpB,MAAM,mBAAmB,CACvB,0BAA0B,CAAC,SAAS,CAAC,EACrC,0BAA0B,CAAC,iBAAiB,EAAE,SAAS,CAAC,CACzD,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,0BAA0B,CACxC,SAA8B;IAE9B,IAAI,mBAAmB,GACrB,OAAO,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,YAAY,KAAK,UAAU,CAAC;IAC5D,OAAO;QACL,YAAY,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;YAC5B,IAAI,CAAC,mBAAmB;gBAAE,OAAO;YACjC,IAAI,CAAC;gBACH,MAAM,iBAAiB,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;YAC1E,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC;oBAAE,MAAM,KAAK,CAAC;gBAC7C,mBAAmB,GAAG,KAAK,CAAC;YAC9B,CAAC;QACH,CAAC;QACD,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;YACvB,MAAM,iBAAiB,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;QACrE,CAAC;KACF,CAAC;AACJ,CAAC"}

package/dist/socket-dbfs-transport.d.ts

@@ -0,0 +1,32 @@
+import type { DirectoryEntry } from "@superblocksteam/shared";
+import type { DbfsFileRef, DbfsTransport, SetHashRequest, SetHashResult } from "./dbfs-transport.js";
+import type { StdISocketRPCClient } from "./rpc-client.js";
+import { type PutManyItems } from "./snapshot/upload-snapshot.js";
+export interface SocketDbfsTransportOptions {
+    /**
+     * The server client. A function is called for every request, so a host
+     * that reconnects with a refreshed credential hands out the new client.
+     */
+    rpcClient: StdISocketRPCClient | (() => StdISocketRPCClient);
+    applicationId: string;
+    branchName: string | undefined;
+}
+/**
+ * {@link DbfsTransport} over the server socket, using the same RPC methods the
+ * live-edit sync service calls.
+ */
+export declare class SocketDbfsTransport implements DbfsTransport {
+    private readonly options;
+    constructor(options: SocketDbfsTransportOptions);
+    private get client();
+    validateMany(items: PutManyItems): Promise<void>;
+    putMany(items: PutManyItems): Promise<string[]>;
+    getDirectory(hash: string): Promise<DirectoryEntry[]>;
+    getContents(rootHash: string, files: DbfsFileRef[]): Promise<Buffer[]>;
+    getBlob(hash: string): Promise<Buffer>;
+    getServerHash(): Promise<string>;
+    setHash({ hash, source }: SetHashRequest): Promise<SetHashResult>;
+    setDraftState(hasDraft: boolean): Promise<void>;
+    private getManyBatch;
+}
+//# sourceMappingURL=socket-dbfs-transport.d.ts.map

package/dist/socket-dbfs-transport.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"socket-dbfs-transport.d.ts","sourceRoot":"","sources":["../src/socket-dbfs-transport.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AAE9D,OAAO,KAAK,EACV,WAAW,EACX,aAAa,EACb,cAAc,EACd,aAAa,EACd,MAAM,qBAAqB,CAAC;AAG7B,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,iBAAiB,CAAC;AAC3D,OAAO,EACL,KAAK,YAAY,EAElB,MAAM,+BAA+B,CAAC;AAKvC,MAAM,WAAW,0BAA0B;IACzC;;;OAGG;IACH,SAAS,EAAE,mBAAmB,GAAG,CAAC,MAAM,mBAAmB,CAAC,CAAC;IAC7D,aAAa,EAAE,MAAM,CAAC;IACtB,UAAU,EAAE,MAAM,GAAG,SAAS,CAAC;CAChC;AAED;;;GAGG;AACH,qBAAa,mBAAoB,YAAW,aAAa;IAC3C,OAAO,CAAC,QAAQ,CAAC,OAAO;IAApC,YAA6B,OAAO,EAAE,0BAA0B,EAAI;IAEpE,OAAO,KAAK,MAAM,GAGjB;IAEK,YAAY,CAAC,KAAK,EAAE,YAAY,GAAG,OAAO,CAAC,IAAI,CAAC,CAErD;IAEK,OAAO,CAAC,KAAK,EAAE,YAAY,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAKpD;IAEK,YAAY,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC,CAK1D;IAEK,WAAW,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CA2B3E;IAEK,OAAO,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAK3C;IAEK,aAAa,IAAI,OAAO,CAAC,MAAM,CAAC,CAQrC;IAEK,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,cAAc,GAAG,OAAO,CAAC,aAAa,CAAC,CAgBtE;IAEK,aAAa,CAAC,QAAQ,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAQpD;YAEa,YAAY;CAoB3B"}

package/dist/socket-dbfs-transport.js

@@ -0,0 +1,126 @@
+import { unwrapResponseDto } from "@superblocksteam/shared";
+import { getDraftOrLiveDirectoryHash } from "./download.js";
+import { getSyncLogger } from "./logger.js";
+import { socketSnapshotUploadTarget, } from "./snapshot/upload-snapshot.js";
+const GET_MANY_BATCH = 256;
+const BLOB_BATCH = 50;
+/**
+ * {@link DbfsTransport} over the server socket, using the same RPC methods the
+ * live-edit sync service calls.
+ */
+export class SocketDbfsTransport {
+    options;
+    constructor(options) {
+        this.options = options;
+    }
+    get client() {
+        const { rpcClient } = this.options;
+        return typeof rpcClient === "function" ? rpcClient() : rpcClient;
+    }
+    async validateMany(items) {
+        await socketSnapshotUploadTarget(this.client).validateMany?.(items);
+    }
+    async putMany(items) {
+        const response = await unwrapResponseDto(this.client.call.v1.dbfs.putMany({ items }));
+        return response.items.map((item) => item.hash);
+    }
+    async getDirectory(hash) {
+        const response = await unwrapResponseDto(this.client.call.v1.dbfs.directoryContents.get({ hash }));
+        return response.contents;
+    }
+    async getContents(rootHash, files) {
+        const results = [];
+        let bulk = typeof this.client.call.v1.dbfs.getMany === "function";
+        for (let idx = 0; idx < files.length;) {
+            if (bulk) {
+                const batch = files.slice(idx, idx + GET_MANY_BATCH);
+                try {
+                    results.push(...(await this.getManyBatch(rootHash, batch)));
+                    idx += batch.length;
+                    continue;
+                }
+                catch (error) {
+                    if (!isGetManyFallbackError(error))
+                        throw error;
+                    getSyncLogger().warn(`DBFS getMany unavailable, invalid, or exceeded bulk limits (${errorMessage(error)}), falling back to per-file download`);
+                    bulk = false;
+                }
+            }
+            const batch = files.slice(idx, idx + BLOB_BATCH);
+            results.push(...(await Promise.all(batch.map((file) => this.getBlob(file.hash)))));
+            idx += batch.length;
+        }
+        return results;
+    }
+    async getBlob(hash) {
+        const response = await unwrapResponseDto(this.client.call.v1.dbfs.fileContents.get({ hash }));
+        return Buffer.from(response.contents, "base64");
+    }
+    async getServerHash() {
+        const { hash } = await getDraftOrLiveDirectoryHash(this.client, this.options.applicationId, this.options.branchName);
+        if (!hash)
+            throw new Error("No live edit hash found");
+        return hash;
+    }
+    async setHash({ hash, source }) {
+        // `baseHash` is not sent: the server has no compare-and-set until the
+        // branching work (#22956) lands it. Until then the BranchWriteGuard
+        // enforces the base hash on the client before this call.
+        const response = await unwrapResponseDto(this.client.call.v3.application.liveEditDirectoryContents.set({
+            applicationId: this.options.applicationId,
+            branchName: this.options.branchName,
+            hash,
+            source,
+        }));
+        return {
+            hash: response.hash ?? hash,
+            ...(response.commitId ? { commitId: response.commitId } : {}),
+        };
+    }
+    async setDraftState(hasDraft) {
+        await unwrapResponseDto(this.client.call.v3.application.draftState.set({
+            applicationId: this.options.applicationId,
+            branchName: this.options.branchName,
+            hasDraft,
+        }));
+    }
+    async getManyBatch(rootHash, batch) {
+        const response = await unwrapResponseDto(this.client.call.v1.dbfs.getMany({
+            rootHash,
+            paths: batch.map((file) => file.path),
+        }));
+        const byPath = new Map(response.items.map((item) => [item.path, item]));
+        return batch.map((file) => {
+            const item = byPath.get(file.path);
+            if (item?.type !== "fileContents" || item.hash !== file.hash)
+                throw new InvalidGetManyResponseError(`DBFS getMany did not return the expected contents for ${file.path}`);
+            return Buffer.from(item.contents, "base64");
+        });
+    }
+}
+class InvalidGetManyResponseError extends Error {
+}
+function errorMessage(error) {
+    if (typeof error === "object" &&
+        error !== null &&
+        "message" in error &&
+        typeof error.message === "string")
+        return error.message;
+    return String(error);
+}
+/** Mirrors the fallback rules of the live-edit download. */
+function isGetManyFallbackError(error) {
+    if (error instanceof InvalidGetManyResponseError)
+        return true;
+    if (typeof error === "object" &&
+        error !== null &&
+        "code" in error &&
+        error.code === 2)
+        return true;
+    const message = errorMessage(error);
+    return (message.includes("is not implemented") ||
+        message.includes("maxObjects") ||
+        message.includes("maxBytes") ||
+        message.includes("bulk read"));
+}
+//# sourceMappingURL=socket-dbfs-transport.js.map

package/dist/socket-dbfs-transport.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"socket-dbfs-transport.js","sourceRoot":"","sources":["../src/socket-dbfs-transport.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AAS5D,OAAO,EAAE,2BAA2B,EAAE,MAAM,eAAe,CAAC;AAC5D,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAE5C,OAAO,EAEL,0BAA0B,GAC3B,MAAM,+BAA+B,CAAC;AAEvC,MAAM,cAAc,GAAG,GAAG,CAAC;AAC3B,MAAM,UAAU,GAAG,EAAE,CAAC;AAYtB;;;GAGG;AACH,MAAM,OAAO,mBAAmB;IACD,OAAO;IAApC,YAA6B,OAAmC;uBAAnC,OAAO;IAA+B,CAAC;IAEpE,IAAY,MAAM;QAChB,MAAM,EAAE,SAAS,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC;QACnC,OAAO,OAAO,SAAS,KAAK,UAAU,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;IACnE,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,KAAmB;QACpC,MAAM,0BAA0B,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,YAAY,EAAE,CAAC,KAAK,CAAC,CAAC;IACtE,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,KAAmB;QAC/B,MAAM,QAAQ,GAAG,MAAM,iBAAiB,CACtC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,CAAC,CAC5C,CAAC;QACF,OAAO,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACjD,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,IAAY;QAC7B,MAAM,QAAQ,GAAG,MAAM,iBAAiB,CACtC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,EAAE,IAAI,EAAE,CAAC,CACzD,CAAC;QACF,OAAO,QAAQ,CAAC,QAAQ,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,QAAgB,EAAE,KAAoB;QACtD,MAAM,OAAO,GAAa,EAAE,CAAC;QAC7B,IAAI,IAAI,GAAG,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,OAAO,KAAK,UAAU,CAAC;QAClE,KAAK,IAAI,GAAG,GAAG,CAAC,EAAE,GAAG,GAAG,KAAK,CAAC,MAAM,GAAG,CAAC;YACtC,IAAI,IAAI,EAAE,CAAC;gBACT,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,EAAE,GAAG,GAAG,cAAc,CAAC,CAAC;gBACrD,IAAI,CAAC;oBACH,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC;oBAC5D,GAAG,IAAI,KAAK,CAAC,MAAM,CAAC;oBACpB,SAAS;gBACX,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,IAAI,CAAC,sBAAsB,CAAC,KAAK,CAAC;wBAAE,MAAM,KAAK,CAAC;oBAChD,aAAa,EAAE,CAAC,IAAI,CAClB,+DAA+D,YAAY,CACzE,KAAK,CACN,sCAAsC,CACxC,CAAC;oBACF,IAAI,GAAG,KAAK,CAAC;gBACf,CAAC;YACH,CAAC;YACD,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,EAAE,GAAG,GAAG,UAAU,CAAC,CAAC;YACjD,OAAO,CAAC,IAAI,CACV,GAAG,CAAC,MAAM,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CACrE,CAAC;YACF,GAAG,IAAI,KAAK,CAAC,MAAM,CAAC;QACtB,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,IAAY;QACxB,MAAM,QAAQ,GAAG,MAAM,iBAAiB,CACtC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,EAAE,IAAI,EAAE,CAAC,CACpD,CAAC;QACF,OAAO,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;IAClD,CAAC;IAED,KAAK,CAAC,aAAa;QACjB,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,2BAA2B,CAChD,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,OAAO,CAAC,aAAa,EAC1B,IAAI,CAAC,OAAO,CAAC,UAAU,CACxB,CAAC;QACF,IAAI,CAAC,IAAI;YAAE,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;QACtD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAkB;QAC5C,sEAAsE;QACtE,oEAAoE;QACpE,yDAAyD;QACzD,MAAM,QAAQ,GAAG,MAAM,iBAAiB,CACtC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,WAAW,CAAC,yBAAyB,CAAC,GAAG,CAAC;YAC5D,aAAa,EAAE,IAAI,CAAC,OAAO,CAAC,aAAa;YACzC,UAAU,EAAE,IAAI,CAAC,OAAO,CAAC,UAAU;YACnC,IAAI;YACJ,MAAM;SACP,CAAC,CACH,CAAC;QACF,OAAO;YACL,IAAI,EAAE,QAAQ,CAAC,IAAI,IAAI,IAAI;YAC3B,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC9D,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,QAAiB;QACnC,MAAM,iBAAiB,CACrB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,WAAW,CAAC,UAAU,CAAC,GAAG,CAAC;YAC7C,aAAa,EAAE,IAAI,CAAC,OAAO,CAAC,aAAa;YACzC,UAAU,EAAE,IAAI,CAAC,OAAO,CAAC,UAAU;YACnC,QAAQ;SACT,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,YAAY,CACxB,QAAgB,EAChB,KAAoB;QAEpB,MAAM,QAAQ,GAAG,MAAM,iBAAiB,CACtC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC;YAC/B,QAAQ;YACR,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;SACtC,CAAC,CACH,CAAC;QACF,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC;QACxE,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;YACxB,MAAM,IAAI,GAAG,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACnC,IAAI,IAAI,EAAE,IAAI,KAAK,cAAc,IAAI,IAAI,CAAC,IAAI,KAAK,IAAI,CAAC,IAAI;gBAC1D,MAAM,IAAI,2BAA2B,CACnC,yDAAyD,IAAI,CAAC,IAAI,EAAE,CACrE,CAAC;YACJ,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;IACL,CAAC;CACF;AAED,MAAM,2BAA4B,SAAQ,KAAK;CAAG;AAElD,SAAS,YAAY,CAAC,KAAc;IAClC,IACE,OAAO,KAAK,KAAK,QAAQ;QACzB,KAAK,KAAK,IAAI;QACd,SAAS,IAAI,KAAK;QAClB,OAAO,KAAK,CAAC,OAAO,KAAK,QAAQ;QAEjC,OAAO,KAAK,CAAC,OAAO,CAAC;IACvB,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;AACvB,CAAC;AAED,4DAA4D;AAC5D,SAAS,sBAAsB,CAAC,KAAc;IAC5C,IAAI,KAAK,YAAY,2BAA2B;QAAE,OAAO,IAAI,CAAC;IAC9D,IACE,OAAO,KAAK,KAAK,QAAQ;QACzB,KAAK,KAAK,IAAI;QACd,MAAM,IAAI,KAAK;QACf,KAAK,CAAC,IAAI,KAAK,CAAC;QAEhB,OAAO,IAAI,CAAC;IACd,MAAM,OAAO,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC;IACpC,OAAO,CACL,OAAO,CAAC,QAAQ,CAAC,oBAAoB,CAAC;QACtC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAC;QAC9B,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAC;QAC5B,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC,CAC9B,CAAC;AACJ,CAAC"}