@tinycloud/node-sdk 3.3.0-beta.28 → 3.3.0-beta.30

package/dist/index.js

@@ -29045,6 +29045,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 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 = join2(root, partition, "replicas");
+    for (const replicaName of await listDir(replicasDir)) {
+      await collectOrphans(join2(replicasDir, replicaName));
+    }
+  }
+}
+async function profileContentBytes(root, contentBytes) {
+  let total = 0;
+  for (const partition of await listDir(root)) {
+    const replicasDir = join2(root, partition, "replicas");
+    for (const replicaName of await listDir(replicasDir)) {
+      total += await contentBytes(join2(replicasDir, replicaName));
+    }
+  }
+  return total;
+}
 function createSqliteReplicaStorage(options) {
   const root = options.dir;
   const profileDir = dirname2(options.gateFile ?? join2(dirname2(root), REPLICA_GATE_FILE));
@@ -30039,10 +30067,29 @@ function createSqliteReplicaStorage(options) {
       const replicaHash = replicaHashOf(spec.prefix, device.did);
       partitionId = `${identityKeyHash}:${replicaHash}`;
       const replicaDir = join2(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: 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 {