Supported refetching the whole scene

Pass true to the refetch and it's the whole scene, otherwise it's just the scene view
const refetch = useRefetch();
refetch(true)
Tried setState(prevContext => nextContext) but this infinitely looped. Don't think it's supported in startTransition

Author: grahammendick

NavigationReact/src/NavigationHandler.tsx

@@ -1,8 +1,9 @@
 'use client'
-import React, { Component } from 'react';
+import React, { Component, startTransition } from 'react';
 import { StateNavigator, State } from 'navigation';
 import AsyncStateNavigator from './AsyncStateNavigator.js';
 import NavigationContext from './NavigationContext.js';
+import SceneRSCContext from './SceneRSCContext.js';
 type NavigationHandlerState = { context: { ignoreCache?: boolean, oldState: State, state: State, data: any, asyncData: any, stateNavigator: AsyncStateNavigator } };
 
 class NavigationHandler extends Component<{ stateNavigator: StateNavigator, children: any }, NavigationHandlerState> {
@@ -13,6 +14,7 @@ class NavigationHandler extends Component<{ stateNavigator: StateNavigator, chil
         const asyncNavigator = new AsyncStateNavigator(this, stateNavigator, stateNavigator.stateContext);
         this.state = { context: { oldState, state, data, asyncData, stateNavigator: asyncNavigator } };
         this.onNavigate = this.onNavigate.bind(this);
+        this.refetch = this.refetch.bind(this);
     }
 
     componentDidMount() {
@@ -33,10 +35,18 @@ class NavigationHandler extends Component<{ stateNavigator: StateNavigator, chil
         }
     }
 
+    refetch() {
+        startTransition(() => {
+            this.setState({ context: { ...this.state.context, ignoreCache: true } });
+        });
+    }
+
     render() {
         return (
             <NavigationContext.Provider value={this.state.context}>
-                {this.props.children}
+                <SceneRSCContext.Provider value={this.refetch}>
+                    {this.props.children}
+                </SceneRSCContext.Provider>
             </NavigationContext.Provider>
         );
     }

NavigationReact/src/SceneRSCContext.ts

@@ -0,0 +1,3 @@
+import { createContext } from 'react';
+
+export default createContext<() => void>(null);

NavigationReact/src/useRefetch.ts

@@ -1,10 +1,12 @@
 import { useContext } from 'react';
 import { StateContext } from 'navigation';
 import RSCContext from './RSCContext.js';
+import SceneRSCContext from './SceneRSCContext.js';
 
 const useRefetch = (refetch?: string[] | ((stateContext: StateContext) => boolean) | null) => {
     const {setRefetch, refetcher} = useContext(RSCContext);
+    const sceneRefetcher = useContext(SceneRSCContext);
     setRefetch(refetch);
-    return refetcher;
+    return (scene = false) => !scene ? refetcher() : sceneRefetcher();
 }
 export default useRefetch;