123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191 |
- /*
- * Copyright (c) Facebook, Inc. and its affiliates.
- *
- * This source code is licensed under the MIT license found in the
- * LICENSE file in the root directory of this source tree.
- */
- #import "RCTSurfacePresenterBridgeAdapter.h"
- #import <cxxreact/MessageQueueThread.h>
- #import <jsi/jsi.h>
- #import <React/RCTAssert.h>
- #import <React/RCTBridge+Private.h>
- #import <React/RCTImageLoader.h>
- #import <React/RCTImageLoaderWithAttributionProtocol.h>
- #import <React/RCTSurfacePresenter.h>
- #import <React/RCTSurfacePresenterStub.h>
- #import <react/utils/ContextContainer.h>
- #import <react/utils/ManagedObjectWrapper.h>
- #import <react/utils/RuntimeExecutor.h>
- using namespace facebook::react;
- @interface RCTBridge ()
- - (std::shared_ptr<facebook::react::MessageQueueThread>)jsMessageThread;
- - (void)invokeAsync:(std::function<void()> &&)func;
- @end
- static ContextContainer::Shared RCTContextContainerFromBridge(RCTBridge *bridge)
- {
- auto contextContainer = std::make_shared<ContextContainer const>();
- RCTImageLoader *imageLoader = RCTTurboModuleEnabled()
- ? [bridge moduleForName:@"RCTImageLoader" lazilyLoadIfNecessary:YES]
- : [bridge moduleForClass:[RCTImageLoader class]];
- contextContainer->insert("Bridge", wrapManagedObjectWeakly(bridge));
- contextContainer->insert("RCTImageLoader", wrapManagedObject((id<RCTImageLoaderWithAttributionProtocol>)imageLoader));
- return contextContainer;
- }
- static RuntimeExecutor RCTRuntimeExecutorFromBridge(RCTBridge *bridge)
- {
- auto bridgeWeakWrapper = wrapManagedObjectWeakly([bridge batchedBridge] ?: bridge);
- RuntimeExecutor runtimeExecutor = [bridgeWeakWrapper](
- std::function<void(facebook::jsi::Runtime & runtime)> &&callback) {
- [unwrapManagedObjectWeakly(bridgeWeakWrapper) invokeAsync:[bridgeWeakWrapper, callback = std::move(callback)]() {
- RCTCxxBridge *batchedBridge = (RCTCxxBridge *)unwrapManagedObjectWeakly(bridgeWeakWrapper);
- if (!batchedBridge) {
- return;
- }
- auto runtime = (facebook::jsi::Runtime *)(batchedBridge.runtime);
- if (!runtime) {
- return;
- }
- callback(*runtime);
- }];
- };
- return runtimeExecutor;
- }
- @implementation RCTSurfacePresenterBridgeAdapter {
- RCTSurfacePresenter *_Nullable _surfacePresenter;
- __weak RCTBridge *_bridge;
- __weak RCTBridge *_batchedBridge;
- }
- - (instancetype)initWithBridge:(RCTBridge *)bridge contextContainer:(ContextContainer::Shared)contextContainer
- {
- if (self = [super init]) {
- contextContainer->update(*RCTContextContainerFromBridge(bridge));
- _surfacePresenter = [[RCTSurfacePresenter alloc] initWithContextContainer:contextContainer
- runtimeExecutor:RCTRuntimeExecutorFromBridge(bridge)];
- _bridge = bridge;
- _batchedBridge = [_bridge batchedBridge] ?: _bridge;
- [self _updateSurfacePresenter];
- [self _addBridgeObservers:_bridge];
- }
- return self;
- }
- - (void)dealloc
- {
- [_surfacePresenter suspend];
- }
- - (RCTBridge *)bridge
- {
- return _bridge;
- }
- - (void)setBridge:(RCTBridge *)bridge
- {
- if (bridge == _bridge) {
- return;
- }
- [self _removeBridgeObservers:_bridge];
- [_surfacePresenter suspend];
- _bridge = bridge;
- _batchedBridge = [_bridge batchedBridge] ?: _bridge;
- [self _updateSurfacePresenter];
- [self _addBridgeObservers:_bridge];
- [_surfacePresenter resume];
- }
- - (void)_updateSurfacePresenter
- {
- _surfacePresenter.runtimeExecutor = RCTRuntimeExecutorFromBridge(_bridge);
- _surfacePresenter.contextContainer->update(*RCTContextContainerFromBridge(_bridge));
- [_bridge setSurfacePresenter:_surfacePresenter];
- [_batchedBridge setSurfacePresenter:_surfacePresenter];
- }
- - (void)_addBridgeObservers:(RCTBridge *)bridge
- {
- if (!bridge) {
- return;
- }
- [[NSNotificationCenter defaultCenter] addObserver:self
- selector:@selector(handleBridgeWillReloadNotification:)
- name:RCTBridgeWillReloadNotification
- object:bridge];
- [[NSNotificationCenter defaultCenter] addObserver:self
- selector:@selector(handleJavaScriptDidLoadNotification:)
- name:RCTJavaScriptDidLoadNotification
- object:bridge];
- [[NSNotificationCenter defaultCenter] addObserver:self
- selector:@selector(handleBridgeWillBeInvalidatedNotification:)
- name:RCTBridgeWillBeInvalidatedNotification
- object:bridge];
- }
- - (void)_removeBridgeObservers:(RCTBridge *)bridge
- {
- if (!bridge) {
- return;
- }
- [[NSNotificationCenter defaultCenter] removeObserver:self name:RCTBridgeWillReloadNotification object:bridge];
- [[NSNotificationCenter defaultCenter] removeObserver:self name:RCTJavaScriptDidLoadNotification object:bridge];
- [[NSNotificationCenter defaultCenter] removeObserver:self name:RCTBridgeWillBeInvalidatedNotification object:bridge];
- }
- #pragma mark - Bridge events
- - (void)handleBridgeWillReloadNotification:(NSNotification *)notification
- {
- [_surfacePresenter suspend];
- }
- - (void)handleBridgeWillBeInvalidatedNotification:(NSNotification *)notification
- {
- [_surfacePresenter suspend];
- }
- - (void)handleJavaScriptDidLoadNotification:(NSNotification *)notification
- {
- RCTBridge *bridge = notification.userInfo[@"bridge"];
- if (bridge == _batchedBridge) {
- // Nothing really changed.
- return;
- }
- _batchedBridge = bridge;
- _batchedBridge.surfacePresenter = _surfacePresenter;
- [self _updateSurfacePresenter];
- [_surfacePresenter resume];
- }
- @end
|