@tinycloud/node-sdk 3.3.0-beta.27 → 3.3.0-beta.29

package/dist/index.cjs

@@ -28902,6 +28902,34 @@ assertSyncEffective_fn = async function() {
     throw new ReplicaStorageError("GRANT_EXPIRED", "The replica grant's effective lifetime has expired.");
   }
 };
+async function listDir(dir) {
+  try {
+    return await (0, import_promises.readdir)(dir);
+  } catch (error) {
+    if (error instanceof Error && "code" in error && (error.code === "ENOENT" || error.code === "ENOTDIR")) {
+      return [];
+    }
+    throw error;
+  }
+}
+async function collectProfileOrphans(root, collectOrphans) {
+  for (const partition of await listDir(root)) {
+    const replicasDir = (0, import_node_path.join)(root, partition, "replicas");
+    for (const replicaName of await listDir(replicasDir)) {
+      await collectOrphans((0, import_node_path.join)(replicasDir, replicaName));
+    }
+  }
+}
+async function profileContentBytes(root, contentBytes) {
+  let total = 0;
+  for (const partition of await listDir(root)) {
+    const replicasDir = (0, import_node_path.join)(root, partition, "replicas");
+    for (const replicaName of await listDir(replicasDir)) {
+      total += await contentBytes((0, import_node_path.join)(replicasDir, replicaName));
+    }
+  }
+  return total;
+}
 function createSqliteReplicaStorage(options) {
   const root = options.dir;
   const profileDir = (0, import_node_path.dirname)(options.gateFile ?? (0, import_node_path.join)((0, import_node_path.dirname)(root), import_contractConstants.REPLICA_GATE_FILE));
@@ -29896,10 +29924,29 @@ function createSqliteReplicaStorage(options) {
       const replicaHash = replicaHashOf(spec.prefix, device.did);
       partitionId = `${identityKeyHash}:${replicaHash}`;
       const replicaDir = (0, import_node_path.join)(idDir, "replicas", replicaHash);
-      const store = await sqlite.SqliteReplicaStore.open(replicaDir, {
-        create: true,
-        guard: storeGuard
-      });
+      const quota = spec.quota?.maxBytes ?? options.maxBytes;
+      if (quota !== void 0 && (!Number.isSafeInteger(quota) || quota < 0)) {
+        throw new TypeError(`maxBytes must be a non-negative safe integer, got ${quota}.`);
+      }
+      if (quota !== void 0) await collectProfileOrphans(root, sqlite.SqliteReplicaStore.collectOrphans);
+      const store = await sqlite.SqliteReplicaStore.open(
+        replicaDir,
+        quota === void 0 ? { create: true, guard: storeGuard } : {
+          create: true,
+          guard: storeGuard,
+          partitionId: (0, import_node_path.basename)(idDir),
+          admit: async (page) => {
+            await collectProfileOrphans(root, sqlite.SqliteReplicaStore.collectOrphans);
+            const used = await profileContentBytes(root, sqlite.SqliteReplicaStore.contentBytes);
+            if (page.blobBytes > 0 && used + page.blobBytes > quota) {
+              throw new replica.ReplicaError(
+                replica.ReplicaErrorCode.STORAGE_BUDGET,
+                `Profile storage holds ${used} bytes; this page adds ${page.blobBytes} against a ${quota}-byte quota. Nothing was committed.`
+              );
+            }
+          }
+        }
+      );
       let generation;
       try {
         try {