fix: fix useSelectedLayoutSegment

hi-ogawa · Jun 28, 2024 · 2b8d9a5 · 2b8d9a5
1 parent eaaf866
commit 2b8d9a5
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/packages/react-server-next/src/compat/navigation.tsx b/packages/react-server-next/src/compat/navigation.tsx
@@ -4,6 +4,7 @@ import {
   routerRevalidate,
   useParams as useParams_,
   useRouter as useRouter_,
+  useSelectedParamEntries,
   useSelectedParams,
 } from "@hiogawa/react-server/client";
 import React from "react";
@@ -32,8 +33,8 @@ export function useSelectedLayoutSegments(_todo?: string): string[] {
 }
 
 export function useSelectedLayoutSegment(_todo?: string): string | null {
-  // TODO: probably useSelectedParamEntries()[0]
-  return useSelectedLayoutSegments()[0] ?? null;
+  const [next] = useSelectedParamEntries();
+  return next?.[1] ?? null;
 }
 
 export function useRouter() {