mild-set 1.1.2 → 1.2.0

package/dist/index.d.ts

@@ -5,5 +5,10 @@ declare class MildSet<V> {
     add(value: V): this;
     delete(value: V): boolean;
     has(value: V): boolean;
+    [Symbol.iterator](): IterableIterator<V>;
+    keys(): IterableIterator<V>;
+    values(): IterableIterator<V>;
+    entries(): IterableIterator<[V, V]>;
+    forEach(callback: (value: V, key: V, set: MildSet<V>) => void, thisArg?: any): undefined;
 }
 export default MildSet;

package/dist/index.js

@@ -5,9 +5,14 @@ class MildSet {
     /* VARIABLES */
     #strong = new Set();
     #weak = new WeakSet();
+    #weakRefs = new Set();
+    #weakRefsMap = new WeakMap();
     #size = 0;
-    #finalizationRegistry = new FinalizationRegistry(() => this.#size -= 1);
     #finalizationTokens = new WeakMap();
+    #finalizationRegistry = new FinalizationRegistry(weakRef => {
+        this.#size -= 1;
+        this.#weakRefs.delete(weakRef);
+    });
     /* CONSTRUCTOR */
     constructor(entries) {
         if (entries) {
@@ -29,8 +34,11 @@ class MildSet {
         if (isWeakReferable(value)) {
             this.#weak.add(value);
             if (!hasValue) {
+                const weakRef = new WeakRef(value);
                 const token = {};
-                this.#finalizationRegistry.register(value, token, token);
+                this.#weakRefs.add(weakRef);
+                this.#weakRefsMap.set(value, weakRef);
+                this.#finalizationRegistry.register(value, weakRef, token);
                 this.#finalizationTokens.set(value, token);
             }
         }
@@ -49,6 +57,11 @@ class MildSet {
             if (token) {
                 this.#finalizationRegistry.unregister(token);
                 this.#finalizationTokens.delete(value);
+                const weakRef = this.#weakRefsMap.get(value);
+                if (weakRef) {
+                    this.#weakRefs.delete(weakRef);
+                    this.#weakRefsMap.delete(value);
+                }
             }
             return this.#weak.delete(value);
         }
@@ -64,6 +77,36 @@ class MildSet {
             return this.#strong.has(value);
         }
     }
+    /* ITERATION API */
+    [Symbol.iterator]() {
+        return this.values();
+    }
+    *keys() {
+        yield* this.values();
+    }
+    *values() {
+        yield* this.#strong.values();
+        for (const weakRef of this.#weakRefs) {
+            const value = weakRef.deref();
+            if (value === undefined)
+                continue;
+            yield value;
+        }
+    }
+    *entries() {
+        yield* this.#strong.entries();
+        for (const weakRef of this.#weakRefs) {
+            const value = weakRef.deref();
+            if (value === undefined)
+                continue;
+            yield [value, value];
+        }
+    }
+    forEach(callback, thisArg) {
+        for (const [key, value] of this.entries()) {
+            callback.call(thisArg, value, key, this);
+        }
+    }
 }
 /* EXPORT */
 export default MildSet;

package/package.json

@@ -3,7 +3,7 @@
   "repository": "github:fabiospampinato/mild-set",
   "description": "A WeakSet that supports any value, it holds strong references to primitives, and weak references to objects.",
   "license": "MIT",
-  "version": "1.1.2",
+  "version": "1.2.0",
   "type": "module",
   "main": "dist/index.js",
   "exports": "./dist/index.js",

package/readme.md

@@ -14,6 +14,8 @@ It has the same API of a WeakSet, except that it just works with primitives too.
 
 It also implements the `size` property, like a Set.
 
+It also implements all the iteration methods, like a Set.
+
 ## License
 
 MIT © Fabio Spampinato