@beam-network/sdk 0.2.12 → 0.2.13

package/dist/client.js

@@ -135,7 +135,7 @@ export class BeamClient {
             const totalWarmupBytes = [...chunksBySource.values()].reduce((sum, chunks) => sum + chunks.reduce((chunkSum, chunk) => chunkSum + chunk.chunk_size, 0), 0);
             if (totalWarmupChunks > 0) {
                 const startedAt = Date.now();
-                emitChecksumEvent(input, {
+                await emitChecksumEvent(input, {
                     event: "checksum_warmup_started",
                     sourceId: "all",
                     chunkCount: totalWarmupChunks,
@@ -149,7 +149,7 @@ export class BeamClient {
                     }
                     await checksumCache.ensureMany(source, chunks);
                 }
-                emitChecksumEvent(input, {
+                await emitChecksumEvent(input, {
                     event: "checksum_warmup_completed",
                     sourceId: "all",
                     chunkCount: totalWarmupChunks,
@@ -158,6 +158,7 @@ export class BeamClient {
                 });
             }
             await input.throwIfCancelled?.();
+            checksumCache.seal();
             preparedSources = await Promise.all(input.sources.map((source, index) => prepareProviderSource(source, { index, expiresIn, fetchImpl: this.fetchImpl })));
             await input.onBeforeTransferPrepare?.();
             prepared = await this.prepareTransfer({
@@ -201,11 +202,7 @@ export class BeamClient {
         const signedUrlFlow = prepared.signed_url_flow ?? requestedSignedUrlFlow;
         if (signedUrlFlow === "signed_url_v1") {
             const chunksBySource = checksumChunksBySource(prepared.chunk_plan, destinationsById);
-            for (const [sourceId, chunks] of chunksBySource) {
-                const source = sourcesById.get(sourceId);
-                if (source)
-                    checksumCache.preloadMany(source, chunks);
-            }
+            checksumCache.assertManyWarmed(chunksBySource);
         }
         const routes = [];
         const multipartUploads = new Map();
@@ -501,20 +498,28 @@ function planSignature(chunks) {
         })
     })));
 }
-function emitChecksumEvent(input, event) {
+async function emitChecksumEvent(input, event) {
     if (!input.onChecksumEvent)
         return;
-    void Promise.resolve(input.onChecksumEvent(event)).catch(() => undefined);
+    await input.onChecksumEvent(event);
 }
 class ChunkChecksumCache {
     options;
     queue = [];
     cache = new Map();
     active = 0;
+    sealed = false;
     constructor(options) {
         this.options = options;
     }
     preloadMany(source, chunks) {
+        if (this.sealed) {
+            const missing = this.uniqueChunks(chunks).filter((chunk) => !this.cache.has(this.key(chunk)));
+            if (missing.length) {
+                throw new Error(`checksum cache missing ${missing.length} chunk(s) after preflight warmup; refusing post-prepare source reads`);
+            }
+            return;
+        }
         const pending = this.uniqueChunks(chunks).filter((chunk) => !this.cache.has(this.key(chunk)));
         for (const chunk of pending) {
             this.cache.set(this.key(chunk), deferredChecksum());
@@ -532,6 +537,9 @@ class ChunkChecksumCache {
         const key = this.key(chunk);
         let existing = this.cache.get(key);
         if (!existing) {
+            if (this.sealed) {
+                return Promise.reject(new Error(`checksum cache missing ${key} after preflight warmup; refusing post-prepare source read`));
+            }
             existing = deferredChecksum();
             this.cache.set(key, existing);
             void this.run(() => this.fillGroup(source, [chunk])).catch(() => undefined);
@@ -550,12 +558,12 @@ class ChunkChecksumCache {
                     continue;
                 const cached = await this.options.provider?.getContentMd5({ source, chunk, algorithm: "content-md5" }).catch(() => undefined);
                 if (cached) {
-                    this.emit({ event: "checksum_cache_hit", sourceId: chunk.source_id, sourceChunkIndex: chunk.source_chunk_index, bytes: chunk.chunk_size });
+                    await this.emit({ event: "checksum_cache_hit", sourceId: chunk.source_id, sourceChunkIndex: chunk.source_chunk_index, bytes: chunk.chunk_size });
                     deferred.resolve(cached);
                     unresolved.delete(key);
                 }
                 else {
-                    this.emit({ event: "checksum_cache_miss", sourceId: chunk.source_id, sourceChunkIndex: chunk.source_chunk_index, bytes: chunk.chunk_size });
+                    await this.emit({ event: "checksum_cache_miss", sourceId: chunk.source_id, sourceChunkIndex: chunk.source_chunk_index, bytes: chunk.chunk_size });
                     misses.push(chunk);
                 }
             }
@@ -579,9 +587,9 @@ class ChunkChecksumCache {
                 const deferred = this.cache.get(this.key(chunk));
                 deferred?.resolve(checksum);
                 unresolved.delete(this.key(chunk));
-                this.emit({ event: "checksum_cold_chunk_cached", sourceId: chunk.source_id, sourceChunkIndex: chunk.source_chunk_index, bytes: chunk.chunk_size });
+                await this.emit({ event: "checksum_cold_chunk_cached", sourceId: chunk.source_id, sourceChunkIndex: chunk.source_chunk_index, bytes: chunk.chunk_size });
             }
-            this.emit({
+            await this.emit({
                 event: "checksum_cold_group_completed",
                 sourceId: misses[0]?.source_id ?? source.id ?? "source",
                 chunkCount: misses.length,
@@ -637,6 +645,21 @@ class ChunkChecksumCache {
     key(chunk) {
         return `${chunk.source_id}:${chunk.source_chunk_index}`;
     }
+    seal() {
+        this.sealed = true;
+    }
+    assertManyWarmed(chunksBySource) {
+        const missing = [];
+        for (const chunks of chunksBySource.values()) {
+            for (const chunk of this.uniqueChunks(chunks)) {
+                if (!this.cache.has(this.key(chunk)))
+                    missing.push(this.key(chunk));
+            }
+        }
+        if (missing.length) {
+            throw new Error(`checksum cache missing ${missing.length} chunk(s) after preflight warmup; refusing post-prepare source reads`);
+        }
+    }
     run(task) {
         return new Promise((resolve, reject) => {
             const start = () => {
@@ -652,10 +675,10 @@ class ChunkChecksumCache {
                 this.queue.push(start);
         });
     }
-    emit(event) {
+    async emit(event) {
         if (!this.options.onEvent)
             return;
-        void Promise.resolve(this.options.onEvent(event)).catch(() => undefined);
+        await this.options.onEvent(event);
     }
 }
 function deferredChecksum() {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@beam-network/sdk",
-  "version": "0.2.12",
+  "version": "0.2.13",
   "description": "TypeScript SDK for BEAM transfer creation and management.",
   "type": "module",
   "license": "MIT",