microcbor 1.2.0 → 1.2.1

package/lib/decodeAsyncIterable.d.ts

@@ -13,6 +13,8 @@ export declare class Decoder<T extends CBORValue = CBORValue> implements AsyncIt
     private readonly constantView;
     private readonly iter;
     private readonly onFree?;
+    private readonly touchedChunks;
+    private readonly freedChunks;
     constructor(source: AsyncIterable<Uint8Array>, options?: AsyncDecodeOptions);
     [Symbol.asyncIterator]: () => this;
     private allocate;

package/lib/decodeAsyncIterable.js

@@ -8,6 +8,8 @@ export class Decoder {
         this.chunks = [];
         this.constantBuffer = new ArrayBuffer(8);
         this.constantView = new DataView(this.constantBuffer);
+        this.touchedChunks = new WeakSet();
+        this.freedChunks = new WeakSet();
         this[_a] = () => this;
         this.constant = (size, f) => {
             return async () => {
@@ -30,6 +32,16 @@ export class Decoder {
         this.iter = source[Symbol.asyncIterator]();
     }
     async allocate(size) {
+        // If we need more data, first call onFree for all touched chunks
+        // This allows the transform stream to provide more chunks
+        if (this.byteLength < size && this.onFree !== undefined) {
+            for (const chunk of this.chunks) {
+                if (this.touchedChunks.has(chunk) && !this.freedChunks.has(chunk)) {
+                    this.freedChunks.add(chunk);
+                    this.onFree(chunk);
+                }
+            }
+        }
         while (this.byteLength < size) {
             const { done, value } = await this.iter.next();
             if (done) {
@@ -38,6 +50,12 @@ export class Decoder {
             else {
                 this.chunks.push(value);
                 this.byteLength += value.byteLength;
+                // If we still need more data after adding this chunk,
+                // immediately call onFree to allow the next chunk to flow
+                if (this.byteLength < size && this.onFree !== undefined && !this.freedChunks.has(value)) {
+                    this.freedChunks.add(value);
+                    this.onFree(value);
+                }
             }
         }
     }
@@ -49,6 +67,10 @@ export class Decoder {
         let deleteCount = 0;
         for (let i = 0; byteLength < target.byteLength; i++) {
             const chunk = this.chunks[i];
+            // Track which chunks we touched
+            if (!this.touchedChunks.has(chunk)) {
+                this.touchedChunks.add(chunk);
+            }
             const capacity = target.byteLength - byteLength;
             const length = chunk.byteLength - this.offset;
             if (length <= capacity) {
@@ -67,9 +89,14 @@ export class Decoder {
                 this.byteLength -= capacity;
             }
         }
+        // Call onFree for chunks that are being removed (fully consumed)
         if (this.onFree !== undefined) {
             for (let i = 0; i < deleteCount; i++) {
-                this.onFree(this.chunks[i]);
+                const chunk = this.chunks[i];
+                if (!this.freedChunks.has(chunk)) {
+                    this.freedChunks.add(chunk);
+                    this.onFree(chunk);
+                }
             }
         }
         this.chunks.splice(0, deleteCount);

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "microcbor",
-	"version": "1.2.0",
+	"version": "1.2.1",
 	"type": "module",
 	"files": [
 		"lib"
@@ -32,7 +32,7 @@
 	},
 	"homepage": "https://github.com/joeltg/microcbor#readme",
 	"devDependencies": {
-		"@ava/typescript": "^5.0.0",
+		"@ava/typescript": "^6.0.0",
 		"@types/node": "^22.13.9",
 		"ava": "^6.2.0",
 		"cbor": "^10.0.3",