add normalizeUrl

codehz · Feb 2, 2024 · f35cc69 · f35cc69
1 parent 7c57728
commit f35cc69
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/router/index.tsx b/router/index.tsx
@@ -102,15 +102,17 @@ export function useRouteState<T extends {}>(key: string, initial: T) {
 
 export const RouterHost = ({
   children,
+  normalizeUrl = (url: string) => url,
   Shell,
   onRouteUpdated,
 }: {
   children: React.ReactElement;
+  normalizeUrl?: (url: string) => string;
   Shell: React.ComponentType<{ children: React.ReactElement; route?: string }>;
   onRouteUpdated?: (path: string) => void;
 }) => {
   const pathname = useLocationProperty(
-    () => location.pathname + location.search,
+    () => normalizeUrl(location.pathname + location.search),
     () => globalX.__INITIAL_ROUTE__
   );
   const [current, setCurrent] = useState(children);