feat: fix tie()

penpenpng · Dec 3, 2023 · 3362d96 · 3362d96
1 parent 0b1a388
commit 3362d96
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/src/__test__/operator.test.ts b/src/__test__/operator.test.ts
@@ -93,22 +93,27 @@ test("tie()", async () => {
         {
           ...packets[0],
           seenOn: ["wss://aaa.example.com"],
+          isNew: true,
         },
         {
           ...packets[2],
           seenOn: ["wss://aaa.example.com"],
+          isNew: true,
         },
         {
           ...packets[3],
           seenOn: ["wss://aaa.example.com"],
+          isNew: true,
         },
         {
           ...packets[4],
           seenOn: ["wss://aaa.example.com", "wss://bbb.example.com"],
+          isNew: false,
         },
         {
           ...packets[5],
           seenOn: ["wss://aaa.example.com", "wss://bbb.example.com"],
+          isNew: false,
         },
         {
           ...packets[6],
@@ -117,6 +122,7 @@ test("tie()", async () => {
             "wss://bbb.example.com",
             "wss://ccc.example.com",
           ],
+          isNew: false,
         }
       )
     );

diff --git a/src/operator.ts b/src/operator.ts
@@ -101,7 +101,7 @@ export function tie<P extends EventPacket>(
  * Create a customizable tie operator.
  */
 export function createTie<P extends EventPacket>(): [
-  OperatorFunction<P, P & { seenOn: Set<string> }>,
+  OperatorFunction<P, P & { seenOn: Set<string>; isNew: boolean }>,
   Map<string, Set<string>>
 ] {
   const memo = new Map<string, Set<string>>();
@@ -111,13 +111,15 @@ export function createTie<P extends EventPacket>(): [
       filter((packet) => !memo.get(packet.event.id)?.has(packet.from)),
       map((packet) => {
         const seenOn = memo.get(packet.event.id) ?? new Set<string>();
+        const isNew = seenOn.size <= 0;
 
         seenOn.add(packet.from);
         memo.set(packet.event.id, seenOn);
 
         return {
           ...packet,
           seenOn,
+          isNew,
         };
       })
     ),