diff --git a/app/@span/(.)[...span]/page.tsx b/app/@span/(.)[...span]/page.tsx
index e58d055..a6af618 100644
--- a/app/@span/(.)[...span]/page.tsx
+++ b/app/@span/(.)[...span]/page.tsx
@@ -1,17 +1,24 @@
import TxData from "@/components/tx-data";
-import { Sheet, SheetContent } from "@/components/ui/sheet";
+import TxSheet from "@/components/tx-data/tx-sheet";
-export default function Tx({ params }: { params: { span: string[] } }) {
- const [txId, spanId] = params.span;
+const traceIdLength = 32;
+
+export default function Tx({ params }: { params: { span: [string] } }) {
+ //NOTE - the params are parsed because this catch-all route merges both traceId and spanId
+ const [txId, spanId] =
+ params.span[0].length > traceIdLength
+ ? [
+ params.span[0].slice(0, traceIdLength),
+ params.span[0].slice(traceIdLength),
+ ]
+ : [params.span[0]];
return (
-
-
-
-
-
+
+
+
);
}
diff --git a/app/[...span]/page.tsx b/app/[...span]/page.tsx
index febffd3..f02b679 100644
--- a/app/[...span]/page.tsx
+++ b/app/[...span]/page.tsx
@@ -1,6 +1,10 @@
import TxData from "@/components/tx-data";
-export default function Tx({ params }: { params: { span: string[] } }) {
+export default function Tx({
+ params,
+}: {
+ params: { span: [string, string | undefined] };
+}) {
const [txId, spanId] = params.span;
return (
diff --git a/components/mermaid.tsx b/components/mermaid.tsx
index c13d40b..a9c7c56 100644
--- a/components/mermaid.tsx
+++ b/components/mermaid.tsx
@@ -2,6 +2,7 @@
import { cn } from "@/lib/utils";
import mermaid from "mermaid";
+import { usePathname } from "next/navigation";
import { useEffect, useState } from "react";
mermaid.initialize({
@@ -36,7 +37,13 @@ const htmlReplacer = {
] ?? "",
};
-export default function Mermaid({ chart }: { chart: string }) {
+type MermaidProps = {
+ chart: string;
+ setSpanId: (spanId: string | undefined) => void;
+};
+
+export default function Mermaid({ chart, setSpanId }: MermaidProps) {
+ const pathname = usePathname();
const [renderStage, setRenderStage] = useState<
"server" | "browser" | "mermaid"
>("server");
@@ -67,6 +74,25 @@ export default function Mermaid({ chart }: { chart: string }) {
htmlReplacer.regex,
htmlReplacer.fn,
);
+
+ const span = msg.firstElementChild;
+ if (!span) {
+ continue;
+ }
+
+ const linkToSpan = span.getAttribute("href");
+ if (!linkToSpan) {
+ continue;
+ }
+
+ span.addEventListener("click", (e) => {
+ e.preventDefault();
+
+ if (!pathname.endsWith(linkToSpan)) {
+ window.history.replaceState(null, "", linkToSpan);
+ setSpanId(linkToSpan.split("/").slice(-1)[0]);
+ }
+ });
}
}
@@ -74,7 +100,7 @@ export default function Mermaid({ chart }: { chart: string }) {
});
return () => clearInterval(interval);
- }, [renderStage]);
+ }, [pathname, renderStage, setSpanId]);
return renderStage !== "server" ? (
diff --git a/components/tx-data.tsx b/components/tx-data/index.tsx
similarity index 68%
rename from components/tx-data.tsx
rename to components/tx-data/index.tsx
index e4ce46d..6aa33b8 100644
--- a/components/tx-data.tsx
+++ b/components/tx-data/index.tsx
@@ -2,27 +2,30 @@
import { useTx } from "@/hooks/api";
import { sequenceDiagramFromSpans } from "@/lib/mermaid";
-import Mermaid from "./mermaid";
+import { useState } from "react";
+import Mermaid from "../mermaid";
type TxDataProps = {
txId: string;
- spanId: string;
+ spanId: string | undefined;
};
export default function TxData({ txId, spanId }: TxDataProps) {
const { isPending, isFetching, error, data: tx } = useTx(txId);
+ const [spanIdToFind, setSpanIdToFind] = useState(spanId);
if (isPending) return "Loading...";
if (error) return "An error has occurred: " + error.message;
if (!tx) return "Couldn't find a Tx with id: " + txId;
const mermaidChart = sequenceDiagramFromSpans(tx.spans);
-
- const span = tx.spans.find((span) => span.spanId === spanId);
+ const span = tx.spans.find((span) => span.spanId === spanIdToFind);
return (
- {!isFetching ?
: null}
+ {!isFetching ? (
+
+ ) : null}
{span ? (
{JSON.stringify(
diff --git a/components/tx-data/tx-sheet.tsx b/components/tx-data/tx-sheet.tsx
new file mode 100644
index 0000000..a4e5e42
--- /dev/null
+++ b/components/tx-data/tx-sheet.tsx
@@ -0,0 +1,22 @@
+"use client";
+
+import { Sheet, SheetContent } from "@/components/ui/sheet";
+import { useRouter } from "next/navigation";
+import { PropsWithChildren } from "react";
+
+export default function TxSheet({ children }: PropsWithChildren) {
+ const router = useRouter();
+
+ return (
+ {
+ if (!open) {
+ router.back();
+ }
+ }}
+ >
+ {children}
+
+ );
+}