Expose way for native modules to modify JSC context

React/Executors/RCTJSCExecutor.h

@@ -11,6 +11,15 @@
 
 #import "RCTJavaScriptExecutor.h"
 
+/**
+ * This notification fires on the JS thread immediately after a `JSContext`
+ * is fully initialized, but before the JS bundle has been loaded. The object
+ * of this notification is the `JSContext`. Native modules should listen for
+ * notification only if they need to install custom functionality into the
+ * context. Note that this notification won't fire when debugging in Chrome.
+ */
+RCT_EXTERN NSString *const RCTJavaScriptContextCreatedNotification;
+
 /**
  * Uses a JavaScriptCore context as the execution engine.
  */

React/Executors/RCTJSCExecutor.m

@@ -26,6 +26,8 @@
 #import "RCTRedBox.h"
 #import "RCTSourceCode.h"
 
+NSString *const RCTJavaScriptContextCreatedNotification = @"RCTJavaScriptContextCreatedNotification";
+
 static NSString *const RCTJSCProfilerEnabledDefaultsKey = @"RCTJSCProfilerEnabled";
 
 @interface RCTJavaScriptContext : NSObject <RCTInvalidating>
@@ -314,7 +316,16 @@ - (void)setUp
   }];
 
   [self executeBlockOnJavaScriptQueue:^{
-    RCTInstallJSCProfiler(_bridge, self.context.ctx);
+    RCTJSCExecutor *strongSelf = weakSelf;
+    if (!strongSelf.valid) {
+      return;
+    }
+
+    JSContext *context = strongSelf.context.context;
+    RCTInstallJSCProfiler(_bridge, context.JSGlobalContextRef);
+
+    [[NSNotificationCenter defaultCenter] postNotificationName:RCTJavaScriptContextCreatedNotification
+                                                        object:context];
   }];
 
   for (NSString *event in @[RCTProfileDidStartProfiling, RCTProfileDidEndProfiling]) {

ReactAndroid/src/main/java/com/facebook/react/bridge/BaseJavaModule.java

@@ -332,6 +332,11 @@ public boolean canOverrideExistingModule() {
     return false;
   }
 
+  @Override
+  public void onReactBridgeInitialized(ReactBridge bridge) {
+    // do nothing
+  }
+
   @Override
   public void onCatalystInstanceDestroy() {
     // do nothing

ReactAndroid/src/main/java/com/facebook/react/bridge/CatalystInstanceImpl.java

@@ -127,6 +127,7 @@ private ReactBridge initializeBridge(
       Systrace.endSection(Systrace.TRACE_TAG_REACT_JAVA_BRIDGE);
     }
 
+    mJavaRegistry.notifyReactBridgeInitialized(bridge);
     return bridge;
   }
 

ReactAndroid/src/main/java/com/facebook/react/bridge/NativeModule.java

@@ -59,6 +59,15 @@ interface NativeMethod {
    */
   boolean canOverrideExistingModule();
 
+  /**
+   * Called on the JS thread after a ReactBridge has been created. This is useful for native modules
+   * that need to do any setup before the JS bundle has been loaded. An example of this would be
+   * installing custom functionality into the JavaScriptCore context.
+   *
+   * @param bridge the ReactBridge instance that has just been created
+   */
+  void onReactBridgeInitialized(ReactBridge bridge);
+
   /**
    * Called before {CatalystInstance#onHostDestroy}
    */

ReactAndroid/src/main/java/com/facebook/react/bridge/NativeModuleRegistry.java

@@ -114,6 +114,19 @@ private NativeModuleRegistry(
     }
   }
 
+  /* package */ void notifyReactBridgeInitialized(ReactBridge bridge) {
+    Systrace.beginSection(
+        Systrace.TRACE_TAG_REACT_JAVA_BRIDGE,
+        "NativeModuleRegistry_notifyReactBridgeInitialized");
+    try {
+      for (NativeModule nativeModule : mModuleInstances.values()) {
+        nativeModule.onReactBridgeInitialized(bridge);
+      }
+    } finally {
+      Systrace.endSection(Systrace.TRACE_TAG_REACT_JAVA_BRIDGE);
+    }
+  }
+
   public void onBatchComplete() {
     for (int i = 0; i < mBatchCompleteListenerModules.size(); i++) {
       mBatchCompleteListenerModules.get(i).onBatchComplete();

ReactAndroid/src/main/java/com/facebook/react/bridge/ReactBridge.java

@@ -82,6 +82,7 @@ private native void initialize(
   public native void callFunction(int moduleId, int methodId, NativeArray arguments);
   public native void invokeCallback(int callbackID, NativeArray arguments);
   public native void setGlobalVariable(String propertyName, String jsonEncodedArgument);
+  public native long getJavaScriptContext();
   public native boolean supportsProfiling();
   public native void startProfiler(String title);
   public native void stopProfiler(String title, String filename);

ReactAndroid/src/main/jni/react/Bridge.cpp

@@ -77,6 +77,10 @@ void Bridge::setGlobalVariable(const std::string& propName, const std::string& j
   m_jsExecutor->setGlobalVariable(propName, jsonValue);
 }
 
+void* Bridge::getJavaScriptContext() {
+  return m_jsExecutor->getJavaScriptContext();
+}
+
 bool Bridge::supportsProfiling() {
   return m_jsExecutor->supportsProfiling();
 }

ReactAndroid/src/main/jni/react/Bridge.h

@@ -60,6 +60,7 @@ class Bridge : public Countable {
     const std::string& startupCode,
     const std::string& sourceURL);
   void setGlobalVariable(const std::string& propName, const std::string& jsonValue);
+  void* getJavaScriptContext();
   bool supportsProfiling();
   void startProfiler(const std::string& title);
   void stopProfiler(const std::string& title, const std::string& filename);

ReactAndroid/src/main/jni/react/Executor.h

@@ -65,6 +65,9 @@ class JSExecutor {
   virtual void setGlobalVariable(
     const std::string& propName,
     const std::string& jsonValue) = 0;
+  virtual void* getJavaScriptContext() {
+    return nullptr;
+  };
   virtual bool supportsProfiling() {
     return false;
   };

ReactAndroid/src/main/jni/react/JSCExecutor.cpp

@@ -237,6 +237,10 @@ void JSCExecutor::setGlobalVariable(const std::string& propName, const std::stri
   JSObjectSetProperty(m_context, globalObject, jsPropertyName, valueToInject, 0, NULL);
 }
 
+void* JSCExecutor::getJavaScriptContext() {
+  return m_context;
+}
+
 bool JSCExecutor::supportsProfiling() {
   #ifdef WITH_FBSYSTRACE
   return true;

ReactAndroid/src/main/jni/react/JSCExecutor.h

@@ -46,6 +46,7 @@ class JSCExecutor : public JSExecutor, public JSCWebWorkerOwner {
   virtual void setGlobalVariable(
     const std::string& propName,
     const std::string& jsonValue) override;
+  virtual void* getJavaScriptContext() override;
   virtual bool supportsProfiling() override;
   virtual void startProfiler(const std::string &titleString) override;
   virtual void stopProfiler(const std::string &titleString, const std::string &filename) override;

ReactAndroid/src/main/jni/react/jni/OnLoad.cpp

@@ -745,6 +745,11 @@ static void setGlobalVariable(JNIEnv* env, jobject obj, jstring propName, jstrin
   bridge->setGlobalVariable(fromJString(env, propName), fromJString(env, jsonValue));
 }
 
+static jlong getJavaScriptContext(JNIEnv *env, jobject obj) {
+  auto bridge = extractRefPtr<Bridge>(env, obj);
+  return (uintptr_t) bridge->getJavaScriptContext();
+}
+
 static jboolean supportsProfiling(JNIEnv* env, jobject obj) {
   auto bridge = extractRefPtr<Bridge>(env, obj);
   return bridge->supportsProfiling() ? JNI_TRUE : JNI_FALSE;
@@ -880,12 +885,12 @@ extern "C" JNIEXPORT jint JNI_OnLoad(JavaVM* vm, void* reserved) {
         makeNativeMethod("callFunction", bridge::callFunction),
         makeNativeMethod("invokeCallback", bridge::invokeCallback),
         makeNativeMethod("setGlobalVariable", bridge::setGlobalVariable),
+        makeNativeMethod("getJavaScriptContext", bridge::getJavaScriptContext),
         makeNativeMethod("supportsProfiling", bridge::supportsProfiling),
         makeNativeMethod("startProfiler", bridge::startProfiler),
         makeNativeMethod("stopProfiler", bridge::stopProfiler),
         makeNativeMethod("handleMemoryPressureModerate", bridge::handleMemoryPressureModerate),
         makeNativeMethod("handleMemoryPressureCritical", bridge::handleMemoryPressureCritical),
-
     });
 
     jclass nativeRunnableClass = env->FindClass("com/facebook/react/bridge/queue/NativeRunnableDeprecated");