From 6fe91f8a1b5aa174d628b6102a28a4226871a159 Mon Sep 17 00:00:00 2001
From: Victor Oliva <olivarra1@gmail.com>
Date: Fri, 1 Apr 2022 10:16:12 +0200
Subject: [PATCH] feat: utils `toKeySet` (#252)

* feat: utils `toKeySet`

* Update packages/utils/src/toKeySet.ts

Co-authored-by: Josep M Sobrepere <jm.sobrepere@gmail.com>

* Update packages/utils/src/toKeySet.test.ts

Co-authored-by: Josep M Sobrepere <jm.sobrepere@gmail.com>

* organize test imports

Co-authored-by: Josep M Sobrepere <jm.sobrepere@gmail.com>
---
 packages/utils/src/index.tsx        |  1 +
 packages/utils/src/toKeySet.test.ts | 98 +++++++++++++++++++++++++++++
 packages/utils/src/toKeySet.ts      | 28 +++++++++
 3 files changed, 127 insertions(+)
 create mode 100644 packages/utils/src/toKeySet.test.ts
 create mode 100644 packages/utils/src/toKeySet.ts

diff --git a/packages/utils/src/index.tsx b/packages/utils/src/index.tsx
index 980cd84..c41d2d5 100644
--- a/packages/utils/src/index.tsx
+++ b/packages/utils/src/index.tsx
@@ -3,6 +3,7 @@ export { createSignal } from "./createSignal"
 export { createKeyedSignal } from "./createKeyedSignal"
 export { mergeWithKey } from "./mergeWithKey"
 export { partitionByKey, KeyChanges } from "./partitionByKey"
+export { toKeySet } from "./toKeySet"
 export { suspend } from "./suspend"
 export { suspended } from "./suspended"
 export { switchMapSuspended } from "./switchMapSuspended"
diff --git a/packages/utils/src/toKeySet.test.ts b/packages/utils/src/toKeySet.test.ts
new file mode 100644
index 0000000..731d5f7
--- /dev/null
+++ b/packages/utils/src/toKeySet.test.ts
@@ -0,0 +1,98 @@
+import { asapScheduler, map, observeOn, of, Subject } from "rxjs"
+import { TestScheduler } from "rxjs/testing"
+import { KeyChanges, toKeySet } from "./"
+
+const scheduler = () =>
+  new TestScheduler((actual, expected) => {
+    expect(actual).toEqual(expected)
+  })
+
+describe("toKeySet", () => {
+  it("transforms key changes to a Set", () => {
+    scheduler().run(({ expectObservable, cold }) => {
+      const expectedStr = "                     xe--f-g--h#"
+      const source$ = cold<KeyChanges<string>>("-a--b-c--d#", {
+        a: {
+          type: "add",
+          keys: ["a", "b"],
+        },
+        b: {
+          type: "remove",
+          keys: ["b", "c"],
+        },
+        c: {
+          type: "add",
+          keys: ["c"],
+        },
+        d: {
+          type: "remove",
+          keys: ["a"],
+        },
+      })
+
+      const result$ = source$.pipe(
+        toKeySet(),
+        map((s) => Array.from(s)),
+      )
+
+      expectObservable(result$).toBe(expectedStr, {
+        x: [],
+        e: ["a", "b"],
+        f: ["a"],
+        g: ["a", "c"],
+        h: ["c"],
+      })
+    })
+  })
+
+  it("emits synchronously on the first subscribe if it receives a synchronous change", () => {
+    const emissions: string[][] = []
+    of<KeyChanges<string>>({
+      type: "add",
+      keys: ["a", "b"],
+    })
+      .pipe(toKeySet())
+      .subscribe((next) => emissions.push(Array.from(next)))
+
+    expect(emissions.length).toBe(1)
+    expect(emissions[0]).toEqual(["a", "b"])
+  })
+
+  it("emits synchronously an empty Set if it doesn't receive a synchronous change", () => {
+    const emissions: string[][] = []
+    of<KeyChanges<string>>({
+      type: "add",
+      keys: ["a", "b"],
+    })
+      .pipe(observeOn(asapScheduler), toKeySet())
+      .subscribe((next) => emissions.push(Array.from(next)))
+
+    expect(emissions.length).toBe(1)
+    expect(emissions[0]).toEqual([])
+  })
+
+  it("resets the Set after unsubscribing", () => {
+    const input$ = new Subject<KeyChanges<string>>()
+    const result$ = input$.pipe(toKeySet())
+
+    let emissions: string[][] = []
+    let sub = result$.subscribe((v) => emissions.push(Array.from(v)))
+    input$.next({
+      type: "add",
+      keys: ["a"],
+    })
+    expect(emissions.length).toBe(2) // [0] is initial empty []
+    expect(emissions[1]).toEqual(["a"])
+    sub.unsubscribe()
+
+    emissions = []
+    sub = result$.subscribe((v) => emissions.push(Array.from(v)))
+    input$.next({
+      type: "add",
+      keys: ["b"],
+    })
+    expect(emissions.length).toBe(2) // [0] is initial empty []
+    expect(emissions[1]).toEqual(["b"])
+    sub.unsubscribe()
+  })
+})
diff --git a/packages/utils/src/toKeySet.ts b/packages/utils/src/toKeySet.ts
new file mode 100644
index 0000000..6ce519c
--- /dev/null
+++ b/packages/utils/src/toKeySet.ts
@@ -0,0 +1,28 @@
+import { Observable, OperatorFunction } from "rxjs"
+import { KeyChanges } from "./partitionByKey"
+
+export function toKeySet<K>(): OperatorFunction<KeyChanges<K>, Set<K>> {
+  return (source$) =>
+    new Observable<Set<K>>((observer) => {
+      const result = new Set<K>()
+      let pristine = true
+      const subscription = source$.subscribe({
+        next({ type, keys }) {
+          const action = type === "add" ? type : "delete"
+          for (let k of keys) {
+            result[action](k)
+          }
+          observer.next(result)
+          pristine = false
+        },
+        error(e) {
+          observer.error(e)
+        },
+        complete() {
+          observer.complete()
+        },
+      })
+      if (pristine) observer.next(result)
+      return subscription
+    })
+}