iOS: allow getting an instance of a js-bound module via the bridge

Reviewed By: sebmarkbage Differential Revision: D6982785 fbshipit-source-id: 7bbcc5416e1d1a3a577328349a7c18af5c0f8577
swissmanu · Feb 14, 2018 · 5f48bd8 · 5f48bd8
1 parent 1aeb925
commit 5f48bd8
Show file tree

Hide file tree

Showing 6 changed files with 42 additions and 0 deletions.
diff --git a/React/Base/RCTBridge.h b/React/Base/RCTBridge.h
@@ -168,6 +168,11 @@ RCT_EXTERN NSString *RCTBridgeModuleNameForClass(Class bridgeModuleClass);
  */
 - (BOOL)moduleIsInitialized:(Class)moduleClass;
 
+/**
+ * Retrieve an extra module that gets bound to the JS context, if any.
+ */
+- (id)jsBoundExtraModuleForClass:(Class)moduleClass;
+
 /**
  * All registered bridge module classes.
  */

diff --git a/React/Base/RCTBridge.m b/React/Base/RCTBridge.m
@@ -251,6 +251,11 @@ - (BOOL)moduleIsInitialized:(Class)moduleClass
   return [self.batchedBridge moduleIsInitialized:moduleClass];
 }
 
+- (id)jsBoundExtraModuleForClass:(Class)moduleClass
+{
+  return [self.batchedBridge jsBoundExtraModuleForClass:moduleClass];
+}
+
 - (void)reload
 {
   #if RCT_ENABLE_INSPECTOR

diff --git a/React/CxxBridge/RCTCxxBridge.mm b/React/CxxBridge/RCTCxxBridge.mm
@@ -454,6 +454,18 @@ - (BOOL)moduleIsInitialized:(Class)moduleClass
   return _moduleDataByName[RCTBridgeModuleNameForClass(moduleClass)].hasInstance;
 }
 
+- (id)jsBoundExtraModuleForClass:(Class)moduleClass
+{
+  if ([self.delegate conformsToProtocol:@protocol(RCTCxxBridgeDelegate)]) {
+    id<RCTCxxBridgeDelegate> cxxDelegate = (id<RCTCxxBridgeDelegate>) self.delegate;
+    if ([cxxDelegate respondsToSelector:@selector(jsBoundExtraModuleForClass:)]) {
+      return [cxxDelegate jsBoundExtraModuleForClass:moduleClass];
+    }
+  }
+
+  return nil;
+}
+
 - (std::shared_ptr<ModuleRegistry>)_buildModuleRegistry
 {
   if (!self.valid) {

diff --git a/React/CxxBridge/RCTCxxBridgeDelegate.h b/React/CxxBridge/RCTCxxBridgeDelegate.h
@@ -40,4 +40,9 @@ class JSExecutorFactory;
  */
 - (void)installExtraJSBinding:(JSGlobalContextRef)jsContextRef;
 
+/**
+ * Experimental: Get the instance of the extra module/class which gets bound via `installExtraJSBinding:`
+ */
+- (id)jsBoundExtraModuleForClass:(Class)moduleClass;
+
 @end
diff --git a/React/Fabric/RCTFabricUIManager.h b/React/Fabric/RCTFabricUIManager.h
@@ -42,3 +42,9 @@
             childSet:(NSArray<RCTShadowView *> *)childSet;
 
 @end
+
+@interface RCTBridge (RCTFabricUIManager)
+
+@property (nonatomic, readonly) RCTFabricUIManager *fabricUIManager;
+
+@end
diff --git a/React/Fabric/RCTFabricUIManager.mm b/React/Fabric/RCTFabricUIManager.mm
@@ -72,3 +72,12 @@ - (void)completeRoot:(nonnull NSNumber *)rootTag
 }
 
 @end
+
+@implementation RCTBridge (RCTFabricUIManager)
+
+- (RCTFabricUIManager *)fabricUIManager
+{
+  return [self jsBoundExtraModuleForClass:[RCTFabricUIManager class]];
+}
+
+@end