Adding getFabricUIManager() APIs to ReactContext (facebook#41506)

Summary:
Pull Request resolved: facebook#41506

Adding APIs for `getFabricUIManager()` to ReactContext and it's subclasses. This will replace the `getJSIModule()` post JSI module deletion.

Thereby replacing the callsite to context.getFabricUIManager() in UIManagerHelper.

NOTE: This still has fallback to getJSIModule() in case the UIManagerProvider is not set

Changelog:
[Internal] internal

Reviewed By: philIip

Differential Revision: D50926218

fbshipit-source-id: f311affb0f82895b254fd4664aa8ea23ab31bac0

packages/react-native/ReactAndroid/src/main/java/com/facebook/react/bridge/ReactContext.java

@@ -26,6 +26,7 @@
 import com.facebook.react.bridge.queue.ReactQueueConfiguration;
 import com.facebook.react.common.LifecycleState;
 import com.facebook.react.common.ReactConstants;
+import com.facebook.react.common.annotations.DeprecatedInNewArchitecture;
 import java.lang.ref.WeakReference;
 import java.util.Collection;
 import java.util.concurrent.CopyOnWriteArraySet;
@@ -554,6 +555,25 @@ public boolean isBridgeless() {
     return mCatalystInstance.getJSIModule(moduleType);
   }
 
+  @DeprecatedInNewArchitecture(
+      message =
+          "This method will be deprecated later as part of Stable APIs with bridge removal and not encouraged usage.")
+  /**
+   * Get the UIManager for Fabric from the CatalystInstance.
+   *
+   * @return The UIManager when CatalystInstance is active.
+   */
+  public @Nullable UIManager getFabricUIManager() {
+    if (!hasActiveReactInstance()) {
+      throw new IllegalStateException(
+          "Unable to retrieve a UIManager if CatalystInstance is not active.");
+    }
+    UIManager uiManager = mCatalystInstance.getFabricUIManager();
+    return uiManager != null
+        ? uiManager
+        : (UIManager) mCatalystInstance.getJSIModule(JSIModuleType.UIManager);
+  }
+
   /**
    * Get the sourceURL for the JS bundle from the CatalystInstance. This method is needed for
    * compatibility with bridgeless mode, which has no CatalystInstance.

packages/react-native/ReactAndroid/src/main/java/com/facebook/react/runtime/BridgelessReactContext.java

@@ -21,6 +21,7 @@
 import com.facebook.react.bridge.ReactApplicationContext;
 import com.facebook.react.bridge.ReactNoCrashBridgeNotAllowedSoftException;
 import com.facebook.react.bridge.ReactSoftExceptionLogger;
+import com.facebook.react.bridge.UIManager;
 import com.facebook.react.bridge.WritableNativeArray;
 import com.facebook.react.config.ReactFeatureFlags;
 import com.facebook.react.devsupport.interfaces.DevSupportManager;
@@ -84,6 +85,11 @@ public void setSourceURL(String sourceURL) {
             + moduleType.name());
   }
 
+  @Override
+  public @Nullable UIManager getFabricUIManager() {
+    return mReactHost.getUIManager();
+  }
+
   @Override
   public CatalystInstance getCatalystInstance() {
     ReactSoftExceptionLogger.logSoftExceptionVerbose(

packages/react-native/ReactAndroid/src/main/java/com/facebook/react/uimanager/ThemedReactContext.java

@@ -15,6 +15,7 @@
 import com.facebook.react.bridge.LifecycleEventListener;
 import com.facebook.react.bridge.ReactApplicationContext;
 import com.facebook.react.bridge.ReactContext;
+import com.facebook.react.bridge.UIManager;
 
 /**
  * Wraps {@link ReactContext} with the base {@link Context} passed into the constructor. It provides
@@ -116,4 +117,12 @@ public JSIModule getJSIModule(JSIModuleType moduleType) {
     }
     return super.getJSIModule(moduleType);
   }
+
+  @Override
+  public UIManager getFabricUIManager() {
+    if (isBridgeless()) {
+      return mReactApplicationContext.getFabricUIManager();
+    }
+    return super.getFabricUIManager();
+  }
 }

packages/react-native/ReactAndroid/src/main/java/com/facebook/react/uimanager/UIManagerHelper.java

@@ -17,7 +17,6 @@
 import androidx.annotation.Nullable;
 import androidx.core.view.ViewCompat;
 import com.facebook.react.bridge.CatalystInstance;
-import com.facebook.react.bridge.JSIModuleType;
 import com.facebook.react.bridge.ReactContext;
 import com.facebook.react.bridge.ReactNoCrashSoftException;
 import com.facebook.react.bridge.ReactSoftExceptionLogger;
@@ -53,7 +52,7 @@ private static UIManager getUIManager(
       @UIManagerType int uiManagerType,
       boolean returnNullIfCatalystIsInactive) {
     if (context.isBridgeless()) {
-      @Nullable UIManager uiManager = (UIManager) context.getJSIModule(JSIModuleType.UIManager);
+      @Nullable UIManager uiManager = context.getFabricUIManager();
       if (uiManager == null) {
         ReactSoftExceptionLogger.logSoftException(
             TAG,
@@ -85,7 +84,7 @@ private static UIManager getUIManager(
     CatalystInstance catalystInstance = context.getCatalystInstance();
     try {
       return uiManagerType == FABRIC
-          ? (UIManager) catalystInstance.getJSIModule(JSIModuleType.UIManager)
+          ? context.getFabricUIManager()
           : catalystInstance.getNativeModule(UIManagerModule.class);
     } catch (IllegalArgumentException ex) {
       // TODO T67518514 Clean this up once we migrate everything over to bridgeless mode

packages/react-native/ReactAndroid/src/test/java/com/facebook/react/animated/NativeAnimatedNodeTraversalTest.kt

@@ -83,6 +83,7 @@ class NativeAnimatedNodeTraversalTest {
     whenever(catalystInstanceMock.getJSIModule(any(JSIModuleType::class.java))).thenAnswer {
       uiManagerMock
     }
+    whenever(reactApplicationContextMock.getFabricUIManager()).thenAnswer { uiManagerMock }
     whenever(catalystInstanceMock.getNativeModule(UIManagerModule::class.java)).thenAnswer {
       uiManagerMock
     }