@localstack/appinspector-ui 1.0.0-dev-a1ff335f → 1.0.0-dev-37c5a6b8

package/dist/index.js

@@ -1,6 +1,6 @@
 // src/pages/spans-list-page.tsx
 import { Box as Box6 } from "@mui/material";
-import { useCallback as useCallback4, useEffect as useEffect3, useRef as useRef3, useState as useState4 } from "react";
+import { useCallback as useCallback5, useEffect as useEffect4, useRef as useRef3, useState as useState5 } from "react";
 import { useNavigate } from "react-router-dom";
 
 // src/api/use-api.tsx
@@ -3254,16 +3254,58 @@ var AppInspectorContextProvider = (props) => {
   ) });
 };
 
+// src/hooks/use-local-storage.tsx
+import { useCallback as useCallback3, useEffect as useEffect2, useState as useState3 } from "react";
+function useLocalStorage(key, initialValue) {
+  const [storedValue, setStoredValue] = useState3(() => {
+    try {
+      const item = globalThis.localStorage.getItem(key);
+      return item ? JSON.parse(item) : initialValue;
+    } catch (error) {
+      console.error(`Error reading localStorage key "${key}":`, error);
+      return initialValue;
+    }
+  });
+  const setValue = useCallback3((value) => {
+    try {
+      setStoredValue(value);
+      globalThis.localStorage.setItem(key, JSON.stringify(value));
+      globalThis.dispatchEvent(new StorageEvent("storage", {
+        key,
+        newValue: JSON.stringify(value)
+      }));
+    } catch (error) {
+      console.error(`Error setting localStorage key "${key}":`, error);
+    }
+  }, [key]);
+  useEffect2(() => {
+    const handleStorageChange = (event) => {
+      if (event.key === key && event.newValue !== null) {
+        try {
+          setStoredValue(JSON.parse(event.newValue));
+        } catch (error) {
+          console.error(`Error parsing localStorage key "${key}":`, error);
+        }
+      }
+    };
+    globalThis.addEventListener("storage", handleStorageChange);
+    return () => {
+      globalThis.removeEventListener("storage", handleStorageChange);
+    };
+  }, [key]);
+  return [storedValue, setValue];
+}
+
 // src/hooks/use-status.tsx
-import { useCallback as useCallback3, useEffect as useEffect2, useRef as useRef2, useState as useState3 } from "react";
+import { useCallback as useCallback4, useEffect as useEffect3, useRef as useRef2, useState as useState4 } from "react";
 var useStatus = () => {
   const api = useAppInspectorApi();
-  const [checking, setChecking] = useState3(true);
-  const [status, setStatus] = useState3();
-  const [error, setError] = useState3();
+  const [checking, setChecking] = useState4(true);
+  const [status, setStatus] = useState4();
+  const [error, setError] = useState4();
   const hasFetchedReference = useRef2(false);
   const abortControllerReference = useRef2(null);
-  const checkStatus = useCallback3(async () => {
+  const checkStatus = useCallback4(async () => {
     if (abortControllerReference.current) {
       abortControllerReference.current.abort();
     }
@@ -3287,7 +3329,7 @@ var useStatus = () => {
       }
     }
   }, [api]);
-  useEffect2(() => {
+  useEffect3(() => {
     if (!hasFetchedReference.current) {
       hasFetchedReference.current = true;
       void checkStatus();
@@ -3301,13 +3343,13 @@ import { jsx as jsx128 } from "react/jsx-runtime";
 var MAX_WIDTH = 1400;
 var SpansListPage = () => {
   const api = useAppInspectorApi();
-  const [spans, setSpans] = useState4();
-  const [loading, setLoading] = useState4(false);
+  const [spans, setSpans] = useState5();
+  const [loading, setLoading] = useState5(false);
   const hasFetched = useRef3(false);
-  const [filterWriteOperations, setFilterWriteOperations] = useState4(true);
-  const [fetchError, setFetchError] = useState4();
+  const [filterWriteOperations, setFilterWriteOperations] = useLocalStorage("filterWriteOperations", true);
+  const [fetchError, setFetchError] = useState5();
   const { checkStatus, error: statusError, status } = useStatus();
-  const fetchSpans = useCallback4(async () => {
+  const fetchSpans = useCallback5(async () => {
     setLoading(true);
     try {
       const response = await api.getSpans({
@@ -3323,26 +3365,26 @@ var SpansListPage = () => {
       setLoading(false);
     }
   }, [api, filterWriteOperations]);
-  const fetchSpansSync = useCallback4(() => {
+  const fetchSpansSync = useCallback5(() => {
     void fetchSpans();
   }, [fetchSpans]);
-  useEffect3(() => {
+  useEffect4(() => {
     if (hasFetched.current) return;
     if (status?.status !== "ENABLED") return;
     hasFetched.current = true;
     void fetchSpans();
   }, [fetchSpans, status]);
-  useEffect3(() => {
+  useEffect4(() => {
     if (!hasFetched.current) return;
     void fetchSpans();
   }, [fetchSpans]);
   const navigate = useNavigate();
   const { resolveLink } = useAppInspector();
-  const navigateToSpan = useCallback4((span) => {
+  const navigateToSpan = useCallback5((span) => {
     const url = resolveLink({ to: "trace-graph", traceId: span.trace_id });
     navigate(`${url}?spanId=${span.span_id}`);
   }, [navigate, resolveLink]);
-  const clearSpans = useCallback4(async () => {
+  const clearSpans = useCallback5(async () => {
     setLoading(true);
     try {
       await api.deleteSpans();
@@ -3356,7 +3398,7 @@ var SpansListPage = () => {
       setLoading(false);
     }
   }, [api, fetchSpans]);
-  const handleRetry = useCallback4(() => {
+  const handleRetry = useCallback5(() => {
     void checkStatus();
   }, [checkStatus]);
   const shouldShowStatusMessage = Boolean(statusError) || status?.status === "DISABLED";
@@ -3417,11 +3459,11 @@ var SpansListPage = () => {
   );
 };
 
-// src/pages/trace-graph.tsx
+// src/pages/trace-graph-page.tsx
 import { Box as Box12, Drawer, Paper as Paper2 } from "@mui/material";
 import { styled } from "@mui/material/styles";
 import { ReactFlowProvider } from "@xyflow/react";
-import { useCallback as useCallback7, useEffect as useEffect5, useState as useState7 } from "react";
+import { useCallback as useCallback8, useEffect as useEffect6, useState as useState8 } from "react";
 import { useParams, useSearchParams } from "react-router-dom";
 
 // src/components/event-details/event-details.tsx
@@ -3462,10 +3504,10 @@ var EventDetailsHeader = ({ event, onClose }) => {
 // src/components/event-details/toggle-section.tsx
 import { KeyboardArrowDown, KeyboardArrowRight } from "@mui/icons-material";
 import { Box as Box8, Typography as Typography5 } from "@mui/material";
-import { useState as useState5 } from "react";
+import { useState as useState6 } from "react";
 import { jsx as jsx130, jsxs as jsxs124 } from "react/jsx-runtime";
 var ToggleSection = ({ children, headline, initialOpen = true }) => {
-  const [isOpen, setIsOpen] = useState5(initialOpen);
+  const [isOpen, setIsOpen] = useState6(initialOpen);
   return /* @__PURE__ */ jsxs124(Box8, { sx: { mt: 4 }, children: [
     /* @__PURE__ */ jsx130(
       Box8,
@@ -3575,13 +3617,13 @@ var EventDetails = ({ selectedEvent }) => {
 import "@xyflow/react/dist/base.css";
 import { Alert as Alert3, Button as Button4 } from "@mui/material";
 import { Background, Controls, ReactFlow, useReactFlow, useStoreApi } from "@xyflow/react";
-import { useCallback as useCallback6, useEffect as useEffect4, useState as useState6 } from "react";
+import { useCallback as useCallback7, useEffect as useEffect5, useState as useState7 } from "react";
 
 // src/components/trace-graph/service-node.tsx
 import { Box as Box11, Stack as Stack2, Typography as Typography7 } from "@mui/material";
 import "@xyflow/react/dist/style.css";
 import { Handle, Position } from "@xyflow/react";
-import { memo, useCallback as useCallback5 } from "react";
+import { memo, useCallback as useCallback6 } from "react";
 
 // src/components/trace-graph/problem-indicator.tsx
 import { Error as ErrorIcon, Info, Warning } from "@mui/icons-material";
@@ -3642,10 +3684,10 @@ var handleStyle = {
   opacity: 0
 };
 var ServiceNode = memo(function ServiceNode2({ data, selected }) {
-  const handleClick = useCallback5(() => {
+  const handleClick = useCallback6(() => {
     data.onEventSelect(data.event);
   }, [data]);
-  const handleKeyPress = useCallback5(
+  const handleKeyPress = useCallback6(
     (event) => {
       if (event.key === "Enter" || event.key === " ") {
         handleClick();
@@ -3838,18 +3880,16 @@ var nodeTypes = {
 var TraceGraph = ({
   initialFocusEventId,
   onEventSelect,
-  traceId,
+  onRefresh,
+  spans,
   viewPortPixelWidth
 }) => {
-  const [nodes, setNodes] = useState6([]);
-  const [edges, setEdges] = useState6([]);
-  const [isLoading, setIsLoading] = useState6(true);
-  const [initialViewSet, setInitialViewSet] = useState6(false);
+  const [nodes, setNodes] = useState7([]);
+  const [edges, setEdges] = useState7([]);
+  const [initialViewSet, setInitialViewSet] = useState7(false);
   const { fitView, setCenter } = useReactFlow();
   const store = useStoreApi();
-  const api = useAppInspectorApi();
-  const [fetchError, setFetchError] = useState6();
-  const updateSelectedNode = useCallback6(
+  const updateSelectedNode = useCallback7(
     (nodeId) => {
       const { addSelectedNodes } = store.getState();
       if (nodeId) {
@@ -3861,57 +3901,40 @@ var TraceGraph = ({
     },
     [nodes, store]
   );
-  const fetchTrace = useCallback6(async () => {
-    setIsLoading(true);
-    try {
-      const response = await api.getSpans({ trace_id: traceId });
-      const spansMap = /* @__PURE__ */ new Map();
-      const rootSpans = [];
-      for (const span of response.spans) {
-        spansMap.set(span.span_id, { ...span, children: [] });
-      }
-      for (const span of response.spans) {
-        const currentSpan = spansMap.get(span.span_id);
-        if (!currentSpan) continue;
-        if (span.parent_span_id && spansMap.has(span.parent_span_id)) {
-          const parentSpan = spansMap.get(span.parent_span_id);
-          if (parentSpan) {
-            parentSpan.children ??= [];
-            parentSpan.children.push(currentSpan);
-          }
-        } else {
-          rootSpans.push(currentSpan);
+  useEffect5(() => {
+    const spansMap = /* @__PURE__ */ new Map();
+    const rootSpans = [];
+    for (const span of spans.data ?? []) {
+      spansMap.set(span.span_id, { ...span, children: [] });
+    }
+    for (const span of spans.data ?? []) {
+      const currentSpan = spansMap.get(span.span_id);
+      if (!currentSpan) continue;
+      if (span.parent_span_id && spansMap.has(span.parent_span_id)) {
+        const parentSpan = spansMap.get(span.parent_span_id);
+        if (parentSpan) {
+          parentSpan.children ??= [];
+          parentSpan.children.push(currentSpan);
         }
+      } else {
+        rootSpans.push(currentSpan);
       }
-      let allNodes = [];
-      let allEdges = [];
-      for (const rootSpan of rootSpans) {
-        const { edges: newEdges, nodes: newNodes } = generateNodesAndEdges(rootSpan, (event) => {
-          onEventSelect(event);
-        });
-        allNodes = [...allNodes, ...newNodes];
-        allEdges = [...allEdges, ...newEdges];
-      }
-      const { edges: layoutedEdges, nodes: layoutedNodes } = getLayoutedNodesAndEdges(allNodes, allEdges);
-      setNodes(layoutedNodes);
-      setEdges(layoutedEdges);
-      setIsLoading(false);
-      setFetchError(void 0);
-    } catch (error) {
-      console.error("Error fetching trace:", error);
-      setFetchError(error instanceof Error ? error : new Error("Unknown error", { cause: error }));
-    } finally {
-      setIsLoading(false);
     }
-  }, [api, traceId, onEventSelect]);
-  const fetchTraceSync = useCallback6(() => {
-    void fetchTrace();
-  }, [fetchTrace]);
-  useEffect4(() => {
-    fetchTraceSync();
-  }, [fetchTraceSync]);
-  useEffect4(() => {
-    const shouldSetInitialView = !isLoading && nodes.length > 0 && !initialViewSet;
+    let allNodes = [];
+    let allEdges = [];
+    for (const rootSpan of rootSpans) {
+      const { edges: newEdges, nodes: newNodes } = generateNodesAndEdges(rootSpan, (event) => {
+        onEventSelect(event);
+      });
+      allNodes = [...allNodes, ...newNodes];
+      allEdges = [...allEdges, ...newEdges];
+    }
+    const { edges: layoutedEdges, nodes: layoutedNodes } = getLayoutedNodesAndEdges(allNodes, allEdges);
+    setNodes(layoutedNodes);
+    setEdges(layoutedEdges);
+  }, [onEventSelect, spans.data]);
+  useEffect5(() => {
+    const shouldSetInitialView = !spans.loading && nodes.length > 0 && !initialViewSet;
     if (shouldSetInitialView) {
       if (initialFocusEventId) {
         const focusNode = nodes.find((node) => node.id === initialFocusEventId);
@@ -3929,26 +3952,28 @@ var TraceGraph = ({
       }
       setInitialViewSet(true);
     }
-  }, [isLoading, nodes, initialFocusEventId, viewPortPixelWidth, setCenter, fitView, onEventSelect, initialViewSet]);
-  useEffect4(() => {
+  }, [spans.loading, nodes, initialFocusEventId, viewPortPixelWidth, setCenter, fitView, onEventSelect, initialViewSet]);
+  useEffect5(() => {
     if (initialFocusEventId) {
       updateSelectedNode(initialFocusEventId);
     }
   }, [initialFocusEventId, updateSelectedNode]);
-  const onNodeClick = useCallback6((_event, node) => {
+  const onNodeClick = useCallback7((_event, node) => {
     onEventSelect(node.data.event);
     updateSelectedNode(node.id);
   }, [onEventSelect, updateSelectedNode]);
   return /* @__PURE__ */ jsxs127(Fragment3, { children: [
-    fetchError && /* @__PURE__ */ jsxs127(
+    spans.error && /* @__PURE__ */ jsxs127(
       Alert3,
       {
-        action: /* @__PURE__ */ jsx134(Button4, { onClick: fetchTraceSync, children: "Retry" }),
+        action: /* @__PURE__ */ jsx134(Button4, { onClick: () => {
+          onRefresh();
+        }, children: "Retry" }),
         severity: "error",
         children: [
           "Unexpected error:",
           " ",
-          fetchError.message
+          spans.error.message
         ]
       }
     ),
@@ -3974,7 +3999,7 @@ var TraceGraph = ({
   ] });
 };
 
-// src/pages/trace-graph.tsx
+// src/pages/trace-graph-page.tsx
 import { Fragment as Fragment4, jsx as jsx135, jsxs as jsxs128 } from "react/jsx-runtime";
 var HEADER_HEIGHT = 56;
 var DRAWER_WIDTH = "35%";
@@ -3995,7 +4020,6 @@ var Main = styled("main", { shouldForwardProp: (property) => property !== "open"
 }));
 var StyledDrawer = styled(Drawer)(() => ({
   "& .MuiDrawer-paper": {
-    backgroundColor: "#ffffff",
     bottom: 0,
     boxSizing: "border-box",
     height: `calc(100% - ${HEADER_HEIGHT.toString()}px)`,
@@ -4011,11 +4035,33 @@ var TraceGraphPage = () => {
   const { traceId } = useParams();
   const [searchParameters] = useSearchParams();
   const spanId = searchParameters.get("spanId");
-  const [open, setOpen] = useState7(true);
-  const [selectedEvent, setSelectedEvent] = useState7();
-  const [viewPortPixelWidth, setViewPortPixelWidth] = useState7(0);
+  const api = useAppInspectorApi();
+  const [spans, setSpans] = useState8();
+  const [loading, setLoading] = useState8(false);
+  const [fetchError, setFetchError] = useState8();
+  const fetchSpans = useCallback8(async () => {
+    setLoading(true);
+    try {
+      const response = await api.getSpans({
+        trace_id: traceId
+      });
+      setSpans(response.spans);
+      setFetchError(void 0);
+    } catch (error) {
+      console.error(error);
+      setFetchError(error instanceof Error ? error : new Error("Unknown error", { cause: error }));
+    } finally {
+      setLoading(false);
+    }
+  }, [api, traceId]);
+  useEffect6(() => {
+    void fetchSpans();
+  }, [fetchSpans]);
+  const [open, setOpen] = useState8(true);
+  const [selectedEvent, setSelectedEvent] = useState8();
+  const [viewPortPixelWidth, setViewPortPixelWidth] = useState8(0);
   const { checkStatus, error: statusError, status } = useStatus();
-  const calculateDrawerWidth = useCallback7(() => {
+  const calculateDrawerWidth = useCallback8(() => {
     const percentageMatch = /(\d+)%/.exec(DRAWER_WIDTH);
     if (percentageMatch && open) {
       const percentage = Number.parseInt(percentageMatch[1], 10);
@@ -4023,7 +4069,7 @@ var TraceGraphPage = () => {
     }
     return 0;
   }, [open]);
-  useEffect5(() => {
+  useEffect6(() => {
     const updateViewportWidth = () => {
       const totalWidth = window.innerWidth;
       const drawerPixelWidth = open ? calculateDrawerWidth() : 0;
@@ -4037,11 +4083,11 @@ var TraceGraphPage = () => {
       clearTimeout(timeoutId);
     };
   }, [open, calculateDrawerWidth]);
-  const handleEventSelect = useCallback7((event) => {
+  const handleEventSelect = useCallback8((event) => {
     setSelectedEvent(event);
     setOpen(true);
   }, []);
-  const handleRetry = useCallback7(() => {
+  const handleRetry = useCallback8(() => {
     void checkStatus();
   }, [checkStatus]);
   const shouldShowStatusMessage = Boolean(statusError) || status?.status === "DISABLED";
@@ -4086,7 +4132,14 @@ var TraceGraphPage = () => {
                   {
                     initialFocusEventId: spanId,
                     onEventSelect: handleEventSelect,
-                    traceId,
+                    onRefresh: () => {
+                      void fetchSpans();
+                    },
+                    spans: {
+                      data: spans,
+                      error: fetchError,
+                      loading
+                    },
                     viewPortPixelWidth
                   }
                 ) })