feat: added filtering for new coming nodes by node type

stakwork · Oct 21, 2024 · 65a4388 · 65a4388
1 parent f083a8c
commit 65a4388
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