fix: granite

Author: gronxb

packages/react-native/src/app/Granite.tsx

@@ -7,7 +7,6 @@ import { AppRoot } from './AppRoot';
 import { HostAppRoot } from './HostAppRoot';
 import { getSchemeUri } from '../constant-bridges';
 import { setupPolyfills } from '../polyfills';
-import { resolveInitialScheme } from './resolveInitialScheme';
 
 export interface GraniteProps {
   /**
@@ -32,7 +31,7 @@ export interface GraniteProps {
    * @description
    * The initial scheme of the app.
    */
-  initialScheme?: string;
+  initialScheme?: string | (() => string);
 
   /**
    * @description
@@ -64,7 +63,7 @@ const createApp = () => {
   }
 
   return {
-    registerApp(
+    registerApp( 
       AppContainer: ComponentType<PropsWithChildren<InitialProps>>,
       { appName, context, router, initialScheme, setIosSwipeGestureEnabled, getInitialUrl }: GraniteProps
     ): (initialProps: InitialProps) => JSX.Element {
@@ -73,15 +72,13 @@ const createApp = () => {
       }
 
       function Root(initialProps: InitialProps) {
+        const initialSchemeValue = (typeof initialScheme === 'function' ? initialScheme() : initialScheme) ?? getSchemeUri();
+
         return (
           <AppRoot
             container={AppContainer}
             initialProps={initialProps}
-            initialScheme={resolveInitialScheme({
-              initialPropsScheme: initialProps.scheme,
-              initialScheme,
-              getSchemeUri,
-            })}
+            initialScheme={initialSchemeValue}
             setIosSwipeGestureEnabled={setIosSwipeGestureEnabled}
             getInitialUrl={getInitialUrl}
             appName={appName}