add onRouteUpdated

Author: codehz

hydrate.tsx

@@ -13,13 +13,16 @@ const globalX = globalThis as unknown as {
 const match = getRouteMatcher(globalX.__ROUTES__);
 
 export async function hydrate(
-  Shell: React.ComponentType<{ children: React.ReactElement } & ServerSideProps>
+  Shell: React.ComponentType<
+    { children: React.ReactElement } & ServerSideProps
+  >,
+  options?: Omit<React.PropsWithoutRef<typeof RouterHost>, "children" | "Shell">
 ) {
   const matched = match(globalX.__INITIAL_ROUTE__.split("?")[0])!;
   const Initial = await import(matched.value);
   return hydrateRoot(
     document,
-    <RouterHost Shell={Shell}>
+    <RouterHost Shell={Shell} {...options}>
       <Shell {...globalX.__SERVERSIDE_PROPS__}>
         <Initial.default {...globalX.__SERVERSIDE_PROPS__?.props} />
       </Shell>

router/index.tsx

@@ -42,9 +42,11 @@ export const ReloadContext = createContext(async (): Promise<void> => {});
 export const RouterHost = ({
   children,
   Shell,
+  onRouteUpdated,
 }: {
   children: React.ReactElement;
   Shell: React.ComponentType<{ children: React.ReactElement }>;
+  onRouteUpdated?: (path: string) => void;
 }) => {
   const pathname = useLocationProperty(
     () => location.pathname + location.search,
@@ -64,6 +66,7 @@ export const RouterHost = ({
         if (props?.redirect) {
           navigate(props.redirect);
         } else {
+          onRouteUpdated?.(target);
           setCurrent(
             <Shell {...props}>
               <module.default {...props?.props} />
@@ -81,6 +84,7 @@ export const RouterHost = ({
         location.href = pathname;
       });
     } else {
+      onRouteUpdated?.(pathname);
       // @ts-ignore
       delete globalX.__INITIAL_ROUTE__;
     }