Merge pull request #2366 from stakwork/feature/realtime-filters-support

feat: added filtering for new coming nodes by node type
stakwork · Oct 21, 2024 · eb95b44 · eb95b44
2 parents f083a8c + 65a4388
commit eb95b44
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/stores/useDataStore/index.ts b/src/stores/useDataStore/index.ts
@@ -323,13 +323,17 @@ export const useDataStore = create<DataStore>()(
       console.log(updatedNode)
     },
     addNewNode: (data) => {
-      const { dataInitial: existingData } = get()
+      const { dataInitial: existingData, filters } = get()
 
       if (!data?.nodes) {
         return
       }
 
-      const uniqueIncomingNodes = deduplicateByRefId(data.nodes || [])
+      const nodesFilteredByFilters = filters.node_type.length
+        ? data.nodes.filter((node) => filters.node_type.some((t) => t === node.node_type))
+        : data.nodes
+
+      const uniqueIncomingNodes = deduplicateByRefId(nodesFilteredByFilters || [])
       const uniqueIncomingEdges = deduplicateByRefId(data.edges || [])
 
       // Step 2: Existing nodes and links from the current state