diff --git a/packages/react-native/ReactCommon/react/nativemodule/core/ReactCommon/TurboModule.h b/packages/react-native/ReactCommon/react/nativemodule/core/ReactCommon/TurboModule.h index fc622dfb526..adf0f2dc56d 100644 --- a/packages/react-native/ReactCommon/react/nativemodule/core/ReactCommon/TurboModule.h +++ b/packages/react-native/ReactCommon/react/nativemodule/core/ReactCommon/TurboModule.h @@ -145,5 +145,7 @@ class JSI_EXPORT TurboModule : public jsi::HostObject { * given a name. */ using TurboModuleProviderFunctionType = std::function(const std::string &name)>; +using TurboModuleProviderFunctionTypeWithRuntime = + std::function(jsi::Runtime &runtime, const std::string &name)>; } // namespace facebook::react diff --git a/packages/react-native/ReactCommon/react/nativemodule/core/ReactCommon/TurboModuleBinding.cpp b/packages/react-native/ReactCommon/react/nativemodule/core/ReactCommon/TurboModuleBinding.cpp index 195811f9876..4d02c5a60f5 100644 --- a/packages/react-native/ReactCommon/react/nativemodule/core/ReactCommon/TurboModuleBinding.cpp +++ b/packages/react-native/ReactCommon/react/nativemodule/core/ReactCommon/TurboModuleBinding.cpp @@ -19,24 +19,25 @@ namespace facebook::react { class BridgelessNativeModuleProxy : public jsi::HostObject { TurboModuleBinding turboBinding_; - std::unique_ptr legacyBinding_; + std::optional legacyBinding_; public: BridgelessNativeModuleProxy( jsi::Runtime& runtime, - TurboModuleProviderFunctionType&& moduleProvider, - TurboModuleProviderFunctionType&& legacyModuleProvider, + TurboModuleProviderFunctionTypeWithRuntime&& moduleProvider, + TurboModuleProviderFunctionTypeWithRuntime&& legacyModuleProvider, std::shared_ptr longLivedObjectCollection) : turboBinding_( runtime, std::move(moduleProvider), longLivedObjectCollection), legacyBinding_( - legacyModuleProvider ? std::make_unique( - runtime, - std::move(legacyModuleProvider), - longLivedObjectCollection) - : nullptr) {} + legacyModuleProvider + ? std::make_optional(TurboModuleBinding( + runtime, + std::move(legacyModuleProvider), + longLivedObjectCollection)) + : std::nullopt) {} jsi::Value get(jsi::Runtime& runtime, const jsi::PropNameID& name) override { /** @@ -88,7 +89,7 @@ class BridgelessNativeModuleProxy : public jsi::HostObject { TurboModuleBinding::TurboModuleBinding( jsi::Runtime& runtime, - TurboModuleProviderFunctionType&& moduleProvider, + TurboModuleProviderFunctionTypeWithRuntime&& moduleProvider, std::shared_ptr longLivedObjectCollection) : runtime_(runtime), moduleProvider_(std::move(moduleProvider)), @@ -99,6 +100,25 @@ void TurboModuleBinding::install( TurboModuleProviderFunctionType&& moduleProvider, TurboModuleProviderFunctionType&& legacyModuleProvider, std::shared_ptr longLivedObjectCollection) { + install( + runtime, + [moduleProvider = std::move(moduleProvider)]( + jsi::Runtime& runtime, const std::string& name) { + return moduleProvider(name); + }, + legacyModuleProvider == nullptr + ? (TurboModuleProviderFunctionTypeWithRuntime) nullptr + : [legacyModuleProvider = std::move(legacyModuleProvider)]( + jsi::Runtime& runtime, + const std::string& name) { return legacyModuleProvider(name); }, + longLivedObjectCollection); +} + +void TurboModuleBinding::install( + jsi::Runtime& runtime, + TurboModuleProviderFunctionTypeWithRuntime&& moduleProvider, + TurboModuleProviderFunctionTypeWithRuntime&& legacyModuleProvider, + std::shared_ptr longLivedObjectCollection) { // TODO(T208105802): We can get this information from the native side! auto isBridgeless = runtime.global().hasProperty(runtime, "RN$Bridgeless"); @@ -153,7 +173,7 @@ jsi::Value TurboModuleBinding::getModule( std::shared_ptr module; { TraceSection s("TurboModuleBinding::moduleProvider", "module", moduleName); - module = moduleProvider_(moduleName); + module = moduleProvider_(runtime, moduleName); } if (module) { TurboModuleWithJSIBindings::installJSIBindings(module, runtime); diff --git a/packages/react-native/ReactCommon/react/nativemodule/core/ReactCommon/TurboModuleBinding.h b/packages/react-native/ReactCommon/react/nativemodule/core/ReactCommon/TurboModuleBinding.h index ef94423c04e..91fc2e7a116 100644 --- a/packages/react-native/ReactCommon/react/nativemodule/core/ReactCommon/TurboModuleBinding.h +++ b/packages/react-native/ReactCommon/react/nativemodule/core/ReactCommon/TurboModuleBinding.h @@ -33,16 +33,22 @@ class TurboModuleBinding final { TurboModuleProviderFunctionType &&legacyModuleProvider = nullptr, std::shared_ptr longLivedObjectCollection = nullptr); - TurboModuleBinding( + static void install( jsi::Runtime &runtime, - TurboModuleProviderFunctionType &&moduleProvider, - std::shared_ptr longLivedObjectCollection); + TurboModuleProviderFunctionTypeWithRuntime &&moduleProvider, + TurboModuleProviderFunctionTypeWithRuntime &&legacyModuleProvider = nullptr, + std::shared_ptr longLivedObjectCollection = nullptr); ~TurboModuleBinding(); private: friend BridgelessNativeModuleProxy; + TurboModuleBinding( + jsi::Runtime &runtime, + TurboModuleProviderFunctionTypeWithRuntime &&moduleProvider, + std::shared_ptr longLivedObjectCollection); + /** * A lookup function exposed to JS to get an instance of a TurboModule * for the given name. @@ -50,7 +56,7 @@ class TurboModuleBinding final { jsi::Value getModule(jsi::Runtime &runtime, const std::string &moduleName) const; jsi::Runtime &runtime_; - TurboModuleProviderFunctionType moduleProvider_; + TurboModuleProviderFunctionTypeWithRuntime moduleProvider_; std::shared_ptr longLivedObjectCollection_; };