mild-set 1.1.1 → 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

@@ -1,25 +1,20 @@
 /* IMPORT */
-import { isWeakReferable } from 'is';
-/* HELPERS */
-const isWeakable = (value) => {
-    return isWeakReferable(value);
-};
+import { isWeakReferable } from './utils.js';
 /* MAIN */
 class MildSet {
     /* VARIABLES */
-    #strong;
-    #weak;
-    #size;
-    #finalizationRegistry;
-    #finalizationTokens;
+    #strong = new Set();
+    #weak = new WeakSet();
+    #weakRefs = new Set();
+    #weakRefsMap = new WeakMap();
+    #size = 0;
+    #finalizationTokens = new WeakMap();
+    #finalizationRegistry = new FinalizationRegistry(weakRef => {
+        this.#size -= 1;
+        this.#weakRefs.delete(weakRef);
+    });
     /* CONSTRUCTOR */
     constructor(entries) {
-        /* VARIABLES */
-        this.#strong = new Set();
-        this.#weak = new WeakSet();
-        this.#size = 0;
-        this.#finalizationRegistry = new FinalizationRegistry(() => this.#size -= 1);
-        this.#finalizationTokens = new WeakMap();
         if (entries) {
             for (const value of entries) {
                 this.add(value);
@@ -36,11 +31,14 @@ class MildSet {
         if (!hasValue) {
             this.#size += 1;
         }
-        if (isWeakable(value)) {
+        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);
             }
         }
@@ -54,11 +52,16 @@ class MildSet {
         if (!hasValue)
             return false;
         this.#size -= 1;
-        if (isWeakable(value)) {
+        if (isWeakReferable(value)) {
             const token = this.#finalizationTokens.get(value);
             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);
         }
@@ -67,13 +70,43 @@ class MildSet {
         }
     }
     has(value) {
-        if (isWeakable(value)) {
+        if (isWeakReferable(value)) {
             return this.#weak.has(value);
         }
         else {
             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/dist/utils.d.ts

@@ -0,0 +1,2 @@
+declare const isWeakReferable: (value: unknown) => value is WeakKey;
+export { isWeakReferable };

package/dist/utils.js

@@ -0,0 +1,19 @@
+/* MAIN */
+const isWeakReferable = (() => {
+    const SYMBOLS_ARE_WEAK_REFERABLE = (() => {
+        try {
+            new WeakSet().add(Symbol());
+            return true;
+        }
+        catch {
+            return false;
+        }
+    })();
+    return (value) => {
+        if (value === null)
+            return false;
+        return typeof value === 'object' || typeof value === 'function' || (SYMBOLS_ARE_WEAK_REFERABLE && typeof value === 'symbol' && !Symbol.keyFor(value));
+    };
+})();
+/* EXPORT */
+export { isWeakReferable };

package/package.json

@@ -2,7 +2,8 @@
   "name": "mild-set",
   "repository": "github:fabiospampinato/mild-set",
   "description": "A WeakSet that supports any value, it holds strong references to primitives, and weak references to objects.",
-  "version": "1.1.1",
+  "license": "MIT",
+  "version": "1.2.0",
   "type": "module",
   "main": "dist/index.js",
   "exports": "./dist/index.js",
@@ -19,12 +20,9 @@
     "strong",
     "set"
   ],
-  "dependencies": {
-    "is": "npm:@fabiospampinato/is@^2.5.0"
-  },
   "devDependencies": {
-    "fava": "^0.2.0",
-    "tsex": "^2.2.0",
-    "typescript": "^4.9.5"
+    "fava": "^0.3.4",
+    "tsex": "^4.0.2",
+    "typescript": "^5.7.3"
   }
 }

package/readme.md

@@ -5,7 +5,7 @@ A WeakSet that supports any value, it holds strong references to primitives, and
 ## Install
 
 ```sh
-npm install --save mild-set
+npm install mild-set
 ```
 
 ## Usage
@@ -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

package/.editorconfig

@@ -1,10 +0,0 @@
-
-root = true
-
-[*]
-charset = utf-8
-end_of_line = lf
-indent_size = 2
-indent_style = space
-insert_final_newline = true
-trim_trailing_whitespace = true

package/src/index.ts

@@ -1,131 +0,0 @@
-
-/* IMPORT */
-
-import {isWeakReferable} from 'is';
-
-/* HELPERS */
-
-const isWeakable = ( value: unknown ): value is object => { //FIXME: Overridden type guard for convenience, delete this once TS updates its types
-  return isWeakReferable ( value );
-};
-
-/* MAIN */
-
-class MildSet<V> {
-
-  /* VARIABLES */
-
-  #strong = new Set<V> ();
-  #weak = new WeakSet<any> ();
-  #size = 0;
-
-  #finalizationRegistry = new FinalizationRegistry ( () => this.#size -= 1 );
-  #finalizationTokens = new WeakMap<object, object> ();
-
-  /* CONSTRUCTOR */
-
-  constructor ( entries?: readonly V[] | null ) {
-
-    if ( entries ) {
-
-      for ( const value of entries ) {
-
-        this.add ( value );
-
-      }
-
-    }
-
-  }
-
-  /* GETTER API */
-
-  get size () {
-
-    return this.#size;
-
-  }
-
-  /* API */
-
-  add ( value: V ): this {
-
-    const hasValue = this.has ( value );
-
-    if ( !hasValue ) {
-
-      this.#size += 1;
-
-    }
-
-    if ( isWeakable ( value ) ) {
-
-      this.#weak.add ( value );
-
-      if ( !hasValue ) {
-
-        const token = {};
-
-        this.#finalizationRegistry.register ( value, token, token );
-        this.#finalizationTokens.set ( value, token );
-
-      }
-
-    } else {
-
-      this.#strong.add ( value );
-
-    }
-
-    return this;
-
-  }
-
-  delete ( value: V ): boolean {
-
-    const hasValue = this.has ( value );
-
-    if ( !hasValue ) return false;
-
-    this.#size -= 1;
-
-    if ( isWeakable ( value ) ) {
-
-      const token = this.#finalizationTokens.get ( value );
-
-      if ( token ) {
-
-        this.#finalizationRegistry.unregister ( token );
-        this.#finalizationTokens.delete ( value );
-
-      }
-
-      return this.#weak.delete ( value );
-
-    } else {
-
-      return this.#strong.delete ( value );
-
-    }
-
-  }
-
-  has ( value: V ): boolean {
-
-    if ( isWeakable ( value ) ) {
-
-      return this.#weak.has ( value );
-
-    } else {
-
-      return this.#strong.has ( value );
-
-    }
-
-  }
-
-}
-
-/* EXPORT */
-
-export default MildSet;

package/test/index.js

@@ -1,64 +0,0 @@
-
-/* IMPORT */
-
-import {describe} from 'fava';
-import {setTimeout as delay} from 'node:timers/promises';
-import MildSet from '../dist/index.js';
-
-/* MAIN */
-
-describe ( 'MildSet', it => {
-
-  it ( 'works', async t => {
-
-    const set = new MildSet ();
-
-    let primitive = 0;
-    let object = {};
-
-    t.is ( set.size, 0 );
-
-    t.false ( set.has ( primitive ) );
-    t.false ( set.has ( object ) );
-
-    t.false ( set.delete ( primitive ) );
-    t.false ( set.delete ( object ) );
-
-    set.add ( primitive );
-    set.add ( object );
-
-    t.is ( set.size, 2 );
-
-    t.true ( set.has ( primitive ) );
-    t.true ( set.has ( object ) );
-
-    t.true ( set.delete ( primitive ) );
-    t.true ( set.delete ( object ) );
-
-    t.is ( set.size, 0 );
-
-    set.add ( primitive );
-    set.add ( object );
-
-    t.is ( set.size, 2 );
-
-    /* CLEANUP */
-
-    let deleted = 0;
-
-    const registry = new FinalizationRegistry ( () => deleted++ );
-
-    registry.register ( object );
-
-    object = null;
-
-    await delay ( 500 );
-    global.gc ();
-    await delay ( 500 );
-
-    t.is ( deleted, 1 );
-    t.is ( set.size, 1 );
-
-  });
-
-});

package/tsconfig.json

@@ -1,6 +0,0 @@
-{
-  "extends": "tsex/tsconfig.json",
-  "compilerOptions": {
-    "target": "es2022"
-  }
-}