Refactor node highlighting

Summary: The previous approach would set the atom many times causing several unnecessary renders

Reviewed By: lblasa

Differential Revision: D47519869

fbshipit-source-id: 536334c892334035a6c0fefc86a2e205b2c9769d

desktop/plugins/public/ui-debugger/index.tsx

@@ -281,16 +281,6 @@ export function plugin(client: PluginClient<Events>) {
     frameworkEvents.update((draft) => {
       if (frameScan?.frameworkEvents) {
         frameScan.frameworkEvents.forEach((frameworkEvent) => {
-          if (
-            uiState.frameworkEventMonitoring.get().get(frameworkEvent.type) ===
-              true &&
-            uiState.isPaused.get() === false
-          ) {
-            highlightedNodes.update((draft) => {
-              draft.add(frameworkEvent.nodeId);
-            });
-          }
-
           const frameworkEventsForNode = draft.get(frameworkEvent.nodeId);
           if (frameworkEventsForNode) {
             frameworkEventsForNode.push(frameworkEvent);
@@ -298,16 +288,36 @@ export function plugin(client: PluginClient<Events>) {
             draft.set(frameworkEvent.nodeId, [frameworkEvent]);
           }
         });
+      }
+    });
+
+    if (uiState.isPaused.get() === true) {
+      return;
+    }
+
+    const monitoredEvents = uiState.frameworkEventMonitoring.get();
+
+    const nodesToHighlight =
+      frameScan.frameworkEvents
+        ?.filter(
+          (frameworkEvent) => monitoredEvents.get(frameworkEvent.type) === true,
+        )
+        .map((event) => event.nodeId) ?? [];
+
+    highlightedNodes.update((draft) => {
+      for (const node of nodesToHighlight) {
+        draft.add(node);
+      }
+    });
+
     setTimeout(() => {
-          highlightedNodes.update((laterDraft) => {
+      highlightedNodes.update((draft) => {
-            for (const event of frameScan.frameworkEvents!!.values()) {
+        for (const nodeId of nodesToHighlight) {
-              laterDraft.delete(event.nodeId);
+          draft.delete(nodeId);
         }
       });
     }, HighlightTime);
   }
-    });
-  }
 
   //todo deal with racecondition, where bloks screen is fetching, takes time then you go back get more recent frame then bloks screen comes and overrites it
   function applyFrameData(