@localstack/appinspector-ui 1.0.34 → 1.0.35

package/dist/index.cjs

@@ -4540,16 +4540,10 @@ var TraceGraph = ({
   spans,
   viewPortPixelWidth
 }) => {
-  const [nodes, setNodes] = (0, import_react18.useState)([]);
+  const [nodes, setNodes, onNodesChange] = (0, import_react17.useNodesState)([]);
   const [edges, setEdges] = (0, import_react18.useState)([]);
   const [initialViewSet, setInitialViewSet] = (0, import_react18.useState)(false);
   const { fitView, setCenter } = (0, import_react17.useReactFlow)();
-  const updateSelectedNode = (0, import_react18.useCallback)((nodeId) => {
-    const toSelect = new Set(nodeId ? [nodeId] : []);
-    setNodes((previousNodes) => {
-      return previousNodes.map((node) => ({ ...node, selected: toSelect.has(node.id) }));
-    });
-  }, []);
   (0, import_react18.useEffect)(() => {
     const spansMap = /* @__PURE__ */ new Map();
     const rootSpans = [];
@@ -4584,7 +4578,7 @@ var TraceGraph = ({
     }
     setNodes(layoutedNodes);
     setEdges(layoutedEdges);
-  }, [spans.data, initialFocusEventId]);
+  }, [spans.data, initialFocusEventId, setNodes]);
   (0, import_react18.useEffect)(() => {
     const shouldSetInitialView = !spans.loading && nodes.length > 0 && !initialViewSet;
     if (shouldSetInitialView) {
@@ -4605,15 +4599,6 @@ var TraceGraph = ({
       setInitialViewSet(true);
     }
   }, [spans.loading, nodes, initialFocusEventId, viewPortPixelWidth, setCenter, fitView, onEventSelect, initialViewSet]);
-  (0, import_react18.useEffect)(() => {
-    if (initialFocusEventId) {
-      updateSelectedNode(initialFocusEventId);
-    }
-  }, [initialFocusEventId, updateSelectedNode]);
-  const onNodeClick = (0, import_react18.useCallback)((_event, node) => {
-    onEventSelect(node.data.event);
-    updateSelectedNode(node.id);
-  }, [onEventSelect, updateSelectedNode]);
   const theme = (0, import_material19.useTheme)();
   return /* @__PURE__ */ (0, import_jsx_runtime140.jsxs)(import_jsx_runtime140.Fragment, { children: [
     spans.error && /* @__PURE__ */ (0, import_jsx_runtime140.jsxs)(
@@ -4641,7 +4626,10 @@ var TraceGraph = ({
         minZoom: 0.1,
         nodes,
         nodeTypes,
-        onNodeClick,
+        onNodeClick: (_event, node) => {
+          onEventSelect(node.data.event);
+        },
+        onNodesChange,
         selectNodesOnDrag: false,
         children: [
           /* @__PURE__ */ (0, import_jsx_runtime140.jsx)(Controls, {}),