RCTSurfacePresenterBridgeAdapter.mm 5.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191
  1. /*
  2. * Copyright (c) Facebook, Inc. and its affiliates.
  3. *
  4. * This source code is licensed under the MIT license found in the
  5. * LICENSE file in the root directory of this source tree.
  6. */
  7. #import "RCTSurfacePresenterBridgeAdapter.h"
  8. #import <cxxreact/MessageQueueThread.h>
  9. #import <jsi/jsi.h>
  10. #import <React/RCTAssert.h>
  11. #import <React/RCTBridge+Private.h>
  12. #import <React/RCTImageLoader.h>
  13. #import <React/RCTImageLoaderWithAttributionProtocol.h>
  14. #import <React/RCTSurfacePresenter.h>
  15. #import <React/RCTSurfacePresenterStub.h>
  16. #import <react/utils/ContextContainer.h>
  17. #import <react/utils/ManagedObjectWrapper.h>
  18. #import <react/utils/RuntimeExecutor.h>
  19. using namespace facebook::react;
  20. @interface RCTBridge ()
  21. - (std::shared_ptr<facebook::react::MessageQueueThread>)jsMessageThread;
  22. - (void)invokeAsync:(std::function<void()> &&)func;
  23. @end
  24. static ContextContainer::Shared RCTContextContainerFromBridge(RCTBridge *bridge)
  25. {
  26. auto contextContainer = std::make_shared<ContextContainer const>();
  27. RCTImageLoader *imageLoader = RCTTurboModuleEnabled()
  28. ? [bridge moduleForName:@"RCTImageLoader" lazilyLoadIfNecessary:YES]
  29. : [bridge moduleForClass:[RCTImageLoader class]];
  30. contextContainer->insert("Bridge", wrapManagedObjectWeakly(bridge));
  31. contextContainer->insert("RCTImageLoader", wrapManagedObject((id<RCTImageLoaderWithAttributionProtocol>)imageLoader));
  32. return contextContainer;
  33. }
  34. static RuntimeExecutor RCTRuntimeExecutorFromBridge(RCTBridge *bridge)
  35. {
  36. auto bridgeWeakWrapper = wrapManagedObjectWeakly([bridge batchedBridge] ?: bridge);
  37. RuntimeExecutor runtimeExecutor = [bridgeWeakWrapper](
  38. std::function<void(facebook::jsi::Runtime & runtime)> &&callback) {
  39. [unwrapManagedObjectWeakly(bridgeWeakWrapper) invokeAsync:[bridgeWeakWrapper, callback = std::move(callback)]() {
  40. RCTCxxBridge *batchedBridge = (RCTCxxBridge *)unwrapManagedObjectWeakly(bridgeWeakWrapper);
  41. if (!batchedBridge) {
  42. return;
  43. }
  44. auto runtime = (facebook::jsi::Runtime *)(batchedBridge.runtime);
  45. if (!runtime) {
  46. return;
  47. }
  48. callback(*runtime);
  49. }];
  50. };
  51. return runtimeExecutor;
  52. }
  53. @implementation RCTSurfacePresenterBridgeAdapter {
  54. RCTSurfacePresenter *_Nullable _surfacePresenter;
  55. __weak RCTBridge *_bridge;
  56. __weak RCTBridge *_batchedBridge;
  57. }
  58. - (instancetype)initWithBridge:(RCTBridge *)bridge contextContainer:(ContextContainer::Shared)contextContainer
  59. {
  60. if (self = [super init]) {
  61. contextContainer->update(*RCTContextContainerFromBridge(bridge));
  62. _surfacePresenter = [[RCTSurfacePresenter alloc] initWithContextContainer:contextContainer
  63. runtimeExecutor:RCTRuntimeExecutorFromBridge(bridge)];
  64. _bridge = bridge;
  65. _batchedBridge = [_bridge batchedBridge] ?: _bridge;
  66. [self _updateSurfacePresenter];
  67. [self _addBridgeObservers:_bridge];
  68. }
  69. return self;
  70. }
  71. - (void)dealloc
  72. {
  73. [_surfacePresenter suspend];
  74. }
  75. - (RCTBridge *)bridge
  76. {
  77. return _bridge;
  78. }
  79. - (void)setBridge:(RCTBridge *)bridge
  80. {
  81. if (bridge == _bridge) {
  82. return;
  83. }
  84. [self _removeBridgeObservers:_bridge];
  85. [_surfacePresenter suspend];
  86. _bridge = bridge;
  87. _batchedBridge = [_bridge batchedBridge] ?: _bridge;
  88. [self _updateSurfacePresenter];
  89. [self _addBridgeObservers:_bridge];
  90. [_surfacePresenter resume];
  91. }
  92. - (void)_updateSurfacePresenter
  93. {
  94. _surfacePresenter.runtimeExecutor = RCTRuntimeExecutorFromBridge(_bridge);
  95. _surfacePresenter.contextContainer->update(*RCTContextContainerFromBridge(_bridge));
  96. [_bridge setSurfacePresenter:_surfacePresenter];
  97. [_batchedBridge setSurfacePresenter:_surfacePresenter];
  98. }
  99. - (void)_addBridgeObservers:(RCTBridge *)bridge
  100. {
  101. if (!bridge) {
  102. return;
  103. }
  104. [[NSNotificationCenter defaultCenter] addObserver:self
  105. selector:@selector(handleBridgeWillReloadNotification:)
  106. name:RCTBridgeWillReloadNotification
  107. object:bridge];
  108. [[NSNotificationCenter defaultCenter] addObserver:self
  109. selector:@selector(handleJavaScriptDidLoadNotification:)
  110. name:RCTJavaScriptDidLoadNotification
  111. object:bridge];
  112. [[NSNotificationCenter defaultCenter] addObserver:self
  113. selector:@selector(handleBridgeWillBeInvalidatedNotification:)
  114. name:RCTBridgeWillBeInvalidatedNotification
  115. object:bridge];
  116. }
  117. - (void)_removeBridgeObservers:(RCTBridge *)bridge
  118. {
  119. if (!bridge) {
  120. return;
  121. }
  122. [[NSNotificationCenter defaultCenter] removeObserver:self name:RCTBridgeWillReloadNotification object:bridge];
  123. [[NSNotificationCenter defaultCenter] removeObserver:self name:RCTJavaScriptDidLoadNotification object:bridge];
  124. [[NSNotificationCenter defaultCenter] removeObserver:self name:RCTBridgeWillBeInvalidatedNotification object:bridge];
  125. }
  126. #pragma mark - Bridge events
  127. - (void)handleBridgeWillReloadNotification:(NSNotification *)notification
  128. {
  129. [_surfacePresenter suspend];
  130. }
  131. - (void)handleBridgeWillBeInvalidatedNotification:(NSNotification *)notification
  132. {
  133. [_surfacePresenter suspend];
  134. }
  135. - (void)handleJavaScriptDidLoadNotification:(NSNotification *)notification
  136. {
  137. RCTBridge *bridge = notification.userInfo[@"bridge"];
  138. if (bridge == _batchedBridge) {
  139. // Nothing really changed.
  140. return;
  141. }
  142. _batchedBridge = bridge;
  143. _batchedBridge.surfacePresenter = _surfacePresenter;
  144. [self _updateSurfacePresenter];
  145. [_surfacePresenter resume];
  146. }
  147. @end